2 * WCMD - Wine-compatible command line interface.
4 * Copyright (C) 1999 - 2001 D A Pickles
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 * - Cannot handle parameters in quotes
24 * - Lots of functionality missing from builtins
29 char *inbuilt[] = {"ATTRIB", "CALL", "CD", "CHDIR", "CLS", "COPY", "CTTY",
30 "DATE", "DEL", "DIR", "ECHO", "ERASE", "FOR", "GOTO",
31 "HELP", "IF", "LABEL", "MD", "MKDIR", "MOVE", "PATH", "PAUSE",
32 "PROMPT", "REM", "REN", "RENAME", "RD", "RMDIR", "SET", "SHIFT",
33 "TIME", "TITLE", "TYPE", "VERIFY", "VER", "VOL", "EXIT" };
37 int echo_mode = 1, verify_mode = 0;
38 char nyi[] = "Not Yet Implemented\n\n";
39 char newline[] = "\n";
40 char version_string[] = "WCMD Version 0.17\n\n";
41 char anykey[] = "Press any key to continue: ";
42 char quals[MAX_PATH], param1[MAX_PATH], param2[MAX_PATH];
43 BATCH_CONTEXT *context = NULL;
45 /*****************************************************************************
46 * Main entry point. This is a console application so we have a main() not a
50 int main (int argc, char *argv[])
56 int opt_c, opt_k, opt_q;
62 if (lstrcmpi(*argv,"/c")==0) {
66 } else if (lstrcmpi(*argv,"/q")==0) {
68 } else if (lstrcmpi(*argv,"/k")==0) {
72 } else if (lstrcmpi(*argv,"/t")==0 || lstrcmpi(*argv,"/x")==0 ||
73 lstrcmpi(*argv,"/y")==0) {
74 /* Ignored for compatibility with Windows */
90 /* Build the command to execute */
92 for (arg = argv; *arg; arg++)
104 if (*a==' ' || *a=='\t') {
106 } else if (*a=='"') {
107 /* doubling of '\' preceeding a '"',
108 * plus escaping of said '"'
116 len+=(a-*arg)+1 /* for the separating space */;
118 len+=2; /* for the quotes */
121 cmd = HeapAlloc(GetProcessHeap(), 0, len);
126 for (arg = argv; *arg; arg++)
128 int has_space,has_quote;
131 /* Check for quotes and spaces in this argument */
132 has_space=has_quote=0;
135 if (*a==' ' || *a=='\t') {
139 } else if (*a=='"') {
147 /* Now transfer it to the command line */
164 /* Double all the '\\' preceeding this '"', plus one */
165 for (i=0;i<=bcount;i++)
184 p--; /* remove last space */
189 /* If we do a "wcmd /c command", we don't want to allocate a new
190 * console since the command returns immediately. Rather, we use
191 * the currently allocated input and output handles. This allows
192 * us to pipe to and read from the command interpreter.
194 WCMD_process_command(cmd);
195 HeapFree(GetProcessHeap(), 0, cmd);
199 SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_LINE_INPUT |
200 ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT);
201 SetConsoleTitle("Wine Command Prompt");
204 WCMD_process_command(cmd);
205 HeapFree(GetProcessHeap(), 0, cmd);
209 * If there is an AUTOEXEC.BAT file, try to execute it.
212 GetFullPathName ("\\autoexec.bat", sizeof(string), string, NULL);
213 h = CreateFile (string, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
214 if (h != INVALID_HANDLE_VALUE) {
217 WCMD_batch_command (string);
222 * Loop forever getting commands and executing them.
228 ReadFile (GetStdHandle(STD_INPUT_HANDLE), string, sizeof(string), &count, NULL);
230 string[count-1] = '\0'; /* ReadFile output is not null-terminated! */
231 if (string[count-2] == '\r') string[count-2] = '\0'; /* Under Windoze we get CRLF! */
232 if (lstrlen (string) != 0) {
233 if (strchr(string,'|') != NULL) {
237 WCMD_process_command (string);
245 /*****************************************************************************
246 * Process one command. If the command is EXIT this routine does not return.
247 * We will recurse through here executing batch files.
251 void WCMD_process_command (char *command) {
257 HANDLE old_stdin = 0, old_stdout = 0, h;
262 * Expand up environment variables.
265 status = ExpandEnvironmentStrings (command, cmd, sizeof(cmd));
272 * Changing default drive has to be handled as a special case.
275 if ((cmd[1] == ':') && IsCharAlpha (cmd[0]) && (strlen(cmd) == 2)) {
276 status = SetCurrentDirectory (cmd);
277 if (!status) WCMD_print_error ();
281 /* Dont issue newline WCMD_output (newline); @JED*/
284 * Redirect stdin and/or stdout if required.
287 if ((p = strchr(cmd,'<')) != NULL) {
288 h = CreateFile (WCMD_parameter (++p, 0, NULL), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING,
289 FILE_ATTRIBUTE_NORMAL, NULL);
290 if (h == INVALID_HANDLE_VALUE) {
294 old_stdin = GetStdHandle (STD_INPUT_HANDLE);
295 SetStdHandle (STD_INPUT_HANDLE, h);
297 if ((p = strchr(cmd,'>')) != NULL) {
298 h = CreateFile (WCMD_parameter (++p, 0, NULL), GENERIC_WRITE, 0, NULL, CREATE_ALWAYS,
299 FILE_ATTRIBUTE_NORMAL, NULL);
300 if (h == INVALID_HANDLE_VALUE) {
304 old_stdout = GetStdHandle (STD_OUTPUT_HANDLE);
305 SetStdHandle (STD_OUTPUT_HANDLE, h);
308 if ((p = strchr(cmd,'<')) != NULL) *p = '\0';
311 * Strip leading whitespaces, and a '@' if supplied
313 whichcmd = WCMD_strtrim_leading_spaces(cmd);
314 if (whichcmd[0] == '@') whichcmd++;
317 * Check if the command entered is internal. If it is, pass the rest of the
318 * line down to the command. If not try to run a program.
322 while (IsCharAlphaNumeric(whichcmd[count])) {
325 for (i=0; i<=WCMD_EXIT; i++) {
326 if (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT,
327 whichcmd, count, inbuilt[i], -1) == 2) break;
329 p = WCMD_strtrim_leading_spaces (&whichcmd[count]);
330 WCMD_parse (p, quals, param1, param2);
334 WCMD_setshow_attrib ();
337 WCMD_batch (param1, p, 1);
341 WCMD_setshow_default ();
344 WCMD_clear_screen ();
353 WCMD_setshow_date ();
363 /* Use the unstripped version of the following data - step over the space */
364 /* but only if a parameter follows */
365 if (strlen(&whichcmd[count]) > 0)
366 WCMD_echo(&whichcmd[count+1]);
368 WCMD_echo(&whichcmd[count]);
393 WCMD_setshow_path (p);
399 WCMD_setshow_prompt ();
412 WCMD_setshow_env (p);
418 WCMD_setshow_time ();
421 if (strlen(&whichcmd[count]) > 0)
422 WCMD_title(&whichcmd[count+1]);
439 WCMD_run_program (whichcmd);
442 CloseHandle (GetStdHandle (STD_INPUT_HANDLE));
443 SetStdHandle (STD_INPUT_HANDLE, old_stdin);
446 CloseHandle (GetStdHandle (STD_OUTPUT_HANDLE));
447 SetStdHandle (STD_OUTPUT_HANDLE, old_stdout);
451 /******************************************************************************
454 * Execute a command line as an external program. If no extension given then
455 * precedence is given to .BAT files. Must allow recursion.
457 * FIXME: Case sensitivity in suffixes!
460 void WCMD_run_program (char *command) {
463 PROCESS_INFORMATION pe;
469 char filetorun[MAX_PATH];
471 WCMD_parse (command, quals, param1, param2); /* Quick way to get the filename */
472 if (!(*param1) && !(*param2))
474 if (strpbrk (param1, "\\:") == NULL) { /* No explicit path given */
475 if ((strchr (param1, '.') == NULL) || (strstr (param1, ".bat") != NULL)) {
476 if (SearchPath (NULL, param1, ".bat", sizeof(filetorun), filetorun, NULL)) {
477 WCMD_batch (filetorun, command, 0);
481 if ((strchr (param1, '.') == NULL) || (strstr (param1, ".cmd") != NULL)) {
482 if (SearchPath (NULL, param1, ".cmd", sizeof(filetorun), filetorun, NULL)) {
483 WCMD_batch (filetorun, command, 0);
488 else { /* Explicit path given */
489 if ((strstr (param1, ".bat") != NULL) ||
490 (strstr (param1, ".cmd") != NULL)) {
491 WCMD_batch (param1, command, 0);
494 if (strchr (param1, '.') == NULL) {
495 strcpy (filetorun, param1);
496 strcat (filetorun, ".bat");
497 h = CreateFile (filetorun, GENERIC_READ, FILE_SHARE_READ,
498 NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
499 if (h != INVALID_HANDLE_VALUE) {
501 WCMD_batch (param1, command, 0);
507 /* No batch file found, assume executable */
509 hinst = FindExecutable (param1, NULL, filetorun);
510 if ((int)hinst < 32) {
514 console = SHGetFileInfo (filetorun, 0, &psfi, sizeof(psfi), SHGFI_EXETYPE);
515 ZeroMemory (&st, sizeof(STARTUPINFO));
516 st.cb = sizeof(STARTUPINFO);
517 status = CreateProcess (NULL, command, NULL, NULL, FALSE,
518 0, NULL, NULL, &st, &pe);
523 if (!console) errorlevel = 0;
526 if (!HIWORD(console)) WaitForSingleObject (pe.hProcess, INFINITE);
527 GetExitCodeProcess (pe.hProcess, &errorlevel);
528 if (errorlevel == STILL_ACTIVE) errorlevel = 0;
532 /******************************************************************************
535 * Display the prompt on STDout
539 void WCMD_show_prompt (void) {
542 char out_string[MAX_PATH], curdir[MAX_PATH], prompt_string[MAX_PATH];
545 status = GetEnvironmentVariable ("PROMPT", prompt_string, sizeof(prompt_string));
546 if ((status == 0) || (status > sizeof(prompt_string))) {
547 lstrcpy (prompt_string, "$P$G");
559 switch (toupper(*p)) {
567 GetDateFormat (LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL, NULL, q, MAX_PATH);
580 status = GetCurrentDirectory (sizeof(curdir), curdir);
586 status = GetCurrentDirectory (sizeof(curdir), curdir);
596 GetTimeFormat (LOCALE_USER_DEFAULT, 0, NULL, NULL, q, MAX_PATH);
600 lstrcat (q, version_string);
611 WCMD_output (out_string);
614 /****************************************************************************
617 * Print the message for GetLastError
620 void WCMD_print_error (void) {
625 error_code = GetLastError ();
626 status = FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
627 NULL, error_code, 0, (LPTSTR) &lpMsgBuf, 0, NULL);
629 WCMD_output ("FIXME: Cannot display message for error %d, status %d\n",
630 error_code, GetLastError());
633 WCMD_output (lpMsgBuf);
634 LocalFree ((HLOCAL)lpMsgBuf);
635 WCMD_output (newline);
639 /*******************************************************************
640 * WCMD_parse - parse a command into parameters and qualifiers.
642 * On exit, all qualifiers are concatenated into q, the first string
643 * not beginning with "/" is in p1 and the
644 * second in p2. Any subsequent non-qualifier strings are lost.
645 * Parameters in quotes are handled.
648 void WCMD_parse (char *s, char *q, char *p1, char *p2) {
652 *q = *p1 = *p2 = '\0';
657 while ((*s != '\0') && (*s != ' ') && *s != '/') {
658 *q++ = toupper (*s++);
667 while ((*s != '\0') && (*s != '"')) {
668 if (p == 0) *p1++ = *s++;
669 else if (p == 1) *p2++ = *s++;
672 if (p == 0) *p1 = '\0';
673 if (p == 1) *p2 = '\0';
680 while ((*s != '\0') && (*s != ' ')) {
681 if (p == 0) *p1++ = *s++;
682 else if (p == 1) *p2++ = *s++;
685 if (p == 0) *p1 = '\0';
686 if (p == 1) *p2 = '\0';
692 /*******************************************************************
693 * WCMD_output - send output to current standard output device.
697 void WCMD_output (char *format, ...) {
703 vsprintf (string, format, ap);
705 WCMD_output_asis(string);
709 static int line_count;
710 static int max_height;
711 static BOOL paged_mode;
713 void WCMD_enter_paged_mode(void)
715 CONSOLE_SCREEN_BUFFER_INFO consoleInfo;
717 if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo))
718 max_height = consoleInfo.dwSize.Y;
722 line_count = 5; /* keep 5 lines from previous output */
725 void WCMD_leave_paged_mode(void)
730 /*******************************************************************
731 * WCMD_output_asis - send output to current standard output device.
732 * without formatting eg. when message contains '%'
735 void WCMD_output_asis (char *message) {
742 if ((ptr = strchr(message, '\n')) != NULL) ptr++;
743 WriteFile (GetStdHandle(STD_OUTPUT_HANDLE), message,
744 (ptr) ? ptr - message : lstrlen(message), &count, NULL);
746 if (++line_count >= max_height - 1) {
748 WCMD_output_asis (anykey);
749 ReadFile (GetStdHandle(STD_INPUT_HANDLE), string, sizeof(string), &count, NULL);
752 } while ((message = ptr) != NULL);
754 WriteFile (GetStdHandle(STD_OUTPUT_HANDLE), message, lstrlen(message), &count, NULL);
759 /***************************************************************************
760 * WCMD_strtrim_leading_spaces
762 * Remove leading spaces from a string. Return a pointer to the first
763 * non-space character. Does not modify the input string
766 char *WCMD_strtrim_leading_spaces (char *string) {
771 while (*ptr == ' ') ptr++;
775 /*************************************************************************
776 * WCMD_strtrim_trailing_spaces
778 * Remove trailing spaces from a string. This routine modifies the input
779 * string by placing a null after the last non-space character
782 void WCMD_strtrim_trailing_spaces (char *string) {
786 ptr = string + lstrlen (string) - 1;
787 while ((*ptr == ' ') && (ptr >= string)) {
793 /*************************************************************************
796 * Handle pipes within a command - the DOS way using temporary files.
799 void WCMD_pipe (char *command) {
802 char temp_path[MAX_PATH], temp_file[MAX_PATH], temp_file2[MAX_PATH], temp_cmd[1024];
804 GetTempPath (sizeof(temp_path), temp_path);
805 GetTempFileName (temp_path, "WCMD", 0, temp_file);
806 p = strchr(command, '|');
808 wsprintf (temp_cmd, "%s > %s", command, temp_file);
809 WCMD_process_command (temp_cmd);
811 while ((p = strchr(command, '|'))) {
813 GetTempFileName (temp_path, "WCMD", 0, temp_file2);
814 wsprintf (temp_cmd, "%s < %s > %s", command, temp_file, temp_file2);
815 WCMD_process_command (temp_cmd);
816 DeleteFile (temp_file);
817 lstrcpy (temp_file, temp_file2);
820 wsprintf (temp_cmd, "%s < %s", command, temp_file);
821 WCMD_process_command (temp_cmd);
822 DeleteFile (temp_file);