| /* |
| * Copyright 2011 Joakim Sindholt <opensource@zhasha.com> |
| * |
| * Permission is hereby granted, free of charge, to any person obtaining a |
| * copy of this software and associated documentation files (the "Software"), |
| * to deal in the Software without restriction, including without limitation |
| * on the rights to use, copy, modify, merge, publish, distribute, sub |
| * license, and/or sell copies of the Software, and to permit persons to whom |
| * the Software is furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice (including the next |
| * paragraph) shall be included in all copies or substantial portions of the |
| * Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL |
| * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, |
| * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR |
| * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE |
| * USE OR OTHER DEALINGS IN THE SOFTWARE. */ |
| |
| #include "iunknown.h" |
| #include "util/u_atomic.h" |
| #include "util/u_hash_table.h" |
| |
| #include "nine_helpers.h" |
| #include "nine_pdata.h" |
| |
| #define DBG_CHANNEL DBG_UNKNOWN |
| |
| HRESULT |
| NineUnknown_ctor( struct NineUnknown *This, |
| struct NineUnknownParams *pParams ) |
| { |
| This->refs = pParams->container ? 0 : 1; |
| This->bind = 0; |
| This->forward = !This->refs; |
| This->container = pParams->container; |
| This->device = pParams->device; |
| if (This->refs && This->device) |
| NineUnknown_AddRef(NineUnknown(This->device)); |
| |
| This->vtable = pParams->vtable; |
| This->vtable_internal = pParams->vtable; |
| This->guids = pParams->guids; |
| This->dtor = pParams->dtor; |
| |
| This->pdata = util_hash_table_create(ht_guid_hash, ht_guid_compare); |
| if (!This->pdata) |
| return E_OUTOFMEMORY; |
| |
| return D3D_OK; |
| } |
| |
| void |
| NineUnknown_dtor( struct NineUnknown *This ) |
| { |
| if (This->refs && This->device) /* Possible only if early exit after a ctor failed */ |
| (void) NineUnknown_Release(NineUnknown(This->device)); |
| |
| if (This->pdata) { |
| util_hash_table_foreach(This->pdata, ht_guid_delete, NULL); |
| util_hash_table_destroy(This->pdata); |
| } |
| |
| FREE(This); |
| } |
| |
| HRESULT NINE_WINAPI |
| NineUnknown_QueryInterface( struct NineUnknown *This, |
| REFIID riid, |
| void **ppvObject ) |
| { |
| unsigned i = 0; |
| char guid_str[64]; |
| |
| DBG("This=%p riid=%p id=%s ppvObject=%p\n", |
| This, riid, riid ? GUID_sprintf(guid_str, riid) : "", ppvObject); |
| |
| (void)guid_str; |
| |
| if (!ppvObject) return E_POINTER; |
| |
| do { |
| if (GUID_equal(This->guids[i], riid)) { |
| *ppvObject = This; |
| /* Tests showed that this call succeeds even on objects with |
| * zero refcount. This can happen if the app released all references |
| * but the resource is still bound. |
| */ |
| NineUnknown_AddRef(This); |
| return S_OK; |
| } |
| } while (This->guids[++i]); |
| |
| *ppvObject = NULL; |
| return E_NOINTERFACE; |
| } |
| |
| ULONG NINE_WINAPI |
| NineUnknown_AddRef( struct NineUnknown *This ) |
| { |
| ULONG r; |
| if (This->forward) |
| return NineUnknown_AddRef(This->container); |
| else |
| r = p_atomic_inc_return(&This->refs); |
| |
| if (r == 1) { |
| if (This->device) |
| NineUnknown_AddRef(NineUnknown(This->device)); |
| /* This shouldn't be necessary: |
| if (This->container) |
| NineUnknown_Bind(NineUnknown(This->container)); */ |
| } |
| return r; |
| } |
| |
| ULONG NINE_WINAPI |
| NineUnknown_Release( struct NineUnknown *This ) |
| { |
| if (This->forward) |
| return NineUnknown_Release(This->container); |
| |
| ULONG r = p_atomic_dec_return(&This->refs); |
| |
| if (r == 0) { |
| if (This->device) { |
| if (NineUnknown_Release(NineUnknown(This->device)) == 0) |
| return r; /* everything's gone */ |
| } |
| if (This->container) { |
| /* NineUnknown_Unbind(NineUnknown(This->container)); */ |
| } else |
| if (This->bind == 0) { |
| This->dtor(This); |
| } |
| } |
| return r; |
| } |
| |
| HRESULT NINE_WINAPI |
| NineUnknown_GetDevice( struct NineUnknown *This, |
| IDirect3DDevice9 **ppDevice ) |
| { |
| user_assert(ppDevice, E_POINTER); |
| NineUnknown_AddRef(NineUnknown(This->device)); |
| *ppDevice = (IDirect3DDevice9 *)This->device; |
| return D3D_OK; |
| } |
| |
| HRESULT NINE_WINAPI |
| NineUnknown_SetPrivateData( struct NineUnknown *This, |
| REFGUID refguid, |
| const void *pData, |
| DWORD SizeOfData, |
| DWORD Flags ) |
| { |
| enum pipe_error err; |
| struct pheader *header; |
| const void *user_data = pData; |
| char guid_str[64]; |
| void *header_data; |
| |
| DBG("This=%p GUID=%s pData=%p SizeOfData=%u Flags=%x\n", |
| This, GUID_sprintf(guid_str, refguid), pData, SizeOfData, Flags); |
| |
| (void)guid_str; |
| |
| if (Flags & D3DSPD_IUNKNOWN) |
| user_assert(SizeOfData == sizeof(IUnknown *), D3DERR_INVALIDCALL); |
| |
| /* data consists of a header and the actual data. avoiding 2 mallocs */ |
| header = CALLOC_VARIANT_LENGTH_STRUCT(pheader, SizeOfData); |
| if (!header) { return E_OUTOFMEMORY; } |
| header->unknown = (Flags & D3DSPD_IUNKNOWN) ? TRUE : FALSE; |
| |
| /* if the refguid already exists, delete it */ |
| NineUnknown_FreePrivateData(This, refguid); |
| |
| /* IUnknown special case */ |
| if (header->unknown) { |
| /* here the pointer doesn't point to the data we want, so point at the |
| * pointer making what we eventually copy is the pointer itself */ |
| user_data = &pData; |
| } |
| |
| header->size = SizeOfData; |
| header_data = (void *)header + sizeof(*header); |
| memcpy(header_data, user_data, header->size); |
| memcpy(&header->guid, refguid, sizeof(header->guid)); |
| |
| err = util_hash_table_set(This->pdata, &header->guid, header); |
| if (err == PIPE_OK) { |
| if (header->unknown) { IUnknown_AddRef(*(IUnknown **)header_data); } |
| return D3D_OK; |
| } |
| |
| FREE(header); |
| if (err == PIPE_ERROR_OUT_OF_MEMORY) { return E_OUTOFMEMORY; } |
| |
| return D3DERR_DRIVERINTERNALERROR; |
| } |
| |
| HRESULT NINE_WINAPI |
| NineUnknown_GetPrivateData( struct NineUnknown *This, |
| REFGUID refguid, |
| void *pData, |
| DWORD *pSizeOfData ) |
| { |
| struct pheader *header; |
| DWORD sizeofdata; |
| char guid_str[64]; |
| void *header_data; |
| |
| DBG("This=%p GUID=%s pData=%p pSizeOfData=%p\n", |
| This, GUID_sprintf(guid_str, refguid), pData, pSizeOfData); |
| |
| (void)guid_str; |
| |
| header = util_hash_table_get(This->pdata, refguid); |
| if (!header) { return D3DERR_NOTFOUND; } |
| |
| user_assert(pSizeOfData, E_POINTER); |
| sizeofdata = *pSizeOfData; |
| *pSizeOfData = header->size; |
| |
| if (!pData) { |
| return D3D_OK; |
| } |
| if (sizeofdata < header->size) { |
| return D3DERR_MOREDATA; |
| } |
| |
| header_data = (void *)header + sizeof(*header); |
| if (header->unknown) { IUnknown_AddRef(*(IUnknown **)header_data); } |
| memcpy(pData, header_data, header->size); |
| |
| return D3D_OK; |
| } |
| |
| HRESULT NINE_WINAPI |
| NineUnknown_FreePrivateData( struct NineUnknown *This, |
| REFGUID refguid ) |
| { |
| struct pheader *header; |
| char guid_str[64]; |
| |
| DBG("This=%p GUID=%s\n", This, GUID_sprintf(guid_str, refguid)); |
| |
| (void)guid_str; |
| |
| header = util_hash_table_get(This->pdata, refguid); |
| if (!header) |
| return D3DERR_NOTFOUND; |
| |
| ht_guid_delete(NULL, header, NULL); |
| util_hash_table_remove(This->pdata, refguid); |
| |
| return D3D_OK; |
| } |