]> git.vomp.tv Git - vompclient-marten.git/blob - dsock.cc
EPG tweaks, EPG made NTSC compatible
[vompclient-marten.git] / dsock.cc
1 /*
2     Copyright 2004-2005 Chris Tallon
3
4     This file is part of VOMP.
5
6     VOMP is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     VOMP is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with VOMP; if not, write to the Free Software
18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19 */
20
21 #include "dsock.h"
22
23 DatagramSocket::DatagramSocket(short port)
24 {
25   theInstance = this;
26   myPort = port;
27   addrlen = sizeof(struct sockaddr);
28
29   if ((socketnum = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
30   { perror("socket"); exit(1); }
31
32   myAddr.sin_family = AF_INET;         // host byte order
33   myAddr.sin_port = htons(myPort);     // short, network byte order
34   myAddr.sin_addr.s_addr = INADDR_ANY; // auto-fill with my IP
35   memset(&(myAddr.sin_zero), 0, 8);    // zero the rest of the struct
36
37   if (bind(socketnum, (struct sockaddr *)&myAddr, addrlen) == -1)
38   { perror("bind"); exit(1); }
39
40   FD_ZERO(&readfds);
41   FD_SET(socketnum, &readfds);
42   tv.tv_sec = 0;
43   tv.tv_usec = 0;
44
45   int allowed = 1;
46   setsockopt(socketnum, SOL_SOCKET, SO_BROADCAST, &allowed, sizeof(allowed));
47 }
48
49 DatagramSocket::~DatagramSocket()
50 {
51   close(socketnum);
52 }
53
54 DatagramSocket* DatagramSocket::theInstance = 0;
55 DatagramSocket* DatagramSocket::getInstance(void)
56 {
57   return theInstance;
58 }
59
60 unsigned char DatagramSocket::waitforMessage(unsigned char how)
61 {
62   /* how = 0 - block
63      how = 1 - start new wait
64      how = 2 - continue wait
65   */
66
67   struct timeval* passToSelect = NULL;
68
69
70   if (how == 0)
71   {
72     passToSelect = NULL;
73   }
74   else if (how == 1)
75   {
76     tv.tv_sec = 1;
77     tv.tv_usec = 500000;
78     passToSelect = &tv;
79   }
80   else if (how == 2)
81   {
82     if ((tv.tv_sec == 0) && (tv.tv_usec == 0))  // protection in case timer = 0
83     {
84       tv.tv_sec = 1;
85       tv.tv_usec = 500000;
86     }
87     passToSelect = &tv;
88   }
89   FD_ZERO(&readfds);
90   FD_SET(socketnum, &readfds);
91
92   if (select(socketnum + 1, &readfds, NULL, NULL, passToSelect) <= 0)
93   {  return 1;  }
94
95   if ((mlength = recvfrom(socketnum, buf, MAXBUFLEN, 0,
96       (struct sockaddr *)&theirAddr, &addrlen)) == -1)
97   { perror("recvfrom"); return 0; }
98   else
99   {
100     memset(&buf[mlength], 0, MAXBUFLEN - mlength);
101     strcpy(fromIPA, inet_ntoa(theirAddr.sin_addr));
102     fromPort = ntohs(theirAddr.sin_port);
103
104     if (DSOCKDEBUG)
105     {
106       //printf("%s:%i\tIN  %i\t", fromIPA, fromPort, mlength);
107       int k;
108       for(k = 0; k < mlength; k++)
109         printf("%u ", (unsigned char)buf[k]);
110       printf("\n");
111     }
112     return 2;
113   }
114
115   /* Return 0, failure
116      Return 1, nothing happened, timer expired
117      Return 2, packet arrived (timer not expired)
118   */
119 }
120
121 int DatagramSocket::getDataLength(void) const
122 {
123   return mlength;
124 }
125
126 char *DatagramSocket::getData(void)             {  return buf;  }
127 char *DatagramSocket::getFromIPA(void)          {  return fromIPA;  }
128 short DatagramSocket::getFromPort(void) const   {  return fromPort; }
129
130 void DatagramSocket::send(char *ipa, short port, char *message, int length)
131 {
132   if (DSOCKDEBUG)
133   {
134     printf("%s:%i\tOUT %i\t", ipa, port, length);
135     int k;
136     uchar l;
137     for (k = 0; k < length; k++)
138       { l = (uchar)message[k]; printf("%u ", l); }
139   }
140
141   int sentLength = 0;
142
143   theirAddr.sin_family = AF_INET;      // host byte order
144   theirAddr.sin_port = htons(port);    // short, network byte order
145   struct in_addr tad;                  // temp struct tad needed to pass to theirAddr.sin_addr
146   tad.s_addr = inet_addr(ipa);
147   theirAddr.sin_addr = tad;            // address
148   memset(&(theirAddr.sin_zero), 0, 8); // zero the rest of the struct
149
150   errno = 0;
151
152   sentLength = sendto(socketnum, message, length, 0, (struct sockaddr *)&theirAddr, addrlen);
153   if (sentLength == length)
154   {
155     if (DSOCKDEBUG) printf(" GOOD\n");
156   }
157   else
158   {
159     if (DSOCKDEBUG)
160     {
161       printf(" --BAD--");  fflush(stdout);
162     }
163     sentLength = sendto(socketnum, message, length, 0, (struct sockaddr *)&theirAddr, addrlen);
164     if (sentLength == length)
165     {
166       if (DSOCKDEBUG) printf(" GOOD\n");
167     }
168     else
169     {
170       if (DSOCKDEBUG && (sentLength != length))
171       {
172
173         printf(" -#-FAILED-#-\n");
174         printf("--------------\n");
175         printf("Sendto failure\n");
176         printf("--------------\n");
177         printf("%s:%i\tOUT %i %i ...\n", ipa, port, length, sentLength);
178         perror("Perror reports");
179         printf("errno value: %d\n", errno);
180         printf("errno translated: %s\n", strerror(errno));
181       //  printf("h_errno value: %d\n", h_errno);
182       //  printf("\nActual address: %s\n", inet_ntoa(tad));
183       //  printf("Actual port: %i\n", ntohs(theirAddr.sin_port));
184         printf("continuing...\n\n");
185
186       }
187     }
188   }
189 }
190