/*++

  Copyright (c) 1995-97 Microsoft Corporation
  All rights reserved.

  Module Name:

        SrvInst.c

  Purpose:


        Server side install code.  This code will be called from a process created by the spooler to do a
        "server" side install of a printer driver.

  Author:

        Patrick Vine (pvine) - 22 March 2000

  Revision History:

--*/

#include "precomp.h"

#pragma hdrstop
#include "srvinst.hxx"

const TCHAR   gcszNTPrint[]  = _TEXT("inf\\ntprint.inf");
const TCHAR   gcszPrintKey[] = _TEXT("SYSTEM\\CurrentControlSet\\Control\\Print");
const TCHAR   gcszTimeOut[]  = _TEXT("ServerInstallTimeOut");
const TCHAR   gcSpace        = _TEXT(' ');
#define DEFAULT_MAX_TIMEOUT    300000 // 5 minute timeout.


/*++

Routine Name:

    ServerInstall

Routine Description:

    Server side install code to be called by a process created by spooler.

Arguments:

    hwnd            - Window handle of stub window.
    hInstance,      - Rundll instance handle.
    pszCmdLine      - Pointer to command line.
    nCmdShow        - Show command value always TRUE.

Return Value:

    Returns the last error code.  This can be read by the spooler by getting the return code from the process.

--*/
DWORD
ServerInstallW(
    IN HWND        hwnd,
    IN HINSTANCE   hInstance,
    IN LPCTSTR     pszCmdLine,
    IN UINT        nCmdShow
    )
{
    CServerInstall Installer;

    if( Installer.ParseCommand( (LPTSTR)pszCmdLine) && Installer.OpenPipe())
    {
        if( Installer.GetInstallParameters() )
            Installer.InstallDriver();

        Installer.SendError();
        Installer.ClosePipe();
    }

    return Installer.GetLastError();
}

////////////////////////////////////////////////////////////////////////////////
//
// Method definitions for CServerInstall Class.
//
////////////////////////////////////////////////////////////////////////////////

CServerInstall::
CServerInstall() : _dwLastError(ERROR_SUCCESS),
                   _tsDriverName(),
                   _tsInf(),
                   _tsSource(),
                   _tsFlags(),
                   _tsPipeName(),
                   _hPipe(INVALID_HANDLE_VALUE)
{
    SetMaxTimeOut();
}

CServerInstall::
~CServerInstall()
{
    ClosePipe();
}

void
CServerInstall::
SetMaxTimeOut()
{
    HKEY  hKey;
    DWORD dwDummy;
    DWORD dwSize = sizeof(_dwMaxTimeOut);

    _dwMaxTimeOut = DEFAULT_MAX_TIMEOUT;
    if( RegOpenKeyEx(HKEY_LOCAL_MACHINE, gcszPrintKey, 0, KEY_READ, &hKey) == ERROR_SUCCESS )
    {
        if(RegQueryValueEx( hKey, gcszTimeOut, 0, &dwDummy, (LPBYTE)&_dwMaxTimeOut, &dwSize ) != ERROR_SUCCESS)
        {
            _dwMaxTimeOut = DEFAULT_MAX_TIMEOUT;
        }
        RegCloseKey( hKey );
    }
}

BOOL
CServerInstall::
InstallDriver()
{

    if(  SetInfDir() &&
         bValidateSourcePath() &&
         DriverNotInstalled() )
    {
        _dwLastError = ::InstallDriverSilently(_tsInf, _tsDriverName, _tsSource);
    }

    return (_dwLastError == ERROR_SUCCESS);
}


/*++

    Parameter structure:

          1st word : Flags = default == 0 for now

          if flags = 0
             2nd word : Pipe name to open

--*/
BOOL
CServerInstall::
ParseCommand( LPTSTR pszCommandStr )
{
    TCHAR * pTemp;
    DWORD   dwCount = 0;

    //
    // If we don't have a valid command string
    //
    if( !pszCommandStr || !*pszCommandStr )
    {
        _dwLastError = ERROR_INVALID_PARAMETER;
        goto Cleanup;
    }

    //
    // Lets grab the flags field
    //
    pTemp = _tcschr( pszCommandStr, gcSpace );
    if( !pTemp )
    {
        //
        // No flags field, fail.
        //
        _dwLastError = ERROR_INVALID_PARAMETER;
        goto Cleanup;
    }

    *(pTemp++) = 0;
    if( !_tsFlags.bUpdate( pszCommandStr ))
    {
        _dwLastError = ::GetLastError();
        goto Cleanup;
    }

    //
    //  Currently we only have one case - so we don't need to worry about the
    //  flags nor branch on them.  however we may want to in the future.
    //

    //
    // The rest of the command line is the pipe's name.
    //
    if( !_tsPipeName.bUpdate( pTemp ))
        _dwLastError = ::GetLastError();

Cleanup:
    return (_dwLastError == ERROR_SUCCESS);
}

BOOL
CServerInstall::
GetInstallParameters()
{
    if(!GetOneParam( &_tsDriverName ))
        goto Done;

    if( _tsDriverName.bEmpty() )
    {
        _dwLastError = ERROR_INVALID_PARAMETER;
        goto Done;
    }

    if(!GetOneParam( &_tsInf ))
        goto Done;

    GetOneParam( &_tsSource );

Done:
    return (_dwLastError == ERROR_SUCCESS);
}


//
// Read the size of the string to follow from the pipe.
// Then reads the string and places it in the TString passed.
//
BOOL
CServerInstall::
GetOneParam( TString * tString )
{
    DWORD  dwSize    = 0;
    DWORD  dwRet     = 0;
    LPTSTR pszString = NULL;
    LPVOID pData     = NULL;

    if( !ReadOverlapped( _hPipe, &dwSize, sizeof(dwSize), &dwRet ))
        goto Done;

    if( dwSize == 0 )
        goto Done;

    //
    // The data that we're receiving will be WCHARs as spooler only works with UNICODE.
    //
    if(!(pData = LocalAllocMem((dwSize + 1)*sizeof(WCHAR))))
    {
        _dwLastError = ::GetLastError();
        goto Done;
    }

    if( !ReadOverlapped( _hPipe, pData, dwSize*sizeof(WCHAR), &dwRet ) )
        goto Done;

    if( dwRet != dwSize*sizeof(WCHAR) )
        _dwLastError = ERROR_INVALID_PARAMETER;

//
// Because ntprint compiles to both unicode and ansi we need to do this conversion.
// The string coming in will be unicode as it comes from spooler which only
// uses wchars.
//
#ifdef UNICODE
    pszString = (LPTSTR)pData;
    pData = NULL;
#else
    //Get the length of the unicode string
    dwSize = WideCharToMultiByte( CP_ACP, 0, (LPWSTR)pData, -1, NULL, 0, NULL, NULL );

    //Create the TCHAR string of the same length
    if( !(pszString = (LPTSTR)LocalAllocMem((dwSize + 1)*sizeof(TCHAR))))
        goto Done;

    //Convert the string from unicode to ansi.
    if( !WideCharToMultiByte( CP_ACP, 0, (LPWSTR)pData, -1, pszString, dwSize, NULL, NULL ) )
    {
        _dwLastError = ::GetLastError();
    }
#endif

Done:
    if( !tString->bUpdate( pszString ))
        _dwLastError = ::GetLastError();

    if( pData )
        LocalFreeMem( pData );

    if( pszString )
        LocalFreeMem( pszString );

    return (_dwLastError == ERROR_SUCCESS);
}


DWORD
CServerInstall::
GetLastError()
{
    SetLastError(_dwLastError);
    return _dwLastError;
}


BOOL
CServerInstall::
SetInfDir()
{
    if( _tsInf.bEmpty() )
    {
        SetInfToNTPRINTDir();
    }
    else
    {
        //
        // The inf name must be fully qualified.
        //
        TCHAR  szFullInfName[MAX_PATH];
        LPTSTR pszDummy;
        DWORD  dwLength = GetFullPathName( _tsInf, COUNTOF( szFullInfName ), szFullInfName, &pszDummy );

        if( !dwLength || !_tsInf.bUpdate( szFullInfName ))
            _dwLastError = ::GetLastError();
    }

    return (_dwLastError == ERROR_SUCCESS);
}


//
//  Returns: TRUE if SUCCESS, FALSE otherwise
//
//  Sets the _stInf string to contain %windir%\inf\ntprint.inf
//
BOOL
CServerInstall::
SetInfToNTPRINTDir()
{
    UINT   uiSize        = 0;
    UINT   uiAllocSize   = 0;
    PTCHAR pData         = NULL;
    TCHAR  szNTPrintInf[MAX_PATH];

    _dwLastError = ERROR_INVALID_DATA;

    //
    //  Get %windir%
    //  If the return is 0 - the call failed.
    //  If the return is greater than MAX_PATH we want to fail as something has managed to change
    //  the system dir to longer than MAX_PATH which is invalid.
    //
    uiSize = GetSystemWindowsDirectory( szNTPrintInf, COUNTOF(szNTPrintInf) );
    if( !uiSize || uiSize > COUNTOF(szNTPrintInf) )
        goto Cleanup;

    //
    // If we don't end in a \ then add one.
    //
    pData = &szNTPrintInf[ _tcslen(szNTPrintInf) ];
    if( *pData != _TEXT('\\') )
        *(pData++) = _TEXT('\\');

    *(pData) = 0;
    uiSize = _tcslen( szNTPrintInf ) + _tcslen( gcszNTPrint ) + 1;

    //
    // If what we've got sums up to a longer string than the allowable length MAX_PATH - fail
    //
    if( uiSize > COUNTOF(szNTPrintInf) )
        goto Cleanup;

    //
    //  Copy the inf\ntprint.inf string onto the end of the %windir%\ string.
    //
    _tcscpy( pData, gcszNTPrint );

    _dwLastError = ERROR_SUCCESS;

Cleanup:

    if( _dwLastError != ERROR_SUCCESS && szNTPrintInf )
    {
        //
        // Got here due to some error.  Get what the called function set the last error to.
        // If the function set a success, set some error code.
        //
        if( (_dwLastError = ::GetLastError()) == ERROR_SUCCESS )
            _dwLastError = ERROR_INVALID_DATA;

        szNTPrintInf[0] = 0;
    }

    if( !_tsInf.bUpdate( szNTPrintInf ) )
        _dwLastError = ::GetLastError();

    return (_dwLastError == ERROR_SUCCESS);
}


BOOL
CServerInstall::
bValidateSourcePath(
    )
{
    if( !_tsSource.bEmpty() &&
        !(GetFileAttributes( (LPCTSTR)_tsSource ) & FILE_ATTRIBUTE_DIRECTORY) )
    {
        _dwLastError = ERROR_DIRECTORY;
    }

    return (_dwLastError == ERROR_SUCCESS);
}


BOOL
CServerInstall::
OpenPipe()
{
    if( !_tsPipeName.bEmpty() )
    {
        _hPipe = CreateFile( _tsPipeName,
                             GENERIC_WRITE | GENERIC_READ,
                             0,
                             NULL,
                             OPEN_EXISTING,
                             FILE_FLAG_OVERLAPPED,
                             NULL );

        if( _hPipe == INVALID_HANDLE_VALUE )
            _dwLastError = ::GetLastError();
    }
    else
        _dwLastError = ERROR_INVALID_HANDLE;

    return (_dwLastError == ERROR_SUCCESS);
}


BOOL
CServerInstall::
SendError()
{
    DWORD dwDontCare;

    return (WriteOverlapped( _hPipe, &_dwLastError, sizeof(_dwLastError), &dwDontCare ));
}


BOOL
CServerInstall::
ClosePipe()
{
    BOOL bRet = TRUE;

    if( _hPipe != INVALID_HANDLE_VALUE )
    {
        bRet = CloseHandle( _hPipe );
        _hPipe = INVALID_HANDLE_VALUE;
    }

    return bRet;
}

BOOL
CServerInstall::
ReadOverlapped( HANDLE  hFile,
                LPVOID  lpBuffer,
                DWORD   nNumberOfBytesToRead,
                LPDWORD lpNumberOfBytesRead )
{
    if( hFile != INVALID_HANDLE_VALUE )
    {
        OVERLAPPED Ov;

        ZeroMemory( &Ov,sizeof(Ov));

        if ( !(Ov.hEvent = CreateEvent(NULL,TRUE,FALSE,NULL)) )
        {
            _dwLastError = ::GetLastError();
            goto Cleanup;
        }

        if( !ReadFile( hFile, lpBuffer, nNumberOfBytesToRead, lpNumberOfBytesRead, &Ov ) &&
            ::GetLastError() != ERROR_IO_PENDING )
        {
            _dwLastError = ::GetLastError();
            goto Cleanup;
        }

        if( WaitForSingleObject(Ov.hEvent, _dwMaxTimeOut) == WAIT_TIMEOUT )
        {
            CancelIo(hFile);
            WaitForSingleObject(Ov.hEvent, INFINITE);
        }

        if( !GetOverlappedResult(hFile, &Ov, lpNumberOfBytesRead, FALSE) )
            _dwLastError = ::GetLastError();

Cleanup:
        if ( Ov.hEvent )
            CloseHandle(Ov.hEvent);
    }
    else
        _dwLastError = ERROR_INVALID_HANDLE;

    return (_dwLastError == ERROR_SUCCESS);
}

BOOL
CServerInstall::
WriteOverlapped( HANDLE  hFile,
                 LPVOID  lpBuffer,
                 DWORD   nNumberOfBytesToRead,
                 LPDWORD lpNumberOfBytesRead )
{
    if( hFile != INVALID_HANDLE_VALUE )
    {
        OVERLAPPED Ov;

        ZeroMemory( &Ov,sizeof(Ov));

        if ( !(Ov.hEvent = CreateEvent(NULL,TRUE,FALSE,NULL)) )
        {
            _dwLastError = ::GetLastError();
            goto Cleanup;
        }

        if( !WriteFile( hFile, lpBuffer, nNumberOfBytesToRead, lpNumberOfBytesRead, &Ov ) &&
            ::GetLastError() != ERROR_IO_PENDING )
        {
            _dwLastError = ::GetLastError();
            goto Cleanup;
        }

        if( WaitForSingleObject(Ov.hEvent, _dwMaxTimeOut) == WAIT_TIMEOUT )
        {
            CancelIo(hFile);
            WaitForSingleObject(Ov.hEvent, INFINITE);
        }

        if( !GetOverlappedResult(hFile, &Ov, lpNumberOfBytesRead, FALSE) )
            _dwLastError = ::GetLastError();

Cleanup:
        if ( Ov.hEvent )
            CloseHandle(Ov.hEvent);
    }
    else
        _dwLastError = ERROR_INVALID_HANDLE;

    return (_dwLastError == ERROR_SUCCESS);
}


/*+

  This function enumerates the drivers and finds if there is one of the same name currently installed.
  If there is then open the inf to install with and verify that the inf's version date is newer than the
  already installed driver.

  Returns: TRUE  - if anything fails or the installed date isn't newer than the inf date.
           FALSE - only if the driver is installed AND it's date is newer than the inf's date.

-*/
BOOL
CServerInstall::
DriverNotInstalled()
{
    LPCTSTR             pszKey       = _TEXT("DriverVer");
    LPTSTR              pszEntry     = NULL;
    LPDRIVER_INFO_6     pDriverInfo6 = NULL;
    LPBYTE              pBuf         = NULL;
    PSP_INF_INFORMATION pInfo        = NULL;
    SYSTEMTIME          Time         = {0};
    BOOL                bRet         = TRUE;
    DWORD               dwLength,
                        dwRet,
                        dwIndex;
    TCHAR               *pTemp,
                        *pTemp2;

    if(!EnumPrinterDrivers( NULL, PlatformEnv[MyPlatform].pszName, 6, pBuf, 0, &dwLength, &dwRet ))
    {
        if( ::GetLastError() != ERROR_INSUFFICIENT_BUFFER )
            return TRUE;

        if( (pBuf = (LPBYTE) AllocMem( dwLength )) == NULL ||
            !EnumPrinterDrivers( NULL, PlatformEnv[MyPlatform].pszName, 6, pBuf, dwLength, &dwLength, &dwRet ))
        {
            _dwLastError = ::GetLastError();
            goto Cleanup;
        }
    }
    else
    {
        //
        // Only way this could succeed is if no drivers installed.
        //
        _dwLastError = ERROR_UNKNOWN_PRINTER_DRIVER;
        return TRUE;
    }

    for( dwIndex = 0, pDriverInfo6 = (LPDRIVER_INFO_6)pBuf; dwIndex < dwRet; dwIndex++, pDriverInfo6++ )
    {
        if( _tcscmp( pDriverInfo6->pName, (LPCTSTR)_tsDriverName ) == 0 )
            break;
    }

    if(dwIndex >= dwRet)
    {
        //
        // Driver not found
        //
        _dwLastError = ERROR_UNKNOWN_PRINTER_DRIVER;
        goto Cleanup;
    }

    //
    // The driver has been found...  Open up inf and look at it's date.
    //

    //
    // Firstly get the size that will be needed for pInfo.
    //
    if( !SetupGetInfInformation( (LPCTSTR)_tsInf, INFINFO_INF_NAME_IS_ABSOLUTE, pInfo, 0, &dwLength ) )
    {
        _dwLastError = ::GetLastError();
        goto Cleanup;
    }

    //
    // Alloc pInfo and fill it.
    //
    if( (pInfo = (PSP_INF_INFORMATION) AllocMem( dwLength )) != NULL &&
        SetupGetInfInformation( (LPCTSTR)_tsInf, INFINFO_INF_NAME_IS_ABSOLUTE, pInfo, dwLength, &dwLength ) )
    {
        //
        //  Get the size of the date string
        //
        if( SetupQueryInfVersionInformation( pInfo, 0, pszKey, pszEntry, 0, &dwLength ))
        {
            //
            // Alloc pszEntry and fill it.
            //
            if( (pszEntry = (LPTSTR) AllocMem( dwLength*sizeof(TCHAR) )) != NULL &&
                SetupQueryInfVersionInformation( pInfo, 0, pszKey, pszEntry, dwLength, &dwLength ))
            {
                //
                // Now convert the date string into a SYSTEMTIME
                // Date is of the form 03/22/2000
                //
                // Get the month - 03 part
                //
                if( (pTemp = _tcschr( pszEntry, _TEXT('/'))) != NULL )
                {
                    *pTemp++ = 0;
                    Time.wMonth = (WORD)_ttoi( pszEntry );
                    pTemp2 = pTemp;

                    //
                    // Get the day - 22 part
                    //
                    if( (pTemp = _tcschr( pTemp2, _TEXT('/'))) != NULL )
                    {
                        *pTemp++ = 0;
                        Time.wDay = (WORD)_ttoi( pTemp2 );
                        pTemp2 = pTemp;

                        //
                        // Get the year - 2000 part
                        //
                        pTemp = _tcschr( pTemp2, _TEXT('/'));
                        if( pTemp )
                            *pTemp = 0;
                        Time.wYear = (WORD)_ttoi( pTemp2 );
                    }
                    else
                        _dwLastError = ERROR_INVALID_PARAMETER;
                }
                else
                    _dwLastError = ERROR_INVALID_PARAMETER;
            }
            else
                _dwLastError = ::GetLastError();
        }
        else
            _dwLastError = ::GetLastError();
    }
    else
        _dwLastError = ::GetLastError();


    //
    //  If we got all the way to filling in the year, we may have something useful...
    //
    if( Time.wYear )
    {
        FILETIME ftTime = {0};
        if(SystemTimeToFileTime( &Time, &ftTime ))
        {
            //
            // If the inf time is more recent than what is installed,
            // reinstall, otherwise don't
            //
            if( CompareFileTime(&ftTime, &pDriverInfo6->ftDriverDate) < 1 )
            {
                bRet = FALSE;
            }
        }
        //
        // Getting here and return TRUE or FALSE is still a successful call.
        //
        _dwLastError = ERROR_SUCCESS;
    }
    else
        _dwLastError = ERROR_INVALID_PARAMETER;

Cleanup:
    if( pBuf )
        FreeMem( pBuf );

    if( pInfo )
        FreeMem( pInfo );

    if( pszEntry )
        FreeMem( pszEntry );

    return bRet;
}