]> git.vomp.tv Git - vompclient-marten.git/blob - wol.cc
Prev channel and prebuffering osd fixes
[vompclient-marten.git] / wol.cc
1 /*
2     Copyright 2006 Matthias Haas <vomp@pompase.net>
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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
19
20     Based on the work by R. Edwards (bob@cs.anu.edu.au) for the wakeonlan stuff
21     Based ont the work of Fred N. van Kempen, <waltje@uwalt.nl.mugnet.org> for the arp stuff
22     Thanks to you guys.
23 */
24
25 #include <ctype.h>
26 #include <errno.h>
27 #ifndef WIN32
28 #include <unistd.h>
29 #include <net/ethernet.h>
30 #include <sys/socket.h>
31 #include <netinet/in.h>
32 #else
33 #define ETH_ALEN                6
34
35 #include <winsock2.h>
36 #include <errno.h>
37 #include <Ws2tcpip.h>
38 #include <Iphlpapi.h>
39 #endif
40 #include "wol.h"
41 #include "log.h"
42
43 #define _PATH_PROCNET_ARP "/proc/net/arp"
44
45 Wol* Wol::instance = NULL;
46
47 Wol::Wol()
48 {
49   if (instance) return;
50   instance = this;
51   bEtherKnown = false;
52   wolEnabled = true;
53   logger = Log::getInstance();
54 #ifdef WIN32
55 //  goo;
56 #endif
57 }
58
59 Wol::~Wol()
60 {
61 }
62
63 Wol* Wol::getInstance()
64 {
65   return instance;
66 }
67
68 void Wol::setEnabled(bool tEnabled)
69 {
70   wolEnabled = tEnabled;
71 }
72 #ifndef WIN32
73 /* Display the contents of the ARP cache in the kernel. */
74 int Wol::find_ether(char *name)
75 {
76   char ip[100];
77   char line[200];
78   FILE *fp;
79   int num;
80
81   /* Open the PROCps kernel table. */
82   if ((fp = fopen(_PATH_PROCNET_ARP, "r")) == NULL)
83   {
84     perror(_PATH_PROCNET_ARP);
85     return -1;
86   }
87   /* Bypass header -- read until newline */
88   if (fgets(line, sizeof(line), fp) != (char *) NULL)
89   {
90     /* Read the ARP cache entries. */
91     for (; fgets(line, sizeof(line), fp);) {
92       num = sscanf(line, "%s 0x%*x 0x%*x %100s %*100s %*100s\n",
93         ip, ether_addr);
94       if (num < 2)
95         break;
96       /* if the user specified address differs, skip it */
97       if (name[0] && !strcmp(ip, name))
98       {
99         logger->log("Wol", Log::NOTICE, "Found etheraddr %s", ether_addr);
100         return 1;
101       }
102     }
103   }
104   fclose(fp);
105   return 0;
106 }
107 #else
108 int Wol::find_ether(char *name)
109 {
110   sockaddr_in sock_address;
111   int sockname_len=sizeof(sock_address);
112   WSAStringToAddress(name,AF_INET,NULL,(sockaddr*)&sock_address,&sockname_len);
113   
114   DWORD ip=sock_address.sin_addr.s_addr;;
115   ULONG size=0;
116   PMIB_IPNETTABLE table;
117
118   GetIpNetTable(NULL,&size,TRUE);
119   table=(PMIB_IPNETTABLE)new char[size];
120   if (table==NULL) return -1;
121   if (GetIpNetTable(table,&size,TRUE)!=NOERROR)
122   {
123       logger->log("Wol", Log::NOTICE, "Error getting IPtable");
124       delete [] table;
125       return -1;
126   }
127   for (int x=0;x<table->dwNumEntries;x++)
128   {
129       if (table->table[x].dwAddr==ip) {
130           char *ethp=ether_addr;
131           for (int i=0;i<table->table[x].dwPhysAddrLen;i++) {
132               ethp+=sprintf(ethp,"%X:",table->table[x].bPhysAddr[i]);
133           }
134           delete [] table;
135           logger->log("Wol", Log::NOTICE, "Found etheraddr %s", ether_addr);
136           return 1;
137       }
138       
139   }
140
141   delete [] table;
142  return 0;
143
144   
145 }
146 #endif
147
148 /* Input an Ethernet address and convert to binary. */
149 int Wol::convertToBinary(char *bufp, unsigned char *addr)
150 {
151     char c, *orig;
152     int i;
153         unsigned char *ptr = addr;
154     unsigned val;
155
156     i = 0;
157     orig = bufp;
158     while ((*bufp != '\0') && (i < ETH_ALEN))
159     {
160         val = 0;
161         c = *bufp++;
162         if (isdigit(c))
163             val = c - '0';
164         else if (c >= 'a' && c <= 'f')
165             val = c - 'a' + 10;
166         else if (c >= 'A' && c <= 'F')
167             val = c - 'A' + 10;
168         else
169         {
170 #ifdef DEBUG
171             fprintf(stderr, "in_ether(%s): invalid ether address!\n", orig);
172 #endif
173             errno = EINVAL;
174             return (-1);
175         }
176         val <<= 4;
177         c = *bufp;
178         if (isdigit(c))
179             val |= c - '0';
180         else if (c >= 'a' && c <= 'f')
181             val |= c - 'a' + 10;
182         else if (c >= 'A' && c <= 'F')
183             val |= c - 'A' + 10;
184         else if (c == ':' || c == 0)
185             val >>= 4;
186         else
187         {
188 #ifdef DEBUG
189             fprintf(stderr, "in_ether(%s): invalid ether address!\n", orig);
190 #endif
191             errno = EINVAL;
192             return (-1);
193         }
194         if (c != 0)
195             bufp++;
196         *ptr++ = (unsigned char) (val & 0377);
197         i++;
198
199         /* We might get a semicolon here - not required. */
200         if (*bufp == ':')
201         {
202             if (i == ETH_ALEN)
203             {
204                     ;           /* nothing */
205             }
206             bufp++;
207         }
208     }
209     return (0);
210 } /* in_ether */
211
212 int Wol::doWakeUp()
213 {
214   if (!wolEnabled) return -2;
215   if (!bEtherKnown) return -3;
216
217   int i, j;
218   int packet;
219   struct sockaddr_in sap;
220   unsigned char ethaddr[8];
221   unsigned char *ptr;
222   unsigned char buf [128];
223   int optval = 1;
224
225   /* Fetch the hardware address. */
226   if (convertToBinary (ether_addr, ethaddr) < 0)
227   {
228     return (-1);
229   }
230
231   /* setup the packet socket */
232   if ((packet = socket (PF_INET, SOCK_DGRAM, 0)) < 0)
233   {
234     return (-1);
235   }
236
237   /* Set socket options */
238 #ifndef WIN32
239   if (setsockopt (packet, SOL_SOCKET, SO_BROADCAST, (caddr_t) &optval,
240      sizeof (optval)) < 0)
241 #else
242   if (setsockopt (packet, SOL_SOCKET, SO_BROADCAST, (char*)&optval,
243      sizeof (optval)) < 0)
244 #endif
245   {
246     fprintf (stderr, "setsocket failed %s\n",
247     strerror (errno));
248     CLOSESOCKET(packet);
249     return (-1);
250   }
251
252   /* Set up broadcast address */
253   sap.sin_family = AF_INET;
254   sap.sin_addr.s_addr = htonl(0xffffffff);        /* broadcast address
255 */
256   sap.sin_port = htons(60000);
257
258   /* Build the message to send - 6 x 0xff then 16 x MAC address */
259   ptr = buf;
260   for (i = 0; i < 6; i++)
261     *ptr++ = 0xff;
262   for (j = 0; j < 16; j++)
263     for (i = 0; i < ETH_ALEN; i++)
264       *ptr++ = ethaddr [i];
265
266   /* Send the packet out */
267   if (sendto (packet,(char*) buf, 102, 0, (struct sockaddr *)&sap, sizeof (sap)) < 0)
268   {
269     fprintf (stderr, " sendto failed, %s\n",
270     strerror(errno));
271     CLOSESOCKET(packet);
272     return (-1);
273   }
274
275   logger->log("Wol", Log::NOTICE, "Send wakeonlan to server");
276
277   CLOSESOCKET(packet);
278   return (0);
279 }
280
281 void Wol::setWakeUpIP(char* ip_addr)
282 {
283   if(find_ether(ip_addr))
284   {
285     bEtherKnown = true;
286     logger->log("Wol", Log::NOTICE, "Server IP set");
287   }
288 }
289