]> git.vomp.tv Git - vompserver.git/blob - vompclient.c
IPv6 support for discovery protocol
[vompserver.git] / vompclient.c
1 /*
2     Copyright 2004-2008 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
19 */
20
21 #include "mediaplayer.h"
22 #include "media.h"
23 #include "servermediafile.h"
24 #include "vdrcommand.h"
25 #include "vompclient.h"
26
27 #include "responsepacket.h"
28
29 #ifndef VOMPSTANDALONE
30 #include <vdr/channels.h>
31 #include <vdr/recording.h>
32 #include <vdr/plugin.h>
33 #include "recplayer.h"
34 #include "mvpreceiver.h"
35 #include "picturereader.h"
36 #endif
37
38
39
40 pthread_mutex_t threadClientMutex;
41 int VompClient::nr_clients = 0;
42 cPlugin *VompClient::scraper = NULL;
43 time_t  VompClient::lastScrapQuery = 0;
44
45 VompClient::VompClient(Config* cfgBase, char* tconfigDir, char* tlogoDir, 
46             char *tresourceDir, char * timageDir, char * tcacheDir, int tsocket)
47  : rrproc(*this), tcp(tsocket), i18n(tconfigDir)
48 {
49 #ifndef VOMPSTANDALONE
50   lp = NULL;
51   recplayer = NULL;
52   pict = new PictureReader(this);
53   if (!scraper) scrapQuery();
54   logoDir = tlogoDir;
55   resourceDir = tresourceDir;
56   imageDir = timageDir;
57   cacheDir = tcacheDir;
58 #endif
59   log = Log::getInstance();
60   loggedIn = false;
61   configDir = tconfigDir;
62   log->log("Client", Log::DEBUG, "Config dir: %s", configDir);
63   baseConfig = cfgBase;
64   incClients();
65   media=new MediaPlayer();
66   mediaprovider=new ServerMediaFile(cfgBase,media);
67   netLogFile = NULL;
68   charcoding=1; //latin1 is default
69   charconvsys=NULL;
70   charconvutf8=NULL;
71   setCharset(charcoding);
72   
73   rrproc.init();
74 }
75
76 VompClient::~VompClient()
77 {
78   log->log("Client", Log::DEBUG, "Vomp client destructor");
79 #ifndef VOMPSTANDALONE  
80   if (lp)
81   {
82     lp->detachMVPReceiver();
83     delete lp;
84     lp = NULL;
85   }
86   else if (recplayer)
87   {
88     writeResumeData();
89
90     delete recplayer;
91     recplayer = NULL;
92   }
93 #endif
94   //if (loggedIn) cleanConfig();
95   decClients();
96   
97   delete pict;
98   
99   delete media;
100   delete mediaprovider;
101
102   if (charconvsys) delete charconvsys;
103   if (charconvutf8) delete charconvutf8;
104
105   
106   if (netLogFile)
107   {
108     fclose(netLogFile);
109     netLogFile = NULL;
110   }
111 }
112
113 cPlugin *VompClient::scrapQuery()
114 {
115     if (scraper) return scraper;
116     if ((time(NULL)-lastScrapQuery) > 5*60) {
117         lastScrapQuery = time(NULL); 
118           if (!scraper) scraper = cPluginManager::GetPlugin("scraper2vdr");
119    }
120    return scraper;
121 }
122
123 void VompClient::setCharset(int charset)
124 {
125    charcoding=charset;
126    cCharSetConv *oldcharconvsys=charconvsys;
127    cCharSetConv *oldcharconvutf8=charconvutf8;
128    switch (charcoding) {
129    case 2: //UTF-8
130    charconvsys=new cCharSetConv(NULL,"UTF-8");
131    charconvutf8=new cCharSetConv("UTF-8","UTF-8");
132    break;
133    case 1:
134    default://latin1
135    charconvsys=new cCharSetConv(NULL,"ISO-8859-1");
136    charconvutf8=new cCharSetConv("UTF-8","ISO-8859-1");
137    break;
138    };
139    if (oldcharconvsys) delete oldcharconvsys;
140    if (oldcharconvutf8) delete oldcharconvutf8;
141
142 }
143
144 void VompClient::incClients()
145 {
146   pthread_mutex_lock(&threadClientMutex);
147   VompClient::nr_clients++;
148   pthread_mutex_unlock(&threadClientMutex);
149 }
150
151 void VompClient::decClients()
152 {
153   pthread_mutex_lock(&threadClientMutex);
154   VompClient::nr_clients--;
155   pthread_mutex_unlock(&threadClientMutex);
156 }
157
158 int VompClient::getNrClients()
159 {
160   int nrClients;
161   pthread_mutex_lock(&threadClientMutex);
162   nrClients = VompClient::nr_clients;
163   pthread_mutex_unlock(&threadClientMutex);
164   return nrClients;
165 }
166
167 /*
168 void VompClient::cleanConfig()
169 {
170   log->log("Client", Log::DEBUG, "Clean config");
171
172 #ifndef VOMPSTANDALONE
173
174   cRecordings Recordings;
175   Recordings.Load();
176
177   int numReturns;
178   int length;
179   char* resumes = config.getSectionKeyNames("ResumeData", numReturns, length);
180   char* position = resumes;
181   for(int k = 0; k < numReturns; k++)
182   {
183     log->log("Client", Log::DEBUG, "EXAMINING: %i %i %p %s", k, numReturns, position, position);
184
185     cRecording* recording = Recordings.GetByName(position);
186     if (!recording)
187     {
188       // doesn't exist anymore
189       log->log("Client", Log::DEBUG, "Found a recording that doesn't exist anymore");
190       config.deleteValue("ResumeData", position);
191     }
192     else
193     {
194       log->log("Client", Log::DEBUG, "This recording still exists");
195     }
196
197     position += strlen(position) + 1;
198   }
199
200   delete[] resumes;
201 #endif
202 } */
203
204 void VompClient::netLog()
205 {
206   // Hook, called from rrproc login after client has logged in.
207   // See if this MVP config says to do network logging, if so open a log
208   // The config object will be usable now since it's set up in login
209   
210   char* doNetLogging = config.getValueString("Advanced", "Network logging");
211   if (doNetLogging)
212   {
213     if (!strcasecmp(doNetLogging, "on"))
214     {
215       char* netLogFileName = config.getValueString("Advanced", "Network logging file");
216       if (netLogFileName)
217       {
218         netLogFile = fopen(netLogFileName, "a");
219         if (netLogFile) log->log("Client", Log::DEBUG, "Client network logging started");
220
221         delete[] netLogFileName;
222       }
223     }
224     
225     delete[] doNetLogging;
226   }
227 }
228
229 void VompClientStartThread(void* arg)
230 {
231   VompClient* m = (VompClient*)arg;
232   m->run2();
233   // Nothing external to this class has a reference to it
234   // This is the end of the thread.. so delete m
235   delete m;
236   pthread_exit(NULL);
237 }
238
239 int VompClient::run()
240 {
241   if (pthread_create(&runThread, NULL, (void*(*)(void*))VompClientStartThread, (void *)this) == -1) return 0;
242   log->log("Client", Log::DEBUG, "VompClient run success");
243   return 1;
244 }
245
246 void VompClient::run2()
247 {
248   // Thread stuff
249   sigset_t sigset;
250   sigfillset(&sigset);
251   pthread_sigmask(SIG_BLOCK, &sigset, NULL);
252   pthread_detach(runThread);  // Detach
253
254 //  tcp.disableReadTimeout();
255
256 //  tcp.setSoKeepTime(3);
257   tcp.setNonBlocking();
258
259   pict->init(&tcp);
260   ULONG channelID;
261   ULONG requestID;
262   ULONG opcode;
263   ULONG extraDataLength;
264   UCHAR* data;
265   
266   ULONG kaTimeStamp;
267   ULONG logStringLen;
268
269   while(1)
270   {
271     log->log("Client", Log::DEBUG, "Waiting");
272     
273     if (!tcp.readData((UCHAR*)&channelID, sizeof(ULONG)))
274     {
275       // If this read fails then the client just hasn't sent anything.
276       // If any of the lower reads fail, then break, the connection is probably dead
277
278       // check connection is ok
279 //      if (tcp.isConnected()) continue;
280       
281       log->log("Client", Log::DEBUG, "Disconnection detected");
282       break;
283     }     
284     
285     channelID = ntohl(channelID);
286     if (channelID == 1)
287     {
288       if (!tcp.readData((UCHAR*)&requestID, sizeof(ULONG))) break;
289       requestID = ntohl(requestID);
290
291       if (!tcp.readData((UCHAR*)&opcode, sizeof(ULONG))) break;
292       opcode = ntohl(opcode);
293
294       if (!tcp.readData((UCHAR*)&extraDataLength, sizeof(ULONG))) break;
295       extraDataLength = ntohl(extraDataLength);
296       if (extraDataLength > 200000) // a random sanity limit
297       {
298         log->log("Client", Log::ERR, "ExtraDataLength > 200000!");
299         break;
300       }
301
302       if (extraDataLength)
303       {
304         data = (UCHAR*)malloc(extraDataLength);
305         if (!data)
306         {
307           log->log("Client", Log::ERR, "Extra data buffer malloc error");
308           break;
309         }
310         
311         if (!tcp.readData(data, extraDataLength))
312         {
313           log->log("Client", Log::ERR, "Could not read extradata");
314           free(data);
315           break;
316         }      
317       }
318       else
319       {
320         data = NULL;
321       }
322
323       log->log("Client", Log::DEBUG, "Received chan=%lu, ser=%lu, op=%lu, edl=%lu", channelID, requestID, opcode, extraDataLength);
324
325       if (!loggedIn && (opcode != 1))
326       {
327         log->log("Client", Log::ERR, "Not logged in and opcode != 1");
328         if (data) free(data);
329         break;
330       }
331
332       RequestPacket* req = new RequestPacket(requestID, opcode, data, extraDataLength);
333       rrproc.recvRequest(req);
334     }
335     else if (channelID == 3)
336     {
337       if (!tcp.readData((UCHAR*)&kaTimeStamp, sizeof(ULONG))) break;
338       kaTimeStamp = ntohl(kaTimeStamp);
339
340       log->log("Client", Log::DEBUG, "Received chan=%lu kats=%lu", channelID, kaTimeStamp);    
341
342       ULONG* p;
343       UCHAR buffer[8];
344       p = (ULONG*)&buffer[0]; *p = htonl(3); // KA CHANNEL
345       p = (ULONG*)&buffer[4]; *p = htonl(kaTimeStamp);
346       if (!tcp.sendPacket(buffer, 8))
347       {
348         log->log("Client", Log::ERR, "Could not send back KA reply");
349         break;
350       }      
351     }
352     else if (channelID == 4)
353     {
354       if (!tcp.readData((UCHAR*)&logStringLen, sizeof(ULONG))) break;
355       logStringLen = ntohl(logStringLen);
356
357       log->log("Client", Log::DEBUG, "Received chan=%lu loglen=%lu", channelID, logStringLen);    
358
359       UCHAR buffer[logStringLen + 1];
360       if (!tcp.readData((UCHAR*)&buffer, logStringLen)) break;
361       buffer[logStringLen] = '\0';
362
363 //      log->log("Client", Log::INFO, "Client said: '%s'", buffer);
364       if (netLogFile)
365       {
366         if (fputs((const char*)buffer, netLogFile) == EOF)
367         {
368           fclose(netLogFile);
369           netLogFile = NULL;
370         }
371         fflush(NULL);
372       }
373     }    
374     else
375     {
376       log->log("Client", Log::ERR, "Incoming channel number unknown");
377       break;
378     }
379   }
380 }
381
382 ULLONG VompClient::ntohll(ULLONG a)
383 {
384   return htonll(a);
385 }
386
387 ULLONG VompClient::htonll(ULLONG a)
388 {
389   #if BYTE_ORDER == BIG_ENDIAN
390     return a;
391   #else
392     ULLONG b = 0;
393
394     b = ((a << 56) & 0xFF00000000000000ULL)
395       | ((a << 40) & 0x00FF000000000000ULL)
396       | ((a << 24) & 0x0000FF0000000000ULL)
397       | ((a <<  8) & 0x000000FF00000000ULL)
398       | ((a >>  8) & 0x00000000FF000000ULL)
399       | ((a >> 24) & 0x0000000000FF0000ULL)
400       | ((a >> 40) & 0x000000000000FF00ULL)
401       | ((a >> 56) & 0x00000000000000FFULL) ;
402
403     return b;
404   #endif
405 }
406
407 #ifndef VOMPSTANDALONE
408
409 void VompClient::writeResumeData()
410 {
411   /*config.setValueLong("ResumeData",
412                           (char*)recplayer->getCurrentRecording()->FileName(),
413                           recplayer->frameNumberFromPosition(recplayer->getLastPosition()) );*/
414
415   /* write to vdr resume file */
416   int resume = recplayer->frameNumberFromPosition(recplayer->getLastPosition());
417   char* ResumeIdC = config.getValueString("General", "ResumeId");
418   int ResumeId;
419   if (ResumeIdC)
420     ResumeId = atoi(ResumeIdC);
421   else
422     ResumeId = 0;
423
424   while (ResumeIDLock)
425           cCondWait::SleepMs(100);
426   ResumeIDLock = true;
427   int OldSetupResumeID = Setup.ResumeID;
428   Setup.ResumeID = ResumeId;                            //UGLY: quickly change resumeid
429 #if VDRVERSNUM < 10703
430   cResumeFile ResumeFile((char*)recplayer->getCurrentRecording()->FileName());  //get corresponding resume file
431 #else
432   cResumeFile ResumeFile((char*)recplayer->getCurrentRecording()->FileName(),(char*)recplayer->getCurrentRecording()->IsPesRecording());        //get corresponding resume file
433 #endif
434   Setup.ResumeID = OldSetupResumeID;                    //and restore it back
435   ResumeIDLock = false;
436   ResumeFile.Save(resume);
437   //isyslog("VOMPDEBUG: Saving resume = %i, ResumeId = %i",resume, ResumeId);
438 }
439
440 #endif
441