/*++ Copyright (c) 1991-1993 Microsoft Corporation Module Name: RdrFsCtl.c Abstract: NetpRdrFsControlTree performs an FSCTL (file system control) operation on a given tree connection name. Author: John Rogers (JohnRo) 26-Mar-1991 Environment: Only runs under NT; has an NT-specific interface (with Win32 types). Requires ANSI C extensions: slash-slash comments, long external names. Revision History: 26-Mar-91 JohnRo Created. 02-Apr-1991 JohnRo Moved NetpRdrFsControlTree to . Use IF_DEBUG and NetpNtStatusToApiStatus(). 10-Apr-1991 JohnRo Various changes suggested by LINT. 16-Apr-1991 JohnRo Added a little more debug output. 07-May-1991 JohnRo Implement UNICODE. Avoid NET_API_FUNCTION. 14-Nov-1991 JohnRo RAID 4407: "NET VIEW" to an NT server gives 2140. Made changes suggested by PC-LINT. Use more FORMAT_ equates. Display unexpected create file error even if trace off. 21-Nov-1991 JohnRo Removed NT dependencies to reduce recompiles. 22-Sep-1992 JohnRo RAID 6739: Browser too slow when not logged into browsed domain. 21-Jun-1993 JohnRo RAID 14180: NetServerEnum never returns (alignment bug in RxpConvertDataStructures). Also quiet some debug output if other machine just isn't there. Added tree name to unexpected error debug messages. Made changes suggested by PC-LINT 5.0 Use NetpKdPrint() where possible. Use PREFIX_ equates. --*/ // These must be included first: #include // IN, etc. (Needed by ntddnfs.h and others.) #include // LPVOID, etc. #include // NET_API_STATUS, etc. // These may be included in any order: #include // IF_DEBUG(). #include // NERR_Success, etc. #include // NetpIsRemoteNameValid(). #include // FORMAT_NTSTATUS, NetpKdPrint(), etc. #include // NetpMemoryAllocate(). #include // My prototype. #include // DD_NFS_DEVICE_NAME, EA_NAME_ equates, etc. #include // NtFsControlFile(). #include // Rtl APIs. #include // NT_SUCCESS(), STATUS_PENDING, etc. #include // PREFIX_ equates. #include // STRCAT(), STRCPY(), STRLEN(). #include // USE_IPC... #include // ALIGN_xxx NET_API_STATUS NetpRdrFsControlTree( IN LPTSTR TreeName, IN LPTSTR TransportName OPTIONAL, IN DWORD ConnectionType, IN DWORD FsControlCode, IN LPVOID SecurityDescriptor OPTIONAL, IN LPVOID InputBuffer OPTIONAL, IN DWORD InputBufferSize, OUT LPVOID OutputBuffer OPTIONAL, IN DWORD OutputBufferSize, IN BOOL NoPermissionRequired ) /*++ Routine Description: NetpRdrFsControlTree performs a given FSCTL (file system control) on a given tree connection name. Arguments: TreeName - Remote name to do fsctl to (in \\server\share format). FsControlCode - function code to pass to the redirector. These are defined in . SecurityDescriptor - optionally points to a security descriptor to be used when creating the tree connection. InputBuffer - optionally points to a structure to be passed to the redirector. InputBufferSize - size of InputBuffer in bytes; must be zero if InputBuffer is a NULL pointer. OutputBuffer - optionally points to a structure to be filled in by the redirector. OutputBufferSize - size of OutputBuffer in bytes; must be zero if OutputBuffer is a NULL pointer. NoPermissionRequired - TRUE if this is a no permission required API. (I.e. TRUE if the null session may be used.) Return Value: NET_API_STATUS --*/ { NET_API_STATUS ApiStatus; IO_STATUS_BLOCK iosb; NTSTATUS ntstatus; // Status from NT operations. OBJECT_ATTRIBUTES objattrTreeConn; // Attrs for tree conn. LPTSTR pszTreeConn; // See strTreeConn below. #ifndef UNICODE STRING strTreeConn; // \Device\LanManRedir\server\share #endif UNICODE_STRING ucTreeConn; HANDLE TreeConnHandle; PFILE_FULL_EA_INFORMATION EaBuffer = NULL; PFILE_FULL_EA_INFORMATION Ea; USHORT TransportNameSize = 0; ULONG EaBufferSize = 0; PWSTR UnicodeTransportName = NULL; UCHAR EaNameDomainNameSize = (UCHAR) (ROUND_UP_COUNT( strlen(EA_NAME_DOMAIN) + sizeof(CHAR), ALIGN_WCHAR ) - sizeof(CHAR)); UCHAR EaNamePasswordSize = (UCHAR) (ROUND_UP_COUNT( strlen(EA_NAME_PASSWORD) + sizeof(CHAR), ALIGN_WCHAR ) - sizeof(CHAR)); UCHAR EaNameTransportNameSize = (UCHAR) (ROUND_UP_COUNT( strlen(EA_NAME_TRANSPORT) + sizeof(CHAR), ALIGN_WCHAR ) - sizeof(CHAR)); UCHAR EaNameTypeSize = (UCHAR) (ROUND_UP_COUNT( strlen(EA_NAME_TYPE) + sizeof(CHAR), ALIGN_DWORD ) - sizeof(CHAR)); UCHAR EaNameUserNameSize = (UCHAR) (ROUND_UP_COUNT( strlen(EA_NAME_USERNAME) + sizeof(CHAR), ALIGN_WCHAR ) - sizeof(CHAR)); USHORT TypeSize = sizeof(ULONG); IF_DEBUG(RDRFSCTL) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: entered, TreeName='" FORMAT_LPTSTR "', " FORMAT_LPTSTR " session.\n", TreeName, NoPermissionRequired ? TEXT("null") : TEXT("non-null") )); } if ((TreeName == NULL) || (TreeName[0] == 0)) { return (ERROR_INVALID_PARAMETER); } if (! NetpIsRemoteNameValid(TreeName)) { return (ERROR_INVALID_PARAMETER); } // // Build NT-style name for what we're connecting to. Note that there is // NOT a pair of backslashes anywhere in this name. // { DWORD NameSize = // /Device/LanManRedirector / server/share \0 #ifdef UNICODE ( ( STRLEN((LPTSTR)DD_NFS_DEVICE_NAME_U) + 1 + STRLEN(TreeName) + 1 ) ) #else ( ( STRLEN((LPTSTR)DD_NFS_DEVICE_NAME) + 1 + STRLEN(TreeName) + 1 ) ) #endif * sizeof(TCHAR); pszTreeConn = (LPTSTR)NetpMemoryAllocate( NameSize ); } if (pszTreeConn == NULL) { return (ERROR_NOT_ENOUGH_MEMORY); } // // Build the tree connect name. // #ifdef UNICODE (void) STRCPY(pszTreeConn, (LPTSTR) DD_NFS_DEVICE_NAME_U); #else (void) STRCPY(pszTreeConn, (LPTSTR) DD_NFS_DEVICE_NAME); #endif // // NOTE: We add 1, (not sizeof(TCHAR)) because pointer arithmetic is done // in terms of multiples of sizeof(*pointer), not bytes // (void) STRCAT(pszTreeConn, TreeName+1); // \server\share #ifdef UNICODE RtlInitUnicodeString(&ucTreeConn, pszTreeConn); #else RtlInitString( & strTreeConn, pszTreeConn); (void) RtlOemStringToUnicodeString(&ucTreeConn, &strTreeConn, TRUE); #endif IF_DEBUG(RDRFSCTL) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: UNICODE name is " FORMAT_LPWSTR ".\n", ucTreeConn.Buffer )); } // // Calculate the number of bytes needed for the EA buffer. // This may have the transport name. For regular sessions, the user // name, password, and domain name are implicit. For null sessions, we // must give 0-len user name, 0-len password, and 0-len domain name. // if (ARGUMENT_PRESENT(TransportName)) { ASSERT(ConnectionType == USE_IPC); #ifdef UNICODE UnicodeTransportName = TransportName; #else UnicodeTransportName = NetpAllocWStrFromStr(TransportName); if (UnicodeTransportName == NULL) { NetpMemoryFree(pszTreeConn); return ERROR_NOT_ENOUGH_MEMORY; } #endif TransportNameSize = (USHORT) (wcslen(UnicodeTransportName) * sizeof(WCHAR)); EaBufferSize += ROUND_UP_COUNT( FIELD_OFFSET(FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameTransportNameSize + sizeof(CHAR) + TransportNameSize, ALIGN_DWORD ); } if (NoPermissionRequired) { // Domain name (0-len). EaBufferSize += ROUND_UP_COUNT( FIELD_OFFSET(FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameDomainNameSize + sizeof(CHAR), ALIGN_DWORD ); // Password (0-len). EaBufferSize += ROUND_UP_COUNT( FIELD_OFFSET(FILE_FULL_EA_INFORMATION, EaName[0]) + EaNamePasswordSize + sizeof(CHAR), ALIGN_DWORD ); // User name (0-len). EaBufferSize += ROUND_UP_COUNT( FIELD_OFFSET(FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameUserNameSize + sizeof(CHAR), ALIGN_DWORD ); } EaBufferSize += ((ULONG)FIELD_OFFSET( FILE_FULL_EA_INFORMATION, EaName[0]))+ EaNameTypeSize + sizeof(CHAR) + TypeSize; // // Allocate the EA buffer // if ((EaBuffer = NetpMemoryAllocate( EaBufferSize )) == NULL) { NetpMemoryFree(pszTreeConn); #ifndef UNICODE if (UnicodeTransportName != NULL) { NetpMemoryFree(UnicodeTransportName); } #endif return ERROR_NOT_ENOUGH_MEMORY; } // // Fill-in the EA buffer. // RtlZeroMemory(EaBuffer, EaBufferSize); Ea = EaBuffer; if (ARGUMENT_PRESENT(TransportName)) { // // Copy the EA name into EA buffer. EA name length does not // include the zero terminator. // strcpy(Ea->EaName, EA_NAME_TRANSPORT); Ea->EaNameLength = EaNameTransportNameSize; // // Copy the EA value into EA buffer. EA value length does not // include the zero terminator. // (VOID) wcscpy( (LPWSTR) &(Ea->EaName[EaNameTransportNameSize + sizeof(CHAR)]), UnicodeTransportName ); Ea->EaValueLength = TransportNameSize; Ea->NextEntryOffset = ROUND_UP_COUNT( FIELD_OFFSET( FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameTransportNameSize + sizeof(CHAR) + TransportNameSize, ALIGN_DWORD ); Ea->Flags = 0; (ULONG) Ea += Ea->NextEntryOffset; } if (NoPermissionRequired) { // // *** DOMAIN NAME *** // Copy the EA name into EA buffer. EA name length does not // include the zero terminator. // (VOID) strcpy(Ea->EaName, EA_NAME_DOMAIN); Ea->EaNameLength = EaNameDomainNameSize; Ea->EaValueLength = 0; // There is no EA value for this. Ea->NextEntryOffset = ROUND_UP_COUNT( FIELD_OFFSET( FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameDomainNameSize + sizeof(CHAR), ALIGN_DWORD ); Ea->Flags = 0; (ULONG) Ea += Ea->NextEntryOffset; // // *** PASSWORD *** // Copy the EA name into EA buffer. EA name length does not // include the zero terminator. // (VOID) strcpy(Ea->EaName, EA_NAME_PASSWORD); Ea->EaNameLength = EaNamePasswordSize; Ea->EaValueLength = 0; // There is no EA value for this. Ea->NextEntryOffset = ROUND_UP_COUNT( FIELD_OFFSET( FILE_FULL_EA_INFORMATION, EaName[0]) + EaNamePasswordSize + sizeof(CHAR), ALIGN_DWORD ); Ea->Flags = 0; (ULONG) Ea += Ea->NextEntryOffset; // // *** USER NAME *** // Copy the EA name into EA buffer. EA name length does not // include the zero terminator. // (VOID) strcpy(Ea->EaName, EA_NAME_USERNAME); Ea->EaNameLength = EaNameUserNameSize; Ea->EaValueLength = 0; // There is no EA value for this. Ea->NextEntryOffset = ROUND_UP_COUNT( FIELD_OFFSET( FILE_FULL_EA_INFORMATION, EaName[0]) + EaNameUserNameSize + sizeof(CHAR), ALIGN_DWORD ); Ea->Flags = 0; (ULONG) Ea += Ea->NextEntryOffset; } // // Copy the EA for the connection type name into EA buffer. EA name length // does not include the zero terminator. // strcpy(Ea->EaName, EA_NAME_TYPE); Ea->EaNameLength = EaNameTypeSize; *((PULONG) &(Ea->EaName[EaNameTypeSize + sizeof(CHAR)])) = ConnectionType; Ea->EaValueLength = TypeSize; Ea->NextEntryOffset = 0; Ea->Flags = 0; // Set object attributes for the tree conn. InitializeObjectAttributes( & objattrTreeConn, // obj attr to init (LPVOID) & ucTreeConn, // string to use OBJ_CASE_INSENSITIVE, // Attributes NULL, // Root directory SecurityDescriptor); // Security Descriptor // // Open a tree connection to the remote server. // IF_DEBUG(RDRFSCTL) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: opening " FORMAT_LPTSTR ".\n", pszTreeConn )); } ntstatus = NtCreateFile( & TreeConnHandle, // ptr to handle SYNCHRONIZE // desired... | GENERIC_READ | GENERIC_WRITE, // ...access & objattrTreeConn, // name & attributes & iosb, // I/O status block. NULL, // alloc size. FILE_ATTRIBUTE_NORMAL, // (ignored) FILE_SHARE_READ | FILE_SHARE_WRITE, // ...access FILE_OPEN_IF, // create disposition FILE_CREATE_TREE_CONNECTION // create... | FILE_SYNCHRONOUS_IO_NONALERT, // ...options EaBuffer, // EA buffer EaBufferSize ); // Ea buffer size #ifndef UNICODE RtlFreeUnicodeString( &ucTreeConn ); #endif if (! NT_SUCCESS(ntstatus)) { #ifndef UNICODE if (UnicodeTransportName != NULL) { NetpMemoryFree(UnicodeTransportName); } #endif if (EaBuffer != NULL) { NetpMemoryFree(EaBuffer); } NetpMemoryFree(pszTreeConn); ApiStatus = NetpNtStatusToApiStatus(ntstatus); if (ApiStatus == ERROR_BAD_NET_NAME) { ApiStatus = NERR_BadTransactConfig; // Special meaning if no IPC$ } if (ApiStatus != ERROR_BAD_NETPATH) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: unexpected create error,\n" " tree name='" FORMAT_LPTSTR "', " "ntstatus=" FORMAT_NTSTATUS ",\n" " iosb.Status=" FORMAT_NTSTATUS ", " "iosb.Info=" FORMAT_HEX_ULONG ", " " returning " FORMAT_API_STATUS ".\n", TreeName, ntstatus, iosb.Status, iosb.Information, ApiStatus )); } return (ApiStatus); } // Do the FSCTL. IF_DEBUG(RDRFSCTL) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: doing fsctl...\n" )); } ntstatus = NtFsControlFile( TreeConnHandle, // handle NULL, // no event NULL, // no APC routine NULL, // no APC context & iosb, // I/O stat blk (set) FsControlCode, // func code InputBuffer, InputBufferSize, OutputBuffer, OutputBufferSize); if (! NT_SUCCESS(ntstatus)) { NTSTATUS Status; Status = NtClose(TreeConnHandle); #if DBG if (!NT_SUCCESS(Status)) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: " "Unexpected error closing tree connect handle: " FORMAT_NTSTATUS "\n", Status )); } #endif NetpMemoryFree(pszTreeConn); #ifndef UNICODE if (UnicodeTransportName != NULL) { NetpMemoryFree(UnicodeTransportName); } #endif if (EaBuffer != NULL) { NetpMemoryFree(EaBuffer); } ApiStatus = NetpNtStatusToApiStatus(ntstatus); NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: unexpected FSCTL error,\n" " tree name='" FORMAT_LPTSTR "', " "ntstatus=" FORMAT_NTSTATUS ".\n" " ApiStatus=" FORMAT_API_STATUS ", " "iosb.Status=" FORMAT_NTSTATUS ", " "iosb.Info=" FORMAT_HEX_ULONG ".\n", TreeName, ntstatus, ApiStatus, iosb.Status, iosb.Information )); return (ApiStatus); } // Clean up. ntstatus = NtClose(TreeConnHandle); #if DBG if (!NT_SUCCESS(ntstatus)) { NetpKdPrint(( PREFIX_NETLIB "NetpRdrFsControlTree: " "Unexpected error closing tree connect handle: " FORMAT_NTSTATUS "\n", ntstatus )); } #endif NetpMemoryFree(pszTreeConn); #ifndef UNICODE if (UnicodeTransportName != NULL) { NetpMemoryFree(UnicodeTransportName); } #endif if (EaBuffer != NULL) { NetpMemoryFree(EaBuffer); } return (NERR_Success); } // NetpRdrFsControlTree