2007-04-26 03:18:29 +02:00
|
|
|
/*
|
|
|
|
* 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 <windows.h>
|
|
|
|
#include "wine/test.h"
|
|
|
|
#include "server.h"
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
|
|
|
|
#define PORT "4114"
|
|
|
|
#define PIPE "\\pipe\\wine_rpcrt4_test"
|
|
|
|
|
|
|
|
#define INT_CODE 4198
|
|
|
|
|
|
|
|
static const char *progname;
|
|
|
|
|
|
|
|
static HANDLE stop_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;
|
|
|
|
}
|
|
|
|
|
2007-04-26 03:22:10 +02:00
|
|
|
int
|
|
|
|
s_sum_fixed_array(int a[5])
|
|
|
|
{
|
|
|
|
return a[0] + a[1] + a[2] + a[3] + a[4];
|
|
|
|
}
|
|
|
|
|
2007-05-10 05:50:49 +02:00
|
|
|
int
|
|
|
|
s_pints_sum(pints_t *pints)
|
|
|
|
{
|
|
|
|
return *pints->pi + **pints->ppi + ***pints->pppi;
|
|
|
|
}
|
|
|
|
|
|
|
|
double
|
|
|
|
s_ptypes_sum(ptypes_t *pt)
|
|
|
|
{
|
|
|
|
return *pt->pc + *pt->ps + *pt->pl + *pt->pf + *pt->pd;
|
|
|
|
}
|
|
|
|
|
2007-05-12 02:27:40 +02:00
|
|
|
int
|
|
|
|
s_dot_pvectors(pvectors_t *p)
|
|
|
|
{
|
|
|
|
return p->pu->x * (*p->pv)->x + p->pu->y * (*p->pv)->y + p->pu->z * (*p->pv)->z;
|
|
|
|
}
|
|
|
|
|
2007-04-26 03:18:29 +02:00
|
|
|
void
|
|
|
|
s_stop(void)
|
|
|
|
{
|
|
|
|
ok(RPC_S_OK == RpcMgmtStopServerListening(NULL), "RpcMgmtStopServerListening\n");
|
|
|
|
ok(RPC_S_OK == RpcServerUnregisterIf(NULL, NULL, FALSE), "RpcServerUnregisterIf\n");
|
|
|
|
ok(SetEvent(stop_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
|
|
|
|
basic_tests(void)
|
|
|
|
{
|
|
|
|
static char string[] = "I am a string";
|
2007-04-26 03:22:10 +02:00
|
|
|
static int f[5] = {1, 3, 0, -2, -4};
|
2007-04-26 03:18:29 +02:00
|
|
|
static vector_t a = {1, 3, 7};
|
2007-05-12 02:27:40 +02:00
|
|
|
static vector_t vec1 = {4, -2, 1}, vec2 = {-5, 2, 3}, *pvec2 = &vec2;
|
|
|
|
static pvectors_t pvecs = {&vec1, &pvec2};
|
2007-05-10 05:50:49 +02:00
|
|
|
pints_t pints;
|
|
|
|
ptypes_t ptypes;
|
|
|
|
int i1, i2, i3, *pi2, *pi3, **ppi3;
|
2007-04-26 03:18:29 +02:00
|
|
|
double u, v;
|
|
|
|
float s, t;
|
|
|
|
long q, r;
|
2007-05-10 05:50:49 +02:00
|
|
|
short h;
|
|
|
|
char c;
|
2007-04-26 03:18:29 +02:00
|
|
|
int x;
|
|
|
|
|
|
|
|
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");
|
2007-04-26 03:22:10 +02:00
|
|
|
|
2007-05-10 05:50:49 +02:00
|
|
|
i1 = 19;
|
|
|
|
i2 = -3;
|
|
|
|
i3 = -29;
|
|
|
|
pi2 = &i2;
|
|
|
|
pi3 = &i3;
|
|
|
|
ppi3 = &pi3;
|
|
|
|
pints.pi = &i1;
|
|
|
|
pints.ppi = &pi2;
|
|
|
|
pints.pppi = &ppi3;
|
|
|
|
ok(pints_sum(&pints) == -13, "RPC pints_sum\n");
|
|
|
|
|
|
|
|
c = 10;
|
|
|
|
h = 3;
|
|
|
|
q = 14;
|
|
|
|
s = -5.0f;
|
|
|
|
u = 11.0;
|
|
|
|
ptypes.pc = &c;
|
|
|
|
ptypes.ps = &h;
|
|
|
|
ptypes.pl = &q;
|
|
|
|
ptypes.pf = &s;
|
|
|
|
ptypes.pd = &u;
|
|
|
|
ok(ptypes_sum(&ptypes) == 33.0, "RPC ptypes_sum\n");
|
|
|
|
|
2007-05-12 02:27:40 +02:00
|
|
|
ok(dot_pvectors(&pvecs) == -21, "RPC dot_pvectors\n");
|
2007-04-26 03:22:10 +02:00
|
|
|
ok(sum_fixed_array(f) == -2, "RPC sum_fixed_array\n");
|
2007-04-26 03:18:29 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
client(const char *test)
|
|
|
|
{
|
|
|
|
if (strcmp(test, "tcp_basic") == 0)
|
|
|
|
{
|
|
|
|
static unsigned char iptcp[] = "ncacn_ip_tcp";
|
|
|
|
static unsigned char address[] = "127.0.0.1";
|
|
|
|
static unsigned char port[] = PORT;
|
|
|
|
unsigned char *binding;
|
|
|
|
|
|
|
|
ok(RPC_S_OK == RpcStringBindingCompose(NULL, iptcp, address, port, NULL, &binding), "RpcStringBindingCompose\n");
|
|
|
|
ok(RPC_S_OK == RpcBindingFromStringBinding(binding, &IServer_IfHandle), "RpcBindingFromStringBinding\n");
|
|
|
|
|
|
|
|
basic_tests();
|
|
|
|
|
|
|
|
ok(RPC_S_OK == RpcStringFree(&binding), "RpcStringFree\n");
|
|
|
|
ok(RPC_S_OK == RpcBindingFree(&IServer_IfHandle), "RpcBindingFree\n");
|
|
|
|
}
|
|
|
|
else if (strcmp(test, "np_basic") == 0)
|
|
|
|
{
|
|
|
|
static unsigned char np[] = "ncacn_np";
|
|
|
|
static unsigned char address[] = "\\\\.";
|
|
|
|
static unsigned char pipe[] = PIPE;
|
|
|
|
unsigned char *binding;
|
|
|
|
|
|
|
|
ok(RPC_S_OK == RpcStringBindingCompose(NULL, np, address, pipe, NULL, &binding), "RpcStringBindingCompose\n");
|
|
|
|
ok(RPC_S_OK == RpcBindingFromStringBinding(binding, &IServer_IfHandle), "RpcBindingFromStringBinding\n");
|
|
|
|
|
|
|
|
basic_tests();
|
|
|
|
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 iptcp[] = "ncacn_ip_tcp";
|
|
|
|
static unsigned char port[] = PORT;
|
|
|
|
static unsigned char np[] = "ncacn_np";
|
|
|
|
static unsigned char pipe[] = PIPE;
|
|
|
|
|
|
|
|
ok(RPC_S_OK == RpcServerUseProtseqEp(iptcp, 20, port, NULL), "RpcServerUseProtseqEp\n");
|
|
|
|
ok(RPC_S_OK == RpcServerRegisterIf(s_IServer_v0_0_s_ifspec, NULL, NULL), "RpcServerRegisterIf\n");
|
|
|
|
ok(RPC_S_OK == RpcServerListen(1, 20, TRUE), "RpcServerListen\n");
|
|
|
|
|
|
|
|
stop_event = CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
|
|
ok(stop_event != NULL, "CreateEvent failed\n");
|
|
|
|
|
|
|
|
ok(run_client("tcp_basic"), "tcp_basic client test failed\n");
|
|
|
|
|
|
|
|
ok(RPC_S_OK == RpcServerUseProtseqEp(np, 0, pipe, NULL), "RpcServerUseProtseqEp\n");
|
|
|
|
ok(run_client("np_basic"), "np_basic client test failed\n");
|
|
|
|
|
|
|
|
ok(WAIT_OBJECT_0 == WaitForSingleObject(stop_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();
|
|
|
|
}
|