/* * Windows regedit.exe registry editor implementation. * * Copyright 2002 Andriy Palamarchuk * * 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 */ #include #include #include #include #include "regproc.h" static const char *usage = "Usage:\n" " regedit filename\n" " regedit /E filename [regpath]\n" " regedit /D regpath\n" "\n" "filename - registry file name\n" "regpath - name of the registry key\n" "\n" "When called without any switches, adds the content of the specified\n" "file to the registry\n" "\n" "Switches:\n" " /E - exports contents of the specified registry key to the specified\n" " file. Exports the whole registry if no key is specified.\n" " /D - deletes specified registry key\n" " /S - silent execution, can be used with any other switch.\n" " Default. The only existing mode, exists for compatibility with Windows regedit.\n" " /V - advanced mode, can be used with any other switch.\n" " Ignored, exists for compatibility with Windows regedit.\n" " /L - location of system.dat file. Can be used with any other switch.\n" " Ignored. Exists for compatibility with Windows regedit.\n" " /R - location of user.dat file. Can be used with any other switch.\n" " Ignored. Exists for compatibility with Windows regedit.\n" " /? - print this help. Any other switches are ignored.\n" " /C - create registry from file. Not implemented.\n" "\n" "The switches are case-insensitive, can be prefixed either by '-' or '/'.\n" "This program is command-line compatible with Microsoft Windows\n" "regedit.\n"; typedef enum { ACTION_UNDEF, ACTION_ADD, ACTION_EXPORT, ACTION_DELETE } REGEDIT_ACTION; const CHAR *getAppName(void) { return "regedit"; } /****************************************************************************** * Copies file name from command line string to the buffer. * Rewinds the command line string pointer to the next non-space character * after the file name. * Buffer contains an empty string if no filename was found; * * params: * command_line - command line current position pointer * where *s[0] is the first symbol of the file name. * file_name - buffer to write the file name to. */ static void get_file_name(CHAR **command_line, CHAR *file_name) { CHAR *s = *command_line; int pos = 0; /* position of pointer "s" in *command_line */ file_name[0] = 0; if (!s[0]) { return; } if (s[0] == '"') { s++; (*command_line)++; while(s[0] != '"') { if (!s[0]) { fprintf(stderr,"%s: Unexpected end of file name!\n", getAppName()); exit(1); } s++; pos++; } } else { while(s[0] && !isspace(s[0])) { s++; pos++; } } memcpy(file_name, *command_line, pos * sizeof((*command_line)[0])); /* Terminate the string and remove the trailing backslash */ if (pos > 0 && file_name[pos - 1] == '\\') { file_name[pos - 1] = '\0'; } else { file_name[pos] = '\0'; } if (s[0]) { s++; pos++; } while(s[0] && isspace(s[0])) { s++; pos++; } (*command_line) += pos; } static BOOL PerformRegAction(REGEDIT_ACTION action, LPSTR s) { switch (action) { case ACTION_ADD: { CHAR filename[MAX_PATH]; FILE *reg_file; get_file_name(&s, filename); if (!filename[0]) { fprintf(stderr,"%s: No file name was specified\n", getAppName()); fprintf(stderr,usage); exit(1); } while(filename[0]) { char* realname = NULL; if (strcmp(filename, "-") == 0) { reg_file = stdin; } else { int size; size = SearchPathA(NULL, filename, NULL, 0, NULL, NULL); if (size > 0) { realname = HeapAlloc(GetProcessHeap(), 0, size); size = SearchPathA(NULL, filename, NULL, size, realname, NULL); } if (size == 0) { fprintf(stderr, "%s: File not found \"%s\" (%d)\n", getAppName(), filename, GetLastError()); exit(1); } reg_file = fopen(realname, "rb"); if (reg_file == NULL) { perror(""); fprintf(stderr, "%s: Can't open file \"%s\"\n", getAppName(), filename); exit(1); } } import_registry_file(reg_file); if (realname) { HeapFree(GetProcessHeap(),0,realname); fclose(reg_file); } get_file_name(&s, filename); } break; } case ACTION_DELETE: { CHAR reg_key_name[KEY_MAX_LEN]; get_file_name(&s, reg_key_name); if (!reg_key_name[0]) { fprintf(stderr,"%s: No registry key was specified for removal\n", getAppName()); fprintf(stderr,usage); exit(1); } else { WCHAR* reg_key_nameW = GetWideString(reg_key_name); delete_registry_key(reg_key_nameW); HeapFree(GetProcessHeap(), 0, reg_key_nameW); } break; } case ACTION_EXPORT: { CHAR filename[MAX_PATH]; WCHAR* filenameW; filename[0] = '\0'; get_file_name(&s, filename); if (!filename[0]) { fprintf(stderr,"%s: No file name was specified\n", getAppName()); fprintf(stderr,usage); exit(1); } filenameW = GetWideString(filename); if (s[0]) { CHAR reg_key_name[KEY_MAX_LEN]; WCHAR* reg_key_nameW; get_file_name(&s, reg_key_name); reg_key_nameW = GetWideString(reg_key_name); export_registry_key(filenameW, reg_key_nameW, REG_FORMAT_4); HeapFree(GetProcessHeap(), 0, reg_key_nameW); } else { export_registry_key(filenameW, NULL, REG_FORMAT_4); } HeapFree(GetProcessHeap(), 0, filenameW); break; } default: fprintf(stderr,"%s: Unhandled action!\n", getAppName()); exit(1); break; } return TRUE; } /** * Process unknown switch. * * Params: * chu - the switch character in upper-case. * s - the command line string where s points to the switch character. */ static void error_unknown_switch(char chu, char *s) { if (isalpha(chu)) { fprintf(stderr,"%s: Undefined switch /%c!\n", getAppName(), chu); } else { fprintf(stderr,"%s: Alphabetic character is expected after '%c' " "in switch specification\n", getAppName(), *(s - 1)); } exit(1); } BOOL ProcessCmdLine(LPSTR lpCmdLine) { REGEDIT_ACTION action = ACTION_UNDEF; LPSTR s = lpCmdLine; /* command line pointer */ CHAR ch = *s; /* current character */ while (ch && ((ch == '-') || (ch == '/'))) { char chu; char ch2; s++; ch = *s; if (!ch || isspace(ch)) { /* '-' is a file name. It indicates we should use stdin */ s--; break; } ch2 = *(s+1); chu = toupper(ch); if (!ch2 || isspace(ch2)) { if (chu == 'S' || chu == 'V') { /* ignore these switches */ } else { switch (chu) { case 'D': action = ACTION_DELETE; break; case 'E': action = ACTION_EXPORT; break; case '?': fprintf(stderr,usage); exit(0); break; default: error_unknown_switch(chu, s); break; } } s++; } else { if (ch2 == ':') { switch (chu) { case 'L': /* fall through */ case 'R': s += 2; while (*s && !isspace(*s)) { s++; } break; default: error_unknown_switch(chu, s); break; } } else { /* this is a file name, starting from '/' */ s--; break; } } /* skip spaces to the next parameter */ ch = *s; while (ch && isspace(ch)) { s++; ch = *s; } } if (*s && action == ACTION_UNDEF) action = ACTION_ADD; if (action == ACTION_UNDEF) return FALSE; return PerformRegAction(action, s); }