[2/2] rpcrt4: Add tests for RPC and WIDL generated code
Dan Hipschman
dsh at linux.ucla.edu
Mon Apr 23 20:08:20 CDT 2007
This patch adds some basic tests for RPC and WIDL generated code.
---
dlls/rpcrt4/tests/Makefile.in | 6 +-
dlls/rpcrt4/tests/server.c | 235 +++++++++++++++++++++++++++++++++++++++++
dlls/rpcrt4/tests/server.idl | 45 ++++++++
3 files changed, 285 insertions(+), 1 deletions(-)
create mode 100644 dlls/rpcrt4/tests/server.c
create mode 100644 dlls/rpcrt4/tests/server.idl
diff --git a/dlls/rpcrt4/tests/Makefile.in b/dlls/rpcrt4/tests/Makefile.in
index 036cd21..7dd008c 100644
--- a/dlls/rpcrt4/tests/Makefile.in
+++ b/dlls/rpcrt4/tests/Makefile.in
@@ -5,12 +5,16 @@ VPATH = @srcdir@
TESTDLL = rpcrt4.dll
IMPORTS = ole32 rpcrt4 kernel32
EXTRALIBS = -luuid
+EXTRAIDLFLAGS = --server-prefix=s_
+IDL_C_SRCS = server.idl
+IDL_S_SRCS = server.idl
CTESTS = \
cstub.c \
generated.c \
ndr_marshall.c \
- rpc.c
+ rpc.c \
+ server.c
@MAKE_TEST_RULES@
diff --git a/dlls/rpcrt4/tests/server.c b/dlls/rpcrt4/tests/server.c
new file mode 100644
index 0000000..cf8d8a9
--- /dev/null
+++ b/dlls/rpcrt4/tests/server.c
@@ -0,0 +1,235 @@
+/*
+ * Tests for WIDL and RPC server/clients.
+ *
+ * Copyright (C) Google 2007 (Dan Hipschman)
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include "wine/test.h"
+#include "server.h"
+
+#include <stdio.h>
+
+#define PROTSEQ "ncacn_ip_tcp"
+#define ENDPOINT "4114"
+#define INT_CODE 4198
+
+static const char *progname;
+
+static HANDLE event;
+
+void __RPC_FAR *__RPC_USER
+midl_user_allocate(size_t n)
+{
+ return malloc(n);
+}
+
+void __RPC_USER
+midl_user_free(void __RPC_FAR *p)
+{
+ free(p);
+}
+
+int
+s_int_return(void)
+{
+ return INT_CODE;
+}
+
+int
+s_square(int x)
+{
+ return x * x;
+}
+
+int
+s_sum(int x, int y)
+{
+ return x + y;
+}
+
+void
+s_square_out(int x, int *y)
+{
+ *y = s_square(x);
+}
+
+void
+s_square_ref(int *x)
+{
+ *x = s_square(*x);
+}
+
+int
+s_str_length(const char *s)
+{
+ return strlen(s);
+}
+
+int
+s_dot_self(vector_t *v)
+{
+ return s_square(v->x) + s_square(v->y) + s_square(v->z);
+}
+
+double
+s_square_half(double x, double *y)
+{
+ *y = x / 2.0;
+ return x * x;
+}
+
+float
+s_square_half_float(float x, float *y)
+{
+ *y = x / 2.0f;
+ return x * x;
+}
+
+long
+s_square_half_long(long x, long *y)
+{
+ *y = x / 2;
+ return x * x;
+}
+
+void
+s_stop(void)
+{
+ ok(RPC_S_OK == RpcMgmtStopServerListening(NULL), "RpcMgmtStopServerListening\n");
+ ok(RPC_S_OK == RpcServerUnregisterIf(NULL, NULL, FALSE), "RpcServerUnregisterIf\n");
+ ok(SetEvent(event), "SetEvent\n");
+}
+
+static void
+make_cmdline(char buffer[MAX_PATH], const char *test)
+{
+ sprintf(buffer, "%s server %s", progname, test);
+}
+
+static int
+run_client(const char *test)
+{
+ char cmdline[MAX_PATH];
+ PROCESS_INFORMATION info;
+ STARTUPINFOA startup;
+ DWORD exitcode;
+
+ memset(&startup, 0, sizeof startup);
+ startup.cb = sizeof startup;
+
+ make_cmdline(cmdline, test);
+ ok(CreateProcessA(NULL, cmdline, NULL, NULL, FALSE, 0L, NULL, NULL, &startup, &info), "CreateProcess\n");
+ ok(WaitForSingleObject(info.hProcess, 30000) == WAIT_OBJECT_0, "Child process termination\n");
+ ok(GetExitCodeProcess(info.hProcess, &exitcode), "GetExitCodeProcess\n");
+ ok(CloseHandle(info.hProcess), "CloseHandle\n");
+ ok(CloseHandle(info.hThread), "CloseHandle\n");
+
+ return exitcode == 0;
+}
+
+static void
+client(const char *test)
+{
+ if (strcmp(test, "basic") == 0)
+ {
+ static unsigned char protseq[] = PROTSEQ;
+ static unsigned char endpoint[] = ENDPOINT;
+ static unsigned char address[] = "127.0.0.1";
+ unsigned char *binding;
+ static char string[] = "I am a string";
+ static vector_t a = {1, 3, 7};
+ double u, v;
+ float s, t;
+ long q, r;
+ int x;
+
+ ok(RPC_S_OK == RpcStringBindingCompose(NULL, protseq, address, endpoint, NULL, &binding), "RpcStringBindingCompose\n");
+ ok(RPC_S_OK == RpcBindingFromStringBinding(binding, &IServer_IfHandle), "RpcBindingFromStringBinding\n");
+
+ ok(int_return() == INT_CODE, "RPC int_return\n");
+
+ ok(square(7) == 49, "RPC square\n");
+ ok(sum(23, -4) == 19, "RPC sum\n");
+
+ x = 0;
+ square_out(11, &x);
+ ok(x == 121, "RPC square_out\n");
+
+ x = 5;
+ square_ref(&x);
+ ok(x == 25, "RPC square_ref\n");
+
+ ok(str_length(string) == strlen(string), "RPC str_length\n");
+ ok(dot_self(&a) == 59, "RPC dot_self\n");
+
+ v = 0.0;
+ u = square_half(3.0, &v);
+ ok(u == 9.0, "RPC square_half\n");
+ ok(v == 1.5, "RPC square_half\n");
+
+ t = 0.0f;
+ s = square_half_float(3.0f, &t);
+ ok(s == 9.0f, "RPC square_half_float\n");
+ ok(t == 1.5f, "RPC square_half_float\n");
+
+ r = 0;
+ q = square_half_long(3, &r);
+ ok(q == 9, "RPC square_half_long\n");
+ ok(r == 1, "RPC square_half_long\n");
+
+ stop();
+
+ ok(RPC_S_OK == RpcStringFree(&binding), "RpcStringFree\n");
+ ok(RPC_S_OK == RpcBindingFree(&IServer_IfHandle), "RpcBindingFree\n");
+ }
+}
+
+static void
+server(void)
+{
+ static unsigned char protseq[] = PROTSEQ;
+ static unsigned char endpoint[] = ENDPOINT;
+
+ ok(RPC_S_OK == RpcServerUseProtseqEp(protseq, 20, endpoint, NULL), "RpcServerUseProtseqEp\n");
+ ok(RPC_S_OK == RpcServerRegisterIf(IServer_v0_0_s_ifspec, NULL, NULL), "RpcServerRegisterIf\n");
+ ok(RPC_S_OK == RpcServerListen(1, 20, TRUE), "RpcServerListen\n");
+
+ event = CreateEvent(NULL, FALSE, FALSE, NULL);
+ ok(event != NULL, "CreateEvent failed\n");
+
+ ok(run_client("basic"), "basic client test failed\n");
+
+ ok(WAIT_OBJECT_0 == WaitForSingleObject(event, 60000), "WaitForSingleObject\n");
+ todo_wine {
+ ok(RPC_S_OK == RpcMgmtWaitServerListen(), "RpcMgmtWaitServerListening\n");
+ }
+}
+
+START_TEST(server)
+{
+ int argc;
+ char **argv;
+
+ argc = winetest_get_mainargs(&argv);
+ progname = argv[0];
+
+ if (argc == 3)
+ client(argv[2]);
+ else
+ server();
+}
+
diff --git a/dlls/rpcrt4/tests/server.idl b/dlls/rpcrt4/tests/server.idl
new file mode 100644
index 0000000..d42a3a9
--- /dev/null
+++ b/dlls/rpcrt4/tests/server.idl
@@ -0,0 +1,45 @@
+/*
+ * A simple interface to test the RPC server.
+ *
+ * Copyright (C) Google 2007 (Dan Hipschman)
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+typedef struct tag_vector
+{
+ int x;
+ int y;
+ int z;
+} vector_t;
+
+[
+ uuid(00000000-4114-0704-2301-000000000000),
+ implicit_handle(handle_t IServer_IfHandle)
+]
+interface IServer
+{
+ int int_return(void);
+ int square(int x);
+ int sum(int x, int y);
+ void square_out(int x, [out] int *y);
+ void square_ref([in, out] int *x);
+ int str_length([string] const char *s);
+ int dot_self(vector_t *v);
+ double square_half(double x, [out] double *y);
+ float square_half_float(float x, [out] float *y);
+ long square_half_long(long x, [out] long *y);
+ void stop(void);
+}
More information about the wine-patches
mailing list