]> git.vomp.tv Git - vompserver.git/blob - tcp.h
Patch from Dave Pickles for VDR 1.3 tv/radio channel detection
[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
41 typedef unsigned char UCHAR;
42 typedef unsigned short USHORT;
43
44 class TCP
45 {
46   public:
47     // Creation
48     TCP(int tsocket);
49     ~TCP();
50     void assignSocket(int tsocket);
51
52     // Setup
53     void disableReadTimeout();
54     void setNonBlocking();
55     int setSoKeepTime(int timeOut);
56
57     int connectTo(char *host, unsigned short port);
58     int sendPacket(UCHAR*, size_t size);
59     UCHAR* receivePacket();
60
61     // Get methods
62     int isConnected();
63     int getDataLength();
64
65     static void dump(unsigned char* data, USHORT size);
66     static UCHAR dcc(UCHAR c);
67
68   private:
69     int sock;
70     int connected;
71     int readTimeoutEnabled;
72     int dataLength;
73
74     void cleanup();
75     int readData(UCHAR* buffer, int totalBytes);
76 };
77
78 #endif