windows-xp/Source/XPSP1/NT/base/ntsetup/opktools/setupmgr/oem/miscapi.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;
}