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