|
|
#include "shole.h"
#include "ids.h"
#include <winnlsp.h>
#include <stdio.h>
#define ARRAYSIZE(a) (sizeof(a)/sizeof(a[0]))
#ifdef SAVE_OBJECTDESCRIPTOR
extern "C" const WCHAR c_wszDescriptor[] = WSTR_SCRAPITEM L"ODS";
HRESULT Scrap_SaveODToStream(IStorage *pstgDoc, OBJECTDESCRIPTOR * pods) { //
// According to Anthony Kitowicz, we must clear this flag.
//
pods->dwStatus &= ~OLEMISC_CANLINKBYOLE1;
IStream *pstm; HRESULT hres = pstgDoc->CreateStream(c_wszDescriptor, STGM_CREATE|STGM_WRITE|STGM_SHARE_EXCLUSIVE, 0, 0, &pstm); if (SUCCEEDED(hres)) { ULONG cbWritten; hres = pstm->Write(pods, pods->cbSize, &cbWritten); DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_SOD descriptor written (%x, %d, %d)"), hres, pods->cbSize, cbWritten); pstm->Release();
if (FAILED(hres) || cbWritten<pods->cbSize) { pstgDoc->DestroyElement(c_wszDescriptor); } } else { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_SOD pstg->CreateStream failed (%x)"), hres); }
return hres; }
HRESULT Scrap_SaveObjectDescriptor(IStorage *pstgDoc, IDataObject *pdtobj, IPersistStorage *pps, BOOL fLink) { STGMEDIUM medium; FORMATETC fmte = {fLink ? CF_LINKSRCDESCRIPTOR : CF_OBJECTDESCRIPTOR, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL}; HRESULT hres = pdtobj->GetData(&fmte, &medium); if (hres == S_OK) { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_SOD found CF_OBJECTDESCRIPTOR (%x)"), medium.hGlobal); LPOBJECTDESCRIPTOR pods = (LPOBJECTDESCRIPTOR)GlobalLock(medium.hGlobal); if (pods) { hres = Scrap_SaveODToStream(pstgDoc, pods); GlobalUnlock(medium.hGlobal); } ReleaseStgMedium(&medium); } //
// Attempt to create object descriptor
//
#if 0
else { hres = E_FAIL; DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_SOD CAN'T find CF_OBJECTDESCRIPTOR (%x)"), hres); LPOLEOBJECT pole; if (SUCCEEDED(pps->QueryInterface(IID_IOleObject, (LPVOID*)&pole))) { IMalloc *pmem; if (SUCCEEDED(CoGetMalloc(MEMCTX_TASK, &pmem))) { LPWSTR pwsz; if (SUCCEEDED(pole->GetUserType(USERCLASSTYPE_FULL, &pwsz))) { UINT cbStr = pmem->GetSize(pwsz) + 1; UINT cb = sizeof(OBJECTDESCRIPTOR) + cbStr; LPOBJECTDESCRIPTOR pods=(LPOBJECTDESCRIPTOR)LocalAlloc(LPTR, cb); if (pods) { pods->cbSize = cb; pole->GetUserClassID(&pods->clsid); pole->GetMiscStatus(DVASPECT_CONTENT, &pods->dwStatus); pole->GetExtent(DVASPECT_CONTENT, &pods->sizel); pods->dwFullUserTypeName = sizeof(OBJECTDESCRIPTOR); MoveMemory(pods+1, pwsz, cbStr); hres = Scrap_SaveODToStream(pstgDoc, pods); } pmem->Free(pwsz); } pmem->Release(); } pole->Release(); } } #endif
return hres; } #else
#define Scrap_SaveObjectDescriptor(pstgDoc, pdtobj, fLink) (0)
#endif // SAVE_OBJECTDESCRIPTOR
#ifdef FIX_ROUNDTRIP
extern "C" const TCHAR c_szCLSID[] = TEXT("CLSID");
//
// This function opens the HKEY for the specified CLSID or its sub-key.
//
// Parameters:
// rclsid -- Specifies the CLSID
// pszSubKey -- Specifies the subkey name, may be NULL
//
// Returns:
// non-NULL, if succeeded; the caller must RegCloseKey it.
// NULL, if failed.
//
HKEY _OpenCLSIDKey(REFCLSID rclsid, LPCTSTR pszSubKey) { #ifdef UNICODE
WCHAR szCLSID[256]; if (StringFromGUID2(rclsid, szCLSID, ARRAYSIZE(szCLSID))) { #else
WCHAR wszCLSID[256]; if (StringFromGUID2(rclsid, wszCLSID, ARRAYSIZE(wszCLSID))) { TCHAR szCLSID[80]; WideCharToMultiByte(CP_ACP, 0, wszCLSID, -1, szCLSID, ARRAYSIZE(szCLSID), NULL, NULL); #endif
TCHAR szKey[256]; if (pszSubKey) { wsprintf(szKey, TEXT("%s\\%s\\%s"), c_szCLSID, szCLSID, pszSubKey); } else { wsprintf(szKey, TEXT("%s\\%s"), c_szCLSID, szCLSID); } DebugMsg(DM_TRACE, TEXT("sc TR - _OpelCLSIDKey RegOpenKey(%s)"), szKey);
HKEY hkey; if (RegOpenKey(HKEY_CLASSES_ROOT, szKey, &hkey)==ERROR_SUCCESS) { return hkey; } } return NULL; }
extern "C" const WCHAR c_wszFormatNames[] = WSTR_SCRAPITEM L"FMT"; #define CCH_FORMATNAMES (ARRAYSIZE(c_wszFormatNames)-1)
//
// This function generates the stream name (UNICODE) for the spcified
// clipboard format.
//
// Parameters:
// pszFormat -- Specifies the clipboard format ("#0"-"#15" for predefined ones)
// wszStreamName -- Specifies the UNICODE buffer.
// cchmax -- Specifies the size of buffer.
//
void _GetCacheStreamName(LPCTSTR pszFormat, LPWSTR wszStreamName, UINT cchMax) { MoveMemory(wszStreamName, c_wszFormatNames, sizeof(c_wszFormatNames)); #ifdef UNICODE
lstrcpyn(wszStreamName+CCH_FORMATNAMES,pszFormat,cchMax-CCH_FORMATNAMES); #ifdef DEBUG
DebugMsg(DM_TRACE, TEXT("sc TR _GetCacheStreamName returning %s"), wszStreamName); #endif
#else
MultiByteToWideChar(CP_ACP, 0, pszFormat, -1, wszStreamName+CCH_FORMATNAMES, cchMax-CCH_FORMATNAMES); #ifdef DEBUG
TCHAR szT[256]; WideCharToMultiByte(CP_ACP, 0, wszStreamName, -1, szT, ARRAYSIZE(szT), NULL, NULL); DebugMsg(DM_TRACE, TEXT("sc TR _GetCacheStreamName returning %s"), szT); #endif
#endif
}
HRESULT Scrap_CacheOnePictureFormat(LPCTSTR pszFormat, FORMATETC * pfmte, STGMEDIUM * pmedium, REFCLSID rclsid, LPSTORAGE pstgDoc, LPDATAOBJECT pdtobj) { LPPERSISTSTORAGE ppstg; HRESULT hres = OleCreateDefaultHandler(rclsid, NULL, IID_IPersistStorage, (LPVOID *)&ppstg); DebugMsg(DM_TRACE, TEXT("sc TR Scrap_CacheOPF OleCreteDefHandler returned %x"), hres); if (SUCCEEDED(hres)) { //
// Generate the stream name based on the clipboard format name.
//
WCHAR wszStorageName[256]; _GetCacheStreamName(pszFormat, wszStorageName, ARRAYSIZE(wszStorageName));
LPSTORAGE pstgPicture; hres = pstgDoc->CreateStorage(wszStorageName, STGM_DIRECT | STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, 0, &pstgPicture); if (SUCCEEDED(hres)) { ppstg->InitNew(pstgPicture);
LPOLECACHE pcache; hres = ppstg->QueryInterface(IID_IOleCache, (LPVOID*)&pcache); DebugMsg(DM_TRACE, TEXT("sc TR Scrap_CacheOPF QI returned %x"), hres); if (SUCCEEDED(hres)) { hres = pcache->Cache(pfmte, ADVF_PRIMEFIRST, NULL); DebugMsg(DM_TRACE, TEXT("sc TR pcache->Cache returned %x"), hres); hres = pcache->SetData(pfmte, pmedium, FALSE); DebugMsg(DM_TRACE, TEXT("sc TR pcache->SetData returned %x"), hres); pcache->Release();
if (SUCCEEDED(hres)) { hres = OleSave(ppstg, pstgPicture, TRUE); DebugMsg(DM_TRACE, TEXT("sc TR Scrap_CacheOnePictureFormat OleSave returned (%x)"), hres); ppstg->HandsOffStorage();
if (SUCCEEDED(hres)) { hres = pstgPicture->Commit(STGC_OVERWRITE); DebugMsg(DM_TRACE, TEXT("sc TR Scrap_CacheOnePictureFormat Commit() returned (%x)"), hres); } } }
pstgPicture->Release();
if (FAILED(hres)) { pstgDoc->DestroyElement(wszStorageName); } }
ppstg->Release(); }
return hres; }
//
// This function stores the specified format of clipboard data.
//
// Parameters:
// pszFormat -- Specifies the clipboard format ("#0"-"#15" for predefined ones)
// pstgDoc -- Specifies the top level IStorage.
// pdtobj -- Sepcified the data object we should get the data from.
//
HRESULT Scrap_CacheOneFormat(LPCTSTR pszFormat, LPSTORAGE pstgDoc, LPDATAOBJECT pdtobj) { UINT cf = RegisterClipboardFormat(pszFormat); STGMEDIUM medium; FORMATETC fmte = {(CLIPFORMAT)cf, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
const CLSID * pclsid = NULL; switch(cf) { case CF_METAFILEPICT: pclsid = &CLSID_Picture_Metafile; fmte.tymed = TYMED_MFPICT; break;
case CF_ENHMETAFILE: pclsid = &CLSID_Picture_EnhMetafile; fmte.tymed = TYMED_ENHMF; break;
case CF_PALETTE: case CF_BITMAP: pclsid = &CLSID_Picture_Dib; fmte.tymed = TYMED_GDI; break; }
//
// Get the specified format of data (TYMED_GLOBAL only)
//
HRESULT hres = pdtobj->GetData(&fmte, &medium); if (hres == S_OK) { if (medium.tymed != TYMED_HGLOBAL) { hres = Scrap_CacheOnePictureFormat(pszFormat, &fmte, &medium, *pclsid, pstgDoc, pdtobj); } else { //
// Global lock the data.
//
UINT cbData = (UINT)GlobalSize(medium.hGlobal); const BYTE * pbData = (const BYTE*)GlobalLock(medium.hGlobal); if (pbData) { //
// Generate the stream name based on the clipboard format name.
//
WCHAR wszStreamName[256]; _GetCacheStreamName(pszFormat, wszStreamName, ARRAYSIZE(wszStreamName));
//
// Create the stream.
//
LPSTREAM pstm; hres = pstgDoc->CreateStream(wszStreamName, STGM_CREATE|STGM_WRITE|STGM_SHARE_EXCLUSIVE, 0, 0, &pstm); if (SUCCEEDED(hres)) { //
// Save the size of data.
//
ULONG cbWritten; hres = pstm->Write(&cbData, sizeof(cbData), &cbWritten); if (SUCCEEDED(hres) && cbWritten==sizeof(cbData)) { //
// Save the data itself.
//
hres = pstm->Write(pbData, cbData, &cbWritten); DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_Save %s written (%x, %d, %d)"), pszFormat, hres, cbData, cbWritten); } pstm->Release();
//
// If anything goes wrong, destroy the stream.
//
if (FAILED(hres) || cbWritten<cbData) { pstgDoc->DestroyElement(wszStreamName); hres = E_FAIL; } } GlobalUnlock(medium.hGlobal); } else { hres = E_FAIL; } } ReleaseStgMedium(&medium); } else { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheOneFormat IDO::GetData(cf=%x,tm=%x) failed (%x)"), fmte.cfFormat, fmte.tymed, hres); }
return hres; }
//
// This function caches the specified format of data if the data object
// support that format.
//
// Parameters:
// szFormat -- Specifies the format to be cahced
// pstgDoc -- Specifies the top level IStorage
// pdtobj -- Specifies the data object from where we get data
// pstm -- Specifies the stream we should write cached format name.
//
// Returns:
// TRUE if the data object support it.
//
BOOL Scrap_MayCacheOneFormat(LPCTSTR szFormat, LPSTORAGE pstgDoc, LPDATAOBJECT pdtobj, LPSTREAM pstm) { //
// Try to cache the format.
//
HRESULT hres = Scrap_CacheOneFormat(szFormat, pstgDoc, pdtobj); if (SUCCEEDED(hres)) { //
// Store the name of format only if we actually
// succeeded to cache the data.
//
#ifdef UNICODE
CHAR szAnsiFormat[128]; WideCharToMultiByte(CP_ACP, 0, szFormat, -1, szAnsiFormat, ARRAYSIZE(szAnsiFormat), NULL, NULL ); USHORT cb = (USHORT)lstrlenA(szAnsiFormat); #else
USHORT cb = (USHORT)lstrlen(szFormat); #endif
pstm->Write(&cb, sizeof(cb), NULL); DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_MayCacheOneFormat writing %s, %d"), szFormat, cb); #ifdef UNICODE
pstm->Write(szAnsiFormat, cb, NULL); #else
pstm->Write(szFormat, cb, NULL); #endif
return TRUE; }
return FALSE; }
//
// Returns:
// TRUE, if the specified format is already cached (from Global list).
//
BOOL Scrap_IsAlreadyCached(UINT acf[], UINT ccf, LPCTSTR szFormat) { if (ccf) { UINT cf = RegisterClipboardFormat(szFormat); for (UINT icf=0; icf<ccf; icf++) { if (acf[icf]==cf) { return TRUE; } } } return FALSE; }
extern "C" const TCHAR c_szCacheFMT[] = TEXT("DataFormats\\PriorityCacheFormats"); #define REGSTR_PATH_SCRAP TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\ShellScrap")
extern "C" const TCHAR c_szGlobalCachedFormats[] = REGSTR_PATH_SCRAP TEXT("\\PriorityCacheFormats");
//
// This function enumerate the list of to-be-cached clipboard data and
// calls Scrap_CacheOneFormat for each of them.
//
// Parameters:
// pstgDoc -- Specifies the top level IStorage.
// pdtobj -- Specifies the data object we'll get the data from.
// pps -- Specifies the "embedded object" (to get CLSID from)
//
void Scrap_CacheClipboardData(LPSTORAGE pstgDoc, LPDATAOBJECT pdtobj, LPPERSIST pps) { //
// Create the stream where we'll store the list of actually
// cached formats, which might be just a subset of to-be-cached
// format specified in the registry.
//
LPSTREAM pstm; HRESULT hres = pstgDoc->CreateStream(c_wszFormatNames, STGM_CREATE|STGM_WRITE|STGM_SHARE_EXCLUSIVE, 0, 0, &pstm);
DebugMsg(DM_TRACE, TEXT("sc TR S_CCD CreateStream returned %x"), hres);
if (SUCCEEDED(hres)) { USHORT cb; HKEY hkey; TCHAR szFormatName[128]; DWORD cchValueName; DWORD dwType; UINT acf[CCF_CACHE_GLOBAL]; UINT ccf = 0;
//
// First, try the formats in the global list.
//
if (RegOpenKey(HKEY_LOCAL_MACHINE, c_szGlobalCachedFormats, &hkey)==ERROR_SUCCESS) { //
// For each global to-be-cached format...
//
for(int iValue=0; iValue<CCF_CACHE_GLOBAL ;iValue++) { //
// Get the value name of the iValue'th value. The value
// name specifies the clipboard format.
// ("#0"-"#15" for predefined formats).
//
cchValueName = ARRAYSIZE(szFormatName); if (RegEnumValue(hkey, iValue, szFormatName, &cchValueName, NULL, &dwType, NULL, NULL)==ERROR_SUCCESS) { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheClipboardData found with %s, %x (Global)"), szFormatName, dwType); if (Scrap_MayCacheOneFormat(szFormatName, pstgDoc, pdtobj, pstm)) { acf[ccf++] = RegisterClipboardFormat(szFormatName); } } else { break; } }
RegCloseKey(hkey); }
//
// Then, try the CLSID specific formats.
//
// Get the CLSID of the "embedded object" (the body of scrap)
//
CLSID clsid; hres = pps->GetClassID(&clsid); if (SUCCEEDED(hres)) { //
// Open the key for the list of to-be-cached formats.
//
hkey = _OpenCLSIDKey(clsid, c_szCacheFMT); if (hkey) { //
// For each class specific to-be-cached format...
//
for(int iValue=0; iValue<CCF_CACHE_CLSID ;iValue++) { //
// Get the value name of the iValue'th value. The value
// name specifies the clipboard format.
// ("#0"-"#15" for predefined formats).
//
cchValueName = ARRAYSIZE(szFormatName);
if (RegEnumValue(hkey, iValue, szFormatName, &cchValueName, NULL, &dwType, NULL, NULL)==ERROR_SUCCESS) { if (!Scrap_IsAlreadyCached(acf, ccf, szFormatName)) { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheClipboardData found with %s, %x (CLSID specific)"), szFormatName, dwType); Scrap_MayCacheOneFormat(szFormatName, pstgDoc, pdtobj, pstm); } else { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheClipboardData skipping %s (already cached)"), szFormatName); } } else { break; } }
//
// HACK: NT 3.5's RegEdit does not support named values...
//
for(iValue=0; iValue<CCF_CACHE_CLSID ;iValue++) { TCHAR szKeyName[128]; if (RegEnumKey(hkey, iValue, szKeyName, ARRAYSIZE(szKeyName))==ERROR_SUCCESS) { LONG cbValue = ARRAYSIZE(szFormatName); if ((RegQueryValue(hkey, szKeyName, szFormatName, &cbValue)==ERROR_SUCCESS) && cbValue) { if (!Scrap_IsAlreadyCached(acf, ccf, szFormatName)) { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheClipboardData found with %s, %x (CLSID specific)"), szFormatName, dwType); Scrap_MayCacheOneFormat(szFormatName, pstgDoc, pdtobj, pstm); } else { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CacheClipboardData skipping %s (already cached)"), szFormatName); } } } else { break; } }
RegCloseKey(hkey); } }
//
// Put the terminator.
//
cb = 0; pstm->Write(&cb, sizeof(cb), NULL); pstm->Release();
}
} #endif // FIX_ROUNDTRIP
// out:
// pszName - short name for object type ("Worksheet", "Word Document", etc)
//
// returns:
//
HRESULT Scrap_Save(IStorage *pstg, IStorage *pstgDoc, IDataObject *pdtobj, BOOL fLink, LPTSTR pszName) { IPersistStorage *pps; HRESULT hres;
if (fLink) { FORMATETC fmte = {CF_METAFILEPICT, NULL, DVASPECT_ICON, -1, TYMED_MFPICT}; hres = OleCreateLinkFromData(pdtobj, IID_IPersistStorage, OLERENDER_FORMAT, &fmte, NULL, pstg, (LPVOID*)&pps); DebugMsg(DM_TRACE, TEXT("sc Scrap_Save OleCreateLinkFromData(FMT) returned (%x)"), hres); } else { hres = OleCreateFromData(pdtobj, IID_IPersistStorage, OLERENDER_DRAW, NULL, NULL, pstg, (LPVOID*)&pps); DebugMsg(DM_TRACE, TEXT("sc Scrap_Save OleCreateFromData(FMT) returned (%x)"), hres); }
if (SUCCEEDED(hres)) { hres = OleSave(pps, pstg, TRUE); // fSameStorage=TRUE
DebugMsg(DM_TRACE, TEXT("sc Scrap_Save OleSave returned (%x)"), hres);
if (SUCCEEDED(hres) && pszName) { LPOLEOBJECT pole; if (SUCCEEDED(pps->QueryInterface(IID_IOleObject, (LPVOID*)&pole))) { IMalloc *pmem; if (SUCCEEDED(CoGetMalloc(MEMCTX_TASK, &pmem))) { LPWSTR pwsz; if (SUCCEEDED(pole->GetUserType(USERCLASSTYPE_SHORT, &pwsz))) { #ifdef UNICODE
lstrcpyn(pszName, pwsz, 64); // What is 64?
#else
WideCharToMultiByte(CP_ACP, 0, pwsz, -1, pszName, 64, NULL, NULL); #endif
DebugMsg(DM_TRACE, TEXT("sc Scrap_Save short name (%s)"), pszName);
// Assert(lstrlen(pszName) < 15); // USERCLASSTYPE_SHORT docs say so
pmem->Free(pwsz); } pmem->Release(); } pole->Release(); } }
// This is no-op if SAVE_OBJECTDESCRIPTOR is not defined.
Scrap_SaveObjectDescriptor(pstgDoc, pdtobj, pps, fLink);
#ifdef FIX_ROUNDTRIP
if (!fLink) { Scrap_CacheClipboardData(pstgDoc, pdtobj, pps); } #endif // FIX_ROUNDTRIP
hres = pps->HandsOffStorage();
pps->Release(); }
return hres; }
// get some text from the data object
//
// out:
// pszOut filled in with text
//
HRESULT Scrap_GetText(IDataObject *pdtobj, LPTSTR pszOut, UINT cchMax) { UINT cbMac = (cchMax-1)*sizeof(pszOut[0]); memset(pszOut, 0, cchMax * sizeof(pszOut[0]));
STGMEDIUM medium; HRESULT hres; #ifdef UNICODE
UINT CodePage = CP_ACP; BOOL bUnicode = TRUE;
FORMATETC fmte = { CF_UNICODETEXT, NULL, DVASPECT_CONTENT, -1, TYMED_ISTREAM | TYMED_HGLOBAL }; hres = pdtobj->QueryGetData( &fmte ); if (hres != S_OK) // S_FALSE means no.
{ fmte.cfFormat = CF_TEXT; bUnicode = FALSE; } hres = pdtobj->GetData(&fmte, &medium);
if (bUnicode == FALSE && hres == S_OK) { //
// We have ANSI text but no UNICODE Text. Look for RTF in order to
// see if we can find a language id so that we can use the correct
// code page for the Ansi to Unicode translation.
//
FORMATETC fetcrtf = { 0, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL | TYMED_ISTREAM }; fetcrtf.cfFormat = (CLIPFORMAT)RegisterClipboardFormat(TEXT("Rich Text Format")); STGMEDIUM mediumrtf;
if (SUCCEEDED(pdtobj->GetData(&fetcrtf, &mediumrtf))) { CHAR szBuf[MAX_PATH * 8]; LPSTR pszRTF = NULL;
if (mediumrtf.tymed == TYMED_ISTREAM) { if (SUCCEEDED(mediumrtf.pstm->Read((LPTSTR)szBuf, ARRAYSIZE(szBuf), NULL))) { pszRTF = szBuf; } } else { pszRTF = (LPSTR)GlobalLock(mediumrtf.hGlobal); }
if (pszRTF) { LPSTR pTmp;
//
// Find the language id used for this text.
//
// Ugly way to search, but can't use c-runtimes in the
// shell.
//
CHAR szLang[5]; UINT LangID = 0;
pTmp = pszRTF; while (*pTmp) { if ((*pTmp == '\\') && *(pTmp + 1) && (*(pTmp + 1) == 'l') && *(pTmp + 2) && (*(pTmp + 2) == 'a') && *(pTmp + 3) && (*(pTmp + 3) == 'n') && *(pTmp + 4) && (*(pTmp + 4) == 'g')) { //
// Get number following the \lang identifier.
//
int ctr;
pTmp += 5; for (ctr = 0; (ctr < 4) && (*(pTmp + ctr)) && ((*(pTmp + ctr)) >= '0') && ((*(pTmp + ctr)) <= '9'); ctr++) { szLang[ctr] = *(pTmp + ctr); } szLang[ctr] = 0;
for (pTmp = szLang; *pTmp; pTmp++) { LangID *= 10; LangID += (*pTmp - '0'); }
break; } pTmp++; } if (LangID) { if (!GetLocaleInfo( LangID, LOCALE_IDEFAULTANSICODEPAGE | LOCALE_RETURN_NUMBER, (LPTSTR)&CodePage, sizeof(UINT) / sizeof(TCHAR) )) { CodePage = CP_ACP; } }
if (mediumrtf.tymed == TYMED_HGLOBAL) { GlobalUnlock(mediumrtf.hGlobal); } } } }
#else
FORMATETC fmte = { CF_TEXT, NULL, DVASPECT_CONTENT, -1, TYMED_ISTREAM|TYMED_HGLOBAL }; hres = pdtobj->GetData(&fmte, &medium); #endif
if (SUCCEEDED(hres)) { DebugMsg(DM_TRACE, TEXT("sh TR - Scrap_GetText found CF_TEXT/CF_UNICODETEXT in %d"), medium.tymed); if (medium.tymed == TYMED_ISTREAM) { hres = medium.pstm->Read(pszOut, cbMac, NULL); } else if (medium.tymed == TYMED_HGLOBAL) { DebugMsg(DM_TRACE, TEXT("sh TR - Scrap_GetText found CF_TEXT/CF_UNICODETEXT in global")); LPCTSTR pszSrc = (LPCTSTR)GlobalLock(medium.hGlobal); if (pszSrc) { #ifdef UNICODE
if ( fmte.cfFormat == CF_TEXT ) { MultiByteToWideChar( CodePage, 0, (LPSTR)pszSrc, cchMax, pszOut, cchMax ); } else #endif
{ MoveMemory(pszOut, pszSrc, cbMac); } GlobalUnlock(medium.hGlobal); } } ReleaseStgMedium(&medium); }
return hres; }
// Avoid linking lots of CRuntime stuff.
#undef isdigit
#undef isalpha
#define isdigit(ch) (ch>=TEXT('0') && ch<=TEXT('9'))
#define isalpha(ch) ((ch&0xdf)>=TEXT('A') && (ch&0xdf)<=TEXT('Z'))
#define CCH_MAXLEADINGSPACE 256
#define CCH_COPY 16
//
// create a fancy name for a scrap/doc shortcut given the data object to get some
// text from
//
// out:
// pszNewName - assumed to be 64 chars at least
//
BOOL Scrap_GetFancyName(IDataObject *pdtobj, UINT idTemplate, LPCTSTR pszPath, LPCTSTR pszTypeName, LPTSTR pszNewName) { TCHAR szText[CCH_MAXLEADINGSPACE + CCH_COPY + 1]; HRESULT hres = Scrap_GetText(pdtobj, szText, ARRAYSIZE(szText));
if (SUCCEEDED(hres)) { #ifdef UNICODE
DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_GetFancyName CF_UNICODETEXT has (%s)"), szText); #else
DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_GetFancyName CF_TEXT has (%s)"), szText); #endif
LPTSTR pszStart; //
// skip leading space/non-printing characters
//
for (pszStart = szText; *pszStart <= TEXT(' '); pszStart++) { if (*pszStart == TEXT('\0')) return FALSE; // empty string
if (pszStart - szText >= CCH_MAXLEADINGSPACE) return FALSE; // too many leading space
} DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_GetFancyName pszStart (%s)"), pszStart);
//
// Chacter conversion
// (1) non-printing characters -> ' '
// (2) invalid characters -> '_'
//
for (LPTSTR pszT = pszStart; *pszT && ((pszT-pszStart) < CCH_COPY); pszT = CharNext(pszT)) { TCHAR ch = *pszT; if (ch <= TEXT(' ')) { *pszT = TEXT(' '); } else if (ch < 127 && !isdigit(ch) && !isalpha(ch)) { switch(ch) { case TEXT('$'): case TEXT('%'): case TEXT('\''): case TEXT('-'): case TEXT('_'): case TEXT('@'): case TEXT('~'): case TEXT('`'): case TEXT('!'): case TEXT('('): case TEXT(')'): case TEXT('{'): case TEXT('}'): case TEXT('^'): case TEXT('#'): case TEXT('&'): break;
default: *pszT = TEXT('_'); break; } } } *pszT = 0;
TCHAR szTemplate[MAX_PATH]; TCHAR szName[MAX_PATH];
LoadString(HINST_THISDLL, idTemplate, szTemplate, ARRAYSIZE(szTemplate)); wsprintf(szName, szTemplate, pszTypeName, pszStart);
PathCombine(szName, pszPath, szName);
if (!PathFileExists(szName)) { DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_GetFancyName (%s)"), szName); lstrcpy(pszNewName, szName); return TRUE; } } return FALSE; }
// *** WARNING ***
//
// Scrap_CreateFromDataObject is a TCHAR export from SHSCRAP.DLL that is used by SHELL32.DLL. If you
// change its calling convention, you must modify shell32's wrapper as well as well.
//
// *** WARNING ***
HRESULT WINAPI Scrap_CreateFromDataObject(LPCTSTR pszPath, IDataObject *pdtobj, BOOL fLink, LPTSTR pszNewFile) { HRESULT hres; TCHAR szTemplateS[32]; TCHAR szTemplateL[128]; TCHAR szTypeName[64]; #ifndef UNICODE
WCHAR wszNewFile[MAX_PATH]; #endif
IStorage *pstg; UINT idErr = 0;
DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CreateFromDataObject called at %s"), pszPath);
LoadString(HINST_THISDLL, fLink ? IDS_BOOKMARK_S : IDS_SCRAP_S, szTemplateS, ARRAYSIZE(szTemplateS)); LoadString(HINST_THISDLL, fLink ? IDS_BOOKMARK_L : IDS_SCRAP_L, szTemplateL, ARRAYSIZE(szTemplateL));
PathYetAnotherMakeUniqueName(pszNewFile, pszPath, szTemplateS, szTemplateL);
DebugMsg(DM_TRACE, TEXT("sc TR - Scrap_CreateFromDataObject creating %s"), pszNewFile);
#ifdef UNICODE
hres = StgCreateDocfile(pszNewFile, STGM_DIRECT | STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, &pstg); #else
MultiByteToWideChar(CP_ACP, 0, pszNewFile, -1, wszNewFile, ARRAYSIZE(wszNewFile));
hres = StgCreateDocfile(wszNewFile, STGM_DIRECT | STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, &pstg); #endif
if (SUCCEEDED(hres)) { IStorage *pstgContents;
hres = pstg->CreateStorage(c_wszContents, STGM_DIRECT | STGM_READWRITE | STGM_CREATE | STGM_SHARE_EXCLUSIVE, 0, 0, &pstgContents);
if (SUCCEEDED(hres)) { hres = Scrap_Save(pstgContents, pstg, pdtobj, fLink, szTypeName); if (SUCCEEDED(hres)) { hres = pstgContents->Commit(STGC_OVERWRITE); if (FAILED(hres)) idErr = IDS_ERR_COMMIT; } else { idErr = IDS_ERR_SCRAPSAVE; } pstgContents->Release(); } else { idErr = IDS_ERR_CREATESTORAGE; }
//
// We need to delete the file, if failed to save/commit.
//
if (SUCCEEDED(hres)) { hres = pstg->Commit(STGC_OVERWRITE); if (FAILED(hres)) idErr = IDS_ERR_COMMIT; }
pstg->Release();
if (FAILED(hres)) DeleteFile(pszNewFile); } else { idErr = IDS_ERR_CREATEDOCFILE; }
if (SUCCEEDED(hres)) { if (IsLFNDrive(pszPath)) { TCHAR szFancyName[MAX_PATH];
if (Scrap_GetFancyName(pdtobj, fLink ? IDS_TEMPLINK : IDS_TEMPSCRAP, pszPath, szTypeName, szFancyName)) { if (MoveFile(pszNewFile, szFancyName)) lstrcpy(pszNewFile, szFancyName); } } } else { DisplayError((HWND)NULL, hres, idErr, pszNewFile); }
return hres; }
#ifdef DEBUG
static const LPFNSCRAPCREATEFROMDATAOBJECT s_pfn = Scrap_CreateFromDataObject; #endif
|