base/shell/cmd/start.c

   1 /*
   2  *  START.C - start internal command.
   3  *
   4  *
   5  *  History:
   6  *
   7  *    24-Jul-1999 (Eric Kohl)
   8  *        Started.
   9  *
  10  *    30-Apr-2005 (Magnus Olsen) <magnus@greatlord.com>)
  11  *        Remove all hardcode string to En.rc
  12  */
  13
  14 #include <precomp.h>
  15
  16 #ifdef INCLUDE_CMD_START
  17
  18 /* Find the end of an option, and turn it into a nul-terminated string
  19  * in place. (It's moved back one character, to make room for the nul) */
  20 static TCHAR *GetParameter(TCHAR **pPointer)
  21 {
  22         BOOL bInQuote = FALSE;
  23         TCHAR *start = *pPointer;
  24         TCHAR *p;
  25         for (p = start; *p; p++)
  26         {
  27                 if (!bInQuote && (*p == _T('/') || _istspace(*p)))
  28                         break;
  29                 bInQuote ^= (*p == _T('"'));
  30                 p[-1] = *p;
  31         }
  32         p[-1] = _T('\0');
  33         *pPointer = p;
  34         return start - 1;
  35 }
  36
  37 INT cmd_start (LPTSTR Rest)
  38 {
  39         TCHAR szFullName[CMDLINE_LENGTH];
  40         TCHAR szUnquotedName[CMDLINE_LENGTH];
  41         TCHAR *param = NULL;
  42         TCHAR *dot;
  43         INT size;
  44         LPTSTR comspec;
  45         BOOL bWait = FALSE;
  46         BOOL bBat  = FALSE;
  47         BOOL bCreate = FALSE;
  48         TCHAR szFullCmdLine [CMDLINE_LENGTH];
  49         PROCESS_INFORMATION prci;
  50         STARTUPINFO stui;
  51 #ifdef UNICODE
  52         DWORD dwCreationFlags = CREATE_NEW_CONSOLE | CREATE_UNICODE_ENVIRONMENT;
  53 #else
  54         DWORD dwCreationFlags = CREATE_NEW_CONSOLE;
  55 #endif
  56         DWORD dwAffinityMask = 0;
  57         LPTSTR lpTitle = NULL;
  58         LPTSTR lpDirectory = NULL;
  59         LPTSTR lpEnvironment = NULL;
  60         WORD wShowWindow = SW_SHOWNORMAL;
  61
  62         while (1)
  63         {
  64                 if (_istspace(*Rest))
  65                 {
  66                         Rest++;
  67                 }
  68                 else if (*Rest == _T('"') && !lpTitle)
  69                 {
  70                         lpTitle = GetParameter(&Rest);
  71                         StripQuotes(lpTitle);
  72                 }
  73                 else if (*Rest == L'/')
  74                 {
  75                         LPTSTR option;
  76                         Rest++;
  77                         option = GetParameter(&Rest);
  78                         if (*option == _T('?'))
  79                         {
  80                                 ConOutResPaging(TRUE,STRING_START_HELP1);
  81                                 return 0;
  82                         }
  83                         else if (_totupper(*option) == _T('D'))
  84                         {
  85                                 lpDirectory = option + 1;
  86                                 if (!*lpDirectory)
  87                                 {
  88                                         while (_istspace(*Rest))
  89                                                 Rest++;
  90                                         lpDirectory = GetParameter(&Rest);
  91                                 }
  92                                 StripQuotes(lpDirectory);
  93                         }
  94                         else if (_totupper(*option) == _T('I'))
  95                         {
  96                                 /* rest of the option is apparently ignored */
  97                                 lpEnvironment = lpOriginalEnvironment;
  98                         }
  99                         else if (!_tcsicmp(option, _T("MIN")))
 100                         {
 101                                 wShowWindow = SW_MINIMIZE;
 102                         }
 103                         else if (!_tcsicmp(option, _T("MAX")))
 104                         {
 105                                 wShowWindow = SW_MAXIMIZE;
 106                         }
 107                         else if (!_tcsicmp(option, _T("AFFINITY")))
 108                         {
 109                                 TCHAR *end;
 110                                 while (_istspace(*Rest))
 111                                         Rest++;
 112                                 option = GetParameter(&Rest);
 113                                 /* Affinity mask is given in hexadecimal */
 114                                 dwAffinityMask = _tcstoul(option, &end, 16);
 115                                 if (*end != _T('\0') || dwAffinityMask == 0 ||
 116                                     dwAffinityMask == (DWORD)-1)
 117                                 {
 118                                         ConErrResPrintf(STRING_ERROR_INVALID_PARAM_FORMAT, option);
 119                                         return 1;
 120                                 }
 121                                 dwCreationFlags |= CREATE_SUSPENDED;
 122                         }
 123                         else if (!_tcsicmp(option, _T("B")))
 124                         {
 125                                 dwCreationFlags &= ~CREATE_NEW_CONSOLE;
 126                                 dwCreationFlags |= CREATE_NEW_PROCESS_GROUP;
 127                         }
 128                         else if (!_tcsicmp(option, _T("LOW")))
 129                         {
 130                                 dwCreationFlags |= IDLE_PRIORITY_CLASS;
 131                         }
 132                         else if (!_tcsicmp(option, _T("NORMAL")))
 133                         {
 134                                 dwCreationFlags |= NORMAL_PRIORITY_CLASS;
 135                         }
 136                         else if (!_tcsicmp(option, _T("HIGH")))
 137                         {
 138                                 dwCreationFlags |= HIGH_PRIORITY_CLASS;
 139                         }
 140                         else if (!_tcsicmp(option, _T("REALTIME")))
 141                         {
 142                                 dwCreationFlags |= REALTIME_PRIORITY_CLASS;
 143                         }
 144                         else if (!_tcsicmp(option, _T("ABOVENORMAL")))
 145                         {
 146                                 dwCreationFlags |= ABOVE_NORMAL_PRIORITY_CLASS;
 147                         }
 148                         else if (!_tcsicmp(option, _T("BELOWNORMAL")))
 149                         {
 150                                 dwCreationFlags |= BELOW_NORMAL_PRIORITY_CLASS;
 151                         }
 152                         else if (!_tcsicmp(option, _T("SEPARATE")))
 153                         {
 154                                 dwCreationFlags |= CREATE_SEPARATE_WOW_VDM;
 155                         }
 156                         else if (!_tcsicmp(option, _T("SHARED")))
 157                         {
 158                                 dwCreationFlags |= CREATE_SHARED_WOW_VDM;
 159                         }
 160                         else if (!_tcsicmp(option, _T("W")) ||
 161                                  !_tcsicmp(option, _T("WAIT")))
 162                         {
 163                                 bWait = TRUE;
 164                         }
 165                         else
 166                         {
 167                                 ConErrResPrintf(STRING_TYPE_ERROR1, option);
 168                                 return 0;
 169                         }
 170                 }
 171                 else
 172                 {
 173                         /* It's not an option - must be the beginning of
 174                          * the actual command. Leave the loop. */
 175                         break;
 176                 }
 177         }
 178
 179         /* get comspec */
 180         comspec = cmd_alloc ( MAX_PATH * sizeof(TCHAR));
 181         if (comspec == NULL)
 182         {
 183                 error_out_of_memory();
 184                 return 1;
 185         }
 186         SetLastError(0);
 187         size = GetEnvironmentVariable (_T("COMSPEC"), comspec, MAX_PATH);
 188         if(GetLastError() == ERROR_ENVVAR_NOT_FOUND)
 189         {
 190                 _tcscpy(comspec, _T("cmd"));
 191         }
 192         else
 193         {
 194                 if (size > MAX_PATH)
 195                 {
 196                         comspec = cmd_realloc(comspec,size * sizeof(TCHAR) );
 197                         if (comspec==NULL)
 198                         {
 199                                 return 1;
 200                         }
 201                         size = GetEnvironmentVariable (_T("COMSPEC"), comspec, size);
 202                 }
 203         }
 204
 205         nErrorLevel = 0;
 206
 207         if (!*Rest)
 208         {
 209                 Rest = _T("cmd.exe");
 210         }
 211         else
 212         /* Parsing the command that gets called by start, and it's parameters */
 213         {
 214                 BOOL bInside = FALSE;
 215                 INT i;
 216                 /* find the end of the command and put the arguments in param */
 217                 for (i = 0; Rest[i]; i++)
 218                 {
 219                         if (Rest[i] == _T('\"'))
 220                                 bInside = !bInside;
 221                         if (_istspace(Rest[i]) && !bInside)
 222                         {
 223                                 param = &Rest[i+1];
 224                                 Rest[i] = _T('\0');
 225                                 break;
 226                         }
 227                 }
 228         }
 229
 230         _tcscpy(szUnquotedName, Rest);
 231         StripQuotes(szUnquotedName);
 232
 233         /* get the PATH environment variable and parse it */
 234         /* search the PATH environment variable for the binary */
 235         if (SearchForExecutable(szUnquotedName, szFullName))
 236         {
 237                 /* check if this is a .BAT or .CMD file */
 238                 dot = _tcsrchr(szFullName, _T('.'));
 239                 if (dot && (!_tcsicmp(dot, _T(".bat")) || !_tcsicmp(dot, _T(".cmd"))))
 240                 {
 241                         bBat = TRUE;
 242                         _stprintf(szFullCmdLine, _T("\"%s\" /K %s"), comspec, Rest);
 243                         TRACE ("[BATCH: %s %s]\n", debugstr_aw(szFullName), debugstr_aw(Rest));
 244                 }
 245                 else
 246                 {
 247                         TRACE ("[EXEC: %s %s]\n", debugstr_aw(szFullName), debugstr_aw(Rest));
 248                         _tcscpy(szFullCmdLine, szFullName);
 249                 }
 250
 251                 /* build command line for CreateProcess() */
 252                 if (param != NULL)
 253                 {
 254                         _tcsncat(szFullCmdLine, _T(" "), CMDLINE_LENGTH - _tcslen(szFullCmdLine));
 255                         _tcsncat(szFullCmdLine, param, CMDLINE_LENGTH - _tcslen(szFullCmdLine));
 256                 }
 257
 258                 /* fill startup info */
 259                 memset (&stui, 0, sizeof (STARTUPINFO));
 260                 stui.cb = sizeof (STARTUPINFO);
 261                 stui.dwFlags = STARTF_USESHOWWINDOW;
 262                 stui.lpTitle = lpTitle;
 263                 stui.wShowWindow = wShowWindow;
 264
 265                 bCreate = CreateProcess(bBat ? comspec : szFullName,
 266                                         szFullCmdLine, NULL, NULL, TRUE, dwCreationFlags,
 267                                         lpEnvironment, lpDirectory, &stui, &prci);
 268                 if (bCreate)
 269                 {
 270                         if (dwAffinityMask)
 271                         {
 272                                 SetProcessAffinityMask(prci.hProcess, dwAffinityMask);
 273                                 ResumeThread(prci.hThread);
 274                         }
 275                         CloseHandle(prci.hThread);
 276                 }
 277         }
 278         else
 279         {
 280                 /* The file name did not seem to be valid, but maybe it's actually a
 281                  * directory or URL, so we still want to pass it to ShellExecute. */
 282                 _tcscpy(szFullName, szUnquotedName);
 283         }
 284
 285         if (!bCreate)
 286         {
 287                 /* CreateProcess didn't work; try ShellExecute */
 288                 DWORD flags = SEE_MASK_NOCLOSEPROCESS;
 289                 if (!(dwCreationFlags & CREATE_NEW_CONSOLE))
 290                         flags |= SEE_MASK_NO_CONSOLE;
 291                 prci.hProcess = RunFile(flags, szFullName, param, lpDirectory, wShowWindow);
 292         }
 293
 294         if (prci.hProcess != NULL)
 295         {
 296                         if (bWait)
 297                         {
 298                                 DWORD dwExitCode;
 299                                 WaitForSingleObject (prci.hProcess, INFINITE);
 300                                 GetExitCodeProcess (prci.hProcess, &dwExitCode);
 301                                 nErrorLevel = (INT)dwExitCode;
 302                         }
 303                         CloseHandle (prci.hProcess);
 304                 /* Get New code page if it has change */
 305                 InputCodePage= GetConsoleCP();
 306         OutputCodePage = GetConsoleOutputCP();
 307                 }
 308                 else
 309                 {
 310                         ErrorMessage(GetLastError (),
 311                                       _T("Error executing CreateProcess()!!\n"));
 312                 }
 313
 314
 315         cmd_free(comspec);
 316         return 0;
 317 }
 318
 319 #endif
 320
 321 /* EOF */