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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
// **********************************************************************
//
// Copyright (c) 2003-2014 ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************
#ifndef ICE_WS_TRANSCEIVER_I_H
#define ICE_WS_TRANSCEIVER_I_H
#include <IceWS/InstanceF.h>
#include <IceWS/Plugin.h>
#include <IceWS/Util.h>
#include <Ice/LoggerF.h>
#include <Ice/StatsF.h>
#include <Ice/Transceiver.h>
#include <Ice/Network.h>
#include <Ice/Buffer.h>
namespace IceWS
{
class ConnectorI;
class AcceptorI;
class TransceiverI : public IceInternal::Transceiver
{
public:
virtual IceInternal::NativeInfoPtr getNativeInfo();
#ifdef ICE_USE_IOCP
virtual IceInternal::AsyncInfo* getAsyncInfo(IceInternal::SocketOperation);
#endif
virtual IceInternal::SocketOperation initialize(IceInternal::Buffer&, IceInternal::Buffer&, bool&);
virtual IceInternal::SocketOperation closing(bool, const Ice::LocalException&);
virtual void close();
virtual IceInternal::SocketOperation write(IceInternal::Buffer&);
virtual IceInternal::SocketOperation read(IceInternal::Buffer&, bool&);
#ifdef ICE_USE_IOCP
virtual bool startWrite(IceInternal::Buffer&);
virtual void finishWrite(IceInternal::Buffer&);
virtual void startRead(IceInternal::Buffer&);
virtual void finishRead(IceInternal::Buffer&);
#endif
virtual std::string protocol() const;
virtual std::string toString() const;
virtual Ice::ConnectionInfoPtr getInfo() const;
virtual void checkSendSize(const IceInternal::Buffer&, size_t);
private:
TransceiverI(const InstancePtr&, const IceInternal::TransceiverPtr&, const std::string&, int, const std::string&);
TransceiverI(const InstancePtr&, const IceInternal::TransceiverPtr&);
virtual ~TransceiverI();
void handleRequest(IceInternal::Buffer&);
void handleResponse();
bool preRead(IceInternal::Buffer&);
bool postRead(IceInternal::Buffer&);
bool preWrite(IceInternal::Buffer&);
bool postWrite(IceInternal::Buffer&);
bool readBuffered(IceInternal::Buffer::Container::size_type);
void prepareWriteHeader(Ice::Byte, IceInternal::Buffer::Container::size_type);
friend class ConnectorI;
friend class AcceptorI;
const InstancePtr _instance;
const IceInternal::TransceiverPtr _delegate;
const std::string _host;
const int _port;
const std::string _resource;
const bool _incoming;
enum State
{
StateInitializeDelegate,
StateConnected,
StateUpgradeRequestPending,
StateUpgradeResponsePending,
StateOpened,
StatePingPending,
StatePongPending,
StateClosingRequestPending,
StateClosingResponsePending,
StateClosed
};
State _state;
State _nextState;
HttpParserPtr _parser;
std::string _key;
enum ReadState
{
ReadStateOpcode,
ReadStateHeader,
ReadStateControlFrame,
ReadStatePayload,
};
ReadState _readState;
IceInternal::Buffer _readBuffer;
IceInternal::Buffer::Container::iterator _readI;
const IceInternal::Buffer::Container::size_type _readBufferSize;
bool _readLastFrame;
int _readOpCode;
size_t _readHeaderLength;
size_t _readPayloadLength;
IceInternal::Buffer::Container::iterator _readStart;
unsigned char _readMask[4];
enum WriteState
{
WriteStateHeader,
WriteStatePayload,
WriteStateControlFrame,
};
WriteState _writeState;
IceInternal::Buffer _writeBuffer;
const IceInternal::Buffer::Container::size_type _writeBufferSize;
unsigned char _writeMask[4];
size_t _writePayloadLength;
bool _closingInitiator;
int _closingReason;
bool _readPending;
bool _writePending;
std::vector<Ice::Byte> _pingPayload;
};
typedef IceUtil::Handle<TransceiverI> TransceiverIPtr;
}
#endif
|
