]> git.vomp.tv Git - vompserver.git/blob - tcp.h
Channel schedules
[vompserver.git] / tcp.h
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20 */
21
22 #ifndef TCP_H
23 #define TCP_H
24
25 #include <stdio.h>
26 #include <unistd.h>
27 #include <string.h>
28 #include <stdlib.h>
29 #include <errno.h>
30
31 // For TCP_KEEPIDLE and co
32 #include <netinet/tcp.h>
33
34 #include <sys/socket.h>
35 #include <netinet/in.h>
36 #include <arpa/inet.h>
37 #include <fcntl.h>
38 #include <ctype.h>
39
40 #include "log.h"
41
42
43 typedef unsigned char UCHAR;
44 typedef unsigned short USHORT;
45
46 class TCP
47 {
48   public:
49     // Creation
50     TCP(int tsocket);
51     ~TCP();
52     void assignSocket(int tsocket);
53
54     // Setup
55     void disableReadTimeout();
56     void setNonBlocking();
57     int setSoKeepTime(int timeOut);
58
59     int connectTo(char *host, unsigned short port);
60     int sendPacket(UCHAR*, size_t size);
61     UCHAR* receivePacket();
62
63     // Get methods
64     int isConnected();
65     int getDataLength();
66
67     static void dump(unsigned char* data, USHORT size);
68     static UCHAR dcc(UCHAR c);
69
70   private:
71     Log* log;
72     int sock;
73     int connected;
74     int readTimeoutEnabled;
75     int dataLength;
76
77     void cleanup();
78     int readData(UCHAR* buffer, int totalBytes);
79 };
80
81 #endif