port.c File Reference

#include <stdio.h>
#include <stdarg.h>
#include "ntstatus.h"
#include "windef.h"
#include "winbase.h"
#include "winuser.h"
#include "winreg.h"
#include "winnls.h"
#include "wine/test.h"
#include "winternl.h"

Include dependency graph for port.c:

Go to the source code of this file.

Classes
struct	_CLIENT_ID
struct	_LPC_SECTION_WRITE
struct	_LPC_SECTION_READ
struct	_LPC_MESSAGE
struct	LPC_MESSAGE64
union	lpc_message

Macros
#define	WIN32_NO_STATUS
#define	UNUSED_MSG_TYPE   0
#define	LPC_REQUEST   1
#define	LPC_REPLY   2
#define	LPC_DATAGRAM   3
#define	LPC_LOST_REPLY   4
#define	LPC_PORT_CLOSED   5
#define	LPC_CLIENT_DIED   6
#define	LPC_EXCEPTION   7
#define	LPC_DEBUG_EVENT   8
#define	LPC_ERROR_EVENT   9
#define	LPC_CONNECTION_REQUEST   10
#define	REQUEST1   "Request1"
#define	REQUEST2   "Request2"
#define	REPLY   "Reply"
#define	MAX_MESSAGE_LEN   30

Typedefs
typedef struct _CLIENT_ID	CLIENT_ID
typedef struct _CLIENT_ID *	PCLIENT_ID
typedef struct _LPC_SECTION_WRITE	LPC_SECTION_WRITE
typedef struct _LPC_SECTION_WRITE *	PLPC_SECTION_WRITE
typedef struct _LPC_SECTION_READ	LPC_SECTION_READ
typedef struct _LPC_SECTION_READ *	PLPC_SECTION_READ
typedef struct _LPC_MESSAGE	LPC_MESSAGE
typedef struct _LPC_MESSAGE *	PLPC_MESSAGE

Functions
static	NTSTATUS (WINAPI *pNtCompleteConnectPort)(HANDLE)
static	BOOL (WINAPI *pIsWow64Process)(HANDLE
static BOOL	init_function_ptrs (void)
static void	ProcessConnectionRequest (union lpc_message *LpcMessage, PHANDLE pAcceptPortHandle)
static void	ProcessLpcRequest (HANDLE PortHandle, union lpc_message *LpcMessage)
static DWORD WINAPI	test_ports_client (LPVOID arg)
static void	test_ports_server (HANDLE PortHandle)
	START_TEST (port)

Variables
static const WCHAR	PORTNAME [] = {'\\','M','y','P','o','r','t',0}
static UNICODE_STRING	port
static HMODULE	hntdll = 0
static	ULONG
static	PULONG
static	POBJECT_ATTRIBUTES
static	PUNICODE_STRING
static	PSECURITY_QUALITY_OF_SERVICE
static	PVOID
static	LPCWSTR
static	PBOOL
static BOOL	is_wow64

Macro Definition Documentation

LPC_CLIENT_DIED

#define LPC_CLIENT_DIED   6

Definition at line 98 of file port.c.

LPC_CONNECTION_REQUEST

#define LPC_CONNECTION_REQUEST   10

Definition at line 102 of file port.c.

LPC_DATAGRAM

#define LPC_DATAGRAM   3

Definition at line 95 of file port.c.

LPC_DEBUG_EVENT

#define LPC_DEBUG_EVENT   8

Definition at line 100 of file port.c.

LPC_ERROR_EVENT

#define LPC_ERROR_EVENT   9

Definition at line 101 of file port.c.

LPC_EXCEPTION

#define LPC_EXCEPTION   7

Definition at line 99 of file port.c.

LPC_LOST_REPLY

#define LPC_LOST_REPLY   4

Definition at line 96 of file port.c.

LPC_PORT_CLOSED

#define LPC_PORT_CLOSED   5

Definition at line 97 of file port.c.

LPC_REPLY

#define LPC_REPLY   2

Definition at line 94 of file port.c.

LPC_REQUEST

#define LPC_REQUEST   1

Definition at line 93 of file port.c.

MAX_MESSAGE_LEN

#define MAX_MESSAGE_LEN   30

Definition at line 110 of file port.c.

REPLY

#define REPLY   "Reply"

Definition at line 108 of file port.c.

REQUEST1

#define REQUEST1   "Request1"

Definition at line 106 of file port.c.

REQUEST2

#define REQUEST2   "Request2"

Definition at line 107 of file port.c.

UNUSED_MSG_TYPE

#define UNUSED_MSG_TYPE   0

Definition at line 92 of file port.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 24 of file port.c.

Typedef Documentation

CLIENT_ID

typedef struct _CLIENT_ID CLIENT_ID

LPC_MESSAGE

typedef struct _LPC_MESSAGE LPC_MESSAGE

LPC_SECTION_READ

typedef struct _LPC_SECTION_READ LPC_SECTION_READ

LPC_SECTION_WRITE

typedef struct _LPC_SECTION_WRITE LPC_SECTION_WRITE

PCLIENT_ID

typedef struct _CLIENT_ID * PCLIENT_ID

PLPC_MESSAGE

static PLPC_MESSAGE

Definition at line 117 of file port.c.

PLPC_SECTION_READ

static PLPC_SECTION_READ

Definition at line 118 of file port.c.

PLPC_SECTION_WRITE

static PLPC_SECTION_WRITE

Definition at line 118 of file port.c.

Function Documentation

BOOL()

static BOOL ( WINAPI *  pIsWow64Process )

static

init_function_ptrs()

static BOOL init_function_ptrs ( void  )

static

Definition at line 135 of file port.c.

{
    hntdll = LoadLibraryA("ntdll.dll");
 
    if (!hntdll)
        return FALSE;
 
    pNtCompleteConnectPort = (void *)GetProcAddress(hntdll, "NtCompleteConnectPort");
    pNtAcceptConnectPort = (void *)GetProcAddress(hntdll, "NtAcceptConnectPort");
    pNtReplyPort = (void *)GetProcAddress(hntdll, "NtReplyPort");
    pNtReplyWaitReceivePort = (void *)GetProcAddress(hntdll, "NtReplyWaitReceivePort");
    pNtCreatePort = (void *)GetProcAddress(hntdll, "NtCreatePort");
    pNtRequestWaitReplyPort = (void *)GetProcAddress(hntdll, "NtRequestWaitReplyPort");
    pNtRequestPort = (void *)GetProcAddress(hntdll, "NtRequestPort");
    pNtRegisterThreadTerminatePort = (void *)GetProcAddress(hntdll, "NtRegisterThreadTerminatePort");
    pNtConnectPort = (void *)GetProcAddress(hntdll, "NtConnectPort");
    pRtlInitUnicodeString = (void *)GetProcAddress(hntdll, "RtlInitUnicodeString");
 
    if (!pNtCompleteConnectPort || !pNtAcceptConnectPort ||
        !pNtReplyWaitReceivePort || !pNtCreatePort || !pNtRequestWaitReplyPort ||
        !pNtRequestPort || !pNtRegisterThreadTerminatePort ||
        !pNtConnectPort || !pRtlInitUnicodeString)
    {
        win_skip("Needed port functions are not available\n");
        FreeLibrary(hntdll);
        return FALSE;
    }
 
    pIsWow64Process = (void *)GetProcAddress(GetModuleHandleA("kernel32.dll"), "IsWow64Process");
    if (!pIsWow64Process || !pIsWow64Process( GetCurrentProcess(), &is_wow64 )) is_wow64 = FALSE;
    return TRUE;
}

Referenced by START_TEST().

NTSTATUS()

static NTSTATUS ( WINAPI *  pNtCompleteConnectPort )

static

ProcessConnectionRequest()

static void ProcessConnectionRequest ( union lpc_message *  LpcMessage, PHANDLE  pAcceptPortHandle  )

static

Definition at line 168 of file port.c.

{
    NTSTATUS status;
 
    if (is_wow64)
    {
        ok(LpcMessage->msg64.MessageType == LPC_CONNECTION_REQUEST,
           "Expected LPC_CONNECTION_REQUEST, got %d\n", LpcMessage->msg64.MessageType);
        ok(!*LpcMessage->msg64.Data, "Expected empty string!\n");
    }
    else
    {
        ok(LpcMessage->msg.MessageType == LPC_CONNECTION_REQUEST,
           "Expected LPC_CONNECTION_REQUEST, got %d\n", LpcMessage->msg.MessageType);
        ok(!*LpcMessage->msg.Data, "Expected empty string!\n");
    }
 
    status = pNtAcceptConnectPort(pAcceptPortHandle, 0, &LpcMessage->msg, 1, NULL, NULL);
    ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
    
    status = pNtCompleteConnectPort(*pAcceptPortHandle);
    ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
}

Referenced by test_ports_server().

ProcessLpcRequest()

static void ProcessLpcRequest ( HANDLE  PortHandle, union lpc_message *  LpcMessage  )

static

Definition at line 192 of file port.c.

{
    NTSTATUS status;
 
    if (is_wow64)
    {
        ok(LpcMessage->msg64.MessageType == LPC_REQUEST,
           "Expected LPC_REQUEST, got %d\n", LpcMessage->msg64.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg64.Data, REQUEST2),
           "Expected %s, got %s\n", REQUEST2, LpcMessage->msg64.Data);
        strcpy((LPSTR)LpcMessage->msg64.Data, REPLY);
 
        status = pNtReplyPort(PortHandle, &LpcMessage->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(LpcMessage->msg64.MessageType == LPC_REQUEST,
           "Expected LPC_REQUEST, got %d\n", LpcMessage->msg64.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg64.Data, REPLY),
           "Expected %s, got %s\n", REPLY, LpcMessage->msg64.Data);
    }
    else
    {
        ok(LpcMessage->msg.MessageType == LPC_REQUEST,
           "Expected LPC_REQUEST, got %d\n", LpcMessage->msg.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg.Data, REQUEST2),
           "Expected %s, got %s\n", REQUEST2, LpcMessage->msg.Data);
        strcpy((LPSTR)LpcMessage->msg.Data, REPLY);
 
        status = pNtReplyPort(PortHandle, &LpcMessage->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(LpcMessage->msg.MessageType == LPC_REQUEST,
           "Expected LPC_REQUEST, got %d\n", LpcMessage->msg.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg.Data, REPLY),
           "Expected %s, got %s\n", REPLY, LpcMessage->msg.Data);
    }
}

Referenced by test_ports_server().

START_TEST()

START_TEST

(

port

)

Definition at line 370 of file port.c.

{
    OBJECT_ATTRIBUTES obj;
    HANDLE port_handle;
    NTSTATUS status;
 
    if (!init_function_ptrs())
        return;
 
    pRtlInitUnicodeString(&port, PORTNAME);
 
    memset(&obj, 0, sizeof(OBJECT_ATTRIBUTES));
    obj.Length = sizeof(OBJECT_ATTRIBUTES);
    obj.ObjectName = &port;
 
    status = pNtCreatePort(&port_handle, &obj, 100, 100, 0);
    if (status == STATUS_ACCESS_DENIED) skip("Not enough rights\n");
    else todo_wine ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %d\n", status);
 
    if (status == STATUS_SUCCESS)
    {
        DWORD id;
        HANDLE thread = CreateThread(NULL, 0, test_ports_client, NULL, 0, &id);
        ok(thread != NULL, "Expected non-NULL thread handle!\n");
 
        test_ports_server( port_handle );
        ok( WaitForSingleObject( thread, 10000 ) == 0, "thread didn't exit\n" );
        CloseHandle(thread);
    }
    FreeLibrary(hntdll);
}

test_ports_client()

static DWORD WINAPI test_ports_client ( LPVOID  arg )

static

Definition at line 228 of file port.c.

{
    SECURITY_QUALITY_OF_SERVICE sqos;
    union lpc_message *LpcMessage, *out;
    HANDLE PortHandle;
    ULONG len, size;
    NTSTATUS status;
 
    sqos.Length = sizeof(SECURITY_QUALITY_OF_SERVICE);
    sqos.ImpersonationLevel = SecurityImpersonation;
    sqos.ContextTrackingMode = SECURITY_STATIC_TRACKING;
    sqos.EffectiveOnly = TRUE;
 
    status = pNtConnectPort(&PortHandle, &port, &sqos, 0, 0, &len, NULL, NULL);
    todo_wine ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
    if (status != STATUS_SUCCESS) return 1;
 
    status = pNtRegisterThreadTerminatePort(PortHandle);
    ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
 
    if (is_wow64)
    {
        size = FIELD_OFFSET(LPC_MESSAGE64, Data[MAX_MESSAGE_LEN]);
        LpcMessage = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, size);
        out = HeapAlloc(GetProcessHeap(), 0, size);
 
        LpcMessage->msg64.DataSize = strlen(REQUEST1) + 1;
        LpcMessage->msg64.MessageSize = FIELD_OFFSET(LPC_MESSAGE64, Data[LpcMessage->msg64.DataSize]);
        strcpy((LPSTR)LpcMessage->msg64.Data, REQUEST1);
 
        status = pNtRequestPort(PortHandle, &LpcMessage->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(LpcMessage->msg64.MessageType == 0, "Expected 0, got %d\n", LpcMessage->msg64.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg64.Data, REQUEST1),
           "Expected %s, got %s\n", REQUEST1, LpcMessage->msg64.Data);
 
        /* Fill in the message */
        memset(LpcMessage, 0, size);
        LpcMessage->msg64.DataSize = strlen(REQUEST2) + 1;
        LpcMessage->msg64.MessageSize = FIELD_OFFSET(LPC_MESSAGE64, Data[LpcMessage->msg64.DataSize]);
        strcpy((LPSTR)LpcMessage->msg64.Data, REQUEST2);
 
        /* Send the message and wait for the reply */
        status = pNtRequestWaitReplyPort(PortHandle, &LpcMessage->msg, &out->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(!strcmp((LPSTR)out->msg64.Data, REPLY), "Expected %s, got %s\n", REPLY, out->msg64.Data);
        ok(out->msg64.MessageType == LPC_REPLY, "Expected LPC_REPLY, got %d\n", out->msg64.MessageType);
    }
    else
    {
        size = FIELD_OFFSET(LPC_MESSAGE, Data[MAX_MESSAGE_LEN]);
        LpcMessage = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, size);
        out = HeapAlloc(GetProcessHeap(), 0, size);
 
        LpcMessage->msg.DataSize = strlen(REQUEST1) + 1;
        LpcMessage->msg.MessageSize = FIELD_OFFSET(LPC_MESSAGE, Data[LpcMessage->msg.DataSize]);
        strcpy((LPSTR)LpcMessage->msg.Data, REQUEST1);
 
        status = pNtRequestPort(PortHandle, &LpcMessage->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(LpcMessage->msg.MessageType == 0, "Expected 0, got %d\n", LpcMessage->msg.MessageType);
        ok(!strcmp((LPSTR)LpcMessage->msg.Data, REQUEST1),
           "Expected %s, got %s\n", REQUEST1, LpcMessage->msg.Data);
 
        /* Fill in the message */
        memset(LpcMessage, 0, size);
        LpcMessage->msg.DataSize = strlen(REQUEST2) + 1;
        LpcMessage->msg.MessageSize = FIELD_OFFSET(LPC_MESSAGE, Data[LpcMessage->msg.DataSize]);
        strcpy((LPSTR)LpcMessage->msg.Data, REQUEST2);
 
        /* Send the message and wait for the reply */
        status = pNtRequestWaitReplyPort(PortHandle, &LpcMessage->msg, &out->msg);
        ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %x\n", status);
        ok(!strcmp((LPSTR)out->msg.Data, REPLY), "Expected %s, got %s\n", REPLY, out->msg.Data);
        ok(out->msg.MessageType == LPC_REPLY, "Expected LPC_REPLY, got %d\n", out->msg.MessageType);
    }
 
    HeapFree(GetProcessHeap(), 0, out);
    HeapFree(GetProcessHeap(), 0, LpcMessage);
 
    return 0;
}

Referenced by START_TEST().

test_ports_server()

static void test_ports_server ( HANDLE  PortHandle )

static

Definition at line 311 of file port.c.

{
    HANDLE AcceptPortHandle;
    union lpc_message *LpcMessage;
    ULONG size;
    NTSTATUS status;
    BOOL done = FALSE;
 
    size = FIELD_OFFSET(LPC_MESSAGE, Data) + MAX_MESSAGE_LEN;
    LpcMessage = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, size);
 
    while (TRUE)
    {
        status = pNtReplyWaitReceivePort(PortHandle, NULL, NULL, &LpcMessage->msg);
        todo_wine
        {
            ok(status == STATUS_SUCCESS, "Expected STATUS_SUCCESS, got %d(%x)\n", status, status);
        }
        /* STATUS_INVALID_HANDLE: win2k without admin rights will perform an
         *                        endless loop here
         */
        if ((status == STATUS_NOT_IMPLEMENTED) ||
            (status == STATUS_INVALID_HANDLE)) return;
 
        switch (is_wow64 ? LpcMessage->msg64.MessageType : LpcMessage->msg.MessageType)
        {
            case LPC_CONNECTION_REQUEST:
                ProcessConnectionRequest(LpcMessage, &AcceptPortHandle);
                break;
 
            case LPC_REQUEST:
                ProcessLpcRequest(PortHandle, LpcMessage);
                done = TRUE;
                break;
 
            case LPC_DATAGRAM:
                if (is_wow64)
                    ok(!strcmp((LPSTR)LpcMessage->msg64.Data, REQUEST1),
                       "Expected %s, got %s\n", REQUEST1, LpcMessage->msg64.Data);
                else
                    ok(!strcmp((LPSTR)LpcMessage->msg.Data, REQUEST1),
                       "Expected %s, got %s\n", REQUEST1, LpcMessage->msg.Data);
                break;
 
            case LPC_CLIENT_DIED:
                ok(done, "Expected LPC request to be completed!\n");
                HeapFree(GetProcessHeap(), 0, LpcMessage);
                return;
 
            default:
                ok(FALSE, "Unexpected message: %d\n",
                   is_wow64 ? LpcMessage->msg64.MessageType : LpcMessage->msg.MessageType);
                break;
        }
    }
 
    HeapFree(GetProcessHeap(), 0, LpcMessage);
}

Referenced by START_TEST().

Variable Documentation

hntdll

HMODULE hntdll = 0

static

Definition at line 115 of file port.c.

Referenced by init_function_ptrs(), and START_TEST().

is_wow64

BOOL is_wow64

static

Definition at line 133 of file port.c.

Referenced by init_function_ptrs(), ProcessConnectionRequest(), ProcessLpcRequest(), test_ports_client(), and test_ports_server().

LPCWSTR

LPCWSTR

Definition at line 130 of file port.c.

PBOOL

PBOOL

Definition at line 131 of file port.c.

POBJECT_ATTRIBUTES

POBJECT_ATTRIBUTES

Definition at line 122 of file port.c.

port

UNICODE_STRING port

static

Definition at line 112 of file port.c.

PORTNAME

const WCHAR PORTNAME[] = {'\\','M','y','P','o','r','t',0}

static

Definition at line 104 of file port.c.

Referenced by START_TEST().

PSECURITY_QUALITY_OF_SERVICE

PSECURITY_QUALITY_OF_SERVICE

Definition at line 127 of file port.c.

PULONG

PULONG

Definition at line 120 of file port.c.

PUNICODE_STRING

PUNICODE_STRING

Definition at line 126 of file port.c.

PVOID

PVOID

Definition at line 129 of file port.c.

ULONG

ULONG

Definition at line 117 of file port.c.