ShellExtGuide2_demo.zip_DEMO_shell编程

// DLLRegShlExt.cpp : Implementation of CDLLRegShlExt #include "stdafx.h" #include "DLLReg.h" #include "DLLRegShlExt.h" #include "ProgressDlg.h" ///////////////////////////////////////////////////////////////////////////// // CDLLRegShlExt construction/destruction CDLLRegShlExt::CDLLRegShlExt() { m_hRegBmp = LoadBitmap ( _Module.GetModuleInstance(), MAKEINTRESOURCE(IDB_REGISTERBMP) ); m_hUnregBmp = LoadBitmap ( _Module.GetModuleInstance(), MAKEINTRESOURCE(IDB_UNREGISTERBMP) ); } ///////////////////////////////////////////////////////////////////////////// // CDLLRegShlExt IShellExtInit methods ////////////////////////////////////////////////////////////////////////// // // Function: Initialize() // // Description: // Reads in the list of selected folders and stores them for later use. // ////////////////////////////////////////////////////////////////////////// HRESULT CDLLRegShlExt::Initialize ( LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDO, HKEY hProgID ) { TCHAR szFile [MAX_PATH]; TCHAR szFolder [MAX_PATH]; TCHAR szCurrDir [MAX_PATH]; TCHAR* pszLastBackslash; UINT uNumFiles; HDROP hdrop; FORMATETC etc = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL }; STGMEDIUM stg = { TYMED_HGLOBAL }; HINSTANCE hinst; bool bChangedDir = false; HRESULT (STDAPICALLTYPE* pfn)(); // Read the list of folders from the data object. They're stored in HDROP // form, so just get the HDROP handle and then use the drag 'n' drop APIs // on it. if ( FAILED( pDO->GetData ( &etc, &stg ))) return E_INVALIDARG; // Get an HDROP handle. hdrop = (HDROP) GlobalLock ( stg.hGlobal ); if ( NULL == hdrop ) { ReleaseStgMedium ( &stg ); return E_INVALIDARG; } // Determine how many files are involved in this operation. uNumFiles = DragQueryFile ( hdrop, 0xFFFFFFFF, NULL, 0 ); for ( UINT uFile = 0; uFile < uNumFiles; uFile++ ) { // Get the next filename. if ( 0 == DragQueryFile ( hdrop, uFile, szFile, MAX_PATH )) continue; // If this is the first time thru the for loop, we need to // change the current directory to the one that was open in // explorer. This is done so the DLLs we're about to load // can load other DLLs they depend on to be in the same // directory. if ( !bChangedDir ) { bChangedDir = true; // Take the current filename and remove the last component. // I could do this using PathRemoveFileSpec() but that's only // available if you have IE 3+ installed. lstrcpy ( szFolder, szFile ); // Find the last backslash and replace it with a zero character. pszLastBackslash = _tcsrchr ( szFolder, '\\' ); ATLASSERT ( NULL != pszLastBackslash ); *pszLastBackslash = '\0'; // But, if szFolder is now just a drive, like "C:", we need // to put the backslash back to make it a valid directory // name. if ( lstrlen ( szFolder ) == 2 ) lstrcat ( szFolder, _T("\\") ); // Save the current directory, and then change it. GetCurrentDirectory ( MAX_PATH, szCurrDir ); SetCurrentDirectory ( szFolder ); } // end if // Try & load the DLL. hinst = LoadLibrary ( szFile ); if ( NULL == hinst ) continue; // Get the address of DllRegisterServer(); (FARPROC&) pfn = GetProcAddress ( hinst, "DllRegisterServer" ); // If it wasn't found, skip the file. if ( NULL == pfn ) { FreeLibrary ( hinst ); continue; } // Get the address of DllUnregisterServer(); (FARPROC&) pfn = GetProcAddress ( hinst, "DllUnregisterServer" ); // If it was found, we can operate on the file, so add it to // our list o' files (m_lsFiles). if ( NULL != pfn ) { m_lsFiles.push_back ( szFile ); } FreeLibrary ( hinst ); } // end for // Change back to the original directory if we changed it. if ( bChangedDir ) SetCurrentDirectory ( szCurrDir ); // Release resources. GlobalUnlock ( stg.hGlobal ); ReleaseStgMedium ( &stg ); // If we found any files we can work with, return S_OK. Otherwise, // return E_INVALIDARG so we don't get called again for this right-click // operation. return ( m_lsFiles.size() > 0 ) ? S_OK : E_INVALIDARG; } ///////////////////////////////////////////////////////////////////////////// // CDLLRegShlExt IContextMenu methods ////////////////////////////////////////////////////////////////////////// // // Function: QueryContextMenu() // // Description: // Adds our items to the supplied menu. // ////////////////////////////////////////////////////////////////////////// HRESULT CDLLRegShlExt::QueryContextMenu ( HMENU hmenu, UINT uMenuIndex, UINT uidFirstCmd, UINT uidLastCmd, UINT uFlags ) { UINT uCmdID = uidFirstCmd; // If the flags include CMF_DEFAULTONLY then we shouldn't do anything. if ( uFlags & CMF_DEFAULTONLY ) { return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, 0 ); } // Add our register/unregister items. InsertMenu ( hmenu, uMenuIndex, MF_STRING | MF_BYPOSITION, uCmdID++, _T("Register server(s)") ); // Set the bitmap for the register item. if ( NULL != m_hRegBmp ) { SetMenuItemBitmaps ( hmenu, uMenuIndex, MF_BYPOSITION, m_hRegBmp, NULL ); } uMenuIndex++; InsertMenu ( hmenu, uMenuIndex, MF_STRING | MF_BYPOSITION, uCmdID++, _T("Unregister server(s)") ); // Set the bitmap for the unregister item. if ( NULL != m_hUnregBmp ) { SetMenuItemBitmaps ( hmenu, uMenuIndex, MF_BYPOSITION, m_hUnregBmp, NULL ); } uMenuIndex++; // The return value tells the shell how many top-level items we added. return MAKE_HRESULT ( SEVERITY_SUCCESS, FACILITY_NULL, 2 ); } ////////////////////////////////////////////////////////////////////////// // // Function: GetCommandString() // // Description: // Sets the flyby help string for the Explorer status bar. // ////////////////////////////////////////////////////////////////////////// HRESULT CDLLRegShlExt::GetCommandString ( UINT uCmdID, UINT uFlags, UINT* puReserved, LPSTR szName, UINT cchMax ) { LPCTSTR szPrompt; USES_CONVERSION; if ( uFlags & GCS_HELPTEXT ) { switch ( uCmdID ) { case 0: szPrompt = _T("Register all selected COM servers"); break; case 1: szPrompt = _T("Unregister all selected COM servers"); break; default: ATLASSERT(0); // should never get here return E_INVALIDARG; break; } // Copy the help text into the supplied buffer. If the shell wants // a Unicode string, we need to case szName to an LPCWSTR. if ( uFlags & GCS_UNICODE ) { lstrcpynW ( (LPWSTR) szName, T2CW(szPrompt), cchMax ); } else