4 Copyright (C) Stefan Metzmacher 2008
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program 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
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "midltests.h"
27 #error "please run 'vcvarsall.bat amd64' -midltests_tcp needs 64-bit support!"
30 #define MIDLTESTS_C_CODE 1
31 #include "midltests.idl"
34 #define LISTEN_IP "127.0.0.1"
38 #define FORWARD_IP "127.0.0.1"
42 #define CONNECT_IP "127.0.0.1"
45 struct NDRTcpThreadCtx
;
47 struct NDRProxyThreadCtx
{
48 const struct NDRTcpThreadCtx
*ctx
;
55 struct NDRTcpThreadCtx
{
63 struct dcerpc_header
{
64 BYTE rpc_vers
; /* RPC version */
65 BYTE rpc_vers_minor
; /* Minor version */
66 BYTE ptype
; /* Packet type */
67 BYTE pfc_flags
; /* Fragmentation flags */
68 BYTE drep
[4]; /* NDR data representation */
69 short frag_length
; /* Total length of fragment */
70 short auth_length
; /* authenticator length */
71 DWORD call_id
; /* Call identifier */
74 static void dump_packet(const char *ctx
, const char *direction
,
75 const unsigned char *buf
, int len
)
77 struct dcerpc_header
*hdr
= (struct dcerpc_header
*)buf
;
79 if (len
< sizeof(struct dcerpc_header
)) {
80 printf("%s:%s: invalid dcerpc pdu len(%d)\n",
86 if (hdr
->rpc_vers
!= 5 || hdr
->rpc_vers_minor
!= 0) {
87 printf("%s:%s: invalid dcerpc pdu len(%d) ver:%d min:%d\n",
89 hdr
->rpc_vers
, hdr
->rpc_vers_minor
);
94 if (hdr
->frag_length
!= len
) {
95 printf("%s:%s: invalid dcerpc pdu len(%d) should be (%d)\n",
96 ctx
, direction
, len
, hdr
->frag_length
);
102 switch (hdr
->ptype
) {
103 case 0: /* request */
104 printf("%s:%s: ptype[request] flen[%d] plen[%d] ahint[%d]\n\n",
105 ctx
, direction
, hdr
->frag_length
,
106 len
- 24, *(DWORD
*)(&buf
[0x10]));
107 dump_data(buf
+ 24, len
- 24);
112 case 2: /* response */
113 printf("\n%s:%s: ptype[response] flen[%d] plen[%d] ahint[%d]\n\n",
114 ctx
, direction
, hdr
->frag_length
,
115 len
- 24, *(DWORD
*)(&buf
[0x10]));
116 dump_data(buf
+ 24, len
- 24);
123 printf("%s:%s: ptype[bind] flen[%d] call[%d] contexts[%d]\n\n"
124 ctx
, direction
, hdr
->frag_length
, hdr
->call_id
,
126 dump_data(buf
+ 24, len
- 24);
132 case 12: /* bind ack */
134 printf("%s:%s: ptype[bind_ack] flen[%d] call[%d]\n\n",
135 ctx
, direction
, hdr
->frag_length
, hdr
->call_id
);
140 case 14: /* alter_req */
142 printf("%s:%s: ptype[alter_req] flen[%d] call[%d] contexts[%d]\n\n",
143 ctx
, direction
, hdr
->frag_length
, hdr
->call_id
,
145 //dump_data(buf + 24, len - 24);
151 case 15: /* alter_ack */
153 printf("%s:%s: ptype[alter_ack] flen[%d] call[%d]\n\n",
154 ctx
, direction
, hdr
->frag_length
, hdr
->call_id
);
160 printf("%s:%s: ptype[%d] flen[%d] call[%d]\n\n",
161 ctx
, direction
, hdr
->ptype
, hdr
->frag_length
, hdr
->call_id
);
167 static void change_packet(const char *ctx
, BOOL ndr64
,
168 unsigned char *buf
, int len
)
170 struct dcerpc_header
*hdr
= (struct dcerpc_header
*)buf
;
171 BOOL is_ndr64
= FALSE
;
172 const unsigned char ndr64_buf
[] = {
173 0x33, 0x05, 0x71, 0x71, 0xBA, 0xBE, 0x37, 0x49,
174 0x83, 0x19, 0xB5, 0xDB, 0xEF, 0x9C, 0xCC, 0x36
177 if (len
< sizeof(struct dcerpc_header
)) {
178 printf("%s: invalid dcerpc pdu len(%d)\n",
184 if (hdr
->rpc_vers
!= 5 || hdr
->rpc_vers_minor
!= 0) {
185 printf("%s: invalid dcerpc pdu len(%d) ver:%d min:%d\n",
187 hdr
->rpc_vers
, hdr
->rpc_vers_minor
);
192 if (hdr
->frag_length
!= len
) {
193 printf("%s: invalid dcerpc pdu len(%d) should be (%d)\n",
194 ctx
, len
, hdr
->frag_length
);
199 switch (hdr
->ptype
) {
201 case 14: /* alter_req */
206 ret
= memcmp(&buf
[0x60], ndr64_buf
, 16);
212 if (is_ndr64
&& !ndr64
) {
214 memset(&buf
[0x60], 0xFF, 16);
215 printf("%s: disable NDR64\n\n", ctx
);
216 } else if (!is_ndr64
&& ndr64
) {
217 printf("\n%s: got NDR32 downgrade\n\n", ctx
);
218 #ifndef DONOT_FORCE_NDR64
219 printf("\n\tERROR!!!\n\n");
220 memset(&buf
[0x34], 0xFF, 16);
221 printf("You may need to run 'vcvarsall.bat amd64' before 'nmake tcp'\n");
224 } else if (is_ndr64
) {
225 printf("%s: got NDR64\n\n", ctx
);
227 printf("%s: got NDR32\n\n", ctx
);
229 //printf("%s: bind with %u pres\n", ctx, buf[24]);
235 static int sock_pending(SOCKET s
)
241 ret
= ioctlsocket(s
, FIONREAD
, &value
);
247 /* this should not be reached */
259 * if no data is available check if the socket is in error state. For
260 * dgram sockets it's the way to return ICMP error messages of
261 * connected sockets to the caller.
263 ret
= getsockopt(s
, SOL_SOCKET
, SO_ERROR
, (char *)&error
, &len
);
273 DWORD WINAPI
NDRProxyThread(LPVOID lpParameter
)
275 struct NDRProxyThreadCtx
*p
= (struct NDRProxyThreadCtx
*)lpParameter
;
277 while (!p
->ctx
->stop
) {
284 ret
= sock_pending(p
->InSocket
);
289 r
= recv(p
->InSocket
, buf
, sizeof(buf
), 0);
291 ret
= WSAGetLastError();
292 printf("%s: recv(in) failed[%d][%d]\n", p
->ctx
->name
, r
, ret
);
297 change_packet(p
->ctx
->name
, p
->ctx
->ndr64
, buf
, r
);
300 dump_packet(p
->ctx
->name
, "in => out", buf
, r
);
304 s
= send(p
->OutSocket
, buf
, r
, 0);
306 ret
= WSAGetLastError();
307 printf("%s: send(out) failed[%d][%d]\n", p
->ctx
->name
, s
, ret
);
312 ret
= sock_pending(p
->OutSocket
);
317 r
= recv(p
->OutSocket
, buf
, sizeof(buf
), 0);
319 ret
= WSAGetLastError();
320 printf("%s: recv(out) failed[%d][%d]\n", p
->ctx
->name
, r
, ret
);
325 dump_packet(p
->ctx
->name
, "out => in", buf
, r
);
328 s
= send(p
->InSocket
, buf
, r
, 0);
330 ret
= WSAGetLastError();
331 printf("%s: send(in) failed[%d][%d]\n", p
->ctx
->name
, s
, ret
);
339 closesocket(p
->InSocket
);
340 closesocket(p
->OutSocket
);
342 printf("NDRTcpThread[%s] stop\n", p
->ctx
->name
);
347 DWORD WINAPI
NDRTcpThread(LPVOID lpParameter
)
349 struct NDRTcpThreadCtx
*ctx
= (struct NDRTcpThreadCtx
*)lpParameter
;
352 struct sockaddr_in saServer
;
353 struct sockaddr_in saClient
;
355 //printf("NDRTcpThread[%s] start\n", ctx->name);
358 ListenSocket
= socket(AF_INET
, SOCK_STREAM
, IPPROTO_TCP
);
359 if (ListenSocket
== INVALID_SOCKET
) {
360 ret
= WSAGetLastError();
361 printf("socket() failed[%d][%d]\n", ListenSocket
, ret
);
366 saServer
.sin_family
= AF_INET
;
367 saServer
.sin_addr
.s_addr
= inet_addr(LISTEN_IP
);
368 saServer
.sin_port
= htons(ctx
->listen_port
);
370 saClient
.sin_family
= AF_INET
;
371 saClient
.sin_addr
.s_addr
= inet_addr(FORWARD_IP
);
372 saClient
.sin_port
= htons(ctx
->client_port
);
374 ret
= bind(ListenSocket
, (SOCKADDR
*)&saServer
, sizeof(saServer
));
375 if (ret
== SOCKET_ERROR
) {
376 ret
= WSAGetLastError();
377 printf("bind() failed[%d]\n", ret
);
382 ret
= listen(ListenSocket
, 10);
383 if (ret
== SOCKET_ERROR
) {
384 ret
= WSAGetLastError();
385 printf("listen() failed[%d]\n", ret
);
391 struct sockaddr_in sa
;
392 int sa_len
= sizeof(sa
);
393 struct NDRProxyThreadCtx
*p
= malloc(sizeof(*p
));
396 p
->InSocket
= accept(ListenSocket
, (SOCKADDR
*)&sa
, &sa_len
);
397 if (p
->InSocket
== INVALID_SOCKET
) {
398 ret
= WSAGetLastError();
399 printf("%s: accept() failed[%d][%d]\n", p
->ctx
->name
, p
->InSocket
, ret
);
404 p
->OutSocket
= socket(AF_INET
, SOCK_STREAM
, IPPROTO_TCP
);
405 if (p
->OutSocket
== INVALID_SOCKET
) {
406 ret
= WSAGetLastError();
407 printf("%s: socket(out) failed[%d][%d]\n", p
->ctx
->name
, p
->OutSocket
, ret
);
409 closesocket(p
->InSocket
);
413 ret
= connect(p
->OutSocket
, (SOCKADDR
*)&saClient
, sizeof(saClient
));
414 if (ret
== SOCKET_ERROR
) {
415 ret
= WSAGetLastError();
416 printf("%s: connect() failed[%d]\n", p
->ctx
->name
, ret
);
418 closesocket(p
->InSocket
);
419 closesocket(p
->OutSocket
);
423 p
->hThread
= CreateThread(
424 NULL
, // default security attributes
425 0, // use default stack size
426 NDRProxyThread
, // thread function name
427 p
, // argument to thread function
428 0, // use default creation flags
429 &p
->dwThreadId
);// returns the thread identifier
430 if (p
->hThread
== NULL
) {
431 printf("failed to create thread ndr32\n");
437 //printf("NDRTcpThread[%s] stop\n", ctx->name);
441 printf("NDRTcpThread[%s] failed[%d]\n", ctx
->name
, ret
);
446 struct NDRRpcThreadCtx
{
451 DWORD WINAPI
NDRRpcThread(LPVOID lpParameter
)
453 struct NDRRpcThreadCtx
*ctx
= (struct NDRRpcThreadCtx
*)lpParameter
;
456 RPC_BINDING_VECTOR
*pBindingVector
;
458 #define RPC_MIN_CALLS 1
459 #define RPC_MAX_CALLS 20
461 //printf("NDRRpcThread[%s] start\n", ctx->name);
463 status
= RpcServerUseProtseqEp("ncacn_ip_tcp", RPC_MAX_CALLS
, "5055", NULL
);
465 printf("Failed to register ncacn_ip_tcp endpoint\n");
470 status
= RpcServerInqBindings(&pBindingVector
);
472 printf("Failed RpcServerInqBindings\n");
478 status
= RpcEpRegister(srv_midltests_v0_0_s_ifspec
, pBindingVector
, NULL
, "midltests server");
480 printf("Failed RpcEpRegister\n");
485 status
= RpcServerRegisterIf(srv_midltests_v0_0_s_ifspec
, NULL
, NULL
);
487 printf("Failed to register interface\n");
492 status
= RpcServerListen(RPC_MIN_CALLS
, RPC_MAX_CALLS
, FALSE
);
494 printf("RpcServerListen returned error %d\n", status
);
499 printf("NDRRpcThread[%s] stop\n", ctx
->name
);
503 printf("NDRRpcThread[%s] failed[%d]\n", ctx
->name
, ret
);
508 int main(int argc
, char **argv
)
511 struct NDRTcpThreadCtx ctx_ndr32
;
512 struct NDRTcpThreadCtx ctx_ndr64
;
513 struct NDRRpcThreadCtx ctx_rpc
;
514 DWORD dwThreadIdArray
[3];
515 HANDLE hThreadArray
[3];
516 WORD wVersionRequested
= MAKEWORD(2, 2);
521 ret
= WSAStartup(wVersionRequested
, &wsaData
);
523 printf("WSAStartup failed with error: %d\n", ret
);
528 ctx_ndr32
.name
= "ndr32";
529 ctx_ndr32
.listen_port
= 5032;
530 ctx_ndr32
.client_port
= 5055;
531 ctx_ndr32
.ndr64
= FALSE
;
532 ctx_ndr32
.stop
= FALSE
;
533 hThreadArray
[0] = CreateThread(
534 NULL
, // default security attributes
535 0, // use default stack size
536 NDRTcpThread
, // thread function name
537 &ctx_ndr32
, // argument to thread function
538 0, // use default creation flags
539 &dwThreadIdArray
[0]); // returns the thread identifier
540 if (hThreadArray
[0] == NULL
) {
541 printf("failed to create thread ndr32\n");
546 ctx_ndr64
.name
= "ndr64";
547 ctx_ndr64
.listen_port
= 5064;
548 ctx_ndr64
.client_port
= 5055;
549 ctx_ndr64
.ndr64
= TRUE
;
550 ctx_ndr64
.stop
= FALSE
;
551 hThreadArray
[1] = CreateThread(
552 NULL
, // default security attributes
553 0, // use default stack size
554 NDRTcpThread
, // thread function name
555 &ctx_ndr64
, // argument to thread function
556 0, // use default creation flags
557 &dwThreadIdArray
[1]); // returns the thread identifier
558 if (hThreadArray
[1] == NULL
) {
559 printf("failed to create thread ndr64\n");
564 ctx_rpc
.name
= "rpc";
565 ctx_rpc
.listen_port
= 5050;
566 hThreadArray
[2] = CreateThread(
567 NULL
, // default security attributes
568 0, // use default stack size
569 NDRRpcThread
, // thread function name
570 &ctx_rpc
, // argument to thread function
571 0, // use default creation flags
572 &dwThreadIdArray
[2]); // returns the thread identifier
573 if (hThreadArray
[2] == NULL
) {
574 printf("failed to create thread rpc\n");
579 printf("Wait for setup of server threads\n");
581 ret
= WaitForMultipleObjects(3, hThreadArray
, TRUE
, 3000);
582 if (ret
== WAIT_TIMEOUT
) {
585 printf("Failed to setup of server threads %d:%d\n",
586 ret
, GetLastError());
592 printf("\nTest NDR32\n\n");
594 binding
= "ncacn_ip_tcp:" CONNECT_IP
"[5032]";
595 status
= RpcBindingFromStringBinding(
597 &midltests_IfHandle
);
599 printf("RpcBindingFromStringBinding returned %d\n", status
);
607 ret
= RpcExceptionCode();
608 printf("NDR32 Runtime error 0x%x\n", ret
);
611 ctx_ndr32
.stop
= TRUE
;
615 printf("\nTest NDR64\n\n");
616 binding
= "ncacn_ip_tcp:" CONNECT_IP
"[5064]";
617 status
= RpcBindingFromStringBinding(
619 &midltests_IfHandle
);
621 printf("RpcBindingFromStringBinding returned %d\n", status
);
629 ret
= RpcExceptionCode();
630 printf("Runtime error 0x%x\n", ret
);
633 ctx_ndr64
.stop
= TRUE
;
635 WaitForMultipleObjects(3, hThreadArray
, TRUE
, 2000);
638 printf("\nTest OK\n");
641 printf("\nTest FAILED[%d]\n", ret
);