// MBShellExt.cpp : Implementation of CMBShellExt
|
|
#include "stdafx.h"
|
#include "MBShellExt.h"
|
|
#pragma comment(lib,"shlwapi.lib")
|
#pragma message("Automatically linking with Shlwapi.dll")
|
|
// CMBShellExt
|
|
#define REG_SUBKEY _T( "Software\\Activesoft\\MBClient" )
|
|
#define PSW_EMKEY ( _T("P512D790") ) // ÊÇAPM\EMÉϵÄÃÜÔ¿
|
|
// CMBShellExt
|
#define AMCADDIN_FLAG_SHELLEX_MENU ( 16384 ) // ÔÊÐí¼ÓÈë×ÊÔ´¹ÜÀíÆ÷ÓÒ¼ü²Ëµ¥ÖÐ
|
|
HRESULT CMBShellExt::Initialize( LPCITEMIDLIST pidlFolder, LPDATAOBJECT pDataObj, HKEY hProgID )
|
{
|
FORMATETC fmt = { CF_HDROP, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL };
|
STGMEDIUM stg = { TYMED_HGLOBAL };
|
HDROP hDrop;
|
|
// Look for CF_HDROP data in the data object.
|
if ( FAILED( pDataObj->GetData( &fmt, &stg ) ) )
|
{
|
// Nope! Return an "invalid argument" error back to Explorer.
|
return E_INVALIDARG;
|
}
|
|
// Get a pointer to the actual data.
|
hDrop = (HDROP)GlobalLock( stg.hGlobal );
|
if ( NULL == hDrop )
|
return E_INVALIDARG;
|
|
// Note that it's vitally important to error-check everything, especially pointers. Since our extension runs in Explorer's process space, if we crash we take down Explorer too. On 9x, such a crash might necessitate rebooting the computer.
|
|
// So, now that we have an HDROP handle, we can get the filename we need.
|
|
// Sanity check ¨C make sure there is at least one filename.
|
UINT uNumFiles = DragQueryFile( hDrop, 0xFFFFFFFF, NULL, 0 );
|
if ( 0 == uNumFiles )
|
{
|
GlobalUnlock( stg.hGlobal );
|
ReleaseStgMedium( &stg );
|
return E_INVALIDARG;
|
}
|
|
HRESULT hr = S_OK;
|
TCHAR szFile[MAX_PATH];
|
UINT nLen;
|
|
// get multi files
|
for ( UINT uFile = 0; uFile < uNumFiles; uFile++ )
|
{
|
// Get the next filename.
|
nLen = DragQueryFile( hDrop, uFile, szFile, MAX_PATH );
|
if ( nLen == 0 )
|
continue;
|
|
if ( PathIsDirectory ( szFile ) )
|
{
|
szFile[nLen] = '\\';
|
szFile[nLen+1] = '\0';
|
}
|
|
m_lsFiles.AddTail( szFile );
|
} // end for
|
|
GlobalUnlock( stg.hGlobal );
|
ReleaseStgMedium( &stg );
|
return hr;
|
}
|
|
HRESULT CMBShellExt::QueryContextMenu( HMENU hmenu, UINT uMenuIndex, UINT uidFirstCmd, UINT uidLastCmd, UINT uFlags )
|
{
|
// If the flags include CMF_DEFAULTONLY then we shouldn't do anything.
|
if ( uFlags & CMF_DEFAULTONLY )
|
return MAKE_HRESULT( SEVERITY_SUCCESS, FACILITY_NULL, 0 );
|
|
int nNum = 0;
|
POSITION pos;
|
CString strDispName;
|
CString strMenuName;
|
|
|
strMenuName = _T("MBCÉÏ´«");//µ½Îĵµ¹ñ
|
InsertMenu( hmenu, uMenuIndex, MF_BYPOSITION, uidFirstCmd + nNum, strMenuName );
|
++nNum;
|
HBITMAP hBitmap= GetSTKBitmap(1);
|
if(hBitmap)
|
{
|
SetMenuItemBitmaps(hmenu,
|
uMenuIndex,
|
MF_BYPOSITION,
|
hBitmap,
|
hBitmap);
|
}
|
/*strMenuName = _T("MBCÉÏ´«µ½¸öÈËÍøÅÌ");
|
InsertMenu( hmenu, uMenuIndex, MF_BYPOSITION, uidFirstCmd + nNum, strMenuName );
|
++nNum;
|
hBitmap= GetSTKBitmap(1);
|
if(hBitmap)
|
{
|
SetMenuItemBitmaps(hmenu,
|
uMenuIndex,
|
MF_BYPOSITION,
|
hBitmap,
|
hBitmap);
|
}*/
|
for ( pos = m_lsAddInDispName.GetHeadPosition(); pos; )
|
{
|
strDispName = m_lsAddInDispName.GetNext( pos );
|
InsertMenu( hmenu, uMenuIndex, MF_BYPOSITION, uidFirstCmd + nNum, strDispName );
|
++nNum;
|
}
|
|
strMenuName = _T("MBCÒ×·¢");
|
InsertMenu( hmenu, uMenuIndex, MF_BYPOSITION, uidFirstCmd + nNum, strMenuName );
|
hBitmap= GetSTKBitmap();
|
if(hBitmap)
|
{
|
SetMenuItemBitmaps(hmenu,
|
uMenuIndex,
|
MF_BYPOSITION,
|
hBitmap,
|
hBitmap);
|
}
|
return MAKE_HRESULT( SEVERITY_SUCCESS, FACILITY_NULL, nNum + 1 );
|
}
|
|
#ifdef _WIN64
|
HRESULT CMBShellExt::GetCommandString(UINT_PTR idCmd, UINT uFlags, UINT* pwReserved, LPSTR pszName, UINT cchMax)
|
{
|
USES_CONVERSION;
|
|
// Check idCmd, it must be 0 since we have only one menu item.
|
if ( idCmd < 0 && idCmd > (UINT)m_lsAddInDispName.GetCount() )
|
return E_INVALIDARG;
|
|
// If Explorer is asking for a help string, copy our string into the
|
// supplied buffer.
|
if ( uFlags & GCS_HELPTEXT )
|
{
|
CString strText;
|
LPCTSTR szText = _T("");
|
if(idCmd == 0)
|
{
|
szText = _T("MBCÒ×·¢");
|
}
|
else
|
{
|
strText.Format( _T("This is the MBC shell extension's help %d"), idCmd);
|
szText = strText;
|
}
|
if ( uFlags & GCS_UNICODE )
|
{
|
// We need to cast pszName to a Unicode string, and then use the
|
// Unicode string copy API.
|
lstrcpynW( (LPWSTR) pszName, T2CW(szText), cchMax );
|
}
|
else
|
{
|
// Use the ANSI string copy API to return the help string.
|
lstrcpynA( pszName, T2CA(szText), cchMax );
|
}
|
|
return S_OK;
|
}
|
|
return E_INVALIDARG;
|
}
|
|
#else
|
HRESULT CMBShellExt::GetCommandString( UINT idCmd, UINT uFlags, UINT* pwReserved, LPSTR pszName, UINT cchMax )
|
{
|
USES_CONVERSION;
|
|
// Check idCmd, it must be 0 since we have only one menu item.
|
if ( idCmd < 0 && idCmd > (UINT)m_lsAddInDispName.GetCount() )
|
return E_INVALIDARG;
|
|
// If Explorer is asking for a help string, copy our string into the
|
// supplied buffer.
|
if ( uFlags & GCS_HELPTEXT )
|
{
|
LPCTSTR szText = _T("This is the MBC shell extension's help");
|
|
if ( uFlags & GCS_UNICODE )
|
{
|
// We need to cast pszName to a Unicode string, and then use the
|
// Unicode string copy API.
|
lstrcpynW( (LPWSTR) pszName, T2CW(szText), cchMax );
|
}
|
else
|
{
|
// Use the ANSI string copy API to return the help string.
|
lstrcpynA( pszName, T2CA(szText), cchMax );
|
}
|
|
return S_OK;
|
}
|
|
return E_INVALIDARG;
|
}
|
#endif
|
|
HRESULT CMBShellExt::InvokeCommand( LPCMINVOKECOMMANDINFO pCmdInfo )
|
{
|
// If lpVerb really points to a string, ignore this function call and bail out.
|
int nMenuID = 0;
|
|
// Get the command index - the only valid one is 0.
|
nMenuID = LOWORD( pCmdInfo->lpVerb );
|
|
|
if ( 0 != HIWORD( pCmdInfo->lpVerb ) )
|
{
|
return E_INVALIDARG;
|
}
|
|
if ( nMenuID >= 0 && nMenuID <= m_lsAddInDispName.GetCount( ) + 2 )
|
{
|
#ifdef _WIN64
|
CString strAMViewPath;
|
|
if ( DoSTKCom( nMenuID, pCmdInfo->hwnd ) )
|
return S_OK;
|
else
|
return S_FALSE;
|
#else
|
if ( DoSTKCom( nMenuID, pCmdInfo->hwnd ) )
|
return S_OK;
|
else
|
return S_FALSE;
|
#endif
|
|
return S_OK;
|
}
|
|
return E_INVALIDARG;
|
}
|
|
bool CMBShellExt::DoSTKCom( long nMenuID, HWND hWnd )
|
{
|
HRESULT hr;
|
CString strMBExe;
|
strMBExe.Format( _T("%s\\mbcU.exe"), GetSTKWorkDir());
|
HANDLE hEventShareMemory = OpenEvent(EVENT_ALL_ACCESS , FALSE, _T("mbcSharedMemoryEvent"));
|
if (NULL == hEventShareMemory)
|
{
|
long nErrorId = GetLastError();
|
CString strMessage;
|
strMessage.Format(_T("OpenEvent mbcSharedMemoryEvent ErrorID:%d"),nErrorId);
|
WriteLog(strMessage);
|
ShellExecute( NULL, _T("open"), strMBExe, _T(""), _T(""), SW_SHOW );
|
//MessageBox(hWnd, strMessage, _T("mbEasySend"), MB_OK | MB_ICONINFORMATION );
|
return false;
|
}
|
HANDLE hShareMemory = OpenFileMapping( PAGE_READONLY,false,_T("mbcSharedMemory") );
|
if (NULL == hShareMemory)
|
{
|
/*m_hShareMemory = CreateFileMapping(HANDLE(0xFFFFFFFF), NULL, PAGE_READWRITE,0, 2560, _T("mbcSharedMemory"));
|
if (NULL == m_hShareMemory)
|
{*/
|
long nErrorId = GetLastError();
|
CString strMessage;
|
strMessage.Format(_T("OpenFileMapping mbcSharedMemory ErrorID:%d"),nErrorId);
|
//MessageBox(hWnd, strMessage, _T("mbEasySend"), MB_OK | MB_ICONINFORMATION );
|
ShellExecute( NULL, _T("open"), strMBExe, _T(""), _T(""), SW_SHOW );
|
WriteLog(strMessage);
|
return false;
|
//}
|
}
|
TCHAR* lpBuffer = (TCHAR*)MapViewOfFile(hShareMemory, FILE_MAP_WRITE, 0, 0, 2560);
|
if (NULL == lpBuffer)
|
{
|
//MessageBox(hWnd, _T("MapViewOfFile mbcSharedMemoryʧ°Ü"), _T("mbEasySend"), MB_OK | MB_ICONINFORMATION );
|
long nErrorId = GetLastError();
|
CString strMessage;
|
strMessage.Format(_T("MapViewOfFile mbcSharedMemory ErrorID:%d"),nErrorId);
|
WriteLog(strMessage);
|
return false;
|
}
|
POSITION pos;
|
CString strBuf;
|
CString strXmlBuf;
|
CString strAddInName = _T("");
|
int nType = 0;
|
if(nMenuID == 0 )//|| nMenuID == 1
|
nType = 1;
|
if ( nMenuID -1 > m_lsAddInName.GetCount() )
|
{
|
pos = m_lsAddInName.FindIndex( nMenuID - 1 );
|
if ( pos )
|
{
|
strAddInName = m_lsAddInName.GetAt( pos );
|
}
|
}
|
strXmlBuf.Format(_T("<body type='%d' addin='%s'>"),nType,strAddInName);
|
CString strFile;
|
for ( pos = m_lsFiles.GetHeadPosition(); pos; )
|
{
|
strFile = m_lsFiles.GetNext( pos );
|
strBuf.Format( _T("<file>%s</file>"),strFile );
|
if(strXmlBuf.GetLength() + strBuf.GetLength()> 2550)
|
{
|
break;
|
}
|
strXmlBuf += strBuf;
|
}
|
strXmlBuf += _T("</body>");
|
_stprintf(lpBuffer,_T("%s"), strXmlBuf );
|
UnmapViewOfFile(lpBuffer);
|
SetEvent(hEventShareMemory);
|
CloseHandle(hEventShareMemory);
|
CloseHandle(hShareMemory);
|
return true;
|
}
|
|
void CMBShellExt::GetAddInMenu( )
|
{
|
HKEY hKey;
|
DWORD dwBuf = MAX_PATH;
|
DWORD dwIndex = 0;
|
CRegKey myKey;
|
CString strSubPath, strDispName;
|
CString strName, strValue;
|
TCHAR szValue[256]= { 0 };
|
TCHAR szName[80] = { 0 };
|
DWORD dwReaded = 0;
|
DWORD dwValue = 0;
|
int nFlag = 0;
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\\ActiveSoft\\MBClient\\AMCAddIn"), 0, KEY_ENUMERATE_SUB_KEYS, &hKey);
|
|
m_lsAddInName.RemoveAll();
|
m_lsAddInDispName.RemoveAll();
|
while (RegEnumKey(hKey, dwIndex, szName, dwBuf) == 0)
|
{
|
dwIndex++;
|
|
szName[dwBuf] = '\0';
|
strName = szName;
|
strDispName = _T("");
|
strSubPath.Format( _T("SOFTWARE\\ActiveSoft\\MBClient\\AMCAddIn\\%s"), strName );
|
myKey.Close( );
|
if ( myKey.Open( HKEY_LOCAL_MACHINE, strSubPath, KEY_READ ) == ERROR_SUCCESS )
|
{
|
dwReaded = 255;
|
nFlag = 0;
|
|
if ( ERROR_SUCCESS == myKey.QueryValue( dwValue, _T( "Flag" ) ) )
|
nFlag = dwValue;
|
|
if ( nFlag & AMCADDIN_FLAG_SHELLEX_MENU )
|
{
|
#ifdef AFX_TARG_CHS
|
if ( ERROR_SUCCESS == myKey.QueryValue( szValue, _T( "DisplayName" ), &dwReaded ) && dwReaded > 0 )
|
{
|
szValue[dwReaded] = '\0';
|
strDispName = szValue;
|
}
|
#else
|
if ( ERROR_SUCCESS == myKey.QueryValue( szValue, _T( "DisplayName_ENU" ), &dwReaded ) && dwReaded > 0 )
|
{
|
szValue[dwReaded] = '\0';
|
strDispName = szValue;
|
}
|
#endif
|
if ( strDispName.IsEmpty() )
|
strDispName = strName;
|
m_lsAddInDispName.AddTail( strDispName );
|
m_lsAddInName.AddTail( strName );
|
}
|
|
myKey.Close( );
|
}
|
else
|
{
|
continue;
|
}
|
}
|
|
RegCloseKey(hKey);
|
}
|
|
bool CMBShellExt::CStringToAnsi( LPCTSTR pszT, LPSTR *ppszA )
|
{
|
ULONG cbAnsi, cCharacters;
|
|
if ( pszT == NULL )
|
{
|
*ppszA= NULL;
|
return false;
|
}
|
#ifdef _UNICODE
|
cCharacters = wcslen(pszT);
|
cbAnsi = ::WideCharToMultiByte( CP_ACP, 0, pszT, -1, NULL, 0, NULL, NULL );
|
*ppszA = (LPSTR) CoTaskMemAlloc(cbAnsi+1);
|
if ( NULL == *ppszA )
|
return false;
|
ZeroMemory(*ppszA,cbAnsi+1);
|
// Convert to ANSI.
|
if (0 == WideCharToMultiByte(CP_ACP, 0, pszT, cCharacters, *ppszA,
|
cbAnsi, NULL, NULL))
|
{
|
CoTaskMemFree(*ppszA);
|
*ppszA = NULL;
|
return false;
|
}
|
#else
|
cCharacters = strlen(pszT);
|
*ppszA = (LPSTR) CoTaskMemAlloc(cCharacters+1);
|
if ( NULL == *ppszA )
|
return false;
|
ZeroMemory(*ppszA,cCharacters+1);
|
memcpy(*ppszA,pszT,cCharacters);
|
#endif
|
return true;
|
}
|
|
CString CMBShellExt::GetSTKWorkDir()
|
{
|
CString strWorkDir;
|
CRegKey reg;
|
CString strSubKey;
|
TCHAR szValue[256] = { 0 };
|
DWORD dwReaded;
|
|
strSubKey.Format( _T("%s"), REG_SUBKEY );
|
if ( ERROR_SUCCESS == reg.Open( HKEY_LOCAL_MACHINE, strSubKey, KEY_READ ) )
|
{
|
dwReaded = 255;
|
if ( ERROR_SUCCESS == reg.QueryStringValue( _T( "WorkDir" ), szValue, &dwReaded ) && dwReaded > 0 )
|
{
|
strWorkDir = szValue;
|
strWorkDir.TrimRight();
|
strWorkDir.TrimRight( '\\' );
|
}
|
|
reg.Close();
|
}
|
if ( strWorkDir.IsEmpty() )
|
strWorkDir =GetModulePath();
|
return strWorkDir;
|
}
|
|
|
|
// дÈÕÖ¾
|
void CMBShellExt::WriteLog( CString strLog )
|
{
|
CTime tmCurrent = CTime::GetCurrentTime();
|
CString strTime = tmCurrent.Format( _T(" %Y-%m-%d %H:%M:%S = ") );
|
CString strEnter = _T(" \r\n ");
|
CString strLogPath = _T(" ");
|
CString strBuff;
|
LPSTR pDest ;
|
CFile file;
|
|
strLogPath.Format( _T("%s\\mbEasySend.log"),GetModulePath());
|
if ( !file.Open(strLogPath, CFile::modeCreate | CFile::modeWrite | CFile::modeNoTruncate) )
|
{
|
return ;
|
}
|
file.Seek( 0, CFile::end );
|
|
strBuff.Format( _T("%s%s%s"),strTime, strLog,strEnter );
|
if ( CStringToAnsi(strBuff, &pDest))
|
{
|
file.Write( (void*)pDest, strlen(pDest));
|
CoTaskMemFree(pDest);
|
}
|
file.Close();
|
}
|
HBITMAP CMBShellExt::GetSTKBitmap(int nMenuType)
|
{
|
HINSTANCE hInstance = NULL;
|
HBITMAP hBitmap = NULL;
|
#ifdef _WIN64
|
hInstance = ::GetModuleHandle( _T("mbEasySendW.dll") );
|
#else
|
hInstance = ::GetModuleHandle( _T("mbEasySend.dll") );
|
#endif
|
if(nMenuType == 0)
|
hBitmap = LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_AM) );//IDB_USERSTAT
|
else
|
hBitmap = LoadBitmap(hInstance, MAKEINTRESOURCE(IDB_MBC) );//IDB_USERSTAT
|
HIMAGELIST hImageList1= ImageList_Create(16,16,ILC_COLORDDB|ILC_MASK ,1,2); //
|
ImageList_SetBkColor(hImageList1, GetSysColor( COLOR_MENU ) );
|
ImageList_AddMasked(hImageList1, hBitmap, RGB(255,0,255));//ÉèÖð×ɫΪ͸Ã÷É«
|
//ImageList_Add( hImageList1, hBitmap,NULL); //AfxGetInstanceHandle()
|
CWnd *pWnd = AfxGetMainWnd(); // Get main window
|
if (pWnd == NULL)
|
pWnd = CWnd::GetDesktopWindow();
|
if(!pWnd)
|
return NULL;
|
CDC *pDC = pWnd->GetDC(); // Get device context
|
if(!pDC)
|
return NULL;
|
HBITMAP hBmp;
|
HICON hIcon = ImageList_GetIcon(hImageList1,0,0);
|
HDC hTmpDc;
|
hTmpDc = CreateCompatibleDC(pDC->GetSafeHdc());
|
hBmp = CreateCompatibleBitmap(pDC->GetSafeHdc(),16,16);
|
HBITMAP hOldBmp = (HBITMAP)::SelectObject(hTmpDc,hBmp);
|
HBRUSH hBrush ;
|
COLORREF newclrBack;
|
newclrBack=GetSysColor(COLOR_3DFACE);
|
hBrush =CreateSolidBrush(newclrBack);
|
::DrawIconEx(
|
hTmpDc,
|
0,
|
0,
|
hIcon,
|
16,
|
16,
|
0,
|
hBrush,
|
DI_NORMAL
|
);
|
(HBITMAP)::SelectObject(hTmpDc, hOldBmp );
|
DeleteDC(hTmpDc);
|
::DestroyIcon(hIcon);
|
pWnd->ReleaseDC(pDC);
|
return hBmp;
|
}
|
// »ñÈ¡Ö´ÐгÌÐòËùÔÚµÄĿ¼
|
CString CMBShellExt::GetModulePath()
|
{
|
CString strFileName;
|
HINSTANCE hInstance = NULL;
|
#ifdef _WIN64
|
hInstance = ::GetModuleHandle( _T("mbEasySendW.dll") );
|
#else
|
hInstance = ::GetModuleHandle( _T("mbEasySend.dll") );
|
#endif
|
if (m_strBinPath.IsEmpty( ))
|
{
|
int nPos;
|
|
GetModuleFileName( hInstance, strFileName.GetBuffer( MAX_PATH ), MAX_PATH );
|
strFileName.ReleaseBuffer();
|
strFileName = GetPathName( strFileName );
|
return strFileName;
|
}
|
else
|
return m_strBinPath;
|
}
|
// ȡȫ·¾¶ÎļþÃûÖеÄ·¾¶
|
CString CMBShellExt::GetPathName( CString strFileName )
|
{
|
int nPos;
|
CString strPathName;
|
|
if ( strFileName.IsEmpty() )
|
return _T( "" );
|
|
nPos = strFileName.ReverseFind( '\\' );
|
if ( nPos < 0 )
|
return _T( "" );
|
|
strPathName = strFileName.Left( nPos );
|
return strPathName;
|
}
|
