103 lines
2.2 KiB
C
103 lines
2.2 KiB
C
|
/***
|
||
|
|
|
||
|
|
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;
|
||
|
|
|
||
|
|
}
|