blob: d1197629b80fdcd38c080ed123642070fdc7c2e5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef REMOTING_HOST_GNUBBY_SOCKET_H_
#define REMOTING_HOST_GNUBBY_SOCKET_H_
#include <string>
#include <vector>
#include "base/callback.h"
#include "base/memory/scoped_ptr.h"
#include "base/threading/non_thread_safe.h"
namespace base {
class Timer;
} // namespace base
namespace net {
class StreamListenSocket;
} // namespace net
namespace remoting {
// Class that manages a socket used for gnubby requests.
class GnubbySocket : public base::NonThreadSafe {
public:
GnubbySocket(scoped_ptr<net::StreamListenSocket> socket,
const base::Closure& timeout_callback);
~GnubbySocket();
// Adds data to the current request.
void AddRequestData(const char* data, int data_len);
// Gets the current request data and clears it.
void GetAndClearRequestData(std::string* data_out);
// Returns true if the current request is complete.
bool IsRequestComplete() const;
// Returns true if the stated request size is larger than the allowed maximum.
bool IsRequestTooLarge() const;
// Sends response data to the socket.
void SendResponse(const std::string& data);
// Sends an SSH error code to the socket.
void SendSshError();
// Returns true if |socket| is the same one owned by this object.
bool IsSocket(net::StreamListenSocket* socket) const;
// Sets a timer for testing.
void SetTimerForTesting(scoped_ptr<base::Timer> timer);
private:
// Returns the stated request length.
size_t GetRequestLength() const;
// Returns the response length bytes.
std::string GetResponseLengthAsBytes(const std::string& response) const;
// Resets the socket activity timer.
void ResetTimer();
// The socket.
scoped_ptr<net::StreamListenSocket> socket_;
// Request data.
std::vector<char> request_data_;
// The activity timer.
scoped_ptr<base::Timer> timer_;
DISALLOW_COPY_AND_ASSIGN(GnubbySocket);
};
} // namespace remoting
#endif // REMOTING_HOST_GNUBBY_SOCKET_H_
|
