[4/4] xolehlp: rudimentary ITransactionDispenser implementation

[Daniel Jeliński]

-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://www.winehq.org/pipermail/wine-patches/attachments/20130430/14ba0b4f/attachment-0001.html>
-------------- next part --------------
From 36368ec75e12f7c01fe43ae7454f5614ea1aee5f Mon Sep 17 00:00:00 2001
From: Daniel Jelinski <djelinski1 at gmail.com>
Date: Tue, 30 Apr 2013 23:37:27 +0200
Subject: [PATCH 4/4] xolehlp: rudimentary ITransactionDispenser
 implementation

---
 dlls/xolehlp/Makefile.in |    1 +
 dlls/xolehlp/xolehlp.c   |  118 +++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 117 insertions(+), 2 deletions(-)

diff --git a/dlls/xolehlp/Makefile.in b/dlls/xolehlp/Makefile.in
index b1389ca..a297f71 100644
--- a/dlls/xolehlp/Makefile.in
+++ b/dlls/xolehlp/Makefile.in
@@ -1,4 +1,5 @@
 MODULE    = xolehlp.dll
+IMPORTS   = uuid
 
 C_SRCS = xolehlp.c
 
diff --git a/dlls/xolehlp/xolehlp.c b/dlls/xolehlp/xolehlp.c
index 475ef67..76565dd 100644
--- a/dlls/xolehlp/xolehlp.c
+++ b/dlls/xolehlp/xolehlp.c
@@ -21,8 +21,119 @@
 #include "winbase.h"
 #include "wine/debug.h"
 
+#define COBJMACROS
+
+#include "transact.h"
+
 WINE_DEFAULT_DEBUG_CHANNEL(xolehlp);
 
+typedef struct {
+    ITransactionDispenser ITransactionDispenser_iface;
+    LONG ref;
+} TransactionManager;
+
+static inline TransactionManager *impl_from_ITransactionDispenser(ITransactionDispenser *iface)
+{
+    return CONTAINING_RECORD(iface, TransactionManager, ITransactionDispenser_iface);
+}
+
+static HRESULT WINAPI TransactionDispenser_QueryInterface(ITransactionDispenser *iface, REFIID iid,
+    void **ppv)
+{
+    TransactionManager *This = impl_from_ITransactionDispenser(iface);
+    TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
+
+    if (!ppv) return E_INVALIDARG;
+
+    if (IsEqualIID(&IID_IUnknown, iid) ||
+        IsEqualIID(&IID_ITransactionDispenser, iid))
+    {
+        *ppv = &This->ITransactionDispenser_iface;
+    }
+    else
+    {
+        FIXME("(%s): not implemented\n", debugstr_guid(iid));
+        *ppv = NULL;
+        return E_NOINTERFACE;
+    }
+
+    IUnknown_AddRef((IUnknown*)*ppv);
+    return S_OK;
+}
+
+static ULONG WINAPI TransactionDispenser_AddRef(ITransactionDispenser *iface)
+{
+    TransactionManager *This = impl_from_ITransactionDispenser(iface);
+    ULONG ref = InterlockedIncrement(&This->ref);
+
+    TRACE("(%p) refcount=%u\n", iface, ref);
+
+    return ref;
+}
+
+static ULONG WINAPI TransactionDispenser_Release(ITransactionDispenser *iface)
+{
+    TransactionManager *This = impl_from_ITransactionDispenser(iface);
+    ULONG ref = InterlockedDecrement(&This->ref);
+
+    TRACE("(%p) refcount=%u\n", iface, ref);
+
+    if (ref == 0)
+    {
+        HeapFree(GetProcessHeap(), 0, This);
+    }
+
+    return ref;
+}
+
+static HRESULT WINAPI TransactionDispenser_GetOptionsObject(ITransactionDispenser *iface,
+        ITransactionOptions **ppOptions)
+{
+    FIXME("(%p, %p): stub\n", iface, ppOptions);
+    if (!ppOptions) return E_INVALIDARG;
+    *ppOptions = NULL;
+    return E_NOTIMPL;
+}
+static HRESULT WINAPI TransactionDispenser_BeginTransaction(ITransactionDispenser *iface,
+        IUnknown *punkOuter,
+        ISOLEVEL isoLevel,
+        ULONG isoFlags,
+        ITransactionOptions *pOptions,
+        ITransaction **ppTransaction)
+{
+    FIXME("(%p, %p, %08x, %08x, %p, %p): stub\n", iface, punkOuter,
+        isoLevel, isoFlags, pOptions, ppTransaction);
+
+    if (!ppTransaction) return E_INVALIDARG;
+    *ppTransaction = NULL;
+    if (punkOuter) return CLASS_E_NOAGGREGATION;
+    return E_NOTIMPL;
+}
+static const ITransactionDispenserVtbl TransactionDispenser_Vtbl = {
+    TransactionDispenser_QueryInterface,
+    TransactionDispenser_AddRef,
+    TransactionDispenser_Release,
+    TransactionDispenser_GetOptionsObject,
+    TransactionDispenser_BeginTransaction
+};
+
+static HRESULT TransactionManager_Create(REFIID riid, void **ppv)
+{
+    TransactionManager *This;
+    HRESULT ret;
+
+    This = HeapAlloc(GetProcessHeap(), 0, sizeof(TransactionManager));
+    if (!This) return E_OUTOFMEMORY;
+
+    This->ITransactionDispenser_iface.lpVtbl = &TransactionDispenser_Vtbl;
+    This->ref = 1;
+
+    ret = ITransactionDispenser_QueryInterface(&This->ITransactionDispenser_iface, riid, ppv);
+    ITransactionDispenser_Release(&This->ITransactionDispenser_iface);
+
+    return ret;
+}
+
 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
 {
     TRACE("%p, %u, %p\n", hinst, reason, reserved);
@@ -41,9 +152,12 @@ BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
 HRESULT CDECL DtcGetTransactionManager(char *host, char *tm_name, REFIID riid,
         DWORD dwReserved1, WORD wcbReserved2, void *pvReserved2, void **ppv)
 {
-    FIXME("(%s, %s, %s, %d, %d, %p, %p): stub\n", debugstr_a(host), debugstr_a(tm_name),
+    FIXME("(%s, %s, %s, %d, %d, %p, %p): semi-stub\n", debugstr_a(host), debugstr_a(tm_name),
           debugstr_guid(riid), dwReserved1, wcbReserved2, pvReserved2, ppv);
-    return E_NOTIMPL;
+
+    if (host)
+        return E_NOTIMPL;
+    return TransactionManager_Create(riid, ppv);
 }
 
 HRESULT CDECL DtcGetTransactionManagerExA(CHAR *host, CHAR *tm_name, REFIID riid,
-- 
1.7.5.4