git.sesse.net Git - vlc/blob - src/misc/win32_specific.c

   1 /*****************************************************************************
   2  * win32_specific.c: Win32 specific features
   3  *****************************************************************************
   4  * Copyright (C) 2001-2004 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Samuel Hocevar <sam@zoy.org>
   8  *          Gildas Bazin <gbazin@videolan.org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 #ifdef HAVE_CONFIG_H
  26 # include "config.h"
  27 #endif
  28
  29 #define UNICODE
  30 #include <vlc_common.h>
  31 #include "../libvlc.h"
  32 #include <vlc_playlist.h>
  33 #include <vlc_charset.h>
  34
  35 #include "../extras/getopt.h"
  36
  37 #if !defined( UNDER_CE )
  38 #   include <io.h>
  39 #   include <fcntl.h>
  40 #   include  <mmsystem.h>
  41 #endif
  42
  43 #include <winsock.h>
  44
  45 /*****************************************************************************
  46  * system_Init: initialize winsock and misc other things.
  47  *****************************************************************************/
  48 void system_Init( libvlc_int_t *p_this, int *pi_argc, const char *ppsz_argv[] )
  49 {
  50     WSADATA Data;
  51
  52     /* Get our full path */
  53     char psz_path[MAX_PATH];
  54     char *psz_vlc;
  55
  56     wchar_t psz_wpath[MAX_PATH];
  57     if( GetModuleFileName( NULL, psz_wpath, MAX_PATH ) )
  58     {
  59         WideCharToMultiByte( CP_ACP, 0, psz_wpath, -1,
  60                              psz_path, MAX_PATH, NULL, NULL );
  61     }
  62     else psz_path[0] = '\0';
  63
  64     if( (psz_vlc = strrchr( psz_path, '\\' )) ) *psz_vlc = '\0';
  65
  66 #ifndef HAVE_RELEASE
  67     {
  68         /* remove trailing \.libs from executable dir path if seen,
  69            we assume we are running vlc through libtool wrapper in build dir */
  70         int offset  = strlen(psz_path)-sizeof("\\.libs")+1;
  71         if( offset > 0 )
  72         {
  73             psz_vlc = psz_path+offset;
  74             if( ! strcmp(psz_vlc, "\\.libs") ) *psz_vlc = '\0';
  75         }
  76     }
  77 #endif
  78
  79     vlc_global()->psz_vlcpath = strdup( psz_path );
  80
  81     /* Set the default file-translation mode */
  82 #if !defined( UNDER_CE )
  83     _fmode = _O_BINARY;
  84     _setmode( _fileno( stdin ), _O_BINARY ); /* Needed for pipes */
  85
  86     timeBeginPeriod(5);
  87 #endif
  88
  89     /* Call mdate() once to make sure it is initialized properly */
  90     mdate();
  91
  92     /* WinSock Library Init. */
  93     if( !WSAStartup( MAKEWORD( 2, 2 ), &Data ) )
  94     {
  95         /* Aah, pretty useless check, we should always have Winsock 2.2
  96          * since it appeared in Win98. */
  97         if( LOBYTE( Data.wVersion ) != 2 || HIBYTE( Data.wVersion ) != 2 )
  98             /* We could not find a suitable WinSock DLL. */
  99             WSACleanup( );
 100         else
 101             /* Everything went ok. */
 102             return;
 103     }
 104
 105     /* Let's try with WinSock 1.1 */
 106     if( !WSAStartup( MAKEWORD( 1, 1 ), &Data ) )
 107     {
 108         /* Confirm that the WinSock DLL supports 1.1.*/
 109         if( LOBYTE( Data.wVersion ) != 1 || HIBYTE( Data.wVersion ) != 1 )
 110             /* We could not find a suitable WinSock DLL. */
 111             WSACleanup( );
 112         else
 113             /* Everything went ok. */
 114             return;
 115     }
 116
 117     fprintf( stderr, "error: can't initialize WinSocks\n" );
 118 }
 119
 120 /*****************************************************************************
 121  * system_Configure: check for system specific configuration options.
 122  *****************************************************************************/
 123 static unsigned __stdcall IPCHelperThread( void * );
 124 LRESULT CALLBACK WMCOPYWNDPROC( HWND, UINT, WPARAM, LPARAM );
 125 static vlc_object_t *p_helper = NULL;
 126 static unsigned long hIPCHelper;
 127 static HANDLE hIPCHelperReady;
 128
 129 typedef struct
 130 {
 131   int argc;
 132   int enqueue;
 133   char data[];
 134 } vlc_ipc_data_t;
 135
 136 void system_Configure( libvlc_int_t *p_this, int *pi_argc, const char *ppsz_argv[] )
 137 {
 138 #if !defined( UNDER_CE )
 139     /* Raise default priority of the current process */
 140 #ifndef ABOVE_NORMAL_PRIORITY_CLASS
 141 #   define ABOVE_NORMAL_PRIORITY_CLASS 0x00008000
 142 #endif
 143     if( config_GetInt( p_this, "high-priority" ) )
 144     {
 145         if( SetPriorityClass( GetCurrentProcess(), ABOVE_NORMAL_PRIORITY_CLASS )
 146              || SetPriorityClass( GetCurrentProcess(), HIGH_PRIORITY_CLASS ) )
 147         {
 148             msg_Dbg( p_this, "raised process priority" );
 149         }
 150         else
 151         {
 152             msg_Dbg( p_this, "could not raise process priority" );
 153         }
 154     }
 155
 156     if( config_GetInt( p_this, "one-instance" )
 157         || ( config_GetInt( p_this, "one-instance-when-started-from-file" )
 158              && config_GetInt( p_this, "started-from-file" ) ) )
 159     {
 160         HANDLE hmutex;
 161
 162         msg_Info( p_this, "one instance mode ENABLED");
 163
 164         /* Use a named mutex to check if another instance is already running */
 165         if( !( hmutex = CreateMutex( 0, TRUE, L"VLC ipc "VERSION ) ) )
 166         {
 167             /* Failed for some reason. Just ignore the option and go on as
 168              * normal. */
 169             msg_Err( p_this, "one instance mode DISABLED "
 170                      "(mutex couldn't be created)" );
 171             return;
 172         }
 173
 174         if( GetLastError() != ERROR_ALREADY_EXISTS )
 175         {
 176             /* We are the 1st instance. */
 177             static const char typename[] = "ipc helper";
 178             p_helper =
 179                 vlc_custom_create( p_this, sizeof(vlc_object_t),
 180                                    VLC_OBJECT_GENERIC, typename );
 181
 182             /* Run the helper thread */
 183             hIPCHelperReady = CreateEvent( NULL, FALSE, FALSE, NULL );
 184             hIPCHelper = _beginthreadex( NULL, 0, IPCHelperThread, p_helper,
 185                                          0, NULL );
 186             if( hIPCHelper )
 187                 WaitForSingleObject( hIPCHelperReady, INFINITE );
 188             else
 189             {
 190                 msg_Err( p_this, "one instance mode DISABLED "
 191                          "(IPC helper thread couldn't be created)" );
 192                 vlc_object_release (p_helper);
 193                 p_helper = NULL;
 194             }
 195             vlc_object_attach (p_helper, p_this);
 196             CloseHandle( hIPCHelperReady );
 197
 198             /* Initialization done.
 199              * Release the mutex to unblock other instances */
 200             ReleaseMutex( hmutex );
 201         }
 202         else
 203         {
 204             /* Another instance is running */
 205
 206             HWND ipcwindow;
 207
 208             /* Wait until the 1st instance is initialized */
 209             WaitForSingleObject( hmutex, INFINITE );
 210
 211             /* Locate the window created by the IPC helper thread of the
 212              * 1st instance */
 213             if( !( ipcwindow = FindWindow( 0, L"VLC ipc "VERSION ) ) )
 214             {
 215                 msg_Err( p_this, "one instance mode DISABLED "
 216                          "(couldn't find 1st instance of program)" );
 217                 ReleaseMutex( hmutex );
 218                 return;
 219             }
 220
 221             /* We assume that the remaining parameters are filenames
 222              * and their input options */
 223             if( *pi_argc - 1 >= optind )
 224             {
 225                 COPYDATASTRUCT wm_data;
 226                 int i_opt;
 227                 vlc_ipc_data_t *p_data;
 228                 size_t i_data = sizeof (*p_data);
 229
 230                 for( i_opt = optind; i_opt < *pi_argc; i_opt++ )
 231                 {
 232                     i_data += sizeof (size_t);
 233                     i_data += strlen( ppsz_argv[ i_opt ] ) + 1;
 234                 }
 235
 236                 p_data = malloc( i_data );
 237                 p_data->argc = *pi_argc - optind;
 238                 p_data->enqueue = config_GetInt( p_this, "playlist-enqueue" );
 239                 i_data = 0;
 240                 for( i_opt = optind; i_opt < *pi_argc; i_opt++ )
 241                 {
 242                     size_t i_len = strlen( ppsz_argv[ i_opt ] ) + 1;
 243                     /* Windows will never switch to an architecture
 244                      * with stronger alignment requirements, right. */
 245                     *((size_t *)(p_data->data + i_data)) = i_len;
 246                     i_data += sizeof (size_t);
 247                     memcpy( &p_data->data[i_data], ppsz_argv[ i_opt ], i_len );
 248                     i_data += i_len;
 249                 }
 250                 i_data += sizeof (*p_data);
 251
 252                 /* Send our playlist items to the 1st instance */
 253                 wm_data.dwData = 0;
 254                 wm_data.cbData = i_data;
 255                 wm_data.lpData = p_data;
 256                 SendMessage( ipcwindow, WM_COPYDATA, 0, (LPARAM)&wm_data );
 257             }
 258
 259             /* Initialization done.
 260              * Release the mutex to unblock other instances */
 261             ReleaseMutex( hmutex );
 262
 263             /* Bye bye */
 264             system_End( p_this );
 265             exit( 0 );
 266         }
 267     }
 268
 269 #endif
 270 }
 271
 272 static unsigned __stdcall IPCHelperThread( void *data )
 273 {
 274     vlc_object_t *p_this = data;
 275     HWND ipcwindow;
 276     MSG message;
 277
 278     ipcwindow =
 279         CreateWindow( L"STATIC",                     /* name of window class */
 280                   L"VLC ipc "VERSION,               /* window title bar text */
 281                   0,                                         /* window style */
 282                   0,                                 /* default X coordinate */
 283                   0,                                 /* default Y coordinate */
 284                   0,                                         /* window width */
 285                   0,                                        /* window height */
 286                   NULL,                                  /* no parent window */
 287                   NULL,                            /* no menu in this window */
 288                   GetModuleHandle(NULL),  /* handle of this program instance */
 289                   NULL );                               /* sent to WM_CREATE */
 290
 291     SetWindowLongPtr( ipcwindow, GWLP_WNDPROC, (LRESULT)WMCOPYWNDPROC );
 292     SetWindowLongPtr( ipcwindow, GWLP_USERDATA, (LONG_PTR)p_this );
 293
 294     /* Signal the creation of the thread and events queue */
 295     SetEvent( hIPCHelperReady );
 296
 297     while( GetMessage( &message, NULL, 0, 0 ) )
 298     {
 299         TranslateMessage( &message );
 300         DispatchMessage( &message );
 301     }
 302     return 0;
 303 }
 304
 305 LRESULT CALLBACK WMCOPYWNDPROC( HWND hwnd, UINT uMsg, WPARAM wParam,
 306                                 LPARAM lParam )
 307 {
 308     if( uMsg == WM_COPYDATA )
 309     {
 310         COPYDATASTRUCT *pwm_data = (COPYDATASTRUCT*)lParam;
 311         vlc_object_t *p_this;
 312         playlist_t *p_playlist;
 313
 314         p_this = (vlc_object_t *)
 315             (uintptr_t)GetWindowLongPtr( hwnd, GWLP_USERDATA );
 316
 317         if( !p_this ) return 0;
 318
 319         /* Add files to the playlist */
 320         p_playlist = pl_Hold( p_this );
 321         if( !p_playlist ) return 0;
 322
 323         if( pwm_data->lpData )
 324         {
 325             char **ppsz_argv;
 326             vlc_ipc_data_t *p_data = (vlc_ipc_data_t *)pwm_data->lpData;
 327             size_t i_data = 0;
 328             int i_argc = p_data->argc, i_opt, i_options;
 329
 330             ppsz_argv = (char **)malloc( i_argc * sizeof(char *) );
 331             for( i_opt = 0; i_opt < i_argc; i_opt++ )
 332             {
 333                 ppsz_argv[i_opt] = p_data->data + i_data + sizeof(int);
 334                 i_data += sizeof(int) + *((int *)(p_data->data + i_data));
 335             }
 336
 337             for( i_opt = 0; i_opt < i_argc; i_opt++ )
 338             {
 339                 i_options = 0;
 340
 341                 /* Count the input options */
 342                 while( i_opt + i_options + 1 < i_argc &&
 343                        *ppsz_argv[ i_opt + i_options + 1 ] == ':' )
 344                 {
 345                     i_options++;
 346                 }
 347                 playlist_AddExt( p_playlist, ppsz_argv[i_opt],
 348                   NULL, PLAYLIST_APPEND |
 349                         ( ( i_opt || p_data->enqueue ) ? 0 : PLAYLIST_GO ),
 350                   PLAYLIST_END, -1,
 351                   (char const **)( i_options ? &ppsz_argv[i_opt+1] : NULL ),
 352                   i_options, true, pl_Unlocked );
 353
 354                 i_opt += i_options;
 355             }
 356
 357             free( ppsz_argv );
 358         }
 359
 360         vlc_object_release( p_playlist );
 361     }
 362
 363     return DefWindowProc( hwnd, uMsg, wParam, lParam );
 364 }
 365
 366 /*****************************************************************************
 367  * system_End: terminate winsock.
 368  *****************************************************************************/
 369 void system_End( libvlc_int_t *p_this )
 370 {
 371     HWND ipcwindow;
 372     if( p_this && vlc_global() )
 373     {
 374         free( vlc_global()->psz_vlcpath );
 375         vlc_global()->psz_vlcpath = NULL;
 376     }
 377
 378     if( ipcwindow = FindWindow( 0, L"VLC ipc "VERSION ) )
 379     {
 380         SendMessage( ipcwindow, WM_QUIT, 0, 0 );
 381     }
 382
 383     if (p_helper && p_helper->p_parent == VLC_OBJECT(p_this) )
 384     {
 385         /* FIXME: thread-safety... */
 386         vlc_object_detach (p_helper);
 387         vlc_object_release (p_helper);
 388         p_helper = NULL;
 389     }
 390
 391 #if !defined( UNDER_CE )
 392     timeEndPeriod(5);
 393 #endif
 394
 395     WSACleanup();
 396 }