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
|
#include "smanager.h"
#include <config.h>
#include <ctype.h>
#include <netdb.h>
#include <sys/utsname.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <strings.h>
SocketManager::SocketManager()
{
max_fd = 0;
FD_ZERO(&read_set);
FD_ZERO(&write_set);
nbWriteable = 0;
}
SocketManager::~SocketManager()
{
clean();
}
void SocketManager::clean()
{
for(uint i=0; i<sockets.size(); i++) delete sockets[i];
sockets.resize(0);
}
int SocketManager::find(int fd)
{
for(uint i=0; i<sockets.size(); i++)
if ( sockets[i]->fd()==fd ) return i;
return -1;
}
uint SocketManager::append(Socket *socket, SocketProperty sp)
{
uint s = sockets.size();
sockets.resize(s+1);
sockets[s] = socket;
max_fd = TQMAX(max_fd, socket->fd());
if ( sp==ReadWrite || sp==ReadOnly ) FD_SET(socket->fd(), &read_set);
if ( sp==ReadWrite || sp==WriteOnly ) {
nbWriteable++;
FD_SET(socket->fd(), &write_set);
}
return s;
}
void SocketManager::remove(uint i, bool del)
{
Socket *so = sockets[i];
uint s = sockets.size()-1;
for(uint j=i; j<s; j++) sockets[j] = sockets[j+1];
sockets.resize(s);
max_fd = 0;
for(uint j=0; j<s; j++)
max_fd = TQMAX(max_fd, sockets[j]->fd());
int fd = so->fd();
if ( FD_ISSET(fd, &read_set) ) FD_CLR(fd, &read_set);
if ( FD_ISSET(fd, &write_set) ) {
nbWriteable--;
FD_CLR(fd, &write_set);
}
if (del) delete so;
}
bool SocketManager::canWriteAll(uint sec, uint usec)
{
write_tmp = write_set;
tv.tv_sec = sec;
tv.tv_usec = usec;
return ( select(max_fd+1, 0, &write_tmp, 0, &tv)==(int)nbWriteable );
}
bool SocketManager::canWrite(uint i, uint sec, uint usec)
{
int fd = sockets[i]->fd();
FD_ZERO(&write_tmp);
FD_SET(fd, &write_tmp);
tv.tv_sec = sec;
tv.tv_usec = usec;
return ( select(fd+1, 0, &write_tmp, 0, &tv)==1 );
}
bool SocketManager::checkPendingData(uint sec, uint usec)
{
read_tmp = read_set;
tv.tv_sec = sec;
tv.tv_usec = usec;
return ( select(max_fd+1, &read_tmp, 0, 0, &tv)!=-1 );
}
bool SocketManager::dataPending(uint i)
{
int fd = sockets[i]->fd();
return FD_ISSET(fd, &read_tmp);
}
bool SocketManager::writeCommon(uint i)
{
return sockets[i]->write(writing.buffer());
}
|
