www.pudn.com > ShellExtGuide4_demo.zip > HardLink.cpp

// HardLink.cpp : Implementation of DLL Exports. 
 
 
// Note: Proxy/Stub Information 
//      To build a separate proxy/stub DLL,  
//      run nmake -f HardLinkps.mk in the project directory. 
 
#include "stdafx.h" 
#include "resource.h" 
#include  
#include "HardLink.h" 
 
#include "HardLink_i.c" 
#include "HardLinkShlExt.h" 
 
 
CComModule _Module; 
 
BEGIN_OBJECT_MAP(ObjectMap) 
OBJECT_ENTRY(CLSID_HardLinkShlExt, CHardLinkShlExt) 
END_OBJECT_MAP() 
 
class CHardLinkApp : public CWinApp 
{ 
public: 
 
// Overrides 
	// ClassWizard generated virtual function overrides 
	//{{AFX_VIRTUAL(CHardLinkApp) 
	public: 
    virtual BOOL InitInstance(); 
    virtual int ExitInstance(); 
	//}}AFX_VIRTUAL 
 
	//{{AFX_MSG(CHardLinkApp) 
		// NOTE - the ClassWizard will add and remove member functions here. 
		//    DO NOT EDIT what you see in these blocks of generated code ! 
	//}}AFX_MSG 
	DECLARE_MESSAGE_MAP() 
}; 
 
BEGIN_MESSAGE_MAP(CHardLinkApp, CWinApp) 
	//{{AFX_MSG_MAP(CHardLinkApp) 
		// NOTE - the ClassWizard will add and remove mapping macros here. 
		//    DO NOT EDIT what you see in these blocks of generated code! 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
CHardLinkApp theApp; 
 
BOOL CHardLinkApp::InitInstance() 
{ 
    _Module.Init(ObjectMap, m_hInstance, &LIBID_HARDLINKLib); 
    return CWinApp::InitInstance(); 
} 
 
int CHardLinkApp::ExitInstance() 
{ 
    _Module.Term(); 
    return CWinApp::ExitInstance(); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// Used to determine whether the DLL can be unloaded by OLE 
 
STDAPI DllCanUnloadNow(void) 
{ 
    AFX_MANAGE_STATE(AfxGetStaticModuleState()); 
    return (AfxDllCanUnloadNow()==S_OK && _Module.GetLockCount()==0) ? S_OK : S_FALSE; 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// Returns a class factory to create an object of the requested type 
 
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv) 
{ 
    return _Module.GetClassObject(rclsid, riid, ppv); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// DllRegisterServer - Adds entries to the system registry 
 
STDAPI DllRegisterServer(void) 
{ 
    // On NT, add ourself to the list of approved shell extensions. 
    if ( 0 == (GetVersion() & 0x80000000) ) 
        { 
        CRegKey reg; 
        LONG    lRet; 
 
        lRet = reg.Open ( HKEY_LOCAL_MACHINE,  
                          _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved"), 
                          KEY_SET_VALUE ); 
 
        if ( ERROR_SUCCESS != lRet ) 
            return E_ACCESSDENIED; 
 
        lRet = reg.SetValue ( _T("Hard Links Shell Extension"), 
                              _T("{3C06DFAE-062F-11D4-8D3B-919D46C1CE19}") ); 
 
        if ( ERROR_SUCCESS != lRet ) 
            return E_ACCESSDENIED; 
        } 
 
    // registers object, typelib and all interfaces in typelib 
    return _Module.RegisterServer(TRUE); 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// DllUnregisterServer - Removes entries from the system registry 
 
STDAPI DllUnregisterServer(void) 
{ 
    // On NT, remove ourself from the list of approved shell extensions. 
    if ( 0 == (GetVersion() & 0x80000000) ) 
        { 
        CRegKey reg; 
        LONG    lRet; 
 
        lRet = reg.Open ( HKEY_LOCAL_MACHINE,  
                          _T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Shell Extensions\\Approved"), 
                          KEY_SET_VALUE ); 
 
        if ( ERROR_SUCCESS == lRet ) 
            { 
            reg.DeleteValue ( _T("{3C06DFAE-062F-11D4-8D3B-919D46C1CE19}") ); 
            reg.Close(); 
            } 
        } 
 
    return _Module.UnregisterServer(TRUE); 
}