462 lines
14 KiB
C
462 lines
14 KiB
C
|
|
||
|
|
/****************************************************************************\
|
||
|
|
|
||
|
|
MISCAPI.C / OPK Wizard (OPKWIZ.EXE)
|
||
|
|
|
||
|
|
Microsoft Confidential
|
||
|
|
Copyright (c) Microsoft Corporation 1999
|
||
|
|
All rights reserved
|
||
|
|
|
||
|
|
Misc. API source file for generic 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.
|
||
|
|
|
||
|
|
09/2000 - Stephen Lodwick (STELO)
|
||
|
|
Ported OPK Wizard to Whistler
|
||
|
|
|
||
|
|
\****************************************************************************/
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Include file(s)
|
||
|
|
//
|
||
|
|
#include "pch.h"
|
||
|
|
#include "resource.h"
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Internal Defined Value(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
#define STR_URLDEF _T("http://")
|
||
|
|
#define STR_EVENT_CANCEL _T("SETUPMGR_EVENT_CANCEL")
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Internal Defined Macro(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
#define MALLOC(cb) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cb)
|
||
|
|
#define FREE(lp) ( (lp != NULL) ? ( (HeapFree(GetProcessHeap(), HEAP_NO_SERIALIZE, (LPVOID) lp)) ? ((lp = NULL) == NULL) : (FALSE) ) : (FALSE) )
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Internal Type Definition(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
typedef struct _COPYDIRDATA
|
||
|
|
{
|
||
|
|
HWND hwndParent;
|
||
|
|
LPTSTR lpSrc;
|
||
|
|
LPTSTR lpDst;
|
||
|
|
HANDLE hEvent;
|
||
|
|
} COPYDIRDATA, *PCOPYDIRDATA, *LPCOPYDIRDATA;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Internal Function Prototype(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
LRESULT CALLBACK CopyDirDlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
||
|
|
DWORD WINAPI CopyDirThread(LPVOID lpVoid);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// External Function(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
// If we find a key name with _Gray then *pfGray == TRUE
|
||
|
|
//
|
||
|
|
void ReadInstallInsKey(TCHAR szSection[], TCHAR szKey[], TCHAR szValue[], INT cchValue, TCHAR szIniFile[], BOOL* pfGray)
|
||
|
|
{
|
||
|
|
TCHAR szTempKey[MAX_PATH];
|
||
|
|
HRESULT hrCat;
|
||
|
|
|
||
|
|
if (!pfGray)
|
||
|
|
return;
|
||
|
|
|
||
|
|
lstrcpyn(szTempKey, szKey, AS(szTempKey));
|
||
|
|
if (!OpkGetPrivateProfileString(szSection, szTempKey, szValue, szValue, cchValue, szIniFile)) {
|
||
|
|
hrCat=StringCchCat(szTempKey, AS(szTempKey), GRAY);
|
||
|
|
if (OpkGetPrivateProfileString(szSection, szTempKey, szValue, szValue, cchValue, szIniFile))
|
||
|
|
*pfGray = TRUE;
|
||
|
|
else
|
||
|
|
*pfGray = TRUE; // default to unchecked if not found!
|
||
|
|
}
|
||
|
|
else
|
||
|
|
*pfGray = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
// If pfGrayed == TRUE then concatenate _Gray to the key name
|
||
|
|
//
|
||
|
|
void WriteInstallInsKey(TCHAR szSection[], TCHAR szKey[], TCHAR szValue[], TCHAR szIniFile[], BOOL fGrayed)
|
||
|
|
{
|
||
|
|
TCHAR szKeyTemp[MAX_PATH];
|
||
|
|
HRESULT hrCat;
|
||
|
|
|
||
|
|
// Clear the old value
|
||
|
|
//
|
||
|
|
lstrcpyn(szKeyTemp, szKey, AS(szKeyTemp));
|
||
|
|
OpkWritePrivateProfileString(szSection, szKeyTemp, NULL, szIniFile);
|
||
|
|
hrCat=StringCchCat(szKeyTemp, AS(szKeyTemp), GRAY);
|
||
|
|
OpkWritePrivateProfileString(szSection, szKeyTemp, NULL, szIniFile);
|
||
|
|
|
||
|
|
// Write the new value
|
||
|
|
lstrcpyn(szKeyTemp, szKey, AS(szKeyTemp));
|
||
|
|
if (fGrayed)
|
||
|
|
hrCat=StringCchCat(szKeyTemp, AS(szKeyTemp), GRAY);
|
||
|
|
OpkWritePrivateProfileString(szSection, szKeyTemp, szValue, szIniFile);
|
||
|
|
}
|
||
|
|
|
||
|
|
// NOTE: pszFileName must point to buffer at least length MAX_PATH
|
||
|
|
void CheckValidBrowseFolder(TCHAR* pszFileName)
|
||
|
|
{
|
||
|
|
if (NULL == pszFileName)
|
||
|
|
return;
|
||
|
|
|
||
|
|
// Last known good browse start folder
|
||
|
|
//
|
||
|
|
PathRemoveFileSpec(pszFileName);
|
||
|
|
if (!lstrlen(pszFileName))
|
||
|
|
lstrcpyn(pszFileName, g_App.szLastKnownBrowseFolder, MAX_PATH);
|
||
|
|
}
|
||
|
|
|
||
|
|
void SetLastKnownBrowseFolder(TCHAR* pszFileName)
|
||
|
|
{
|
||
|
|
if (NULL == pszFileName)
|
||
|
|
return;
|
||
|
|
|
||
|
|
// Save Last known good browse start folder
|
||
|
|
//
|
||
|
|
PathCombine(g_App.szLastKnownBrowseFolder, pszFileName, NULL);
|
||
|
|
PathRemoveFileSpec(g_App.szLastKnownBrowseFolder);
|
||
|
|
}
|
||
|
|
|
||
|
|
// NOTE: lpszURL is assumed to point to a buffer at least MAX_URL in length
|
||
|
|
BOOL ValidURL(LPTSTR lpszURL)
|
||
|
|
{
|
||
|
|
BOOL bResult = TRUE;
|
||
|
|
TCHAR szBuffer[MAX_PATH] = NULLSTR;
|
||
|
|
HRESULT hrCat;
|
||
|
|
|
||
|
|
// Check if valid URL
|
||
|
|
//
|
||
|
|
if ( !PathIsURL(lpszURL) )
|
||
|
|
{
|
||
|
|
// Check if empty string first
|
||
|
|
//
|
||
|
|
if (0 == lstrlen(lpszURL))
|
||
|
|
bResult = FALSE;
|
||
|
|
else {
|
||
|
|
// Currently not a valid URL, we are now going to prepend the
|
||
|
|
// URL with http:// and then test the validity again
|
||
|
|
//
|
||
|
|
lstrcpyn(szBuffer, STR_URLDEF, AS(szBuffer));
|
||
|
|
hrCat=StringCchCat(szBuffer, AS(szBuffer), lpszURL);
|
||
|
|
|
||
|
|
// Still not a valid URL or we were unable to copy the string
|
||
|
|
//
|
||
|
|
if ( !PathIsURL(szBuffer) ||
|
||
|
|
!lstrcpyn(lpszURL, szBuffer, MAX_URL) )
|
||
|
|
bResult = FALSE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return bResult;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL IsFolderShared(LPWSTR lpFolder, LPWSTR lpShare, DWORD cbShare)
|
||
|
|
{
|
||
|
|
LPSHARE_INFO_502 lpsi502 = NULL;
|
||
|
|
DWORD dwRead = 0,
|
||
|
|
dwTotal = 0;
|
||
|
|
NET_API_STATUS nas;
|
||
|
|
BOOL bRet = FALSE,
|
||
|
|
bBest = FALSE,
|
||
|
|
bBuffer = ( lpShare && cbShare );
|
||
|
|
PACL paclOld = NULL;
|
||
|
|
TCHAR szUnc[MAX_COMPUTERNAME_LENGTH + 4] = NULLSTR;
|
||
|
|
|
||
|
|
// Success or failure, we will always atleast pass back the computer
|
||
|
|
// name if they passed in a buffer. So here is where we create the
|
||
|
|
// computer name part of the path.
|
||
|
|
//
|
||
|
|
if ( bBuffer )
|
||
|
|
{
|
||
|
|
DWORD cbUnc = AS(szUnc) - 2;
|
||
|
|
HRESULT hrCat;
|
||
|
|
|
||
|
|
// We want to return the UNC path, so first need the \\ plus the
|
||
|
|
// computer name.
|
||
|
|
//
|
||
|
|
// NOTE: We hard coded the length of the "\\" string below as 2
|
||
|
|
// in two different places. Once just above, and once below
|
||
|
|
// in the GetComputerName() call. We also hard code the "\"
|
||
|
|
// string below as 1 when adding to the lenght of the string
|
||
|
|
// after adding the computer name. So don't forget these things
|
||
|
|
// if you make some changes here.
|
||
|
|
//
|
||
|
|
lstrcpyn(szUnc, _T("\\\\"), cbUnc);
|
||
|
|
if ( ( GetComputerName(szUnc + 2, &cbUnc) ) &&
|
||
|
|
( AS(szUnc) > ((DWORD) lstrlen(szUnc) + 1) ) )
|
||
|
|
{
|
||
|
|
// Added on a backslash so we can add the share name.
|
||
|
|
//
|
||
|
|
hrCat=StringCchCat(szUnc,AS(szUnc), _T("\\"));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// If GetComputerName() fails, that is bad. But we will just
|
||
|
|
// return the share name. That is about all we can do.
|
||
|
|
//
|
||
|
|
szUnc[0] = NULLCHR;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// Now share time, first retrieve all the shares on this machine.
|
||
|
|
//
|
||
|
|
nas = NetShareEnum(NULL, 502, (unsigned char **) &lpsi502, MAX_PREFERRED_LENGTH, &dwRead, &dwTotal, NULL);
|
||
|
|
|
||
|
|
// Make sure we got a list of shares, otherwise there is nothing.
|
||
|
|
// we can do. Because we specify MAX_PREFERRED_LENGTH, we should
|
||
|
|
// never get ERROR_MORE_DATA, but if for some reason we do there is
|
||
|
|
// no reason not to loop through the ones we did get.
|
||
|
|
//
|
||
|
|
if ( ( lpsi502 ) &&
|
||
|
|
( ( nas == NERR_Success ) || ( nas == ERROR_MORE_DATA ) ) )
|
||
|
|
{
|
||
|
|
int iLength = lstrlen(lpFolder);
|
||
|
|
LPTSTR lpSearch = lpFolder + iLength;
|
||
|
|
HRESULT hrCat;
|
||
|
|
|
||
|
|
// Trailing backslash is only bad if not the root folder.
|
||
|
|
//
|
||
|
|
if ( iLength > 3 )
|
||
|
|
{
|
||
|
|
// See if the folder has a trailing backslash.
|
||
|
|
//
|
||
|
|
lpSearch = CharPrev(lpFolder, lpSearch);
|
||
|
|
if ( *lpSearch == _T('\\') )
|
||
|
|
{
|
||
|
|
iLength--;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Go through all the shares until we fine the best
|
||
|
|
// one for this directory.
|
||
|
|
//
|
||
|
|
while ( dwRead-- && !bBest )
|
||
|
|
{
|
||
|
|
// See if this share is a disk share and is the
|
||
|
|
// same path passed in.
|
||
|
|
//
|
||
|
|
if ( ( lpsi502[dwRead].shi502_type == STYPE_DISKTREE ) &&
|
||
|
|
( StrCmpNI(lpsi502[dwRead].shi502_path, lpFolder, iLength) == 0 ) &&
|
||
|
|
( lstrlen(lpsi502[dwRead].shi502_path) == iLength ) )
|
||
|
|
{
|
||
|
|
// If this directory is shared more than once, we want to use
|
||
|
|
// the first one we fine with no security descriptor, because
|
||
|
|
// then it is most likely shared out to everyone.
|
||
|
|
//
|
||
|
|
if ( lpsi502[dwRead].shi502_security_descriptor == NULL )
|
||
|
|
{
|
||
|
|
// If there is no security descriptor, then everyone should have
|
||
|
|
// access and this is a good share.
|
||
|
|
//
|
||
|
|
bBest = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
// If we have no ACL, or we reset it because the new one is better,
|
||
|
|
// then we want to copy off the share name into our return buffer.
|
||
|
|
//
|
||
|
|
if ( !bRet || bBest )
|
||
|
|
{
|
||
|
|
// Return the share name for this directory in the supplied buffer
|
||
|
|
// (if the buffer is NULL or zero size, we just return TRUE so they
|
||
|
|
// know that the folder is shared even if they don't care what the
|
||
|
|
// name of the share is).
|
||
|
|
//
|
||
|
|
if ( bBuffer )
|
||
|
|
{
|
||
|
|
// Find out what we have room for in the return buffer.
|
||
|
|
//
|
||
|
|
if ( cbShare > (DWORD) (lstrlen(lpsi502[dwRead].shi502_netname) + lstrlen(szUnc)) )
|
||
|
|
{
|
||
|
|
// Copy the computer name and share into return buffer.
|
||
|
|
//
|
||
|
|
lstrcpyn(lpShare, szUnc, cbShare);
|
||
|
|
hrCat=StringCchCat(lpShare, cbShare, lpsi502[dwRead].shi502_netname);
|
||
|
|
}
|
||
|
|
else if ( cbShare > (DWORD) lstrlen(lpsi502[dwRead].shi502_netname) )
|
||
|
|
{
|
||
|
|
// Return buffer not big enough for both computer name and
|
||
|
|
// share name, so just return the share name.
|
||
|
|
//
|
||
|
|
lstrcpyn(lpShare, lpsi502[dwRead].shi502_netname,cbShare);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Not big enough for both, so return TRUE because we found a
|
||
|
|
// share, but don't return anything in the buffer.
|
||
|
|
//
|
||
|
|
*lpShare = NULLCHR;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// We found one, so always set this to TRUE.
|
||
|
|
//
|
||
|
|
bRet = TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Make sure and free the buffer returned by NetShareEnum().
|
||
|
|
//
|
||
|
|
if ( lpsi502 )
|
||
|
|
NetApiBufferFree(lpsi502);
|
||
|
|
|
||
|
|
// Now check to see if we didn't find the share, because we can
|
||
|
|
// still just return the computer name.
|
||
|
|
//
|
||
|
|
if ( ( !bRet && bBuffer ) &&
|
||
|
|
( cbShare > (DWORD) lstrlen(szUnc) ) )
|
||
|
|
{
|
||
|
|
lstrcpyn(lpShare, szUnc, cbShare);
|
||
|
|
}
|
||
|
|
|
||
|
|
return bRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CopyDirectoryDialog(HINSTANCE hInstance, HWND hwnd, LPTSTR lpSrc, LPTSTR lpDst)
|
||
|
|
{
|
||
|
|
COPYDIRDATA cdd;
|
||
|
|
|
||
|
|
// Pass in via the structure the source and destination the dialog needs
|
||
|
|
// to know about.
|
||
|
|
//
|
||
|
|
cdd.lpSrc = lpSrc;
|
||
|
|
cdd.lpDst = lpDst;
|
||
|
|
|
||
|
|
// Create the progress dialog.
|
||
|
|
//
|
||
|
|
return ( DialogBoxParam(hInstance, MAKEINTRESOURCE(IDD_PROGRESS), hwnd, CopyDirDlgProc, (LPARAM) &cdd) != 0 );
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL CopyResetFileErr(HWND hwnd, LPCTSTR lpSource, LPCTSTR lpTarget)
|
||
|
|
{
|
||
|
|
BOOL bReturn;
|
||
|
|
|
||
|
|
if ( !(bReturn = CopyResetFile(lpSource, lpTarget)) && hwnd )
|
||
|
|
MsgBox(hwnd, IDS_MISSINGFILE, IDS_APPNAME, MB_ERRORBOX, lpSource);
|
||
|
|
return bReturn;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// Internal Function(s):
|
||
|
|
//
|
||
|
|
|
||
|
|
LRESULT CALLBACK CopyDirDlgProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
||
|
|
{
|
||
|
|
static LPCOPYDIRDATA lpcdd = NULL;
|
||
|
|
|
||
|
|
switch (uMsg)
|
||
|
|
{
|
||
|
|
case WM_INITDIALOG:
|
||
|
|
|
||
|
|
// Make sure we have out copy directory data structure.
|
||
|
|
//
|
||
|
|
if ( lParam )
|
||
|
|
{
|
||
|
|
HANDLE hThread;
|
||
|
|
DWORD dwThreadId;
|
||
|
|
|
||
|
|
// Save off our lParam.
|
||
|
|
//
|
||
|
|
lpcdd = (LPCOPYDIRDATA) lParam;
|
||
|
|
|
||
|
|
// Replace the old parent with the new progress dialog parent.
|
||
|
|
//
|
||
|
|
lpcdd->hwndParent = hwnd;
|
||
|
|
|
||
|
|
// Need to pass in the cancel event as well.
|
||
|
|
//
|
||
|
|
lpcdd->hEvent = CreateEvent(NULL, TRUE, FALSE, STR_EVENT_CANCEL);
|
||
|
|
|
||
|
|
// Now create the thread that will copy the actual files.
|
||
|
|
//
|
||
|
|
if ( hThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) CopyDirThread, (LPVOID) lpcdd, 0, &dwThreadId) )
|
||
|
|
CloseHandle(hThread);
|
||
|
|
else
|
||
|
|
EndDialog(hwnd, 0);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
EndDialog(hwnd, 0);
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
case WM_COMMAND:
|
||
|
|
case WM_CLOSE:
|
||
|
|
|
||
|
|
// If we have an event, signal it, or just end the dialog.
|
||
|
|
//
|
||
|
|
if ( lpcdd && lpcdd->hEvent )
|
||
|
|
SetEvent(lpcdd->hEvent);
|
||
|
|
else
|
||
|
|
EndDialog(hwnd, 0);
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
case WM_DESTROY:
|
||
|
|
|
||
|
|
// If there is an event, get rid of it.
|
||
|
|
//
|
||
|
|
if ( lpcdd && lpcdd->hEvent )
|
||
|
|
{
|
||
|
|
CloseHandle(lpcdd->hEvent);
|
||
|
|
lpcdd->hEvent = NULL;
|
||
|
|
}
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
default:
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD WINAPI CopyDirThread(LPVOID lpVoid)
|
||
|
|
{
|
||
|
|
LPCOPYDIRDATA lpcdd = (LPCOPYDIRDATA) lpVoid;
|
||
|
|
HWND hwnd = lpcdd->hwndParent,
|
||
|
|
hwndProgress = GetDlgItem(hwnd, IDC_PROGRESS);
|
||
|
|
HANDLE hEvent = lpcdd->hEvent;
|
||
|
|
DWORD dwRet = 0;
|
||
|
|
LPTSTR lpSrc = lpcdd->lpSrc,
|
||
|
|
lpDst = lpcdd->lpDst;
|
||
|
|
|
||
|
|
// First we need to create the path.
|
||
|
|
//
|
||
|
|
if ( CreatePath(lpDst) )
|
||
|
|
{
|
||
|
|
// Setup the progress bar.
|
||
|
|
//
|
||
|
|
SendMessage(hwndProgress, PBM_SETSTEP, 1, 0L);
|
||
|
|
SendMessage(hwndProgress, PBM_SETRANGE32, 0, (LPARAM) FileCount(lpSrc));
|
||
|
|
|
||
|
|
// Now copy the directory.
|
||
|
|
//
|
||
|
|
if ( CopyDirectoryProgressCancel(hwndProgress, hEvent, lpSrc, lpDst) )
|
||
|
|
dwRet = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Now end the dialog with our error code and return.
|
||
|
|
//
|
||
|
|
EndDialog(hwnd, dwRet);
|
||
|
|
return dwRet;
|
||
|
|
}
|