summaryrefslogtreecommitdiffstats
path: root/private/nw/ndsutils/getps.c
blob: a0634bad67b4bfc286f89398d1eed3e0a0d6bfeb (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/***

Copyright (c) 1995  Microsoft Corporation

Module Name:

    GetPs.c

Abstract:

    Command line test for getting the preferred server.

Author:

    Cory West       [corywest] 14-Nov-95

***/

#include "ndsapi32.h"

int
_cdecl main(
    int argc,
    char **argv
) {

    NTSTATUS ntstatus;
    IO_STATUS_BLOCK IoStatusBlock;
    OBJECT_ATTRIBUTES ObjectAttributes;
    ACCESS_MASK DesiredAccess = SYNCHRONIZE | FILE_LIST_DIRECTORY;
    HANDLE hRdr;

    WCHAR OpenString[] = L"\\Device\\Nwrdr\\*";
    UNICODE_STRING OpenName;

    BYTE Reply[64];

    //
    // Check the arguments.
    //

    if ( argc != 1 ) {
        printf( "Usage: getps\n" );
        printf( "Retrieves the current preferred server.\n" );
        return -1;
    }

    //
    // Set up the object attributes.
    //

    RtlInitUnicodeString( &OpenName, OpenString );

    InitializeObjectAttributes( &ObjectAttributes,
                                &OpenName,
                                OBJ_CASE_INSENSITIVE,
                                NULL,
                                NULL );

    ntstatus = NtOpenFile( &hRdr,
                           DesiredAccess,
                           &ObjectAttributes,
                           &IoStatusBlock,
                           FILE_SHARE_VALID_FLAGS,
                           FILE_SYNCHRONOUS_IO_NONALERT );

    if ( !NT_SUCCESS(ntstatus) )
        return ntstatus;

    //
    // Call the nwrdr.
    //

    ntstatus = NtFsControlFile( hRdr,
                                NULL,
                                NULL,
                                NULL,
                                &IoStatusBlock,
                                FSCTL_NWR_GET_PREFERRED_SERVER,
                                NULL,
                                0,
                                (PVOID) Reply,
                                sizeof( Reply ) );

    if ( !NT_SUCCESS( ntstatus ) ) {
        printf( "No preferred server is currently set.\n" );
        goto ExitWithClose;
    }

    //
    // On success the output buffer contains a UNICODE_STRING
    // with the string packed in afterwards.
    //

    printf( "Preferred Server: %wZ\n", (PUNICODE_STRING) Reply );

ExitWithClose:

   NtClose( hRdr );
   return ntstatus;

}