584033a956f888d676f9d44862382b61b41d59ce
[reactos.git] / rosapps / sysutils / regexpl / RegistryExplorer.cpp
1 /* $Id: RegistryExplorer.cpp,v 1.5 2001/04/15 22:17:50 narnaoud Exp $
2 *
3 * regexpl - Console Registry Explorer
4 *
5 * Copyright (C) 2000,2001 Nedko Arnaoudov <nedkohome@atia.com>
6 *
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License as published by
9 * the Free Software Foundation; either version 2 of the License, or
10 * (at your option) any later version.
11 *
12 * This program 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
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; see the file COPYING. If not, write to
19 * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20 * Boston, MA 02111-1307, USA.
21 */
22
23 // Registry Explorer.cpp : Defines the entry point for the console application.
24 //
25
26 #include "ph.h"
27 #include "RegistryExplorer.h"
28 #include "Console.h"
29 #include "RegistryKey.h"
30 #include "RegistryTree.h"
31 #include "SecurityDescriptor.h"
32 #include "ArgumentParser.h"
33 #include "ShellCommandsLinkedList.h"
34 #include "ShellCommandExit.h"
35 #include "ShellCommandVersion.h"
36 #include "ShellCommandHelp.h"
37 #include "ShellCommandDir.h"
38 #include "ShellCommandChangeKey.h"
39 #include "ShellCommandValue.h"
40 #include "ShellCommandOwner.h"
41 #include "ShellCommandDACL.h"
42 #include "ShellCommandSACL.h"
43 //#include "ShellCommandDOKA.h"
44 #include "ShellCommandConnect.h"
45 #include "ShellCommandNewKey.h"
46 #include "ShellCommandDeleteKey.h"
47 #include "ShellCommandSetValue.h"
48 #include "ShellCommandDeleteValue.h"
49
50 TCHAR pchCurrentKey[PROMPT_BUFFER_SIZE];
51
52 CRegistryTree Tree;
53 CConsole Console;
54
55 BOOL blnCommandExecutionInProgress = FALSE;
56
57 BOOL WINAPI HandlerRoutine(DWORD dwCtrlType)
58 {
59 switch(dwCtrlType)
60 {
61 case CTRL_C_EVENT:
62 case CTRL_BREAK_EVENT:
63 if (blnCommandExecutionInProgress)
64 {
65 /* Beep(1000,100);
66 Beep(2000,100);
67 Beep(3000,100);
68 Beep(4000,100);
69 Beep(5000,100);
70 Beep(4000,100);
71 Beep(3000,100);
72 Beep(2000,100);
73 Beep(1000,100);*/
74 Console.Write(_T("\n"));
75 Console.DisableWrite();
76 }
77 return TRUE;
78 default:
79 return FALSE;
80 }
81 }
82
83 //int _tmain(/*int argc, TCHAR* argv[], TCHAR* envp[]*/)
84 int main ()
85 {
86 CShellCommandsLinkedList CommandsList(Console);
87
88 CShellCommandExit ExitCommand;
89 CommandsList.AddCommand(&ExitCommand);
90
91 CShellCommandVersion VersionCommand;
92 CommandsList.AddCommand(&VersionCommand);
93
94 CShellCommandHelp HelpCommand(CommandsList);
95 CommandsList.AddCommand(&HelpCommand);
96
97 CShellCommandDir DirCommand(Tree);
98 CommandsList.AddCommand(&DirCommand);
99
100 CShellCommandChangeKey ChangeKeyCommand(Tree);
101 CommandsList.AddCommand(&ChangeKeyCommand);
102
103 CShellCommandValue ValueCommand(Tree);
104 CommandsList.AddCommand(&ValueCommand);
105
106 CShellCommandOwner OwnerCommand(Tree);
107 CommandsList.AddCommand(&OwnerCommand);
108
109 CShellCommandDACL DACLCommand(Tree);
110 CommandsList.AddCommand(&DACLCommand);
111
112 CShellCommandSACL SACLCommand(Tree);
113 CommandsList.AddCommand(&SACLCommand);
114
115 //CShellCommandDOKA DOKACommand(Tree);
116 //CommandsList.AddCommand(&DOKACommand);
117
118 CShellCommandConnect ConnectCommand(Tree);
119 CommandsList.AddCommand(&ConnectCommand);
120
121 CShellCommandNewKey NewKeyCommand(Tree);
122 CommandsList.AddCommand(&NewKeyCommand);
123
124 CShellCommandDeleteKey DeleteKeyCommand(Tree);
125 CommandsList.AddCommand(&DeleteKeyCommand);
126
127 CShellCommandSetValue SetValueCommand(Tree);
128 CommandsList.AddCommand(&SetValueCommand);
129
130 CShellCommandDeleteValue DeleteValueCommand(Tree);
131 CommandsList.AddCommand(&DeleteValueCommand);
132
133 CArgumentParser Parser;
134
135 int nRetCode = 0;
136
137 // input buffer size in chars
138 #define INPUT_BUFFER_SIZE 1024
139 //#define INPUT_BUFFER_SIZE 128
140 //#define INPUT_BUFFER_SIZE 10
141
142 TCHAR *pchCommand;
143
144 pchCommand = Console.Init(INPUT_BUFFER_SIZE,10);
145 if (pchCommand == NULL)
146 {
147 _ftprintf(stderr,_T("Cannot initialize console.\n"));
148 goto Abort;
149 }
150
151 Console.SetReplaceCompletionCallback(CompletionCallback);
152
153 WORD wOldConsoleAttribute;
154 if (!Console.GetTextAttribute(wOldConsoleAttribute)) goto Abort;
155
156 Console.SetTitle(_T("Registry Explorer"));
157 Console.SetTextAttribute(FOREGROUND_BLUE|FOREGROUND_GREEN|FOREGROUND_RED/*|FOREGROUND_INTENSITY*/);
158
159 VERIFY(SetConsoleCtrlHandler((PHANDLER_ROUTINE)HandlerRoutine,TRUE));
160
161 if (!Console.Write(HELLO_MSG
162 //(_L(__TIMESTAMP__))
163 )) goto Abort;
164
165 //Tree.SetDesiredOpenKeyAccess(KEY_READ);
166
167 const TCHAR *pszCurrentPath;
168 GetCommand:
169 // prompt
170 // TODO: make prompt user-customizable
171 Console.EnableWrite();
172 pszCurrentPath = Tree.GetCurrentPath();
173 Console.Write(pszCurrentPath?pszCurrentPath:_T("NULL (Internal Error)"));
174 Console.Write(_T("\n# "));
175 Console.FlushInputBuffer();
176
177 blnCommandExecutionInProgress = FALSE;
178
179 // Set command line color
180 // Console.SetTextAttribute(/*FOREGROUND_BLUE|*/FOREGROUND_GREEN|FOREGROUND_RED|FOREGROUND_INTENSITY);
181 if (!Console.ReadLine())
182 goto Abort;
183
184 // Set normal color
185 // Console.SetTextAttribute(FOREGROUND_BLUE|FOREGROUND_GREEN|FOREGROUND_RED|FOREGROUND_INTENSITY);
186
187 Console.BeginScrollingOperation();
188 blnCommandExecutionInProgress = TRUE;
189
190 // Parse command line (1st step - convert to multi sz)
191 Parser.SetArgumentList(pchCommand);
192
193 int nCommandReturnValue;
194 switch(CommandsList.Execute(Parser,nCommandReturnValue))
195 {
196 case -1: // not recognized command
197 {
198 Parser.ResetArgumentIteration();
199 TCHAR *pchCommandItself = Parser.GetNextArgument();
200 size_t cmdlen = _tcslen(pchCommandItself);
201 if ((!cmdlen)||
202 (pchCommandItself[cmdlen-1] != _T('\\'))||
203 (Parser.GetNextArgument())||
204 (!Tree.ChangeCurrentKey(pchCommandItself)))
205 {
206 Console.Write(_T("Unknown command \""));
207 Console.Write(pchCommandItself);
208 Console.Write(_T("\"\n"));
209 }
210 }
211 case -2: // empty line
212 goto GetCommand;
213 case 0: // exit command
214 nRetCode = 0;
215 Console.SetTextAttribute(wOldConsoleAttribute);
216 goto Exit;
217 default:
218 Console.Write(_T("\n"));
219 goto GetCommand;
220 }
221
222 Abort:
223 _ftprintf(stderr,_T("Abnormal program termination.\nPlease report bugs to ") EMAIL _T("\n"));
224 nRetCode = 1;
225 Exit:
226 return nRetCode;
227 }