1 /**************************************************************
2 *
3 * Licensed to the Apache Software Foundation (ASF) under one
4 * or more contributor license agreements. See the NOTICE file
5 * distributed with this work for additional information
6 * regarding copyright ownership. The ASF licenses this file
7 * to you under the Apache License, Version 2.0 (the
8 * "License"); you may not use this file except in compliance
9 * with the License. You may obtain a copy of the License at
10 *
11 * http://www.apache.org/licenses/LICENSE-2.0
12 *
13 * Unless required by applicable law or agreed to in writing,
14 * software distributed under the License is distributed on an
15 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16 * KIND, either express or implied. See the License for the
17 * specific language governing permissions and limitations
18 * under the License.
19 *
20 *************************************************************/
21
22
23
24 // MARKER(update_precomp.py): autogen include statement, do not remove
25 #include "precompiled_shell.hxx"
26
27 #if defined _MSC_VER
28 #pragma warning(push, 1)
29 #endif
30 #include <windows.h>
31 #if defined _MSC_VER
32 #pragma warning(pop)
33 #endif
34
35 typedef HRESULT (__stdcall *lpfnDllRegisterServer)();
36 typedef HRESULT (__stdcall *lpfnDllUnregisterServer)();
37
38 /**
39 */
IsUnregisterParameter(const char * Param)40 bool IsUnregisterParameter(const char* Param)
41 {
42 return ((0 == _stricmp(Param, "/u")) ||
43 (0 == _stricmp(Param, "-u")));
44 }
45
46 /**
47 */
main(int argc,char * argv[])48 int main(int argc, char* argv[])
49 {
50 HMODULE hmod;
51 HRESULT hr = E_FAIL;
52 lpfnDllRegisterServer lpfn_register;
53 lpfnDllUnregisterServer lpfn_unregister;
54
55 if (2 == argc)
56 {
57 hmod = LoadLibraryA(argv[1]);
58
59 if (hmod)
60 {
61 lpfn_register = (lpfnDllRegisterServer)GetProcAddress(
62 hmod, "DllRegisterServer");
63
64 if (lpfn_register)
65 hr = lpfn_register();
66
67 FreeLibrary(hmod);
68 }
69 }
70 else if (3 == argc && IsUnregisterParameter(argv[1]))
71 {
72 hmod = LoadLibraryA(argv[2]);
73
74 if (hmod)
75 {
76 lpfn_unregister = (lpfnDllUnregisterServer)GetProcAddress(
77 hmod, "DllUnregisterServer");
78
79 if (lpfn_unregister)
80 hr = lpfn_unregister();
81
82 FreeLibrary(hmod);
83 }
84 }
85
86 return 0;
87 }
88