pop b0283e26815279d45d201d5585820bb1d1997663
[wine/hacks.git] / programs / cmd / wcmdmain.c
blobb86801c310d30eeeaaa25ee03e6158d09fc45b06
1 /*
2 * CMD - Wine-compatible command line interface.
4 * Copyright (C) 1999 - 2001 D A Pickles
5 * Copyright (C) 2007 J Edmeades
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
23 * FIXME:
24 * - Cannot handle parameters in quotes
25 * - Lots of functionality missing from builtins
28 #include "config.h"
29 #include "wcmd.h"
30 #include "wine/debug.h"
32 WINE_DEFAULT_DEBUG_CHANNEL(cmd);
34 const WCHAR inbuilt[][10] = {
35 {'A','T','T','R','I','B','\0'},
36 {'C','A','L','L','\0'},
37 {'C','D','\0'},
38 {'C','H','D','I','R','\0'},
39 {'C','L','S','\0'},
40 {'C','O','P','Y','\0'},
41 {'C','T','T','Y','\0'},
42 {'D','A','T','E','\0'},
43 {'D','E','L','\0'},
44 {'D','I','R','\0'},
45 {'E','C','H','O','\0'},
46 {'E','R','A','S','E','\0'},
47 {'F','O','R','\0'},
48 {'G','O','T','O','\0'},
49 {'H','E','L','P','\0'},
50 {'I','F','\0'},
51 {'L','A','B','E','L','\0'},
52 {'M','D','\0'},
53 {'M','K','D','I','R','\0'},
54 {'M','O','V','E','\0'},
55 {'P','A','T','H','\0'},
56 {'P','A','U','S','E','\0'},
57 {'P','R','O','M','P','T','\0'},
58 {'R','E','M','\0'},
59 {'R','E','N','\0'},
60 {'R','E','N','A','M','E','\0'},
61 {'R','D','\0'},
62 {'R','M','D','I','R','\0'},
63 {'S','E','T','\0'},
64 {'S','H','I','F','T','\0'},
65 {'T','I','M','E','\0'},
66 {'T','I','T','L','E','\0'},
67 {'T','Y','P','E','\0'},
68 {'V','E','R','I','F','Y','\0'},
69 {'V','E','R','\0'},
70 {'V','O','L','\0'},
71 {'E','N','D','L','O','C','A','L','\0'},
72 {'S','E','T','L','O','C','A','L','\0'},
73 {'P','U','S','H','D','\0'},
74 {'P','O','P','D','\0'},
75 {'A','S','S','O','C','\0'},
76 {'C','O','L','O','R','\0'},
77 {'F','T','Y','P','E','\0'},
78 {'M','O','R','E','\0'},
79 {'E','X','I','T','\0'}
82 HINSTANCE hinst;
83 DWORD errorlevel;
84 int echo_mode = 1, verify_mode = 0, defaultColor = 7;
85 static int opt_c, opt_k, opt_s;
86 const WCHAR newline[] = {'\r','\n','\0'};
87 static const WCHAR equalsW[] = {'=','\0'};
88 static const WCHAR closeBW[] = {')','\0'};
89 WCHAR anykey[100];
90 WCHAR version_string[100];
91 WCHAR quals[MAX_PATH], param1[MAXSTRING], param2[MAXSTRING];
92 BATCH_CONTEXT *context = NULL;
93 extern struct env_stack *pushd_directories;
94 static const WCHAR *pagedMessage = NULL;
95 static char *output_bufA = NULL;
96 #define MAX_WRITECONSOLE_SIZE 65535
97 BOOL unicodePipes = FALSE;
100 /*******************************************************************
101 * WCMD_output_asis_len - send output to current standard output
103 * Output a formatted unicode string. Ideally this will go to the console
104 * and hence required WriteConsoleW to output it, however if file i/o is
105 * redirected, it needs to be WriteFile'd using OEM (not ANSI) format
107 static void WCMD_output_asis_len(const WCHAR *message, int len, HANDLE device) {
109 DWORD nOut= 0;
110 DWORD res = 0;
112 /* If nothing to write, return (MORE does this sometimes) */
113 if (!len) return;
115 /* Try to write as unicode assuming it is to a console */
116 res = WriteConsoleW(device, message, len, &nOut, NULL);
118 /* If writing to console fails, assume its file
119 i/o so convert to OEM codepage and output */
120 if (!res) {
121 BOOL usedDefaultChar = FALSE;
122 DWORD convertedChars;
124 if (!unicodePipes) {
126 * Allocate buffer to use when writing to file. (Not freed, as one off)
128 if (!output_bufA) output_bufA = HeapAlloc(GetProcessHeap(), 0,
129 MAX_WRITECONSOLE_SIZE);
130 if (!output_bufA) {
131 WINE_FIXME("Out of memory - could not allocate ansi 64K buffer\n");
132 return;
135 /* Convert to OEM, then output */
136 convertedChars = WideCharToMultiByte(GetConsoleOutputCP(), 0, message,
137 len, output_bufA, MAX_WRITECONSOLE_SIZE,
138 "?", &usedDefaultChar);
139 WriteFile(device, output_bufA, convertedChars,
140 &nOut, FALSE);
141 } else {
142 WriteFile(device, message, len*sizeof(WCHAR),
143 &nOut, FALSE);
146 return;
149 /*******************************************************************
150 * WCMD_output - send output to current standard output device.
154 void WCMD_output (const WCHAR *format, ...) {
156 va_list ap;
157 WCHAR string[1024];
158 int ret;
160 va_start(ap,format);
161 ret = vsnprintfW(string, sizeof(string)/sizeof(WCHAR), format, ap);
162 if( ret >= (sizeof(string)/sizeof(WCHAR))) {
163 WINE_ERR("Output truncated in WCMD_output\n" );
164 ret = (sizeof(string)/sizeof(WCHAR)) - 1;
165 string[ret] = '\0';
167 va_end(ap);
168 WCMD_output_asis_len(string, ret, GetStdHandle(STD_OUTPUT_HANDLE));
172 static int line_count;
173 static int max_height;
174 static int max_width;
175 static BOOL paged_mode;
176 static int numChars;
178 void WCMD_enter_paged_mode(const WCHAR *msg)
180 CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
182 if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo)) {
183 max_height = consoleInfo.dwSize.Y;
184 max_width = consoleInfo.dwSize.X;
185 } else {
186 max_height = 25;
187 max_width = 80;
189 paged_mode = TRUE;
190 line_count = 0;
191 numChars = 0;
192 pagedMessage = (msg==NULL)? anykey : msg;
195 void WCMD_leave_paged_mode(void)
197 paged_mode = FALSE;
198 pagedMessage = NULL;
201 /***************************************************************************
202 * WCMD_Readfile
204 * Read characters in from a console/file, returning result in Unicode
205 * with signature identical to ReadFile
207 BOOL WCMD_ReadFile(const HANDLE hIn, WCHAR *intoBuf, const DWORD maxChars,
208 LPDWORD charsRead, const LPOVERLAPPED unused) {
210 BOOL res;
212 /* Try to read from console as Unicode */
213 res = ReadConsoleW(hIn, intoBuf, maxChars, charsRead, NULL);
215 /* If reading from console has failed we assume its file
216 i/o so read in and convert from OEM codepage */
217 if (!res) {
219 DWORD numRead;
221 * Allocate buffer to use when reading from file. Not freed
223 if (!output_bufA) output_bufA = HeapAlloc(GetProcessHeap(), 0,
224 MAX_WRITECONSOLE_SIZE);
225 if (!output_bufA) {
226 WINE_FIXME("Out of memory - could not allocate ansi 64K buffer\n");
227 return 0;
230 /* Read from file (assume OEM codepage) */
231 res = ReadFile(hIn, output_bufA, maxChars, &numRead, unused);
233 /* Convert from OEM */
234 *charsRead = MultiByteToWideChar(GetConsoleCP(), 0, output_bufA, numRead,
235 intoBuf, maxChars);
238 return res;
241 /*******************************************************************
242 * WCMD_output_asis - send output to current standard output device.
243 * without formatting eg. when message contains '%'
245 void WCMD_output_asis (const WCHAR *message) {
246 DWORD count;
247 const WCHAR* ptr;
248 WCHAR string[1024];
250 if (paged_mode) {
251 do {
252 ptr = message;
253 while (*ptr && *ptr!='\n' && (numChars < max_width)) {
254 numChars++;
255 ptr++;
257 if (*ptr == '\n') ptr++;
258 WCMD_output_asis_len(message, (ptr) ? ptr - message : strlenW(message),
259 GetStdHandle(STD_OUTPUT_HANDLE));
260 if (ptr) {
261 numChars = 0;
262 if (++line_count >= max_height - 1) {
263 line_count = 0;
264 WCMD_output_asis_len(pagedMessage, strlenW(pagedMessage),
265 GetStdHandle(STD_OUTPUT_HANDLE));
266 WCMD_ReadFile (GetStdHandle(STD_INPUT_HANDLE), string,
267 sizeof(string)/sizeof(WCHAR), &count, NULL);
270 } while (((message = ptr) != NULL) && (*ptr));
271 } else {
272 WCMD_output_asis_len(message, lstrlenW(message),
273 GetStdHandle(STD_OUTPUT_HANDLE));
277 /****************************************************************************
278 * WCMD_print_error
280 * Print the message for GetLastError
283 void WCMD_print_error (void) {
284 LPVOID lpMsgBuf;
285 DWORD error_code;
286 int status;
288 error_code = GetLastError ();
289 status = FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
290 NULL, error_code, 0, (LPWSTR) &lpMsgBuf, 0, NULL);
291 if (!status) {
292 WINE_FIXME ("Cannot display message for error %d, status %d\n",
293 error_code, GetLastError());
294 return;
297 WCMD_output_asis_len(lpMsgBuf, lstrlenW(lpMsgBuf),
298 GetStdHandle(STD_ERROR_HANDLE));
299 LocalFree (lpMsgBuf);
300 WCMD_output_asis_len (newline, lstrlenW(newline),
301 GetStdHandle(STD_ERROR_HANDLE));
302 return;
305 /******************************************************************************
306 * WCMD_show_prompt
308 * Display the prompt on STDout
312 static void WCMD_show_prompt (void) {
314 int status;
315 WCHAR out_string[MAX_PATH], curdir[MAX_PATH], prompt_string[MAX_PATH];
316 WCHAR *p, *q;
317 DWORD len;
318 static const WCHAR envPrompt[] = {'P','R','O','M','P','T','\0'};
320 len = GetEnvironmentVariableW(envPrompt, prompt_string,
321 sizeof(prompt_string)/sizeof(WCHAR));
322 if ((len == 0) || (len >= (sizeof(prompt_string)/sizeof(WCHAR)))) {
323 const WCHAR dfltPrompt[] = {'$','P','$','G','\0'};
324 strcpyW (prompt_string, dfltPrompt);
326 p = prompt_string;
327 q = out_string;
328 *q++ = '\r';
329 *q++ = '\n';
330 *q = '\0';
331 while (*p != '\0') {
332 if (*p != '$') {
333 *q++ = *p++;
334 *q = '\0';
336 else {
337 p++;
338 switch (toupper(*p)) {
339 case '$':
340 *q++ = '$';
341 break;
342 case 'A':
343 *q++ = '&';
344 break;
345 case 'B':
346 *q++ = '|';
347 break;
348 case 'C':
349 *q++ = '(';
350 break;
351 case 'D':
352 GetDateFormatW(LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL, NULL, q, MAX_PATH);
353 while (*q) q++;
354 break;
355 case 'E':
356 *q++ = '\E';
357 break;
358 case 'F':
359 *q++ = ')';
360 break;
361 case 'G':
362 *q++ = '>';
363 break;
364 case 'H':
365 *q++ = '\b';
366 break;
367 case 'L':
368 *q++ = '<';
369 break;
370 case 'N':
371 status = GetCurrentDirectoryW(sizeof(curdir)/sizeof(WCHAR), curdir);
372 if (status) {
373 *q++ = curdir[0];
375 break;
376 case 'P':
377 status = GetCurrentDirectoryW(sizeof(curdir)/sizeof(WCHAR), curdir);
378 if (status) {
379 strcatW (q, curdir);
380 while (*q) q++;
382 break;
383 case 'Q':
384 *q++ = '=';
385 break;
386 case 'S':
387 *q++ = ' ';
388 break;
389 case 'T':
390 GetTimeFormatW(LOCALE_USER_DEFAULT, 0, NULL, NULL, q, MAX_PATH);
391 while (*q) q++;
392 break;
393 case 'V':
394 strcatW (q, version_string);
395 while (*q) q++;
396 break;
397 case '_':
398 *q++ = '\n';
399 break;
400 case '+':
401 if (pushd_directories) {
402 memset(q, '+', pushd_directories->u.stackdepth);
403 q = q + pushd_directories->u.stackdepth;
405 break;
407 p++;
408 *q = '\0';
411 WCMD_output_asis (out_string);
415 /*************************************************************************
416 * WCMD_strdupW
417 * A wide version of strdup as its missing from unicode.h
419 WCHAR *WCMD_strdupW(WCHAR *input) {
420 int len=strlenW(input)+1;
421 WCHAR *result = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
422 memcpy(result, input, len * sizeof(WCHAR));
423 return result;
426 /*************************************************************************
427 * WCMD_strsubstW
428 * Replaces a portion of a Unicode string with the specified string.
429 * It's up to the caller to ensure there is enough space in the
430 * destination buffer.
432 void WCMD_strsubstW(WCHAR* start, WCHAR* next, WCHAR* insert, int len) {
434 if (len < 0)
435 len=insert ? lstrlenW(insert) : 0;
436 if (start+len != next)
437 memmove(start+len, next, (strlenW(next) + 1) * sizeof(*next));
438 if (insert)
439 memcpy(start, insert, len * sizeof(*insert));
442 /***************************************************************************
443 * WCMD_strtrim_leading_spaces
445 * Remove leading spaces from a string. Return a pointer to the first
446 * non-space character. Does not modify the input string
448 WCHAR *WCMD_strtrim_leading_spaces (WCHAR *string) {
450 WCHAR *ptr;
452 ptr = string;
453 while (*ptr == ' ') ptr++;
454 return ptr;
457 /*************************************************************************
458 * WCMD_opt_s_strip_quotes
460 * Remove first and last quote WCHARacters, preserving all other text
462 void WCMD_opt_s_strip_quotes(WCHAR *cmd) {
463 WCHAR *src = cmd + 1, *dest = cmd, *lastq = NULL;
464 while((*dest=*src) != '\0') {
465 if (*src=='\"')
466 lastq=dest;
467 dest++, src++;
469 if (lastq) {
470 dest=lastq++;
471 while ((*dest++=*lastq++) != 0)
477 /*************************************************************************
478 * WCMD_expand_envvar
480 * Expands environment variables, allowing for WCHARacter substitution
482 static WCHAR *WCMD_expand_envvar(WCHAR *start, WCHAR *forVar, WCHAR *forVal) {
483 WCHAR *endOfVar = NULL, *s;
484 WCHAR *colonpos = NULL;
485 WCHAR thisVar[MAXSTRING];
486 WCHAR thisVarContents[MAXSTRING];
487 WCHAR savedchar = 0x00;
488 int len;
490 static const WCHAR ErrorLvl[] = {'E','R','R','O','R','L','E','V','E','L','\0'};
491 static const WCHAR ErrorLvlP[] = {'%','E','R','R','O','R','L','E','V','E','L','%','\0'};
492 static const WCHAR Date[] = {'D','A','T','E','\0'};
493 static const WCHAR DateP[] = {'%','D','A','T','E','%','\0'};
494 static const WCHAR Time[] = {'T','I','M','E','\0'};
495 static const WCHAR TimeP[] = {'%','T','I','M','E','%','\0'};
496 static const WCHAR Cd[] = {'C','D','\0'};
497 static const WCHAR CdP[] = {'%','C','D','%','\0'};
498 static const WCHAR Random[] = {'R','A','N','D','O','M','\0'};
499 static const WCHAR RandomP[] = {'%','R','A','N','D','O','M','%','\0'};
500 static const WCHAR Delims[] = {'%',' ',':','\0'};
502 WINE_TRACE("Expanding: %s (%s,%s)\n", wine_dbgstr_w(start),
503 wine_dbgstr_w(forVal), wine_dbgstr_w(forVar));
505 /* Find the end of the environment variable, and extract name */
506 endOfVar = strpbrkW(start+1, Delims);
508 if (endOfVar == NULL || *endOfVar==' ') {
510 /* In batch program, missing terminator for % and no following
511 ':' just removes the '%' */
512 if (context) {
513 WCMD_strsubstW(start, start + 1, NULL, 0);
514 return start;
515 } else {
517 /* In command processing, just ignore it - allows command line
518 syntax like: for %i in (a.a) do echo %i */
519 return start+1;
523 /* If ':' found, process remaining up until '%' (or stop at ':' if
524 a missing '%' */
525 if (*endOfVar==':') {
526 WCHAR *endOfVar2 = strchrW(endOfVar+1, '%');
527 if (endOfVar2 != NULL) endOfVar = endOfVar2;
530 memcpy(thisVar, start, ((endOfVar - start) + 1) * sizeof(WCHAR));
531 thisVar[(endOfVar - start)+1] = 0x00;
532 colonpos = strchrW(thisVar+1, ':');
534 /* If there's complex substitution, just need %var% for now
535 to get the expanded data to play with */
536 if (colonpos) {
537 *colonpos = '%';
538 savedchar = *(colonpos+1);
539 *(colonpos+1) = 0x00;
542 WINE_TRACE("Retrieving contents of %s\n", wine_dbgstr_w(thisVar));
544 /* Expand to contents, if unchanged, return */
545 /* Handle DATE, TIME, ERRORLEVEL and CD replacements allowing */
546 /* override if existing env var called that name */
547 if ((CompareStringW(LOCALE_USER_DEFAULT,
548 NORM_IGNORECASE | SORT_STRINGSORT,
549 thisVar, 12, ErrorLvlP, -1) == 2) &&
550 (GetEnvironmentVariableW(ErrorLvl, thisVarContents, 1) == 0) &&
551 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
552 static const WCHAR fmt[] = {'%','d','\0'};
553 wsprintfW(thisVarContents, fmt, errorlevel);
554 len = strlenW(thisVarContents);
556 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
557 NORM_IGNORECASE | SORT_STRINGSORT,
558 thisVar, 6, DateP, -1) == 2) &&
559 (GetEnvironmentVariableW(Date, thisVarContents, 1) == 0) &&
560 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
562 GetDateFormatW(LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL,
563 NULL, thisVarContents, MAXSTRING);
564 len = strlenW(thisVarContents);
566 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
567 NORM_IGNORECASE | SORT_STRINGSORT,
568 thisVar, 6, TimeP, -1) == 2) &&
569 (GetEnvironmentVariableW(Time, thisVarContents, 1) == 0) &&
570 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
571 GetTimeFormatW(LOCALE_USER_DEFAULT, TIME_NOSECONDS, NULL,
572 NULL, thisVarContents, MAXSTRING);
573 len = strlenW(thisVarContents);
575 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
576 NORM_IGNORECASE | SORT_STRINGSORT,
577 thisVar, 4, CdP, -1) == 2) &&
578 (GetEnvironmentVariableW(Cd, thisVarContents, 1) == 0) &&
579 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
580 GetCurrentDirectoryW(MAXSTRING, thisVarContents);
581 len = strlenW(thisVarContents);
583 } else if ((CompareStringW(LOCALE_USER_DEFAULT,
584 NORM_IGNORECASE | SORT_STRINGSORT,
585 thisVar, 8, RandomP, -1) == 2) &&
586 (GetEnvironmentVariableW(Random, thisVarContents, 1) == 0) &&
587 (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) {
588 static const WCHAR fmt[] = {'%','d','\0'};
589 wsprintfW(thisVarContents, fmt, rand() % 32768);
590 len = strlenW(thisVarContents);
592 /* Look for a matching 'for' variable */
593 } else if (forVar &&
594 (CompareStringW(LOCALE_USER_DEFAULT,
595 SORT_STRINGSORT,
596 thisVar,
597 (colonpos - thisVar) - 1,
598 forVar, -1) == 2)) {
599 strcpyW(thisVarContents, forVal);
600 len = strlenW(thisVarContents);
602 } else {
604 len = ExpandEnvironmentStringsW(thisVar, thisVarContents,
605 sizeof(thisVarContents)/sizeof(WCHAR));
608 if (len == 0)
609 return endOfVar+1;
611 /* In a batch program, unknown env vars are replaced with nothing,
612 note syntax %garbage:1,3% results in anything after the ':'
613 except the %
614 From the command line, you just get back what you entered */
615 if (lstrcmpiW(thisVar, thisVarContents) == 0) {
617 /* Restore the complex part after the compare */
618 if (colonpos) {
619 *colonpos = ':';
620 *(colonpos+1) = savedchar;
623 /* Command line - just ignore this */
624 if (context == NULL) return endOfVar+1;
627 /* Batch - replace unknown env var with nothing */
628 if (colonpos == NULL) {
629 WCMD_strsubstW(start, endOfVar + 1, NULL, 0);
630 } else {
631 len = strlenW(thisVar);
632 thisVar[len-1] = 0x00;
633 /* If %:...% supplied, : is retained */
634 if (colonpos == thisVar+1) {
635 WCMD_strsubstW(start, endOfVar + 1, colonpos, -1);
636 } else {
637 WCMD_strsubstW(start, endOfVar + 1, colonpos + 1, -1);
640 return start;
644 /* See if we need to do complex substitution (any ':'s), if not
645 then our work here is done */
646 if (colonpos == NULL) {
647 WCMD_strsubstW(start, endOfVar + 1, thisVarContents, -1);
648 return start;
651 /* Restore complex bit */
652 *colonpos = ':';
653 *(colonpos+1) = savedchar;
656 Handle complex substitutions:
657 xxx=yyy (replace xxx with yyy)
658 *xxx=yyy (replace up to and including xxx with yyy)
659 ~x (from x WCHARs in)
660 ~-x (from x WCHARs from the end)
661 ~x,y (from x WCHARs in for y WCHARacters)
662 ~x,-y (from x WCHARs in until y WCHARacters from the end)
665 /* ~ is substring manipulation */
666 if (savedchar == '~') {
668 int substrposition, substrlength = 0;
669 WCHAR *commapos = strchrW(colonpos+2, ',');
670 WCHAR *startCopy;
672 substrposition = atolW(colonpos+2);
673 if (commapos) substrlength = atolW(commapos+1);
675 /* Check bounds */
676 if (substrposition >= 0) {
677 startCopy = &thisVarContents[min(substrposition, len)];
678 } else {
679 startCopy = &thisVarContents[max(0, len+substrposition-1)];
682 if (commapos == NULL) {
683 /* Copy the lot */
684 WCMD_strsubstW(start, endOfVar + 1, startCopy, -1);
685 } else if (substrlength < 0) {
687 int copybytes = (len+substrlength-1)-(startCopy-thisVarContents);
688 if (copybytes > len) copybytes = len;
689 else if (copybytes < 0) copybytes = 0;
690 WCMD_strsubstW(start, endOfVar + 1, startCopy, copybytes);
691 } else {
692 WCMD_strsubstW(start, endOfVar + 1, startCopy, substrlength);
695 return start;
697 /* search and replace manipulation */
698 } else {
699 WCHAR *equalspos = strstrW(colonpos, equalsW);
700 WCHAR *replacewith = equalspos+1;
701 WCHAR *found = NULL;
702 WCHAR *searchIn;
703 WCHAR *searchFor;
705 if (equalspos == NULL) return start+1;
706 s = WCMD_strdupW(endOfVar + 1);
708 /* Null terminate both strings */
709 thisVar[strlenW(thisVar)-1] = 0x00;
710 *equalspos = 0x00;
712 /* Since we need to be case insensitive, copy the 2 buffers */
713 searchIn = WCMD_strdupW(thisVarContents);
714 CharUpperBuffW(searchIn, strlenW(thisVarContents));
715 searchFor = WCMD_strdupW(colonpos+1);
716 CharUpperBuffW(searchFor, strlenW(colonpos+1));
718 /* Handle wildcard case */
719 if (*(colonpos+1) == '*') {
720 /* Search for string to replace */
721 found = strstrW(searchIn, searchFor+1);
723 if (found) {
724 /* Do replacement */
725 strcpyW(start, replacewith);
726 strcatW(start, thisVarContents + (found-searchIn) + strlenW(searchFor+1));
727 strcatW(start, s);
728 } else {
729 /* Copy as is */
730 strcpyW(start, thisVarContents);
731 strcatW(start, s);
734 } else {
735 /* Loop replacing all instances */
736 WCHAR *lastFound = searchIn;
737 WCHAR *outputposn = start;
739 *start = 0x00;
740 while ((found = strstrW(lastFound, searchFor))) {
741 lstrcpynW(outputposn,
742 thisVarContents + (lastFound-searchIn),
743 (found - lastFound)+1);
744 outputposn = outputposn + (found - lastFound);
745 strcatW(outputposn, replacewith);
746 outputposn = outputposn + strlenW(replacewith);
747 lastFound = found + strlenW(searchFor);
749 strcatW(outputposn,
750 thisVarContents + (lastFound-searchIn));
751 strcatW(outputposn, s);
753 HeapFree(GetProcessHeap(), 0, s);
754 HeapFree(GetProcessHeap(), 0, searchIn);
755 HeapFree(GetProcessHeap(), 0, searchFor);
756 return start;
758 return start+1;
761 /*****************************************************************************
762 * Expand the command. Native expands lines from batch programs as they are
763 * read in and not again, except for 'for' variable substitution.
764 * eg. As evidence, "echo %1 && shift && echo %1" or "echo %%path%%"
766 static void handleExpansion(WCHAR *cmd, BOOL justFors, WCHAR *forVariable, WCHAR *forValue) {
768 /* For commands in a context (batch program): */
769 /* Expand environment variables in a batch file %{0-9} first */
770 /* including support for any ~ modifiers */
771 /* Additionally: */
772 /* Expand the DATE, TIME, CD, RANDOM and ERRORLEVEL special */
773 /* names allowing environment variable overrides */
774 /* NOTE: To support the %PATH:xxx% syntax, also perform */
775 /* manual expansion of environment variables here */
777 WCHAR *p = cmd;
778 WCHAR *t;
779 int i;
781 while ((p = strchrW(p, '%'))) {
783 WINE_TRACE("Translate command:%s %d (at: %s)\n",
784 wine_dbgstr_w(cmd), justFors, wine_dbgstr_w(p));
785 i = *(p+1) - '0';
787 /* Don't touch %% unless its in Batch */
788 if (!justFors && *(p+1) == '%') {
789 if (context) {
790 WCMD_strsubstW(p, p+1, NULL, 0);
792 p+=1;
794 /* Replace %~ modifications if in batch program */
795 } else if (*(p+1) == '~') {
796 WCMD_HandleTildaModifiers(&p, forVariable, forValue, justFors);
797 p++;
799 /* Replace use of %0...%9 if in batch program*/
800 } else if (!justFors && context && (i >= 0) && (i <= 9)) {
801 t = WCMD_parameter (context -> command, i + context -> shift_count[i], NULL);
802 WCMD_strsubstW(p, p+2, t, -1);
804 /* Replace use of %* if in batch program*/
805 } else if (!justFors && context && *(p+1)=='*') {
806 WCHAR *startOfParms = NULL;
807 t = WCMD_parameter (context -> command, 1, &startOfParms);
808 if (startOfParms != NULL)
809 WCMD_strsubstW(p, p+2, startOfParms, -1);
810 else
811 WCMD_strsubstW(p, p+2, NULL, 0);
813 } else if (forVariable &&
814 (CompareStringW(LOCALE_USER_DEFAULT,
815 SORT_STRINGSORT,
817 strlenW(forVariable),
818 forVariable, -1) == 2)) {
819 WCMD_strsubstW(p, p + strlenW(forVariable), forValue, -1);
821 } else if (!justFors) {
822 p = WCMD_expand_envvar(p, forVariable, forValue);
824 /* In a FOR loop, see if this is the variable to replace */
825 } else { /* Ignore %'s on second pass of batch program */
826 p++;
830 return;
834 /*******************************************************************
835 * WCMD_parse - parse a command into parameters and qualifiers.
837 * On exit, all qualifiers are concatenated into q, the first string
838 * not beginning with "/" is in p1 and the
839 * second in p2. Any subsequent non-qualifier strings are lost.
840 * Parameters in quotes are handled.
842 static void WCMD_parse (WCHAR *s, WCHAR *q, WCHAR *p1, WCHAR *p2)
844 int p = 0;
846 *q = *p1 = *p2 = '\0';
847 while (TRUE) {
848 switch (*s) {
849 case '/':
850 *q++ = *s++;
851 while ((*s != '\0') && (*s != ' ') && *s != '/') {
852 *q++ = toupperW (*s++);
854 *q = '\0';
855 break;
856 case ' ':
857 case '\t':
858 s++;
859 break;
860 case '"':
861 s++;
862 while ((*s != '\0') && (*s != '"')) {
863 if (p == 0) *p1++ = *s++;
864 else if (p == 1) *p2++ = *s++;
865 else s++;
867 if (p == 0) *p1 = '\0';
868 if (p == 1) *p2 = '\0';
869 p++;
870 if (*s == '"') s++;
871 break;
872 case '\0':
873 return;
874 default:
875 while ((*s != '\0') && (*s != ' ') && (*s != '\t')
876 && (*s != '=') && (*s != ',') ) {
877 if (p == 0) *p1++ = *s++;
878 else if (p == 1) *p2++ = *s++;
879 else s++;
881 /* Skip concurrent parms */
882 while ((*s == ' ') || (*s == '\t') || (*s == '=') || (*s == ',') ) s++;
884 if (p == 0) *p1 = '\0';
885 if (p == 1) *p2 = '\0';
886 p++;
891 static void init_msvcrt_io_block(STARTUPINFOW* st)
893 STARTUPINFOW st_p;
894 /* fetch the parent MSVCRT info block if any, so that the child can use the
895 * same handles as its grand-father
897 st_p.cb = sizeof(STARTUPINFOW);
898 GetStartupInfoW(&st_p);
899 st->cbReserved2 = st_p.cbReserved2;
900 st->lpReserved2 = st_p.lpReserved2;
901 if (st_p.cbReserved2 && st_p.lpReserved2)
903 /* Override the entries for fd 0,1,2 if we happened
904 * to change those std handles (this depends on the way wcmd sets
905 * it's new input & output handles)
907 size_t sz = max(sizeof(unsigned) + (sizeof(char) + sizeof(HANDLE)) * 3, st_p.cbReserved2);
908 BYTE* ptr = HeapAlloc(GetProcessHeap(), 0, sz);
909 if (ptr)
911 unsigned num = *(unsigned*)st_p.lpReserved2;
912 char* flags = (char*)(ptr + sizeof(unsigned));
913 HANDLE* handles = (HANDLE*)(flags + num * sizeof(char));
915 memcpy(ptr, st_p.lpReserved2, st_p.cbReserved2);
916 st->cbReserved2 = sz;
917 st->lpReserved2 = ptr;
919 #define WX_OPEN 0x01 /* see dlls/msvcrt/file.c */
920 if (num <= 0 || (flags[0] & WX_OPEN))
922 handles[0] = GetStdHandle(STD_INPUT_HANDLE);
923 flags[0] |= WX_OPEN;
925 if (num <= 1 || (flags[1] & WX_OPEN))
927 handles[1] = GetStdHandle(STD_OUTPUT_HANDLE);
928 flags[1] |= WX_OPEN;
930 if (num <= 2 || (flags[2] & WX_OPEN))
932 handles[2] = GetStdHandle(STD_ERROR_HANDLE);
933 flags[2] |= WX_OPEN;
935 #undef WX_OPEN
940 /******************************************************************************
941 * WCMD_run_program
943 * Execute a command line as an external program. Must allow recursion.
945 * Precedence:
946 * Manual testing under windows shows PATHEXT plays a key part in this,
947 * and the search algorithm and precedence appears to be as follows.
949 * Search locations:
950 * If directory supplied on command, just use that directory
951 * If extension supplied on command, look for that explicit name first
952 * Otherwise, search in each directory on the path
953 * Precedence:
954 * If extension supplied on command, look for that explicit name first
955 * Then look for supplied name .* (even if extension supplied, so
956 * 'garbage.exe' will match 'garbage.exe.cmd')
957 * If any found, cycle through PATHEXT looking for name.exe one by one
958 * Launching
959 * Once a match has been found, it is launched - Code currently uses
960 * findexecutable to achieve this which is left untouched.
963 void WCMD_run_program (WCHAR *command, int called) {
965 WCHAR temp[MAX_PATH];
966 WCHAR pathtosearch[MAXSTRING];
967 WCHAR *pathposn;
968 WCHAR stemofsearch[MAX_PATH]; /* maximum allowed executable name is
969 MAX_PATH, including null character */
970 WCHAR *lastSlash;
971 WCHAR pathext[MAXSTRING];
972 BOOL extensionsupplied = FALSE;
973 BOOL launched = FALSE;
974 BOOL status;
975 BOOL assumeInternal = FALSE;
976 DWORD len;
977 static const WCHAR envPath[] = {'P','A','T','H','\0'};
978 static const WCHAR envPathExt[] = {'P','A','T','H','E','X','T','\0'};
979 static const WCHAR delims[] = {'/','\\',':','\0'};
981 WCMD_parse (command, quals, param1, param2); /* Quick way to get the filename */
982 if (!(*param1) && !(*param2))
983 return;
985 /* Calculate the search path and stem to search for */
986 if (strpbrkW (param1, delims) == NULL) { /* No explicit path given, search path */
987 static const WCHAR curDir[] = {'.',';','\0'};
988 strcpyW(pathtosearch, curDir);
989 len = GetEnvironmentVariableW(envPath, &pathtosearch[2], (sizeof(pathtosearch)/sizeof(WCHAR))-2);
990 if ((len == 0) || (len >= (sizeof(pathtosearch)/sizeof(WCHAR)) - 2)) {
991 static const WCHAR curDir[] = {'.','\0'};
992 strcpyW (pathtosearch, curDir);
994 if (strchrW(param1, '.') != NULL) extensionsupplied = TRUE;
995 if (strlenW(param1) >= MAX_PATH)
997 WCMD_output_asis(WCMD_LoadMessage(WCMD_LINETOOLONG));
998 return;
1001 strcpyW(stemofsearch, param1);
1003 } else {
1005 /* Convert eg. ..\fred to include a directory by removing file part */
1006 GetFullPathNameW(param1, sizeof(pathtosearch)/sizeof(WCHAR), pathtosearch, NULL);
1007 lastSlash = strrchrW(pathtosearch, '\\');
1008 if (lastSlash && strchrW(lastSlash, '.') != NULL) extensionsupplied = TRUE;
1009 strcpyW(stemofsearch, lastSlash+1);
1011 /* Reduce pathtosearch to a path with trailing '\' to support c:\a.bat and
1012 c:\windows\a.bat syntax */
1013 if (lastSlash) *(lastSlash + 1) = 0x00;
1016 /* Now extract PATHEXT */
1017 len = GetEnvironmentVariableW(envPathExt, pathext, sizeof(pathext)/sizeof(WCHAR));
1018 if ((len == 0) || (len >= (sizeof(pathext)/sizeof(WCHAR)))) {
1019 static const WCHAR dfltPathExt[] = {'.','b','a','t',';',
1020 '.','c','o','m',';',
1021 '.','c','m','d',';',
1022 '.','e','x','e','\0'};
1023 strcpyW (pathext, dfltPathExt);
1026 /* Loop through the search path, dir by dir */
1027 pathposn = pathtosearch;
1028 WINE_TRACE("Searching in '%s' for '%s'\n", wine_dbgstr_w(pathtosearch),
1029 wine_dbgstr_w(stemofsearch));
1030 while (!launched && pathposn) {
1032 WCHAR thisDir[MAX_PATH] = {'\0'};
1033 WCHAR *pos = NULL;
1034 BOOL found = FALSE;
1035 const WCHAR slashW[] = {'\\','\0'};
1037 /* Work on the first directory on the search path */
1038 pos = strchrW(pathposn, ';');
1039 if (pos) {
1040 memcpy(thisDir, pathposn, (pos-pathposn) * sizeof(WCHAR));
1041 thisDir[(pos-pathposn)] = 0x00;
1042 pathposn = pos+1;
1044 } else {
1045 strcpyW(thisDir, pathposn);
1046 pathposn = NULL;
1049 /* Since you can have eg. ..\.. on the path, need to expand
1050 to full information */
1051 strcpyW(temp, thisDir);
1052 GetFullPathNameW(temp, MAX_PATH, thisDir, NULL);
1054 /* 1. If extension supplied, see if that file exists */
1055 strcatW(thisDir, slashW);
1056 strcatW(thisDir, stemofsearch);
1057 pos = &thisDir[strlenW(thisDir)]; /* Pos = end of name */
1059 /* 1. If extension supplied, see if that file exists */
1060 if (extensionsupplied) {
1061 if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
1062 found = TRUE;
1066 /* 2. Any .* matches? */
1067 if (!found) {
1068 HANDLE h;
1069 WIN32_FIND_DATAW finddata;
1070 static const WCHAR allFiles[] = {'.','*','\0'};
1072 strcatW(thisDir,allFiles);
1073 h = FindFirstFileW(thisDir, &finddata);
1074 FindClose(h);
1075 if (h != INVALID_HANDLE_VALUE) {
1077 WCHAR *thisExt = pathext;
1079 /* 3. Yes - Try each path ext */
1080 while (thisExt) {
1081 WCHAR *nextExt = strchrW(thisExt, ';');
1083 if (nextExt) {
1084 memcpy(pos, thisExt, (nextExt-thisExt) * sizeof(WCHAR));
1085 pos[(nextExt-thisExt)] = 0x00;
1086 thisExt = nextExt+1;
1087 } else {
1088 strcpyW(pos, thisExt);
1089 thisExt = NULL;
1092 if (GetFileAttributesW(thisDir) != INVALID_FILE_ATTRIBUTES) {
1093 found = TRUE;
1094 thisExt = NULL;
1100 /* Internal programs won't be picked up by this search, so even
1101 though not found, try one last createprocess and wait for it
1102 to complete.
1103 Note: Ideally we could tell between a console app (wait) and a
1104 windows app, but the API's for it fail in this case */
1105 if (!found && pathposn == NULL) {
1106 WINE_TRACE("ASSUMING INTERNAL\n");
1107 assumeInternal = TRUE;
1108 } else {
1109 WINE_TRACE("Found as %s\n", wine_dbgstr_w(thisDir));
1112 /* Once found, launch it */
1113 if (found || assumeInternal) {
1114 STARTUPINFOW st;
1115 PROCESS_INFORMATION pe;
1116 SHFILEINFOW psfi;
1117 DWORD console;
1118 HINSTANCE hinst;
1119 WCHAR *ext = strrchrW( thisDir, '.' );
1120 static const WCHAR batExt[] = {'.','b','a','t','\0'};
1121 static const WCHAR cmdExt[] = {'.','c','m','d','\0'};
1123 launched = TRUE;
1125 /* Special case BAT and CMD */
1126 if (ext && !strcmpiW(ext, batExt)) {
1127 WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE);
1128 return;
1129 } else if (ext && !strcmpiW(ext, cmdExt)) {
1130 WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE);
1131 return;
1132 } else {
1134 /* thisDir contains the file to be launched, but with what?
1135 eg. a.exe will require a.exe to be launched, a.html may be iexplore */
1136 hinst = FindExecutableW (thisDir, NULL, temp);
1137 if ((INT_PTR)hinst < 32)
1138 console = 0;
1139 else
1140 console = SHGetFileInfoW(temp, 0, &psfi, sizeof(psfi), SHGFI_EXETYPE);
1142 ZeroMemory (&st, sizeof(STARTUPINFOW));
1143 st.cb = sizeof(STARTUPINFOW);
1144 init_msvcrt_io_block(&st);
1146 /* Launch the process and if a CUI wait on it to complete
1147 Note: Launching internal wine processes cannot specify a full path to exe */
1148 status = CreateProcessW(assumeInternal?NULL : thisDir,
1149 command, NULL, NULL, TRUE, 0, NULL, NULL, &st, &pe);
1150 if ((opt_c || opt_k) && !opt_s && !status
1151 && GetLastError()==ERROR_FILE_NOT_FOUND && command[0]=='\"') {
1152 /* strip first and last quote WCHARacters and try again */
1153 WCMD_opt_s_strip_quotes(command);
1154 opt_s=1;
1155 WCMD_run_program(command, called);
1156 return;
1159 if (!status)
1160 break;
1162 if (!assumeInternal && !console) errorlevel = 0;
1163 else
1165 /* Always wait when called in a batch program context */
1166 if (assumeInternal || context || !HIWORD(console)) WaitForSingleObject (pe.hProcess, INFINITE);
1167 GetExitCodeProcess (pe.hProcess, &errorlevel);
1168 if (errorlevel == STILL_ACTIVE) errorlevel = 0;
1170 CloseHandle(pe.hProcess);
1171 CloseHandle(pe.hThread);
1172 return;
1177 /* Not found anywhere - give up */
1178 SetLastError(ERROR_FILE_NOT_FOUND);
1179 WCMD_print_error ();
1181 /* If a command fails to launch, it sets errorlevel 9009 - which
1182 does not seem to have any associated constant definition */
1183 errorlevel = 9009;
1184 return;
1188 /*****************************************************************************
1189 * Process one command. If the command is EXIT this routine does not return.
1190 * We will recurse through here executing batch files.
1192 void WCMD_execute (WCHAR *command, WCHAR *redirects,
1193 WCHAR *forVariable, WCHAR *forValue,
1194 CMD_LIST **cmdList)
1196 WCHAR *cmd, *p, *redir;
1197 int status, i;
1198 DWORD count, creationDisposition;
1199 HANDLE h;
1200 WCHAR *whichcmd;
1201 SECURITY_ATTRIBUTES sa;
1202 WCHAR *new_cmd = NULL;
1203 WCHAR *new_redir = NULL;
1204 HANDLE old_stdhandles[3] = {GetStdHandle (STD_INPUT_HANDLE),
1205 GetStdHandle (STD_OUTPUT_HANDLE),
1206 GetStdHandle (STD_ERROR_HANDLE)};
1207 DWORD idx_stdhandles[3] = {STD_INPUT_HANDLE,
1208 STD_OUTPUT_HANDLE,
1209 STD_ERROR_HANDLE};
1210 BOOL piped = FALSE;
1212 WINE_TRACE("command on entry:%s (%p), with '%s'='%s'\n",
1213 wine_dbgstr_w(command), cmdList,
1214 wine_dbgstr_w(forVariable), wine_dbgstr_w(forValue));
1216 /* If the next command is a pipe then we implement pipes by redirecting
1217 the output from this command to a temp file and input into the
1218 next command from that temp file.
1219 FIXME: Use of named pipes would make more sense here as currently this
1220 process has to finish before the next one can start but this requires
1221 a change to not wait for the first app to finish but rather the pipe */
1222 if (cmdList && (*cmdList)->nextcommand &&
1223 (*cmdList)->nextcommand->prevDelim == CMD_PIPE) {
1225 WCHAR temp_path[MAX_PATH];
1226 static const WCHAR cmdW[] = {'C','M','D','\0'};
1228 /* Remember piping is in action */
1229 WINE_TRACE("Output needs to be piped\n");
1230 piped = TRUE;
1232 /* Generate a unique temporary filename */
1233 GetTempPathW(sizeof(temp_path)/sizeof(WCHAR), temp_path);
1234 GetTempFileNameW(temp_path, cmdW, 0, (*cmdList)->nextcommand->pipeFile);
1235 WINE_TRACE("Using temporary file of %s\n",
1236 wine_dbgstr_w((*cmdList)->nextcommand->pipeFile));
1239 /* Move copy of the command onto the heap so it can be expanded */
1240 new_cmd = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
1241 if (!new_cmd)
1243 WINE_ERR("Could not allocate memory for new_cmd\n");
1244 return;
1246 strcpyW(new_cmd, command);
1248 /* Move copy of the redirects onto the heap so it can be expanded */
1249 new_redir = HeapAlloc( GetProcessHeap(), 0, MAXSTRING * sizeof(WCHAR));
1250 if (!new_redir)
1252 WINE_ERR("Could not allocate memory for new_redir\n");
1253 HeapFree( GetProcessHeap(), 0, new_cmd );
1254 return;
1257 /* If piped output, send stdout to the pipe by appending >filename to redirects */
1258 if (piped) {
1259 static const WCHAR redirOut[] = {'%','s',' ','>',' ','%','s','\0'};
1260 wsprintfW (new_redir, redirOut, redirects, (*cmdList)->nextcommand->pipeFile);
1261 WINE_TRACE("Redirects now %s\n", wine_dbgstr_w(new_redir));
1262 } else {
1263 strcpyW(new_redir, redirects);
1266 /* Expand variables in command line mode only (batch mode will
1267 be expanded as the line is read in, except for 'for' loops) */
1268 handleExpansion(new_cmd, (context != NULL), forVariable, forValue);
1269 handleExpansion(new_redir, (context != NULL), forVariable, forValue);
1270 cmd = new_cmd;
1273 * Changing default drive has to be handled as a special case.
1276 if ((cmd[1] == ':') && IsCharAlphaW(cmd[0]) && (strlenW(cmd) == 2)) {
1277 WCHAR envvar[5];
1278 WCHAR dir[MAX_PATH];
1280 /* According to MSDN CreateProcess docs, special env vars record
1281 the current directory on each drive, in the form =C:
1282 so see if one specified, and if so go back to it */
1283 strcpyW(envvar, equalsW);
1284 strcatW(envvar, cmd);
1285 if (GetEnvironmentVariableW(envvar, dir, MAX_PATH) == 0) {
1286 static const WCHAR fmt[] = {'%','s','\\','\0'};
1287 wsprintfW(cmd, fmt, cmd);
1288 WINE_TRACE("No special directory settings, using dir of %s\n", wine_dbgstr_w(cmd));
1290 WINE_TRACE("Got directory %s as %s\n", wine_dbgstr_w(envvar), wine_dbgstr_w(cmd));
1291 status = SetCurrentDirectoryW(cmd);
1292 if (!status) WCMD_print_error ();
1293 HeapFree( GetProcessHeap(), 0, cmd );
1294 HeapFree( GetProcessHeap(), 0, new_redir );
1295 return;
1298 sa.nLength = sizeof(sa);
1299 sa.lpSecurityDescriptor = NULL;
1300 sa.bInheritHandle = TRUE;
1303 * Redirect stdin, stdout and/or stderr if required.
1306 /* STDIN could come from a preceding pipe, so delete on close if it does */
1307 if (cmdList && (*cmdList)->pipeFile[0] != 0x00) {
1308 WINE_TRACE("Input coming from %s\n", wine_dbgstr_w((*cmdList)->pipeFile));
1309 h = CreateFileW((*cmdList)->pipeFile, GENERIC_READ,
1310 FILE_SHARE_READ, &sa, OPEN_EXISTING,
1311 FILE_ATTRIBUTE_NORMAL | FILE_FLAG_DELETE_ON_CLOSE, NULL);
1312 if (h == INVALID_HANDLE_VALUE) {
1313 WCMD_print_error ();
1314 HeapFree( GetProcessHeap(), 0, cmd );
1315 HeapFree( GetProcessHeap(), 0, new_redir );
1316 return;
1318 SetStdHandle (STD_INPUT_HANDLE, h);
1320 /* No need to remember the temporary name any longer once opened */
1321 (*cmdList)->pipeFile[0] = 0x00;
1323 /* Otherwise STDIN could come from a '<' redirect */
1324 } else if ((p = strchrW(new_redir,'<')) != NULL) {
1325 h = CreateFileW(WCMD_parameter (++p, 0, NULL), GENERIC_READ, FILE_SHARE_READ, &sa, OPEN_EXISTING,
1326 FILE_ATTRIBUTE_NORMAL, NULL);
1327 if (h == INVALID_HANDLE_VALUE) {
1328 WCMD_print_error ();
1329 HeapFree( GetProcessHeap(), 0, cmd );
1330 HeapFree( GetProcessHeap(), 0, new_redir );
1331 return;
1333 SetStdHandle (STD_INPUT_HANDLE, h);
1336 /* Scan the whole command looking for > and 2> */
1337 redir = new_redir;
1338 while (redir != NULL && ((p = strchrW(redir,'>')) != NULL)) {
1339 int handle = 0;
1341 if (*(p-1)!='2') {
1342 handle = 1;
1343 } else {
1344 handle = 2;
1347 p++;
1348 if ('>' == *p) {
1349 creationDisposition = OPEN_ALWAYS;
1350 p++;
1352 else {
1353 creationDisposition = CREATE_ALWAYS;
1356 /* Add support for 2>&1 */
1357 redir = p;
1358 if (*p == '&') {
1359 int idx = *(p+1) - '0';
1361 if (DuplicateHandle(GetCurrentProcess(),
1362 GetStdHandle(idx_stdhandles[idx]),
1363 GetCurrentProcess(),
1365 0, TRUE, DUPLICATE_SAME_ACCESS) == 0) {
1366 WINE_FIXME("Duplicating handle failed with gle %d\n", GetLastError());
1368 WINE_TRACE("Redirect %d (%p) to %d (%p)\n", handle, GetStdHandle(idx_stdhandles[idx]), idx, h);
1370 } else {
1371 WCHAR *param = WCMD_parameter (p, 0, NULL);
1372 h = CreateFileW(param, GENERIC_WRITE, 0, &sa, creationDisposition,
1373 FILE_ATTRIBUTE_NORMAL, NULL);
1374 if (h == INVALID_HANDLE_VALUE) {
1375 WCMD_print_error ();
1376 HeapFree( GetProcessHeap(), 0, cmd );
1377 HeapFree( GetProcessHeap(), 0, new_redir );
1378 return;
1380 if (SetFilePointer (h, 0, NULL, FILE_END) ==
1381 INVALID_SET_FILE_POINTER) {
1382 WCMD_print_error ();
1384 WINE_TRACE("Redirect %d to '%s' (%p)\n", handle, wine_dbgstr_w(param), h);
1387 SetStdHandle (idx_stdhandles[handle], h);
1391 * Strip leading whitespaces, and a '@' if supplied
1393 whichcmd = WCMD_strtrim_leading_spaces(cmd);
1394 WINE_TRACE("Command: '%s'\n", wine_dbgstr_w(cmd));
1395 if (whichcmd[0] == '@') whichcmd++;
1398 * Check if the command entered is internal. If it is, pass the rest of the
1399 * line down to the command. If not try to run a program.
1402 count = 0;
1403 while (IsCharAlphaNumericW(whichcmd[count])) {
1404 count++;
1406 for (i=0; i<=WCMD_EXIT; i++) {
1407 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1408 whichcmd, count, inbuilt[i], -1) == 2) break;
1410 p = WCMD_strtrim_leading_spaces (&whichcmd[count]);
1411 WCMD_parse (p, quals, param1, param2);
1412 WINE_TRACE("param1: %s, param2: %s\n", wine_dbgstr_w(param1), wine_dbgstr_w(param2));
1414 if((p[0] == '/') && (p[1] == '?')) {
1416 /*this is a help request for a program*/
1417 i = WCMD_HELP;
1418 memcpy(p, whichcmd, count * sizeof(WCHAR));
1419 p[count] = '\0';
1423 switch (i) {
1425 case WCMD_ATTRIB:
1426 WCMD_setshow_attrib ();
1427 break;
1428 case WCMD_CALL:
1429 WCMD_call (p);
1430 break;
1431 case WCMD_CD:
1432 case WCMD_CHDIR:
1433 WCMD_setshow_default (p);
1434 break;
1435 case WCMD_CLS:
1436 WCMD_clear_screen ();
1437 break;
1438 case WCMD_COPY:
1439 WCMD_copy ();
1440 break;
1441 case WCMD_CTTY:
1442 WCMD_change_tty ();
1443 break;
1444 case WCMD_DATE:
1445 WCMD_setshow_date ();
1446 break;
1447 case WCMD_DEL:
1448 case WCMD_ERASE:
1449 WCMD_delete (p, TRUE);
1450 break;
1451 case WCMD_DIR:
1452 WCMD_directory (p);
1453 break;
1454 case WCMD_ECHO:
1455 WCMD_echo(&whichcmd[count]);
1456 break;
1457 case WCMD_FOR:
1458 WCMD_for (p, cmdList);
1459 break;
1460 case WCMD_GOTO:
1461 WCMD_goto (cmdList);
1462 break;
1463 case WCMD_HELP:
1464 WCMD_give_help (p);
1465 break;
1466 case WCMD_IF:
1467 WCMD_if (p, cmdList);
1468 break;
1469 case WCMD_LABEL:
1470 WCMD_volume (1, p);
1471 break;
1472 case WCMD_MD:
1473 case WCMD_MKDIR:
1474 WCMD_create_dir ();
1475 break;
1476 case WCMD_MOVE:
1477 WCMD_move ();
1478 break;
1479 case WCMD_PATH:
1480 WCMD_setshow_path (p);
1481 break;
1482 case WCMD_PAUSE:
1483 WCMD_pause ();
1484 break;
1485 case WCMD_PROMPT:
1486 WCMD_setshow_prompt ();
1487 break;
1488 case WCMD_REM:
1489 break;
1490 case WCMD_REN:
1491 case WCMD_RENAME:
1492 WCMD_rename ();
1493 break;
1494 case WCMD_RD:
1495 case WCMD_RMDIR:
1496 WCMD_remove_dir (p);
1497 break;
1498 case WCMD_SETLOCAL:
1499 WCMD_setlocal(p);
1500 break;
1501 case WCMD_ENDLOCAL:
1502 WCMD_endlocal();
1503 break;
1504 case WCMD_SET:
1505 WCMD_setshow_env (p);
1506 break;
1507 case WCMD_SHIFT:
1508 WCMD_shift (p);
1509 break;
1510 case WCMD_TIME:
1511 WCMD_setshow_time ();
1512 break;
1513 case WCMD_TITLE:
1514 if (strlenW(&whichcmd[count]) > 0)
1515 WCMD_title(&whichcmd[count+1]);
1516 break;
1517 case WCMD_TYPE:
1518 WCMD_type (p);
1519 break;
1520 case WCMD_VER:
1521 WCMD_version ();
1522 break;
1523 case WCMD_VERIFY:
1524 WCMD_verify (p);
1525 break;
1526 case WCMD_VOL:
1527 WCMD_volume (0, p);
1528 break;
1529 case WCMD_PUSHD:
1530 WCMD_pushd(p);
1531 break;
1532 case WCMD_POPD:
1533 WCMD_popd();
1534 break;
1535 case WCMD_ASSOC:
1536 WCMD_assoc(p, TRUE);
1537 break;
1538 case WCMD_COLOR:
1539 WCMD_color();
1540 break;
1541 case WCMD_FTYPE:
1542 WCMD_assoc(p, FALSE);
1543 break;
1544 case WCMD_MORE:
1545 WCMD_more(p);
1546 break;
1547 case WCMD_EXIT:
1548 WCMD_exit (cmdList);
1549 break;
1550 default:
1551 WCMD_run_program (whichcmd, 0);
1553 HeapFree( GetProcessHeap(), 0, cmd );
1554 HeapFree( GetProcessHeap(), 0, new_redir );
1556 /* Restore old handles */
1557 for (i=0; i<3; i++) {
1558 if (old_stdhandles[i] != GetStdHandle(idx_stdhandles[i])) {
1559 CloseHandle (GetStdHandle (idx_stdhandles[i]));
1560 SetStdHandle (idx_stdhandles[i], old_stdhandles[i]);
1564 /*************************************************************************
1565 * WCMD_LoadMessage
1566 * Load a string from the resource file, handling any error
1567 * Returns string retrieved from resource file
1569 WCHAR *WCMD_LoadMessage(UINT id) {
1570 static WCHAR msg[2048];
1571 static const WCHAR failedMsg[] = {'F','a','i','l','e','d','!','\0'};
1573 if (!LoadStringW(GetModuleHandleW(NULL), id, msg, sizeof(msg)/sizeof(WCHAR))) {
1574 WINE_FIXME("LoadString failed with %d\n", GetLastError());
1575 strcpyW(msg, failedMsg);
1577 return msg;
1580 /***************************************************************************
1581 * WCMD_DumpCommands
1583 * Dumps out the parsed command line to ensure syntax is correct
1585 static void WCMD_DumpCommands(CMD_LIST *commands) {
1586 CMD_LIST *thisCmd = commands;
1588 WINE_TRACE("Parsed line:\n");
1589 while (thisCmd != NULL) {
1590 WINE_TRACE("%p %d %2.2d %p %s Redir:%s\n",
1591 thisCmd,
1592 thisCmd->prevDelim,
1593 thisCmd->bracketDepth,
1594 thisCmd->nextcommand,
1595 wine_dbgstr_w(thisCmd->command),
1596 wine_dbgstr_w(thisCmd->redirects));
1597 thisCmd = thisCmd->nextcommand;
1601 /***************************************************************************
1602 * WCMD_addCommand
1604 * Adds a command to the current command list
1606 static void WCMD_addCommand(WCHAR *command, int *commandLen,
1607 WCHAR *redirs, int *redirLen,
1608 WCHAR **copyTo, int **copyToLen,
1609 CMD_DELIMITERS prevDelim, int curDepth,
1610 CMD_LIST **lastEntry, CMD_LIST **output) {
1612 CMD_LIST *thisEntry = NULL;
1614 /* Allocate storage for command */
1615 thisEntry = HeapAlloc(GetProcessHeap(), 0, sizeof(CMD_LIST));
1617 /* Copy in the command */
1618 if (command) {
1619 thisEntry->command = HeapAlloc(GetProcessHeap(), 0,
1620 (*commandLen+1) * sizeof(WCHAR));
1621 memcpy(thisEntry->command, command, *commandLen * sizeof(WCHAR));
1622 thisEntry->command[*commandLen] = 0x00;
1624 /* Copy in the redirects */
1625 thisEntry->redirects = HeapAlloc(GetProcessHeap(), 0,
1626 (*redirLen+1) * sizeof(WCHAR));
1627 memcpy(thisEntry->redirects, redirs, *redirLen * sizeof(WCHAR));
1628 thisEntry->redirects[*redirLen] = 0x00;
1629 thisEntry->pipeFile[0] = 0x00;
1631 /* Reset the lengths */
1632 *commandLen = 0;
1633 *redirLen = 0;
1634 *copyToLen = commandLen;
1635 *copyTo = command;
1637 } else {
1638 thisEntry->command = NULL;
1639 thisEntry->redirects = NULL;
1640 thisEntry->pipeFile[0] = 0x00;
1643 /* Fill in other fields */
1644 thisEntry->nextcommand = NULL;
1645 thisEntry->prevDelim = prevDelim;
1646 thisEntry->bracketDepth = curDepth;
1647 if (*lastEntry) {
1648 (*lastEntry)->nextcommand = thisEntry;
1649 } else {
1650 *output = thisEntry;
1652 *lastEntry = thisEntry;
1656 /***************************************************************************
1657 * WCMD_IsEndQuote
1659 * Checks if the quote pointet to is the end-quote.
1661 * Quotes end if:
1663 * 1) The current parameter ends at EOL or at the beginning
1664 * of a redirection or pipe and not in a quote section.
1666 * 2) If the next character is a space and not in a quote section.
1668 * Returns TRUE if this is an end quote, and FALSE if it is not.
1671 static BOOL WCMD_IsEndQuote(WCHAR *quote, int quoteIndex)
1673 int quoteCount = quoteIndex;
1674 int i;
1676 /* If we are not in a quoted section, then we are not an end-quote */
1677 if(quoteIndex == 0)
1679 return FALSE;
1682 /* Check how many quotes are left for this parameter */
1683 for(i=0;quote[i];i++)
1685 if(quote[i] == '"')
1687 quoteCount++;
1690 /* Quote counting ends at EOL, redirection, space or pipe if current quote is complete */
1691 else if(((quoteCount % 2) == 0)
1692 && ((quote[i] == '<') || (quote[i] == '>') || (quote[i] == '|') || (quote[i] == ' ')))
1694 break;
1698 /* If the quote is part of the last part of a series of quotes-on-quotes, then it must
1699 be an end-quote */
1700 if(quoteIndex >= (quoteCount / 2))
1702 return TRUE;
1705 /* No cigar */
1706 return FALSE;
1709 /***************************************************************************
1710 * WCMD_ReadAndParseLine
1712 * Either uses supplied input or
1713 * Reads a file from the handle, and then...
1714 * Parse the text buffer, spliting into separate commands
1715 * - unquoted && strings split 2 commands but the 2nd is flagged as
1716 * following an &&
1717 * - ( as the first character just ups the bracket depth
1718 * - unquoted ) when bracket depth > 0 terminates a bracket and
1719 * adds a CMD_LIST structure with null command
1720 * - Anything else gets put into the command string (including
1721 * redirects)
1723 WCHAR *WCMD_ReadAndParseLine(WCHAR *optionalcmd, CMD_LIST **output, HANDLE readFrom) {
1725 WCHAR *curPos;
1726 int inQuotes = 0;
1727 WCHAR curString[MAXSTRING];
1728 int curStringLen = 0;
1729 WCHAR curRedirs[MAXSTRING];
1730 int curRedirsLen = 0;
1731 WCHAR *curCopyTo;
1732 int *curLen;
1733 int curDepth = 0;
1734 CMD_LIST *lastEntry = NULL;
1735 CMD_DELIMITERS prevDelim = CMD_NONE;
1736 static WCHAR *extraSpace = NULL; /* Deliberately never freed */
1737 const WCHAR remCmd[] = {'r','e','m',' ','\0'};
1738 const WCHAR forCmd[] = {'f','o','r',' ','\0'};
1739 const WCHAR ifCmd[] = {'i','f',' ','\0'};
1740 const WCHAR ifElse[] = {'e','l','s','e',' ','\0'};
1741 BOOL inRem = FALSE;
1742 BOOL inFor = FALSE;
1743 BOOL inIn = FALSE;
1744 BOOL inIf = FALSE;
1745 BOOL inElse= FALSE;
1746 BOOL onlyWhiteSpace = FALSE;
1747 BOOL lastWasWhiteSpace = FALSE;
1748 BOOL lastWasDo = FALSE;
1749 BOOL lastWasIn = FALSE;
1750 BOOL lastWasElse = FALSE;
1751 BOOL lastWasRedirect = TRUE;
1753 /* Allocate working space for a command read from keyboard, file etc */
1754 if (!extraSpace)
1755 extraSpace = HeapAlloc(GetProcessHeap(), 0, (MAXSTRING+1) * sizeof(WCHAR));
1756 if (!extraSpace)
1758 WINE_ERR("Could not allocate memory for extraSpace\n");
1759 return NULL;
1762 /* If initial command read in, use that, otherwise get input from handle */
1763 if (optionalcmd != NULL) {
1764 strcpyW(extraSpace, optionalcmd);
1765 } else if (readFrom == INVALID_HANDLE_VALUE) {
1766 WINE_FIXME("No command nor handle supplied\n");
1767 } else {
1768 if (WCMD_fgets(extraSpace, MAXSTRING, readFrom) == NULL) return NULL;
1770 curPos = extraSpace;
1772 /* Handle truncated input - issue warning */
1773 if (strlenW(extraSpace) == MAXSTRING -1) {
1774 WCMD_output_asis(WCMD_LoadMessage(WCMD_TRUNCATEDLINE));
1775 WCMD_output_asis(extraSpace);
1776 WCMD_output_asis(newline);
1779 /* Replace env vars if in a batch context */
1780 if (context) handleExpansion(extraSpace, FALSE, NULL, NULL);
1781 /* Show prompt before batch line IF echo is on and in batch program */
1782 if (context && echo_mode && extraSpace[0] && (extraSpace[0] != '@')) {
1783 WCMD_show_prompt();
1784 WCMD_output_asis(extraSpace);
1785 WCMD_output_asis(newline);
1788 /* Start with an empty string, copying to the command string */
1789 curStringLen = 0;
1790 curRedirsLen = 0;
1791 curCopyTo = curString;
1792 curLen = &curStringLen;
1793 lastWasRedirect = FALSE; /* Required for eg spaces between > and filename */
1795 /* Parse every character on the line being processed */
1796 while (*curPos != 0x00) {
1798 WCHAR thisChar;
1800 /* Debugging AID:
1801 WINE_TRACE("Looking at '%c' (len:%d, lws:%d, ows:%d)\n", *curPos, *curLen,
1802 lastWasWhiteSpace, onlyWhiteSpace);
1805 /* Certain commands need special handling */
1806 if (curStringLen == 0 && curCopyTo == curString) {
1807 const WCHAR forDO[] = {'d','o',' ','\0'};
1809 /* If command starts with 'rem', ignore any &&, ( etc */
1810 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1811 curPos, 4, remCmd, -1) == 2) {
1812 inRem = TRUE;
1814 /* If command starts with 'for', handle ('s mid line after IN or DO */
1815 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1816 curPos, 4, forCmd, -1) == 2) {
1817 inFor = TRUE;
1819 /* If command starts with 'if' or 'else', handle ('s mid line. We should ensure this
1820 is only true in the command portion of the IF statement, but this
1821 should suffice for now
1822 FIXME: Silly syntax like "if 1(==1( (
1823 echo they equal
1824 )" will be parsed wrong */
1825 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1826 curPos, 3, ifCmd, -1) == 2) {
1827 inIf = TRUE;
1829 } else if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1830 curPos, 5, ifElse, -1) == 2) {
1831 inElse = TRUE;
1832 lastWasElse = TRUE;
1833 onlyWhiteSpace = TRUE;
1834 memcpy(&curCopyTo[*curLen], curPos, 5*sizeof(WCHAR));
1835 (*curLen)+=5;
1836 curPos+=5;
1837 continue;
1839 /* In a for loop, the DO command will follow a close bracket followed by
1840 whitespace, followed by DO, ie closeBracket inserts a NULL entry, curLen
1841 is then 0, and all whitespace is skipped */
1842 } else if (inFor &&
1843 (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1844 curPos, 3, forDO, -1) == 2)) {
1845 WINE_TRACE("Found DO\n");
1846 lastWasDo = TRUE;
1847 onlyWhiteSpace = TRUE;
1848 memcpy(&curCopyTo[*curLen], curPos, 3*sizeof(WCHAR));
1849 (*curLen)+=3;
1850 curPos+=3;
1851 continue;
1853 } else if (curCopyTo == curString) {
1855 /* Special handling for the 'FOR' command */
1856 if (inFor && lastWasWhiteSpace) {
1857 const WCHAR forIN[] = {'i','n',' ','\0'};
1859 WINE_TRACE("Found 'FOR', comparing next parm: '%s'\n", wine_dbgstr_w(curPos));
1861 if (CompareStringW(LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
1862 curPos, 3, forIN, -1) == 2) {
1863 WINE_TRACE("Found IN\n");
1864 lastWasIn = TRUE;
1865 onlyWhiteSpace = TRUE;
1866 memcpy(&curCopyTo[*curLen], curPos, 3*sizeof(WCHAR));
1867 (*curLen)+=3;
1868 curPos+=3;
1869 continue;
1874 /* Nothing 'ends' a REM statement and &&, quotes etc are ineffective,
1875 so just use the default processing ie skip character specific
1876 matching below */
1877 if (!inRem) thisChar = *curPos;
1878 else thisChar = 'X'; /* Character with no special processing */
1880 lastWasWhiteSpace = FALSE; /* Will be reset below */
1882 switch (thisChar) {
1884 case '=': /* drop through - ignore token delimiters at the start of a command */
1885 case ',': /* drop through - ignore token delimiters at the start of a command */
1886 case '\t':/* drop through - ignore token delimiters at the start of a command */
1887 case ' ':
1888 /* If a redirect in place, it ends here */
1889 if (!inQuotes && !lastWasRedirect) {
1891 /* If finishing off a redirect, add a whitespace delimiter */
1892 if (curCopyTo == curRedirs) {
1893 curCopyTo[(*curLen)++] = ' ';
1895 curCopyTo = curString;
1896 curLen = &curStringLen;
1898 if (*curLen > 0) {
1899 curCopyTo[(*curLen)++] = *curPos;
1902 /* Remember just processed whitespace */
1903 lastWasWhiteSpace = TRUE;
1905 break;
1907 case '>': /* drop through - handle redirect chars the same */
1908 case '<':
1909 /* Make a redirect start here */
1910 if (!inQuotes) {
1911 curCopyTo = curRedirs;
1912 curLen = &curRedirsLen;
1913 lastWasRedirect = TRUE;
1916 /* See if 1>, 2> etc, in which case we have some patching up
1917 to do */
1918 if (curPos != extraSpace &&
1919 *(curPos-1)>='1' && *(curPos-1)<='9') {
1921 curStringLen--;
1922 curString[curStringLen] = 0x00;
1923 curCopyTo[(*curLen)++] = *(curPos-1);
1926 curCopyTo[(*curLen)++] = *curPos;
1928 /* If a redirect is immediately followed by '&' (ie. 2>&1) then
1929 do not process that ampersand as an AND operator */
1930 if (thisChar == '>' && *(curPos+1) == '&') {
1931 curCopyTo[(*curLen)++] = *(curPos+1);
1932 curPos++;
1934 break;
1936 case '|': /* Pipe character only if not || */
1937 if (!inQuotes) {
1938 lastWasRedirect = FALSE;
1940 /* Add an entry to the command list */
1941 if (curStringLen > 0) {
1943 /* Add the current command */
1944 WCMD_addCommand(curString, &curStringLen,
1945 curRedirs, &curRedirsLen,
1946 &curCopyTo, &curLen,
1947 prevDelim, curDepth,
1948 &lastEntry, output);
1952 if (*(curPos+1) == '|') {
1953 curPos++; /* Skip other | */
1954 prevDelim = CMD_ONFAILURE;
1955 } else {
1956 prevDelim = CMD_PIPE;
1958 } else {
1959 curCopyTo[(*curLen)++] = *curPos;
1961 break;
1963 case '"': if (WCMD_IsEndQuote(curPos, inQuotes)) {
1964 inQuotes--;
1965 } else {
1966 inQuotes++; /* Quotes within quotes are fun! */
1968 curCopyTo[(*curLen)++] = *curPos;
1969 lastWasRedirect = FALSE;
1970 break;
1972 case '(': /* If a '(' is the first non whitespace in a command portion
1973 ie start of line or just after &&, then we read until an
1974 unquoted ) is found */
1975 WINE_TRACE("Found '(' conditions: curLen(%d), inQ(%d), onlyWS(%d)"
1976 ", for(%d, In:%d, Do:%d)"
1977 ", if(%d, else:%d, lwe:%d)\n",
1978 *curLen, inQuotes,
1979 onlyWhiteSpace,
1980 inFor, lastWasIn, lastWasDo,
1981 inIf, inElse, lastWasElse);
1982 lastWasRedirect = FALSE;
1984 /* Ignore open brackets inside the for set */
1985 if (*curLen == 0 && !inIn) {
1986 curDepth++;
1988 /* If in quotes, ignore brackets */
1989 } else if (inQuotes) {
1990 curCopyTo[(*curLen)++] = *curPos;
1992 /* In a FOR loop, an unquoted '(' may occur straight after
1993 IN or DO
1994 In an IF statement just handle it regardless as we don't
1995 parse the operands
1996 In an ELSE statement, only allow it straight away after
1997 the ELSE and whitespace
1999 } else if (inIf ||
2000 (inElse && lastWasElse && onlyWhiteSpace) ||
2001 (inFor && (lastWasIn || lastWasDo) && onlyWhiteSpace)) {
2003 /* If entering into an 'IN', set inIn */
2004 if (inFor && lastWasIn && onlyWhiteSpace) {
2005 WINE_TRACE("Inside an IN\n");
2006 inIn = TRUE;
2009 /* Add the current command */
2010 WCMD_addCommand(curString, &curStringLen,
2011 curRedirs, &curRedirsLen,
2012 &curCopyTo, &curLen,
2013 prevDelim, curDepth,
2014 &lastEntry, output);
2016 curDepth++;
2017 } else {
2018 curCopyTo[(*curLen)++] = *curPos;
2020 break;
2022 case '&': if (!inQuotes) {
2023 lastWasRedirect = FALSE;
2025 /* Add an entry to the command list */
2026 if (curStringLen > 0) {
2028 /* Add the current command */
2029 WCMD_addCommand(curString, &curStringLen,
2030 curRedirs, &curRedirsLen,
2031 &curCopyTo, &curLen,
2032 prevDelim, curDepth,
2033 &lastEntry, output);
2037 if (*(curPos+1) == '&') {
2038 curPos++; /* Skip other & */
2039 prevDelim = CMD_ONSUCCESS;
2040 } else {
2041 prevDelim = CMD_NONE;
2043 } else {
2044 curCopyTo[(*curLen)++] = *curPos;
2046 break;
2048 case ')': if (!inQuotes && curDepth > 0) {
2049 lastWasRedirect = FALSE;
2051 /* Add the current command if there is one */
2052 if (curStringLen) {
2054 /* Add the current command */
2055 WCMD_addCommand(curString, &curStringLen,
2056 curRedirs, &curRedirsLen,
2057 &curCopyTo, &curLen,
2058 prevDelim, curDepth,
2059 &lastEntry, output);
2062 /* Add an empty entry to the command list */
2063 prevDelim = CMD_NONE;
2064 WCMD_addCommand(NULL, &curStringLen,
2065 curRedirs, &curRedirsLen,
2066 &curCopyTo, &curLen,
2067 prevDelim, curDepth,
2068 &lastEntry, output);
2069 curDepth--;
2071 /* Leave inIn if necessary */
2072 if (inIn) inIn = FALSE;
2073 } else {
2074 curCopyTo[(*curLen)++] = *curPos;
2076 break;
2077 default:
2078 lastWasRedirect = FALSE;
2079 curCopyTo[(*curLen)++] = *curPos;
2082 curPos++;
2084 /* At various times we need to know if we have only skipped whitespace,
2085 so reset this variable and then it will remain true until a non
2086 whitespace is found */
2087 if ((thisChar != ' ') && (thisChar != '\n')) onlyWhiteSpace = FALSE;
2089 /* Flag end of interest in FOR DO and IN parms once something has been processed */
2090 if (!lastWasWhiteSpace) {
2091 lastWasIn = lastWasDo = FALSE;
2094 /* If we have reached the end, add this command into the list */
2095 if (*curPos == 0x00 && *curLen > 0) {
2097 /* Add an entry to the command list */
2098 WCMD_addCommand(curString, &curStringLen,
2099 curRedirs, &curRedirsLen,
2100 &curCopyTo, &curLen,
2101 prevDelim, curDepth,
2102 &lastEntry, output);
2105 /* If we have reached the end of the string, see if bracketing outstanding */
2106 if (*curPos == 0x00 && curDepth > 0 && readFrom != INVALID_HANDLE_VALUE) {
2107 inRem = FALSE;
2108 prevDelim = CMD_NONE;
2109 inQuotes = 0;
2110 memset(extraSpace, 0x00, (MAXSTRING+1) * sizeof(WCHAR));
2112 /* Read more, skipping any blank lines */
2113 while (*extraSpace == 0x00) {
2114 if (!context) WCMD_output_asis( WCMD_LoadMessage(WCMD_MOREPROMPT));
2115 if (WCMD_fgets(extraSpace, MAXSTRING, readFrom) == NULL) break;
2117 curPos = extraSpace;
2118 if (context) handleExpansion(extraSpace, FALSE, NULL, NULL);
2119 /* Continue to echo commands IF echo is on and in batch program */
2120 if (context && echo_mode && extraSpace[0] && (extraSpace[0] != '@')) {
2121 WCMD_output_asis(extraSpace);
2122 WCMD_output_asis(newline);
2127 /* Dump out the parsed output */
2128 WCMD_DumpCommands(*output);
2130 return extraSpace;
2133 /***************************************************************************
2134 * WCMD_process_commands
2136 * Process all the commands read in so far
2138 CMD_LIST *WCMD_process_commands(CMD_LIST *thisCmd, BOOL oneBracket,
2139 WCHAR *var, WCHAR *val) {
2141 int bdepth = -1;
2143 if (thisCmd && oneBracket) bdepth = thisCmd->bracketDepth;
2145 /* Loop through the commands, processing them one by one */
2146 while (thisCmd) {
2148 CMD_LIST *origCmd = thisCmd;
2150 /* If processing one bracket only, and we find the end bracket
2151 entry (or less), return */
2152 if (oneBracket && !thisCmd->command &&
2153 bdepth <= thisCmd->bracketDepth) {
2154 WINE_TRACE("Finished bracket @ %p, next command is %p\n",
2155 thisCmd, thisCmd->nextcommand);
2156 return thisCmd->nextcommand;
2159 /* Ignore the NULL entries a ')' inserts (Only 'if' cares
2160 about them and it will be handled in there)
2161 Also, skip over any batch labels (eg. :fred) */
2162 if (thisCmd->command && thisCmd->command[0] != ':') {
2163 WINE_TRACE("Executing command: '%s'\n", wine_dbgstr_w(thisCmd->command));
2164 WCMD_execute (thisCmd->command, thisCmd->redirects, var, val, &thisCmd);
2167 /* Step on unless the command itself already stepped on */
2168 if (thisCmd == origCmd) thisCmd = thisCmd->nextcommand;
2170 return NULL;
2173 /***************************************************************************
2174 * WCMD_free_commands
2176 * Frees the storage held for a parsed command line
2177 * - This is not done in the process_commands, as eventually the current
2178 * pointer will be modified within the commands, and hence a single free
2179 * routine is simpler
2181 void WCMD_free_commands(CMD_LIST *cmds) {
2183 /* Loop through the commands, freeing them one by one */
2184 while (cmds) {
2185 CMD_LIST *thisCmd = cmds;
2186 cmds = cmds->nextcommand;
2187 HeapFree(GetProcessHeap(), 0, thisCmd->command);
2188 HeapFree(GetProcessHeap(), 0, thisCmd->redirects);
2189 HeapFree(GetProcessHeap(), 0, thisCmd);
2194 /*****************************************************************************
2195 * Main entry point. This is a console application so we have a main() not a
2196 * winmain().
2199 int wmain (int argc, WCHAR *argvW[])
2201 int args;
2202 WCHAR *cmd = NULL;
2203 WCHAR string[1024];
2204 WCHAR envvar[4];
2205 HANDLE h;
2206 int opt_q;
2207 int opt_t = 0;
2208 static const WCHAR autoexec[] = {'\\','a','u','t','o','e','x','e','c','.',
2209 'b','a','t','\0'};
2210 static const WCHAR promptW[] = {'P','R','O','M','P','T','\0'};
2211 static const WCHAR defaultpromptW[] = {'$','P','$','G','\0'};
2212 char ansiVersion[100];
2213 CMD_LIST *toExecute = NULL; /* Commands left to be executed */
2215 srand(time(NULL));
2217 /* Pre initialize some messages */
2218 strcpy(ansiVersion, PACKAGE_VERSION);
2219 MultiByteToWideChar(CP_ACP, 0, ansiVersion, -1, string, 1024);
2220 wsprintfW(version_string, WCMD_LoadMessage(WCMD_VERSION), string);
2221 strcpyW(anykey, WCMD_LoadMessage(WCMD_ANYKEY));
2223 args = argc;
2224 opt_c=opt_k=opt_q=opt_s=0;
2225 while (args > 0)
2227 WCHAR c;
2228 WINE_TRACE("Command line parm: '%s'\n", wine_dbgstr_w(*argvW));
2229 if ((*argvW)[0]!='/' || (*argvW)[1]=='\0') {
2230 argvW++;
2231 args--;
2232 continue;
2235 c=(*argvW)[1];
2236 if (tolowerW(c)=='c') {
2237 opt_c=1;
2238 } else if (tolowerW(c)=='q') {
2239 opt_q=1;
2240 } else if (tolowerW(c)=='k') {
2241 opt_k=1;
2242 } else if (tolowerW(c)=='s') {
2243 opt_s=1;
2244 } else if (tolowerW(c)=='a') {
2245 unicodePipes=FALSE;
2246 } else if (tolowerW(c)=='u') {
2247 unicodePipes=TRUE;
2248 } else if (tolowerW(c)=='t' && (*argvW)[2]==':') {
2249 opt_t=strtoulW(&(*argvW)[3], NULL, 16);
2250 } else if (tolowerW(c)=='x' || tolowerW(c)=='y') {
2251 /* Ignored for compatibility with Windows */
2254 if ((*argvW)[2]==0) {
2255 argvW++;
2256 args--;
2258 else /* handle `cmd /cnotepad.exe` and `cmd /x/c ...` */
2260 *argvW+=2;
2263 if (opt_c || opt_k) /* break out of parsing immediately after c or k */
2264 break;
2267 if (opt_q) {
2268 const WCHAR eoff[] = {'O','F','F','\0'};
2269 WCMD_echo(eoff);
2272 if (opt_c || opt_k) {
2273 int len,qcount;
2274 WCHAR** arg;
2275 int argsLeft;
2276 WCHAR* p;
2278 /* opt_s left unflagged if the command starts with and contains exactly
2279 * one quoted string (exactly two quote characters). The quoted string
2280 * must be an executable name that has whitespace and must not have the
2281 * following characters: &<>()@^| */
2283 /* Build the command to execute */
2284 len = 0;
2285 qcount = 0;
2286 argsLeft = args;
2287 for (arg = argvW; argsLeft>0; arg++,argsLeft--)
2289 int has_space,bcount;
2290 WCHAR* a;
2292 has_space=0;
2293 bcount=0;
2294 a=*arg;
2295 if( !*a ) has_space=1;
2296 while (*a!='\0') {
2297 if (*a=='\\') {
2298 bcount++;
2299 } else {
2300 if (*a==' ' || *a=='\t') {
2301 has_space=1;
2302 } else if (*a=='"') {
2303 /* doubling of '\' preceding a '"',
2304 * plus escaping of said '"'
2306 len+=2*bcount+1;
2307 qcount++;
2309 bcount=0;
2311 a++;
2313 len+=(a-*arg) + 1; /* for the separating space */
2314 if (has_space)
2316 len+=2; /* for the quotes */
2317 qcount+=2;
2321 if (qcount!=2)
2322 opt_s=1;
2324 /* check argvW[0] for a space and invalid characters */
2325 if (!opt_s) {
2326 opt_s=1;
2327 p=*argvW;
2328 while (*p!='\0') {
2329 if (*p=='&' || *p=='<' || *p=='>' || *p=='(' || *p==')'
2330 || *p=='@' || *p=='^' || *p=='|') {
2331 opt_s=1;
2332 break;
2334 if (*p==' ')
2335 opt_s=0;
2336 p++;
2340 cmd = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
2341 if (!cmd)
2342 exit(1);
2344 p = cmd;
2345 argsLeft = args;
2346 for (arg = argvW; argsLeft>0; arg++,argsLeft--)
2348 int has_space,has_quote;
2349 WCHAR* a;
2351 /* Check for quotes and spaces in this argument */
2352 has_space=has_quote=0;
2353 a=*arg;
2354 if( !*a ) has_space=1;
2355 while (*a!='\0') {
2356 if (*a==' ' || *a=='\t') {
2357 has_space=1;
2358 if (has_quote)
2359 break;
2360 } else if (*a=='"') {
2361 has_quote=1;
2362 if (has_space)
2363 break;
2365 a++;
2368 /* Now transfer it to the command line */
2369 if (has_space)
2370 *p++='"';
2371 if (has_quote) {
2372 int bcount;
2373 WCHAR* a;
2375 bcount=0;
2376 a=*arg;
2377 while (*a!='\0') {
2378 if (*a=='\\') {
2379 *p++=*a;
2380 bcount++;
2381 } else {
2382 if (*a=='"') {
2383 int i;
2385 /* Double all the '\\' preceding this '"', plus one */
2386 for (i=0;i<=bcount;i++)
2387 *p++='\\';
2388 *p++='"';
2389 } else {
2390 *p++=*a;
2392 bcount=0;
2394 a++;
2396 } else {
2397 strcpyW(p,*arg);
2398 p+=strlenW(*arg);
2400 if (has_space)
2401 *p++='"';
2402 *p++=' ';
2404 if (p > cmd)
2405 p--; /* remove last space */
2406 *p = '\0';
2408 WINE_TRACE("/c command line: '%s'\n", wine_dbgstr_w(cmd));
2410 /* strip first and last quote characters if opt_s; check for invalid
2411 * executable is done later */
2412 if (opt_s && *cmd=='\"')
2413 WCMD_opt_s_strip_quotes(cmd);
2416 if (opt_c) {
2417 /* If we do a "wcmd /c command", we don't want to allocate a new
2418 * console since the command returns immediately. Rather, we use
2419 * the currently allocated input and output handles. This allows
2420 * us to pipe to and read from the command interpreter.
2423 /* Parse the command string, without reading any more input */
2424 WCMD_ReadAndParseLine(cmd, &toExecute, INVALID_HANDLE_VALUE);
2425 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2426 WCMD_free_commands(toExecute);
2427 toExecute = NULL;
2429 HeapFree(GetProcessHeap(), 0, cmd);
2430 return errorlevel;
2433 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_LINE_INPUT |
2434 ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT);
2435 SetConsoleTitleW(WCMD_LoadMessage(WCMD_CONSTITLE));
2437 /* Note: cmd.exe /c dir does not get a new color, /k dir does */
2438 if (opt_t) {
2439 if (!(((opt_t & 0xF0) >> 4) == (opt_t & 0x0F))) {
2440 defaultColor = opt_t & 0xFF;
2441 param1[0] = 0x00;
2442 WCMD_color();
2444 } else {
2445 /* Check HKCU\Software\Microsoft\Command Processor
2446 Then HKLM\Software\Microsoft\Command Processor
2447 for defaultcolour value
2448 Note Can be supplied as DWORD or REG_SZ
2449 Note2 When supplied as REG_SZ it's in decimal!!! */
2450 HKEY key;
2451 DWORD type;
2452 DWORD value=0, size=4;
2453 static const WCHAR regKeyW[] = {'S','o','f','t','w','a','r','e','\\',
2454 'M','i','c','r','o','s','o','f','t','\\',
2455 'C','o','m','m','a','n','d',' ','P','r','o','c','e','s','s','o','r','\0'};
2456 static const WCHAR dfltColorW[] = {'D','e','f','a','u','l','t','C','o','l','o','r','\0'};
2458 if (RegOpenKeyExW(HKEY_CURRENT_USER, regKeyW,
2459 0, KEY_READ, &key) == ERROR_SUCCESS) {
2460 WCHAR strvalue[4];
2462 /* See if DWORD or REG_SZ */
2463 if (RegQueryValueExW(key, dfltColorW, NULL, &type,
2464 NULL, NULL) == ERROR_SUCCESS) {
2465 if (type == REG_DWORD) {
2466 size = sizeof(DWORD);
2467 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2468 (LPBYTE)&value, &size);
2469 } else if (type == REG_SZ) {
2470 size = sizeof(strvalue)/sizeof(WCHAR);
2471 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2472 (LPBYTE)strvalue, &size);
2473 value = strtoulW(strvalue, NULL, 10);
2476 RegCloseKey(key);
2479 if (value == 0 && RegOpenKeyExW(HKEY_LOCAL_MACHINE, regKeyW,
2480 0, KEY_READ, &key) == ERROR_SUCCESS) {
2481 WCHAR strvalue[4];
2483 /* See if DWORD or REG_SZ */
2484 if (RegQueryValueExW(key, dfltColorW, NULL, &type,
2485 NULL, NULL) == ERROR_SUCCESS) {
2486 if (type == REG_DWORD) {
2487 size = sizeof(DWORD);
2488 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2489 (LPBYTE)&value, &size);
2490 } else if (type == REG_SZ) {
2491 size = sizeof(strvalue)/sizeof(WCHAR);
2492 RegQueryValueExW(key, dfltColorW, NULL, NULL,
2493 (LPBYTE)strvalue, &size);
2494 value = strtoulW(strvalue, NULL, 10);
2497 RegCloseKey(key);
2500 /* If one found, set the screen to that colour */
2501 if (!(((value & 0xF0) >> 4) == (value & 0x0F))) {
2502 defaultColor = value & 0xFF;
2503 param1[0] = 0x00;
2504 WCMD_color();
2509 /* Save cwd into appropriate env var */
2510 GetCurrentDirectoryW(1024, string);
2511 if (IsCharAlphaW(string[0]) && string[1] == ':') {
2512 static const WCHAR fmt[] = {'=','%','c',':','\0'};
2513 wsprintfW(envvar, fmt, string[0]);
2514 SetEnvironmentVariableW(envvar, string);
2515 WINE_TRACE("Set %s to %s\n", wine_dbgstr_w(envvar), wine_dbgstr_w(string));
2518 if (opt_k) {
2519 /* Parse the command string, without reading any more input */
2520 WCMD_ReadAndParseLine(cmd, &toExecute, INVALID_HANDLE_VALUE);
2521 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2522 WCMD_free_commands(toExecute);
2523 toExecute = NULL;
2524 HeapFree(GetProcessHeap(), 0, cmd);
2528 * If there is an AUTOEXEC.BAT file, try to execute it.
2531 GetFullPathNameW (autoexec, sizeof(string)/sizeof(WCHAR), string, NULL);
2532 h = CreateFileW(string, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
2533 if (h != INVALID_HANDLE_VALUE) {
2534 CloseHandle (h);
2535 #if 0
2536 WCMD_batch (autoexec, autoexec, 0, NULL, INVALID_HANDLE_VALUE);
2537 #endif
2541 * Loop forever getting commands and executing them.
2544 SetEnvironmentVariableW(promptW, defaultpromptW);
2545 WCMD_version ();
2546 while (TRUE) {
2548 /* Read until EOF (which for std input is never, but if redirect
2549 in place, may occur */
2550 WCMD_show_prompt ();
2551 if (WCMD_ReadAndParseLine(NULL, &toExecute,
2552 GetStdHandle(STD_INPUT_HANDLE)) == NULL)
2553 break;
2554 WCMD_process_commands(toExecute, FALSE, NULL, NULL);
2555 WCMD_free_commands(toExecute);
2556 toExecute = NULL;
2558 return 0;