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
|
#ifndef RUNTIME_NETWORK_SOCKETSTREAMS_H
#define RUNTIME_NETWORK_SOCKETSTREAMS_H
#if ENABLE_SOCKETS //|| UNITY_WINRT
#include "Sockets.h"
#include "Runtime/Containers/ExtendedRingbuffer.h"
#include "Runtime/Threads/Mutex.h"
#include "Runtime/Threads/Thread.h"
class SocketStream : public Socket
{
public:
SocketStream(TSocketHandle socketHandle, bool block);
virtual ~SocketStream() {};
virtual int Send(const void* data, UInt32 data_len);
virtual int Recv(void* data, UInt32 data_len);
virtual bool IsBlocking() const { return m_IsBlocking; };
virtual bool IsConnected() const { return m_IsConnected; };
virtual bool WouldBlockError() { return Socket::WouldBlockError(); }
virtual bool Shutdown();
bool SendAll(const void* data, UInt32 data_len);
bool RecvAll(void* data, UInt32 data_len);
virtual bool CanSendNonblocking(UInt32 data_len);
protected:
virtual bool Poll(UInt64 /*timeoutMS*/ = kDefaultPollTime) { return false; }
virtual void OnSocketError();
protected:
bool m_IsBlocking;
volatile bool m_IsConnected;
};
class BufferedSocketStream : public SocketStream
{
public:
BufferedSocketStream(TSocketHandle socketHandle, UInt32 sendbufferMaxSize = kDefaultBufferSize, UInt32 recvbufferMaxSize = kDefaultBufferSize);
virtual ~BufferedSocketStream() {};
virtual int Send(const void* data, UInt32 data_len);
virtual int Recv(void* data, UInt32 data_len);
virtual bool IsBlocking() const { return false; };
virtual bool IsConnected() const { return (m_IsArtificiallyConnected && m_Recvbuffer.GetAvailableSize() != 0) || m_IsConnected; };
virtual bool WouldBlockError() { return IsConnected(); }
virtual bool Shutdown();
virtual bool Poll(UInt64 timeoutMS = kDefaultPollTime);
ExtendedGrowingRingbuffer& SendBuffer() { return m_Sendbuffer; };
ExtendedGrowingRingbuffer& RecvBuffer() { return m_Recvbuffer; };
const ExtendedGrowingRingbuffer& SendBuffer() const { return m_Sendbuffer; };
const ExtendedGrowingRingbuffer& RecvBuffer() const { return m_Recvbuffer; };
virtual bool CanSendNonblocking(UInt32 data_len);
protected:
BufferedSocketStream(TSocketHandle socketHandle, UInt32 sendbufferMaxSize, UInt32 recvbufferMaxSize, bool block);
void OnSocketError();
bool FlushSendbuffer();
bool FillRecvbuffer();
private:
volatile bool m_IsArtificiallyConnected;
ExtendedGrowingRingbuffer m_Sendbuffer;
ExtendedGrowingRingbuffer m_Recvbuffer;
Mutex m_PollMutex;
};
#if SUPPORT_THREADS
class ThreadedSocketStream : public BufferedSocketStream
{
public:
ThreadedSocketStream(TSocketHandle socketHandle, UInt32 sendbufferMaxSize = kDefaultBufferSize, UInt32 recvbufferMaxSize = kDefaultBufferSize);
virtual ~ThreadedSocketStream();
virtual bool Poll(UInt64 /*timeoutMS*/ = kDefaultPollTime) { return IsConnected(); }
private:
static void* WriterLoop(void* _arg);
static void* ReaderLoop(void* _arg);
Thread m_Reader;
Thread m_Writer;
};
#endif // SUPPORT_THREADS
#endif // ENABLE_SOCKETS
#endif // RUNTIME_NETWORK_SOCKETSTREAMS_H
|
