| /* |
| * CMD - Wine-compatible command line interface. |
| * |
| * Copyright (C) 1999 - 2001 D A Pickles |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| /* |
| * FIXME: |
| * - Cannot handle parameters in quotes |
| * - Lots of functionality missing from builtins |
| */ |
| |
| #include "config.h" |
| #include "wcmd.h" |
| #include "wine/debug.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(cmd); |
| |
| const char * const inbuilt[] = {"ATTRIB", "CALL", "CD", "CHDIR", "CLS", "COPY", "CTTY", |
| "DATE", "DEL", "DIR", "ECHO", "ERASE", "FOR", "GOTO", |
| "HELP", "IF", "LABEL", "MD", "MKDIR", "MOVE", "PATH", "PAUSE", |
| "PROMPT", "REM", "REN", "RENAME", "RD", "RMDIR", "SET", "SHIFT", |
| "TIME", "TITLE", "TYPE", "VERIFY", "VER", "VOL", |
| "ENDLOCAL", "SETLOCAL", "PUSHD", "POPD", "ASSOC", "COLOR", "FTYPE", |
| "EXIT" }; |
| |
| HINSTANCE hinst; |
| DWORD errorlevel; |
| int echo_mode = 1, verify_mode = 0, defaultColor = 7; |
| static int opt_c, opt_k, opt_s; |
| const char nyi[] = "Not Yet Implemented\n\n"; |
| const char newline[] = "\n"; |
| const char version_string[] = "CMD Version " PACKAGE_VERSION "\n\n"; |
| const char anykey[] = "Press Return key to continue: "; |
| char quals[MAX_PATH], param1[MAX_PATH], param2[MAX_PATH]; |
| BATCH_CONTEXT *context = NULL; |
| extern struct env_stack *pushd_directories; |
| |
| static char *WCMD_expand_envvar(char *start); |
| |
| /***************************************************************************** |
| * Main entry point. This is a console application so we have a main() not a |
| * winmain(). |
| */ |
| |
| int main (int argc, char *argv[]) |
| { |
| char string[1024]; |
| char envvar[4]; |
| char* cmd=NULL; |
| DWORD count; |
| HANDLE h; |
| int opt_q; |
| int opt_t = 0; |
| |
| opt_c=opt_k=opt_q=opt_s=0; |
| while (*argv!=NULL) |
| { |
| char c; |
| if ((*argv)[0]!='/' || (*argv)[1]=='\0') { |
| argv++; |
| continue; |
| } |
| |
| c=(*argv)[1]; |
| if (tolower(c)=='c') { |
| opt_c=1; |
| } else if (tolower(c)=='q') { |
| opt_q=1; |
| } else if (tolower(c)=='k') { |
| opt_k=1; |
| } else if (tolower(c)=='s') { |
| opt_s=1; |
| } else if (tolower(c)=='t' && (*argv)[2]==':') { |
| opt_t=strtoul(&(*argv)[3], NULL, 16); |
| } else if (tolower(c)=='x' || tolower(c)=='y') { |
| /* Ignored for compatibility with Windows */ |
| } |
| |
| if ((*argv)[2]==0) |
| argv++; |
| else /* handle `cmd /cnotepad.exe` and `cmd /x/c ...` */ |
| *argv+=2; |
| |
| if (opt_c || opt_k) /* break out of parsing immediately after c or k */ |
| break; |
| } |
| |
| if (opt_q) { |
| WCMD_echo("OFF"); |
| } |
| |
| if (opt_c || opt_k) { |
| int len,qcount; |
| char** arg; |
| char* p; |
| |
| /* opt_s left unflagged if the command starts with and contains exactly |
| * one quoted string (exactly two quote characters). The quoted string |
| * must be an executable name that has whitespace and must not have the |
| * following characters: &<>()@^| */ |
| |
| /* Build the command to execute */ |
| len = 0; |
| qcount = 0; |
| for (arg = argv; *arg; arg++) |
| { |
| int has_space,bcount; |
| char* a; |
| |
| has_space=0; |
| bcount=0; |
| a=*arg; |
| if( !*a ) has_space=1; |
| while (*a!='\0') { |
| if (*a=='\\') { |
| bcount++; |
| } else { |
| if (*a==' ' || *a=='\t') { |
| has_space=1; |
| } else if (*a=='"') { |
| /* doubling of '\' preceding a '"', |
| * plus escaping of said '"' |
| */ |
| len+=2*bcount+1; |
| qcount++; |
| } |
| bcount=0; |
| } |
| a++; |
| } |
| len+=(a-*arg)+1 /* for the separating space */; |
| if (has_space) |
| { |
| len+=2; /* for the quotes */ |
| qcount+=2; |
| } |
| } |
| |
| if (qcount!=2) |
| opt_s=1; |
| |
| /* check argv[0] for a space and invalid characters */ |
| if (!opt_s) { |
| opt_s=1; |
| p=*argv; |
| while (*p!='\0') { |
| if (*p=='&' || *p=='<' || *p=='>' || *p=='(' || *p==')' |
| || *p=='@' || *p=='^' || *p=='|') { |
| opt_s=1; |
| break; |
| } |
| if (*p==' ') |
| opt_s=0; |
| p++; |
| } |
| } |
| |
| cmd = HeapAlloc(GetProcessHeap(), 0, len); |
| if (!cmd) |
| exit(1); |
| |
| p = cmd; |
| for (arg = argv; *arg; arg++) |
| { |
| int has_space,has_quote; |
| char* a; |
| |
| /* Check for quotes and spaces in this argument */ |
| has_space=has_quote=0; |
| a=*arg; |
| if( !*a ) has_space=1; |
| while (*a!='\0') { |
| if (*a==' ' || *a=='\t') { |
| has_space=1; |
| if (has_quote) |
| break; |
| } else if (*a=='"') { |
| has_quote=1; |
| if (has_space) |
| break; |
| } |
| a++; |
| } |
| |
| /* Now transfer it to the command line */ |
| if (has_space) |
| *p++='"'; |
| if (has_quote) { |
| int bcount; |
| char* a; |
| |
| bcount=0; |
| a=*arg; |
| while (*a!='\0') { |
| if (*a=='\\') { |
| *p++=*a; |
| bcount++; |
| } else { |
| if (*a=='"') { |
| int i; |
| |
| /* Double all the '\\' preceding this '"', plus one */ |
| for (i=0;i<=bcount;i++) |
| *p++='\\'; |
| *p++='"'; |
| } else { |
| *p++=*a; |
| } |
| bcount=0; |
| } |
| a++; |
| } |
| } else { |
| strcpy(p,*arg); |
| p+=strlen(*arg); |
| } |
| if (has_space) |
| *p++='"'; |
| *p++=' '; |
| } |
| if (p > cmd) |
| p--; /* remove last space */ |
| *p = '\0'; |
| |
| /* strip first and last quote characters if opt_s; check for invalid |
| * executable is done later */ |
| if (opt_s && *cmd=='\"') |
| WCMD_opt_s_strip_quotes(cmd); |
| } |
| |
| if (opt_c) { |
| /* If we do a "wcmd /c command", we don't want to allocate a new |
| * console since the command returns immediately. Rather, we use |
| * the currently allocated input and output handles. This allows |
| * us to pipe to and read from the command interpreter. |
| */ |
| if (strchr(cmd,'|') != NULL) |
| WCMD_pipe(cmd); |
| else |
| WCMD_process_command(cmd); |
| HeapFree(GetProcessHeap(), 0, cmd); |
| return errorlevel; |
| } |
| |
| SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE), ENABLE_LINE_INPUT | |
| ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT); |
| SetConsoleTitle("Wine Command Prompt"); |
| |
| /* Note: cmd.exe /c dir does not get a new color, /k dir does */ |
| if (opt_t) { |
| if (!(((opt_t & 0xF0) >> 4) == (opt_t & 0x0F))) { |
| defaultColor = opt_t & 0xFF; |
| param1[0] = 0x00; |
| WCMD_color(); |
| } |
| } else { |
| /* Check HKCU\Software\Microsoft\Command Processor |
| Then HKLM\Software\Microsoft\Command Processor |
| for defaultcolour value |
| Note Can be supplied as DWORD or REG_SZ |
| Note2 When supplied as REG_SZ it's in decimal!!! */ |
| HKEY key; |
| DWORD type; |
| DWORD value=0, size=4; |
| |
| if (RegOpenKeyEx(HKEY_CURRENT_USER, "Software\\Microsoft\\Command Processor", |
| 0, KEY_READ, &key) == ERROR_SUCCESS) { |
| char strvalue[4]; |
| |
| /* See if DWORD or REG_SZ */ |
| if (RegQueryValueEx(key, "DefaultColor", NULL, &type, |
| NULL, NULL) == ERROR_SUCCESS) { |
| if (type == REG_DWORD) { |
| size = sizeof(DWORD); |
| RegQueryValueEx(key, "DefaultColor", NULL, NULL, |
| (LPBYTE)&value, &size); |
| } else if (type == REG_SZ) { |
| size = sizeof(strvalue); |
| RegQueryValueEx(key, "DefaultColor", NULL, NULL, |
| (LPBYTE)strvalue, &size); |
| value = strtoul(strvalue, NULL, 10); |
| } |
| } |
| } |
| |
| if (value == 0 && RegOpenKeyEx(HKEY_LOCAL_MACHINE, |
| "Software\\Microsoft\\Command Processor", |
| 0, KEY_READ, &key) == ERROR_SUCCESS) { |
| char strvalue[4]; |
| |
| /* See if DWORD or REG_SZ */ |
| if (RegQueryValueEx(key, "DefaultColor", NULL, &type, |
| NULL, NULL) == ERROR_SUCCESS) { |
| if (type == REG_DWORD) { |
| size = sizeof(DWORD); |
| RegQueryValueEx(key, "DefaultColor", NULL, NULL, |
| (LPBYTE)&value, &size); |
| } else if (type == REG_SZ) { |
| size = sizeof(strvalue); |
| RegQueryValueEx(key, "DefaultColor", NULL, NULL, |
| (LPBYTE)strvalue, &size); |
| value = strtoul(strvalue, NULL, 10); |
| } |
| } |
| } |
| |
| /* If one found, set the screen to that colour */ |
| if (!(((value & 0xF0) >> 4) == (value & 0x0F))) { |
| defaultColor = value & 0xFF; |
| param1[0] = 0x00; |
| WCMD_color(); |
| } |
| |
| } |
| |
| /* Save cwd into appropriate env var */ |
| GetCurrentDirectory(1024, string); |
| if (IsCharAlpha(string[0]) && string[1] == ':') { |
| sprintf(envvar, "=%c:", string[0]); |
| SetEnvironmentVariable(envvar, string); |
| } |
| |
| if (opt_k) { |
| WCMD_process_command(cmd); |
| HeapFree(GetProcessHeap(), 0, cmd); |
| } |
| |
| /* |
| * If there is an AUTOEXEC.BAT file, try to execute it. |
| */ |
| |
| GetFullPathName ("\\autoexec.bat", sizeof(string), string, NULL); |
| h = CreateFile (string, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); |
| if (h != INVALID_HANDLE_VALUE) { |
| CloseHandle (h); |
| #if 0 |
| WCMD_batch ((char *)"\\autoexec.bat", (char *)"\\autoexec.bat", 0, NULL, INVALID_HANDLE_VALUE); |
| #endif |
| } |
| |
| /* |
| * Loop forever getting commands and executing them. |
| */ |
| |
| WCMD_version (); |
| while (TRUE) { |
| WCMD_show_prompt (); |
| ReadFile (GetStdHandle(STD_INPUT_HANDLE), string, sizeof(string), &count, NULL); |
| if (count > 1) { |
| string[count-1] = '\0'; /* ReadFile output is not null-terminated! */ |
| if (string[count-2] == '\r') string[count-2] = '\0'; /* Under Windoze we get CRLF! */ |
| if (lstrlen (string) != 0) { |
| if (strchr(string,'|') != NULL) { |
| WCMD_pipe (string); |
| } |
| else { |
| WCMD_process_command (string); |
| } |
| } |
| } |
| } |
| } |
| |
| |
| /***************************************************************************** |
| * Process one command. If the command is EXIT this routine does not return. |
| * We will recurse through here executing batch files. |
| */ |
| |
| |
| void WCMD_process_command (char *command) |
| { |
| char *cmd, *p, *s, *t, *redir; |
| int status, i; |
| DWORD count, creationDisposition; |
| HANDLE h; |
| char *whichcmd; |
| SECURITY_ATTRIBUTES sa; |
| char *new_cmd; |
| char *first_redir = NULL; |
| HANDLE old_stdhandles[3] = {INVALID_HANDLE_VALUE, |
| INVALID_HANDLE_VALUE, |
| INVALID_HANDLE_VALUE}; |
| DWORD idx_stdhandles[3] = {STD_INPUT_HANDLE, |
| STD_OUTPUT_HANDLE, |
| STD_ERROR_HANDLE}; |
| |
| /* Move copy of the command onto the heap so it can be expanded */ |
| new_cmd = HeapAlloc( GetProcessHeap(), 0, MAXSTRING ); |
| strcpy(new_cmd, command); |
| |
| /* For commands in a context (batch program): */ |
| /* Expand environment variables in a batch file %{0-9} first */ |
| /* including support for any ~ modifiers */ |
| /* Additionally: */ |
| /* Expand the DATE, TIME, CD, RANDOM and ERRORLEVEL special */ |
| /* names allowing environment variable overrides */ |
| /* NOTE: To support the %PATH:xxx% syntax, also perform */ |
| /* manual expansion of environment variables here */ |
| |
| p = new_cmd; |
| while ((p = strchr(p, '%'))) { |
| i = *(p+1) - '0'; |
| |
| /* Replace %~ modifications if in batch program */ |
| if (context && *(p+1) == '~') { |
| WCMD_HandleTildaModifiers(&p, NULL); |
| p++; |
| |
| /* Replace use of %0...%9 if in batch program*/ |
| } else if (context && (i >= 0) && (i <= 9)) { |
| s = strdup (p+2); |
| t = WCMD_parameter (context -> command, i + context -> shift_count[i], NULL); |
| strcpy (p, t); |
| strcat (p, s); |
| free (s); |
| |
| /* Replace use of %* if in batch program*/ |
| } else if (context && *(p+1)=='*') { |
| char *startOfParms = NULL; |
| s = strdup (p+2); |
| t = WCMD_parameter (context -> command, 1, &startOfParms); |
| if (startOfParms != NULL) strcpy (p, startOfParms); |
| else *p = 0x00; |
| strcat (p, s); |
| free (s); |
| |
| } else { |
| p = WCMD_expand_envvar(p); |
| } |
| } |
| cmd = new_cmd; |
| |
| /* In a batch program, unknown variables are replace by nothing */ |
| /* so remove any remaining %var% */ |
| if (context) { |
| p = cmd; |
| while ((p = strchr(p, '%'))) { |
| s = strchr(p+1, '%'); |
| if (!s) { |
| *p=0x00; |
| } else { |
| t = strdup(s+1); |
| strcpy(p, t); |
| free(t); |
| } |
| } |
| |
| /* Show prompt before batch line IF echo is on and in batch program */ |
| if (echo_mode && (cmd[0] != '@')) { |
| WCMD_show_prompt(); |
| WCMD_output_asis ( cmd); |
| WCMD_output_asis ( "\n"); |
| } |
| } |
| |
| /* |
| * Changing default drive has to be handled as a special case. |
| */ |
| |
| if ((cmd[1] == ':') && IsCharAlpha (cmd[0]) && (strlen(cmd) == 2)) { |
| char envvar[5]; |
| char dir[MAX_PATH]; |
| |
| /* According to MSDN CreateProcess docs, special env vars record |
| the current directory on each drive, in the form =C: |
| so see if one specified, and if so go back to it */ |
| strcpy(envvar, "="); |
| strcat(envvar, cmd); |
| if (GetEnvironmentVariable(envvar, dir, MAX_PATH) == 0) { |
| sprintf(cmd, "%s\\", cmd); |
| } |
| status = SetCurrentDirectory (cmd); |
| if (!status) WCMD_print_error (); |
| HeapFree( GetProcessHeap(), 0, cmd ); |
| return; |
| } |
| |
| sa.nLength = sizeof(sa); |
| sa.lpSecurityDescriptor = NULL; |
| sa.bInheritHandle = TRUE; |
| |
| /* |
| * Redirect stdin, stdout and/or stderr if required. |
| */ |
| |
| if ((p = strchr(cmd,'<')) != NULL) { |
| if (first_redir == NULL) first_redir = p; |
| h = CreateFile (WCMD_parameter (++p, 0, NULL), GENERIC_READ, FILE_SHARE_READ, &sa, OPEN_EXISTING, |
| FILE_ATTRIBUTE_NORMAL, NULL); |
| if (h == INVALID_HANDLE_VALUE) { |
| WCMD_print_error (); |
| HeapFree( GetProcessHeap(), 0, cmd ); |
| return; |
| } |
| old_stdhandles[0] = GetStdHandle (STD_INPUT_HANDLE); |
| SetStdHandle (STD_INPUT_HANDLE, h); |
| } |
| |
| /* Scan the whole command looking for > and 2> */ |
| redir = cmd; |
| while (redir != NULL && ((p = strchr(redir,'>')) != NULL)) { |
| int handle = 0; |
| |
| if (*(p-1)!='2') { |
| if (first_redir == NULL) first_redir = p; |
| handle = 1; |
| } else { |
| if (first_redir == NULL) first_redir = (p-1); |
| handle = 2; |
| } |
| |
| p++; |
| if ('>' == *p) { |
| creationDisposition = OPEN_ALWAYS; |
| p++; |
| } |
| else { |
| creationDisposition = CREATE_ALWAYS; |
| } |
| |
| /* Add support for 2>&1 */ |
| redir = p; |
| if (*p == '&') { |
| int idx = *(p+1) - '0'; |
| |
| if (DuplicateHandle(GetCurrentProcess(), |
| GetStdHandle(idx_stdhandles[idx]), |
| GetCurrentProcess(), |
| &h, |
| 0, TRUE, DUPLICATE_SAME_ACCESS) == 0) { |
| WINE_FIXME("Duplicating handle failed with gle %d\n", GetLastError()); |
| } |
| WINE_TRACE("Redirect %d (%p) to %d (%p)\n", handle, GetStdHandle(idx_stdhandles[idx]), idx, h); |
| |
| } else { |
| char *param = WCMD_parameter (p, 0, NULL); |
| h = CreateFile (param, GENERIC_WRITE, 0, &sa, creationDisposition, |
| FILE_ATTRIBUTE_NORMAL, NULL); |
| if (h == INVALID_HANDLE_VALUE) { |
| WCMD_print_error (); |
| HeapFree( GetProcessHeap(), 0, cmd ); |
| return; |
| } |
| if (SetFilePointer (h, 0, NULL, FILE_END) == |
| INVALID_SET_FILE_POINTER) { |
| WCMD_print_error (); |
| } |
| WINE_TRACE("Redirect %d to '%s' (%p)\n", handle, param, h); |
| } |
| |
| old_stdhandles[handle] = GetStdHandle (idx_stdhandles[handle]); |
| SetStdHandle (idx_stdhandles[handle], h); |
| } |
| |
| /* Terminate the command string at <, or first 2> or > */ |
| if (first_redir != NULL) *first_redir = '\0'; |
| |
| /* |
| * Strip leading whitespaces, and a '@' if supplied |
| */ |
| whichcmd = WCMD_strtrim_leading_spaces(cmd); |
| WINE_TRACE("Command: '%s'\n", cmd); |
| if (whichcmd[0] == '@') whichcmd++; |
| |
| /* |
| * Check if the command entered is internal. If it is, pass the rest of the |
| * line down to the command. If not try to run a program. |
| */ |
| |
| count = 0; |
| while (IsCharAlphaNumeric(whichcmd[count])) { |
| count++; |
| } |
| for (i=0; i<=WCMD_EXIT; i++) { |
| if (CompareString (LOCALE_USER_DEFAULT, NORM_IGNORECASE | SORT_STRINGSORT, |
| whichcmd, count, inbuilt[i], -1) == 2) break; |
| } |
| p = WCMD_strtrim_leading_spaces (&whichcmd[count]); |
| WCMD_parse (p, quals, param1, param2); |
| switch (i) { |
| |
| case WCMD_ATTRIB: |
| WCMD_setshow_attrib (); |
| break; |
| case WCMD_CALL: |
| WCMD_call (p); |
| break; |
| case WCMD_CD: |
| case WCMD_CHDIR: |
| WCMD_setshow_default (p); |
| break; |
| case WCMD_CLS: |
| WCMD_clear_screen (); |
| break; |
| case WCMD_COPY: |
| WCMD_copy (); |
| break; |
| case WCMD_CTTY: |
| WCMD_change_tty (); |
| break; |
| case WCMD_DATE: |
| WCMD_setshow_date (); |
| break; |
| case WCMD_DEL: |
| case WCMD_ERASE: |
| WCMD_delete (p); |
| break; |
| case WCMD_DIR: |
| WCMD_directory (); |
| break; |
| case WCMD_ECHO: |
| WCMD_echo(&whichcmd[count]); |
| break; |
| case WCMD_FOR: |
| WCMD_for (p); |
| break; |
| case WCMD_GOTO: |
| WCMD_goto (); |
| break; |
| case WCMD_HELP: |
| WCMD_give_help (p); |
| break; |
| case WCMD_IF: |
| WCMD_if (p); |
| break; |
| case WCMD_LABEL: |
| WCMD_volume (1, p); |
| break; |
| case WCMD_MD: |
| case WCMD_MKDIR: |
| WCMD_create_dir (); |
| break; |
| case WCMD_MOVE: |
| WCMD_move (); |
| break; |
| case WCMD_PATH: |
| WCMD_setshow_path (p); |
| break; |
| case WCMD_PAUSE: |
| WCMD_pause (); |
| break; |
| case WCMD_PROMPT: |
| WCMD_setshow_prompt (); |
| break; |
| case WCMD_REM: |
| break; |
| case WCMD_REN: |
| case WCMD_RENAME: |
| WCMD_rename (); |
| break; |
| case WCMD_RD: |
| case WCMD_RMDIR: |
| WCMD_remove_dir (p); |
| break; |
| case WCMD_SETLOCAL: |
| WCMD_setlocal(p); |
| break; |
| case WCMD_ENDLOCAL: |
| WCMD_endlocal(); |
| break; |
| case WCMD_SET: |
| WCMD_setshow_env (p); |
| break; |
| case WCMD_SHIFT: |
| WCMD_shift (p); |
| break; |
| case WCMD_TIME: |
| WCMD_setshow_time (); |
| break; |
| case WCMD_TITLE: |
| if (strlen(&whichcmd[count]) > 0) |
| WCMD_title(&whichcmd[count+1]); |
| break; |
| case WCMD_TYPE: |
| WCMD_type (p); |
| break; |
| case WCMD_VER: |
| WCMD_version (); |
| break; |
| case WCMD_VERIFY: |
| WCMD_verify (p); |
| break; |
| case WCMD_VOL: |
| WCMD_volume (0, p); |
| break; |
| case WCMD_PUSHD: |
| WCMD_pushd(p); |
| break; |
| case WCMD_POPD: |
| WCMD_popd(); |
| break; |
| case WCMD_ASSOC: |
| WCMD_assoc(p, TRUE); |
| break; |
| case WCMD_COLOR: |
| WCMD_color(); |
| break; |
| case WCMD_FTYPE: |
| WCMD_assoc(p, FALSE); |
| break; |
| case WCMD_EXIT: |
| WCMD_exit (); |
| break; |
| default: |
| WCMD_run_program (whichcmd, 0); |
| } |
| HeapFree( GetProcessHeap(), 0, cmd ); |
| |
| /* Restore old handles */ |
| for (i=0; i<3; i++) { |
| if (old_stdhandles[i] != INVALID_HANDLE_VALUE) { |
| CloseHandle (GetStdHandle (idx_stdhandles[i])); |
| SetStdHandle (idx_stdhandles[i], old_stdhandles[i]); |
| } |
| } |
| } |
| |
| static void init_msvcrt_io_block(STARTUPINFO* st) |
| { |
| STARTUPINFO st_p; |
| /* fetch the parent MSVCRT info block if any, so that the child can use the |
| * same handles as its grand-father |
| */ |
| st_p.cb = sizeof(STARTUPINFO); |
| GetStartupInfo(&st_p); |
| st->cbReserved2 = st_p.cbReserved2; |
| st->lpReserved2 = st_p.lpReserved2; |
| if (st_p.cbReserved2 && st_p.lpReserved2) |
| { |
| /* Override the entries for fd 0,1,2 if we happened |
| * to change those std handles (this depends on the way wcmd sets |
| * it's new input & output handles) |
| */ |
| size_t sz = max(sizeof(unsigned) + (sizeof(char) + sizeof(HANDLE)) * 3, st_p.cbReserved2); |
| BYTE* ptr = HeapAlloc(GetProcessHeap(), 0, sz); |
| if (ptr) |
| { |
| unsigned num = *(unsigned*)st_p.lpReserved2; |
| char* flags = (char*)(ptr + sizeof(unsigned)); |
| HANDLE* handles = (HANDLE*)(flags + num * sizeof(char)); |
| |
| memcpy(ptr, st_p.lpReserved2, st_p.cbReserved2); |
| st->cbReserved2 = sz; |
| st->lpReserved2 = ptr; |
| |
| #define WX_OPEN 0x01 /* see dlls/msvcrt/file.c */ |
| if (num <= 0 || (flags[0] & WX_OPEN)) |
| { |
| handles[0] = GetStdHandle(STD_INPUT_HANDLE); |
| flags[0] |= WX_OPEN; |
| } |
| if (num <= 1 || (flags[1] & WX_OPEN)) |
| { |
| handles[1] = GetStdHandle(STD_OUTPUT_HANDLE); |
| flags[1] |= WX_OPEN; |
| } |
| if (num <= 2 || (flags[2] & WX_OPEN)) |
| { |
| handles[2] = GetStdHandle(STD_ERROR_HANDLE); |
| flags[2] |= WX_OPEN; |
| } |
| #undef WX_OPEN |
| } |
| } |
| } |
| |
| /****************************************************************************** |
| * WCMD_run_program |
| * |
| * Execute a command line as an external program. Must allow recursion. |
| * |
| * Precedence: |
| * Manual testing under windows shows PATHEXT plays a key part in this, |
| * and the search algorithm and precedence appears to be as follows. |
| * |
| * Search locations: |
| * If directory supplied on command, just use that directory |
| * If extension supplied on command, look for that explicit name first |
| * Otherwise, search in each directory on the path |
| * Precedence: |
| * If extension supplied on command, look for that explicit name first |
| * Then look for supplied name .* (even if extension supplied, so |
| * 'garbage.exe' will match 'garbage.exe.cmd') |
| * If any found, cycle through PATHEXT looking for name.exe one by one |
| * Launching |
| * Once a match has been found, it is launched - Code currently uses |
| * findexecutable to acheive this which is left untouched. |
| */ |
| |
| void WCMD_run_program (char *command, int called) { |
| |
| char temp[MAX_PATH]; |
| char pathtosearch[MAX_PATH]; |
| char *pathposn; |
| char stemofsearch[MAX_PATH]; |
| char *lastSlash; |
| char pathext[MAXSTRING]; |
| BOOL extensionsupplied = FALSE; |
| BOOL launched = FALSE; |
| BOOL status; |
| DWORD len; |
| |
| |
| WCMD_parse (command, quals, param1, param2); /* Quick way to get the filename */ |
| if (!(*param1) && !(*param2)) |
| return; |
| |
| /* Calculate the search path and stem to search for */ |
| if (strpbrk (param1, "/\\:") == NULL) { /* No explicit path given, search path */ |
| strcpy(pathtosearch,".;"); |
| len = GetEnvironmentVariable ("PATH", &pathtosearch[2], sizeof(pathtosearch)-2); |
| if ((len == 0) || (len >= sizeof(pathtosearch) - 2)) { |
| lstrcpy (pathtosearch, "."); |
| } |
| if (strchr(param1, '.') != NULL) extensionsupplied = TRUE; |
| strcpy(stemofsearch, param1); |
| |
| } else { |
| |
| /* Convert eg. ..\fred to include a directory by removing file part */ |
| GetFullPathName(param1, MAX_PATH, pathtosearch, NULL); |
| lastSlash = strrchr(pathtosearch, '\\'); |
| if (lastSlash) *lastSlash = 0x00; |
| if (strchr(lastSlash, '.') != NULL) extensionsupplied = TRUE; |
| strcpy(stemofsearch, lastSlash+1); |
| } |
| |
| /* Now extract PATHEXT */ |
| len = GetEnvironmentVariable ("PATHEXT", pathext, sizeof(pathext)); |
| if ((len == 0) || (len >= sizeof(pathext))) { |
| lstrcpy (pathext, ".bat;.com;.cmd;.exe"); |
| } |
| |
| /* Loop through the search path, dir by dir */ |
| pathposn = pathtosearch; |
| while (!launched && pathposn) { |
| |
| char thisDir[MAX_PATH] = ""; |
| char *pos = NULL; |
| BOOL found = FALSE; |
| |
| /* Work on the first directory on the search path */ |
| pos = strchr(pathposn, ';'); |
| if (pos) { |
| strncpy(thisDir, pathposn, (pos-pathposn)); |
| thisDir[(pos-pathposn)] = 0x00; |
| pathposn = pos+1; |
| |
| } else { |
| strcpy(thisDir, pathposn); |
| pathposn = NULL; |
| } |
| |
| /* Since you can have eg. ..\.. on the path, need to expand |
| to full information */ |
| strcpy(temp, thisDir); |
| GetFullPathName(temp, MAX_PATH, thisDir, NULL); |
| |
| /* 1. If extension supplied, see if that file exists */ |
| strcat(thisDir, "\\"); |
| strcat(thisDir, stemofsearch); |
| pos = &thisDir[strlen(thisDir)]; /* Pos = end of name */ |
| |
| if (GetFileAttributes(thisDir) != INVALID_FILE_ATTRIBUTES) { |
| found = TRUE; |
| } |
| |
| /* 2. Any .* matches? */ |
| if (!found) { |
| HANDLE h; |
| WIN32_FIND_DATA finddata; |
| |
| strcat(thisDir,".*"); |
| h = FindFirstFile(thisDir, &finddata); |
| FindClose(h); |
| if (h != INVALID_HANDLE_VALUE) { |
| |
| char *thisExt = pathext; |
| |
| /* 3. Yes - Try each path ext */ |
| while (thisExt) { |
| char *nextExt = strchr(thisExt, ';'); |
| |
| if (nextExt) { |
| strncpy(pos, thisExt, (nextExt-thisExt)); |
| pos[(nextExt-thisExt)] = 0x00; |
| thisExt = nextExt+1; |
| } else { |
| strcpy(pos, thisExt); |
| thisExt = NULL; |
| } |
| |
| if (GetFileAttributes(thisDir) != INVALID_FILE_ATTRIBUTES) { |
| found = TRUE; |
| thisExt = NULL; |
| } |
| } |
| } |
| } |
| |
| /* Once found, launch it */ |
| if (found) { |
| STARTUPINFO st; |
| PROCESS_INFORMATION pe; |
| SHFILEINFO psfi; |
| DWORD console; |
| HINSTANCE hinst; |
| char *ext = strrchr( thisDir, '.' ); |
| launched = TRUE; |
| |
| /* Special case BAT and CMD */ |
| if (ext && !strcasecmp(ext, ".bat")) { |
| WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE); |
| return; |
| } else if (ext && !strcasecmp(ext, ".cmd")) { |
| WCMD_batch (thisDir, command, called, NULL, INVALID_HANDLE_VALUE); |
| return; |
| } else { |
| |
| /* thisDir contains the file to be launched, but with what? |
| eg. a.exe will require a.exe to be launched, a.html may be iexplore */ |
| hinst = FindExecutable (param1, NULL, temp); |
| if ((INT_PTR)hinst < 32) |
| console = 0; |
| else |
| console = SHGetFileInfo (temp, 0, &psfi, sizeof(psfi), SHGFI_EXETYPE); |
| |
| ZeroMemory (&st, sizeof(STARTUPINFO)); |
| st.cb = sizeof(STARTUPINFO); |
| init_msvcrt_io_block(&st); |
| |
| /* Launch the process and if a CUI wait on it to complete */ |
| status = CreateProcess (thisDir, command, NULL, NULL, TRUE, |
| 0, NULL, NULL, &st, &pe); |
| if ((opt_c || opt_k) && !opt_s && !status |
| && GetLastError()==ERROR_FILE_NOT_FOUND && command[0]=='\"') { |
| /* strip first and last quote characters and try again */ |
| WCMD_opt_s_strip_quotes(command); |
| opt_s=1; |
| WCMD_run_program(command, called); |
| return; |
| } |
| if (!status) { |
| WCMD_print_error (); |
| /* If a command fails to launch, it sets errorlevel 9009 - which |
| does not seem to have any associated constant definition */ |
| errorlevel = 9009; |
| return; |
| } |
| if (!console) errorlevel = 0; |
| else |
| { |
| if (!HIWORD(console)) WaitForSingleObject (pe.hProcess, INFINITE); |
| GetExitCodeProcess (pe.hProcess, &errorlevel); |
| if (errorlevel == STILL_ACTIVE) errorlevel = 0; |
| } |
| CloseHandle(pe.hProcess); |
| CloseHandle(pe.hThread); |
| return; |
| } |
| } |
| } |
| |
| /* Not found anywhere - give up */ |
| SetLastError(ERROR_FILE_NOT_FOUND); |
| WCMD_print_error (); |
| |
| /* If a command fails to launch, it sets errorlevel 9009 - which |
| does not seem to have any associated constant definition */ |
| errorlevel = 9009; |
| return; |
| |
| } |
| |
| /****************************************************************************** |
| * WCMD_show_prompt |
| * |
| * Display the prompt on STDout |
| * |
| */ |
| |
| void WCMD_show_prompt (void) { |
| |
| int status; |
| char out_string[MAX_PATH], curdir[MAX_PATH], prompt_string[MAX_PATH]; |
| char *p, *q; |
| DWORD len; |
| |
| len = GetEnvironmentVariable ("PROMPT", prompt_string, sizeof(prompt_string)); |
| if ((len == 0) || (len >= sizeof(prompt_string))) { |
| lstrcpy (prompt_string, "$P$G"); |
| } |
| p = prompt_string; |
| q = out_string; |
| *q = '\0'; |
| while (*p != '\0') { |
| if (*p != '$') { |
| *q++ = *p++; |
| *q = '\0'; |
| } |
| else { |
| p++; |
| switch (toupper(*p)) { |
| case '$': |
| *q++ = '$'; |
| break; |
| case 'A': |
| *q++ = '&'; |
| break; |
| case 'B': |
| *q++ = '|'; |
| break; |
| case 'C': |
| *q++ = '('; |
| break; |
| case 'D': |
| GetDateFormat (LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL, NULL, q, MAX_PATH); |
| while (*q) q++; |
| break; |
| case 'E': |
| *q++ = '\E'; |
| break; |
| case 'F': |
| *q++ = ')'; |
| break; |
| case 'G': |
| *q++ = '>'; |
| break; |
| case 'H': |
| *q++ = '\b'; |
| break; |
| case 'L': |
| *q++ = '<'; |
| break; |
| case 'N': |
| status = GetCurrentDirectory (sizeof(curdir), curdir); |
| if (status) { |
| *q++ = curdir[0]; |
| } |
| break; |
| case 'P': |
| status = GetCurrentDirectory (sizeof(curdir), curdir); |
| if (status) { |
| lstrcat (q, curdir); |
| while (*q) q++; |
| } |
| break; |
| case 'Q': |
| *q++ = '='; |
| break; |
| case 'S': |
| *q++ = ' '; |
| break; |
| case 'T': |
| GetTimeFormat (LOCALE_USER_DEFAULT, 0, NULL, NULL, q, MAX_PATH); |
| while (*q) q++; |
| break; |
| case 'V': |
| lstrcat (q, version_string); |
| while (*q) q++; |
| break; |
| case '_': |
| *q++ = '\n'; |
| break; |
| case '+': |
| if (pushd_directories) { |
| memset(q, '+', pushd_directories->u.stackdepth); |
| q = q + pushd_directories->u.stackdepth; |
| } |
| break; |
| } |
| p++; |
| *q = '\0'; |
| } |
| } |
| WCMD_output_asis (out_string); |
| } |
| |
| /**************************************************************************** |
| * WCMD_print_error |
| * |
| * Print the message for GetLastError |
| */ |
| |
| void WCMD_print_error (void) { |
| LPVOID lpMsgBuf; |
| DWORD error_code; |
| int status; |
| |
| error_code = GetLastError (); |
| status = FormatMessage (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, |
| NULL, error_code, 0, (LPTSTR) &lpMsgBuf, 0, NULL); |
| if (!status) { |
| WCMD_output ("FIXME: Cannot display message for error %d, status %d\n", |
| error_code, GetLastError()); |
| return; |
| } |
| WCMD_output_asis (lpMsgBuf); |
| LocalFree ((HLOCAL)lpMsgBuf); |
| WCMD_output_asis (newline); |
| return; |
| } |
| |
| /******************************************************************* |
| * WCMD_parse - parse a command into parameters and qualifiers. |
| * |
| * On exit, all qualifiers are concatenated into q, the first string |
| * not beginning with "/" is in p1 and the |
| * second in p2. Any subsequent non-qualifier strings are lost. |
| * Parameters in quotes are handled. |
| */ |
| |
| void WCMD_parse (char *s, char *q, char *p1, char *p2) { |
| |
| int p = 0; |
| |
| *q = *p1 = *p2 = '\0'; |
| while (TRUE) { |
| switch (*s) { |
| case '/': |
| *q++ = *s++; |
| while ((*s != '\0') && (*s != ' ') && *s != '/') { |
| *q++ = toupper (*s++); |
| } |
| *q = '\0'; |
| break; |
| case ' ': |
| case '\t': |
| s++; |
| break; |
| case '"': |
| s++; |
| while ((*s != '\0') && (*s != '"')) { |
| if (p == 0) *p1++ = *s++; |
| else if (p == 1) *p2++ = *s++; |
| else s++; |
| } |
| if (p == 0) *p1 = '\0'; |
| if (p == 1) *p2 = '\0'; |
| p++; |
| if (*s == '"') s++; |
| break; |
| case '\0': |
| return; |
| default: |
| while ((*s != '\0') && (*s != ' ') && (*s != '\t')) { |
| if (p == 0) *p1++ = *s++; |
| else if (p == 1) *p2++ = *s++; |
| else s++; |
| } |
| if (p == 0) *p1 = '\0'; |
| if (p == 1) *p2 = '\0'; |
| p++; |
| } |
| } |
| } |
| |
| /******************************************************************* |
| * WCMD_output - send output to current standard output device. |
| * |
| */ |
| |
| void WCMD_output (const char *format, ...) { |
| |
| va_list ap; |
| char string[1024]; |
| int ret; |
| |
| va_start(ap,format); |
| ret = vsnprintf (string, sizeof( string), format, ap); |
| va_end(ap); |
| if( ret >= sizeof( string)) { |
| WCMD_output_asis("ERR: output truncated in WCMD_output\n" ); |
| string[sizeof( string) -1] = '\0'; |
| } |
| WCMD_output_asis(string); |
| } |
| |
| |
| static int line_count; |
| static int max_height; |
| static BOOL paged_mode; |
| |
| void WCMD_enter_paged_mode(void) |
| { |
| CONSOLE_SCREEN_BUFFER_INFO consoleInfo; |
| |
| if (GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &consoleInfo)) |
| max_height = consoleInfo.dwSize.Y; |
| else |
| max_height = 25; |
| paged_mode = TRUE; |
| line_count = 5; /* keep 5 lines from previous output */ |
| } |
| |
| void WCMD_leave_paged_mode(void) |
| { |
| paged_mode = FALSE; |
| } |
| |
| /******************************************************************* |
| * WCMD_output_asis - send output to current standard output device. |
| * without formatting eg. when message contains '%' |
| */ |
| |
| void WCMD_output_asis (const char *message) { |
| DWORD count; |
| char* ptr; |
| char string[1024]; |
| |
| if (paged_mode) { |
| do { |
| if ((ptr = strchr(message, '\n')) != NULL) ptr++; |
| WriteFile (GetStdHandle(STD_OUTPUT_HANDLE), message, |
| (ptr) ? ptr - message : lstrlen(message), &count, NULL); |
| if (ptr) { |
| if (++line_count >= max_height - 1) { |
| line_count = 0; |
| WCMD_output_asis (anykey); |
| ReadFile (GetStdHandle(STD_INPUT_HANDLE), string, sizeof(string), &count, NULL); |
| } |
| } |
| } while ((message = ptr) != NULL); |
| } else { |
| WriteFile (GetStdHandle(STD_OUTPUT_HANDLE), message, lstrlen(message), &count, NULL); |
| } |
| } |
| |
| |
| /*************************************************************************** |
| * WCMD_strtrim_leading_spaces |
| * |
| * Remove leading spaces from a string. Return a pointer to the first |
| * non-space character. Does not modify the input string |
| */ |
| |
| char *WCMD_strtrim_leading_spaces (char *string) { |
| |
| char *ptr; |
| |
| ptr = string; |
| while (*ptr == ' ') ptr++; |
| return ptr; |
| } |
| |
| /************************************************************************* |
| * WCMD_strtrim_trailing_spaces |
| * |
| * Remove trailing spaces from a string. This routine modifies the input |
| * string by placing a null after the last non-space character |
| */ |
| |
| void WCMD_strtrim_trailing_spaces (char *string) { |
| |
| char *ptr; |
| |
| ptr = string + lstrlen (string) - 1; |
| while ((*ptr == ' ') && (ptr >= string)) { |
| *ptr = '\0'; |
| ptr--; |
| } |
| } |
| |
| /************************************************************************* |
| * WCMD_opt_s_strip_quotes |
| * |
| * Remove first and last quote characters, preserving all other text |
| */ |
| |
| void WCMD_opt_s_strip_quotes(char *cmd) { |
| char *src = cmd + 1, *dest = cmd, *lastq = NULL; |
| while((*dest=*src) != '\0') { |
| if (*src=='\"') |
| lastq=dest; |
| dest++, src++; |
| } |
| if (lastq) { |
| dest=lastq++; |
| while ((*dest++=*lastq++) != 0) |
| ; |
| } |
| } |
| |
| /************************************************************************* |
| * WCMD_pipe |
| * |
| * Handle pipes within a command - the DOS way using temporary files. |
| */ |
| |
| void WCMD_pipe (char *command) { |
| |
| char *p; |
| char temp_path[MAX_PATH], temp_file[MAX_PATH], temp_file2[MAX_PATH], temp_cmd[1024]; |
| |
| GetTempPath (sizeof(temp_path), temp_path); |
| GetTempFileName (temp_path, "CMD", 0, temp_file); |
| p = strchr(command, '|'); |
| *p++ = '\0'; |
| wsprintf (temp_cmd, "%s > %s", command, temp_file); |
| WCMD_process_command (temp_cmd); |
| command = p; |
| while ((p = strchr(command, '|'))) { |
| *p++ = '\0'; |
| GetTempFileName (temp_path, "CMD", 0, temp_file2); |
| wsprintf (temp_cmd, "%s < %s > %s", command, temp_file, temp_file2); |
| WCMD_process_command (temp_cmd); |
| DeleteFile (temp_file); |
| lstrcpy (temp_file, temp_file2); |
| command = p; |
| } |
| wsprintf (temp_cmd, "%s < %s", command, temp_file); |
| WCMD_process_command (temp_cmd); |
| DeleteFile (temp_file); |
| } |
| |
| /************************************************************************* |
| * WCMD_expand_envvar |
| * |
| * Expands environment variables, allowing for character substitution |
| */ |
| static char *WCMD_expand_envvar(char *start) { |
| char *endOfVar = NULL, *s; |
| char *colonpos = NULL; |
| char thisVar[MAXSTRING]; |
| char thisVarContents[MAXSTRING]; |
| char savedchar = 0x00; |
| int len; |
| |
| /* Find the end of the environment variable, and extract name */ |
| endOfVar = strchr(start+1, '%'); |
| if (endOfVar == NULL) { |
| /* FIXME: Some special conditions here depending opn whether |
| in batch, complex or not, and whether env var exists or not! */ |
| return start+1; |
| } |
| strncpy(thisVar, start, (endOfVar - start)+1); |
| thisVar[(endOfVar - start)+1] = 0x00; |
| colonpos = strchr(thisVar+1, ':'); |
| |
| /* If there's complex substitution, just need %var% for now |
| to get the expanded data to play with */ |
| if (colonpos) { |
| *colonpos = '%'; |
| savedchar = *(colonpos+1); |
| *(colonpos+1) = 0x00; |
| } |
| |
| /* Expand to contents, if unchanged, return */ |
| /* Handle DATE, TIME, ERRORLEVEL and CD replacements allowing */ |
| /* override if existing env var called that name */ |
| if ((CompareString (LOCALE_USER_DEFAULT, |
| NORM_IGNORECASE | SORT_STRINGSORT, |
| thisVar, 12, "%ERRORLEVEL%", -1) == 2) && |
| (GetEnvironmentVariable("ERRORLEVEL", thisVarContents, 1) == 0) && |
| (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) { |
| sprintf(thisVarContents, "%d", errorlevel); |
| len = strlen(thisVarContents); |
| |
| } else if ((CompareString (LOCALE_USER_DEFAULT, |
| NORM_IGNORECASE | SORT_STRINGSORT, |
| thisVar, 6, "%DATE%", -1) == 2) && |
| (GetEnvironmentVariable("DATE", thisVarContents, 1) == 0) && |
| (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) { |
| |
| GetDateFormat(LOCALE_USER_DEFAULT, DATE_SHORTDATE, NULL, |
| NULL, thisVarContents, MAXSTRING); |
| len = strlen(thisVarContents); |
| |
| } else if ((CompareString (LOCALE_USER_DEFAULT, |
| NORM_IGNORECASE | SORT_STRINGSORT, |
| thisVar, 6, "%TIME%", -1) == 2) && |
| (GetEnvironmentVariable("TIME", thisVarContents, 1) == 0) && |
| (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) { |
| GetTimeFormat(LOCALE_USER_DEFAULT, TIME_NOSECONDS, NULL, |
| NULL, thisVarContents, MAXSTRING); |
| len = strlen(thisVarContents); |
| |
| } else if ((CompareString (LOCALE_USER_DEFAULT, |
| NORM_IGNORECASE | SORT_STRINGSORT, |
| thisVar, 4, "%CD%", -1) == 2) && |
| (GetEnvironmentVariable("CD", thisVarContents, 1) == 0) && |
| (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) { |
| GetCurrentDirectory (MAXSTRING, thisVarContents); |
| len = strlen(thisVarContents); |
| |
| } else if ((CompareString (LOCALE_USER_DEFAULT, |
| NORM_IGNORECASE | SORT_STRINGSORT, |
| thisVar, 8, "%RANDOM%", -1) == 2) && |
| (GetEnvironmentVariable("RANDOM", thisVarContents, 1) == 0) && |
| (GetLastError() == ERROR_ENVVAR_NOT_FOUND)) { |
| sprintf(thisVarContents, "%d", rand() % 32768); |
| len = strlen(thisVarContents); |
| |
| } else { |
| |
| len = ExpandEnvironmentStrings(thisVar, thisVarContents, |
| sizeof(thisVarContents)); |
| } |
| |
| if (len == 0) |
| return endOfVar+1; |
| |
| /* In a batch program, unknown env vars are replaced with nothing, |
| note syntax %garbage:1,3% results in anything after the ':' |
| except the % |
| From the command line, you just get back what you entered */ |
| if (lstrcmpi(thisVar, thisVarContents) == 0) { |
| |
| /* Restore the complex part after the compare */ |
| if (colonpos) { |
| *colonpos = ':'; |
| *(colonpos+1) = savedchar; |
| } |
| |
| s = strdup (endOfVar + 1); |
| |
| /* Command line - just ignore this */ |
| if (context == NULL) return endOfVar+1; |
| |
| /* Batch - replace unknown env var with nothing */ |
| if (colonpos == NULL) { |
| strcpy (start, s); |
| |
| } else { |
| len = strlen(thisVar); |
| thisVar[len-1] = 0x00; |
| /* If %:...% supplied, : is retained */ |
| if (colonpos == thisVar+1) { |
| strcpy (start, colonpos); |
| } else { |
| strcpy (start, colonpos+1); |
| } |
| strcat (start, s); |
| } |
| free (s); |
| return start; |
| |
| } |
| |
| /* See if we need to do complex substitution (any ':'s), if not |
| then our work here is done */ |
| if (colonpos == NULL) { |
| s = strdup (endOfVar + 1); |
| strcpy (start, thisVarContents); |
| strcat (start, s); |
| free(s); |
| return start; |
| } |
| |
| /* Restore complex bit */ |
| *colonpos = ':'; |
| *(colonpos+1) = savedchar; |
| |
| /* |
| Handle complex substitutions: |
| xxx=yyy (replace xxx with yyy) |
| *xxx=yyy (replace up to and including xxx with yyy) |
| ~x (from x chars in) |
| ~-x (from x chars from the end) |
| ~x,y (from x chars in for y characters) |
| ~x,-y (from x chars in until y characters from the end) |
| */ |
| |
| /* ~ is substring manipulation */ |
| if (savedchar == '~') { |
| |
| int substrposition, substrlength = 0; |
| char *commapos = strchr(colonpos+2, ','); |
| char *startCopy; |
| |
| substrposition = atol(colonpos+2); |
| if (commapos) substrlength = atol(commapos+1); |
| |
| s = strdup (endOfVar + 1); |
| |
| /* Check bounds */ |
| if (substrposition >= 0) { |
| startCopy = &thisVarContents[min(substrposition, len)]; |
| } else { |
| startCopy = &thisVarContents[max(0, len+substrposition-1)]; |
| } |
| |
| if (commapos == NULL) { |
| strcpy (start, startCopy); /* Copy the lot */ |
| } else if (substrlength < 0) { |
| |
| int copybytes = (len+substrlength-1)-(startCopy-thisVarContents); |
| if (copybytes > len) copybytes = len; |
| else if (copybytes < 0) copybytes = 0; |
| strncpy (start, startCopy, copybytes); /* Copy the lot */ |
| start[copybytes] = 0x00; |
| } else { |
| strncpy (start, startCopy, substrlength); /* Copy the lot */ |
| start[substrlength] = 0x00; |
| } |
| |
| strcat (start, s); |
| free(s); |
| return start; |
| |
| /* search and replace manipulation */ |
| } else { |
| char *equalspos = strstr(colonpos, "="); |
| char *replacewith = equalspos+1; |
| char *found = NULL; |
| char *searchIn; |
| char *searchFor; |
| |
| s = strdup (endOfVar + 1); |
| if (equalspos == NULL) return start+1; |
| |
| /* Null terminate both strings */ |
| thisVar[strlen(thisVar)-1] = 0x00; |
| *equalspos = 0x00; |
| |
| /* Since we need to be case insensitive, copy the 2 buffers */ |
| searchIn = strdup(thisVarContents); |
| CharUpperBuff(searchIn, strlen(thisVarContents)); |
| searchFor = strdup(colonpos+1); |
| CharUpperBuff(searchFor, strlen(colonpos+1)); |
| |
| |
| /* Handle wildcard case */ |
| if (*(colonpos+1) == '*') { |
| /* Search for string to replace */ |
| found = strstr(searchIn, searchFor+1); |
| |
| if (found) { |
| /* Do replacement */ |
| strcpy(start, replacewith); |
| strcat(start, thisVarContents + (found-searchIn) + strlen(searchFor+1)); |
| strcat(start, s); |
| free(s); |
| } else { |
| /* Copy as it */ |
| strcpy(start, thisVarContents); |
| strcat(start, s); |
| } |
| |
| } else { |
| /* Loop replacing all instances */ |
| char *lastFound = searchIn; |
| char *outputposn = start; |
| |
| *start = 0x00; |
| while ((found = strstr(lastFound, searchFor))) { |
| strncpy(outputposn, |
| thisVarContents + (lastFound-searchIn), |
| (found - lastFound)); |
| outputposn = outputposn + (found - lastFound); |
| *outputposn = 0x00; |
| strcat(outputposn, replacewith); |
| outputposn = outputposn + strlen(replacewith); |
| lastFound = found + strlen(searchFor); |
| } |
| strcat(outputposn, |
| thisVarContents + (lastFound-searchIn)); |
| strcat(outputposn, s); |
| } |
| free(searchIn); |
| free(searchFor); |
| return start; |
| } |
| return start+1; |
| } |
