]> git.vomp.tv Git - vompserver.git/blob - tcp.c
IPv6 support for main protocol
[vompserver.git] / tcp.c
1 /*
2     Copyright 2004-2005 Chris Tallon
3     Copyright 2003-2004 University Of Bradford
4
5     This file is part of VOMP.
6
7     VOMP is free software; you can redistribute it and/or modify
8     it under the terms of the GNU General Public License as published by
9     the Free Software Foundation; either version 2 of the License, or
10     (at your option) any later version.
11
12     VOMP is distributed in the hope that it will be useful,
13     but WITHOUT ANY WARRANTY; without even the implied warranty of
14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15     GNU General Public License for more details.
16
17     You should have received a copy of the GNU General Public License
18     along with VOMP; if not, write to the Free Software
19     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
20 */
21
22 #include "tcp.h"
23
24 TCP::TCP(int tsocket)
25 {
26   log = Log::getInstance();
27   sock = -1;
28   connected = 0;
29   readTimeoutEnabled = 1;
30   pthread_mutex_init(&sendLock, NULL);
31
32   if (tsocket)
33   {
34     sock = tsocket;
35     connected = 1;
36   }
37 }
38
39 TCP::~TCP()
40 {
41   if (connected) cleanup();
42 }
43
44 void TCP::cleanup()
45 {
46   close(sock);
47   sock = -1;
48   connected = 0;
49   log->log("TCP", Log::DEBUG, "TCP has closed socket");
50 }
51
52
53 void TCP::disableReadTimeout()
54 {
55   readTimeoutEnabled = 0;
56 }
57
58 int TCP::connectTo(char* host, unsigned short port)
59 {
60   sock = socket(PF_INET, SOCK_STREAM, 0);
61   if (sock == -1) return 0;
62
63   struct sockaddr_in dest_addr;
64   dest_addr.sin_family = AF_INET;
65   dest_addr.sin_port = htons(port);
66
67   if (!inet_aton(host, &dest_addr.sin_addr))
68   {
69     cleanup();
70     return 0;
71   }
72
73   memset(&(dest_addr.sin_zero), '\0', 8);
74
75   int success = connect(sock, (struct sockaddr *)&dest_addr, sizeof(struct sockaddr));
76   if (success == -1)
77   {
78     cleanup();
79     return 0;
80   }
81
82   connected = 1;
83   return 1;
84 }
85
86 void TCP::setNonBlocking()
87 {
88   int oldflags = fcntl(sock, F_GETFL, 0);
89   oldflags |= O_NONBLOCK;
90   fcntl(sock, F_SETFL, oldflags);
91 }
92
93 int TCP::setSoKeepTime(int timeOut)
94 {
95   int option;
96   int s1, s2, s3, s4;
97
98   option = 1;
99   s1 = setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, &option, sizeof(option));
100   log->log("TCP", Log::DEBUG, "SO_KEEPALIVE = %i", s1);
101
102   option = timeOut;
103   s2 = setsockopt(sock, SOL_TCP, TCP_KEEPIDLE, &option, sizeof(option));
104   log->log("TCP", Log::DEBUG, "TCP_KEEPIDLE = %i", s2);
105
106   s3 = setsockopt(sock, SOL_TCP, TCP_KEEPINTVL, &option, sizeof(option));
107   log->log("TCP", Log::DEBUG, "TCP_KEEPINTVL = %i", s3);
108
109   option = 2;
110   s4 = setsockopt(sock, SOL_TCP, TCP_KEEPCNT, &option, sizeof(option));
111   log->log("TCP", Log::DEBUG, "TCP_KEEPCNT = %i", s4);
112
113   if (s1 || s2 || s3 || s4) return 0;
114   return 1;
115 }
116
117 void TCP::assignSocket(int tsocket)
118 {
119   sock = tsocket;
120   connected = 1;
121 }
122
123 int TCP::isConnected()
124 {
125   return connected;
126 }
127
128 /*
129 UCHAR* TCP::receivePacket()
130 {
131   if (!connected) return NULL;
132
133   int packetLength;
134   int success;
135
136   success = readData((UCHAR*)&packetLength, sizeof(int));
137   if (!success)
138   {
139     cleanup();
140     return NULL;
141   }
142
143   packetLength = ntohl(packetLength);
144
145   if (packetLength > 200000) return NULL;
146   UCHAR* buffer = (UCHAR*) malloc(packetLength);
147
148   success = readData(buffer, packetLength);
149   if (!success)
150   {
151     cleanup();
152     free(buffer);
153     return NULL;
154   }
155
156 //  dump((unsigned char*)buffer, packetLength);
157
158   dataLength = packetLength;
159   return buffer;
160 }
161 */
162
163 int TCP::getDataLength()
164 {
165   return dataLength;
166 }
167
168 int TCP::readData(UCHAR* buffer, int totalBytes)
169 {
170   if (!connected) return 0;
171
172   int bytesRead = 0;
173   int thisRead;
174   int readTries = 0;
175   int success;
176   fd_set readSet;
177   struct timeval timeout;
178   struct timeval* passToSelect;
179
180   if (readTimeoutEnabled) passToSelect = &timeout;
181   else passToSelect = NULL;
182
183   while(1)
184   {
185     FD_ZERO(&readSet);
186     FD_SET(sock, &readSet);
187     timeout.tv_sec = 20;
188     timeout.tv_usec = 0;
189     success = select(sock + 1, &readSet, NULL, NULL, passToSelect);
190     if (success < 1)
191     {
192 //      cleanup();
193       log->log("TCP", Log::DEBUG, "Select finished with %i", success);
194       return 0;  // error, or timeout
195     }
196
197     thisRead = read(sock, &buffer[bytesRead], totalBytes - bytesRead);
198     if (!thisRead)
199     {
200       // if read returns 0 then connection is closed
201       // in non-blocking mode if read is called with no data available, it returns -1
202       // and sets errno to EGAGAIN. but we use select so it wouldn't do that anyway.
203       cleanup();
204       return 0;
205     }
206     bytesRead += thisRead;
207
208 //    log->log("TCP", Log::DEBUG, "Bytes read now: %u", bytesRead);
209     if (bytesRead == totalBytes)
210     {
211       return 1;
212     }
213     else
214     {
215       if (++readTries == 100)
216       {
217         cleanup();
218         log->log("TCP", Log::DEBUG, "too many reads");
219         return 0;
220       }
221     }
222   }
223 }
224
225 int TCP::sendPacket(UCHAR* buf, size_t count)
226 {
227   pthread_mutex_lock(&sendLock);
228   
229   if (!connected)
230   {
231     pthread_mutex_unlock(&sendLock);
232     return 0;
233   }
234
235   unsigned int bytesWritten = 0;
236   int thisWrite;
237   int writeTries = 0;
238   int success;
239   fd_set writeSet;
240   struct timeval timeout;
241
242   while(1)
243   {
244     FD_ZERO(&writeSet);
245     FD_SET(sock, &writeSet);
246     timeout.tv_sec = 10;
247     timeout.tv_usec = 0;
248     success = select(sock + 1, NULL, &writeSet, NULL, &timeout);
249     if (success < 1)
250     {
251       cleanup();
252       log->log("TCP", Log::DEBUG, "TCP: error or timeout");
253       pthread_mutex_unlock(&sendLock);
254       return 0;  // error, or timeout
255     }
256
257     thisWrite = write(sock, &buf[bytesWritten], count - bytesWritten);
258 //  log->log("TCP", Log::DEBUG, "written %i", thisWrite);
259     if (!thisWrite)
260     {
261       // if write returns 0 then connection is closed ?
262       // in non-blocking mode if read is called with no data available, it returns -1
263       // and sets errno to EGAGAIN. but we use select so it wouldn't do that anyway.
264       cleanup();
265       log->log("TCP", Log::DEBUG, "Detected connection closed");
266       pthread_mutex_unlock(&sendLock);      
267       return 0;
268     }
269     bytesWritten += thisWrite;
270
271 //    log->log("TCP", Log::DEBUG, "Bytes written now: %u", bytesWritten);
272     if (bytesWritten == count)
273     {
274       pthread_mutex_unlock(&sendLock);
275       return 1;
276     }
277     else
278     {
279       if (++writeTries == 100)
280       {
281         cleanup();
282         log->log("TCP", Log::DEBUG, "too many writes");
283         pthread_mutex_unlock(&sendLock);
284         return 0;
285       }
286     }
287   }
288 }
289
290 void TCP::dump(unsigned char* data, USHORT size)
291 {
292   printf("Size = %u\n", size);
293
294   USHORT c = 0;
295   while(c < size)
296   {
297     if ((size - c) > 15)
298     {
299       printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X  %c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c\n",
300         data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
301         data[c+8], data[c+9], data[c+10], data[c+11], data[c+12], data[c+13], data[c+14], data[c+15],
302         dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
303         dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]), dcc(data[c+11]), dcc(data[c+12]), dcc(data[c+13]), dcc(data[c+14]), dcc(data[c+15]));
304       c += 16;
305     }
306     else
307     {
308       switch (size - c)
309       {
310         case 15:
311           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X     %c%c%c%c%c%c%c%c%c%c%c%c%c%c%c\n",
312             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
313             data[c+8], data[c+9], data[c+10], data[c+11], data[c+12], data[c+13], data[c+14],
314             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
315             dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]), dcc(data[c+11]), dcc(data[c+12]), dcc(data[c+13]), dcc(data[c+14]));
316           c += 15;
317           break;
318         case 14:
319           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X        %c%c%c%c%c%c%c%c%c%c%c%c%c%c\n",
320             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
321             data[c+8], data[c+9], data[c+10], data[c+11], data[c+12], data[c+13],
322             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
323             dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]), dcc(data[c+11]), dcc(data[c+12]), dcc(data[c+13]));
324           c += 14;
325           break;
326         case 13:
327           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X  %02X           %c%c%c%c%c%c%c%c%c%c%c%c%c\n",
328             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
329             data[c+8], data[c+9], data[c+10], data[c+11], data[c+12],
330             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
331             dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]), dcc(data[c+11]), dcc(data[c+12]));
332           c += 13;
333           break;
334         case 12:
335           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X %02X               %c%c%c%c%c%c%c%c%c%c%c%c\n",
336             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
337             data[c+8], data[c+9], data[c+10], data[c+11],
338             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
339             dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]), dcc(data[c+11]));
340           c += 12;
341           break;
342         case 11:
343           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X %02X                  %c%c%c%c%c%c%c%c%c%c%c\n",
344             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
345             data[c+8], data[c+9], data[c+10],
346             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
347             dcc(data[c+8]), dcc(data[c+9]), dcc(data[c+10]));
348           c += 11;
349           break;
350         case 10:
351           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X %02X                     %c%c%c%c%c%c%c%c%c%c\n",
352             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
353             data[c+8], data[c+9],
354             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
355             dcc(data[c+8]), dcc(data[c+9]));
356           c += 10;
357           break;
358         case 9:
359           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X  %02X                        %c%c%c%c%c%c%c%c%c\n",
360             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
361             data[c+8],
362             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]),
363             dcc(data[c+8]));
364           c += 9;
365           break;
366         case 8:
367           printf(" %02X %02X %02X %02X  %02X %02X %02X %02X                            %c%c%c%c%c%c%c%c\n",
368             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6], data[c+7],
369             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]), dcc(data[c+7]));
370           c += 8;
371           break;
372         case 7:
373           printf(" %02X %02X %02X %02X  %02X %02X %02X                               %c%c%c%c%c%c%c\n",
374             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5], data[c+6],
375             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]), dcc(data[c+6]));
376           c += 7;
377           break;
378         case 6:
379           printf(" %02X %02X %02X %02X  %02X %02X                                  %c%c%c%c%c%c\n",
380             data[c], data[c+1], data[c+2], data[c+3], data[c+4], data[c+5],
381             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]), dcc(data[c+5]));
382           c += 6;
383           break;
384         case 5:
385           printf(" %02X %02X %02X %02X  %02X                                     %c%c%c%c%c\n",
386             data[c], data[c+1], data[c+2], data[c+3], data[c+4],
387             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]), dcc(data[c+4]));
388           c += 5;
389           break;
390         case 4:
391           printf(" %02X %02X %02X %02X                                         %c%c%c%c\n",
392             data[c], data[c+1], data[c+2], data[c+3],
393             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]), dcc(data[c+3]));
394           c += 4;
395           break;
396         case 3:
397           printf(" %02X %02X %02X                                            %c%c%c\n",
398             data[c], data[c+1], data[c+2],
399             dcc(data[c]), dcc(data[c+1]), dcc(data[c+2]));
400           c += 3;
401           break;
402         case 2:
403           printf(" %02X %02X                                               %c%c\n",
404             data[c], data[c+1],
405             dcc(data[c]), dcc(data[c+1]));
406           c += 2;
407           break;
408         case 1:
409           printf(" %02X                                                  %c\n",
410             data[c],
411             dcc(data[c]));
412           c += 1;
413           break;
414       }
415     }
416   }
417 }
418
419 unsigned char TCP::dcc(UCHAR c)
420 {
421   if (isspace(c)) return ' ';
422   if (isprint(c)) return c;
423   return '.';
424 }