NpfsConnect.c File Reference

#include <kmt_test.h>
#include "npfs.h"

Include dependency graph for NpfsConnect.c:

Go to the source code of this file.

Macros
#define	MAX_INSTANCES   5
#define	IN_QUOTA   4096
#define	OUT_QUOTA   4096
#define	CheckServer(ServerHandle, State)
#define	CheckClient(ClientHandle, State)

Functions
static VOID	ConnectPipe (IN OUT PTHREAD_CONTEXT Context)
static VOID	ListenPipe (IN OUT PTHREAD_CONTEXT Context)
static BOOLEAN	CheckConnectPipe (IN PTHREAD_CONTEXT Context, IN PCWSTR PipePath, IN ULONG MilliSeconds)
static BOOLEAN	CheckListenPipe (IN PTHREAD_CONTEXT Context, IN HANDLE ServerHandle, IN ULONG MilliSeconds)
static VOID	TestConnect (IN HANDLE ServerHandle, IN PCWSTR PipePath)
static VOID NTAPI	RunTest (IN PVOID Context)
	START_TEST (NpfsConnect)

Variables
static KSTART_ROUTINE	RunTest

Macro Definition Documentation

CheckClient

#define CheckClient	(		ClientHandle,
			State
	)

Value:

    NpCheckClientPipe(ClientHandle,                             \
                      BYTE_STREAM, QUEUE, BYTE_STREAM, DUPLEX,  \
                      MAX_INSTANCES, 1,                         \
                      IN_QUOTA, 0,                              \
                      OUT_QUOTA, OUT_QUOTA,                     \
                      State)

Definition at line 23 of file NpfsConnect.c.

CheckServer

#define CheckServer	(		ServerHandle,
			State
	)

Value:

    NpCheckServerPipe(ServerHandle,                             \
                      BYTE_STREAM, QUEUE, BYTE_STREAM, DUPLEX,  \
                      MAX_INSTANCES, 1,                         \
                      IN_QUOTA, 0,                              \
                      OUT_QUOTA, OUT_QUOTA,                     \
                      State)

Definition at line 15 of file NpfsConnect.c.

IN_QUOTA

#define IN_QUOTA   4096

Definition at line 12 of file NpfsConnect.c.

MAX_INSTANCES

#define MAX_INSTANCES   5

Definition at line 11 of file NpfsConnect.c.

OUT_QUOTA

#define OUT_QUOTA   4096

Definition at line 13 of file NpfsConnect.c.

Function Documentation

CheckConnectPipe()

static BOOLEAN CheckConnectPipe ( IN PTHREAD_CONTEXT  Context, IN PCWSTR  PipePath, IN ULONG  MilliSeconds  )

static

Definition at line 55 of file NpfsConnect.c.

{
    Context->Work = ConnectPipe;
    Context->Connect.PipePath = PipePath;
    return TriggerWork(Context, MilliSeconds);
}

Referenced by TestConnect().

CheckListenPipe()

static BOOLEAN CheckListenPipe ( IN PTHREAD_CONTEXT  Context, IN HANDLE  ServerHandle, IN ULONG  MilliSeconds  )

static

Definition at line 67 of file NpfsConnect.c.

{
    Context->Work = ListenPipe;
    Context->Listen.ServerHandle = ServerHandle;
    return TriggerWork(Context, MilliSeconds);
}

Referenced by TestConnect().

ConnectPipe()

static VOID ConnectPipe ( IN OUT PTHREAD_CONTEXT  Context )

static

Definition at line 33 of file NpfsConnect.c.

{
    HANDLE ClientHandle;
 
    ClientHandle = NULL;
    Context->Connect.Status = NpOpenPipe(&ClientHandle,
                                         Context->Connect.PipePath,
                                         FILE_PIPE_FULL_DUPLEX);
    Context->Connect.ClientHandle = ClientHandle;
}

Referenced by CheckConnectPipe().

ListenPipe()

static VOID ListenPipe ( IN OUT PTHREAD_CONTEXT  Context )

static

Definition at line 47 of file NpfsConnect.c.

{
    Context->Listen.Status = NpListenPipe(Context->Listen.ServerHandle);
}

Referenced by CheckListenPipe().

RunTest()

static VOID NTAPI RunTest ( IN PVOID  Context )

static

Definition at line 242 of file NpfsConnect.c.

{
    NTSTATUS Status;
    HANDLE ServerHandle;
 
    UNREFERENCED_PARAMETER(Context);
 
    ServerHandle = INVALID_HANDLE_VALUE;
    Status = NpCreatePipe(&ServerHandle,
                          DEVICE_NAMED_PIPE L"\\KmtestNpfsConnectTestPipe",
                          BYTE_STREAM, QUEUE, BYTE_STREAM, DUPLEX,
                          MAX_INSTANCES,
                          IN_QUOTA,
                          OUT_QUOTA);
    ok_eq_hex(Status, STATUS_SUCCESS);
    ok(ServerHandle != NULL && ServerHandle != INVALID_HANDLE_VALUE, "ServerHandle = %p\n", ServerHandle);
    if (!skip(NT_SUCCESS(Status) && ServerHandle != NULL && ServerHandle != INVALID_HANDLE_VALUE, "No pipe\n"))
    {
        CheckServer(ServerHandle, FILE_PIPE_LISTENING_STATE);
        TestConnect(ServerHandle, DEVICE_NAMED_PIPE L"\\KmtestNpfsConnectTestPipe");
    }
}

START_TEST()

START_TEST

(

NpfsConnect

)

Definition at line 266 of file NpfsConnect.c.

{
    PKTHREAD Thread;
 
    Thread = KmtStartThread(RunTest, NULL);
    KmtFinishThread(Thread, NULL);
}

TestConnect()

static VOID TestConnect ( IN HANDLE  ServerHandle, IN PCWSTR  PipePath  )

static

Definition at line 79 of file NpfsConnect.c.

{
    NTSTATUS Status;
    THREAD_CONTEXT ConnectContext;
    THREAD_CONTEXT ListenContext;
    BOOLEAN Okay;
    HANDLE ClientHandle;
 
    StartWorkerThread(&ConnectContext);
    StartWorkerThread(&ListenContext);
 
    /* Server should start out listening */
    CheckServer(ServerHandle, FILE_PIPE_LISTENING_STATE);
 
    /* Connect a client */
    ClientHandle = NULL;
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_SUCCESS);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
    {
        ClientHandle = ConnectContext.Connect.ClientHandle;
        CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    }
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Connect another client */
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_PIPE_NOT_AVAILABLE);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
        ObCloseHandle(ConnectContext.Connect.ClientHandle, KernelMode);
    CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Disconnecting the client should fail */
    Status = NpDisconnectPipe(ClientHandle);
    ok_eq_hex(Status, STATUS_ILLEGAL_FUNCTION);
    CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Listening on the client should fail */
    Status = NpListenPipe(ClientHandle);
    ok_eq_hex(Status, STATUS_ILLEGAL_FUNCTION);
    CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Close client */
    if (ClientHandle)
        ObCloseHandle(ClientHandle, KernelMode);
    CheckServer(ServerHandle, FILE_PIPE_CLOSING_STATE);
 
    /* Connecting a client now should fail */
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_PIPE_NOT_AVAILABLE);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
        ObCloseHandle(ConnectContext.Connect.ClientHandle, KernelMode);
    CheckServer(ServerHandle, FILE_PIPE_CLOSING_STATE);
 
    /* Listening should fail */
    Okay = CheckListenPipe(&ListenContext, ServerHandle, 100);
    ok_bool_true(Okay, "CheckListenPipe returned");
    if (!skip(Okay, "Listen succeeded unexpectedly\n"))
        CheckServer(ServerHandle, FILE_PIPE_CLOSING_STATE);
 
    /* Disconnect server */
    Status = NpDisconnectPipe(ServerHandle);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckServer(ServerHandle, FILE_PIPE_DISCONNECTED_STATE);
 
    /* Disconnecting again should fail */
    Status = NpDisconnectPipe(ServerHandle);
    ok_eq_hex(Status, STATUS_PIPE_DISCONNECTED);
    CheckServer(ServerHandle, FILE_PIPE_DISCONNECTED_STATE);
 
    /* Connecting a client now should fail */
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_PIPE_NOT_AVAILABLE);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
        ObCloseHandle(ConnectContext.Connect.ClientHandle, KernelMode);
    CheckServer(ServerHandle, FILE_PIPE_DISCONNECTED_STATE);
 
    /**************************************************************************/
    /* Now listen again */
    Okay = CheckListenPipe(&ListenContext, ServerHandle, 100);
    ok_bool_false(Okay, "CheckListenPipe returned");
    //blocks: CheckServer(ServerHandle, FILE_PIPE_LISTENING_STATE);
 
    /* Connect client */
    ClientHandle = NULL;
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_SUCCESS);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
    {
        ClientHandle = ConnectContext.Connect.ClientHandle;
        CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    }
    Okay = WaitForWork(&ListenContext, 100);
    ok_bool_true(Okay, "WaitForWork returned");
    ok_eq_hex(ListenContext.Listen.Status, STATUS_SUCCESS);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Listening again should fail */
    Okay = CheckListenPipe(&ListenContext, ServerHandle, 100);
    ok_bool_true(Okay, "CheckListenPipe returned");
    ok_eq_hex(ListenContext.Listen.Status, STATUS_PIPE_CONNECTED);
    CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Disconnect server */
    Status = NpDisconnectPipe(ServerHandle);
    ok_eq_hex(Status, STATUS_SUCCESS);
    NpQueryPipe(ClientHandle, STATUS_PIPE_DISCONNECTED);
    CheckServer(ServerHandle, FILE_PIPE_DISCONNECTED_STATE);
 
    /* Close client */
    if (ClientHandle)
        ObCloseHandle(ClientHandle, KernelMode);
    CheckServer(ServerHandle, FILE_PIPE_DISCONNECTED_STATE);
 
    /**************************************************************************/
    /* Listen once more */
    Okay = CheckListenPipe(&ListenContext, ServerHandle, 100);
    ok_bool_false(Okay, "CheckListenPipe returned");
    //blocks: CheckServer(ServerHandle, FILE_PIPE_LISTENING_STATE);
 
    /* Connect client */
    ClientHandle = NULL;
    Okay = CheckConnectPipe(&ConnectContext, PipePath, 100);
    ok_bool_true(Okay, "CheckConnectPipe returned");
    ok_eq_hex(ConnectContext.Connect.Status, STATUS_SUCCESS);
    if (NT_SUCCESS(ConnectContext.Connect.Status))
    {
        ClientHandle = ConnectContext.Connect.ClientHandle;
        CheckClient(ClientHandle, FILE_PIPE_CONNECTED_STATE);
    }
    Okay = WaitForWork(&ListenContext, 100);
    ok_bool_true(Okay, "WaitForWork returned");
    ok_eq_hex(ListenContext.Listen.Status, STATUS_SUCCESS);
    CheckServer(ServerHandle, FILE_PIPE_CONNECTED_STATE);
 
    /* Close server */
    Status = ObCloseHandle(ServerHandle, KernelMode);
    ok_eq_hex(Status, STATUS_SUCCESS);
    CheckClient(ClientHandle, FILE_PIPE_CLOSING_STATE);
 
    /* Close client */
    if (ClientHandle)
        ObCloseHandle(ClientHandle, KernelMode);
 
    FinishWorkerThread(&ListenContext);
    FinishWorkerThread(&ConnectContext);
}

Variable Documentation

RunTest

KSTART_ROUTINE RunTest

static

Definition at line 238 of file NpfsConnect.c.

Referenced by main(), CWineTest::Run(), and START_TEST().