Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

676 lines
20 KiB

/****************************************************************************\
DISKAPI.C / OPK Wizard (OPKWIZ.EXE)
Microsoft Confidential
Copyright (c) Microsoft Corporation 1999
All rights reserved
Disk API source file for custom disk APIs used in the OPK Wizard.
4/99 - Jason Cohen (JCOHEN)
Added this new source file for the OPK Wizard as part of the
Millennium rewrite.
\****************************************************************************/
//
// Include file(s)
//
#include <pch.h>
#include <commdlg.h>
#include <tchar.h>
#include <shlobj.h>
//
// Internal Define(s):
//
#define IDC_BROWSE_EDIT 0x3744 // Common dialogs edit box in the SHBrowseForFolder function.
//
// Internal Function Prototype(s):
//
static DWORD CopyDirectoryEngine(HWND hwnd, HANDLE hEvent, LPCTSTR lpSrc, LPCTSTR lpDst, BOOL fCount);
static CALLBACK BrowseCallbackProc(HWND, UINT, LPARAM, LPARAM);
//
// External Function(s):
//
BOOL DirectoryExists(LPCTSTR lpDirectory)
{
DWORD dwAttr;
return ( ( lpDirectory != NULL ) &&
( *lpDirectory != NULLCHR ) &&
( (dwAttr = GetFileAttributes(lpDirectory)) != 0xFFFFFFFF ) &&
( dwAttr & FILE_ATTRIBUTE_DIRECTORY ) );
}
BOOL FileExists(LPCTSTR lpFile)
{
DWORD dwAttr;
return ( ( lpFile != NULL ) &&
( *lpFile != NULLCHR ) &&
( (dwAttr = GetFileAttributes(lpFile)) != 0xFFFFFFFF ) &&
( !(dwAttr & FILE_ATTRIBUTE_DIRECTORY) ) );
}
BOOL CopyResetFile(LPCTSTR lpSource, LPCTSTR lpTarget)
{
if ( !CopyFile(lpSource, lpTarget, FALSE) )
return FALSE;
SetFileAttributes(lpTarget, FILE_ATTRIBUTE_NORMAL);
return TRUE;
}
DWORD IfGetLongPathName(LPCTSTR lpszShortPath, LPTSTR lpszLongPath, DWORD cchBuffer)
{
//
// See also \nt\base\win32\client\vdm.c.
//
DWORD dwReturn = 0;
#if defined(_WIN64) // _WIN64 postdates the introduction of GetLongPathName.
typedef (WINAPI* PFNGetLongPathNameA)( PCSTR lpszShortPath, PSTR lpszLongPath, DWORD cchBuffer);
typedef (WINAPI* PFNGetLongPathNameW)(PCWSTR lpszShortPath, PWSTR lpszLongPath, DWORD cchBuffer);
#ifdef UNICODE
typedef PFNGetLongPathNameW PFNGetLongPathName;
const static char ProcName[] = "GetLongPathNameW";
#else
typedef PFNGetLongPathNameA PFNGetLongPathName;
const static char ProcName[] = "GetLongPathNameA";
#endif
static PFNGetLongPathName hGetLongPathName = NULL;
static BOOL fInited = FALSE;
if (!fInited)
{
//
// GetModuleHandle is in kernel32, so as long as this lib code
// is around, the handle to kernel32 is constant and the result of
// GetProcAccess is valid.
//
// The old code that called LoadLibrary/FreeLibrary would lose the
// value of GetLastError by calling FreeLibrary.
//
HMODULE hKernel32;
if (hKernel32 = GetModuleHandle(TEXT("Kernel32.dll")))
hGetLongPathName = (PFNGetLongPathName)(GetProcAddress(hKernel32, ProcName));
fInited = TRUE;
}
if (hGetLongPathName)
{
dwReturn = hGetLongPathName(lpszShortPath, lpszLongPath, cchBuffer);
}
#else
dwReturn = GetLongPathName(lpszShortPath, lpszLongPath, cchBuffer);
#endif
return dwReturn;
}
BOOL CreatePath(LPCTSTR lpPath)
{
LPTSTR lpFind = (LPTSTR) lpPath;
while ( lpFind = _tcschr(lpFind + 1, CHR_BACKSLASH) )
{
if ( !((lpFind - lpPath <= 2) && (*(lpFind - 1) == _T(':'))) )
{
*lpFind = NULLCHR;
if ( !DirectoryExists(lpPath) )
CreateDirectory(lpPath, NULL);
*lpFind = CHR_BACKSLASH;
}
}
if ( !DirectoryExists(lpPath) )
CreateDirectory(lpPath, NULL);
return DirectoryExists(lpPath);
}
BOOL DeletePath(LPCTSTR lpDirectory)
{
WIN32_FIND_DATA FileFound;
HANDLE hFile;
// Validate the parameters.
//
if ( ( lpDirectory == NULL ) ||
( *lpDirectory == NULLCHR ) ||
( !SetCurrentDirectory(lpDirectory) ) )
{
return TRUE;
}
// Process all the files and directories in the directory passed in.
//
SetCurrentDirectory(lpDirectory);
if ( (hFile = FindFirstFile(_T("*"), &FileFound)) != INVALID_HANDLE_VALUE )
{
do
{
// First check to see if this is a file (not a directory).
//
if ( !( FileFound.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) )
{
// Make sure we clear the readonly flag
//
SetFileAttributes(FileFound.cFileName, FILE_ATTRIBUTE_NORMAL);
DeleteFile(FileFound.cFileName);
}
// Otherwise, make sure the directory is not "." or "..".
//
else if ( ( lstrcmp(FileFound.cFileName, _T(".")) ) &&
( lstrcmp(FileFound.cFileName, _T("..")) ) )
{
DeletePath(FileFound.cFileName);
}
}
while ( FindNextFile(hFile, &FileFound) );
FindClose(hFile);
}
// Go to the parent directory and remove the current one.
// We have to make sure and reset the readonly attributes
// on the dir also.
//
SetCurrentDirectory(_T(".."));
SetFileAttributes(lpDirectory, FILE_ATTRIBUTE_NORMAL);
return RemoveDirectory(lpDirectory);
}
BOOL DeleteFilesEx(LPCTSTR lpDirectory, LPCTSTR lpFileSpec)
{
WIN32_FIND_DATA FileFound;
HANDLE hFile;
TCHAR szCurDir[MAX_PATH];
// Validate the parameters.
//
if ( ( lpDirectory == NULL ) ||
( *lpDirectory == NULLCHR ) ||
( !SetCurrentDirectory(lpDirectory) ) )
{
return FALSE;
}
// Get our current directory so we can set ourself back
//
GetCurrentDirectory(MAX_PATH, szCurDir);
// Process all the files and directories in the directory passed in.
//
SetCurrentDirectory(lpDirectory);
if ( (hFile = FindFirstFile(lpFileSpec, &FileFound)) != INVALID_HANDLE_VALUE )
{
do
{
// First check to see if this is a file (not a directory).
//
if ( !( FileFound.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) )
{
DeleteFile(FileFound.cFileName);
}
}
while ( FindNextFile(hFile, &FileFound) );
FindClose(hFile);
}
SetCurrentDirectory(szCurDir);
return TRUE;
}
LPTSTR AddPathN(LPTSTR lpPath, LPCTSTR lpName, DWORD cbPath)
{
LPTSTR lpTemp = lpPath;
// Validate the parameters passed in.
//
if ( ( lpPath == NULL ) ||
( lpName == NULL ) )
{
return NULL;
}
// Find the end of the path.
//
while ( *lpTemp )
{
lpTemp = CharNext(lpTemp);
if ( cbPath )
{
cbPath--;
}
}
// If no trailing backslash on the path then add one.
//
if ( ( lpTemp > lpPath ) &&
( *CharPrev(lpPath, lpTemp) != CHR_BACKSLASH ) )
{
// Make sure there is room in the path buffer to
// add the backslash and the null terminator.
//
if ( cbPath < 2 )
{
return NULL;
}
*lpTemp = CHR_BACKSLASH;
lpTemp = CharNext(lpTemp);
cbPath--;
}
else
{
// Make sure there is at least room for the null
// terminator.
//
if ( cbPath < 1 )
{
return NULL;
}
}
// Make sure there is no preceeding spaces or backslashes
// on the name to add.
//
while ( ( *lpName == CHR_SPACE ) ||
( *lpName == CHR_BACKSLASH ) )
{
lpName = CharNext(lpName);
}
// Add the new name to existing path.
//
lstrcpyn(lpTemp, lpName, cbPath);
// Trim trailing spaces from result.
//
while ( ( lpTemp > lpPath ) &&
( *(lpTemp = CharPrev(lpPath, lpTemp)) == CHR_SPACE ) )
{
*lpTemp = NULLCHR;
}
return lpPath;
}
LPTSTR AddPath(LPTSTR lpPath, LPCTSTR lpName)
{
return AddPathN(lpPath, lpName, 0xFFFFFFFF);
}
DWORD ExpandFullPath(LPTSTR lpszPath, LPTSTR lpszReturn, DWORD cbReturn)
{
LPTSTR lpszExpanded = AllocateExpand(lpszPath ? lpszPath : lpszReturn),
lpszDontCare;
DWORD dwRet;
*lpszReturn = NULLCHR;
if ( NULL == lpszExpanded )
{
return 0;
}
dwRet = GetFullPathName(lpszExpanded, cbReturn, lpszReturn, &lpszDontCare);
FREE(lpszExpanded);
return dwRet;
}
BOOL CopyDirectory(LPCTSTR lpSrc, LPCTSTR lpDst)
{
return ( CopyDirectoryEngine(NULL, NULL, lpSrc, lpDst, FALSE) != 0 );
}
BOOL CopyDirectoryProgress(HWND hwnd, LPCTSTR lpSrc, LPCTSTR lpDst)
{
return ( CopyDirectoryEngine(hwnd, NULL, lpSrc, lpDst, FALSE) != 0 );
}
BOOL CopyDirectoryProgressCancel(HWND hwnd, HANDLE hEvent, LPCTSTR lpSrc, LPCTSTR lpDst)
{
return ( CopyDirectoryEngine(hwnd, hEvent, lpSrc, lpDst, FALSE) != 0 );
}
DWORD FileCount(LPCTSTR lpSrc)
{
return CopyDirectoryEngine(NULL, NULL, lpSrc, NULL, TRUE);
}
BOOL BrowseForFolder(HWND hwndParent, INT iString, LPTSTR lpDirBuf, DWORD dwFlags)
{
BROWSEINFO bi = {0};
TCHAR szBuffer[MAX_PATH],
szPath[MAX_PATH],
szTitle[256] = NULLSTR;
LPITEMIDLIST lpil;
// Copy the current directory into the buffer so
// we start out from that folder.
//
lstrcpyn(szPath, lpDirBuf, AS(szPath));
// Load the instructional text for the dialog.
//
if ( iString )
LoadString(NULL, iString, szTitle, sizeof(szTitle) / sizeof(TCHAR));
// Setup the BrowseInfo struct.
//
bi.hwndOwner = hwndParent;
bi.pidlRoot = NULL;
bi.pszDisplayName = szBuffer;
bi.lpszTitle = szTitle;
bi.ulFlags = dwFlags ? dwFlags : BIF_RETURNONLYFSDIRS;
bi.lpfn = (BFFCALLBACK) BrowseCallbackProc;
bi.lParam = (LPARAM) szPath;
// Return the new path if we got one.
//
if ( ( (lpil = SHBrowseForFolder(&bi)) != NULL ) &&
( SHGetPathFromIDList(lpil, szPath) && szPath[0] && DirectoryExists(szPath) ) )
{
lstrcpy(lpDirBuf, szPath);
return TRUE;
}
return FALSE;
}
BOOL BrowseForFile(HWND hwnd, INT iTitle, INT iFilter, INT iExtension, LPTSTR lpFileName, DWORD cbFileName, LPTSTR lpDirectory, DWORD dwFlags)
{
OPENFILENAME ofn = {sizeof(ofn)};
TCHAR szTitle[256] = NULLSTR,
szFilter[256] = NULLSTR,
szExtension[256] = NULLSTR,
szFullPath[MAX_PATH] = NULLSTR;
LPTSTR lpSearch,
lpNext,
lpFilePart = NULL;
// Load all the strings we need for the open file structure.
//
if ( iTitle )
LoadString(NULL, iTitle, szTitle, sizeof(szTitle) / sizeof(TCHAR));
if ( iFilter )
LoadString(NULL, iFilter, szFilter, sizeof(szFilter) / sizeof(TCHAR));
if ( iExtension )
LoadString(NULL, iExtension, szExtension, sizeof(szExtension) / sizeof(TCHAR));
// Replace all the | in the filter string with \0.
//
lpSearch = szFilter;
while ( *lpSearch )
{
lpNext = CharNext(lpSearch);
if ( *lpSearch == _T('|') )
*lpSearch = NULLCHR;
lpSearch = lpNext;
}
// Figure out what the default directory and file will be.
//
if ( *lpFileName && GetFullPathName(lpFileName, STRSIZE(szFullPath), szFullPath, &lpFilePart) && szFullPath[0] )
{
// If the whole path is a directory, there is no file part.
//
if ( DirectoryExists(szFullPath) )
lpFilePart = NULL;
// Copy off the file name part.
//
if ( lpFilePart && ( (DWORD) lstrlen(lpFilePart) < cbFileName ) )
lstrcpy(lpFileName, lpFilePart);
else
*lpFileName = NULLCHR;
// Now chop off the file name so we are left with the directory.
//
if ( lpFilePart )
*lpFilePart = NULLCHR;
}
else
{
// No cool default directory or file name to use, so we use the
// directory passed in and no file name.
//
*lpFileName = NULLCHR;
szFullPath[0] = NULLCHR;
}
// Setup the open file struture.
//
ofn.hwndOwner = hwnd;
ofn.lpstrFilter = szFilter[0] ? szFilter : NULL;
ofn.nFilterIndex = szFilter[0] ? 1 : 0;
ofn.lpstrFile = lpFileName;
ofn.nMaxFile = cbFileName;
ofn.lpstrInitialDir = ( szFullPath[0] && DirectoryExists(szFullPath) ) ? szFullPath : lpDirectory;
ofn.lpstrTitle = szTitle[0] ? szTitle : NULL;
ofn.lpstrDefExt = szExtension[0] ? szExtension : NULL;
ofn.Flags = dwFlags ? dwFlags : (OFN_HIDEREADONLY | OFN_FILEMUSTEXIST);
// Make sure the buffer is zero'ed out if the function failes.
//
if ( !GetOpenFileName(&ofn) )
*lpFileName = NULLCHR;
// Return true only if we are passing back a file name.
//
return ( *lpFileName != NULLCHR );
}
//
// Internal Functions:
//
static DWORD CopyDirectoryEngine(HWND hwnd, HANDLE hEvent, LPCTSTR lpSrc, LPCTSTR lpDst, BOOL fCount)
{
WIN32_FIND_DATA FileFound;
HANDLE hFile;
BOOL bReturn = TRUE;
DWORD dwReturn = 0;
TCHAR szDst[MAX_PATH];
LPTSTR lpFileName,
lpSearch = NULL;
// If a source directory was passed in, set the current directory
// to it because that is we we are going to search for files.
//
if ( lpSrc )
{
// If the source isn't a directory, it is a file or file pattern we are
// copying.
//
if ( DirectoryExists(lpSrc) )
{
// Now make sure we set the current directory to the source directory.
//
bReturn = SetCurrentDirectory(lpSrc);
}
else
{
// We have to separate the path from the file or file pattern.
//
if ( lpSearch = _tcsrchr(lpSrc, CHR_BACKSLASH) )
{
// Set the current directory to the path part of the source buffer.
//
TCHAR szPath[MAX_PATH];
lstrcpyn(szPath, lpSrc, 1 + (int)(lpSearch - lpSrc));
if ( *(lpSearch = CharNext(lpSearch)) == NULLCHR )
lpSearch = NULL;
bReturn = SetCurrentDirectory(szPath);
}
else
lpSearch = (LPTSTR) lpSrc;
}
}
// Make sure the source directory existed, create the
// destination directory, and make sure it exists also.
//
if ( bReturn && ( fCount || ( bReturn = CreatePath(lpDst) ) ) )
{
// Setup the destination buffer with a pointer to the
// end of the path.
//
if ( !fCount )
{
lstrcpy(szDst, lpDst);
AddPath(szDst, NULLSTR);
lpFileName = szDst + lstrlen(szDst);
}
// Process all the files and directories in the directory passed in.
//
if ( (hFile = FindFirstFile(lpSearch ? lpSearch : _T("*"), &FileFound)) != INVALID_HANDLE_VALUE )
{
do
{
// Create the full path destination name.
//
if ( !fCount )
lstrcpy(lpFileName, FileFound.cFileName);
// First check to see if this is a file (not a directory).
//
if ( !( FileFound.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ) )
{
// Copy the file from the source to the destination.
//
fCount ? (dwReturn++) : (bReturn = CopyResetFile(FileFound.cFileName, szDst));
// Increase the progress bar. This is the only difference between
// CopyDirectroy() and CopyDirectoryProgress().
//
if ( hwnd )
SendMessage(hwnd, PBM_STEPIT, 0, 0);
}
// Otherwise, make sure the directory is not "." or "..".
//
else if ( lstrcmp(FileFound.cFileName, _T(".")) &&
lstrcmp(FileFound.cFileName, _T("..")) &&
SetCurrentDirectory(FileFound.cFileName) )
{
// Process all the files there.
//
DWORD dwBuffer = CopyDirectoryEngine(hwnd, hEvent, NULL, szDst, fCount);
fCount ? (dwReturn += dwBuffer) : (bReturn = (dwBuffer != 0));
SetCurrentDirectory(_T(".."));
}
// Check event to see if the user canceled.
//
if ( hEvent && ( WaitForSingleObject(hEvent, 0) != WAIT_TIMEOUT ) )
bReturn = FALSE;
}
while ( bReturn && FindNextFile(hFile, &FileFound) );
FindClose(hFile);
}
}
return bReturn ? (fCount ? dwReturn : 1) : 0;
}
static CALLBACK BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lParam, LPARAM lpData)
{
TCHAR szPathName[MAX_PATH];
LPTSTR lpszData = (LPTSTR) lpData;
switch ( uMsg )
{
case BFFM_INITIALIZED:
// Initialize the dialog with the OK button and current directory.
//
if ( lpszData && *lpszData )
{
LPTSTR lpEnd;
// Make sure there is a trailing backslash so that a drive passed in
// works (like c:).
//
szPathName[0] = NULLCHR;
if ( GetFullPathName(lpszData, STRSIZE(szPathName), szPathName, NULL) && szPathName[0] )
lstrcpy(lpszData, szPathName);
// For some dumb reason, the BFFM_SETSELECTION doesn't like it when there
// is a trailing backslash on the path.
//
if ( ( lstrlen(lpszData) > 3 ) &&
( lpEnd = CharPrev(lpszData, lpszData + lstrlen(lpszData)) ) &&
( *lpEnd == CHR_BACKSLASH ) )
{
*lpEnd = NULLCHR;
}
// Update the tree with the default dir and enable/disable the OK button
// if there is a valid directory.
//
SendMessage(hwnd, BFFM_SETSELECTION, TRUE, lpData);
SendMessage(hwnd, BFFM_ENABLEOK, 0, (DirectoryExists(lpszData) != 0));
}
else
SendMessage(hwnd, BFFM_ENABLEOK, 0, 0);
break;
case BFFM_SELCHANGED:
// Turn the id into a folder name.
//
szPathName[0] = NULLCHR;
if ( SHGetPathFromIDList((LPITEMIDLIST) lParam, szPathName) && szPathName[0] && DirectoryExists(szPathName) )
{
SetDlgItemText(hwnd, IDC_BROWSE_EDIT, szPathName);
SendMessage(hwnd, BFFM_ENABLEOK, 0, 1);
}
else
SendMessage(hwnd, BFFM_ENABLEOK, 0, 0);
break;
case BFFM_VALIDATEFAILED:
SendMessage(hwnd, BFFM_ENABLEOK, 0, 0);
return TRUE;
}
return 0;
}
BOOL CreateUnicodeFile(LPCTSTR lpFile)
{
HANDLE hFile;
DWORD dwWritten = 0;
WCHAR cHeader = 0xFEFF;
BOOL bReturn = FALSE;
// If we have a file name and the file does not exist, attempt to create
//
if ( lpFile && *lpFile && !FileExists(lpFile))
{
if ( (hFile = CreateFile(lpFile, GENERIC_WRITE, FILE_SHARE_READ, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL)) != INVALID_HANDLE_VALUE)
{
WriteFile(hFile, &cHeader, sizeof(cHeader), &dwWritten, NULL);
CloseHandle(hFile);
bReturn = TRUE;
}
}
return bReturn;
}