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
148
149
150
151
|
/************************************************************************************
PublicHeader: n/a
Filename : OVR_Win32_Socket.h
Content : Windows-specific socket-based networking implementation
Created : June 10, 2014
Authors : Kevin Jenkins
Copyright : Copyright 2014 Oculus VR, LLC All Rights reserved.
Licensed under the Oculus VR Rift SDK License Version 3.2 (the "License");
you may not use the Oculus VR Rift SDK except in compliance with the License,
which is provided at the time of installation or download, or which
otherwise accompanies this software in either electronic or hard copy form.
You may obtain a copy of the License at
http://www.oculusvr.com/licenses/LICENSE-3.2
Unless required by applicable law or agreed to in writing, the Oculus VR SDK
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
************************************************************************************/
#ifndef OVR_Win32_Socket_h
#define OVR_Win32_Socket_h
#include "OVR_Socket.h"
#include "OVR_BitStream.h"
#include <WinSock2.h>
#include <WS2tcpip.h>
#define WIN32_LEAN_AND_MEAN
#include <Windows.h>
#include <io.h>
namespace OVR { namespace Net {
//-----------------------------------------------------------------------------
// SockAddr
// Abstraction for IPV6 socket address, with various convenience functions
class SockAddr
{
public:
SockAddr();
SockAddr(SockAddr* sa);
SockAddr(sockaddr_storage* sa);
SockAddr(sockaddr_in6* sa);
SockAddr(const char* hostAddress, uint16_t port, int sockType);
public:
void Set(const sockaddr_storage* sa);
void Set(const sockaddr_in6* sa);
void Set(const char* hostAddress, uint16_t port, int sockType); // SOCK_DGRAM or SOCK_STREAM
uint16_t GetPort();
String ToString(bool writePort, char portDelineator) const;
bool IsLocalhost() const;
void Serialize(BitStream* bs);
bool Deserialize(BitStream);
bool operator==( const SockAddr& right ) const;
bool operator!=( const SockAddr& right ) const;
bool operator >( const SockAddr& right ) const;
bool operator <( const SockAddr& right ) const;
public:
sockaddr_in6 Addr6;
};
//-----------------------------------------------------------------------------
// UDP Socket
// Windows version of TCP socket
class UDPSocket : public UDPSocketBase
{
public:
UDPSocket();
virtual ~UDPSocket();
public:
virtual SocketHandle Bind(BerkleyBindParameters* pBindParameters);
virtual int Send(const void* pData, int bytes, SockAddr* address);
virtual void Poll(SocketEvent_UDP* eventHandler);
protected:
static const int RecvBufSize = 1048576;
uint8_t* RecvBuf;
virtual void OnRecv(SocketEvent_UDP* eventHandler, uint8_t* pData,
int bytesRead, SockAddr* address);
};
//-----------------------------------------------------------------------------
// TCP Socket
// Windows version of TCP socket
class TCPSocket : public TCPSocketBase
{
friend class TCPSocketPollState;
public:
TCPSocket();
TCPSocket(SocketHandle boundHandle, bool isListenSocket);
virtual ~TCPSocket();
public:
virtual SocketHandle Bind(BerkleyBindParameters* pBindParameters);
virtual int Listen();
virtual int Connect(SockAddr* address);
virtual int Send(const void* pData, int bytes);
protected:
virtual void OnRecv(SocketEvent_TCP* eventHandler, uint8_t* pData,
int bytesRead);
public:
bool IsConnecting; // Is in the process of connecting?
};
//-----------------------------------------------------------------------------
// TCPSocketPollState
// Polls multiple blocking TCP sockets at once
class TCPSocketPollState
{
fd_set readFD, exceptionFD, writeFD;
SocketHandle largestDescriptor;
public:
TCPSocketPollState();
bool IsValid() const;
void Add(TCPSocket* tcpSocket);
bool Poll(long usec = 30000, long seconds = 0);
void HandleEvent(TCPSocket* tcpSocket, SocketEvent_TCP* eventHandler);
};
}} // OVR::Net
#endif
|
