d78a225ab1c742f87258bbb767c7d352a278f39f
[reactos.git] / reactos / base / shell / cmd / assoc.c
1 /*
2 * Assoc.C - assoc internal command.
3 *
4 *
5 * History:
6 *
7 * 14-Mar-2009 Lee C. Baker
8 * - initial implementation
9 *
10 * 15-Mar-2009 Lee C. Baker
11 * - Don't write to (or use) HKEY_CLASSES_ROOT directly
12 * - Externalize strings
13 *
14 * TODO:
15 * - PrintAllAssociations might could be optimized to not fetch all registry subkeys under 'Classes', just the ones that start with '.'
16 * - Make sure that non-administrator users can list associations, and get appropriate error messages when they don't have sufficient
17 * priveleges to perform an operation
18 */
19
20 #include "precomp.h"
21
22 #ifdef INCLUDE_CMD_ASSOC
23
24 static INT
25 PrintAssociation(LPTSTR extension)
26 {
27 DWORD return_val;
28 HKEY hKey = NULL, hInsideKey = NULL;
29
30 DWORD fileTypeLength = 0;
31 LPTSTR fileType = NULL;
32
33 return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
34
35 if (return_val != ERROR_SUCCESS)
36 {
37 RegCloseKey(hKey);
38 return -1;
39 }
40
41 return_val = RegOpenKeyEx(hKey, extension, 0, KEY_READ, &hInsideKey);
42
43 if (return_val != ERROR_SUCCESS)
44 {
45 RegCloseKey(hKey);
46 RegCloseKey(hInsideKey);
47 return 0;
48 }
49
50 /* obtain string length */
51 return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, NULL, &fileTypeLength);
52
53 if (return_val == ERROR_FILE_NOT_FOUND) /* no default value, don't display */
54 {
55 RegCloseKey(hInsideKey);
56 RegCloseKey(hKey);
57 return 0;
58 }
59
60 if (return_val != ERROR_SUCCESS)
61 {
62 RegCloseKey(hInsideKey);
63 RegCloseKey(hKey);
64 return -2;
65 }
66
67 fileType = cmd_alloc(fileTypeLength * sizeof(TCHAR));
68
69 /* obtain actual file type */
70 return_val = RegQueryValueEx(hInsideKey, NULL, NULL, NULL, (LPBYTE) fileType, &fileTypeLength);
71
72 RegCloseKey(hInsideKey);
73 RegCloseKey(hKey);
74
75 if (return_val != ERROR_SUCCESS)
76 {
77 cmd_free(fileType);
78 return -2;
79 }
80
81 if (fileTypeLength != 0) /* if there is a default key, display relevant information */
82 {
83 ConOutPrintf(_T("%s=%s\n"), extension, fileType);
84 }
85
86 if (fileTypeLength)
87 cmd_free(fileType);
88
89 return 1;
90 }
91
92 static INT
93 PrintAllAssociations()
94 {
95 DWORD return_val = 0;
96 HKEY hKey = NULL;
97 DWORD numKeys = 0;
98
99 DWORD extLength = 0;
100 LPTSTR extName = NULL;
101 DWORD keyCtr = 0;
102
103 return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_READ, &hKey);
104
105 if (return_val != ERROR_SUCCESS)
106 {
107 RegCloseKey(hKey);
108 return -1;
109 }
110
111 return_val = RegQueryInfoKey(hKey, NULL, NULL, NULL, &numKeys, &extLength, NULL, NULL, NULL, NULL, NULL, NULL);
112
113 if (return_val != ERROR_SUCCESS)
114 {
115 RegCloseKey(hKey);
116 return -2;
117 }
118
119 extName = cmd_alloc(extLength * sizeof(TCHAR));
120
121 for(keyCtr = 0; keyCtr < numKeys; keyCtr++)
122 {
123 DWORD buffer_size = extLength;
124 return_val = RegEnumKeyEx(hKey, keyCtr, extName, &buffer_size, NULL, NULL, NULL, NULL);
125
126 if (return_val == ERROR_SUCCESS || return_val == ERROR_MORE_DATA)
127 {
128 if (*extName == _T('.'))
129 PrintAssociation(extName);
130 }
131 else
132 {
133 cmd_free(extName);
134 RegCloseKey(hKey);
135 return -1;
136 }
137 }
138
139 RegCloseKey(hKey);
140
141 if (extName)
142 cmd_free(extName);
143
144 return numKeys;
145 }
146
147 static INT
148 AddAssociation(LPTSTR extension, LPTSTR type)
149 {
150 DWORD return_val;
151 HKEY hKey = NULL, insideKey = NULL;
152
153 return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
154
155 if (return_val != ERROR_SUCCESS)
156 return -1;
157
158 return_val = RegCreateKeyEx(hKey, extension, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &insideKey, NULL);
159
160 if (return_val != ERROR_SUCCESS)
161 {
162 RegCloseKey(hKey);
163 return -1;
164 }
165
166 return_val = RegSetValueEx(insideKey, NULL, 0, REG_SZ, (LPBYTE)type, (_tcslen(type) + 1) * sizeof(TCHAR));
167
168 if (return_val != ERROR_SUCCESS)
169 {
170 RegCloseKey(insideKey);
171 RegCloseKey(hKey);
172 return -2;
173 }
174
175 RegCloseKey(insideKey);
176 RegCloseKey(hKey);
177 return 0;
178 }
179
180
181 static int
182 RemoveAssociation(LPTSTR extension)
183 {
184 DWORD return_val;
185 HKEY hKey;
186
187 return_val = RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\Classes"), 0, KEY_ALL_ACCESS, &hKey);
188
189 if (return_val != ERROR_SUCCESS)
190 return -1;
191
192 return_val = RegDeleteKey(hKey, extension);
193
194 if (return_val != ERROR_SUCCESS)
195 {
196 RegCloseKey(hKey);
197 return -2;
198 }
199
200 RegCloseKey(hKey);
201 return 0;
202 }
203
204
205
206 INT CommandAssoc (LPTSTR param)
207 {
208 /* print help */
209 if (!_tcsncmp (param, _T("/?"), 2))
210 {
211 ConOutResPaging(TRUE,STRING_ASSOC_HELP);
212 return 0;
213 }
214
215 nErrorLevel = 0;
216
217 if (_tcslen(param) == 0)
218 PrintAllAssociations();
219 else
220 {
221 LPTSTR lpEqualSign = _tcschr(param, _T('='));
222 if (lpEqualSign != NULL)
223 {
224 LPTSTR fileType = lpEqualSign + 1;
225 LPTSTR extension = cmd_alloc((lpEqualSign - param + 1) * sizeof(TCHAR));
226
227 _tcsncpy(extension, param, lpEqualSign - param);
228 extension[lpEqualSign - param] = (TCHAR)0;
229
230 if (_tcslen(fileType) == 0)
231 /* if the equal sign is the last character
232 in the string, then delete the key */
233 {
234 RemoveAssociation(extension);
235 }
236 else
237 /* otherwise, add the key and print out the association*/
238 {
239 AddAssociation( extension, fileType);
240 PrintAssociation(extension);
241 }
242
243 cmd_free(extension);
244 }
245 else
246 {
247 /* no equal sign, print all associations */
248 INT retval = PrintAssociation(param);
249
250 if (retval == 0) /* if nothing printed out */
251 ConOutResPrintf(STRING_ASSOC_ERROR, param);
252 }
253 }
254
255 return 0;
256 }
257
258 #endif /* INCLUDE_CMD_ASSOC */