Line data Source code
1 : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
2 : /*
3 : * This file is part of the LibreOffice project.
4 : *
5 : * This Source Code Form is subject to the terms of the Mozilla Public
6 : * License, v. 2.0. If a copy of the MPL was not distributed with this
7 : * file, You can obtain one at http://mozilla.org/MPL/2.0/.
8 : *
9 : * This file incorporates work covered by the following license notice:
10 : *
11 : * Licensed to the Apache Software Foundation (ASF) under one or more
12 : * contributor license agreements. See the NOTICE file distributed
13 : * with this work for additional information regarding copyright
14 : * ownership. The ASF licenses this file to you under the Apache
15 : * License, Version 2.0 (the "License"); you may not use this file
16 : * except in compliance with the License. You may obtain a copy of
17 : * the License at http://www.apache.org/licenses/LICENSE-2.0 .
18 : */
19 :
20 : #include <stdio.h>
21 : #include <string.h>
22 :
23 : #include "registry/registry.hxx"
24 : #include "registry/reflread.hxx"
25 : #include <rtl/ustring.hxx>
26 : #include <rtl/alloc.h>
27 : #include <osl/process.h>
28 : #include <osl/diagnose.h>
29 : #include <osl/thread.h>
30 : #include <osl/file.hxx>
31 :
32 : #ifdef SAL_UNX
33 : #define SEPARATOR '/'
34 : #else
35 : #define SEPARATOR '\\'
36 : #endif
37 :
38 : using namespace ::rtl;
39 : using namespace ::osl;
40 :
41 0 : sal_Bool isFileUrl(const OString& fileName)
42 : {
43 0 : if (fileName.indexOf("file://") == 0 )
44 0 : return sal_True;
45 0 : return sal_False;
46 : }
47 :
48 0 : OUString convertToFileUrl(const OString& fileName)
49 : {
50 0 : if ( isFileUrl(fileName) )
51 : {
52 0 : return OStringToOUString(fileName, osl_getThreadTextEncoding());
53 : }
54 :
55 0 : OUString uUrlFileName;
56 0 : OUString uFileName(fileName.getStr(), fileName.getLength(), osl_getThreadTextEncoding());
57 0 : if ( fileName.indexOf('.') == 0 || fileName.indexOf(SEPARATOR) < 0 )
58 : {
59 0 : OUString uWorkingDir;
60 0 : if (osl_getProcessWorkingDir(&uWorkingDir.pData) != osl_Process_E_None)
61 : {
62 : OSL_ASSERT(false);
63 : }
64 0 : if (FileBase::getAbsoluteFileURL(uWorkingDir, uFileName, uUrlFileName)
65 : != FileBase::E_None)
66 : {
67 : OSL_ASSERT(false);
68 0 : }
69 : } else
70 : {
71 0 : if (FileBase::getFileURLFromSystemPath(uFileName, uUrlFileName)
72 : != FileBase::E_None)
73 : {
74 : OSL_ASSERT(false);
75 : }
76 : }
77 :
78 0 : return uUrlFileName;
79 : }
80 :
81 : #define U2S( s ) \
82 : OUStringToOString(s, RTL_TEXTENCODING_UTF8).getStr()
83 : #define S2U( s ) \
84 : OStringToOUString(s, RTL_TEXTENCODING_UTF8)
85 :
86 : struct LessString
87 : {
88 : sal_Bool operator()(const OUString& str1, const OUString& str2) const
89 : {
90 : return (str1 < str2);
91 : }
92 : };
93 :
94 : enum Command {
95 : DELETEKEY
96 : };
97 :
98 : class Options
99 : {
100 : public:
101 0 : Options()
102 0 : : m_bVerbose(false)
103 0 : {}
104 0 : ~Options()
105 0 : {}
106 :
107 : bool initOptions(int ac, char* av[]);
108 :
109 : OString prepareHelp();
110 : OString prepareVersion();
111 :
112 0 : const OString& getProgramName() const
113 0 : { return m_program; }
114 0 : const OString& getTypeReg() const
115 0 : { return m_typeRegName; }
116 0 : const OString& getKeyName() const
117 0 : { return m_keyName; }
118 0 : Command getCommand() const
119 0 : { return m_command; }
120 0 : bool verbose() const
121 0 : { return m_bVerbose; }
122 : protected:
123 : OString m_program;
124 : OString m_typeRegName;
125 : OString m_keyName;
126 : Command m_command;
127 : bool m_bVerbose;
128 : };
129 :
130 0 : bool Options::initOptions(int ac, char* av[])
131 : {
132 0 : bool bRet = true;
133 0 : sal_uInt16 i=1;
134 :
135 0 : if (ac < 2)
136 : {
137 0 : fprintf(stderr, "%s", prepareHelp().getStr());
138 0 : bRet = sal_False;
139 : }
140 :
141 0 : m_program = av[0];
142 0 : sal_Int32 index = -1;
143 0 : if ((index=m_program.lastIndexOf(SEPARATOR)) > 0)
144 0 : m_program = av[0]+index+1;
145 :
146 0 : char *s=NULL;
147 0 : for (; i < ac; i++)
148 : {
149 0 : if (av[i][0] == '-')
150 : {
151 0 : switch (av[i][1])
152 : {
153 : case 'r':
154 : case 'R':
155 0 : if (av[i][2] == '\0')
156 : {
157 0 : if (i < ac - 1 && av[i+1][0] != '-')
158 : {
159 0 : i++;
160 0 : s = av[i];
161 : } else
162 : {
163 0 : fprintf(stderr, "%s: invalid option '%s'\n", m_program.getStr(), av[i]);
164 0 : bRet = sal_False;
165 0 : break;
166 : }
167 : } else
168 : {
169 0 : s = av[i] + 2;
170 : }
171 0 : m_typeRegName = OString(s);
172 0 : break;
173 : case 'd':
174 : case 'D':
175 0 : if (av[i][2] == '\0')
176 : {
177 0 : if (i < ac - 1 && av[i+1][0] != '-')
178 : {
179 0 : i++;
180 0 : s = av[i];
181 : } else
182 : {
183 0 : fprintf(stderr, "%s: invalid option '%s'\n", m_program.getStr(), av[i]);
184 0 : bRet = sal_False;
185 0 : break;
186 : }
187 : } else
188 : {
189 0 : s = av[i] + 2;
190 : }
191 0 : m_keyName = OString(s);
192 0 : break;
193 : case 'v':
194 : case 'V':
195 0 : if (av[i][2] != '\0')
196 : {
197 0 : fprintf(stderr, "%s: invalid option '%s'\n", m_program.getStr(), av[i]);
198 0 : bRet = sal_False;
199 : }
200 0 : m_bVerbose = true;
201 0 : break;
202 : case 'h':
203 : case '?':
204 0 : if (av[i][2] != '\0')
205 : {
206 0 : fprintf(stderr, "%s: invalid option '%s'\n", m_program.getStr(), av[i]);
207 0 : bRet = false;
208 : } else
209 : {
210 0 : fprintf(stdout, "%s", prepareHelp().getStr());
211 0 : exit(0);
212 : }
213 0 : break;
214 : default:
215 0 : fprintf(stderr, "%s: unknown option '%s'\n", m_program.getStr(), av[i]);
216 0 : bRet = false;
217 0 : break;
218 : }
219 : } else
220 : {
221 0 : fprintf(stderr, "%s: unknown option '%s'\n", m_program.getStr(), av[i]);
222 0 : bRet = false;
223 : }
224 : }
225 :
226 0 : return bRet;
227 : }
228 :
229 0 : OString Options::prepareHelp()
230 : {
231 0 : OString help("\nusing: ");
232 0 : help += m_program + " -r<filename> <command>\n";
233 0 : help += " -r<filename> = filename specifies the name of the type registry.\n";
234 0 : help += "Commands:\n";
235 0 : help += " -d <keyname> = delete the specified key from the registry. Keyname\n";
236 0 : help += " specifies the name of the key that get deleted.\n";
237 0 : help += " -v = verbose output.\n";
238 0 : help += " -h|-? = print this help message and exit.\n";
239 0 : help += prepareVersion();
240 :
241 0 : return help;
242 : }
243 :
244 0 : OString Options::prepareVersion()
245 : {
246 0 : OString version(m_program);
247 0 : version += " Version 1.0\n\n";
248 0 : return version;
249 : }
250 :
251 0 : static Options options;
252 :
253 :
254 : #if (defined UNX) || (defined __MINGW32__)
255 0 : int main( int argc, char * argv[] )
256 : #else
257 : int _cdecl main( int argc, char * argv[] )
258 : #endif
259 : {
260 0 : if ( !options.initOptions(argc, argv) )
261 : {
262 0 : exit(1);
263 : }
264 :
265 0 : OUString typeRegName( convertToFileUrl(options.getTypeReg()) );
266 :
267 0 : Registry typeReg;
268 :
269 0 : if ( typeReg.open(typeRegName, REG_READWRITE) )
270 : {
271 : fprintf(stderr, "%s: open registry \"%s\" failed\n",
272 0 : options.getProgramName().getStr(), options.getTypeReg().getStr());
273 0 : exit(2);
274 : }
275 :
276 0 : RegistryKey typeRoot;
277 0 : if ( typeReg.openRootKey(typeRoot) )
278 : {
279 : fprintf(stderr, "%s: open root key of registry \"%s\" failed\n",
280 0 : options.getProgramName().getStr(), options.getTypeReg().getStr());
281 0 : exit(3);
282 : }
283 :
284 0 : if ( options.getCommand() == DELETEKEY )
285 : {
286 0 : if ( typeRoot.deleteKey(S2U(options.getKeyName())) )
287 : {
288 : fprintf(stderr, "%s: delete key \"%s\" of registry \"%s\" failed\n",
289 0 : options.getProgramName().getStr(), options.getKeyName().getStr(), options.getTypeReg().getStr());
290 0 : exit(4);
291 : } else {
292 0 : if (options.verbose())
293 : fprintf(stderr, "%s: delete key \"%s\" of registry \"%s\"\n",
294 0 : options.getProgramName().getStr(), options.getKeyName().getStr(), options.getTypeReg().getStr());
295 : }
296 : }
297 :
298 0 : typeRoot.releaseKey();
299 0 : if ( typeReg.close() )
300 : {
301 : fprintf(stderr, "%s: closing registry \"%s\" failed\n",
302 0 : options.getProgramName().getStr(), options.getTypeReg().getStr());
303 0 : exit(5);
304 0 : }
305 0 : }
306 :
307 : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|
