programs/start/start.c

   1 /*
   2  * Start a program using ShellExecuteEx, optionally wait for it to finish
   3  * Compatible with Microsoft's "c:\windows\command\start.exe"
   4  *
   5  * Copyright 2003 Dan Kegel
   6  * Copyright 2007 Lyutin Anatoly (Etersoft)
   7  *
   8  * This program is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License as published by the Free Software Foundation; either
  11  * version 2.1 of the License, or (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public
  19  * License along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  21  */
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <windows.h>
  26 #include <shlobj.h>
  27 #include <shellapi.h>
  28
  29 #include <wine/unicode.h>
  30 #include <wine/debug.h>
  31
  32 #include "resources.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(start);
  35
  36 /**
  37  Output given message to stdout without formatting.
  38 */
  39 static void output(const WCHAR *message)
  40 {
  41         DWORD count;
  42         DWORD   res;
  43         int    wlen = strlenW(message);
  44
  45         if (!wlen) return;
  46
  47         res = WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE), message, wlen, &count, NULL);
  48
  49         /* If writing to console fails, assume it's file
  50          * i/o so convert to OEM codepage and output
  51          */
  52         if (!res)
  53         {
  54                 DWORD len;
  55                 char  *mesA;
  56                 /* Convert to OEM, then output */
  57                 len = WideCharToMultiByte( GetConsoleOutputCP(), 0, message, wlen, NULL, 0, NULL, NULL );
  58                 mesA = HeapAlloc(GetProcessHeap(), 0, len*sizeof(char));
  59                 if (!mesA) return;
  60                 WideCharToMultiByte( GetConsoleOutputCP(), 0, message, wlen, mesA, len, NULL, NULL );
  61                 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), mesA, len, &count, FALSE);
  62                 HeapFree(GetProcessHeap(), 0, mesA);
  63         }
  64 }
  65
  66 /**
  67  Output given message from string table,
  68  followed by ": ",
  69  followed by description of given GetLastError() value to stdout,
  70  followed by a trailing newline,
  71  then terminate.
  72 */
  73
  74 static void fatal_error(const WCHAR *msg, DWORD error_code, const WCHAR *filename)
  75 {
  76     DWORD_PTR args[1];
  77     LPVOID lpMsgBuf;
  78     int status;
  79     static const WCHAR colonsW[] = { ':', ' ', 0 };
  80     static const WCHAR newlineW[] = { '\n', 0 };
  81
  82     output(msg);
  83     output(colonsW);
  84     args[0] = (DWORD_PTR)filename;
  85     status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ARGUMENT_ARRAY,
  86                             NULL, error_code, 0, (LPWSTR)&lpMsgBuf, 0, (__ms_va_list *)args );
  87     if (!status)
  88     {
  89         WINE_ERR("FormatMessage failed\n");
  90     } else
  91     {
  92         output(lpMsgBuf);
  93         LocalFree((HLOCAL) lpMsgBuf);
  94         output(newlineW);
  95     }
  96     ExitProcess(1);
  97 }
  98
  99 static void fatal_string_error(int which, DWORD error_code, const WCHAR *filename)
 100 {
 101         WCHAR msg[2048];
 102
 103         if (!LoadStringW(GetModuleHandleW(NULL), which, msg, ARRAY_SIZE(msg)))
 104                 WINE_ERR("LoadString failed, error %d\n", GetLastError());
 105
 106         fatal_error(msg, error_code, filename);
 107 }
 108
 109 static void fatal_string(int which)
 110 {
 111         WCHAR msg[2048];
 112
 113         if (!LoadStringW(GetModuleHandleW(NULL), which, msg, ARRAY_SIZE(msg)))
 114                 WINE_ERR("LoadString failed, error %d\n", GetLastError());
 115
 116         output(msg);
 117         ExitProcess(1);
 118 }
 119
 120 static void usage(void)
 121 {
 122         fatal_string(STRING_USAGE);
 123 }
 124
 125 static WCHAR *build_args( int argc, WCHAR **argvW )
 126 {
 127         int i, wlen = 1;
 128         WCHAR *ret, *p;
 129         static const WCHAR FormatQuotesW[] = { ' ', '\"', '%', 's', '\"', 0 };
 130         static const WCHAR FormatW[] = { ' ', '%', 's', 0 };
 131
 132         for (i = 0; i < argc; i++ )
 133         {
 134                 wlen += strlenW(argvW[i]) + 1;
 135                 if (strchrW(argvW[i], ' '))
 136                         wlen += 2;
 137         }
 138         ret = HeapAlloc( GetProcessHeap(), 0, wlen*sizeof(WCHAR) );
 139         ret[0] = 0;
 140
 141         for (i = 0, p = ret; i < argc; i++ )
 142         {
 143                 if (strchrW(argvW[i], ' '))
 144                         p += sprintfW(p, FormatQuotesW, argvW[i]);
 145                 else
 146                         p += sprintfW(p, FormatW, argvW[i]);
 147         }
 148         return ret;
 149 }
 150
 151 static WCHAR *get_parent_dir(WCHAR* path)
 152 {
 153         WCHAR *last_slash;
 154         WCHAR *result;
 155         int len;
 156
 157         last_slash = strrchrW( path, '\\' );
 158         if (last_slash == NULL)
 159                 len = 1;
 160         else
 161                 len = last_slash - path + 1;
 162
 163         result = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 164         CopyMemory(result, path, (len-1)*sizeof(WCHAR));
 165         result[len-1] = '\0';
 166
 167         return result;
 168 }
 169
 170 static BOOL is_option(const WCHAR* arg, const WCHAR* opt)
 171 {
 172     return CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE,
 173                           arg, -1, opt, -1) == CSTR_EQUAL;
 174 }
 175
 176 int wmain (int argc, WCHAR *argv[])
 177 {
 178         SHELLEXECUTEINFOW sei;
 179         DWORD creation_flags;
 180         WCHAR *args = NULL;
 181         int i;
 182         BOOL unix_mode = FALSE;
 183         BOOL progid_open = FALSE;
 184         WCHAR *title = NULL;
 185         WCHAR *dos_filename = NULL;
 186         WCHAR *parent_directory = NULL;
 187         DWORD binary_type;
 188
 189         static const WCHAR bW[] = { '/', 'b', 0 };
 190         static const WCHAR minW[] = { '/', 'm', 'i', 'n', 0 };
 191         static const WCHAR maxW[] = { '/', 'm', 'a', 'x', 0 };
 192         static const WCHAR lowW[] = { '/', 'l', 'o', 'w', 0 };
 193         static const WCHAR normalW[] = { '/', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
 194         static const WCHAR highW[] = { '/', 'h', 'i', 'g', 'h', 0 };
 195         static const WCHAR realtimeW[] = { '/', 'r', 'e', 'a', 'l', 't', 'i', 'm', 'e', 0 };
 196         static const WCHAR abovenormalW[] = { '/', 'a', 'b', 'o', 'v', 'e', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
 197         static const WCHAR belownormalW[] = { '/', 'b', 'e', 'l', 'o', 'w', 'n', 'o', 'r', 'm', 'a', 'l', 0 };
 198         static const WCHAR separateW[] = { '/', 's', 'e', 'p', 'a', 'r', 'a', 't', 'e', 0 };
 199         static const WCHAR sharedW[] = { '/', 's', 'h', 'a', 'r', 'e', 'd', 0 };
 200         static const WCHAR nodeW[] = { '/', 'n', 'o', 'd', 'e', 0 };
 201         static const WCHAR affinityW[] = { '/', 'a', 'f', 'f', 'i', 'n', 'i', 't', 'y', 0 };
 202         static const WCHAR wW[] = { '/', 'w', 0 };
 203         static const WCHAR waitW[] = { '/', 'w', 'a', 'i', 't', 0 };
 204         static const WCHAR helpW[] = { '/', '?', 0 };
 205         static const WCHAR unixW[] = { '/', 'u', 'n', 'i', 'x', 0 };
 206         static const WCHAR progIDOpenW[] =
 207                 { '/', 'p', 'r', 'o', 'g', 'I', 'D', 'O', 'p', 'e', 'n', 0};
 208         static const WCHAR openW[] = { 'o', 'p', 'e', 'n', 0 };
 209         static const WCHAR cmdW[] = { 'c', 'm', 'd', '.', 'e', 'x', 'e', 0 };
 210
 211         memset(&sei, 0, sizeof(sei));
 212         sei.cbSize = sizeof(sei);
 213         sei.lpVerb = openW;
 214         sei.nShow = SW_SHOWNORMAL;
 215         /* Dunno what these mean, but it looks like winMe's start uses them */
 216         sei.fMask = SEE_MASK_FLAG_DDEWAIT|
 217                     SEE_MASK_FLAG_NO_UI;
 218         sei.lpDirectory = NULL;
 219         creation_flags = CREATE_NEW_CONSOLE;
 220
 221         /* Canonical Microsoft commandline flag processing:
 222          * flags start with / and are case insensitive.
 223          */
 224         for (i=1; i<argc; i++) {
 225                 /* parse first quoted argument as console title */
 226                 if (!title && argv[i][0] == '"') {
 227                         title = argv[i];
 228                         continue;
 229                 }
 230                 if (argv[i][0] != '/')
 231                         break;
 232
 233                 /* Unix paths can start with / so we have to assume anything following /unix is not a flag */
 234                 if (unix_mode || progid_open)
 235                         break;
 236
 237                 if (argv[i][0] == '/' && (argv[i][1] == 'd' || argv[i][1] == 'D')) {
 238                         if (argv[i][2])
 239                                 /* The start directory was concatenated to the option */
 240                                 sei.lpDirectory = argv[i]+2;
 241                         else if (i+1 == argc) {
 242                                 WINE_ERR("you must specify a directory path for the /d option\n");
 243                                 usage();
 244                         } else
 245                                 sei.lpDirectory = argv[++i];
 246                 }
 247                 else if (is_option(argv[i], bW)) {
 248                         creation_flags &= ~CREATE_NEW_CONSOLE;
 249                 }
 250                 else if (argv[i][0] == '/' && (argv[i][1] == 'i' || argv[i][1] == 'I')) {
 251                     TRACE("/i is ignored\n"); /* FIXME */
 252                 }
 253                 else if (is_option(argv[i], minW)) {
 254                         sei.nShow = SW_SHOWMINIMIZED;
 255                 }
 256                 else if (is_option(argv[i], maxW)) {
 257                         sei.nShow = SW_SHOWMAXIMIZED;
 258                 }
 259                 else if (is_option(argv[i], lowW)) {
 260                         creation_flags |= IDLE_PRIORITY_CLASS;
 261                 }
 262                 else if (is_option(argv[i], normalW)) {
 263                         creation_flags |= NORMAL_PRIORITY_CLASS;
 264                 }
 265                 else if (is_option(argv[i], highW)) {
 266                         creation_flags |= HIGH_PRIORITY_CLASS;
 267                 }
 268                 else if (is_option(argv[i], realtimeW)) {
 269                         creation_flags |= REALTIME_PRIORITY_CLASS;
 270                 }
 271                 else if (is_option(argv[i], abovenormalW)) {
 272                         creation_flags |= ABOVE_NORMAL_PRIORITY_CLASS;
 273                 }
 274                 else if (is_option(argv[i], belownormalW)) {
 275                         creation_flags |= BELOW_NORMAL_PRIORITY_CLASS;
 276                 }
 277                 else if (is_option(argv[i], separateW)) {
 278                         TRACE("/separate is ignored\n"); /* FIXME */
 279                 }
 280                 else if (is_option(argv[i], sharedW)) {
 281                         TRACE("/shared is ignored\n"); /* FIXME */
 282                 }
 283                 else if (is_option(argv[i], nodeW)) {
 284                         if (i+1 == argc) {
 285                                 WINE_ERR("you must specify a numa node for the /node option\n");
 286                                 usage();
 287                         } else
 288                         {
 289                                 TRACE("/node is ignored\n"); /* FIXME */
 290                                 i++;
 291                         }
 292                 }
 293                 else if (is_option(argv[i], affinityW))
 294                 {
 295                         if (i+1 == argc) {
 296                                 WINE_ERR("you must specify a numa node for the /node option\n");
 297                                 usage();
 298                         } else
 299                         {
 300                                 TRACE("/affinity is ignored\n"); /* FIXME */
 301                                 i++;
 302                         }
 303                 }
 304                 else if (is_option(argv[i], wW) || is_option(argv[i], waitW)) {
 305                         sei.fMask |= SEE_MASK_NOCLOSEPROCESS;
 306                 }
 307                 else if (is_option(argv[i], helpW)) {
 308                         usage();
 309                 }
 310
 311                 /* Wine extensions */
 312
 313                 else if (is_option(argv[i], unixW)) {
 314                         unix_mode = TRUE;
 315                 }
 316                 else if (is_option(argv[i], progIDOpenW)) {
 317                         progid_open = TRUE;
 318                 } else
 319
 320                 {
 321                         WINE_ERR("Unknown option '%s'\n", wine_dbgstr_w(argv[i]));
 322                         usage();
 323                 }
 324         }
 325
 326         if (progid_open) {
 327                 if (i == argc)
 328                         usage();
 329                 sei.lpClass = argv[i++];
 330                 sei.fMask |= SEE_MASK_CLASSNAME;
 331         }
 332
 333         if (i == argc) {
 334                 if (progid_open || unix_mode)
 335                         usage();
 336                 sei.lpFile = cmdW;
 337         }
 338         else
 339                 sei.lpFile = argv[i++];
 340
 341         args = build_args( argc - i, &argv[i] );
 342         sei.lpParameters = args;
 343
 344         if (unix_mode || progid_open) {
 345                 LPWSTR (*CDECL wine_get_dos_file_name_ptr)(LPCSTR);
 346                 char* multibyte_unixpath;
 347                 int multibyte_unixpath_len;
 348
 349                 wine_get_dos_file_name_ptr = (void*)GetProcAddress(GetModuleHandleA("KERNEL32"), "wine_get_dos_file_name");
 350
 351                 if (!wine_get_dos_file_name_ptr)
 352                         fatal_string(STRING_UNIXFAIL);
 353
 354                 multibyte_unixpath_len = WideCharToMultiByte(CP_UNIXCP, 0, sei.lpFile, -1, NULL, 0, NULL, NULL);
 355                 multibyte_unixpath = HeapAlloc(GetProcessHeap(), 0, multibyte_unixpath_len);
 356
 357                 WideCharToMultiByte(CP_UNIXCP, 0, sei.lpFile, -1, multibyte_unixpath, multibyte_unixpath_len, NULL, NULL);
 358
 359                 dos_filename = wine_get_dos_file_name_ptr(multibyte_unixpath);
 360
 361                 HeapFree(GetProcessHeap(), 0, multibyte_unixpath);
 362
 363                 if (!dos_filename)
 364                         fatal_string(STRING_UNIXFAIL);
 365
 366                 sei.lpFile = dos_filename;
 367                 if (!sei.lpDirectory)
 368                         sei.lpDirectory = parent_directory = get_parent_dir(dos_filename);
 369                 sei.fMask &= ~SEE_MASK_FLAG_NO_UI;
 370
 371                 if (GetBinaryTypeW(sei.lpFile, &binary_type)) {
 372                     WCHAR *commandline;
 373                     STARTUPINFOW startup_info;
 374                     PROCESS_INFORMATION process_information;
 375                     static const WCHAR commandlineformat[] = {'"','%','s','"','%','s',0};
 376
 377                     /* explorer on windows always quotes the filename when running a binary on windows (see bug 5224) so we have to use CreateProcessW in this case */
 378
 379                     commandline = HeapAlloc(GetProcessHeap(), 0, (strlenW(sei.lpFile)+3+strlenW(sei.lpParameters))*sizeof(WCHAR));
 380                     sprintfW(commandline, commandlineformat, sei.lpFile, sei.lpParameters);
 381
 382                     ZeroMemory(&startup_info, sizeof(startup_info));
 383                     startup_info.cb = sizeof(startup_info);
 384                     startup_info.lpTitle = title;
 385
 386                     if (!CreateProcessW(
 387                             NULL, /* lpApplicationName */
 388                             commandline, /* lpCommandLine */
 389                             NULL, /* lpProcessAttributes */
 390                             NULL, /* lpThreadAttributes */
 391                             FALSE, /* bInheritHandles */
 392                             creation_flags, /* dwCreationFlags */
 393                             NULL, /* lpEnvironment */
 394                             sei.lpDirectory, /* lpCurrentDirectory */
 395                             &startup_info, /* lpStartupInfo */
 396                             &process_information /* lpProcessInformation */ ))
 397                     {
 398                         fatal_string_error(STRING_EXECFAIL, GetLastError(), sei.lpFile);
 399                     }
 400                     sei.hProcess = process_information.hProcess;
 401                     goto done;
 402                 }
 403         }
 404
 405         if (!ShellExecuteExW(&sei))
 406             fatal_string_error(STRING_EXECFAIL, GetLastError(), sei.lpFile);
 407
 408 done:
 409         HeapFree( GetProcessHeap(), 0, args );
 410         HeapFree( GetProcessHeap(), 0, dos_filename );
 411         HeapFree( GetProcessHeap(), 0, parent_directory );
 412
 413         if (sei.fMask & SEE_MASK_NOCLOSEPROCESS) {
 414                 DWORD exitcode;
 415                 WaitForSingleObject(sei.hProcess, INFINITE);
 416                 GetExitCodeProcess(sei.hProcess, &exitcode);
 417                 ExitProcess(exitcode);
 418         }
 419
 420         ExitProcess(0);
 421 }