]> git.sesse.net Git - vlc/blob - bin/winvlc.c
Detect dangerous use of environment variables at run-time
[vlc] / bin / winvlc.c
1 /*****************************************************************************
2  * winvlc.c: the Windows VLC player
3  *****************************************************************************
4  * Copyright (C) 1998-2008 the VideoLAN team
5  *
6  * Authors: Vincent Seguin <seguin@via.ecp.fr>
7  *          Samuel Hocevar <sam@zoy.org>
8  *          Gildas Bazin <gbazin@videolan.org>
9  *          Derk-Jan Hartman <hartman at videolan dot org>
10  *          Lots of other people, see the libvlc AUTHORS file
11  *
12  * This program is free software; you can redistribute it and/or modify
13  * it under the terms of the GNU General Public License as published by
14  * the Free Software Foundation; either version 2 of the License, or
15  * (at your option) any later version.
16  *
17  * This program is distributed in the hope that it will be useful,
18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20  * GNU General Public License for more details.
21  *
22  * You should have received a copy of the GNU General Public License
23  * along with this program; if not, write to the Free Software
24  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
25  *****************************************************************************/
26
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #define UNICODE
32 #include <vlc/vlc.h>
33 #include <string.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <windows.h>
37
38 #if !defined(UNDER_CE)
39 # ifndef _WIN32_IE
40 #   define  _WIN32_IE 0x501
41 # endif
42 #   include <shlobj.h>
43 #   include <tlhelp32.h>
44 #   include <wininet.h>
45 # ifndef _WIN64
46 static void check_crashdump(void);
47 LONG WINAPI vlc_exception_filter(struct _EXCEPTION_POINTERS *lpExceptionInfo);
48 # endif
49 typedef enum _HEAP_INFORMATION_CLASS {
50         HeapCompatibilityInformation,
51         HeapEnableTerminationOnCorruption
52 } HEAP_INFORMATION_CLASS;
53 WINBASEAPI BOOL WINAPI HeapSetInformation(HANDLE,HEAP_INFORMATION_CLASS,PVOID,SIZE_T);
54 #define HeapEnableTerminationOnCorruption (HEAP_INFORMATION_CLASS)1
55 #endif
56
57 #ifndef UNDER_CE
58 static char *FromWide (const wchar_t *wide)
59 {
60     size_t len;
61     len = WideCharToMultiByte (CP_UTF8, 0, wide, -1, NULL, 0, NULL, NULL);
62
63     char *out = (char *)malloc (len);
64     if (out)
65         WideCharToMultiByte (CP_UTF8, 0, wide, -1, out, len, NULL, NULL);
66     return out;
67 }
68 #else
69 static int parse_cmdline (char *line, char ***argvp)
70 {
71     char **argv = malloc (sizeof (char *));
72     int argc = 0;
73
74     while (*line != '\0')
75     {
76         char quote = 0;
77
78         /* Skips white spaces */
79         while (strchr ("\t ", *line))
80             line++;
81         if (!*line)
82             break;
83
84         /* Starts a new parameter */
85         argv = realloc (argv, (argc + 2) * sizeof (char *));
86         if (*line == '"')
87         {
88             quote = '"';
89             line++;
90         }
91         argv[argc++] = line;
92
93     more:
94             while (*line && !strchr ("\t ", *line))
95             line++;
96
97     if (line > argv[argc - 1] && line[-1] == quote)
98         /* End of quoted parameter */
99         line[-1] = 0;
100     else
101         if (*line && quote)
102     {
103         /* Space within a quote */
104         line++;
105         goto more;
106     }
107     else
108         /* End of unquoted parameter */
109         if (*line)
110             *line++ = 0;
111     }
112     argv[argc] = NULL;
113     *argvp = argv;
114     return argc;
115 }
116 #endif
117
118 int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance,
119 #ifndef UNDER_CE
120                     LPSTR lpCmdLine,
121 #else
122                     LPWSTR lpCmdLine,
123 #endif
124                     int nCmdShow )
125 {
126     int argc;
127 #ifndef UNDER_CE
128     HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
129     wchar_t **wargv = CommandLineToArgvW (GetCommandLine (), &argc);
130     if (wargv == NULL)
131         return 1;
132
133     char *argv[argc + 1];
134     BOOL crash_handling = TRUE;
135     int j = 0;
136
137     argv[j++] = FromWide( L"--no-ignore-config" );
138     for (int i = 1; i < argc; i++)
139     {
140         if(!wcscmp(wargv[i], L"--no-crashdump"))
141         {
142             crash_handling = FALSE;
143         }
144         else
145         {
146             argv[j] = FromWide (wargv[i]);
147             j++;
148         }
149     }
150
151     argc = j;
152     argv[argc] = NULL;
153     LocalFree (wargv);
154
155 # ifndef _WIN64
156     if(crash_handling)
157     {
158         check_crashdump();
159         SetUnhandledExceptionFilter(vlc_exception_filter);
160     }
161 # endif /* WIN64 */
162
163 #else
164     char **argv, psz_cmdline[wcslen(lpCmdLine) * 4];
165
166     WideCharToMultiByte( CP_UTF8, 0, lpCmdLine, -1,
167                          psz_cmdline, sizeof (psz_cmdline), NULL, NULL );
168
169     argc = parse_cmdline (psz_cmdline, &argv);
170 #endif
171
172     /* Initialize libvlc */
173     libvlc_instance_t *vlc;
174     vlc = libvlc_new (argc, (const char **)argv);
175     if (vlc != NULL)
176     {
177         libvlc_add_intf (vlc, "globalhotkeys,none");
178         libvlc_add_intf (vlc, NULL);
179         libvlc_playlist_play (vlc, -1, 0, NULL);
180         libvlc_wait (vlc);
181         libvlc_release (vlc);
182     }
183
184     for (int i = 0; i < argc; i++)
185         free (argv[i]);
186
187     (void)hInstance; (void)hPrevInstance; (void)lpCmdLine; (void)nCmdShow;
188     return 0;
189 }
190
191 #if !defined( UNDER_CE ) && !defined( _WIN64 )
192
193 static void get_crashdump_path(wchar_t * wdir)
194 {
195     if( S_OK != SHGetFolderPathW( NULL,
196                         CSIDL_APPDATA | CSIDL_FLAG_CREATE,
197                         NULL, SHGFP_TYPE_CURRENT, wdir ) )
198         fprintf( stderr, "Can't open the vlc conf PATH\n" );
199
200     swprintf( wdir+wcslen( wdir ), L"%s", L"\\vlc\\crashdump" );
201 }
202
203 static void check_crashdump()
204 {
205     wchar_t * wdir = (wchar_t *)malloc(sizeof(wchar_t)*MAX_PATH);
206     get_crashdump_path(wdir);
207
208     FILE * fd = _wfopen ( wdir, L"r, ccs=UTF-8" );
209     if( fd )
210     {
211         fclose( fd );
212         int answer = MessageBox( NULL, L"VLC media player just crashed." \
213         " Do you want to send a bug report to the developers team?",
214         L"VLC crash reporting", MB_YESNO);
215
216         if(answer == IDYES)
217         {
218             HINTERNET Hint = InternetOpen(L"VLC Crash Reporter", INTERNET_OPEN_TYPE_PRECONFIG, NULL,NULL,0);
219             if(Hint)
220             {
221                 HINTERNET ftp = InternetConnect(Hint, L"crash.videolan.org", INTERNET_DEFAULT_FTP_PORT,
222                                                 NULL, NULL, INTERNET_SERVICE_FTP, 0, 0);
223                 if(ftp)
224                 {
225                     SYSTEMTIME now;
226                     GetSystemTime(&now);
227                     wchar_t remote_file[MAX_PATH];
228                     swprintf( remote_file, L"/crashs/%04d%02d%02d%02d%02d%02d",now.wYear,
229                             now.wMonth, now.wDay, now.wHour, now.wMinute, now.wSecond  );
230
231                     FtpPutFile( ftp, wdir, remote_file, FTP_TRANSFER_TYPE_BINARY, 0);
232                     InternetCloseHandle(ftp);
233                 }
234                 else
235                     fprintf(stderr,"Can't connect to FTP server%d\n",GetLastError());
236                 InternetCloseHandle(Hint);
237             }
238         }
239
240         _wremove(wdir);
241     }
242     free((void *)wdir);
243 }
244
245 /*****************************************************************************
246  * vlc_exception_filter: handles unhandled exceptions, like segfaults
247  *****************************************************************************/
248 LONG WINAPI vlc_exception_filter(struct _EXCEPTION_POINTERS *lpExceptionInfo)
249 {
250     if(IsDebuggerPresent())
251     {
252         //If a debugger is present, pass the exception to the debugger with EXCEPTION_CONTINUE_SEARCH
253         return EXCEPTION_CONTINUE_SEARCH;
254     }
255     else
256     {
257         fprintf( stderr, "unhandled vlc exception\n" );
258
259         wchar_t * wdir = (wchar_t *)malloc(sizeof(wchar_t)*MAX_PATH);
260         get_crashdump_path(wdir);
261         FILE * fd = _wfopen ( wdir, L"w, ccs=UTF-8" );
262         free((void *)wdir);
263
264         if( !fd )
265         {
266             fprintf( stderr, "\nerror while opening file" );
267             exit( 1 );
268         }
269
270         OSVERSIONINFO osvi;
271         ZeroMemory( &osvi, sizeof(OSVERSIONINFO) );
272         osvi.dwOSVersionInfoSize = sizeof( OSVERSIONINFO );
273         GetVersionEx( &osvi );
274
275         fwprintf( fd, L"[version]\nOS=%d.%d.%d.%d.%s\nVLC=" VERSION_MESSAGE, osvi.dwMajorVersion,
276                                                                osvi.dwMinorVersion,
277                                                                osvi.dwBuildNumber,
278                                                                osvi.dwPlatformId,
279                                                                osvi.szCSDVersion);
280
281         const CONTEXT *const pContext = (const CONTEXT *)lpExceptionInfo->ContextRecord;
282         const EXCEPTION_RECORD *const pException = (const EXCEPTION_RECORD *)lpExceptionInfo->ExceptionRecord;
283         /*No nested exceptions for now*/
284         fwprintf( fd, L"\n\n[exceptions]\n%08x at %08x",pException->ExceptionCode,
285                                                 pException->ExceptionAddress );
286         if( pException->NumberParameters > 0 )
287         {
288             unsigned int i;
289             for( i = 0; i < pException->NumberParameters; i++ )
290                 fprintf( fd, " | %08x", pException->ExceptionInformation[i] );
291         }
292
293         fwprintf( fd, L"\n\n[context]\nEDI:%08x\nESI:%08x\n" \
294                     "EBX:%08x\nEDX:%08x\nECX:%08x\nEAX:%08x\n" \
295                     "EBP:%08x\nEIP:%08x\nESP:%08x\n",
296                         pContext->Edi,pContext->Esi,pContext->Ebx,
297                         pContext->Edx,pContext->Ecx,pContext->Eax,
298                         pContext->Ebp,pContext->Eip,pContext->Esp );
299
300         fwprintf( fd, L"\n[stacktrace]\n#EIP|base|module\n" );
301
302         wchar_t module[ 256 ];
303         MEMORY_BASIC_INFORMATION mbi ;
304         VirtualQuery( (DWORD *)pContext->Eip, &mbi, sizeof( mbi ) ) ;
305         HINSTANCE hInstance = mbi.AllocationBase;
306         GetModuleFileName( hInstance, module, 256 ) ;
307         fwprintf( fd, L"%08x|%s\n", pContext->Eip, module );
308
309         DWORD pEbp = pContext->Ebp;
310         DWORD caller = *((DWORD*)pEbp + 1);
311
312         do
313         {
314             VirtualQuery( (DWORD *)caller, &mbi, sizeof( mbi ) ) ;
315             HINSTANCE hInstance = mbi.AllocationBase;
316             GetModuleFileName( hInstance, module, 256 ) ;
317             fwprintf( fd, L"%08x|%s\n", caller, module );
318             pEbp = *(DWORD*)pEbp ;
319             caller = *((DWORD*)pEbp + 1) ;
320             /*The last EBP points to NULL!*/
321         }while(caller);
322
323         fclose( fd );
324         fflush( stderr );
325         exit( 1 );
326     }
327 }
328 #endif