2 * Tests for WIDL and RPC server/clients.
4 * Copyright (C) Google 2007 (Dan Hipschman)
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.
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.
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
22 #include "wine/test.h"
24 #include "server_defines.h"
31 #define PIPE "\\pipe\\wine_rpcrt4_test"
35 static const char *progname
;
37 static HANDLE stop_event
;
39 static void (WINAPI
*pNDRSContextMarshall2
)(RPC_BINDING_HANDLE
, NDR_SCONTEXT
, void*, NDR_RUNDOWN
, void*, ULONG
);
40 static NDR_SCONTEXT (WINAPI
*pNDRSContextUnmarshall2
)(RPC_BINDING_HANDLE
, void*, ULONG
, void*, ULONG
);
42 static void InitFunctionPointers(void)
44 HMODULE hrpcrt4
= GetModuleHandleA("rpcrt4.dll");
46 pNDRSContextMarshall2
= (void *)GetProcAddress(hrpcrt4
, "NDRSContextMarshall2");
47 pNDRSContextUnmarshall2
= (void *)GetProcAddress(hrpcrt4
, "NDRSContextUnmarshall2");
50 void __RPC_FAR
*__RPC_USER
51 midl_user_allocate(size_t n
)
57 midl_user_free(void __RPC_FAR
*p
)
63 xstrdup(const char *s
)
65 char *d
= HeapAlloc(GetProcessHeap(), 0, strlen(s
) + 1);
89 s_square_out(int x
, int *y
)
101 s_str_length(const char *s
)
107 s_str_t_length(str_t s
)
113 s_cstr_length(const char *s
, int n
)
116 while (0 < n
-- && *s
++)
122 s_dot_self(vector_t
*v
)
124 return s_square(v
->x
) + s_square(v
->y
) + s_square(v
->z
);
128 s_square_half(double x
, double *y
)
135 s_square_half_float(float x
, float *y
)
142 s_square_half_long(long x
, long *y
)
149 s_sum_fixed_array(int a
[5])
151 return a
[0] + a
[1] + a
[2] + a
[3] + a
[4];
155 s_pints_sum(pints_t
*pints
)
157 return *pints
->pi
+ **pints
->ppi
+ ***pints
->pppi
;
161 s_ptypes_sum(ptypes_t
*pt
)
163 return *pt
->pc
+ *pt
->ps
+ *pt
->pl
+ *pt
->pf
+ *pt
->pd
;
167 s_dot_pvectors(pvectors_t
*p
)
169 return p
->pu
->x
* (*p
->pv
)->x
+ p
->pu
->y
* (*p
->pv
)->y
+ p
->pu
->z
* (*p
->pv
)->z
;
175 return sp
->x
+ sp
->s
->x
;
179 s_square_sun(sun_t
*su
)
183 case SUN_I
: return su
->u
.i
* su
->u
.i
;
185 case SUN_F2
: return su
->u
.f
* su
->u
.f
;
186 case SUN_PI
: return (*su
->u
.pi
) * (*su
->u
.pi
);
193 s_test_list_length(test_list_t
*list
)
195 return (list
->t
== TL_LIST
196 ? 1 + s_test_list_length(list
->u
.tail
)
201 s_sum_fixed_int_3d(int m
[2][3][4])
206 for (i
= 0; i
< 2; ++i
)
207 for (j
= 0; j
< 3; ++j
)
208 for (k
= 0; k
< 4; ++k
)
215 s_sum_conf_array(int x
[], int n
)
217 int *p
= x
, *end
= p
+ n
;
227 s_sum_unique_conf_array(int x
[], int n
)
229 return s_sum_conf_array(x
, n
);
233 s_sum_unique_conf_ptr(int *x
, int n
)
235 return x
? s_sum_conf_array(x
, n
) : 0;
239 s_sum_var_array(int x
[20], int n
)
241 ok(0 <= n
, "RPC sum_var_array\n");
242 ok(n
<= 20, "RPC sum_var_array\n");
244 return s_sum_conf_array(x
, n
);
248 s_dot_two_vectors(vector_t vs
[2])
250 return vs
[0].x
* vs
[1].x
+ vs
[0].y
* vs
[1].y
+ vs
[0].z
* vs
[1].z
;
256 return s_sum_conf_array(cs
->ca
, cs
->n
);
260 s_sum_cps(cps_t
*cps
)
265 for (i
= 0; i
< *cps
->pn
; ++i
)
268 for (i
= 0; i
< 2 * cps
->n
; ++i
)
275 s_sum_cpsc(cpsc_t
*cpsc
)
279 for (i
= 0; i
< (cpsc
->c
? cpsc
->a
: cpsc
->b
); ++i
)
285 s_square_puint(puint_t p
)
292 s_sum_puints(puints_t
*p
)
296 for (i
= 0; i
< p
->n
; ++i
)
297 sum
+= atoi(p
->ps
[i
]);
302 s_sum_cpuints(cpuints_t
*p
)
306 for (i
= 0; i
< p
->n
; ++i
)
307 sum
+= atoi(p
->ps
[i
]);
312 s_dot_copy_vectors(vector_t u
, vector_t v
)
314 return u
.x
* v
.x
+ u
.y
* v
.y
+ u
.z
* v
.z
;
318 s_square_test_us(test_us_t
*tus
)
320 int n
= atoi(tus
->us
.x
);
325 s_square_encu(encu_t
*eu
)
329 case ENCU_I
: return eu
->tagged_union
.i
* eu
->tagged_union
.i
;
330 case ENCU_F
: return eu
->tagged_union
.f
* eu
->tagged_union
.f
;
337 s_square_unencu(int t
, unencu_t
*eu
)
341 case ENCU_I
: return eu
->i
* eu
->i
;
342 case ENCU_F
: return eu
->f
* eu
->f
;
351 ok(s
->f
== E2
, "check_se2\n");
355 s_sum_parr(int *a
[3])
357 return s_sum_pcarr(a
, 3);
361 s_sum_pcarr(int *a
[], int n
)
364 for (i
= 0; i
< n
; ++i
)
384 s_square_encue(encue_t
*eue
)
388 case E1
: return eue
->tagged_union
.i1
* eue
->tagged_union
.i1
;
389 case E2
: return eue
->tagged_union
.f2
* eue
->tagged_union
.f2
;
396 s_sum_toplev_conf_2n(int *x
, int n
)
400 for (i
= 0; i
< 2 * n
; ++i
)
406 s_sum_toplev_conf_cond(int *x
, int a
, int b
, int c
)
411 for (i
= 0; i
< n
; ++i
)
417 s_sum_aligns(aligns_t
*a
)
419 return a
->c
+ a
->i
+ a
->s
+ a
->d
;
423 s_sum_padded(padded_t
*p
)
429 s_sum_padded2(padded_t ps
[2])
431 return s_sum_padded(&ps
[0]) + s_sum_padded(&ps
[1]);
435 s_sum_padded_conf(padded_t
*ps
, int n
)
439 for (i
= 0; i
< n
; ++i
)
440 sum
+= s_sum_padded(&ps
[i
]);
445 s_sum_bogus(bogus_t
*b
)
447 return *b
->h
.p1
+ *b
->p2
+ *b
->p3
+ b
->c
;
451 s_check_null(int *null
)
453 ok(!null
, "RPC check_null\n");
457 s_str_struct_len(str_struct_t
*s
)
459 return lstrlenA(s
->s
);
463 s_wstr_struct_len(wstr_struct_t
*s
)
465 return lstrlenW(s
->s
);
469 s_sum_doub_carr(doub_carr_t
*dc
)
473 for (i
= 0; i
< dc
->n
; ++i
)
474 for (j
= 0; j
< dc
->a
[i
]->n
; ++j
)
475 sum
+= dc
->a
[i
]->a
[j
];
480 s_make_pyramid_doub_carr(unsigned char n
, doub_carr_t
**dc
)
484 t
= MIDL_user_allocate(FIELD_OFFSET(doub_carr_t
, a
[n
]));
486 for (i
= 0; i
< n
; ++i
)
489 t
->a
[i
] = MIDL_user_allocate(FIELD_OFFSET(doub_carr_1_t
, a
[v
]));
491 for (j
= 0; j
< v
; ++j
)
492 t
->a
[i
]->a
[j
] = j
+ 1;
498 s_hash_bstr(bstr_t b
)
504 for (i
= 0; i
< n
; ++i
)
505 hash
= 5 * hash
+ (unsigned) s
[i
];
510 s_get_name(name_t
*name
)
512 const char bossman
[] = "Jeremy White";
513 memcpy(name
->name
, bossman
, min(name
->size
, sizeof(bossman
)));
514 /* ensure nul-termination */
515 if (name
->size
< sizeof(bossman
))
516 name
->name
[name
->size
- 1] = 0;
520 s_sum_pcarr2(int n
, int **pa
)
522 return s_sum_conf_array(*pa
, n
);
526 s_sum_L1_norms(int n
, vector_t
*vs
)
530 for (i
= 0; i
< n
; ++i
)
531 sum
+= abs(vs
[i
].x
) + abs(vs
[i
].y
) + abs(vs
[i
].z
);
538 s123_t
*s
= MIDL_user_allocate(sizeof *s
);
548 return (char *)__FILE__
;
552 s_context_handle_test(void)
555 RPC_BINDING_HANDLE binding
;
557 unsigned char buf
[20];
558 static RPC_SERVER_INTERFACE server_if
=
560 sizeof(RPC_SERVER_INTERFACE
),
561 {{0x00000000,0x4114,0x0704,{0x23,0x01,0x00,0x00,0x00,0x00,0x00,0x00}},{1,0}},
562 {{0x8a885d04,0x1ceb,0x11c9,{0x9f,0xe8,0x08,0x00,0x2b,0x10,0x48,0x60}},{2,0}},
571 binding
= I_RpcGetCurrentCallHandle();
572 ok(binding
!= NULL
, "I_RpcGetCurrentCallHandle returned NULL\n");
574 h
= pNDRSContextUnmarshall2(binding
, NULL
, NDR_LOCAL_DATA_REPRESENTATION
, NULL
, 0);
575 ok(h
!= NULL
, "NDRSContextUnmarshall2 returned NULL\n");
577 /* marshal a context handle with NULL userContext */
578 memset(buf
, 0xcc, sizeof(buf
));
579 pNDRSContextMarshall2(binding
, h
, buf
, NULL
, NULL
, 0);
580 ok(*(ULONG
*)buf
== 0, "attributes should have been set to 0 instead of 0x%x\n", *(ULONG
*)buf
);
581 ok(UuidIsNil((UUID
*)&buf
[4], &status
), "uuid should have been nil\n");
583 h
= pNDRSContextUnmarshall2(binding
, NULL
, NDR_LOCAL_DATA_REPRESENTATION
, NULL
, 0);
584 ok(h
!= NULL
, "NDRSContextUnmarshall2 returned NULL\n");
586 /* marshal a context handle with non-NULL userContext */
587 memset(buf
, 0xcc, sizeof(buf
));
588 h
->userContext
= (void *)0xdeadbeef;
589 pNDRSContextMarshall2(binding
, h
, buf
, NULL
, NULL
, 0);
590 ok(*(ULONG
*)buf
== 0, "attributes should have been set to 0 instead of 0x%x\n", *(ULONG
*)buf
);
591 ok(!UuidIsNil((UUID
*)&buf
[4], &status
), "uuid should not have been nil\n");
593 h
= pNDRSContextUnmarshall2(binding
, buf
, NDR_LOCAL_DATA_REPRESENTATION
, NULL
, 0);
594 ok(h
!= NULL
, "NDRSContextUnmarshall2 returned NULL\n");
595 ok(h
->userContext
== (void *)0xdeadbeef, "userContext of interface didn't unmarshal properly: %p\n", h
->userContext
);
597 /* marshal a context handle with an interface specified */
598 h
= pNDRSContextUnmarshall2(binding
, NULL
, NDR_LOCAL_DATA_REPRESENTATION
, &server_if
.InterfaceId
, 0);
599 ok(h
!= NULL
, "NDRSContextUnmarshall2 returned NULL\n");
601 memset(buf
, 0xcc, sizeof(buf
));
602 h
->userContext
= (void *)0xcafebabe;
603 pNDRSContextMarshall2(binding
, h
, buf
, NULL
, &server_if
.InterfaceId
, 0);
604 ok(*(ULONG
*)buf
== 0, "attributes should have been set to 0 instead of 0x%x\n", *(ULONG
*)buf
);
605 ok(!UuidIsNil((UUID
*)&buf
[4], &status
), "uuid should not have been nil\n");
607 h
= pNDRSContextUnmarshall2(binding
, buf
, NDR_LOCAL_DATA_REPRESENTATION
, &server_if
.InterfaceId
, 0);
608 ok(h
!= NULL
, "NDRSContextUnmarshall2 returned NULL\n");
609 ok(h
->userContext
== (void *)0xcafebabe, "userContext of interface didn't unmarshal properly: %p\n", h
->userContext
);
611 /* test same interface data, but different pointer */
612 /* raises ERROR_INVALID_HANDLE exception */
615 RPC_SERVER_INTERFACE server_if_clone
= server_if
;
617 pNDRSContextUnmarshall2(binding
, buf
, NDR_LOCAL_DATA_REPRESENTATION
, &server_if_clone
.InterfaceId
, 0);
620 /* test different interface data, but different pointer */
621 /* raises ERROR_INVALID_HANDLE exception */
624 static RPC_SERVER_INTERFACE server_if2
=
626 sizeof(RPC_SERVER_INTERFACE
),
627 {{0x00000000,0x4114,0x0704,{0x23,0x01,0x00,0x00,0x00,0x00,0x00,0x00}},{1,0}},
628 {{0x8a885d04,0x1ceb,0x11c9,{0x9f,0xe8,0x08,0x00,0x2b,0x10,0x48,0x60}},{2,0}},
636 pNDRSContextMarshall2(binding
, h
, buf
, NULL
, &server_if
.InterfaceId
, 0);
638 pNDRSContextUnmarshall2(binding
, buf
, NDR_LOCAL_DATA_REPRESENTATION
, &server_if2
.InterfaceId
, 0);
643 s_get_5numbers(int count
, pints_t n
[5])
646 for (i
= 0; i
< count
; i
++)
648 n
[i
].pi
= midl_user_allocate(sizeof(*n
[i
].pi
));
656 s_get_numbers(int length
, int size
, pints_t n
[])
659 for (i
= 0; i
< length
; i
++)
661 n
[i
].pi
= midl_user_allocate(sizeof(*n
[i
].pi
));
669 s_get_numbers_struct(numbers_struct_t
**ns
)
672 *ns
= midl_user_allocate(FIELD_OFFSET(numbers_struct_t
, numbers
[5]));
676 for (i
= 0; i
< (*ns
)->length
; i
++)
678 (*ns
)->numbers
[i
].pi
= NULL
;
679 (*ns
)->numbers
[i
].ppi
= NULL
;
680 (*ns
)->numbers
[i
].pppi
= NULL
;
682 (*ns
)->numbers
[0].pi
= midl_user_allocate(sizeof(*(*ns
)->numbers
[i
].pi
));
683 *(*ns
)->numbers
[0].pi
= 5;
689 ok(RPC_S_OK
== RpcMgmtStopServerListening(NULL
), "RpcMgmtStopServerListening\n");
690 ok(RPC_S_OK
== RpcServerUnregisterIf(NULL
, NULL
, FALSE
), "RpcServerUnregisterIf\n");
691 ok(SetEvent(stop_event
), "SetEvent\n");
695 make_cmdline(char buffer
[MAX_PATH
], const char *test
)
697 sprintf(buffer
, "%s server %s", progname
, test
);
701 run_client(const char *test
)
703 char cmdline
[MAX_PATH
];
704 PROCESS_INFORMATION info
;
705 STARTUPINFOA startup
;
707 memset(&startup
, 0, sizeof startup
);
708 startup
.cb
= sizeof startup
;
710 make_cmdline(cmdline
, test
);
711 ok(CreateProcessA(NULL
, cmdline
, NULL
, NULL
, FALSE
, 0L, NULL
, NULL
, &startup
, &info
), "CreateProcess\n");
712 winetest_wait_child_process( info
.hProcess
);
713 ok(CloseHandle(info
.hProcess
), "CloseHandle\n");
714 ok(CloseHandle(info
.hThread
), "CloseHandle\n");
720 char string
[] = "I am a string";
721 WCHAR wstring
[] = {'I',' ','a','m',' ','a',' ','w','s','t','r','i','n','g', 0};
722 int f
[5] = {1, 3, 0, -2, -4};
723 vector_t a
= {1, 3, 7};
724 vector_t vec1
= {4, -2, 1}, vec2
= {-5, 2, 3}, *pvec2
= &vec2
;
725 pvectors_t pvecs
= {&vec1
, &pvec2
};
726 sp_inner_t spi
= {42};
727 sp_t sp
= {-13, &spi
};
734 int i1
, i2
, i3
, *pi2
, *pi3
, **ppi3
;
741 str_struct_t ss
= {string
};
742 wstr_struct_t ws
= {wstring
};
746 ok(int_return() == INT_CODE
, "RPC int_return\n");
748 ok(square(7) == 49, "RPC square\n");
749 ok(sum(23, -4) == 19, "RPC sum\n");
753 ok(x
== 121, "RPC square_out\n");
757 ok(x
== 25, "RPC square_ref\n");
759 ok(str_length(string
) == strlen(string
), "RPC str_length\n");
760 ok(str_t_length(string
) == strlen(string
), "RPC str_length\n");
761 ok(dot_self(&a
) == 59, "RPC dot_self\n");
763 ok(str_struct_len(&ss
) == lstrlenA(string
), "RPC str_struct_len\n");
764 ok(wstr_struct_len(&ws
) == lstrlenW(wstring
), "RPC str_struct_len\n");
767 u
= square_half(3.0, &v
);
768 ok(u
== 9.0, "RPC square_half\n");
769 ok(v
== 1.5, "RPC square_half\n");
772 s
= square_half_float(3.0f
, &t
);
773 ok(s
== 9.0f
, "RPC square_half_float\n");
774 ok(t
== 1.5f
, "RPC square_half_float\n");
777 q
= square_half_long(3, &r
);
778 ok(q
== 9, "RPC square_half_long\n");
779 ok(r
== 1, "RPC square_half_long\n");
790 ok(pints_sum(&pints
) == -13, "RPC pints_sum\n");
802 ok(ptypes_sum(&ptypes
) == 33.0, "RPC ptypes_sum\n");
804 ok(dot_pvectors(&pvecs
) == -21, "RPC dot_pvectors\n");
805 ok(dot_copy_vectors(vec1
, vec2
) == -21, "RPC dot_copy_vectors\n");
806 ok(sum_fixed_array(f
) == -2, "RPC sum_fixed_array\n");
807 ok(sum_sp(&sp
) == 29, "RPC sum_sp\n");
809 ok(enum_ord(E1
) == 1, "RPC enum_ord\n");
810 ok(enum_ord(E2
) == 2, "RPC enum_ord\n");
811 ok(enum_ord(E3
) == 3, "RPC enum_ord\n");
812 ok(enum_ord(E4
) == 4, "RPC enum_ord\n");
817 memset(&aligns
, 0, sizeof(aligns
));
822 ok(sum_aligns(&aligns
) == 18.0, "RPC sum_aligns\n");
826 ok(sum_padded(&padded
) == 5, "RPC sum_padded\n");
831 ok(sum_padded2(padded2
) == 6, "RPC sum_padded2\n");
836 ok(sum_padded_conf(padded2
, 2) == 6, "RPC sum_padded_conf\n");
845 ok(sum_bogus(&bogus
) == 12, "RPC sum_bogus\n");
849 str
= get_filename();
850 ok(!strcmp(str
, __FILE__
), "get_filename() returned %s instead of %s\n", str
, __FILE__
);
865 ok(square_sun(&su
) == 81.0, "RPC square_sun\n");
869 ok(square_sun(&su
) == 25.0, "RPC square_sun\n");
873 ok(square_sun(&su
) == 4.0, "RPC square_sun\n");
878 ok(square_sun(&su
) == 121.0, "RPC square_sun\n");
881 eu
.tagged_union
.i
= 7;
882 ok(square_encu(&eu
) == 49.0, "RPC square_encu\n");
885 eu
.tagged_union
.f
= 3.0;
886 ok(square_encu(&eu
) == 9.0, "RPC square_encu\n");
889 ok(square_unencu(ENCU_I
, &uneu
) == 16.0, "RPC square_unencu\n");
892 ok(square_unencu(ENCU_F
, &uneu
) == 25.0, "RPC square_unencu\n");
895 eue
.tagged_union
.i1
= 8;
896 ok(square_encue(&eue
) == 64.0, "RPC square_encue\n");
899 eue
.tagged_union
.f2
= 10.0;
900 ok(square_encue(&eue
) == 100.0, "RPC square_encue\n");
906 test_list_t
*n
= HeapAlloc(GetProcessHeap(), 0, sizeof *n
);
913 make_list(test_list_t
*tail
)
915 test_list_t
*n
= HeapAlloc(GetProcessHeap(), 0, sizeof *n
);
922 free_list(test_list_t
*list
)
924 if (list
->t
== TL_LIST
)
925 free_list(list
->u
.tail
);
926 HeapFree(GetProcessHeap(), 0, list
);
930 puint_t_UserSize(ULONG
*flags
, ULONG start
, puint_t
*p
)
932 return start
+ sizeof(int);
935 unsigned char * __RPC_USER
936 puint_t_UserMarshal(ULONG
*flags
, unsigned char *buffer
, puint_t
*p
)
939 memcpy(buffer
, &n
, sizeof n
);
940 return buffer
+ sizeof n
;
943 unsigned char * __RPC_USER
944 puint_t_UserUnmarshal(ULONG
*flags
, unsigned char *buffer
, puint_t
*p
)
947 memcpy(&n
, buffer
, sizeof n
);
948 *p
= HeapAlloc(GetProcessHeap(), 0, 10);
949 sprintf(*p
, "%d", n
);
950 return buffer
+ sizeof n
;
954 puint_t_UserFree(ULONG
*flags
, puint_t
*p
)
956 HeapFree(GetProcessHeap(), 0, *p
);
960 us_t_UserSize(ULONG
*flags
, ULONG start
, us_t
*pus
)
962 return start
+ sizeof(struct wire_us
);
965 unsigned char * __RPC_USER
966 us_t_UserMarshal(ULONG
*flags
, unsigned char *buffer
, us_t
*pus
)
969 wus
.x
= atoi(pus
->x
);
970 memcpy(buffer
, &wus
, sizeof wus
);
971 return buffer
+ sizeof wus
;
974 unsigned char * __RPC_USER
975 us_t_UserUnmarshal(ULONG
*flags
, unsigned char *buffer
, us_t
*pus
)
978 memcpy(&wus
, buffer
, sizeof wus
);
979 pus
->x
= HeapAlloc(GetProcessHeap(), 0, 10);
980 sprintf(pus
->x
, "%d", wus
.x
);
981 return buffer
+ sizeof wus
;
985 us_t_UserFree(ULONG
*flags
, us_t
*pus
)
987 HeapFree(GetProcessHeap(), 0, pus
->x
);
991 bstr_t_UserSize(ULONG
*flags
, ULONG start
, bstr_t
*b
)
993 return start
+ FIELD_OFFSET(wire_bstr_t
, data
[(*b
)[-1]]);
996 unsigned char * __RPC_USER
997 bstr_t_UserMarshal(ULONG
*flags
, unsigned char *buffer
, bstr_t
*b
)
999 wire_bstr_t
*wb
= (wire_bstr_t
*) buffer
;
1001 memcpy(&wb
->data
, *b
, wb
->n
* sizeof wb
->data
[0]);
1002 return buffer
+ FIELD_OFFSET(wire_bstr_t
, data
[wb
->n
]);
1005 unsigned char * __RPC_USER
1006 bstr_t_UserUnmarshal(ULONG
*flags
, unsigned char *buffer
, bstr_t
*b
)
1008 wire_bstr_t
*wb
= (wire_bstr_t
*) buffer
;
1009 short *data
= HeapAlloc(GetProcessHeap(), 0, (wb
->n
+ 1) * sizeof *data
);
1011 memcpy(&data
[1], wb
->data
, wb
->n
* sizeof data
[1]);
1013 return buffer
+ FIELD_OFFSET(wire_bstr_t
, data
[wb
->n
]);
1017 bstr_t_UserFree(ULONG
*flags
, bstr_t
*b
)
1019 HeapFree(GetProcessHeap(), 0, &((*b
)[-1]));
1025 int a
[] = {1, 2, 3, 4};
1027 test_list_t
*list
= make_list(make_list(make_list(null_list())));
1028 test_us_t tus
= {{p1
}};
1032 short bstr_data
[] = { 5, 'H', 'e', 'l', 'l', 'o' };
1033 bstr_t bstr
= &bstr_data
[1];
1039 ok(test_list_length(list
) == 3, "RPC test_list_length\n");
1040 ok(square_puint(p1
) == 121, "RPC square_puint\n");
1042 pus
.ps
= HeapAlloc(GetProcessHeap(), 0, pus
.n
* sizeof pus
.ps
[0]);
1043 pus
.ps
[0] = xstrdup("5");
1044 pus
.ps
[1] = xstrdup("6");
1045 pus
.ps
[2] = xstrdup("7");
1046 pus
.ps
[3] = xstrdup("8");
1047 ok(sum_puints(&pus
) == 26, "RPC sum_puints\n");
1048 HeapFree(GetProcessHeap(), 0, pus
.ps
[0]);
1049 HeapFree(GetProcessHeap(), 0, pus
.ps
[1]);
1050 HeapFree(GetProcessHeap(), 0, pus
.ps
[2]);
1051 HeapFree(GetProcessHeap(), 0, pus
.ps
[3]);
1052 HeapFree(GetProcessHeap(), 0, pus
.ps
);
1054 cpus
.ps
= HeapAlloc(GetProcessHeap(), 0, cpus
.n
* sizeof cpus
.ps
[0]);
1055 cpus
.ps
[0] = xstrdup("5");
1056 cpus
.ps
[1] = xstrdup("6");
1057 cpus
.ps
[2] = xstrdup("7");
1058 cpus
.ps
[3] = xstrdup("8");
1059 ok(sum_cpuints(&cpus
) == 26, "RPC sum_puints\n");
1060 HeapFree(GetProcessHeap(), 0, cpus
.ps
[0]);
1061 HeapFree(GetProcessHeap(), 0, cpus
.ps
[1]);
1062 HeapFree(GetProcessHeap(), 0, cpus
.ps
[2]);
1063 HeapFree(GetProcessHeap(), 0, cpus
.ps
[3]);
1064 HeapFree(GetProcessHeap(), 0, cpus
.ps
);
1065 ok(square_test_us(&tus
) == 121, "RPC square_test_us\n");
1070 ok(sum_parr(pa
) == 6, "RPC sum_parr\n");
1076 ok(sum_pcarr(pa
, 4) == 10, "RPC sum_pcarr\n");
1078 ok(hash_bstr(bstr
) == s_hash_bstr(bstr
), "RPC hash_bstr_data\n");
1083 name
.name
= buffer
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, name
.size
);
1085 ok(name
.name
== buffer
, "[in,out] pointer should have stayed as %p but instead changed to %p\n", name
.name
, buffer
);
1086 ok(!strcmp(name
.name
, "Jeremy Wh"), "name didn't unmarshall properly, expected \"Jeremy Wh\", but got \"%s\"\n", name
.name
);
1087 HeapFree(GetProcessHeap(), 0, name
.name
);
1090 ok(sum_pcarr2(4, &pa2
) == 10, "RPC sum_pcarr2\n");
1093 ok(s123
->f1
== 1 && s123
->f2
== 2 && s123
->f3
== 3, "RPC get_s123\n");
1094 MIDL_user_free(s123
);
1098 check_pyramid_doub_carr(doub_carr_t
*dc
)
1101 for (i
= 0; i
< dc
->n
; ++i
)
1102 for (j
= 0; j
< dc
->a
[i
]->n
; ++j
)
1103 if (dc
->a
[i
]->a
[j
] != j
+ 1)
1109 free_pyramid_doub_carr(doub_carr_t
*dc
)
1112 for (i
= 0; i
< dc
->n
; ++i
)
1113 MIDL_user_free(dc
->a
[i
]);
1120 const char str1
[25] = "Hello";
1123 {{1, 2, 3, 4}, {-1, -3, -5, -7}, {0, 2, 4, 6}},
1124 {{1, -2, 3, -4}, {2, 3, 5, 7}, {-4, -1, -14, 4114}}
1126 int c
[] = {0, 1, 2, 3, 4, 5, 6, 7, 8, 9};
1127 vector_t vs
[2] = {{1, -2, 3}, {4, -5, -6}};
1132 int ca
[5] = {1, -2, 3, -4, 5};
1136 numbers_struct_t
*ns
;
1138 ok(cstr_length(str1
, sizeof str1
) == strlen(str1
), "RPC cstr_length\n");
1140 ok(sum_fixed_int_3d(m
) == 4116, "RPC sum_fixed_int_3d\n");
1142 ok(sum_conf_array(c
, 10) == 45, "RPC sum_conf_array\n");
1143 ok(sum_conf_array(&c
[5], 2) == 11, "RPC sum_conf_array\n");
1144 ok(sum_conf_array(&c
[7], 1) == 7, "RPC sum_conf_array\n");
1145 ok(sum_conf_array(&c
[2], 0) == 0, "RPC sum_conf_array\n");
1147 ok(sum_unique_conf_array(ca
, 4) == -2, "RPC sum_unique_conf_array\n");
1148 ok(sum_unique_conf_ptr(ca
, 5) == 3, "RPC sum_unique_conf_array\n");
1149 ok(sum_unique_conf_ptr(NULL
, 10) == 0, "RPC sum_unique_conf_array\n");
1151 ok(sum_var_array(c
, 10) == 45, "RPC sum_conf_array\n");
1152 ok(sum_var_array(&c
[5], 2) == 11, "RPC sum_conf_array\n");
1153 ok(sum_var_array(&c
[7], 1) == 7, "RPC sum_conf_array\n");
1154 ok(sum_var_array(&c
[2], 0) == 0, "RPC sum_conf_array\n");
1156 ok(dot_two_vectors(vs
) == -4, "RPC dot_two_vectors\n");
1157 cs
= HeapAlloc(GetProcessHeap(), 0, FIELD_OFFSET(cs_t
, ca
[5]));
1164 ok(sum_cs(cs
) == 1, "RPC sum_cs\n");
1165 HeapFree(GetProcessHeap(), 0, cs
);
1172 ok(sum_cps(&cps
) == 53, "RPC sum_cps\n");
1178 ok(sum_cpsc(&cpsc
) == 6, "RPC sum_cpsc\n");
1183 ok(sum_cpsc(&cpsc
) == 10, "RPC sum_cpsc\n");
1185 ok(sum_toplev_conf_2n(c
, 3) == 15, "RPC sum_toplev_conf_2n\n");
1186 ok(sum_toplev_conf_cond(c
, 5, 6, 1) == 10, "RPC sum_toplev_conf_cond\n");
1187 ok(sum_toplev_conf_cond(c
, 5, 6, 0) == 15, "RPC sum_toplev_conf_cond\n");
1189 dc
= malloc(FIELD_OFFSET(doub_carr_t
, a
[2]));
1191 dc
->a
[0] = malloc(FIELD_OFFSET(doub_carr_1_t
, a
[3]));
1196 dc
->a
[1] = malloc(FIELD_OFFSET(doub_carr_1_t
, a
[2]));
1200 ok(sum_doub_carr(dc
) == 19, "RPC sum_doub_carr\n");
1206 make_pyramid_doub_carr(4, &dc
);
1207 ok(check_pyramid_doub_carr(dc
), "RPC make_pyramid_doub_carr\n");
1208 free_pyramid_doub_carr(dc
);
1210 ok(sum_L1_norms(2, vs
) == 21, "RPC sum_L1_norms\n");
1212 memset(api
, 0, sizeof(api
));
1213 pi
= HeapAlloc(GetProcessHeap(), 0, sizeof(*pi
));
1216 get_5numbers(1, api
);
1217 ok(api
[0].pi
== pi
, "RPC varying array [out] pointer changed from %p to %p\n", pi
, api
[0].pi
);
1218 ok(*api
[0].pi
== 0, "pi unmarshalled incorrectly %d\n", *api
[0].pi
);
1221 get_numbers(1, 1, api
);
1222 ok(api
[0].pi
== pi
, "RPC conformant varying array [out] pointer changed from %p to %p\n", pi
, api
[0].pi
);
1223 ok(*api
[0].pi
== 0, "pi unmarshalled incorrectly %d\n", *api
[0].pi
);
1225 ns
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, FIELD_OFFSET(numbers_struct_t
, numbers
[5]));
1228 ns
->numbers
[0].pi
= pi
;
1229 get_numbers_struct(&ns
);
1230 ok(ns
->numbers
[0].pi
== pi
, "RPC conformant varying struct embedded pointer changed from %p to %p\n", pi
, ns
->numbers
[0].pi
);
1231 ok(*ns
->numbers
[0].pi
== 5, "pi unmarshalled incorrectly %d\n", *ns
->numbers
[0].pi
);
1233 HeapFree(GetProcessHeap(), 0, ns
);
1234 HeapFree(GetProcessHeap(), 0, pi
);
1244 context_handle_test();
1248 client(const char *test
)
1250 if (strcmp(test
, "tcp_basic") == 0)
1252 static unsigned char iptcp
[] = "ncacn_ip_tcp";
1253 static unsigned char address
[] = "127.0.0.1";
1254 static unsigned char port
[] = PORT
;
1255 unsigned char *binding
;
1257 ok(RPC_S_OK
== RpcStringBindingCompose(NULL
, iptcp
, address
, port
, NULL
, &binding
), "RpcStringBindingCompose\n");
1258 ok(RPC_S_OK
== RpcBindingFromStringBinding(binding
, &IServer_IfHandle
), "RpcBindingFromStringBinding\n");
1262 ok(RPC_S_OK
== RpcStringFree(&binding
), "RpcStringFree\n");
1263 ok(RPC_S_OK
== RpcBindingFree(&IServer_IfHandle
), "RpcBindingFree\n");
1265 else if (strcmp(test
, "np_basic") == 0)
1267 static unsigned char np
[] = "ncacn_np";
1268 static unsigned char address
[] = "\\\\.";
1269 static unsigned char pipe
[] = PIPE
;
1270 unsigned char *binding
;
1272 ok(RPC_S_OK
== RpcStringBindingCompose(NULL
, np
, address
, pipe
, NULL
, &binding
), "RpcStringBindingCompose\n");
1273 ok(RPC_S_OK
== RpcBindingFromStringBinding(binding
, &IServer_IfHandle
), "RpcBindingFromStringBinding\n");
1278 ok(RPC_S_OK
== RpcStringFree(&binding
), "RpcStringFree\n");
1279 ok(RPC_S_OK
== RpcBindingFree(&IServer_IfHandle
), "RpcBindingFree\n");
1286 static unsigned char iptcp
[] = "ncacn_ip_tcp";
1287 static unsigned char port
[] = PORT
;
1288 static unsigned char np
[] = "ncacn_np";
1289 static unsigned char pipe
[] = PIPE
;
1290 RPC_STATUS status
, iptcp_status
, np_status
;
1291 RPC_STATUS (RPC_ENTRY
*pRpcServerRegisterIfEx
)(RPC_IF_HANDLE
,UUID
*,
1292 RPC_MGR_EPV
*, unsigned int,unsigned int,RPC_IF_CALLBACK_FN
*);
1295 iptcp_status
= RpcServerUseProtseqEp(iptcp
, 20, port
, NULL
);
1296 ok(iptcp_status
== RPC_S_OK
, "RpcServerUseProtseqEp(ncacn_ip_tcp) failed with status %ld\n", iptcp_status
);
1297 np_status
= RpcServerUseProtseqEp(np
, 0, pipe
, NULL
);
1298 ok(np_status
== RPC_S_OK
, "RpcServerUseProtseqEp(ncacn_np) failed with status %ld\n", np_status
);
1300 pRpcServerRegisterIfEx
= (void *)GetProcAddress(GetModuleHandle("rpcrt4.dll"), "RpcServerRegisterIfEx");
1301 if (pRpcServerRegisterIfEx
)
1303 trace("Using RpcServerRegisterIfEx\n");
1304 status
= pRpcServerRegisterIfEx(s_IServer_v0_0_s_ifspec
, NULL
, NULL
,
1305 RPC_IF_ALLOW_CALLBACKS_WITH_NO_AUTH
,
1306 RPC_C_LISTEN_MAX_CALLS_DEFAULT
, NULL
);
1309 status
= RpcServerRegisterIf(s_IServer_v0_0_s_ifspec
, NULL
, NULL
);
1310 ok(status
== RPC_S_OK
, "RpcServerRegisterIf failed with status %ld\n", status
);
1311 status
= RpcServerListen(1, 20, TRUE
);
1312 ok(status
== RPC_S_OK
, "RpcServerListen failed with status %ld\n", status
);
1313 stop_event
= CreateEvent(NULL
, FALSE
, FALSE
, NULL
);
1314 ok(stop_event
!= NULL
, "CreateEvent failed with error %d\n", GetLastError());
1316 if (iptcp_status
== RPC_S_OK
)
1317 run_client("tcp_basic");
1319 skip("tcp_basic tests skipped due to earlier failure\n");
1321 if (np_status
== RPC_S_OK
)
1322 run_client("np_basic");
1325 skip("np_basic tests skipped due to earlier failure\n");
1326 /* np client is what signals stop_event, so bail out if we didn't run do it */
1330 ret
= WaitForSingleObject(stop_event
, 1000);
1331 ok(WAIT_OBJECT_0
== ret
, "WaitForSingleObject\n");
1332 /* if the stop event didn't fire then RpcMgmtWaitServerListen will wait
1333 * forever, so don't bother calling it in this case */
1334 if (ret
== WAIT_OBJECT_0
)
1336 status
= RpcMgmtWaitServerListen();
1338 ok(status
== RPC_S_OK
, "RpcMgmtWaitServerListening failed with status %ld\n", status
);
1348 InitFunctionPointers();
1350 argc
= winetest_get_mainargs(&argv
);
1361 trace("Exception %d\n", RpcExceptionCode());