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.
 
 
 
 
 
 

834 lines
24 KiB

//----------------------------------------------------------------------------
//
// Copyright (c) 1997-1999 Microsoft Corporation
// All rights reserved.
//
// File Name:
// distfold.c
//
// Description:
// This file contains the dialog proc for the IDD_DISTFOLD page.
//
//----------------------------------------------------------------------------
#include "pch.h"
#include "resource.h"
#include <lm.h>
#include <winnetwk.h>
static TCHAR *StrSelectDistFolder;
static TCHAR *StrWindowsDistibFolder;
//----------------------------------------------------------------------------
//
// Function: OnSetActiveDistFolder
//
// Purpose: Called when the page is about to display.
//
//----------------------------------------------------------------------------
VOID
OnSetActiveDistFolder(HWND hwnd)
{
SetDlgItemText(hwnd, IDT_DISTFOLDER, WizGlobals.DistFolder);
SetDlgItemText(hwnd, IDT_SHARENAME, WizGlobals.DistShareName);
WIZ_BUTTONS(hwnd, PSWIZB_BACK | PSWIZB_NEXT);
}
//----------------------------------------------------------------------------
//
// Function: ConvertRelativePathToUncPath
//
// Purpose: If the path is a local path or a network path, it is converted
// to its UNC equivalent. If the path is already a UNC path then
// it is just copied and returned. The output variable
// UncDistFolder is assumed to be of MAX_PATH length.
//
// Arguments:
//
// Returns: BOOL
// TRUE - on success,
// FALSE - on failure
//
//----------------------------------------------------------------------------
BOOL
ConvertRelativePathToUncPath( IN const TCHAR *szRelativePath,
OUT TCHAR *UncDistFolder,
IN DWORD cbSize) {
TCHAR szLocalName[MAX_PATH];
TCHAR szDistribFolderDirPath[MAX_PATH];
TCHAR szUncPath[MAX_PATH];
const TCHAR *pDirString;
DWORD dwReturnValue;
DWORD dwSize;
HRESULT hrPrintf;
//
// Check and see if it is already a UNC Path, just checking to see if it
// begins with a \
//
if( szRelativePath[0] == _T('\\') ) {
lstrcpyn( UncDistFolder, szRelativePath, cbSize );
return( TRUE );
}
if( IsPathOnLocalDiskDrive( szRelativePath ) )
{
const TCHAR *pDirPath;
TCHAR szLocalComputerName[MAX_COMPUTERNAME_LENGTH + 1];
HRESULT hrCat;
dwSize = MAX_COMPUTERNAME_LENGTH + 1;
if( ! GetComputerName( szLocalComputerName, &dwSize ) )
{
ReportErrorId( NULL,
MSGTYPE_ERR | MSGTYPE_WIN32,
IDS_ERR_CANNOT_GET_LOCAL_COMPNAME );
return( FALSE );
}
hrPrintf=StringCchPrintf( UncDistFolder, cbSize, _T("\\\\%s\\"), szLocalComputerName );
pDirPath = szRelativePath;
//
// Set the dir path to just past the first \
//
pDirPath = pDirPath + 3;
hrCat=StringCchCat( UncDistFolder, cbSize, pDirPath );
return( TRUE );
}
else
{
hrPrintf=StringCchPrintf( szLocalName, AS(szLocalName), _T("%c:"), szRelativePath[0] );
pDirString = szRelativePath + 3;
lstrcpyn( szDistribFolderDirPath, pDirString, AS(szDistribFolderDirPath) );
dwSize = StrBuffSize( szUncPath );
dwReturnValue = WNetGetConnection( szLocalName, szUncPath, &dwSize );
if( dwReturnValue == NO_ERROR ) {
lstrcpyn( UncDistFolder, szUncPath, cbSize );
// Note:ConcatenatePaths will truncate to avoid buffer overrun
ConcatenatePaths( UncDistFolder,
szDistribFolderDirPath,
NULL );
return( TRUE );
}
else if( dwReturnValue == ERROR_BAD_DEVICE ) {
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_BAD_DEVICE );
}
else if( dwReturnValue == ERROR_NOT_CONNECTED ) {
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_NOT_CONNECTED );
}
else if( dwReturnValue == ERROR_CONNECTION_UNAVAIL ) {
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_CONNECTION_UNAVAIL );
}
else if( dwReturnValue == ERROR_NO_NETWORK ) {
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_NO_NETWORK );
}
else if( dwReturnValue == ERROR_NO_NET_OR_BAD_PATH ) {
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_NO_NETWORK );
}
else if( dwReturnValue == ERROR_EXTENDED_ERROR ) {
TCHAR szErrorString[MAX_STRING_LEN];
TCHAR szProviderName[MAX_STRING_LEN];
TCHAR szErrorCode[MAX_STRING_LEN];
DWORD dwErrorCode;
DWORD dwErrorSize = StrBuffSize( szErrorString );
DWORD dwProviderSize = StrBuffSize( szProviderName );
WNetGetLastError( &dwErrorCode,
szErrorString,
dwErrorSize,
szProviderName,
dwProviderSize );
_itot( dwErrorCode, szErrorCode, 10 );
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_EXTENDED_ERROR,
szProviderName,
szErrorCode,
szErrorString );
}
else {
//
// Unknown error
//
ReportErrorId( NULL,
MSGTYPE_ERR,
IDS_ERR_NETWORK_UNKNOWN_ERROR );
}
return( FALSE );
}
}
// NTRAID#NTBUG9-551874-2002/02/27-stelo,swamip - CreateDistFolder, ShareTheDistFolder should use the code from OEM mode, reduce attack surface
//
//----------------------------------------------------------------------------
//
// Function: CreateDistFolder
//
// Purpose: Creates the distribution folder and reports any error.
//
// Arguments:
// HWND hwnd - current window
//
// Returns:
// TRUE - all is ok
// FALSE - the error was reported, stay on this page
//
//----------------------------------------------------------------------------
BOOL
CreateDistFolder(HWND hwnd)
{
//
// Don't just CreateDir in case user says d:\foo\bar\fud and 'foo'
// doesn't exist. We want to make foo and bar and fud for the user.
//
// On the other hand, if d:\foo\bar\fud already exists, we want to
// report an error, because the user explicitly said "New Folder"
// on the dialog.
//
// So, check if it's there first, then just go silently make everything
// needed if it doesn't exist. EnsureDirExists() will be silent in
// the case that d:\foo\bar\fud already exists.
//
if ( DoesPathExist(WizGlobals.DistFolder) ) {
UINT iRet;
iRet = ReportErrorId(hwnd,
MSGTYPE_YESNO,
IDS_ERR_DISTFOLD_EXISTS,
WizGlobals.DistFolder);
if ( iRet != IDYES )
return FALSE;
}
if ( ! EnsureDirExists(WizGlobals.DistFolder) ) {
ReportErrorId(hwnd,
MSGTYPE_ERR | MSGTYPE_WIN32,
IDS_ERR_CREATING_DISTFOLD,
WizGlobals.DistFolder);
return FALSE;
}
return TRUE;
}
//----------------------------------------------------------------------------
//
// Function: CheckDistFolder
//
// Purpose: This function is called only by OnWizNextDistFolder. It
// assumes that WizGlobals.DistFolder is up to date. We
// create a folder or check if one exists as the user asks.
//
// Returns:
// FALSE - Problems, don't let wizard proceed
// TRUE - All is cool, go to the next page
//
// Notes:
// - This routine directly reports any errors and prompts the
// user as needed.
//
//----------------------------------------------------------------------------
BOOL
CheckDistFolder(HWND hwnd)
{
TCHAR PathBuffer[MAX_PATH];
//
// If user selected a file instead of a folder, trap that now, (just
// for a more informative error message).
//
if ( DoesFileExist(WizGlobals.DistFolder) ) {
ReportErrorId(hwnd,
MSGTYPE_ERR,
IDS_ERR_FOLDER_IS_FILE,
WizGlobals.DistFolder);
return FALSE;
}
if ( ! WizGlobals.bCreateNewDistFolder ) {
//
// We're trying to edit an existing distfolder. Popup if
//
// 1. It doesn't exist OR
// 2. Can't find dosnet.inf
//
// In the case of #1, make user change path or radio button.
// In the case of #2, let user edit it anyway via YES/NO popup.
//
if ( ! DoesFolderExist(WizGlobals.DistFolder) ) {
ReportErrorId(hwnd,
MSGTYPE_ERR,
IDS_ERR_FOLDER_NOT_EXIST,
WizGlobals.DistFolder);
return FALSE;
}
//
// If dosnet.inf doesn't exist, this isn't a distribution folder.
// Note, this is a really minimal check.
//
lstrcpyn(PathBuffer, WizGlobals.DistFolder, AS(PathBuffer));
ConcatenatePaths(PathBuffer, WizGlobals.Architecture, _T("dosnet.inf"), NULL);
if ( ! DoesFileExist(PathBuffer) ) {
UINT iRet;
iRet = ReportErrorId(
hwnd,
MSGTYPE_YESNO,
IDS_ERR_NOT_VALID_DISTFOLD,
WizGlobals.DistFolder);
if ( iRet != IDYES )
return FALSE;
}
} else {
//
// We're creating a new distfolder. Go do the work.
//
if ( ! CreateDistFolder(hwnd) )
return FALSE;
}
return TRUE;
}
//--------------------------------------------------------------------------
//
// Function: ShareTheDistFolder
//
// Purpose: Called by the Next procedure to put up the share.
//
//--------------------------------------------------------------------------
BOOL
ShareTheDistFolder(HWND hwnd, TCHAR *UncDistFolder)
{
LPTSTR pszServerPath;
TCHAR szServerName[MAX_PATH];
NET_API_STATUS nas;
SHARE_INFO_502 si502, *psi502;
BOOL bStatus = TRUE;
//
// Query this computer for info about the sharename the user
// just typed in.
//
nas = NetShareGetInfo(
NULL, // this computer
WizGlobals.DistShareName, // shrname
502, // amount of info
(LPBYTE*) &psi502); // the info (output)
//
// If this sharename is in use, we have to look at the directory that
// is being shared.
//
if ( nas == NERR_Success ) {
TCHAR szUncPath[MAX_PATH + 1];
//
// All is ok if this sharename is for the DistFolder
// We have to check both the relative and the UNC path
//
if( ! ConvertRelativePathToUncPath( psi502->shi502_path,
szUncPath,
AS(szUncPath)) )
{
lstrcpyn( szUncPath, psi502->shi502_path, AS(szUncPath) );
}
if ( lstrcmpi(WizGlobals.DistFolder, psi502->shi502_path) == 0 ||
lstrcmpi(WizGlobals.DistFolder, szUncPath) == 0 ) {
NetApiBufferFree(psi502);
return TRUE;
}
//
// The sharename is in use, however, it's being used by a
// different directory. Report an error telling user to enter
// a different sharename.
//
ReportErrorId(hwnd,
MSGTYPE_ERR,
IDS_ERR_SHARENAME_INUSE,
WizGlobals.DistShareName);
NetApiBufferFree(psi502);
return FALSE;
}
//
// Put up the share
//
si502.shi502_netname = WizGlobals.DistShareName;
si502.shi502_type = STYPE_DISKTREE;
si502.shi502_remark = StrWindowsDistibFolder;
si502.shi502_permissions = 0;
si502.shi502_max_uses = SHI_USES_UNLIMITED;
si502.shi502_current_uses = 0;
si502.shi502_path = WizGlobals.DistFolder;
si502.shi502_passwd = NULL;
si502.shi502_reserved = 0;
si502.shi502_security_descriptor = NULL;
//
// Set the server path to NULL if it is on the local machine, or the
// computer name if it is on a remote machine
//
if( IsPathOnLocalDiskDrive(WizGlobals.DistFolder) ) {
pszServerPath = NULL;
}
else {
GetComputerNameFromUnc( UncDistFolder,
szServerName,
AS(szServerName));
pszServerPath = szServerName;
}
nas = NetShareAdd(pszServerPath,
502, // info-level
(LPBYTE) &si502, // info-buffer
NULL); // don't bother with parm
//
// If the NetShareAdd fails for some reason, report the error code
// to the user and give the user a chance to continue the wizard
// without enabling the share. The user might not have privelege
// to do this.
//
//
// ISSUE-2002/02/27-stelo - The Net apis don't set GetLastError(). So how is one
// supposed to report the error message to the user, like
// 'access denied'???
//
// Run this test: Logon to an account without admin privelege.
// Now try to put up the share. I bet the error message is
// useless. This is a common scenario and should be addressed.
//
if ( nas != NERR_Success ) {
UINT iRet;
iRet = ReportErrorId(hwnd, MSGTYPE_YESNO, IDS_ERR_ENABLE_SHARE, nas);
if ( iRet != IDYES )
return FALSE;
}
return TRUE;
}
//----------------------------------------------------------------------------
//
// Function: OnDistFolderInitDialog
//
// Purpose:
//
// Arguments: IN HWND hwnd - handle to the dialog
//
// Returns: VOID
//
//----------------------------------------------------------------------------
static VOID
OnDistFolderInitDialog( IN HWND hwnd )
{
//
// Load text strings
//
StrSelectDistFolder = MyLoadString( IDS_SELECT_DISTRIB_FOLDER );
StrWindowsDistibFolder = MyLoadString( IDS_WINDOWS_DISTRIB_FOLDER );
//
// Set the text limit on the edit boxes
//
SendDlgItemMessage( hwnd,
IDT_DISTFOLDER,
EM_LIMITTEXT,
(WPARAM) MAX_DIST_FOLDER,
(LPARAM) 0 );
SendDlgItemMessage( hwnd,
IDT_SHARENAME,
EM_LIMITTEXT,
(WPARAM) MAX_SHARENAME,
(LPARAM) 0 );
}
//----------------------------------------------------------------------------
//
// Function: OnWizNextDistFolder
//
// Purpose: Called when user pushes the NEXT button
//
//----------------------------------------------------------------------------
BOOL
OnWizNextDistFolder(HWND hwnd)
{
BOOL bResult = TRUE;
//
// Get the control settings
//
GetDlgItemText(hwnd, IDT_DISTFOLDER, WizGlobals.DistFolder, MAX_DIST_FOLDER + 1);
GetDlgItemText(hwnd,
IDT_SHARENAME,
WizGlobals.DistShareName,
MAX_SHARENAME + 1);
// ISSUE-2002/02/27-stelo - need to check they entered a valid distrib folder here, i.e. that it is a valid path (local or UNC)
// which of these are valid?
//
// c
// c:
// c:\
// c\sdjf
// \somedir
// //somename
// asfdj
//
MyGetFullPath(WizGlobals.DistFolder);
//
// Make sure they filled in the edit boxes
//
if( WizGlobals.DistFolder[0] == _T('\0') ) {
ReportErrorId( hwnd, MSGTYPE_ERR, IDS_ERR_ENTER_DIST_FOLDER );
bResult = FALSE;
}
if( WizGlobals.DistShareName[0] == _T('\0') ) {
ReportErrorId( hwnd, MSGTYPE_ERR, IDS_ERR_ENTER_SHARENAME );
bResult = FALSE;
}
if( !bResult ) {
SetWindowLongPtr(hwnd, DWLP_MSGRESULT, -1);
return bResult;
}
// ISSUE-2002/02/27-stelo - if they specified a network path, it may take a while for the
// connection to restore if they haven't used it in a while, we should probably
// do a pop-up here to tell them we are looking for it
//
// Do some checking for the dist fold, create it if necessary, etc...
//
if ( ! CheckDistFolder(hwnd) )
bResult = FALSE;
if( ! ConvertRelativePathToUncPath( WizGlobals.DistFolder,
WizGlobals.UncDistFolder,
AS(WizGlobals.UncDistFolder)) )
{
lstrcpyn( WizGlobals.UncDistFolder, WizGlobals.DistFolder, AS(WizGlobals.UncDistFolder) );
}
//
// Share it out if we haven't already reported an error
//
if ( bResult ) {
if ( ! ShareTheDistFolder( hwnd, WizGlobals.UncDistFolder ) ) {
bResult = FALSE;
}
}
//
// If OemFilesPath doesn't have a value, give it one.
//
if ( bResult && WizGlobals.OemFilesPath[0] == _T('\0') ) {
lstrcpyn( WizGlobals.OemFilesPath, WizGlobals.DistFolder, AS(WizGlobals.OemFilesPath) );
//Note:ConcatenatePaths truncates to avoid buffer overrun
ConcatenatePaths( WizGlobals.OemFilesPath,
WizGlobals.Architecture,
_T("$oem$"),
NULL );
}
//
// Force creation of the $oem$ dir (if it doesn't exist already)
//
if ( bResult && ! EnsureDirExists(WizGlobals.OemFilesPath) ) {
ReportErrorId(hwnd,
MSGTYPE_ERR | MSGTYPE_WIN32,
IDS_ERR_CREATE_FOLDER,
WizGlobals.OemFilesPath);
bResult = FALSE;
}
//
// Route to the next wizard page
//
return bResult;
}
//---------------------------------------------------------------------------
//
// Function: BrowseForDistFolder
//
// Purpose: Calls SHBrowseForFolder to allow user to browse for a
// distribution folder.
//
// Arguments:
// HWND hwnd - owning window
// LPTSTR PathBuffer - MAX_PATH buffer to receive results
//
// Returns: BOOL - success
//
//---------------------------------------------------------------------------
BOOL
BrowseForDistFolder(HWND hwnd, LPTSTR PathBuffer)
{
BROWSEINFO BrowseInf;
LPITEMIDLIST lpIdList;
UINT ulFlags = BIF_EDITBOX |
BIF_RETURNONLYFSDIRS;
BrowseInf.hwndOwner = hwnd;
BrowseInf.pidlRoot = NULL; // no initial root
BrowseInf.pszDisplayName = PathBuffer; // output
BrowseInf.lpszTitle = StrSelectDistFolder;
BrowseInf.ulFlags = ulFlags;
BrowseInf.lpfn = NULL;
BrowseInf.lParam = (LPARAM) 0;
BrowseInf.iImage = 0; // no image
lpIdList = SHBrowseForFolder(&BrowseInf);
//
// Get the pathname out of this idlist returned and free up the memory
//
if ( lpIdList == NULL )
return FALSE;
SHGetPathFromIDList(lpIdList, PathBuffer);
ILFreePriv(lpIdList);
return TRUE;
}
//---------------------------------------------------------------------------
//
// Function: OnBrowseDistFolder
//
// Purpose: Called when user pushes the BROWSE button
//
//---------------------------------------------------------------------------
VOID
OnBrowseDistFolder(HWND hwnd)
{
//
// NTRAID#NTBUG9-551874-2002/02/27-stelo,swamip - CreateDistFolder, ShareTheDistFolder should use the code from OEM mode, reduce attack surface
//
if ( BrowseForDistFolder(hwnd, WizGlobals.DistFolder) ) {
SendDlgItemMessage(hwnd,
IDT_DISTFOLDER,
WM_SETTEXT,
(WPARAM) MAX_PATH,
(LPARAM) WizGlobals.DistFolder);
}
}
//----------------------------------------------------------------------------
//
// Function: DlgDistFolderPage
//
// Purpose: This is the dialog procedure IDD_DISTFOLDER.
//
//----------------------------------------------------------------------------
INT_PTR CALLBACK
DlgDistFolderPage(
IN HWND hwnd,
IN UINT uMsg,
IN WPARAM wParam,
IN LPARAM lParam)
{
BOOL bStatus = TRUE;
switch (uMsg) {
case WM_INITDIALOG:
OnDistFolderInitDialog( hwnd );
break;
case WM_COMMAND:
{
int nButtonId=LOWORD(wParam);
switch ( nButtonId ) {
case IDC_BROWSE:
if ( HIWORD(wParam) == BN_CLICKED )
OnBrowseDistFolder(hwnd);
break;
default:
bStatus = FALSE;
break;
}
}
break;
case WM_NOTIFY:
{
LPNMHDR pnmh = (LPNMHDR)lParam;
// ISSUE-2002/02/27-stelo,swamip - should check for valid pointer (possible dereference)
//
switch( pnmh->code ) {
case PSN_QUERYCANCEL:
WIZ_CANCEL(hwnd);
break;
case PSN_SETACTIVE:
g_App.dwCurrentHelp = IDH_DIST_FLDR;
if ( (WizGlobals.iProductInstall != PRODUCT_UNATTENDED_INSTALL) ||
WizGlobals.bStandAloneScript)
WIZ_SKIP( hwnd );
else
OnSetActiveDistFolder(hwnd);
break;
case PSN_WIZBACK:
bStatus = FALSE;
break;
case PSN_WIZNEXT:
if ( !OnWizNextDistFolder(hwnd) )
WIZ_SKIP(hwnd);
else
bStatus = FALSE;
break;
case PSN_HELP:
WIZ_HELP();
break;
default:
bStatus = FALSE;
break;
}
}
break;
default:
bStatus = FALSE;
break;
}
return bStatus;
}