home
***
CD-ROM
|
disk
|
FTP
|
other
***
search
/
Tricks of the Windows Gam…ming Gurus (2nd Edition)
/
Disc2.iso
/
vc98
/
mfc
/
src
/
olereg.cpp
< prev
next >
Wrap
C/C++ Source or Header
|
1998-06-16
|
16KB
|
538 lines
// This is a part of the Microsoft Foundation Classes C++ library.
// Copyright (C) 1992-1998 Microsoft Corporation
// All rights reserved.
//
// This source code is only intended as a supplement to the
// Microsoft Foundation Classes Reference and related
// electronic documentation provided with the library.
// See these sources for detailed information regarding the
// Microsoft Foundation Classes product.
#include "stdafx.h"
#include <shellapi.h>
#ifdef AFX_OLE4_SEG
#pragma code_seg(AFX_OLE4_SEG)
#endif
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define new DEBUG_NEW
// from docmgr.cpp
extern BOOL AFXAPI _AfxDeleteRegKey(LPCTSTR lpszKey);
//////////////////////////////////////////////////////////////////////////////
// data for UpdateRegistry functionality
// %1 - class ID
// %2 - class name
// %3 - SFN executable path
// %4 - short type name
// %5 - long type name
// %6 - long application name
// %7 - icon index
// %8 - Creator(xxxxxxxx) [mac-only]
// %8 - File extension Description [non-mac only]
// %9 - (not used yet) [mac-only]
// %9 - File extension *.TLA [non-mac only]
// %A - (not used yet)
#define NUM_REG_VARS 10
class CAfxOleSymbolTable
{
protected:
LPTSTR* m_strEntries;
int m_nEntries;
public:
CAfxOleSymbolTable(int nEntries);
~CAfxOleSymbolTable();
LPCTSTR* GetArray() { return (LPCTSTR*) m_strEntries; }
LPCTSTR GetAt(int nIndex) const;
void SetAt(int nIndex, LPCTSTR pstr);
LPCTSTR operator[](int nIndex) const { return GetAt(nIndex); }
};
CAfxOleSymbolTable::CAfxOleSymbolTable(int nEntries)
{
m_strEntries = new LPTSTR[nEntries];
memset(m_strEntries, 0, sizeof(LPTSTR) * nEntries);
m_nEntries = nEntries;
}
CAfxOleSymbolTable::~CAfxOleSymbolTable()
{
int nIndex;
for (nIndex = 0; nIndex < m_nEntries; nIndex++)
free(m_strEntries[nIndex]);
delete [] m_strEntries;
}
void CAfxOleSymbolTable::SetAt(int nIndex, LPCTSTR pstr)
{
ASSERT(nIndex < m_nEntries && nIndex >= 0);
free(m_strEntries[nIndex]);
m_strEntries[nIndex] = pstr ? _tcsdup(pstr) : NULL;
}
LPCTSTR CAfxOleSymbolTable::GetAt(int nIndex) const
{
if (nIndex < m_nEntries || nIndex < 0)
return m_strEntries[nIndex];
else
return NULL;
}
static const TCHAR sz00[] = _T("%2\0") _T("%5");
static const TCHAR sz01[] = _T("%2\\CLSID\0") _T("%1");
static const TCHAR sz02[] = _T("%2\\Insertable\0") _T("");
static const TCHAR sz03[] = _T("%2\\protocol\\StdFileEditing\\verb\\0\0") _T("&Edit");
static const TCHAR sz04[] = _T("%2\\protocol\\StdFileEditing\\server\0") _T("%3");
static const TCHAR sz05[] = _T("CLSID\\%1\0") _T("%5");
static const TCHAR sz06[] = _T("CLSID\\%1\\ProgID\0") _T("%2");
static const TCHAR sz07[] = _T("CLSID\\%1\\InprocHandler32\0") _T("ole32.dll");
static const TCHAR sz08[] = _T("CLSID\\%1\\LocalServer32\0") _T("%3");
static const TCHAR sz09[] = _T("CLSID\\%1\\Verb\\0\0") _T("&Edit,0,2");
static const TCHAR sz10[] = _T("CLSID\\%1\\Verb\\1\0") _T("&Open,0,2");
static const TCHAR sz11[] = _T("CLSID\\%1\\Insertable\0") _T("");
static const TCHAR sz12[] = _T("CLSID\\%1\\AuxUserType\\2\0") _T("%4");
static const TCHAR sz13[] = _T("CLSID\\%1\\AuxUserType\\3\0") _T("%6");
static const TCHAR sz14[] = _T("CLSID\\%1\\DefaultIcon\0") _T("%3,%7");
static const TCHAR sz15[] = _T("CLSID\\%1\\MiscStatus\0") _T("32");
static const TCHAR sz16[] = _T("\0") _T("");
static const TCHAR sz17[] = _T("CLSID\\%1\\InProcServer32\0") _T("%3");
static const TCHAR sz18[] = _T("CLSID\\%1\\DocObject\0" _T("0")); // CLSIDDocObject
static const TCHAR sz19[] = _T("%2\\DocObject\0" _T("0")); // ProgIDDocObject
static const TCHAR sz20[] = _T("CLSID\\%1\\Printable\0"); // szPrintable
static const TCHAR sz21[] = _T("CLSID\\%1\\DefaultExtension\0%9, %8"); // szDefaultExt
// registration for OAT_INPLACE_SERVER
static const LPCTSTR rglpszInPlaceRegister[] =
{
sz00, sz02, sz03, sz05, sz09, sz10, sz11, sz12,
sz13, sz15,
NULL
};
// registration for OAT_SERVER
static const LPCTSTR rglpszServerRegister[] =
{
sz00, sz02, sz03, sz05, sz09, sz11, sz12,
sz13, sz15,
NULL
};
// registration for OAT_DOC_OBJECT_SERVER
static const LPCTSTR rglpszDocObjectRegister[] =
{
sz00, sz02, sz03, sz05, sz09, sz10, sz11, sz12,
sz13, sz15, sz18, sz19, sz20,
NULL
};
// overwrite entries for OAT_SERVER & OAT_INPLACE_SERVER
static const LPCTSTR rglpszServerOverwrite[] =
{
sz01, sz04, sz06, sz07, sz08, sz14, NULL
};
// overwrite entries for OAT_SERVER & OAT_INPLACE_SERVER (dll)
static const LPCTSTR rglpszServerOverwriteDLL[] =
{
sz01, sz04, sz06,
sz17,
sz14,
NULL
};
// registration for OAT_CONTAINER
static const LPCTSTR rglpszContainerRegister[] =
{
sz00, sz05, NULL
};
// overwrite entries for OAT_CONTAINER
static const LPCTSTR rglpszContainerOverwrite[] =
{
sz01, sz06, sz07, sz08, sz14, NULL
};
// registration for OAT_DISPATCH_OBJECT
static const LPCTSTR rglpszDispatchRegister[] =
{
sz00, sz05, NULL
};
// overwrite entries for OAT_DISPATCH_OBJECT
static const LPCTSTR rglpszDispatchOverwrite[] =
{
sz01, sz06, sz07, sz08, NULL
};
// overwrite entries for OAT_DISPATCH_OBJECT (dll)
static const LPCTSTR rglpszDispatchOverwriteDLL[] =
{
sz01, sz06,
sz17,
NULL
};
// overwrite entries for OAT_DOC_OBJECT_SERVER
static const LPCTSTR rglpszDocObjectOverwrite[] =
{
sz01, sz04, sz06, sz07, sz08, sz14, sz21,
NULL
};
struct STANDARD_ENTRY
{
const LPCTSTR* rglpszRegister;
const LPCTSTR* rglpszOverwrite;
};
static const STANDARD_ENTRY rgStdEntries[] =
{
{ rglpszInPlaceRegister, rglpszServerOverwrite },
{ rglpszServerRegister, rglpszServerOverwrite },
{ rglpszContainerRegister, rglpszContainerOverwrite },
{ rglpszDispatchRegister, rglpszDispatchOverwrite },
{ rglpszDocObjectRegister, rglpszDocObjectOverwrite },
};
static const STANDARD_ENTRY rgStdEntriesDLL[] =
{
{ rglpszInPlaceRegister, rglpszServerOverwriteDLL },
{ rglpszServerRegister, rglpszServerOverwriteDLL },
{ rglpszContainerRegister, rglpszContainerOverwrite },
{ rglpszDispatchRegister, rglpszDispatchOverwriteDLL },
{ rglpszDocObjectRegister, rglpszDocObjectOverwrite },
};
/////////////////////////////////////////////////////////////////////////////
// Special registration for apps that wish not to use REGLOAD
BOOL AFXAPI AfxOleRegisterServerClassCompat(
REFCLSID clsid, LPCTSTR lpszClassName,
LPCTSTR lpszShortTypeName, LPCTSTR lpszLongTypeName,
OLE_APPTYPE nAppType, LPCTSTR* rglpszRegister, LPCTSTR* rglpszOverwrite)
{
return AfxOleRegisterServerClass(clsid, lpszClassName, lpszShortTypeName,
lpszLongTypeName, nAppType, rglpszRegister, rglpszOverwrite);
}
AFX_STATIC BOOL AFXAPI _AfxOleMakeSymbolTable(CAfxOleSymbolTable& refTable,
REFCLSID clsid, LPCTSTR lpszClassName, LPCTSTR lpszShortTypeName,
LPCTSTR lpszLongTypeName, int nIconIndex,
LPCTSTR lpszFilterName, LPCTSTR lpszFilterExt)
{
// 0 - class ID
// 1 - class name
// 2 - SFN executable path
// 3 - short type name
// 4 - long type name
// 5 - long application name
// 6 - icon index
// 7 - Creator(xxxxxxxx) [Mac only]
// 8 - Creator(xxxxxxxx) [mac-only]
// 9 - Filter description
// convert the CLSID to a string
LPTSTR lpszClassID;
LPOLESTR lpOleStr;
::StringFromCLSID(clsid, &lpOleStr);
lpszClassID = TASKSTRINGOLE2T(lpOleStr);
if (lpszClassID == NULL)
{
TRACE0("Warning: StringFromCLSID failed in AfxOleRegisterServerName --\n");
TRACE0("\tperhaps AfxOleInit() has not been called.\n");
return FALSE;
}
refTable.SetAt(0, lpszClassID);
refTable.SetAt(1, lpszClassName);
// free memory for class ID
ASSERT(lpszClassID != NULL);
CoTaskMemFree(lpszClassID);
// get path name to server
CString strPathName;
AfxGetModuleShortFileName(AfxGetInstanceHandle(), strPathName);
refTable.SetAt(2, strPathName);
// fill in rest of symbols
refTable.SetAt(3, lpszShortTypeName);
refTable.SetAt(4, lpszLongTypeName);
refTable.SetAt(5, AfxGetAppName()); // will usually be long, readable name
CString strIconIndex;
if (nIconIndex != 0)
{
HICON hIcon = ::ExtractIcon(AfxGetInstanceHandle(), strPathName, nIconIndex);
if (hIcon != NULL)
DestroyIcon(hIcon);
else
nIconIndex = 0; // couldn't find specified icon so use default
}
strIconIndex.Format(_T("%d"), nIconIndex);
refTable.SetAt(6, strIconIndex);
refTable.SetAt(7, lpszFilterName);
CString strFileExtension;
if (lpszFilterExt != NULL && *lpszFilterExt != 0)
{
// use file extension provided
strFileExtension = lpszFilterExt;
}
else
{
// otherwise, try to find the extension from the description
// parse the actual extension (eg "*.TLA") from the
// filter name (eg, "Three Letter Acronym Files (*.TLA)")
strFileExtension = lpszFilterName;
int nBeginning = strFileExtension.Find('(');
if (nBeginning == -1)
strFileExtension.Empty();
else
{
strFileExtension = strFileExtension.Mid(1+nBeginning);
nBeginning = strFileExtension.Find('.');
if (nBeginning == -1)
strFileExtension.Empty();
else
{
strFileExtension = strFileExtension.Mid(nBeginning);
int nEnd = strFileExtension.Find(')');
if (nEnd == -1)
strFileExtension.Empty();
else
strFileExtension = strFileExtension.Left(nEnd);
}
}
}
refTable.SetAt(8, strFileExtension);
return TRUE;
}
BOOL AFXAPI AfxOleRegisterServerClass(
REFCLSID clsid, LPCTSTR lpszClassName, LPCTSTR lpszShortTypeName,
LPCTSTR lpszLongTypeName, OLE_APPTYPE nAppType, LPCTSTR* rglpszRegister,
LPCTSTR* rglpszOverwrite, int nIconIndex,
LPCTSTR lpszFilterName)
{
return AfxOleRegisterServerClass(clsid, lpszClassName, lpszShortTypeName,
lpszLongTypeName, nAppType, rglpszRegister, rglpszOverwrite, nIconIndex,
lpszFilterName, NULL);
}
BOOL AFXAPI AfxOleRegisterServerClass(
REFCLSID clsid, LPCTSTR lpszClassName, LPCTSTR lpszShortTypeName,
LPCTSTR lpszLongTypeName, OLE_APPTYPE nAppType, LPCTSTR* rglpszRegister,
LPCTSTR* rglpszOverwrite, int nIconIndex,
LPCTSTR lpszFilterName, LPCTSTR lpszFilterExt)
{
ASSERT(AfxIsValidString(lpszClassName));
ASSERT(AfxIsValidString(lpszShortTypeName));
ASSERT(*lpszShortTypeName != 0);
ASSERT(AfxIsValidString(lpszLongTypeName));
ASSERT(*lpszLongTypeName != 0);
ASSERT(nAppType == OAT_INPLACE_SERVER || nAppType == OAT_SERVER ||
nAppType == OAT_CONTAINER || nAppType == OAT_DISPATCH_OBJECT ||
nAppType == OAT_DOC_OBJECT_SERVER);
ASSERT(nAppType >= 0 && nAppType < _countof(rgStdEntries));
// use standard registration entries if non given
if (rglpszRegister == NULL)
rglpszRegister = (LPCTSTR*)rgStdEntries[nAppType].rglpszRegister;
if (rglpszOverwrite == NULL)
{
// DLL contexts have special strings
if (!afxContextIsDLL)
rglpszOverwrite = (LPCTSTR*)rgStdEntries[nAppType].rglpszOverwrite;
else
rglpszOverwrite = (LPCTSTR*)rgStdEntriesDLL[nAppType].rglpszOverwrite;
}
CAfxOleSymbolTable table(NUM_REG_VARS);
if (!_AfxOleMakeSymbolTable(table, clsid, lpszClassName,
lpszShortTypeName, lpszLongTypeName,
nIconIndex, lpszFilterName, lpszFilterExt))
{
return FALSE;
}
// protect against registering an invalid DocObject server
ASSERT(nAppType != OAT_DOC_OBJECT_SERVER ||
(lstrlen(table.GetAt(8)) != 0 && lstrcmp(table.GetAt(8), _T(".*")) != 0));
// update the registry with helper function
BOOL bResult;
bResult = AfxOleRegisterHelper(rglpszRegister, table.GetArray(),
NUM_REG_VARS, FALSE);
if (bResult && rglpszOverwrite != NULL)
{
bResult = AfxOleRegisterHelper(rglpszOverwrite, table.GetArray(),
NUM_REG_VARS, TRUE);
}
// free memory for class ID
return bResult;
}
BOOL AFXAPI AfxOleUnregisterServerClass(
REFCLSID clsid, LPCTSTR lpszClassName, LPCTSTR lpszShortTypeName,
LPCTSTR lpszLongTypeName, OLE_APPTYPE nAppType, LPCTSTR* rglpszRegister,
LPCTSTR* rglpszOverwrite)
{
// use standard registration entries if non given
if (rglpszRegister == NULL)
rglpszRegister = (LPCTSTR*)rgStdEntries[nAppType].rglpszRegister;
if (rglpszOverwrite == NULL)
{
// DLL contexts have special strings
if (!afxContextIsDLL)
rglpszOverwrite = (LPCTSTR*)rgStdEntries[nAppType].rglpszOverwrite;
else
rglpszOverwrite = (LPCTSTR*)rgStdEntriesDLL[nAppType].rglpszOverwrite;
}
CAfxOleSymbolTable table(NUM_REG_VARS);
if (!_AfxOleMakeSymbolTable(table, clsid, lpszClassName,
lpszShortTypeName, lpszLongTypeName, 0, NULL, NULL))
{
return FALSE;
}
// clean up the the registry with helper function
BOOL bResult;
bResult = AfxOleUnregisterHelper(rglpszRegister, table.GetArray(),
NUM_REG_VARS);
if (bResult && rglpszOverwrite != NULL)
{
bResult = AfxOleUnregisterHelper(rglpszOverwrite, table.GetArray(),
NUM_REG_VARS);
}
return bResult;
}
// removes key/value pairs from system registry
BOOL AFXAPI AfxOleUnregisterHelper(LPCTSTR const* rglpszRegister,
LPCTSTR const* rglpszSymbols, int nSymbols,
HKEY hKeyRoot /* = HKEY_CLASSES_ROOT */)
{
ASSERT(rglpszRegister != NULL);
ASSERT(nSymbols == 0 || rglpszSymbols != NULL);
CString strKey;
CString strValue;
// keeping a key open makes this go a bit faster
HKEY hKeyTemp = NULL;
if (hKeyRoot == HKEY_CLASSES_ROOT)
RegOpenKey(HKEY_CLASSES_ROOT, _T("CLSID"), &hKeyTemp);
BOOL bResult = TRUE;
while (*rglpszRegister != NULL)
{
LPCTSTR lpszKey = *rglpszRegister++;
if ((hKeyRoot == HKEY_CLASSES_ROOT) && (*lpszKey == '\0'))
continue;
AfxFormatStrings(strKey, lpszKey, rglpszSymbols, nSymbols);
if ((hKeyRoot == HKEY_CLASSES_ROOT) && strKey.IsEmpty())
{
TRACE1("Warning: skipping empty key '%s'.\n", lpszKey);
continue;
}
_AfxDeleteRegKey(strKey);
}
if (hKeyTemp != NULL)
RegCloseKey(hKeyTemp);
return bResult;
}
// writes key/value pairs to system registry
BOOL AFXAPI AfxOleRegisterHelper(LPCTSTR const* rglpszRegister,
LPCTSTR const* rglpszSymbols, int nSymbols, BOOL bReplace,
HKEY hKeyRoot /* = HKEY_CLASSES_ROOT */)
{
ASSERT(rglpszRegister != NULL);
ASSERT(nSymbols == 0 || rglpszSymbols != NULL);
CString strKey;
CString strValue;
// keeping a key open makes this go a bit faster
HKEY hKeyTemp = NULL;
if (hKeyRoot == HKEY_CLASSES_ROOT)
RegOpenKey(HKEY_CLASSES_ROOT, _T("CLSID"), &hKeyTemp);
BOOL bResult = TRUE;
while (*rglpszRegister != NULL)
{
LPCTSTR lpszKey = *rglpszRegister++;
if ((hKeyRoot == HKEY_CLASSES_ROOT) && (*lpszKey == '\0'))
continue;
LPCTSTR lpszValue = lpszKey + lstrlen(lpszKey) + 1;
AfxFormatStrings(strKey, lpszKey, rglpszSymbols, nSymbols);
AfxFormatStrings(strValue, lpszValue, rglpszSymbols, nSymbols);
if ((hKeyRoot == HKEY_CLASSES_ROOT) && strKey.IsEmpty())
{
TRACE1("Warning: skipping empty key '%s'.\n", lpszKey);
continue;
}
if (!bReplace)
{
TCHAR szBuffer[256];
LONG lSize = sizeof(szBuffer);
if (::RegQueryValue(hKeyRoot, strKey, szBuffer, &lSize) ==
ERROR_SUCCESS)
{
#ifdef _DEBUG
if (strValue != szBuffer)
{
TRACE2("Warning: Leaving value '%s' for key '%s' in registry\n",
szBuffer, (LPCTSTR)strKey);
TRACE1("\tintended value was '%s'.\n", (LPCTSTR)strValue);
}
#endif
continue;
}
}
if (::RegSetValue(hKeyRoot, strKey, REG_SZ, strValue, lstrlen(strValue) * sizeof(TCHAR))
!= ERROR_SUCCESS)
{
TRACE2("Error: failed setting key '%s' to value '%s'.\n",
(LPCTSTR)strKey, (LPCTSTR)strValue);
bResult = FALSE;
break;
}
}
if (hKeyTemp != NULL)
RegCloseKey(hKeyTemp);
return bResult;
}
/////////////////////////////////////////////////////////////////////////////