dlls/rpcrt4/tests/rpc_async.c

   1 /*
   2  * Unit test suite for rpc functions
   3  *
   4  * Copyright 2008 Robert Shearman (for CodeWeavers)
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "wine/test.h"
  24
  25 #include <rpc.h>
  26 #include <rpcasync.h>
  27
  28 static RPC_STATUS (RPC_ENTRY *pRpcAsyncInitializeHandle)(PRPC_ASYNC_STATE,unsigned int);
  29 static RPC_STATUS (RPC_ENTRY *pRpcAsyncGetCallStatus)(PRPC_ASYNC_STATE);
  30
  31 static void test_RpcAsyncInitializeHandle(void)
  32 {
  33     char buffer[256];
  34     RPC_ASYNC_STATE async;
  35     RPC_STATUS status;
  36     int i;
  37     void *unset_ptr;
  38
  39     status = pRpcAsyncInitializeHandle((PRPC_ASYNC_STATE)buffer, sizeof(buffer));
  40     ok(status == ERROR_INVALID_PARAMETER, "RpcAsyncInitializeHandle with large Size should have returned ERROR_INVALID_PARAMETER instead of %d\n", status);
  41
  42     status = pRpcAsyncInitializeHandle(&async, sizeof(async) - 1);
  43     ok(status == ERROR_INVALID_PARAMETER, "RpcAsyncInitializeHandle with small Size should have returned ERROR_INVALID_PARAMETER instead of %d\n", status);
  44
  45     memset(&async, 0xcc, sizeof(async));
  46     memset(&unset_ptr, 0xcc, sizeof(unset_ptr));
  47     status = pRpcAsyncInitializeHandle(&async, sizeof(async));
  48     ok(status == RPC_S_OK, "RpcAsyncInitializeHandle failed with error %d\n", status);
  49
  50     ok(async.Size == sizeof(async), "async.Size wrong: %d\n", async.Size);
  51     ok(async.Signature == 0x43595341, "async.Signature should be 0x43595341, but is 0x%x instead\n", async.Signature);
  52     ok(async.Lock == 0, "async.Lock should be 0, but is %d instead\n", async.Lock);
  53     ok(async.Flags == 0, "async.Flags should be 0, but is %d instead\n", async.Flags);
  54     ok(async.StubInfo == NULL, "async.StubInfo should be NULL, not %p\n", async.StubInfo);
  55     ok(async.UserInfo == unset_ptr, "async.UserInfo should be unset, not %p\n", async.UserInfo);
  56     ok(async.RuntimeInfo == NULL, "async.RuntimeInfo should be NULL, not %p\n", async.RuntimeInfo);
  57     ok(async.Event == 0xcccccccc, "async.Event should be unset, not %d\n", async.Event);
  58     ok(async.NotificationType == 0xcccccccc, "async.NotificationType should be unset, not %d\n", async.NotificationType);
  59     for (i = 0; i < 4; i++)
  60         ok(async.Reserved[i] == 0x0, "async.Reserved[%d] should be 0x0, not 0x%lx\n", i, async.Reserved[i]);
  61 }
  62
  63 static void test_RpcAsyncGetCallStatus(void)
  64 {
  65     RPC_ASYNC_STATE async;
  66     RPC_STATUS status;
  67
  68     status = pRpcAsyncInitializeHandle(&async, sizeof(async));
  69     ok(status == RPC_S_OK, "RpcAsyncInitializeHandle failed with error %d\n", status);
  70
  71     status = pRpcAsyncGetCallStatus(&async);
  72     todo_wine
  73     ok(status == RPC_S_INVALID_BINDING, "RpcAsyncGetCallStatus should have returned RPC_S_INVALID_BINDING instead of %d\n", status);
  74
  75     memset(&async, 0, sizeof(async));
  76     status = pRpcAsyncGetCallStatus(&async);
  77     todo_wine
  78     ok(status == RPC_S_INVALID_BINDING, "RpcAsyncGetCallStatus should have returned RPC_S_INVALID_BINDING instead of %d\n", status);
  79 }
  80
  81 START_TEST( rpc_async )
  82 {
  83     HMODULE hRpcRt4 = GetModuleHandleA("rpcrt4.dll");
  84     pRpcAsyncInitializeHandle = (void *)GetProcAddress(hRpcRt4, "RpcAsyncInitializeHandle");
  85     pRpcAsyncGetCallStatus = (void *)GetProcAddress(hRpcRt4, "RpcAsyncGetCallStatus");
  86     if (!pRpcAsyncInitializeHandle || !pRpcAsyncGetCallStatus)
  87     {
  88         win_skip("asynchronous functions not available\n");
  89         return;
  90     }
  91     test_RpcAsyncInitializeHandle();
  92     test_RpcAsyncGetCallStatus();
  93 }