/*****************************************************************************/ /* Page05NtCreate.cpp Copyright (c) Ladislav Zezula 2004 */ /*---------------------------------------------------------------------------*/ /* Description: */ /*---------------------------------------------------------------------------*/ /* Date Ver Who Comment */ /* -------- ---- --- ------- */ /* 06.03.04 1.00 Lad The first version of Page05NtCreate.cpp */ /*****************************************************************************/ #include "FileTest.h" #include "resource.h" #define FILE_ID_SIZE 16 //----------------------------------------------------------------------------- // Flags TFlagInfo AccessMaskValues[] = { {{"FILE_READ_DATA/FILE_LIST_DIRECTORY"}, FILE_READ_DATA, FILE_READ_DATA}, {{"FILE_WRITE_DATA/FILE_ADD_FILE"}, FILE_WRITE_DATA, FILE_WRITE_DATA}, {{"FILE_APPEND_DATA/FILE_ADD_SUBDIRECTORY"}, FILE_APPEND_DATA, FILE_APPEND_DATA}, {{"FILE_READ_EA"}, FILE_READ_EA, FILE_READ_EA}, {{"FILE_WRITE_EA"}, FILE_WRITE_EA, FILE_WRITE_EA}, {{"FILE_EXECUTE/FILE_TRAVERSE"}, FILE_EXECUTE, FILE_EXECUTE}, {{"FILE_DELETE_CHILD"}, FILE_DELETE_CHILD, FILE_DELETE_CHILD}, {{"FILE_READ_ATTRIBUTES"}, FILE_READ_ATTRIBUTES, FILE_READ_ATTRIBUTES}, {{"FILE_WRITE_ATTRIBUTES"}, FILE_WRITE_ATTRIBUTES, FILE_WRITE_ATTRIBUTES}, FLAGINFO_BITV(DELETE), FLAGINFO_BITV(READ_CONTROL), FLAGINFO_BITV(WRITE_DAC), FLAGINFO_BITV(WRITE_OWNER), FLAGINFO_BITV(SYNCHRONIZE), FLAGINFO_BITV(ACCESS_SYSTEM_SECURITY), FLAGINFO_BITV(GENERIC_READ), FLAGINFO_BITV(GENERIC_WRITE), FLAGINFO_BITV(GENERIC_EXECUTE), FLAGINFO_BITV(GENERIC_ALL), FLAGINFO_BITV(MAXIMUM_ALLOWED), FLAGINFO_END() }; // Also necessary in Page04FileOps.cpp TFlagInfo FileAttributesValues[] = { FLAGINFO_BITV(FILE_ATTRIBUTE_READONLY), FLAGINFO_BITV(FILE_ATTRIBUTE_HIDDEN), FLAGINFO_BITV(FILE_ATTRIBUTE_SYSTEM), FLAGINFO_BITV(OLD_DOS_VOLID), FLAGINFO_BITV(FILE_ATTRIBUTE_DIRECTORY), FLAGINFO_BITV(FILE_ATTRIBUTE_ARCHIVE), FLAGINFO_BITV(FILE_ATTRIBUTE_DEVICE), FLAGINFO_BITV(FILE_ATTRIBUTE_NORMAL), FLAGINFO_BITV(FILE_ATTRIBUTE_TEMPORARY), FLAGINFO_BITV(FILE_ATTRIBUTE_SPARSE_FILE), FLAGINFO_BITV(FILE_ATTRIBUTE_REPARSE_POINT), FLAGINFO_BITV(FILE_ATTRIBUTE_COMPRESSED), FLAGINFO_BITV(FILE_ATTRIBUTE_OFFLINE), FLAGINFO_BITV(FILE_ATTRIBUTE_NOT_CONTENT_INDEXED), FLAGINFO_BITV(FILE_ATTRIBUTE_ENCRYPTED), FLAGINFO_BITV(FILE_ATTRIBUTE_INTEGRITY_STREAM), FLAGINFO_BITV(FILE_ATTRIBUTE_VIRTUAL), FLAGINFO_BITV(FILE_ATTRIBUTE_NO_SCRUB_DATA), FLAGINFO_BITV(FILE_ATTRIBUTE_RECALL_ON_OPEN), FLAGINFO_BITV(FILE_ATTRIBUTE_PINNED), FLAGINFO_BITV(FILE_ATTRIBUTE_UNPINNED), FLAGINFO_BITV(FILE_ATTRIBUTE_RECALL_ON_DATA_ACCESS), FLAGINFO_BITV(FILE_ATTRIBUTE_STRICTLY_SEQUENTIAL), FLAGINFO_END() }; TFlagInfo CreateOptionsValues[] = { FLAGINFO_BITV(FILE_DIRECTORY_FILE), FLAGINFO_BITV(FILE_WRITE_THROUGH), FLAGINFO_BITV(FILE_SEQUENTIAL_ONLY), FLAGINFO_BITV(FILE_NO_INTERMEDIATE_BUFFERING), FLAGINFO_BITV(FILE_SYNCHRONOUS_IO_ALERT), FLAGINFO_BITV(FILE_SYNCHRONOUS_IO_NONALERT), FLAGINFO_BITV(FILE_NON_DIRECTORY_FILE), FLAGINFO_BITV(FILE_CREATE_TREE_CONNECTION), FLAGINFO_BITV(FILE_COMPLETE_IF_OPLOCKED), FLAGINFO_BITV(FILE_NO_EA_KNOWLEDGE), FLAGINFO_BITV(FILE_OPEN_FOR_RECOVERY), FLAGINFO_BITV(FILE_RANDOM_ACCESS), FLAGINFO_BITV(FILE_DELETE_ON_CLOSE), FLAGINFO_BITV(FILE_OPEN_BY_FILE_ID), FLAGINFO_BITV(FILE_OPEN_FOR_BACKUP_INTENT), FLAGINFO_BITV(FILE_NO_COMPRESSION), FLAGINFO_BITV(FILE_OPEN_REQUIRING_OPLOCK), FLAGINFO_BITV(FILE_DISALLOW_EXCLUSIVE), FLAGINFO_BITV(FILE_SESSION_AWARE), FLAGINFO_BITV(FILE_RESERVE_OPFILTER), FLAGINFO_BITV(FILE_OPEN_REPARSE_POINT), FLAGINFO_BITV(FILE_OPEN_NO_RECALL), FLAGINFO_BITV(FILE_OPEN_FOR_FREE_SPACE_QUERY), FLAGINFO_END() }; static TFlagInfo ObjAttrFlagsValues[] = { FLAGINFO_BITV(OBJ_PROTECT_CLOSE), FLAGINFO_BITV(OBJ_INHERIT), FLAGINFO_BITV(OBJ_AUDIT_OBJECT_CLOSE), FLAGINFO_BITV(OBJ_NO_RIGHTS_UPGRADE), FLAGINFO_BITV(OBJ_PERMANENT), FLAGINFO_BITV(OBJ_EXCLUSIVE), FLAGINFO_BITV(OBJ_CASE_INSENSITIVE), FLAGINFO_BITV(OBJ_OPENIF), FLAGINFO_BITV(OBJ_OPENLINK), FLAGINFO_BITV(OBJ_KERNEL_HANDLE), FLAGINFO_BITV(OBJ_FORCE_ACCESS_CHECK), FLAGINFO_BITV(OBJ_IGNORE_IMPERSONATED_DEVICEMAP), FLAGINFO_BITV(OBJ_DONT_REPARSE), FLAGINFO_BITV(OBJ_KERNEL_EXCLUSIVE), FLAGINFO_END() }; //----------------------------------------------------------------------------- // Local functions static NTSTATUS MyCreateDirectory(TFileTestData * pData, POBJECT_ATTRIBUTES pObjAttr, PIO_STATUS_BLOCK pIoStatus) { NTSTATUS Status; HANDLE SaveTransactionHandle = NULL; HANDLE DirectoryHandle = NULL; if(pData->bUseTransaction) { SaveTransactionHandle = pfnRtlGetCurrentTransaction(); pfnRtlSetCurrentTransaction(pData->hTransaction); } Status = NtCreateFile(&DirectoryHandle, FILE_ADD_FILE, pObjAttr, pIoStatus, NULL, FILE_ATTRIBUTE_DIRECTORY, FILE_SHARE_READ, FILE_OPEN_IF, FILE_DIRECTORY_FILE, NULL, 0); if(DirectoryHandle != NULL) NtClose(DirectoryHandle); if(pData->bUseTransaction) { pfnRtlSetCurrentTransaction(SaveTransactionHandle); } // If the directory already exists, take it as success return Status; } static int QuickAccessSelection(HWND hDlg, DWORD dwDesiredAccess, DWORD dwShareAccess, bool bSynchronous) { DWORD dwCreateOptions; // Retrieve the create options from the dialog DlgText2Hex32(hDlg, IDC_CREATE_OPTIONS, &dwCreateOptions); // Fix desired access according to bAsynchronous if(bSynchronous) { dwCreateOptions |= FILE_SYNCHRONOUS_IO_NONALERT; dwDesiredAccess |= SYNCHRONIZE; } else { dwCreateOptions &= ~(FILE_SYNCHRONOUS_IO_ALERT | FILE_SYNCHRONOUS_IO_NONALERT); dwDesiredAccess &= ~SYNCHRONIZE; } // Set or reset DeleteOnClose flag if(dwDesiredAccess & DELETE) dwCreateOptions |= FILE_DELETE_ON_CLOSE; else dwCreateOptions &= ~FILE_DELETE_ON_CLOSE; // Apply the create options to the dialog controls Hex2DlgText32(hDlg, IDC_DESIRED_ACCESS, dwDesiredAccess); Hex2DlgText32(hDlg, IDC_SHARE_ACCESS, dwShareAccess); Hex2DlgText32(hDlg, IDC_CREATE_OPTIONS, dwCreateOptions); return TRUE; } static int SaveDialog(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); HWND hCombo = GetDlgItem(hDlg, IDC_CREATE_DISPOSITION); int nError; GetDlgItemText(hDlg, IDC_DIRECTORY_NAME, pData->szDirName, _countof(pData->szDirName)); GetDlgItemText(hDlg, IDC_FILE_NAME, pData->szFileName1, _countof(pData->szFileName1)); if((nError = DlgText2Hex32(hDlg, IDC_OBJ_ATTR_FLAGS, &pData->dwObjAttrFlags)) != ERROR_SUCCESS) return nError; if((nError = DlgText2Hex32(hDlg, IDC_DESIRED_ACCESS, &pData->dwDesiredAccess)) != ERROR_SUCCESS) return nError; if((nError = DlgText2Hex32(hDlg, IDC_FILE_ATTRIBUTES, &pData->dwFlagsAndAttributes)) != ERROR_SUCCESS) return nError; if((nError = DlgText2Hex32(hDlg, IDC_SHARE_ACCESS, &pData->dwShareAccess)) != ERROR_SUCCESS) return nError; if((nError = DlgText2Hex32(hDlg, IDC_CREATE_OPTIONS, &pData->dwCreateOptions)) != ERROR_SUCCESS) return nError; if((nError = DlgText2Hex64(hDlg, IDC_ALLOCATION_SIZE, &pData->AllocationSize)) != ERROR_SUCCESS) return nError; pData->dwCreateDisposition2 = ComboBox_GetCurSel(hCombo); pData->bUseTransaction = (IsDlgButtonChecked(hDlg, IDC_TRANSACTED) == BST_CHECKED); return ERROR_SUCCESS; } //----------------------------------------------------------------------------- // Message handlers static TAnchors * pAnchors = NULL; static int OnNtCloseClick(HWND hDlg); static int OnInitDialog(HWND hDlg, LPARAM lParam) { TFileTestData * pData; PROPSHEETPAGE * pPage = (PROPSHEETPAGE *)lParam; HWND hCombo = GetDlgItem(hDlg, IDC_CREATE_DISPOSITION); SetDialogData(hDlg, pPage->lParam); pData = (TFileTestData *)pPage->lParam; // Configure dialog resizing if(pData->bEnableResizing) { pAnchors = new TAnchors(); pAnchors->AddAnchor(hDlg, IDC_MAIN_FRAME, akAll); pAnchors->AddAnchor(hDlg, IDC_DIRECTORY_NAME, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_DIRECTORY_NAME_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_FILE_NAME, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_FILE_NAME_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_OBJ_ATTR_FLAGS, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_OBJ_ATTR_FLAGS_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_DESIRED_ACCESS, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_DESIRED_ACCESS_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_ALLOCATION_SIZE, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_ALLOCATION_SIZE_UPDOWN, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_FILE_ATTRIBUTES, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_FILE_ATTRIBUTES_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_SHARE_ACCESS, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_SHARE_ACCESS_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_CREATE_DISPOSITION, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_CREATE_OPTIONS, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_CREATE_OPTIONS_BROWSE, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_EXTENDED_ATTRIBUTES, akLeft | akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_EXTENDED_ATTRIBUTES_EDIT, akTop | akRight); pAnchors->AddAnchor(hDlg, IDC_TRANSACTED, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_VIRTUALIZATION, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_BREAKPOINT, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_PRIVILEGES, akLeft | akBottom); pAnchors->AddAnchor(hDlg, IDC_MAKE_DIRECTORY, akLeft | akBottom); pAnchors->AddAnchor(hDlg, IDC_CREATE_FILE, akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_CLOSE_HANDLE, akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_RESULT_FRAME, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_ERROR_CODE_TITLE, akLeft | akBottom); pAnchors->AddAnchor(hDlg, IDC_ERROR_CODE, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_HANDLE_TITLE, akLeft | akBottom); pAnchors->AddAnchor(hDlg, IDC_HANDLE, akLeft | akRight | akBottom); pAnchors->AddAnchor(hDlg, IDC_INFORMATION_TITLE, akLeft | akBottom); pAnchors->AddAnchor(hDlg, IDC_INFORMATION, akLeft | akRight | akBottom); } // Initialize the "Relative File" hyperlink InitURLButton(hDlg, IDC_RELATIVE_FILE_HELP, FALSE); // Initialize the combo box InitDialogControls(hDlg, MAKEINTRESOURCE(IDD_PAGE02_NTCREATE)); if(hCombo != NULL) ComboBox_SetCurSel(hCombo, pData->dwCreateDisposition2); // If we have a tooltip window, init tooltips g_Tooltip.AddToolTip(hDlg, IDC_OBJ_ATTR_FLAGS, ObjAttrFlagsValues); g_Tooltip.AddToolTip(hDlg, IDC_DESIRED_ACCESS, AccessMaskValues); g_Tooltip.AddToolTip(hDlg, IDC_FILE_ATTRIBUTES, FileAttributesValues); g_Tooltip.AddToolTip(hDlg, IDC_SHARE_ACCESS, ShareAccessValues); g_Tooltip.AddToolTip(hDlg, IDC_CREATE_OPTIONS, CreateOptionsValues); // On post-Vista, enable the virtualization button if(GetTokenVirtualizationEnabled(NULL)) EnableDlgItems(hDlg, TRUE, IDC_VIRTUALIZATION, 0); return TRUE; } static int OnSetActive(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); TCHAR szEaInfo[128]; BOOL bEnabled; int nChecked; // Set directory name and file name SetDlgItemText(hDlg, IDC_DIRECTORY_NAME, pData->szDirName); SetDlgItemText(hDlg, IDC_FILE_NAME, pData->szFileName1); // Convert both to NT name. if(GetWindowTextLength(GetDlgItem(hDlg, IDC_DIRECTORY_NAME)) == 0) ConvertToNtName(hDlg, IDC_FILE_NAME); ConvertToNtName(hDlg, IDC_DIRECTORY_NAME); // Set the various create options Hex2DlgText32(hDlg, IDC_OBJ_ATTR_FLAGS, pData->dwObjAttrFlags); Hex2DlgText32(hDlg, IDC_DESIRED_ACCESS, pData->dwDesiredAccess); Hex2DlgText64(hDlg, IDC_ALLOCATION_SIZE, pData->AllocationSize); Hex2DlgText32(hDlg, IDC_FILE_ATTRIBUTES, pData->dwFlagsAndAttributes); Hex2DlgText32(hDlg, IDC_SHARE_ACCESS, pData->dwShareAccess); Hex2DlgText32(hDlg, IDC_CREATE_OPTIONS, pData->dwCreateOptions); // Update the info about extended attributes rsprintf(szEaInfo, _countof(szEaInfo), IDS_EA_INFO, pData->pFileEa, pData->dwEaSize); SetDlgItemText(hDlg, IDC_EXTENDED_ATTRIBUTES, szEaInfo); // Enable/disable transaction bEnabled = (pfnRtlSetCurrentTransaction != NULL && IsHandleValid(pData->hTransaction)); EnableDlgItems(hDlg, bEnabled, IDC_TRANSACTED, 0); nChecked = (bEnabled && pData->bUseTransaction) ? BST_CHECKED : BST_UNCHECKED; CheckDlgButton(hDlg, IDC_TRANSACTED, nChecked); // Check/uncheck virtualization nChecked = (GetTokenVirtualizationEnabled(&bEnabled) && bEnabled) ? BST_CHECKED : BST_UNCHECKED; CheckDlgButton(hDlg, IDC_VIRTUALIZATION, nChecked); // Enable/disable "NtClose" bEnabled = IsHandleValid(pData->hFile) ? TRUE : FALSE; EnableDlgItems(hDlg, bEnabled, IDC_CLOSE_HANDLE, 0); return TRUE; } static int OnKillActive(HWND hDlg) { SaveDialog(hDlg); return TRUE; } static int OnRelativeFileHelp(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); LPTSTR szBuffer; TCHAR szAccessMask[0x100]; TCHAR szShareAccess[0x100]; TCHAR szOpenOptions[0x100]; TCHAR szMsgFormat[512]; size_t cchBuffer = 0x1000; int nLength = 0; // Load both parts of the message nLength = LoadString(g_hInst, IDS_RELATIVE_FILE_HELP, szMsgFormat, _countof(szMsgFormat)); if(nLength > 0) { // Allocate big buffer for the entire text szBuffer = new TCHAR[cchBuffer]; if(szBuffer != NULL) { // Format the result string StringCchPrintf(szBuffer, cchBuffer, szMsgFormat, pData->dwDesiredAccessRF, FlagsToString(AccessMaskValues, szAccessMask, _countof(szAccessMask), pData->dwDesiredAccessRF, false), pData->dwShareAccessRF, FlagsToString(ShareAccessValues, szShareAccess, _countof(szShareAccess), pData->dwShareAccessRF, false), pData->dwOpenOptionsRF, FlagsToString(CreateOptionsValues, szOpenOptions, _countof(szOpenOptions), pData->dwOpenOptionsRF, false)); // Display the message box MessageBoxRc(hDlg, IDS_INFO, (UINT_PTR)szBuffer); delete [] szBuffer; } } return TRUE; } static int OnBrowseDirClick(HWND hDlg) { BrowseForDirectory(hDlg, MAKEINTRESOURCE(IDC_DIRECTORY_NAME), IDS_SELECT_DIRECTORY, MAX_PATH); ConvertToNtName(hDlg, IDC_DIRECTORY_NAME); return TRUE; } static int OnBrowseFileClick(HWND hDlg) { OPENFILENAME ofn; InitOpenFileName(&ofn); ofn.lpstrFile = MAKEINTRESOURCE(IDC_FILE_NAME); ofn.lpstrTitle = MAKEINTRESOURCE(IDS_SELECT_FILE); ofn.lpstrFilter = MAKEINTRESOURCE(IDS_FILTER_ALL); if(GetOpenFileNameRc(hDlg, &ofn)) { SetDlgItemText(hDlg, IDC_DIRECTORY_NAME, _T("")); ConvertToNtName(hDlg, IDC_FILE_NAME); } return TRUE; } static int OnObjAtributesFlags(HWND hDlg) { FlagsDialog_OnControl(hDlg, IDS_OBJECT_ATTRIBUTES_FLAGS, ObjAttrFlagsValues, IDC_OBJ_ATTR_FLAGS); return TRUE; } static int OnDesiredAccessClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); bool bRelativeFile = (GetAsyncKeyState(VK_SHIFT) < 0); // Shall we edit desired access for relative file? if(bRelativeFile) { FlagsDialog(hDlg, IDS_DESIRED_ACCESS_RF, AccessMaskValues, pData->dwDesiredAccessRF); return TRUE; } // Show the dialog for desired access FlagsDialog_OnControl(hDlg, IDS_DESIRED_ACCESS, AccessMaskValues, IDC_DESIRED_ACCESS); return TRUE; } static int OnFileAttributesClick(HWND hDlg) { FlagsDialog_OnControl(hDlg, IDS_FILE_ATTRIBUTES, FileAttributesValues, IDC_FILE_ATTRIBUTES); return TRUE; } static int OnShareAccessClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); bool bRelativeFile = (GetAsyncKeyState(VK_SHIFT) < 0); // Shall we edit desired access for relative file? if(bRelativeFile) { FlagsDialog(hDlg, IDS_SHARE_ACCESS_RF, ShareAccessValues, pData->dwShareAccessRF); return TRUE; } // Show the dialog for desired access FlagsDialog_OnControl(hDlg, IDS_SHARE_ACCESS, ShareAccessValues, IDC_SHARE_ACCESS); return TRUE; } static int OnCreateOptionsClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); bool bRelativeFile = (GetAsyncKeyState(VK_SHIFT) < 0); // Shall we edit desired access for relative file? if(bRelativeFile) { FlagsDialog(hDlg,IDS_OPEN_OPTIONS_RF, CreateOptionsValues, pData->dwOpenOptionsRF); return TRUE; } // Show the dialog for desired access FlagsDialog_OnControl(hDlg, IDS_CREATE_OPTIONS, CreateOptionsValues, IDC_CREATE_OPTIONS); return TRUE; } static int OnEditEaClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); TCHAR szEaInfo[128]; // Invoke the editor of the extended attributes if(ExtendedAtributesEditorDialog(hDlg, pData) == IDOK) { // Update the info about extended attributes rsprintf(szEaInfo, _countof(szEaInfo), IDS_EA_INFO, pData->pFileEa, pData->dwEaSize); SetDlgItemText(hDlg, IDC_EXTENDED_ATTRIBUTES, szEaInfo); } return TRUE; } static int OnVirtualization(HWND hDlg) { BOOL bEnabled = (IsDlgButtonChecked(hDlg, IDC_VIRTUALIZATION) == BST_CHECKED) ? TRUE : FALSE; SetTokenVirtualizationEnabled(bEnabled); return TRUE; } static int OnMakeDirectoryClick(HWND hDlg) { OBJECT_ATTRIBUTES ObjAttr; IO_STATUS_BLOCK IoStatus = {0, 0}; UNICODE_STRING FileName; TFileTestData * pData = GetDialogData(hDlg); NTSTATUS Status = STATUS_SUCCESS; LPTSTR szDirectory = pData->szDirName; LPTSTR szPathPart = pData->szDirName; LPTSTR szTemp; USHORT SaveLength; // Get the values from dialog controls to the dialog data if(SaveDialog(hDlg) != ERROR_SUCCESS) return FALSE; // Initialize object attributes and the unicode string InitializeObjectAttributes(&ObjAttr, &FileName, OBJ_CASE_INSENSITIVE, NULL, NULL); RtlInitUnicodeString(&FileName, pData->szDirName); SaveLength = FileName.Length; // Now parse the directory as-is, create every sub-directory if(szDirectory[0] != 0) { // Now find the begin of the first directory part szPathPart = FindDirectoryPathPart(szDirectory); if(szPathPart != NULL) { while(szPathPart[0] != 0) { // Find either next backslash or end of string szTemp = FindNextPathSeparator(szPathPart); // Create the directory part FileName.Length = (USHORT)((szTemp - szDirectory) * sizeof(WCHAR)); Status = MyCreateDirectory(pData, &ObjAttr, &IoStatus); if(!NT_SUCCESS(Status)) break; // Go to the next part of the path FileName.Length = SaveLength; szPathPart = szTemp; } } else { Status = MyCreateDirectory(pData, &ObjAttr, &IoStatus); } } else { Status = MyCreateDirectory(pData, &ObjAttr, &IoStatus); } SetResultInfo(hDlg, RSI_NTSTATUS | RSI_NTCREATE, Status, &IoStatus); return TRUE; } static int OnCreateFileClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); OBJECT_ATTRIBUTES ObjAttr; IO_STATUS_BLOCK IoStatus = {0}; UNICODE_STRING FileName; UNICODE_STRING DirName; LARGE_INTEGER AllocationSize; NTSTATUS Status = STATUS_SUCCESS; HANDLE SaveTransactionHandle = NULL; TCHAR szFileName[MAX_NT_PATH]; TCHAR szDirName[MAX_NT_PATH]; DWORD cbObjectID = 0; BYTE ObjectID[0x10]; // Close the handle, if already open if(IsHandleValid(pData->hDirectory) || IsHandleValid(pData->hFile)) OnNtCloseClick(hDlg); // Get the various create options if(SaveDialog(hDlg) != ERROR_SUCCESS) return FALSE; if(pData->bUseTransaction && pfnRtlSetCurrentTransaction != NULL) { SaveTransactionHandle = pfnRtlGetCurrentTransaction(); pfnRtlSetCurrentTransaction(pData->hTransaction); } // Get the directory name from the dialog data StringCchCopy(szDirName, _countof(szDirName), pData->szDirName); StringCchCopy(szFileName, _countof(szFileName), pData->szFileName1); // If we are about to open a file by ID, and we have no relative directory, // try to take the directory from the file name if(szDirName[0] == 0 && (pData->dwCreateOptions & FILE_OPEN_BY_FILE_ID)) { LPTSTR szFullName = pData->szFileName1; LPTSTR szFileId = _tcsrchr(szFullName, _T('\\')); size_t nLength = szFileId - szFullName + 1; if(szFileId != NULL) { StringCchCopy(szDirName, nLength, szFullName); StringCchCopy(szFileName, _countof(szFileName), szFileId + 1); } } // Open the relative file (if any) if(szDirName[0] != 0) { RtlInitUnicodeString(&DirName, szDirName); InitializeObjectAttributes(&ObjAttr, &DirName, pData->dwObjAttrFlags, 0, NULL); Status = NtOpenFile(&pData->hDirectory, pData->dwDesiredAccessRF, &ObjAttr, &IoStatus, pData->dwShareAccessRF, pData->dwOpenOptionsRF); if(!NT_SUCCESS(Status)) { SetResultInfo(hDlg, RSI_NTSTATUS | RSI_HANDLE, Status, pData->hDirectory); return TRUE; } } // Prepare the file open if(NT_SUCCESS(Status)) { RtlInitUnicodeString(&FileName, szFileName); // If open by ID required, set the ID to the string if(pData->dwCreateOptions & FILE_OPEN_BY_FILE_ID) { // Convert object ID to binary value if(StringToFileID(szFileName, NULL, ObjectID, &cbObjectID) != ERROR_SUCCESS) return TRUE; // Set the object ID to the UNICODE_STRING FileName.MaximumLength = FileName.Length = (USHORT)cbObjectID; FileName.Buffer = (PWSTR)ObjectID; } ZeroMemory(&IoStatus, sizeof(IO_STATUS_BLOCK)); InitializeObjectAttributes(&ObjAttr, &FileName, pData->dwObjAttrFlags, pData->hDirectory, NULL); AllocationSize.QuadPart = (LONGLONG)pData->AllocationSize; // Invoke breakpoint if the user wants to if(IsDlgButtonChecked(hDlg, IDC_BREAKPOINT) == BST_CHECKED) { __debugbreak(); } Status = NtCreateFile(&pData->hFile, pData->dwDesiredAccess, &ObjAttr, &IoStatus, &AllocationSize, pData->dwFlagsAndAttributes, pData->dwShareAccess, pData->dwCreateDisposition2, pData->dwCreateOptions, pData->pFileEa, pData->dwEaSize); SetResultInfo(hDlg, RSI_NTSTATUS | RSI_HANDLE | RSI_NTCREATE, Status, pData->hFile, &IoStatus); // If this operation failed, we close the directory as well if(!NT_SUCCESS(Status) && pData->hDirectory != NULL) { NtClose(pData->hDirectory); pData->hDirectory = NULL; } } if(pData->bUseTransaction && pfnRtlSetCurrentTransaction != NULL) { pfnRtlSetCurrentTransaction(SaveTransactionHandle); SaveTransactionHandle = NULL; } return TRUE; } static int OnNtCloseClick(HWND hDlg) { TFileTestData * pData = GetDialogData(hDlg); NTSTATUS Status = STATUS_SUCCESS; // Invoke breakpoint if the user wants to if(IsDlgButtonChecked(hDlg, IDC_BREAKPOINT) == BST_CHECKED) { __debugbreak(); } // Close file handle first if(IsHandleValid(pData->hFile)) Status = NtClose(pData->hFile); pData->hFile = NULL; // Close directory handle last if(IsHandleValid(pData->hDirectory)) NtClose(pData->hDirectory); pData->hDirectory = NULL; SetResultInfo(hDlg, RSI_NTSTATUS | RSI_HANDLE, Status, pData->hFile); return TRUE; } static int OnDeltaPos(HWND hDlg, NMUPDOWN * pNMUpDown) { LONGLONG AllocationSize; DlgText2Hex64(hDlg, IDC_ALLOCATION_SIZE, &AllocationSize); AllocationSize -= pNMUpDown->iDelta * 0x100; if(AllocationSize < 0) AllocationSize = 0; Hex2DlgText64(hDlg, IDC_ALLOCATION_SIZE, AllocationSize); return TRUE; } static int OnCommand(HWND hDlg, UINT nNotify, UINT nIDCtrl) { if(nNotify == BN_CLICKED) { switch(nIDCtrl) { case IDC_RELATIVE_FILE_HELP: return OnRelativeFileHelp(hDlg); case IDC_DIRECTORY_NAME_BROWSE: return OnBrowseDirClick(hDlg); case IDC_FILE_NAME_BROWSE: return OnBrowseFileClick(hDlg); case IDC_OBJ_ATTR_FLAGS_BROWSE: return OnObjAtributesFlags(hDlg); case IDC_DESIRED_ACCESS_BROWSE: return OnDesiredAccessClick(hDlg); case IDC_FILE_ATTRIBUTES_BROWSE: return OnFileAttributesClick(hDlg); case IDC_SHARE_ACCESS_BROWSE: return OnShareAccessClick(hDlg); case IDC_CREATE_OPTIONS_BROWSE: return OnCreateOptionsClick(hDlg); case IDC_EXTENDED_ATTRIBUTES_EDIT: return OnEditEaClick(hDlg); case IDC_VIRTUALIZATION: return OnVirtualization(hDlg); case IDC_PRIVILEGES: PrivilegesDialog(hDlg); return TRUE; case IDC_MAKE_DIRECTORY: return OnMakeDirectoryClick(hDlg); case IDC_RDWR_ASYNC: return QuickAccessSelection(hDlg, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, false); case IDC_RDWR_SYNC: return QuickAccessSelection(hDlg, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, true); case IDC_READ_ASYNC: return QuickAccessSelection(hDlg, GENERIC_READ, FILE_SHARE_READ, false); case IDC_READ_SYNC: return QuickAccessSelection(hDlg, GENERIC_READ, FILE_SHARE_READ, true); case IDC_QUERY_INFO_ASYNC: return QuickAccessSelection(hDlg, FILE_READ_ATTRIBUTES, 0, false); case IDC_QUERY_INFO_SYNC: return QuickAccessSelection(hDlg, FILE_READ_ATTRIBUTES, 0, true); case IDC_DELETE_ON_CLOSE: return QuickAccessSelection(hDlg, DELETE, 0, false); case IDC_CREATE_FILE: return OnCreateFileClick(hDlg); case IDC_CLOSE_HANDLE: return OnNtCloseClick(hDlg); } } // Convert the directory/file name to the NT name, if needed // Removed: Keep it as it is. We want to test NtCreateFile // with whatever the user entered. // if(nNotify == EN_KILLFOCUS) // { // if(nIDCtrl == IDC_DIRECTORY_NAME) // ConvertToNtName(hDlg, nIDCtrl); // if(nIDCtrl == IDC_FILE_NAME && GetWindowTextLength(GetDlgItem(hDlg, IDC_DIRECTORY_NAME)) == 0) // ConvertToNtName(hDlg, nIDCtrl); // return TRUE; // } return FALSE; } static int OnNotify(HWND hDlg, NMHDR * pNMHDR) { switch(pNMHDR->code) { case PSN_SETACTIVE: return OnSetActive(hDlg); case PSN_KILLACTIVE: return OnKillActive(hDlg); case UDN_DELTAPOS: return OnDeltaPos(hDlg, (NMUPDOWN *)pNMHDR); } return FALSE; } //----------------------------------------------------------------------------- // Public functions INT_PTR CALLBACK PageProc02(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam) { // Call tooltip to handle messages g_Tooltip.HandleMessages(hDlg, uMsg, wParam, lParam, NULL); // Handle other messages switch(uMsg) { case WM_INITDIALOG: return OnInitDialog(hDlg, lParam); case WM_SIZE: if(pAnchors != NULL) pAnchors->OnSize(); return FALSE; case WM_DRAWITEM: if(wParam == IDC_RELATIVE_FILE_HELP) DrawURLButton(hDlg, (LPDRAWITEMSTRUCT)lParam); return TRUE; case WM_CONTEXTMENU: return ExecuteContextMenu(hDlg, FindContextMenu(IDR_NTCREATE_MENU), lParam); case WM_COMMAND: return OnCommand(hDlg, HIWORD(wParam), LOWORD(wParam)); case WM_NOTIFY: return OnNotify(hDlg, (NMHDR *)lParam); case WM_DESTROY: if(pAnchors != NULL) delete pAnchors; pAnchors = NULL; return FALSE; } return FALSE; } #ifdef _DEBUG void DebugCode_TEST() { //DWORD dwBitMask = 0x80030080; //FlagsDialog(NULL, // IDS_FLAGS_AND_ATTRIBUTES, // FileAttributesValues, // dwBitMask); //ExitProcess(3); } #endif