blob: a334d64f70f47a3fac3d5a95965386b51060c71b (
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
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
|
/* Starshatter: The Open Source Project
Copyright (c) 2021-2022, Starshatter: The Open Source Project Contributors
Copyright (c) 2011-2012, Starshatter OpenSource Distribution Contributors
Copyright (c) 1997-2006, Destroyer Studios LLC.
AUTHOR: John DiCamillo
OVERVIEW
========
Network Server Pump for HTTP Server
*/
#include "NetClient.h"
#include "NetHost.h"
#include "NetLayer.h"
#include <mmsystem.h>
// +-------------------------------------------------------------------+
NetClient::NetClient(const NetAddr& server_addr)
: addr(server_addr), sock(0), delta(0), time(0), err(0)
{
}
NetClient::~NetClient()
{
delete sock;
}
// +--------------------------------------------------------------------+
bool
NetClient::Send(Text msg)
{
if (msg.length() > 0) {
if (sock)
delete sock;
sock = new NetSock(addr, true);
delta = 0;
time = timeGetTime();
if (!sock) {
err = ERR_NOBUFS;
return false;
}
err = sock->send(msg);
if (err < 0) {
err = NetLayer::GetLastError();
return false;
}
err = sock->shutdown_output();
if (err < 0) {
err = NetLayer::GetLastError();
return false;
}
return true;
}
else {
delete sock;
sock = 0;
}
return false;
}
Text
NetClient::Recv()
{
Text response;
if (sock) {
int ready = sock->select();
while (!ready && timeGetTime() - time < 2000) {
Sleep(5);
ready = sock->select();
}
if (ready) {
Text msg = sock->recv();
while (msg.length() > 0) {
response += msg;
msg = sock->recv();
}
delta = timeGetTime() - time;
}
delete sock;
sock = 0;
}
return response;
}
Text
NetClient::SendRecv(Text msg)
{
Text response;
if (msg.length() > 0 && Send(msg)) {
response = Recv();
}
return response;
}
|