]> git.vomp.tv Git - vompclient.git/blob - main.cc
Two segfaults fixed for when there are multiple servers
[vompclient.git] / main.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 <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <signal.h>
25 #include <unistd.h>
26 #include <endian.h>
27
28 #include "defines.h"
29 #include "log.h"
30 #include "remote.h"
31 #include "led.h"
32 #include "mtd.h"
33 #include "video.h"
34 #include "audio.h"
35 #include "vdr.h"
36 #include "osd.h"
37 #include "viewman.h"
38 #include "command.h"
39
40 void sighandler(int signalReceived);
41 void shutdown(int code);
42
43 // Global variables --------------------------------------------------------------------------------------------------
44 int debugEnabled = 0;
45 Log* logger;
46 Remote* remote;
47 Mtd* mtd;
48 Led* led;
49 Osd* osd;
50 ViewMan* viewman;
51 Command* command;
52 VDR* vdr;
53 Video* video;
54 Audio* audio;
55
56 int main(int argc, char** argv)
57 {
58   if ((argc > 1) && (!strcmp(argv[1], "-d"))) debugEnabled = 1;
59
60
61   // Init global vars ------------------------------------------------------------------------------------------------
62
63   logger     = new Log();
64   remote     = new Remote();
65   mtd        = new Mtd();
66   led        = new Led();
67   osd        = new Osd();
68   vdr        = new VDR();
69   video      = new Video();
70   audio      = new Audio();
71   viewman    = new ViewMan();
72   command    = new Command();
73
74   if (!logger || !remote || !mtd || !led || !osd || !video || !audio || !viewman || !command)
75   {
76     printf("Could not create objects. Memory problems?\n");
77     shutdown(1);
78   }
79
80   // Get logging module started --------------------------------------------------------------------------------------
81
82   if (!logger->init(Log::DEBUG, "dummy", debugEnabled))
83   {
84     printf("Could not initialise log object. Aborting.\n");
85     shutdown(1);
86   }
87
88   logger->log("Core", Log::INFO, "Starting up...");
89
90   // Daemonize if not -d
91
92   if (!debugEnabled)
93   {
94     // Fork away
95     pid_t forkTest = fork();
96     if (forkTest == -1)
97     { printf("Cannot fork (1).\n"); exit(1); }
98     if (forkTest != 0) _exit(0); // PID returned, I am the parent
99     // otherwise, I am the child
100     setsid();
101     forkTest = fork();
102     if (forkTest == -1)
103     { printf("Cannot fork (2).\n"); exit(1); }
104     if (forkTest != 0) _exit(0); // PID returned, I am the parent
105     // otherwise, I am the child
106     close(0);
107     close(1);
108     close(2);
109   }
110
111   // Set up signal handling ------------------------------------------------------------------------------------------
112
113   sighandler_t sigtest;
114
115   sigtest = signal(SIGPIPE, SIG_IGN);
116   if (sigtest == SIG_ERR)
117   {
118     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGPIPE. Aborting.");
119     shutdown(1);
120   }
121   sigtest = signal(SIGINT, sighandler);
122   if (sigtest == SIG_ERR)
123   {
124     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGINT. Aborting.");
125     shutdown(1);
126   }
127   sigtest = signal(SIGTERM, sighandler);
128   if (sigtest == SIG_ERR)
129   {
130     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGTERM. Aborting.");
131     shutdown(1);
132   }
133   sigtest = signal(SIGUSR1, sighandler);
134   if (sigtest == SIG_ERR)
135   {
136     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGUSR1. Aborting.");
137     shutdown(1);
138   }
139   sigtest = signal(SIGUSR2, sighandler);
140   if (sigtest == SIG_ERR)
141   {
142     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGUSR2. Aborting.");
143     shutdown(1);
144   }
145   sigtest = signal(SIGURG, sighandler);
146   if (sigtest == SIG_ERR)
147   {
148     logger->log("Core", Log::EMERG, "Could not set up signal handler for SIGURG. Aborting.");
149     shutdown(1);
150   }
151
152   logger->log("Core", Log::INFO, "Signal handlers set up successfully");
153
154
155   // Init modules ----------------------------------------------------------------------------------------------------
156   int success;
157
158   success = remote->init("/dev/rawir");
159   if (success)
160   {
161     logger->log("Core", Log::INFO, "Remote module initialised");
162   }
163   else
164   {
165     logger->log("Core", Log::EMERG, "Remote module failed to initialise");
166     shutdown(1);
167   }
168
169   success = led->init(remote->getDevice());
170   if (success)
171   {
172     logger->log("Core", Log::INFO, "LED module initialised");
173   }
174   else
175   {
176     logger->log("Core", Log::EMERG, "LED module failed to initialise");
177     shutdown(1);
178   }
179
180   success = mtd->init("/dev/mtd1");
181   if (success)
182   {
183     logger->log("Core", Log::INFO, "Mtd module initialised");
184   }
185   else
186   {
187     logger->log("Core", Log::EMERG, "Mtd module failed to initialise");
188     shutdown(1);
189   }
190
191   UCHAR videoFormat = (UCHAR)mtd->getPALorNTSC();
192   if      (videoFormat == Video::PAL)  logger->log("Core", Log::INFO, "Read from MTD: PAL 720x576");
193   else if (videoFormat == Video::NTSC) logger->log("Core", Log::INFO, "Read from MTD: NTSC 720x480");
194   else                                 logger->log("Core", Log::INFO, "No help from MTD. Assuming NTSC 720x480");
195
196   success = video->init(videoFormat);
197   if (success)
198   {
199     logger->log("Core", Log::INFO, "Video module initialised");
200   }
201   else
202   {
203     logger->log("Core", Log::EMERG, "Video module failed to initialise");
204     shutdown(1);
205   }
206
207   success = osd->init("/dev/stbgfx", 1);
208   if (success)
209   {
210     logger->log("Core", Log::INFO, "OSD module initialised");
211   }
212   else
213   {
214     logger->log("Core", Log::EMERG, "OSD module failed to initialise");
215     shutdown(1);
216   }
217
218   success = audio->init(Audio::MPEG2_PES);
219   if (success)
220   {
221     logger->log("Core", Log::INFO, "Audio module initialised");
222   }
223   else
224   {
225     logger->log("Core", Log::EMERG, "Audio module failed to initialise");
226     shutdown(1);
227   }
228
229   success = vdr->init(3024);
230   if (success)
231   {
232     logger->log("Core", Log::INFO, "VDR module initialised");
233   }
234   else
235   {
236     logger->log("Core", Log::EMERG, "VDR module failed to initialise");
237     shutdown(1);
238   }
239
240   success = viewman->init();
241   if (success)
242   {
243     logger->log("Core", Log::INFO, "ViewMan module initialised");
244   }
245   else
246   {
247     logger->log("Core", Log::EMERG, "ViewMan module failed to initialise");
248     shutdown(1);
249   }
250
251   success = command->init();
252   if (success)
253   {
254     logger->log("Core", Log::INFO, "Command module initialised");
255   }
256   else
257   {
258     logger->log("Core", Log::EMERG, "Command module failed to initialise");
259     shutdown(1);
260   }
261
262   // Other init ------------------------------------------------------------------------------------------------------
263
264   logger->log("Core", Log::NOTICE, "Startup successful");
265
266   // Run main loop ---------------------------------------------------------------------------------------------------
267
268   // Ok, all major device components and other bits are loaded and ready
269   command->run();
270
271   // When that returns quit ------------------------------------------------------------------------------------------
272
273   shutdown(0);
274   return 0;
275 }
276
277 // -------------------------------------------------------------------------------------------------------------------
278
279 void shutdown(int code)
280 {
281   if (command)
282   {
283     command->shutdown();
284     delete command;
285     logger->log("Core", Log::NOTICE, "Command module shut down");
286   }
287
288   if (viewman)
289   {
290     viewman->shutdown();
291     delete viewman;
292     logger->log("Core", Log::NOTICE, "ViewMan module shut down");
293   }
294
295   if (vdr)
296   {
297     vdr->shutdown();
298     delete vdr;
299     logger->log("Core", Log::NOTICE, "VDR module shut down");
300   }
301
302   if (osd)
303   {
304     osd->shutdown();
305     delete osd;
306     logger->log("Core", Log::NOTICE, "OSD module shut down");
307   }
308
309   if (audio)
310   {
311     audio->shutdown();
312     delete audio;
313     logger->log("Core", Log::NOTICE, "Audio module shut down");
314   }
315
316   if (video)
317   {
318     video->shutdown();
319     delete video;
320     logger->log("Core", Log::NOTICE, "Video module shut down");
321   }
322
323   if (mtd)
324   {
325     mtd->shutdown();
326     delete mtd;
327     logger->log("Core", Log::NOTICE, "MTD module shut down");
328   }
329
330   if (led)
331   {
332     led->shutdown();
333     delete led;
334     logger->log("Core", Log::NOTICE, "LED module shut down");
335   }
336
337   if (remote)
338   {
339     remote->shutdown();
340     delete remote;
341     logger->log("Core", Log::NOTICE, "Remote module shut down");
342   }
343
344   if (logger)
345   {
346     logger->log("Core", Log::NOTICE, "Log module shutting down... bye!\n\n");
347     logger->shutdown();
348     delete logger;
349   }
350
351   exit(code);
352 }
353
354 // -------------------------------------------------------------------------------------------------------------------
355
356 void sighandler(int signalReceived)
357 {
358   logger->log("Core", Log::NOTICE, "Signal %i received", signal);
359
360   switch (signalReceived)
361   {
362     case SIGINT:
363     {
364       logger->log("Core", Log::NOTICE, "Interrupt signal, shutting down...");
365       command->stop(); // FIXME this is probably not safe - use the messaging system / is that even safe?
366       break;
367     }
368     case SIGTERM:
369     {
370       logger->log("Core", Log::NOTICE, "Term signal, shutting down...");
371       command->stop(); // FIXME this is probably not safe - use the messaging system / is that even safe?
372       break;
373     }
374     case SIGUSR1:
375     {
376       logger->log("Core", Log::DEBUG, "SIGUSR1 caught");
377       logger->upLogLevel();
378       break;
379     }
380     case SIGUSR2:
381     {
382       logger->log("Core", Log::DEBUG, "SIGUSR2 caught");
383       logger->downLogLevel();
384       break;
385     }
386     case SIGURG:
387     {
388       logger->log("Core", Log::DEBUG, "SIGURG caught");
389       break;
390     }
391   }
392 }
393
394 // -------------------------------------------------------------------------------------------------------------------
395
396 ULLONG ntohll(ULLONG a)
397 {
398   return htonll(a);
399 }
400
401 ULLONG htonll(ULLONG a)
402 {
403   #if BYTE_ORDER == BIG_ENDIAN
404     return a;
405   #else
406     ULLONG b = 0;
407
408     b = ((a << 56) & 0xFF00000000000000ULL)
409       | ((a << 40) & 0x00FF000000000000ULL)
410       | ((a << 24) & 0x0000FF0000000000ULL)
411       | ((a <<  8) & 0x000000FF00000000ULL)
412       | ((a >>  8) & 0x00000000FF000000ULL)
413       | ((a >> 24) & 0x0000000000FF0000ULL)
414       | ((a >> 40) & 0x000000000000FF00ULL)
415       | ((a >> 56) & 0x00000000000000FFULL) ;
416
417     return b;
418   #endif
419 }