Fix abortion if killing signals are re-used.
[vlc.git] / src / vlc.c
1 /*****************************************************************************
2  * vlc.c: the vlc player
3  *****************************************************************************
4  * Copyright (C) 1998-2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Vincent Seguin <seguin@via.ecp.fr>
8  *          Samuel Hocevar <sam@zoy.org>
9  *          Gildas Bazin <gbazin@videolan.org>
10  *          Derk-Jan Hartman <hartman at videolan dot org>
11  *          Lots of other people, see the libvlc AUTHORS file
12  *
13  * This program is free software; you can redistribute it and/or modify
14  * it under the terms of the GNU General Public License as published by
15  * the Free Software Foundation; either version 2 of the License, or
16  * (at your option) any later version.
17  *
18  * This program is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
26  *****************************************************************************/
27
28 #include "config.h"
29
30 #include <vlc/vlc.h>
31 #include <stdio.h>                                              /* fprintf() */
32 #include <stdlib.h>                                  /* putenv(), strtol(),  */
33
34
35 /*****************************************************************************
36  * Local prototypes.
37  *****************************************************************************/
38 #ifdef WIN32
39 #include <windows.h>
40 extern void __wgetmainargs(int *argc, wchar_t ***wargv, wchar_t ***wenviron,
41                            int expand_wildcards, int *startupinfo);
42 #else
43
44 # include <signal.h>
45 # include <time.h>
46 # include <pthread.h>
47 # include <stdbool.h>
48
49 static struct
50 {
51     bool flag;
52     pthread_mutex_t lock;
53 } live = { true, PTHREAD_MUTEX_INITIALIZER };
54
55 static void *SigHandler (void *set);
56 #endif
57
58 /*****************************************************************************
59  * main: parse command line, start interface and spawn threads.
60  *****************************************************************************/
61 int main( int i_argc, char *ppsz_argv[] )
62 {
63     int i_ret;
64
65 #ifndef __APPLE__
66     /* This clutters OSX GUI error logs */
67     fprintf( stderr, "VLC media player %s\n", VLC_Version() );
68 #endif
69
70 #ifdef HAVE_PUTENV
71 #   ifdef DEBUG
72     /* Activate malloc checking routines to detect heap corruptions. */
73     putenv( (char*)"MALLOC_CHECK_=2" );
74
75     /* Disable the ugly Gnome crash dialog so that we properly segfault */
76     putenv( (char *)"GNOME_DISABLE_CRASH_DIALOG=1" );
77 #   endif
78
79     /* If the user isn't using VLC_VERBOSE, set it to 0 by default */
80     if( getenv( "VLC_VERBOSE" ) == NULL )
81     {
82         putenv( (char *)"VLC_VERBOSE=0" );
83     }
84 #endif
85
86 #if defined (HAVE_GETEUID) && !defined (SYS_BEOS)
87     /* FIXME: rootwrap (); */
88 #endif
89
90     /* Create a libvlc structure */
91     i_ret = VLC_Create();
92     if( i_ret < 0 )
93     {
94         return i_ret;
95     }
96
97 #if !defined(WIN32) && !defined(UNDER_CE)
98     /* Synchronously intercepted signals. Thy request a clean shutdown,
99      * and force an unclean shutdown if they are triggered again 2+ seconds
100      * later. We have to handle SIGTERM cleanly because of daemon mode.
101      * Note that we set the signals after the vlc_create call. */
102     static const int sigs[] = { SIGINT, SIGHUP, SIGQUIT, SIGTERM };
103     /* Ignored signals */
104     static const int ignored[] = { SIGALRM, SIGPIPE };
105
106     sigset_t set;
107     pthread_t sigth;
108
109     sigemptyset (&set);
110     for (unsigned i = 0; i < sizeof (sigs) / sizeof (sigs[0]); i++)
111         sigaddset (&set, sigs[i]);
112     for (unsigned i = 0; i < sizeof (ignored) / sizeof (ignored[0]); i++)
113         sigaddset (&set, ignored[i]);
114
115     /* Block all these signals */
116     pthread_sigmask (SIG_BLOCK, &set, NULL);
117
118     for (unsigned i = 0; i < sizeof (ignored) / sizeof (ignored[0]); i++)
119         sigdelset (&set, ignored[i]);
120
121     pthread_create (&sigth, NULL, SigHandler, &set);
122 #endif
123
124 #ifdef WIN32
125     /* Replace argv[1..n] with unicode for Windows NT and above */
126     if( GetVersion() < 0x80000000 )
127     {
128         wchar_t **wargv, **wenvp;
129         int i,i_wargc;
130         int si = { 0 };
131         __wgetmainargs(&i_wargc, &wargv, &wenvp, 0, &si);
132
133         for( i = 1; i < i_wargc; i++ )
134         {
135             int len = WideCharToMultiByte(CP_UTF8, 0, wargv[i], -1, NULL, 0, NULL, NULL);
136             if( len > 0 )
137             {
138                 if( len > 1 ) {
139                     char *utf8arg = (char *)malloc(len);
140                     if( NULL != utf8arg )
141                     {
142                         WideCharToMultiByte(CP_UTF8, 0, wargv[i], -1, utf8arg, len, NULL, NULL);
143                         ppsz_argv[i] = utf8arg;
144                     }
145                     else
146                     {
147                         /* failed!, quit */
148                         return -1;
149                     }
150                 }
151                 else
152                 {
153                     ppsz_argv[i] = "";
154                 }
155             }
156             else
157             {
158                 /* failed!, quit */
159                 return -1;
160             }
161         }
162     }
163 #endif
164
165     /* Initialize libvlc */
166     i_ret = VLC_Init( 0, i_argc, ppsz_argv );
167     if( i_ret < 0 )
168     {
169         VLC_Destroy( 0 );
170         return i_ret == VLC_EEXITSUCCESS ? 0 : i_ret;
171     }
172
173     i_ret = VLC_AddIntf( 0, NULL, VLC_TRUE, VLC_TRUE );
174
175     /* Finish the threads */
176     VLC_CleanUp( 0 );
177
178     /* Destroy the libvlc structure */
179     VLC_Destroy( 0 );
180
181 #if !defined(WIN32) && !defined(UNDER_CE)
182     pthread_cancel (sigth);
183     pthread_join (sigth, NULL);
184 #endif
185
186     return i_ret;
187 }
188
189 #if !defined(WIN32) && !defined(UNDER_CE)
190 /*****************************************************************************
191  * SigHandler: system signal handler
192  *****************************************************************************
193  * This thread receives all handled signals synchronously.
194  * It tries to end the program in a clean way.
195  *****************************************************************************/
196 static void *SigHandler (void *data)
197 {
198     const sigset_t *set = (sigset_t *)data;
199     time_t abort_time = 0;
200     vlc_bool_t b_die = VLC_FALSE;
201
202     for (;;)
203     {
204         int i_signal, state;
205         (void)sigwait (set, &i_signal);
206
207         /* Once a signal has been trapped, the termination sequence will be
208          * armed and subsequent signals will be ignored to avoid sending
209          * signals to a libvlc structure having been destroyed */
210
211         pthread_setcancelstate (PTHREAD_CANCEL_DISABLE, &state);
212         if (!b_die)
213         {
214             b_die = VLC_TRUE;
215             abort_time = time (NULL);
216
217             fprintf (stderr, "signal %d received, terminating vlc - do it "
218                             "again in case it gets stuck\n", i_signal);
219
220             /* Acknowledge the signal received */
221             pthread_mutex_lock (&live.lock);
222             if (live.flag)
223             {
224                 VLC_Die (0);
225                 live.flag = false;
226             }
227             pthread_mutex_unlock (&live.lock);
228         }
229         else if( time( NULL ) > abort_time + 2 )
230         {
231             /* If user asks again 1 or 2 seconds later, die badly */
232             pthread_sigmask (SIG_UNBLOCK, set, NULL);
233             fprintf (stderr, "user insisted too much, dying badly\n");
234             abort ();
235         }
236         pthread_setcancelstate (state, NULL);
237     }
238     /* Never reached */
239 }
240 #endif
241
242 #if defined(UNDER_CE)
243 #   if defined( _MSC_VER ) && defined( UNDER_CE )
244 #       include "vlc_common.h"
245 #   endif
246 /*****************************************************************************
247  * WinMain: parse command line, start interface and spawn threads. (WinCE only)
248  *****************************************************************************/
249 int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,
250                     LPTSTR lpCmdLine, int nCmdShow )
251 {
252     char **argv, psz_cmdline[MAX_PATH];
253     int argc, i_ret;
254
255     WideCharToMultiByte( CP_ACP, 0, lpCmdLine, -1,
256                          psz_cmdline, MAX_PATH, NULL, NULL );
257
258     argv = vlc_parse_cmdline( psz_cmdline, &argc );
259     argv = realloc( argv, (argc + 1) * sizeof(char *) );
260     if( !argv ) return -1;
261
262     if( argc ) memmove( argv + 1, argv, argc * sizeof(char *) );
263     argv[0] = ""; /* Fake program path */
264
265     i_ret = main( argc + 1, argv );
266
267     /* No need to free the argv memory */
268     return i_ret;
269 }
270 #endif