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 */ 40 bool IsUnregisterParameter(const char* Param) 41 { 42 return ((0 == _stricmp(Param, "/u")) || 43 (0 == _stricmp(Param, "-u"))); 44 } 45 46 /** 47 */ 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