1 /* Unit test suite for Ntdll Port API functions
3 * Copyright 2006 James Hawkins
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2.1 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #define WIN32_NO_STATUS
30 #include "wine/test.h"
33 #ifndef __WINE_WINTERNL_H
35 typedef struct _CLIENT_ID
39 } CLIENT_ID
, *PCLIENT_ID
;
41 typedef struct _LPC_SECTION_WRITE
49 } LPC_SECTION_WRITE
, *PLPC_SECTION_WRITE
;
51 typedef struct _LPC_SECTION_READ
56 } LPC_SECTION_READ
, *PLPC_SECTION_READ
;
58 typedef struct _LPC_MESSAGE
63 USHORT VirtualRangesOffset
;
67 UCHAR Data
[ANYSIZE_ARRAY
];
68 } LPC_MESSAGE
, *PLPC_MESSAGE
;
72 /* Types of LPC messages */
73 #define UNUSED_MSG_TYPE 0
76 #define LPC_DATAGRAM 3
77 #define LPC_LOST_REPLY 4
78 #define LPC_PORT_CLOSED 5
79 #define LPC_CLIENT_DIED 6
80 #define LPC_EXCEPTION 7
81 #define LPC_DEBUG_EVENT 8
82 #define LPC_ERROR_EVENT 9
83 #define LPC_CONNECTION_REQUEST 10
85 static const WCHAR PORTNAME
[] = {'\\','M','y','P','o','r','t',0};
87 #define REQUEST1 "Request1"
88 #define REQUEST2 "Request2"
91 #define MAX_MESSAGE_LEN 30
94 static char selfname
[MAX_PATH
];
98 /* Function pointers for ntdll calls */
99 static HMODULE hntdll
= 0;
100 static NTSTATUS (WINAPI
*pNtCompleteConnectPort
)(HANDLE
);
101 static NTSTATUS (WINAPI
*pNtAcceptConnectPort
)(PHANDLE
,ULONG
,PLPC_MESSAGE
,ULONG
,
102 ULONG
,PLPC_SECTION_READ
);
103 static NTSTATUS (WINAPI
*pNtReplyPort
)(HANDLE
,PLPC_MESSAGE
);
104 static NTSTATUS (WINAPI
*pNtReplyWaitReceivePort
)(PHANDLE
,PULONG
,PLPC_MESSAGE
,
106 static NTSTATUS (WINAPI
*pNtCreatePort
)(PHANDLE
,POBJECT_ATTRIBUTES
,ULONG
,ULONG
,ULONG
);
107 static NTSTATUS (WINAPI
*pNtRequestWaitReplyPort
)(HANDLE
,PLPC_MESSAGE
,PLPC_MESSAGE
);
108 static NTSTATUS (WINAPI
*pNtRequestPort
)(HANDLE
,PLPC_MESSAGE
);
109 static NTSTATUS (WINAPI
*pNtRegisterThreadTerminatePort
)(HANDLE
);
110 static NTSTATUS (WINAPI
*pNtConnectPort
)(PHANDLE
,PUNICODE_STRING
,
111 PSECURITY_QUALITY_OF_SERVICE
,
112 PLPC_SECTION_WRITE
,PLPC_SECTION_READ
,
114 static NTSTATUS (WINAPI
*pRtlInitUnicodeString
)(PUNICODE_STRING
,LPCWSTR
);
115 static NTSTATUS (WINAPI
*pNtWaitForSingleObject
)(HANDLE
,BOOLEAN
,PLARGE_INTEGER
);
117 static BOOL
init_function_ptrs(void)
119 hntdll
= LoadLibraryA("ntdll.dll");
124 pNtCompleteConnectPort
= (void *)GetProcAddress(hntdll
, "NtCompleteConnectPort");
125 pNtAcceptConnectPort
= (void *)GetProcAddress(hntdll
, "NtAcceptConnectPort");
126 pNtReplyPort
= (void *)GetProcAddress(hntdll
, "NtReplyPort");
127 pNtReplyWaitReceivePort
= (void *)GetProcAddress(hntdll
, "NtReplyWaitReceivePort");
128 pNtCreatePort
= (void *)GetProcAddress(hntdll
, "NtCreatePort");
129 pNtRequestWaitReplyPort
= (void *)GetProcAddress(hntdll
, "NtRequestWaitReplyPort");
130 pNtRequestPort
= (void *)GetProcAddress(hntdll
, "NtRequestPort");
131 pNtRegisterThreadTerminatePort
= (void *)GetProcAddress(hntdll
, "NtRegisterThreadTerminatePort");
132 pNtConnectPort
= (void *)GetProcAddress(hntdll
, "NtConnectPort");
133 pRtlInitUnicodeString
= (void *)GetProcAddress(hntdll
, "RtlInitUnicodeString");
134 pNtWaitForSingleObject
= (void *)GetProcAddress(hntdll
, "NtWaitForSingleObject");
136 if (!pNtCompleteConnectPort
|| !pNtAcceptConnectPort
||
137 !pNtReplyWaitReceivePort
|| !pNtCreatePort
|| !pNtRequestWaitReplyPort
||
138 !pNtRequestPort
|| !pNtRegisterThreadTerminatePort
||
139 !pNtConnectPort
|| !pRtlInitUnicodeString
)
148 static void ProcessConnectionRequest(PLPC_MESSAGE LpcMessage
, PHANDLE pAcceptPortHandle
)
152 ok(LpcMessage
->MessageType
== LPC_CONNECTION_REQUEST
,
153 "Expected LPC_CONNECTION_REQUEST, got %d\n", LpcMessage
->MessageType
);
154 ok(!*LpcMessage
->Data
, "Expected empty string!\n");
156 status
= pNtAcceptConnectPort(pAcceptPortHandle
, 0, LpcMessage
, 1, 0, NULL
);
157 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
159 status
= pNtCompleteConnectPort(*pAcceptPortHandle
);
160 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
163 static void ProcessLpcRequest(HANDLE PortHandle
, PLPC_MESSAGE LpcMessage
)
167 ok(LpcMessage
->MessageType
== LPC_REQUEST
,
168 "Expected LPC_REQUEST, got %d\n", LpcMessage
->MessageType
);
169 ok(!lstrcmp((LPSTR
)LpcMessage
->Data
, REQUEST2
),
170 "Expected %s, got %s\n", REQUEST2
, LpcMessage
->Data
);
172 lstrcpy((LPSTR
)LpcMessage
->Data
, REPLY
);
174 status
= pNtReplyPort(PortHandle
, LpcMessage
);
175 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
176 ok(LpcMessage
->MessageType
== LPC_REQUEST
,
177 "Expected LPC_REQUEST, got %d\n", LpcMessage
->MessageType
);
178 ok(!lstrcmp((LPSTR
)LpcMessage
->Data
, REPLY
),
179 "Expected %s, got %s\n", REPLY
, LpcMessage
->Data
);
182 static DWORD WINAPI
test_ports_client(LPVOID arg
)
184 SECURITY_QUALITY_OF_SERVICE sqos
;
185 LPC_MESSAGE
*LpcMessage
, *out
;
190 sqos
.Length
= sizeof(SECURITY_QUALITY_OF_SERVICE
);
191 sqos
.ImpersonationLevel
= SecurityImpersonation
;
192 sqos
.ContextTrackingMode
= SECURITY_STATIC_TRACKING
;
193 sqos
.EffectiveOnly
= TRUE
;
195 status
= pNtConnectPort(&PortHandle
, &port
, &sqos
, 0, 0, &len
, NULL
, NULL
);
196 todo_wine
ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
197 if (status
!= STATUS_SUCCESS
) return 1;
199 status
= pNtRegisterThreadTerminatePort(PortHandle
);
200 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
202 size
= FIELD_OFFSET(LPC_MESSAGE
, Data
) + MAX_MESSAGE_LEN
;
203 LpcMessage
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, size
);
204 out
= HeapAlloc(GetProcessHeap(), 0, size
);
206 LpcMessage
->DataSize
= lstrlen(REQUEST1
) + 1;
207 LpcMessage
->MessageSize
= FIELD_OFFSET(LPC_MESSAGE
, Data
) + LpcMessage
->DataSize
;
208 lstrcpy((LPSTR
)LpcMessage
->Data
, REQUEST1
);
210 status
= pNtRequestPort(PortHandle
, LpcMessage
);
211 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
212 ok(LpcMessage
->MessageType
== 0, "Expected 0, got %d\n", LpcMessage
->MessageType
);
213 ok(!lstrcmp((LPSTR
)LpcMessage
->Data
, REQUEST1
),
214 "Expected %s, got %s\n", REQUEST1
, LpcMessage
->Data
);
216 /* Fill in the message */
217 memset(LpcMessage
, 0, size
);
218 LpcMessage
->DataSize
= lstrlen(REQUEST2
) + 1;
219 LpcMessage
->MessageSize
= FIELD_OFFSET(LPC_MESSAGE
, Data
) + LpcMessage
->DataSize
;
220 lstrcpy((LPSTR
)LpcMessage
->Data
, REQUEST2
);
222 /* Send the message and wait for the reply */
223 status
= pNtRequestWaitReplyPort(PortHandle
, LpcMessage
, out
);
224 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
225 ok(!lstrcmp((LPSTR
)out
->Data
, REPLY
), "Expected %s, got %s\n", REPLY
, out
->Data
);
226 ok(out
->MessageType
== LPC_REPLY
, "Expected LPC_REPLY, got %d\n", out
->MessageType
);
228 HeapFree(GetProcessHeap(), 0, out
);
229 HeapFree(GetProcessHeap(), 0, LpcMessage
);
234 static void test_ports_server(void)
236 OBJECT_ATTRIBUTES obj
;
238 HANDLE AcceptPortHandle
;
239 PLPC_MESSAGE LpcMessage
;
244 pRtlInitUnicodeString(&port
, PORTNAME
);
246 memset(&obj
, 0, sizeof(OBJECT_ATTRIBUTES
));
247 obj
.Length
= sizeof(OBJECT_ATTRIBUTES
);
248 obj
.ObjectName
= &port
;
250 status
= pNtCreatePort(&PortHandle
, &obj
, 100, 100, 0);
251 if (status
== STATUS_ACCESS_DENIED
)
253 skip("Not enough rights\n");
258 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d\n", status
);
260 if (status
!= STATUS_SUCCESS
) return;
262 size
= FIELD_OFFSET(LPC_MESSAGE
, Data
) + MAX_MESSAGE_LEN
;
263 LpcMessage
= HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY
, size
);
267 status
= pNtReplyWaitReceivePort(PortHandle
, NULL
, NULL
, LpcMessage
);
270 ok(status
== STATUS_SUCCESS
, "Expected STATUS_SUCCESS, got %d(%x)\n", status
, status
);
272 /* STATUS_INVALID_HANDLE: win2k without admin rights will perform an
275 if ((status
== STATUS_NOT_IMPLEMENTED
) ||
276 (status
== STATUS_INVALID_HANDLE
)) return;
278 switch (LpcMessage
->MessageType
)
280 case LPC_CONNECTION_REQUEST
:
281 ProcessConnectionRequest(LpcMessage
, &AcceptPortHandle
);
285 ProcessLpcRequest(PortHandle
, LpcMessage
);
290 ok(!lstrcmp((LPSTR
)LpcMessage
->Data
, REQUEST1
),
291 "Expected %s, got %s\n", REQUEST1
, LpcMessage
->Data
);
294 case LPC_CLIENT_DIED
:
295 ok(done
, "Expected LPC request to be completed!\n");
296 HeapFree(GetProcessHeap(), 0, LpcMessage
);
300 ok(FALSE
, "Unexpected message: %d\n", LpcMessage
->MessageType
);
305 HeapFree(GetProcessHeap(), 0, LpcMessage
);
313 if (!init_function_ptrs())
316 myARGC
= winetest_get_mainargs(&myARGV
);
317 strcpy(selfname
, myARGV
[0]);
319 thread
= CreateThread(NULL
, 0, test_ports_client
, NULL
, 0, &id
);
320 ok(thread
!= NULL
, "Expected non-NULL thread handle!\n");