* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
-
-#include <stdarg.h>
-
-#define COBJMACROS
-
-#include "windef.h"
-#include "winbase.h"
-#include "winuser.h"
-#include "winreg.h"
-#include "objbase.h"
-#include "oaidl.h"
-
-#define ATL_INITGUID
-#include "atliface.h"
-#include "atlbase.h"
-
-#include "wine/debug.h"
-#include "wine/unicode.h"
-
-WINE_DEFAULT_DEBUG_CHANNEL(atl);
-
-static LONG dll_count;
+#include <precomp.h>
/**************************************************************
* ATLRegistrar implementation
} rep_list;
typedef struct {
- const IRegistrarVtbl *lpVtbl;
+ IRegistrar IRegistrar_iface;
LONG ref;
rep_list *rep;
} Registrar;
DWORD len;
} strbuf;
+static inline Registrar *impl_from_IRegistrar(IRegistrar *iface)
+{
+ return CONTAINING_RECORD(iface, Registrar, IRegistrar_iface);
+}
+
static void strbuf_init(strbuf *buf)
{
buf->str = HeapAlloc(GetProcessHeap(), 0, 128*sizeof(WCHAR));
LONG lres;
HKEY hkey = 0;
strbuf name;
-
+
enum {
NORMAL,
NO_REMOVE,
IS_VAL,
FORCE_REMOVE,
DO_DELETE
- } key_type = NORMAL;
+ } key_type = NORMAL;
static const WCHAR wstrNoRemove[] = {'N','o','R','e','m','o','v','e',0};
static const WCHAR wstrForceRemove[] = {'F','o','r','c','e','R','e','m','o','v','e',0};
break;
}
TRACE("name = %s\n", debugstr_w(buf->str));
-
+
if(do_register) {
if(key_type == IS_VAL) {
hkey = parent_key;
}
break;
case 'd': {
- WCHAR *end;
DWORD dw;
- if(*iter == '0' && iter[1] == 'x') {
- iter += 2;
- dw = strtolW(iter, &end, 16);
- }else {
- dw = strtolW(iter, &end, 10);
- }
- iter = end;
+ hres = get_word(&iter, buf);
+ if(FAILED(hres))
+ break;
+ dw = atoiW(buf->str);
lres = RegSetValueExW(hkey, name.len ? name.str : NULL, 0, REG_DWORD,
(PBYTE)&dw, sizeof(dw));
if(lres != ERROR_SUCCESS) {
}
break;
}
+ case 'b': {
+ BYTE *bytes;
+ DWORD count;
+ DWORD i;
+ hres = get_word(&iter, buf);
+ if(FAILED(hres))
+ break;
+ count = (lstrlenW(buf->str) + 1) / 2;
+ bytes = HeapAlloc(GetProcessHeap(), 0, count);
+ if(bytes == NULL) {
+ hres = E_OUTOFMEMORY;
+ break;
+ }
+ for(i = 0; i < count && buf->str[2*i]; i++) {
+ WCHAR digits[3];
+ if(!isxdigitW(buf->str[2*i]) || !isxdigitW(buf->str[2*i + 1])) {
+ hres = E_FAIL;
+ break;
+ }
+ digits[0] = buf->str[2*i];
+ digits[1] = buf->str[2*i + 1];
+ digits[2] = 0;
+ bytes[i] = (BYTE) strtoulW(digits, NULL, 16);
+ }
+ if(SUCCEEDED(hres)) {
+ lres = RegSetValueExW(hkey, name.len ? name.str : NULL, 0, REG_BINARY,
+ bytes, count);
+ if(lres != ERROR_SUCCESS) {
+ WARN("Could not set value of key: 0x%08x\n", lres);
+ hres = HRESULT_FROM_WIN32(lres);
+ }
+ }
+ HeapFree(GetProcessHeap(), 0, bytes);
+ break;
+ }
default:
WARN("Wrong resource type: %s\n", debugstr_w(buf->str));
hres = DISP_E_EXCEPTION;
RegCloseKey(hkey);
hkey = 0;
name.len = 0;
-
+
hres = get_word(&iter, buf);
if(FAILED(hres))
break;
DWORD filelen, len;
LPWSTR regstrw;
LPSTR regstra;
- LRESULT lres;
HRESULT hres;
- file = CreateFileW(fileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
+ file = CreateFileW(fileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
if(file != INVALID_HANDLE_VALUE) {
filelen = GetFileSize(file, NULL);
regstra = HeapAlloc(GetProcessHeap(), 0, filelen);
- lres = ReadFile(file, regstra, filelen, NULL, NULL);
- if(lres == ERROR_SUCCESS) {
+ if(ReadFile(file, regstra, filelen, NULL, NULL)) {
len = MultiByteToWideChar(CP_ACP, 0, regstra, filelen, NULL, 0)+1;
regstrw = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len*sizeof(WCHAR));
MultiByteToWideChar(CP_ACP, 0, regstra, filelen, regstrw, len);
regstrw[len-1] = '\0';
-
+
hres = string_register(This, regstrw, do_register);
HeapFree(GetProcessHeap(), 0, regstrw);
}else {
- WARN("Failed to read faile\n");
- hres = HRESULT_FROM_WIN32(lres);
+ WARN("Failed to read file %s\n", debugstr_w(fileName));
+ hres = HRESULT_FROM_WIN32(GetLastError());
}
HeapFree(GetProcessHeap(), 0, regstra);
CloseHandle(file);
}else {
- WARN("Could not open file\n");
+ WARN("Could not open file %s\n", debugstr_w(fileName));
hres = HRESULT_FROM_WIN32(GetLastError());
}
{
TRACE("(%p)->(%s %p\n", iface, debugstr_guid(riid), ppvObject);
- if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IRegistrar, riid)) {
+ if(IsEqualGUID(&IID_IUnknown, riid)
+ || IsEqualGUID(&IID_IRegistrar, riid)
+ || IsEqualGUID(&IID_IRegistrarBase, riid)) {
IRegistrar_AddRef(iface);
*ppvObject = iface;
return S_OK;
static ULONG WINAPI Registrar_AddRef(IRegistrar *iface)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
ULONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) ->%d\n", This, ref);
return ref;
static ULONG WINAPI Registrar_Release(IRegistrar *iface)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
ULONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) ->%d\n", This, ref);
if(!ref) {
IRegistrar_ClearReplacements(iface);
HeapFree(GetProcessHeap(), 0, This);
- InterlockedDecrement(&dll_count);
}
return ref;
}
static HRESULT WINAPI Registrar_AddReplacement(IRegistrar *iface, LPCOLESTR Key, LPCOLESTR item)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
int len;
rep_list *new_rep;
new_rep->next = This->rep;
This->rep = new_rep;
-
+
return S_OK;
}
static HRESULT WINAPI Registrar_ClearReplacements(IRegistrar *iface)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
rep_list *iter, *iter2;
TRACE("(%p)\n", This);
static HRESULT WINAPI Registrar_ResourceRegisterSz(IRegistrar* iface, LPCOLESTR resFileName,
LPCOLESTR szID, LPCOLESTR szType)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s %s %s)\n", This, debugstr_w(resFileName), debugstr_w(szID), debugstr_w(szType));
return resource_register(This, resFileName, szID, szType, TRUE);
}
static HRESULT WINAPI Registrar_ResourceUnregisterSz(IRegistrar* iface, LPCOLESTR resFileName,
LPCOLESTR szID, LPCOLESTR szType)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s %s %s)\n", This, debugstr_w(resFileName), debugstr_w(szID), debugstr_w(szType));
return resource_register(This, resFileName, szID, szType, FALSE);
}
static HRESULT WINAPI Registrar_FileRegister(IRegistrar* iface, LPCOLESTR fileName)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s)\n", This, debugstr_w(fileName));
return file_register(This, fileName, TRUE);
}
static HRESULT WINAPI Registrar_FileUnregister(IRegistrar* iface, LPCOLESTR fileName)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
FIXME("(%p)->(%s)\n", This, debugstr_w(fileName));
return file_register(This, fileName, FALSE);
}
static HRESULT WINAPI Registrar_StringRegister(IRegistrar* iface, LPCOLESTR data)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s)\n", This, debugstr_w(data));
return string_register(This, data, TRUE);
}
static HRESULT WINAPI Registrar_StringUnregister(IRegistrar* iface, LPCOLESTR data)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s)\n", This, debugstr_w(data));
return string_register(This, data, FALSE);
}
static HRESULT WINAPI Registrar_ResourceRegister(IRegistrar* iface, LPCOLESTR resFileName,
UINT nID, LPCOLESTR szType)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s %d %s)\n", iface, debugstr_w(resFileName), nID, debugstr_w(szType));
return resource_register(This, resFileName, MAKEINTRESOURCEW(nID), szType, TRUE);
}
static HRESULT WINAPI Registrar_ResourceUnregister(IRegistrar* iface, LPCOLESTR resFileName,
UINT nID, LPCOLESTR szType)
{
- Registrar *This = (Registrar*)iface;
+ Registrar *This = impl_from_IRegistrar(iface);
TRACE("(%p)->(%s %d %s)\n", This, debugstr_w(resFileName), nID, debugstr_w(szType));
return resource_register(This, resFileName, MAKEINTRESOURCEW(nID), szType, FALSE);
}
Registrar_ResourceUnregister,
};
-static HRESULT Registrar_create(const IUnknown *pUnkOuter, REFIID riid, void **ppvObject)
-{
- Registrar *ret;
-
- if(!IsEqualGUID(&IID_IUnknown, riid) && !IsEqualGUID(&IID_IRegistrar, riid))
- return E_NOINTERFACE;
-
- ret = HeapAlloc(GetProcessHeap(), 0, sizeof(Registrar));
- ret->lpVtbl = &RegistrarVtbl;
- ret->ref = 1;
- ret->rep = NULL;
- *ppvObject = ret;
-
- InterlockedIncrement(&dll_count);
-
- return S_OK;
-}
-
-/**************************************************************
- * ClassFactory implementation
+/***********************************************************************
+ * AtlCreateRegistrar [atl100.@]
*/
-
-static HRESULT WINAPI RegistrarCF_QueryInterface(IClassFactory *iface, REFIID riid, void **ppvObject)
-{
- TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);
-
- if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
- *ppvObject = iface;
- IClassFactory_AddRef( iface );
- return S_OK;
- }
-
- return E_NOINTERFACE;
-}
-
-static ULONG WINAPI RegistrarCF_AddRef(IClassFactory *iface)
-{
- InterlockedIncrement(&dll_count);
- return 2;
-}
-
-static ULONG WINAPI RegistrarCF_Release(IClassFactory *iface)
-{
- InterlockedDecrement(&dll_count);
- return 1;
-}
-
-static HRESULT WINAPI RegistrarCF_CreateInstance(IClassFactory *iface, LPUNKNOWN pUnkOuter,
- REFIID riid, void **ppvObject)
+HRESULT WINAPI AtlCreateRegistrar(IRegistrar **ret)
{
- TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);
- return Registrar_create(pUnkOuter, riid, ppvObject);
-}
+ Registrar *registrar;
-static HRESULT WINAPI RegistrarCF_LockServer(IClassFactory *iface, BOOL lock)
-{
- TRACE("(%p)->(%x)\n", iface, lock);
+ registrar = HeapAlloc(GetProcessHeap(), 0, sizeof(*registrar));
+ if(!registrar)
+ return E_OUTOFMEMORY;
- if(lock)
- InterlockedIncrement(&dll_count);
- else
- InterlockedDecrement(&dll_count);
+ registrar->IRegistrar_iface.lpVtbl = &RegistrarVtbl;
+ registrar->ref = 1;
+ registrar->rep = NULL;
+ *ret = ®istrar->IRegistrar_iface;
return S_OK;
}
-static const IClassFactoryVtbl IRegistrarCFVtbl = {
- RegistrarCF_QueryInterface,
- RegistrarCF_AddRef,
- RegistrarCF_Release,
- RegistrarCF_CreateInstance,
- RegistrarCF_LockServer
-};
-
-static IClassFactory RegistrarCF = { &IRegistrarCFVtbl };
-
-/**************************************************************
- * DllGetClassObject (ATL.2)
+/***********************************************************************
+ * AtlUpdateRegistryFromResourceD [atl100.@]
*/
-HRESULT WINAPI DllGetClassObject(REFCLSID clsid, REFIID riid, LPVOID *ppvObject)
-{
- TRACE("(%s %s %p)\n", debugstr_guid(clsid), debugstr_guid(riid), ppvObject);
-
- if(IsEqualGUID(&CLSID_Registrar, clsid))
- return IClassFactory_QueryInterface( &RegistrarCF, riid, ppvObject );
-
- FIXME("Not supported class %s\n", debugstr_guid(clsid));
- return CLASS_E_CLASSNOTAVAILABLE;
-}
-
-extern HINSTANCE hInst;
-
-static HRESULT do_register_dll_server(IRegistrar *pRegistrar, LPCOLESTR wszDll,
- LPCOLESTR wszId, BOOL do_register,
- const struct _ATL_REGMAP_ENTRY* pMapEntries)
+HRESULT WINAPI AtlUpdateRegistryFromResourceD(HINSTANCE inst, LPCOLESTR res,
+ BOOL bRegister, struct _ATL_REGMAP_ENTRY *pMapEntries, IRegistrar *pReg)
{
+ const struct _ATL_REGMAP_ENTRY *iter;
+ WCHAR module_name[MAX_PATH];
IRegistrar *registrar;
HRESULT hres;
- const struct _ATL_REGMAP_ENTRY *pMapEntry;
-
- static const WCHAR wszModule[] = {'M','O','D','U','L','E',0};
- static const WCHAR wszRegistry[] = {'R','E','G','I','S','T','R','Y',0};
-
- if (pRegistrar)
- registrar = pRegistrar;
- else
- Registrar_create(NULL, &IID_IRegistrar, (void**)®istrar);
-
- IRegistrar_AddReplacement(registrar, wszModule, wszDll);
-
- for (pMapEntry = pMapEntries; pMapEntry && pMapEntry->szKey; pMapEntry++)
- IRegistrar_AddReplacement(registrar, pMapEntry->szKey, pMapEntry->szData);
-
- if(do_register)
- hres = IRegistrar_ResourceRegisterSz(registrar, wszDll, wszId, wszRegistry);
- else
- hres = IRegistrar_ResourceUnregisterSz(registrar, wszDll, wszId, wszRegistry);
-
- if(registrar != pRegistrar)
- IRegistrar_Release(registrar);
- return hres;
-}
-
-static HRESULT do_register_server(BOOL do_register)
-{
- static const WCHAR CLSID_RegistrarW[] =
- {'C','L','S','I','D','_','R','e','g','i','s','t','r','a','r',0};
- static const WCHAR atl_dllW[] = {'a','t','l','.','d','l','l',0};
- WCHAR clsid_str[40];
- const struct _ATL_REGMAP_ENTRY reg_map[] = {{CLSID_RegistrarW, clsid_str}, {NULL,NULL}};
+ static const WCHAR moduleW[] = {'M','O','D','U','L','E',0};
+ static const WCHAR registryW[] = {'R','E','G','I','S','T','R','Y',0};
- StringFromGUID2(&CLSID_Registrar, clsid_str, sizeof(clsid_str)/sizeof(WCHAR));
- return do_register_dll_server(NULL, atl_dllW, MAKEINTRESOURCEW(101), do_register, reg_map);
-}
-
-/***********************************************************************
- * AtlModuleUpdateRegistryFromResourceD [ATL.@]
- *
- */
-HRESULT WINAPI AtlModuleUpdateRegistryFromResourceD(_ATL_MODULEW* pM, LPCOLESTR lpszRes,
- BOOL bRegister, struct _ATL_REGMAP_ENTRY* pMapEntries, IRegistrar* pReg)
-{
- HINSTANCE lhInst = pM->m_hInst;
- /* everything inside this function below this point
- * should go into atl71.AtlUpdateRegistryFromResourceD
- */
- WCHAR module_name[MAX_PATH];
-
- if(!GetModuleFileNameW(lhInst, module_name, MAX_PATH)) {
- FIXME("hinst %p: did not get module name\n",
- lhInst);
+ if(!GetModuleFileNameW(inst, module_name, MAX_PATH)) {
+ FIXME("hinst %p: did not get module name\n", inst);
return E_FAIL;
}
- TRACE("%p (%s), %s, %d, %p, %p\n", hInst, debugstr_w(module_name),
- debugstr_w(lpszRes), bRegister, pMapEntries, pReg);
+ TRACE("%p (%s), %s, %d, %p, %p\n", inst, debugstr_w(module_name),
+ debugstr_w(res), bRegister, pMapEntries, pReg);
- return do_register_dll_server(pReg, module_name, lpszRes, bRegister, pMapEntries);
-}
+ if(pReg) {
+ registrar = pReg;
+ }else {
+ hres = AtlCreateRegistrar(®istrar);
+ if(FAILED(hres))
+ return hres;
+ }
-/***********************************************************************
- * DllRegisterServer (ATL.@)
- */
-HRESULT WINAPI DllRegisterServer(void)
-{
- TRACE("\n");
- return do_register_server(TRUE);
-}
+ IRegistrar_AddReplacement(registrar, moduleW, module_name);
-/***********************************************************************
- * DllRegisterServer (ATL.@)
- */
-HRESULT WINAPI DllUnregisterServer(void)
-{
- TRACE("\n");
- return do_register_server(FALSE);
-}
+ for (iter = pMapEntries; iter && iter->szKey; iter++)
+ IRegistrar_AddReplacement(registrar, iter->szKey, iter->szData);
-/***********************************************************************
- * DllCanUnloadNow (ATL.@)
- */
-HRESULT WINAPI DllCanUnloadNow(void)
-{
- TRACE("dll_count = %u\n", dll_count);
- return dll_count ? S_FALSE : S_OK;
+ if(bRegister)
+ hres = IRegistrar_ResourceRegisterSz(registrar, module_name, res, registryW);
+ else
+ hres = IRegistrar_ResourceUnregisterSz(registrar, module_name, res, registryW);
+
+ if(registrar != pReg)
+ IRegistrar_Release(registrar);
+ return hres;
}
projects / reactos.git / blobdiff