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.
 
 
 
 
 
 

826 lines
17 KiB

/*++
Copyright (C) Microsoft Corporation, 1995 - 1998
All rights reserved.
Module Name:
F:\nt\private\windows\spooler\printui.pri\procdlg.cxx
Abstract:
Print Processor dialog.
Author:
Steve Kiraly (SteveKi) 11/10/95
Revision History:
--*/
#include "precomp.hxx"
#pragma hdrstop
#include "procdlg.hxx"
#include "psetup.hxx"
#include "drvver.hxx"
/********************************************************************
Print Processor Dialog class
********************************************************************/
/*++
Routine Name:
TPrintProcessor
Routine Description:
Contstucts the print processor dialog.
Arguments:
hWnd - Parent window handle
strServerName - Currnet printer server name
strPrintProcessor - Current print processor
strDatatype - Current datatype
Return Value:
Nothing. bValid() inidicated valid object.
--*/
TPrintProcessor::
TPrintProcessor(
IN HWND hWnd,
IN LPCTSTR pszServerName,
IN TString &strPrintProcessor,
IN TString &strDatatype,
IN BOOL bAdministrator
) : _hWnd( hWnd ),
_pPrintProcessors( NULL ),
_cPrintProcessors( 0 ),
_hctlPrintProcessorList( 0 ),
_hctlDatatypeList( 0 ),
_bAdministrator( bAdministrator ),
_bValid( FALSE ),
_pszServerName( pszServerName )
{
DBGMSG( DBG_TRACE, ( "TPrintProcessor::ctor\n") );
DBGMSG( DBG_TRACE, ( "ServerName = " TSTR "\n", DBGSTR( (LPCTSTR)pszServerName ) ) );
DBGMSG( DBG_TRACE, ( "PrintProcessorName = " TSTR "\n", (LPCTSTR)strPrintProcessor ) );
DBGMSG( DBG_TRACE, ( "DataType = " TSTR "\n", (LPCTSTR)strDatatype ) );
//
// Update the string oobjects.
//
if( !_strPrintProcessor.bUpdate( strPrintProcessor ) ||
!_strDatatype.bUpdate( strDatatype ) ){
DBGMSG( DBG_WARN, ( "String update failed with %d\n", GetLastError() ) );
return;
}
//
// Do loading for this object.
//
if( !bLoad( ) ){
return;
}
_bValid = TRUE;
}
/*++
Routine Name:
~TPrintProcessor
Routine Description:
Destructs the print processor dialog.
Arguments:
None.
Return Value:
Nothing.
--*/
TPrintProcessor::
~TPrintProcessor(
)
{
DBGMSG( DBG_TRACE, ( "TPrintProcessor::dtor\n") );
FreeMem( _pPrintProcessors );
}
/*++
Routine Name:
bValid
Routine Description:
Returns valid object indicator.
Arguments:
None.
Return Value:
Nothing.
--*/
BOOL
TPrintProcessor::
bValid(
VOID
)
{
return _bValid;
}
/*++
Routine Name:
bDoModal
Routine Description:
Start modal execution of dialog.
Arguments:
None.
Return Value:
TRUE dialog ok button chosen.
FALSE cancel button chosen.
--*/
BOOL
TPrintProcessor::
bDoModal(
VOID
)
{
//
// Create a modal dialog.
//
return (BOOL)DialogBoxParam( ghInst,
MAKEINTRESOURCE( TPrintProcessor::kResourceId ),
_hWnd,
MGenericDialog::SetupDlgProc,
(LPARAM)this );
}
/*++
Routine Name:
bSetUI
Routine Description:
Sets the data on the dialog.
Arguments:
None.
Return Value:
TRUE data set on UI successfully.
FALSE error occurred setting UI data.
--*/
BOOL
TPrintProcessor::
bSetUI(
VOID
)
{
//
// Create local copies of the control ID's this saves
// some execution time and alot of typing.
//
_hctlPrintProcessorList = GetDlgItem( _hDlg, IDC_PRINT_PROCESSOR_LIST );
_hctlDatatypeList = GetDlgItem( _hDlg, IDC_PRINT_DATATYPE_LIST );
//
// Set the UI controls.
//
if( !_hctlPrintProcessorList ||
!_hctlDatatypeList ||
!bSetPrintProcessorList() ||
!bSetDatatypeList() ||
!bDataTypeAssociation( TRUE ) ){
DBGMSG( DBG_WARN, ( "bSetUI failed %d\n", GetLastError( )));
return FALSE;
}
//
// If not an administrator disable the controls.
//
vEnableCtl( _hDlg, IDC_PRINT_PROCESSOR_LIST, _bAdministrator );
vEnableCtl( _hDlg, IDC_PRINT_DATATYPE_LIST, _bAdministrator );
vEnableCtl( _hDlg, IDC_SPOOL_DATATYPE, _bAdministrator );
vEnableCtl( _hDlg, IDOK, _bAdministrator );
vEnableCtl( _hDlg, IDC_PRINT_PROCESSOR_DESC, _bAdministrator );
vEnableCtl( _hDlg, IDC_PRINT_PROCESSOR_TEXT, _bAdministrator );
vEnableCtl( _hDlg, IDC_PRINT_DATATYPE_TEXT, _bAdministrator );
return TRUE;
}
/*++
Routine Name:
bReadUI
Routine Description:
Reads the UI data back into the public members.
Arguments:
None.
Return Value:
TRUE data read ok.
FALSE error reading UI data.
--*/
BOOL
TPrintProcessor::
bReadUI(
VOID
)
{
UINT uSel;
CAutoHandleHLOCAL spText;
//
// If we do not have administrator privilages.
//
if ( !_bAdministrator )
return FALSE;
//
// Read the selected print proccessor form the list box.
//
spText = NULL;
uSel = ListBox_GetCurSel( _hctlPrintProcessorList );
if(( uSel == LB_ERR ) ||
FAILED( LBGetTextHelper( _hctlPrintProcessorList, uSel, (LPWSTR *)&spText ) ) ||
!_strPrintProcessor.bUpdate( spText.GetPtrAs<LPWSTR>() )){
DBGMSG( DBG_WARN, ( "Read print processor listbox failed %d\n", GetLastError( )));
return FALSE;
}
//
// Read the selected datatype form the list box.
//
spText = NULL;
uSel = ListBox_GetCurSel( _hctlDatatypeList );
if(( uSel == LB_ERR ) ||
FAILED( LBGetTextHelper( _hctlDatatypeList, uSel, (LPWSTR *)&spText ) ) ||
!_strDatatype.bUpdate( spText.GetPtrAs<LPWSTR>() )){
DBGMSG( DBG_WARN, ( "Read datatype listbox failed %d\n", GetLastError( )));
return FALSE;
}
//
// Trace message to display UI data.
//
DBGMSG( DBG_TRACE, ( "PrintProcessorName = " TSTR "\n", (LPCTSTR)strPrintProcessor() ) );
DBGMSG( DBG_TRACE, ( "DataType = " TSTR "\n", (LPCTSTR)strDatatype() ) );
return TRUE;
}
/*++
Routine Name:
bSetPrintProcessorList
Routine Description:
Fills the print processors list box.
Arguments:
None.
Return Value:
TRUE list box fill successfully.
FALSE if error occurred.
--*/
BOOL
TPrintProcessor::
bSetPrintProcessorList(
VOID
)
{
//
// Reset the list box in case we are called to refresh.
//
ListBox_ResetContent( _hctlPrintProcessorList );
//
// Build list of print processors.
//
UINT i;
for( i = 0; i < _cPrintProcessors; i++ ){
ListBox_InsertString( _hctlPrintProcessorList, -1, (LPARAM)_pPrintProcessors[i].pName );
ListBox_SetItemData( _hctlPrintProcessorList, i, 0 );
}
//
// Set the highlight on the current print processor.
//
UINT uSel;
uSel = ListBox_FindString( _hctlPrintProcessorList, -1, _strPrintProcessor );
uSel = ( uSel == LB_ERR ) ? 0 : uSel;
ListBox_SetCurSel( _hctlPrintProcessorList, uSel );
return TRUE;
}
/*++
Routine Name:
bSetDatatypeList
Routine Description:
Fills the datatype list box.
Arguments:
None.
Return Value:
TRUE list box fill successfully.
FALSE if error occurred.
--*/
BOOL
TPrintProcessor::
bSetDatatypeList(
VOID
)
{
HRESULT hr = S_OK;
BOOL bStatus = FALSE;
CAutoHandleHLOCAL spText;
UINT uSel;
//
// Get the currently selected print processor name.
//
uSel = ListBox_GetCurSel( _hctlPrintProcessorList );
hr = LBGetTextHelper( _hctlPrintProcessorList, uSel, (LPWSTR *)&spText );
bStatus = SUCCEEDED(hr);
if( bStatus ){
//
// Enumerate the data types.
//
DATATYPES_INFO_1 *pDatatypes = NULL;
DWORD cDatatypes = 0;
bStatus = bEnumPrintProcessorDatatypes( (LPTSTR)_pszServerName,
spText.GetPtrAs<LPWSTR>(),
1,
(PVOID *)&pDatatypes,
&cDatatypes );
if( bStatus ){
//
// Reset the list box in case we are called to refresh.
//
ListBox_ResetContent( _hctlDatatypeList );
//
// Build list of datatypes.
//
UINT i;
for( i = 0; i < cDatatypes; i++ ){
ListBox_InsertString( _hctlDatatypeList, -1, (LPARAM)pDatatypes[i].pName );
}
}
//
// Clean up any allocated resources.
//
FreeMem( pDatatypes );
//
// Select the correct datatype for the slected print processor.
//
ListBox_SetCurSel( _hctlDatatypeList,
ListBox_GetItemData( _hctlPrintProcessorList,
ListBox_GetCurSel( _hctlPrintProcessorList ) ) );
} else {
//
// LBGetTextHelper failed. Set the last error.
//
SetLastError(HRESULT_CODE(hr));
}
return bStatus;
}
/*++
Routine Name:
bDataTypeAssociation
Routine Description:
The currently selected datatype item is tracted for
each print processor. The print processor list box item
data contains the index of its corresponding datatype.
Arguments:
TRUE to select the highlight on the default datatype.
FALSE to the association for the currently selected datatype.
Return Value:
Always returns success
--*/
BOOL
TPrintProcessor::
bDataTypeAssociation(
IN BOOL bSetDatatype
)
{
//
// Set the highlight using the current datatype.
//
if( bSetDatatype ){
//
// Locate and select the index of the default datatype. The
// default datatype is the datatype string passed into this object.
//
UINT uSel;
uSel = ListBox_FindString( _hctlDatatypeList, -1, _strDatatype );
uSel = ( uSel == LB_ERR ) ? 0 : uSel;
ListBox_SetCurSel( _hctlDatatypeList, uSel );
}
//
// Get the currently selected print processor and set the item data
// to associated selected data type.
//
ListBox_SetItemData( _hctlPrintProcessorList,
ListBox_GetCurSel( _hctlPrintProcessorList ),
ListBox_GetCurSel( _hctlDatatypeList ) );
return TRUE;
}
/*++
Routine Name:
bLoad
Routine Description:
Gets the list of print processors.
Arguments:
None.
Return Value:
TRUE list box fill successfully.
FALSE if error occurred.
--*/
BOOL
TPrintProcessor::
bLoad(
VOID
)
{
//
// Get the current driver / version.
//
DWORD dwDriverVersion = 0;
if( !bGetCurrentDriver( _pszServerName, &dwDriverVersion ) ){
DBGMSG( DBG_WARN, ( "bGetDriverVersion failed.\n" ) );
return FALSE;
}
DBGMSG( DBG_TRACE, ( "Driver Version %d\n", dwDriverVersion ) );
//
// Convert the driver / version to spooler usable environment string.
//
TString strDriverEnv;
if( !bGetDriverEnv( dwDriverVersion, strDriverEnv ) ){
DBGMSG( DBG_WARN, ( "bGetDriverEnv failed.\n" ) );
return FALSE;
}
DBGMSG( DBG_TRACE, ( "Driver Environment " TSTR "\n", (LPCTSTR)strDriverEnv ) );
//
// Enumerate the currently installed print processors.
//
BOOL bStatus;
bStatus = bEnumPrintProcessors( (LPTSTR)_pszServerName,
(LPTSTR)(LPCTSTR)strDriverEnv,
1,
(PVOID *)&_pPrintProcessors,
&_cPrintProcessors );
if( !bStatus ){
DBGMSG( DBG_ERROR, ( "bEnumPrintProccessors failed = %d\n", GetLastError () ) );
return FALSE;
}
return TRUE;
}
/*++
Routine Name:
bHandleMesage
Routine Description:
Dialog message handler.
Arguments:
IN UINT uMsg,
IN WPARAM wParam,
IN LPARAM lParam
Return Value:
TRUE message was handled.
FALSE message was not handled.
--*/
BOOL
TPrintProcessor::
bHandleMessage(
UINT uMsg,
WPARAM wParam,
LPARAM lParam
)
{
UNREFERENCED_PARAMETER( lParam );
BOOL bStatus = FALSE;
switch( uMsg ){
case WM_INITDIALOG:
bSetUI( );
bStatus = TRUE;
break;
case WM_HELP:
case WM_CONTEXTMENU:
bStatus = PrintUIHelp( uMsg, _hDlg, wParam, lParam );
break;
case WM_COMMAND:
switch( GET_WM_COMMAND_ID( wParam, lParam )){
//
// Read back the UI data and indicate success.
//
case IDOK:
bStatus = bReadUI();
EndDialog( _hDlg, bStatus );
break;
//
// Indicate cancel request.
//
case IDCANCEL:
bStatus = TRUE;
EndDialog( _hDlg, FALSE );
break;
//
// Handle the print processor list change.
//
case IDC_PRINT_PROCESSOR_LIST:
switch ( GET_WM_COMMAND_CMD( wParam, lParam ) ){
case LBN_SELCHANGE:
bSetDatatypeList();
bStatus = TRUE;
break;
}
break;
//
// Handle the data type list change.
//
case IDC_PRINT_DATATYPE_LIST:
switch ( GET_WM_COMMAND_CMD( wParam, lParam ) ){
case LBN_SELCHANGE:
bDataTypeAssociation( FALSE );
bStatus = TRUE;
break;
}
break;
default:
bStatus = FALSE;
break;
}
default:
bStatus = FALSE;
break;
}
return bStatus;
}
/*++
Routine Name:
bEnumPrintProcessors
Routine Description:
Enumerates the installed print processors.
Arguments:
Return Value:
TRUE list box fill successfully.
FALSE if error occurred.
--*/
BOOL
TPrintProcessor::
bEnumPrintProcessors(
IN LPTSTR pszServerName,
IN LPTSTR pszEnvironment,
IN DWORD dwLevel,
OUT PVOID *ppvBuffer,
OUT PDWORD pcReturned
)
{
DWORD dwNeeded;
DWORD dwReturned;
PBYTE pBuf = NULL;
BOOL bStatus = FALSE;
//
// First query spooler for installed print processors.
//
if ( !EnumPrintProcessors( pszServerName, pszEnvironment, dwLevel, NULL, 0, &dwNeeded, &dwReturned) ) {
if ( GetLastError() != ERROR_INSUFFICIENT_BUFFER ||
((pBuf = (PBYTE)AllocMem( dwNeeded )) == NULL) ||
!EnumPrintProcessors( pszServerName,
pszEnvironment,
dwLevel,
pBuf,
dwNeeded,
&dwNeeded,
&dwReturned) ) {
bStatus = FALSE;
} else {
bStatus = TRUE;
}
}
//
// If success copy back the data.
//
if( bStatus ){
*ppvBuffer = pBuf;
*pcReturned = dwReturned;
}
return bStatus;
}
/*++
Routine Name:
bEnumPrintProcessorDatatypes
Routine Description:
Enumerates the print processors datatypes.
Arguments:
Return Value:
TRUE list box fill successfully.
FALSE if error occurred.
--*/
BOOL
TPrintProcessor::
bEnumPrintProcessorDatatypes(
IN LPTSTR pszServerName,
IN LPTSTR pszPrintProcessor,
IN DWORD dwLevel,
OUT PVOID *ppvBuffer,
OUT PDWORD pcReturned
)
{
DWORD dwNeeded;
DWORD dwReturned;
PBYTE pBuf = NULL;
BOOL bStatus = FALSE;
//
// First query spooler for installed print processors.
//
if ( !EnumPrintProcessorDatatypes( pszServerName, pszPrintProcessor, dwLevel, NULL, 0, &dwNeeded, &dwReturned) ) {
if ( GetLastError() != ERROR_INSUFFICIENT_BUFFER ||
((pBuf = (PBYTE)AllocMem( dwNeeded )) == NULL) ||
!EnumPrintProcessorDatatypes( pszServerName,
pszPrintProcessor,
dwLevel,
pBuf,
dwNeeded,
&dwNeeded,
&dwReturned) ) {
bStatus = FALSE;
} else {
bStatus = TRUE;
}
}
//
// If success copy back the data.
//
if( bStatus ){
*ppvBuffer = pBuf;
*pcReturned = dwReturned;
}
return bStatus;
}