able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/printscan/ui/printui/browse.cxx

5230 lines
147 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (C) Microsoft Corporation, 1990 - 1999
All rights reserved
Module Name:
browse.cxx
Abstract:
Handles the browse dialog for printer connections.
Author:
Created by AndrewBe on 1 Dec 1992
Steve Kiraly (SteveKi) 1 May 1998
Environment:
User Mode Win32
Revision History:
1 May 1998 moved from winspool.drv to printui.dll
--*/
#include "precomp.hxx"
#pragma hdrstop
#include "result.hxx"
#include "asyncdlg.hxx"
#include "addprn.hxx"
#include "browse.hxx"
#include "persist.hxx"
#define SECURITY_WIN32
#include <wincred.h>
#include <wincrui.h>
//
// Global variables used only in this file.
//
static HDC hdcBitmap = NULL;
static HBITMAP hbmBitmap = NULL;
static HBITMAP hbmDefault = NULL;
static HANDLE hRes = NULL;
static DWORD SysColorHighlight = 0;
static DWORD SysColorWindow = 0;
static INT iBackground = 0;
static INT iBackgroundSel = 0;
static INT iButtonFace = 0;
static INT iButtonShadow = 0;
static BOOL ColorIndicesInitialised = FALSE;
/*++
ConnectTo objects
In the ConnectTo dialog, we initially call EnumPrinters with Flags == 0.
This returns an array of top-level objects represented by a PrinterInfo
structure, e.g:
LanMan Windows NT
Banyan
etc...
We create a chain of ConnectTo objects, each of which points to a PrinterInfo
structure. Initially, pSubObject is NULL.
The Flags field in the PrinterInfo structure indicates whether we can enumerate
on the object. If so, this is indicated to the user through the display of
an appropriate icon in the list box.
If the user clicks on such an enumerable object, we call
EnumPrinters( PRINTER_ENUM_NAME, pName, ... ), which returns a further buffer
of PrinterInfo structures. These may represent servers on the network,
which may in turn be enumerated on to give the names of printers.
Each time an object is enumerated on, we create a new array of ConnectTo objects
which pSubObject points to:
pPrinterInfo[n] pPrinterInfo[n+1]
+-----------------+ +-----------------+
| FLAG_ENUMERABLE | | FLAG_ENUMERABLE |
| <description> | | <description> |
| "LanMan NT" | .... | "Banyan" |
| "local network" | | "other network" |
+-----------------+ +-----------------+
A A
| |
+--------------+ | +--------------+ |
| pPrinterInfo |--+ | pPrinterInfo |--+
+--------------+ ..... +--------------+
+--| pSubObject | | (NULL) |
| +--------------+ +--------------+
| | sizeof(Inf)*2| | 0 |
| +--------------+ +--------------+
|
| =======================================================================
|
| pPrinterInfo[n+m] pPrinterInfo[n+m+1]
| +-----------------+ +-----------------+
| | FLAG_ENUMERABLE | | FLAG_ENUMERABLE |
| | "LanMan Server" | | "LanMan Server" |
| | "Server A" | .... | "Server B" |
| | "daft comment" | | "other comment" |
| +-----------------+ +-----------------+
| A A
| | |
| +--------------+ | +--------------+ |
+->| pPrinterInfo |--+ | pPrinterInfo |--+
+--------------+ +--------------+
+--| pSubObject | | (NULL) |
| +--------------+ ..... +--------------+
| | sizeof(Inf)*2| | 0 |
| +--------------+ +--------------+
|
| =======================================================================
|
| pPrinterInfo[n+m+k] pPrinterInfo[n+m+k+1]
| +-----------------+ +-----------------+
| | 0 | | 0 |
| | "HP Laserjet" | | "Epson" |
| | "Fave Printer" | .... | "Epson Printer" |
| | "good quality" | | "Epson thingy" |
| +-----------------+ +-----------------+
| A A
| | |
| +--------------+ | +--------------+ |
+->| pPrinterInfo |--+ | pPrinterInfo |--+
+--------------+ +--------------+
| (NULL) | | (NULL) |
+--------------+ ..... +--------------+
| 0 | | 0 |
+--------------+ +--------------+
In the list box, the name of each object is displayed, with icon and indentation
to indicate enumerations possible. The simple example above would look like this:
+----------------------+-+
| - LanMan NT |A|
| * Fave Printer + +
| * Epson Printer | |
| + Banyan | |
| | |
| + +
| |V|
+----------------------+-+
--*/
/* ConnectToPrinterDlg
*
* Initializes bitmaps, fonts and cursors the first time it is invoked,
* then calls the ConnectTo dialog.
*
* Parameters:
*
* hwnd - Owner window handle
*
* Returns:
*
* The handle of the printer connected to,
* NULL if no printer was selected or an error occurred.
*
* Author: andrewbe, August 1992
*/
HANDLE
WINAPI
ConnectToPrinterDlg(
HWND hwnd,
DWORD Flags
)
{
PBROWSE_DLG_DATA pBrowseDlgData = new BROWSE_DLG_DATA;
HANDLE hPrinter = NULL;
if( pBrowseDlgData )
{
pBrowseDlgData->vIncRef();
if( pBrowseDlgData->bValid() )
{
pBrowseDlgData->hwndParent = hwnd;
pBrowseDlgData->Flags = Flags;
//
// Make sure COM is initialized first.
//
COleComInitializer com;
//
// Show up the dialog box now.
//
INT_PTR iResult = DialogBoxParam( ghInst,
MAKEINTRESOURCE(DLG_CONNECTTO),
hwnd,
(DLGPROC)ConnectToDlg,
(LPARAM)pBrowseDlgData );
if( iResult == IDOK )
{
hPrinter = pBrowseDlgData->hPrinter;
}
}
pBrowseDlgData->cDecRef();
}
return hPrinter;
}
UINT CALLBACK PropSheetPageCallBack(
IN HWND hwnd,
IN UINT uMsg,
IN LPPROPSHEETPAGE ppsp
)
/*++
Routine Description:
This function gets called when the property sheet
returned from ConnectToPrinterPropertyPage() gets
created/released
Arguments:
hwnd - reserved; must be NULL
uMsg - PSPCB_CREATE - when the page gets created
(return TRUE to allow page creation)
PSPCB_RELEASE - when the page is being destroyed
(return value is ignored)
ppsp - Address of a PROPSHEETPAGE structure that defines
the page being created or destroyed.
Return Value:
Depends on the message - see above.
--*/
{
UINT uReturn = TRUE;
switch( uMsg )
{
case PSPCB_CREATE:
{
//
// Just allow page creation. Do nothing
// but return TRUE
//
}
break;
case PSPCB_RELEASE:
{
//
// We must unhook the UI from the page here
//
PBROWSE_DLG_DATA pBrowseDlgData = reinterpret_cast<PBROWSE_DLG_DATA>( ppsp->lParam );
pBrowseDlgData->cDecRef( );
}
break;
default:
{
//
// This message is not processed.
//
uReturn = FALSE;
}
break;
}
return uReturn;
}
/*
*
*/
HRESULT
ConnectToPrinterPropertyPage(
OUT HPROPSHEETPAGE *phPsp,
OUT UINT *puPageID,
IN IPageSwitch *pPageSwitchController
)
/*++
Routine Description:
Creates a property page, which is identical to the
ConnectToPrinterDlg dialog box
Arguments:
phpsp - Pointer to handle of property sheet page created
ppageID - The resource identifier of the created page
nNextPageID - Page ID to switch if next is pressed
nPrevPageID - Page ID to switch if prev is pressed
Return Value:
S_OK - if everything is fine
E_FAIL (or other error code) if something goes wrong
--*/
{
//
// Assume success
//
HRESULT hResult = S_OK;
//
// Create the dialog data structure
//
PBROWSE_DLG_DATA pBrowseDlgData = NULL;
pBrowseDlgData = new BROWSE_DLG_DATA;
if( pBrowseDlgData )
{
pBrowseDlgData->bInPropertyPage = TRUE;
pBrowseDlgData->pPageSwitchController = pPageSwitchController;
pBrowseDlgData->vIncRef( );
if( pBrowseDlgData->bValid() )
{
//
// Create a property page and return a handle to it.
//
PROPSHEETPAGE psp = {0};
psp.dwSize = sizeof( psp );
psp.hInstance = ghInst;
psp.pfnDlgProc = reinterpret_cast<DLGPROC>( ConnectToPropertyPage );
psp.dwFlags = PSP_DEFAULT | PSP_USEHEADERTITLE | PSP_USEHEADERSUBTITLE | PSP_USECALLBACK;
psp.pfnCallback = PropSheetPageCallBack;
//
// Set the title and subtitle.
//
TStatusB bStatus;
TString strTitle;
TString strSubTitle;
bStatus DBGCHK = strTitle.bLoadString( ghInst, IDS_WIZ_BROWSE_TITLE );
bStatus DBGCHK = strSubTitle.bLoadString( ghInst, IDS_WIZ_BROWSE_SUBTITLE );
psp.pszHeaderTitle = strTitle;
psp.pszHeaderSubTitle = strSubTitle;
psp.pszTemplate = MAKEINTRESOURCE( DLG_WIZ_BROWSE );
psp.lParam = reinterpret_cast<LPARAM>( pBrowseDlgData );
//
// Create the property page handle
//
*phPsp = CreatePropertySheetPage( &psp );
if( NULL == *phPsp )
{
bStatus DBGCHK = FALSE;
hResult = E_FAIL;
}
else
{
//
// Everything looks fine - so return the page ID
//
*puPageID = DLG_WIZ_BROWSE;
}
}
else
{
//
// An error occured ...
//
hResult = E_FAIL;
}
}
else
{
//
// Memory allocation failure
//
hResult = E_OUTOFMEMORY;
}
//
// Check if something failed to prevent
// leaking the BROWSE_DLG_DATA structure
//
if( FAILED( hResult ) )
{
if( pBrowseDlgData )
{
pBrowseDlgData->cDecRef();
}
}
return hResult;
}
/*
*
*/
LRESULT
WINAPI
ConnectToPropertyPage(
HWND hWnd,
UINT uMsg,
WPARAM wParam,
LPARAM lParam
)
/*++
Routine Description:
The window proc for the ConnectToPrinter property page.
Just cracks the lParam to point to PBROWSE_DLG_DATA
structure and pass the control to the ConnectTo proc.
Arguments:
Standard window procedure parameters -
see WindowProc for more details
Return Value:
--*/
{
BOOL bProcessed = TRUE;
//
// Message processing switch
//
switch( uMsg )
{
case WM_INITDIALOG:
{
//
// Crack the lParam parameter, so we could leverage
// the original ConnectToDlg dialog proc for initialization
//
lParam = ( (LPPROPSHEETPAGE )lParam )->lParam;
//
// Impersonate message is not processed
//
bProcessed = FALSE;
}
break;
case WM_DESTROY:
{
//
// Just send a message to the background thread to terminate
// and unhook from the data.
//
ConnectToDestroy( hWnd );
SET_BROWSE_DLG_DATA( hWnd, static_cast<PBROWSE_DLG_DATA>(NULL) );
}
break;
case WM_NOTIFY:
{
//
// Assume we didn't process the message until
// otherwise happened
//
bProcessed = FALSE;
PBROWSE_DLG_DATA pBrowseDlgData = GET_BROWSE_DLG_DATA( hWnd );
SPLASSERT( pBrowseDlgData );
//
// Check if we are in a property page
//
if( pBrowseDlgData->bInPropertyPage )
{
LPNMHDR pnmh = (LPNMHDR)lParam;
switch( pnmh->code )
{
case PSN_WIZBACK:
{
bProcessed = PropertyPageWizardBack( hWnd );
}
break;
case PSN_WIZNEXT:
{
bProcessed = ConnectToOK( hWnd, TRUE );
}
break;
case PSN_QUERYCANCEL:
{
bProcessed = ConnectToCancel( hWnd );
}
break;
}
}
}
break;
default:
{
//
// Pass the message for standard
// processing to ConnectToDlg( ... )
//
bProcessed = FALSE;
}
break;
}
if( !bProcessed )
{
//
// Transfer the control to the ConnectToDlg( ... ) proc
//
bProcessed = (BOOL)ConnectToDlg( hWnd, uMsg, wParam, lParam );
}
return bProcessed;
}
/*
*
*/
BOOL
PropertyPageWizardBack( HWND hWnd )
/*++
Routine Description:
This function moves the property sheet wizard
by one page back
Arguments:
hWnd - handle to the property page
Return Value:
TRUE - if the message is processed
FALSE - otherwise
--*/
{
PBROWSE_DLG_DATA pBrowseDlgData = GET_BROWSE_DLG_DATA( hWnd );
SPLASSERT( pBrowseDlgData );
BOOL bProcessed = TRUE;
//
// Check if there is page switch controller provided
//
if( pBrowseDlgData->pPageSwitchController )
{
UINT uPrevPage;
HRESULT hr = pBrowseDlgData->pPageSwitchController->GetPrevPageID( &uPrevPage );
//
// if S_OK == hr then just go to the provided
// prev page (no problem)
//
if( S_OK == hr )
{
SetWindowLong( hWnd, DWLP_MSGRESULT, uPrevPage );
}
else
{
//
// Don't switch the page - or per client request
// or in case of an error
//
SetWindowLong( hWnd, DWLP_MSGRESULT, -1 );
}
}
else
{
//
// There is no page switch controller provided -
// just go to the natural prev page ...
//
SetWindowLong( hWnd, DWLP_MSGRESULT, 0 );
}
return bProcessed;
}
/*
*
*/
BOOL SetDevMode( HANDLE hPrinter )
{
PPRINTER_INFO_2 pPrinter = NULL;
DWORD cbPrinter = 0;
LONG cbDevMode;
PDEVMODE pNewDevMode;
BOOL Success = FALSE;
//
// Gather the information.
//
if( VDataRefresh::bGetPrinter( hPrinter,
2,
(PVOID*)&pPrinter,
&cbPrinter ) )
{
cbDevMode = DocumentProperties(NULL,
hPrinter,
pPrinter->pPrinterName,
NULL,
pPrinter->pDevMode,
0);
if (cbDevMode > 0)
{
if (pNewDevMode = (PDEVMODE)AllocSplMem(cbDevMode))
{
if (DocumentProperties(NULL,
hPrinter,
pPrinter->pPrinterName,
pNewDevMode,
pPrinter->pDevMode,
DM_COPY) == IDOK)
{
pPrinter->pDevMode = pNewDevMode;
if( SetPrinter( hPrinter, 2, (LPBYTE)pPrinter, 0 ) )
Success = TRUE;
}
FreeSplMem(pNewDevMode);
pPrinter->pDevMode = NULL;
}
}
FreeMem( pPrinter );
}
else
{
DBGMSG( DBG_WARN, ( "GetPrinter failed: Error %d\n", GetLastError( ) ) );
}
return Success;
}
/////////////////////////////////////////////////////////////////////////////
//
// ConnectToDlg
//
// This is the window procedure manages the ConnectTo dialog which allows
// for the selection and creation of a new printer for use by the system.
//
// TO DO:
// error checking for spooler api calls
// IDOK - creating/saving new Printer settings
// Limit text on editbox input fields ???
// Implement
// case IDD_AP_HELP
//
//
//
/////////////////////////////////////////////////////////////////////////////
LRESULT
WINAPI
ConnectToDlg(
HWND hWnd,
UINT usMsg,
WPARAM wParam,
LPARAM lParam
)
{
PBROWSE_DLG_DATA pBrowseDlgData = NULL;
switch (usMsg)
{
case WM_INITDIALOG:
return ConnectToInitDialog( hWnd, (PBROWSE_DLG_DATA)lParam );
case WM_DRAWITEM:
if( ConnectToDrawItem( hWnd, (LPDRAWITEMSTRUCT)lParam ) )
return TRUE;
break;
case WM_CHARTOITEM:
//
// If the key entered is space, well will not do the search;
// instead, we send a fake double click message to the list
// box to expand/collaps the selected item.
//
if( LOWORD( wParam ) == VK_SPACE )
{
HWND hListbox = NULL;
if( hListbox = GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ))
{
SendMessage( hWnd,
WM_COMMAND,
MAKEWPARAM(GetDlgCtrlID(hListbox), LBN_DBLCLK),
(LPARAM)hListbox );
}
return -1;
}
else
{
return ConnectToCharToItem( hWnd, LOWORD( wParam ) );
}
case WM_VKEYTOITEM:
switch (LOWORD(wParam))
{
case VK_RETURN:
ConnectToSelectLbDblClk( hWnd, (HWND)lParam );
/* fall through ... */
default:
return -1;
}
case WM_DESTROY:
ConnectToDestroy( hWnd );
return 0;
case WM_COMMAND:
switch (LOWORD(wParam))
{
case IDD_BROWSE_SELECT_LB:
switch (HIWORD(wParam))
{
case LBN_SELCHANGE:
ConnectToSelectLbSelChange( hWnd );
break;
case LBN_DBLCLK:
ConnectToSelectLbDblClk( hWnd, (HWND)lParam );
break;
}
break;
case IDOK:
return ConnectToOK( hWnd, FALSE );
case IDCANCEL:
return ConnectToCancel( hWnd );
case IDD_BROWSE_DEFAULTEXPAND:
{
//
// Imediately save the checkbox state in the registry
//
return SetRegShowLogonDomainFlag( (BOOL )SendDlgItemMessage(hWnd,IDD_BROWSE_DEFAULTEXPAND, BM_GETCHECK, 0, 0L) );
}
}
break;
case WM_MOUSEMOVE:
ConnectToMouseMove( hWnd, (LONG)LOWORD( lParam ), (LONG)HIWORD( lParam ) );
break;
case WM_SETCURSOR:
return ConnectToSetCursor( hWnd );
case WM_ENUM_OBJECTS_COMPLETE:
pBrowseDlgData = (PBROWSE_DLG_DATA)lParam;
ConnectToEnumObjectsComplete( hWnd,
(PCONNECTTO_OBJECT)pBrowseDlgData->wParam );
break;
case WM_GET_PRINTER_COMPLETE:
pBrowseDlgData = (PBROWSE_DLG_DATA)lParam;
ConnectToGetPrinterComplete( hWnd,
(LPTSTR)pBrowseDlgData->wParam,
(PPRINTER_INFO_2)pBrowseDlgData->lParam,
NO_ERROR );
break;
case WM_GET_PRINTER_ERROR:
pBrowseDlgData = (PBROWSE_DLG_DATA)lParam;
ConnectToGetPrinterComplete( hWnd,
(LPTSTR)pBrowseDlgData->wParam,
NULL,
(DWORD)pBrowseDlgData->lParam );
break;
case WM_HELP:
case WM_CONTEXTMENU:
PrintUIHelp( usMsg, hWnd, wParam, lParam );
break;
}
return FALSE;
}
/*
*
*/
BOOL ConnectToInitDialog( HWND hWnd, PBROWSE_DLG_DATA pBrowseDlgData )
{
//
// Start the initial browse request.
//
DBGMSG( DBG_TRACE, ( "Sending initial browse request\n" ) );
if( !pBrowseDlgData->bInitializeBrowseThread( hWnd ) )
{
iMessage( hWnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTSTARTBROWSETHREAD,
MB_OK | MB_ICONSTOP,
kMsgNone,
NULL );
if( pBrowseDlgData->bInPropertyPage )
{
PropSheet_PressButton( GetParent( hWnd ), PSBTN_CANCEL );
}
else
{
EndDialog( hWnd, IDCANCEL );
}
return FALSE;
}
//
// Set up the initial UI.
//
SET_BROWSE_DLG_DATA( hWnd, pBrowseDlgData );
SendDlgItemMessage(hWnd, IDD_BROWSE_PRINTER, EM_LIMITTEXT, kPrinterBufMax-1, 0L );
if( pBrowseDlgData->Status & BROWSE_STATUS_INITIAL )
{
SETLISTCOUNT(hWnd, 1);
DISABLE_LIST(hWnd);
if( !pBrowseDlgData->bInPropertyPage )
{
//
// When we are in a property page this checkbox
// will not exist at all.
//
SendDlgItemMessage( hWnd, IDD_BROWSE_DEFAULTEXPAND, BM_SETCHECK, 1, 0L );
}
}
/* Set focus initially to the Printer entry field;
* when enumeration is complete (if we're enumerating)
* we'll set it to the list:
*/
SetFocus( GetDlgItem( hWnd, IDD_BROWSE_PRINTER ) );
//
// Enable autocomplete for the printer share/name edit box
//
ShellServices::InitPrintersAutoComplete(GetDlgItem(hWnd, IDD_BROWSE_PRINTER));
return FALSE; /* FALSE == don't set default keyboard focus */
}
/*
*
*/
BOOL ConnectToDrawItem( HWND hwnd, LPDRAWITEMSTRUCT pdis )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pConnectToData;
PCONNECTTO_OBJECT pConnectToObject;
TCHAR Working[255]; /* String to display when we're expanding initially */
DWORD ObjectsFound = 0;
DWORD Depth = 0;
RECT LineRect;
BOOL Selected;
int xIcon; // Coordinates of icon
int yIcon; // in the resource bitmap
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hwnd) ) )
return FALSE;
pConnectToData = GET_CONNECTTO_DATA(hwnd);
if( !pConnectToData || ( pdis->itemID == (UINT)-1 ) )
return FALSE;
/* If this is the first item when we're expanding,
* put "Working..." in the list box:
*/
if( ( pBrowseDlgData->Status & BROWSE_STATUS_INITIAL ) && pdis->itemID == 0 )
{
LoadString( ghInst, IDS_WORKING, Working,
COUNTOF(Working));
pdis->rcItem.left += 3;
DrawLine( pdis->hDC, &pdis->rcItem, Working, FALSE );
return TRUE;
}
LineRect = pdis->rcItem;
Selected = ( pdis->itemState & ODS_SELECTED );
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
pdis->itemID,
NULL,
&ObjectsFound,
&Depth );
if( pConnectToObject )
{
DWORD Flags;
//
// If the object is not a container and it is a provider do not
// display this provider, because there is nothing underneath it
// we should not show it.
//
if( pConnectToObject->pPrinterInfo->Flags & PRINTER_ENUM_HIDE )
{
return FALSE;
}
if (Selected) {
SetBkColor(pdis->hDC, GetSysColor(COLOR_HIGHLIGHT) );
SetTextColor(pdis->hDC, GetSysColor(COLOR_HIGHLIGHTTEXT) );
} else {
SetBkColor(pdis->hDC, GetSysColor(COLOR_WINDOW) );
SetTextColor(pdis->hDC, GetSysColor(COLOR_WINDOWTEXT) );
}
/* Draw the indentation:
*/
LineRect.right = ( LineRect.left + ( Depth * STATUS_BITMAP_SPACE / 4 ) );
DrawLine( pdis->hDC, &LineRect, TEXT(""), Selected );
LineRect.left = LineRect.right;
/* We need to handle 8 different types of icon here:
*/
Flags = pConnectToObject->pPrinterInfo->Flags;
/* Find out the x-coordinate of the icon we need
* to display in the listbox:
*/
switch( Flags & PRINTER_ENUM_ICONMASK )
{
case PRINTER_ENUM_ICON1:
xIcon = ( STATUS_BITMAP_WIDTH * 0 );
break;
case PRINTER_ENUM_ICON2:
xIcon = ( STATUS_BITMAP_WIDTH * 1 );
break;
case PRINTER_ENUM_ICON3:
xIcon = ( STATUS_BITMAP_WIDTH * 2 );
break;
case PRINTER_ENUM_ICON4:
xIcon = ( STATUS_BITMAP_WIDTH * 3 );
break;
case PRINTER_ENUM_ICON5:
xIcon = ( STATUS_BITMAP_WIDTH * 4 );
break;
case PRINTER_ENUM_ICON6:
xIcon = ( STATUS_BITMAP_WIDTH * 5 );
break;
case PRINTER_ENUM_ICON7:
xIcon = ( STATUS_BITMAP_WIDTH * 6 );
break;
case PRINTER_ENUM_ICON8:
default:
xIcon = ( STATUS_BITMAP_WIDTH * 7 );
break;
}
/* If there are enumerated subobjects, pick the appropriate icon:
*/
if( pConnectToObject->pSubObject )
yIcon = BM_IND_CONNECTTO_DOMEXPAND;
else
yIcon = BM_IND_CONNECTTO_DOMPLUS;
/* Ensure that the highlight will extend right across:
*/
LineRect.right = pdis->rcItem.right;
DisplayStatusIcon( pdis->hDC, &LineRect, xIcon, yIcon, Selected );
if( pConnectToObject->pPrinterInfo->Flags & PRINTER_ENUM_CONTAINER )
{
/* Draw the description as is for containers:
*/
DrawLine( pdis->hDC, &LineRect,
pConnectToObject->pPrinterInfo->pDescription,
Selected );
}
else
{
/* ... but insert tabs for the printers:
*/
DrawLineWithTabs( pdis->hDC, &LineRect,
pConnectToObject->pPrinterInfo->pDescription,
Selected );
}
}
if( Selected && ( pdis->itemState & ODS_FOCUS ) )
DrawFocusRect( pdis->hDC, &pdis->rcItem );
return TRUE;
}
/* Need to define LBS_WANTKEYBOARDINPUT for this to work
*
*/
LONG ConnectToCharToItem( HWND hWnd, WORD Key )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pConnectToData;
PCONNECTTO_OBJECT pConnectToObject;
LONG_PTR CurSel;
LONG_PTR i;
LONG_PTR ListCount;
DWORD ObjectsFound;
DWORD Depth;
BOOL Found = FALSE;
TCHAR Char[2];
CurSel = SendDlgItemMessage(hWnd, IDD_BROWSE_SELECT_LB, LB_GETCURSEL, 0, 0L );
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return FALSE;
ENTER_CRITICAL( pBrowseDlgData );
pConnectToData = GET_CONNECTTO_DATA(hWnd);
if( pConnectToData )
{
/* Ensure character is upper case:
*/
Char[0] = (TCHAR)Key;
Char[1] = (TCHAR)0;
CharUpper( Char );
ListCount = SendDlgItemMessage( hWnd, IDD_BROWSE_SELECT_LB, LB_GETCOUNT, 0, 0 );
i = ( CurSel + 1 );
while( !Found && ( i < ListCount ) )
{
ObjectsFound = 0;
Depth = 0;
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
(DWORD)i,
NULL,
&ObjectsFound,
&Depth );
if( pConnectToObject
&&( *pConnectToObject->pPrinterInfo->pDescription == *Char ) )
Found = TRUE;
else
i++;
}
if( !Found )
i = 0;
while( !Found && ( i < CurSel ) )
{
ObjectsFound = 0;
Depth = 0;
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
(DWORD)i,
NULL,
&ObjectsFound,
&Depth );
if( pConnectToObject
&&( *pConnectToObject->pPrinterInfo->pDescription == *Char ) )
Found = TRUE;
else
i++;
}
}
LEAVE_CRITICAL( pBrowseDlgData );
if( Found )
return (LONG)i;
else
return -1;
}
/*
*
*/
VOID ConnectToMouseMove( HWND hWnd, LONG x, LONG y )
{
PBROWSE_DLG_DATA pBrowseDlgData;
POINT pt;
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return;
if( GetCursor() != pBrowseDlgData->hcursorArrow && GetCursor() != pBrowseDlgData->hcursorWait )
{
return;
}
if( pBrowseDlgData->Status & BROWSE_STATUS_EXPAND )
{
pt.x = x;
pt.y = y;
if( ChildWindowFromPoint( hWnd, pt ) == GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) )
{
SetCursor( pBrowseDlgData->hcursorWait );
}
else
SetCursor( pBrowseDlgData->hcursorArrow );
}
else
SetCursor( pBrowseDlgData->hcursorArrow );
}
/* Return TRUE if we want control of the cursor.
* This will be the case if we're over the browse list and
* currently expanding the list.
*/
BOOL ConnectToSetCursor( HWND hWnd )
{
PBROWSE_DLG_DATA pBrowseDlgData;
POINT pt;
BOOL rc = FALSE;
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return rc;
if( pBrowseDlgData->Status & BROWSE_STATUS_EXPAND )
{
if( !GetCursorPos( &pt ) )
{
DBGMSG( DBG_WARN, ( "GetCursorPos failed in ConnectToSetCursor: Error %d\n", GetLastError( ) ) );
}
ScreenToClient( hWnd, &pt );
if( ChildWindowFromPoint( hWnd, pt ) == GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) )
rc = TRUE;
}
return rc;
}
/*
*
*/
VOID SetCursorShape( HWND hWnd )
{
POINT CursorPos;
if( !GetCursorPos( &CursorPos ) )
{
DBGMSG( DBG_WARN, ( "GetCursorPos failed in SetCursorShape: Error %d\n", GetLastError( ) ) );
}
ScreenToClient( hWnd, &CursorPos );
ConnectToMouseMove( hWnd, CursorPos.x, CursorPos.y );
}
/*
*
*/
VOID ConnectToEnumObjectsComplete(
HWND hWnd,
PCONNECTTO_OBJECT pConnectToObject )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pDefaultExpand;
DWORD Index;
TCHAR PrinterName[10];
DWORD ObjectsAdded;
DWORD dwExtent;
INT iLevel;
PCONNECTTO_OBJECT pConnectToData;
DWORD Depth = 0;
DWORD DepthExtent = 0;
DWORD ObjectsFound = 0;
HDC hDC;
LPTSTR pszLine;
LPTSTR pszPrevLine;
SIZE size;
DWORD dwCurExtent;
PCONNECTTO_OBJECT pConnectToObjectChild;
DBGMSG( DBG_TRACE, ( "EnumObjectsComplete\n" ) );
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return;
ObjectsAdded = pConnectToObject->cSubObjects;
//
// Before entering critical section, calculated extents
//
hDC = GetDC(NULL);
if (hDC)
{
pConnectToData = GET_CONNECTTO_DATA(hWnd);
if (pConnectToData)
{
dwExtent = pBrowseDlgData->dwExtent;
GetConnectToObject(pConnectToData->pSubObject,
pConnectToData->cSubObjects,
0,
pConnectToObject,
&ObjectsFound,
&Depth);
DepthExtent = (Depth + 2) * STATUS_BITMAP_SPACE / 4 +
STATUS_BITMAP_SPACE;
for (Index = 0, pConnectToObjectChild = pConnectToObject->pSubObject;
Index < ObjectsAdded;
Index++, pConnectToObjectChild++)
{
pszLine = pConnectToObjectChild->pPrinterInfo->pDescription;
for (iLevel = 0; pszLine;) {
pszPrevLine = pszLine;
pszLine = _tcschr(pszLine, TEXT(','));
if (pszLine) {
iLevel++;
pszLine++;
}
}
if (GetTextExtentPoint32(hDC,
pszPrevLine,
_tcslen(pszPrevLine),
&size))
{
dwCurExtent = size.cx +
iLevel * (COLUMN_WIDTH + COLUMN_SEPARATOR_WIDTH) +
DepthExtent;
dwExtent = dwExtent > dwCurExtent ? dwExtent : dwCurExtent;
}
}
if (pBrowseDlgData->dwExtent != dwExtent)
{
SendDlgItemMessage(hWnd,
IDD_BROWSE_SELECT_LB,
LB_SETHORIZONTALEXTENT,
dwExtent,
0L);
pBrowseDlgData->dwExtent = dwExtent;
}
}
ReleaseDC(NULL, hDC);
}
ENTER_CRITICAL( pBrowseDlgData );
if( pBrowseDlgData->Status & BROWSE_STATUS_INITIAL )
{
pBrowseDlgData->cExpandObjects += ObjectsAdded;
pDefaultExpand = GetDefaultExpand( pConnectToObject->pSubObject,
pConnectToObject->cSubObjects,
&Index );
if( pDefaultExpand )
{
DBGMSG( DBG_TRACE, ( "Expanding next level @08%x\n", pDefaultExpand ) );
pBrowseDlgData->ExpandSelection += ( Index + 1 );
SEND_BROWSE_THREAD_REQUEST( pBrowseDlgData,
BROWSE_THREAD_ENUM_OBJECTS,
pDefaultExpand->pPrinterInfo->pName,
pDefaultExpand );
}
else
{
DBGMSG( DBG_TRACE, ( "No more levels to expand: Count = %d; Selection = %d\n",
pBrowseDlgData->cExpandObjects,
pBrowseDlgData->ExpandSelection ) );
/* Put the selection on the name of the last enumerated node,
* not the first printer under that node:
*/
pBrowseDlgData->ExpandSelection--;
SendDlgItemMessage( hWnd, IDD_BROWSE_SELECT_LB, WM_SETREDRAW, 0, 0L );
SETLISTCOUNT( hWnd, pBrowseDlgData->cExpandObjects );
SETLISTSEL( hWnd, pBrowseDlgData->ExpandSelection );
SendDlgItemMessage( hWnd, IDD_BROWSE_SELECT_LB, LB_SETTOPINDEX,
pBrowseDlgData->ExpandSelection, 0 );
SendDlgItemMessage( hWnd, IDD_BROWSE_SELECT_LB, WM_SETREDRAW, 1, 0L );
ENABLE_LIST( hWnd );
SetCursorShape( hWnd );
/* If the user hasn't typed into the printer name field,
* set the focus to the list:
*/
if( !GetDlgItemText( hWnd, IDD_BROWSE_PRINTER,
PrinterName, COUNTOF(PrinterName) ) )
{
//
// Check if the window is visible or not.
// This is a workaround for the case when we are in a property page
// where the property page may not be visible at
// this moment and we didn't want the current control
// to lose the focus because the background thread has finished
// its job.
//
if( IsWindowVisible(hWnd) )
{
SetFocus( GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) );
}
}
pBrowseDlgData->Status &= ~BROWSE_STATUS_INITIAL;
pBrowseDlgData->Status &= ~BROWSE_STATUS_EXPAND;
}
}
else
{
UpdateList( hWnd, (INT)pConnectToObject->cSubObjects );
if( GETLISTSEL( hWnd ) == LB_ERR )
SETLISTSEL( hWnd, 0 );
ENABLE_LIST( hWnd );
pBrowseDlgData->Status &= ~BROWSE_STATUS_EXPAND;
SetCursor( pBrowseDlgData->hcursorArrow );
//
// If no one has focus, set it to the list box.
// (Common case: double click on machine, listbox
// is disabled, updated, enabled)
//
if ( !GetFocus() )
SetFocus( GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) );
}
LEAVE_CRITICAL( pBrowseDlgData );
}
VOID ConnectToGetPrinterComplete(
HWND hWnd,
LPTSTR pPrinterName,
PPRINTER_INFO_2 pPrinter,
DWORD Error )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pConnectToData;
PCONNECTTO_OBJECT pConnectToObject;
LONG_PTR i;
DWORD ObjectsFound = 0;
DWORD Depth = 0;
DBGMSG( DBG_TRACE, ( "GetPrinterComplete\n" ) );
i = GETLISTSEL(hWnd);
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return;
pConnectToData = GET_CONNECTTO_DATA(hWnd);
ENTER_CRITICAL( pBrowseDlgData );
if( pConnectToData )
{
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
(DWORD)i,
NULL,
&ObjectsFound,
&Depth );
if( !pConnectToObject || !pPrinterName ||
!pConnectToObject->pPrinterInfo->pName ||
_tcscmp( pConnectToObject->pPrinterInfo->pName, pPrinterName ) ) {
pPrinter = NULL;
}
}
UpdateError( hWnd, Error );
if( Error == NO_ERROR )
SetInfoFields( hWnd, pPrinter );
LEAVE_CRITICAL( pBrowseDlgData );
}
VOID ConnectToDestroy( HWND hWnd )
{
PBROWSE_DLG_DATA pBrowseDlgData;
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return;
DBGMSG( DBG_TRACE, ( "Terminating browse thread\n" ) );
ENTER_CRITICAL( pBrowseDlgData );
DBGMSG( DBG_TRACE, ( "Entered critical section\n" ) );
SEND_BROWSE_THREAD_REQUEST( pBrowseDlgData,
BROWSE_THREAD_TERMINATE,
NULL, NULL );
DBGMSG( DBG_TRACE, ( "Sent BROWSE_THREAD_TERMINATE\n" ) );
LEAVE_CRITICAL( pBrowseDlgData );
DBGMSG( DBG_TRACE, ( "Left critical section\n" ) );
FreeBitmaps( );
}
/*
*
*/
VOID ConnectToSelectLbSelChange( HWND hWnd )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pConnectToData;
PCONNECTTO_OBJECT pConnectToObject;
LONG_PTR i;
DWORD ObjectsFound = 0;
DWORD Depth = 0;
i = GETLISTSEL(hWnd);
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return;
pConnectToData = GET_CONNECTTO_DATA(hWnd);
ENTER_CRITICAL( pBrowseDlgData );
SetInfoFields( hWnd, NULL );
if( pConnectToData )
{
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
(DWORD)i,
NULL,
&ObjectsFound,
&Depth );
if( pConnectToObject )
{
DBGMSG( DBG_TRACE, ( "Selection: " TSTR "\n", pConnectToObject->pPrinterInfo->pName ) );
if( !( pConnectToObject->pPrinterInfo->Flags & PRINTER_ENUM_CONTAINER ) )
{
SetDlgItemText(hWnd, IDD_BROWSE_PRINTER,
pConnectToObject->pPrinterInfo->pName);
SEND_BROWSE_THREAD_REQUEST( pBrowseDlgData,
BROWSE_THREAD_GET_PRINTER,
pConnectToObject->pPrinterInfo->pName,
pConnectToObject );
}
else
{
SetDlgItemText(hWnd, IDD_BROWSE_PRINTER, TEXT(""));
}
}
}
LEAVE_CRITICAL( pBrowseDlgData );
}
/*
*
*/
VOID ConnectToSelectLbDblClk( HWND hwnd, HWND hwndListbox )
{
PBROWSE_DLG_DATA pBrowseDlgData;
PCONNECTTO_OBJECT pConnectToData;
PCONNECTTO_OBJECT pConnectToObject;
LONG_PTR CurSel;
DWORD ObjectsFound = 0;
DWORD Depth = 0;
CurSel = SendMessage(hwndListbox, LB_GETCURSEL, 0, 0L );
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA( hwnd ) ) )
return;
ENTER_CRITICAL( pBrowseDlgData );
pConnectToData = GET_CONNECTTO_DATA(hwnd);
if( pConnectToData )
{
pConnectToObject = GetConnectToObject( pConnectToData->pSubObject,
pConnectToData->cSubObjects,
(DWORD)CurSel,
NULL,
&ObjectsFound,
&Depth );
if( pConnectToObject )
{
/* If this object is a container, and has not yet been enumerated,
* call EnumPrinters on this node. If the node has already been
* expanded, close the subtree:
*/
if( pConnectToObject->pPrinterInfo->Flags & PRINTER_ENUM_CONTAINER )
ToggleExpandConnectToObject( hwnd, pConnectToObject );
else
{
//
// Check if we are in a property page force the
// wizard to advance to the next page - not only
// to create the printer connection
//
if( pBrowseDlgData->bInPropertyPage )
{
PropSheet_PressButton( GetParent( hwnd ), PSBTN_NEXT );
}
else
{
ConnectToOK( hwnd, TRUE );
}
}
}
}
LEAVE_CRITICAL( pBrowseDlgData );
}
BOOL
ConnectToOK(
HWND hWnd,
BOOL ForceClose
)
/*++
Routine Description:
We have a remote printer name, try and form the connection. We
may need to create a local printer (the masq case) if the print
providor doesn't support AddPrinterConnection.
Arguments:
Return Value:
--*/
{
PBROWSE_DLG_DATA pBrowseDlgData;
TCHAR szPrinter[kPrinterBufMax];
LPPRINTER_INFO_1 pPrinter=NULL;
LPTSTR pListName=NULL; // The name selected in the list
LPTSTR pConnectToName=NULL; // The name we try to connect to
DWORD ObjectsFound = 0;
DWORD Depth = 0;
BOOL bAdded;
BOOL bStatus = FALSE;
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hWnd) ) )
return FALSE;
SetCursor( pBrowseDlgData->hcursorWait );
//
// Fake a double-click if the focus is on the list box:
//
if( !ForceClose &&
( GetFocus( ) == GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) ) ) {
SendMessage( hWnd, WM_COMMAND,
MAKEWPARAM( IDD_BROWSE_SELECT_LB, LBN_DBLCLK ),
(LPARAM)GetDlgItem( hWnd, IDD_BROWSE_SELECT_LB ) );
return 0;
}
//
// Get the name from the edit box:
//
if( !GetDlgItemText(hWnd, IDD_BROWSE_PRINTER, szPrinter, COUNTOF(szPrinter)) )
{
//
// Check if we are in a property page -
// to prevent closing
//
BOOL bResult = TRUE;
if( pBrowseDlgData->bInPropertyPage )
{
//
// Display a messge to the user and prevent
// advancing to the next page
//
iMessage( hWnd,
IDS_ERR_ADD_PRINTER_TITLE,
IDS_ERR_MISSING_PRINTER_NAME,
MB_OK|MB_ICONHAND,
kMsgNone,
NULL );
SetWindowLong( hWnd, DWLP_MSGRESULT, -1 );
}
else
{
//
// This is the dialog case -
// just close the dialog
//
bResult = ConnectToCancel( hWnd );
}
return bResult;
}
//
// Printer names cannot have trailing white spaces.
//
vStripTrailWhiteSpace( szPrinter );
//
// Add the printer connection, also displaying progress UI.
//
pBrowseDlgData->hPrinter = AddPrinterConnectionUI( hWnd,
szPrinter,
&bAdded );
//
// Check if the printer connection has been sucessfully
// created/added - this covers also the case if the printer
// connection already exist (so we don't add new connection)
//
if( pBrowseDlgData->hPrinter )
bStatus = HandleSuccessfulPrinterConnection( hWnd, pBrowseDlgData );
return bStatus;
}
BOOL
HandleSuccessfulPrinterConnection(
HWND hWnd,
PBROWSE_DLG_DATA pBrowseDlgData
)
/*++
Routine Description:
Handles the situation when successful printer
connection has been established
Arguments:
hWnd - Handle to the dialog/property page
pBrowseDlgData - The already extracted dialog data
Return Value:
TRUE - the message is proccessed fine
FALSE - an error occured
--*/
{
SPLASSERT( pBrowseDlgData );
//
// Assume success
//
BOOL bStatus = TRUE;
//
// Check if we are in a property page ...
//
if( pBrowseDlgData->bInPropertyPage )
{
if( pBrowseDlgData->pPageSwitchController )
{
//
// Extract the printer information
//
TString strPrinterName, strLocation, strComment, strShareName;
bStatus = PrintUIGetPrinterInformation( pBrowseDlgData->hPrinter, &strPrinterName, &strLocation, &strComment, &strShareName );
if( bStatus )
{
//
// Notify the client for the extracted printer information
//
pBrowseDlgData->pPageSwitchController->SetPrinterInfo( strPrinterName, strLocation, strComment, strShareName );
}
}
//
// Prevent the user from leaking the printer handle
// when going to the next page
//
ClosePrinter( pBrowseDlgData->hPrinter );
pBrowseDlgData->hPrinter = NULL;
}
//
// Perform the actual message processing here
//
if( bStatus )
{
//
// Check if we are in a propery page
//
if( pBrowseDlgData->bInPropertyPage )
{
//
// Is there a page switch controller provided
//
if( pBrowseDlgData->pPageSwitchController )
{
UINT uNextPage;
HRESULT hr = pBrowseDlgData->pPageSwitchController->GetNextPageID( &uNextPage );
//
// if S_OK == hr then go to the next page provided
//
if( S_OK == hr )
{
SetWindowLong( hWnd, DWLP_MSGRESULT, uNextPage );
}
else
{
//
// Disable advancing to the next page here
// per client request (S_FALSE) or in case
// of an error
//
SetWindowLong( hWnd, DWLP_MSGRESULT, -1 );
}
}
else
{
//
// There is no page switch controller provided -
// just go to the natural next page in the wizard
//
SetWindowLong( hWnd, DWLP_MSGRESULT, 0 );
}
}
else
{
//
// This is the dialog case ...
// just close the dialog
//
EndDialog( hWnd, IDOK );
}
}
return bStatus;
}
/////////////////////////////////////////////////////////
// CredUI prototypes
typedef
CREDUIAPI
DWORD
(WINAPI *PFN_CredUIPromptForCredentialsW)(
PCREDUI_INFOW pUiInfo,
PCWSTR pszTargetName,
PCtxtHandle pContext,
DWORD dwAuthError,
PWSTR pszUserName,
ULONG ulUserNameMaxChars,
PWSTR pszPassword,
ULONG ulPasswordMaxChars,
BOOL *save,
DWORD dwFlags
);
typedef
CREDUIAPI
void
(WINAPI *PFN_CredUIConfirmCredentialsW)(
PCWSTR pszTargetName,
BOOL bConfirm
);
/////////////////////////////////////////////////////////
// CCredUILoader
class CCredUILoader: public CDllLoader
{
public:
CCredUILoader():
CDllLoader(TEXT("credui.dll")),
m_pfnCredUIPromptForCredentials(NULL),
m_pfnCredUIConfirmCredentials(NULL)
{
if (CDllLoader::operator BOOL())
{
// if the DLL is loaded then do GetProcAddress to the functions we care about
m_pfnCredUIPromptForCredentials =
(PFN_CredUIPromptForCredentialsW )GetProcAddress("CredUIPromptForCredentialsW");
m_pfnCredUIConfirmCredentials =
(PFN_CredUIConfirmCredentialsW)GetProcAddress("CredUIConfirmCredentialsW");
}
}
operator BOOL () const
{
return
((CDllLoader::operator BOOL()) &&
m_pfnCredUIPromptForCredentials &&
m_pfnCredUIConfirmCredentials);
}
PFN_CredUIPromptForCredentialsW m_pfnCredUIPromptForCredentials;
PFN_CredUIConfirmCredentialsW m_pfnCredUIConfirmCredentials;
};
/////////////////////////////////////////////////////////
// OpenPrinter_CredUI
inline static BOOL
IsRPC_SMB(LPTSTR pszPrinter)
{
// should have 2 leading slashes
return (
lstrlen(pszPrinter) > 2 &&
TEXT('\\') == pszPrinter[0] &&
TEXT('\\') == pszPrinter[1]);
}
static BOOL
OpenPrinter_CredUI(HWND hwnd, LPTSTR pszPrinter, LPHANDLE phPrinter, LPPRINTER_DEFAULTS pDefault)
{
ASSERT(phPrinter);
BOOL bRet = FALSE;
// open the printer to see if we have access to it.
bRet = OpenPrinter(pszPrinter, phPrinter, pDefault);
DWORD dwError = GetLastError();
if (IsRPC_SMB(pszPrinter) && (!bRet && ERROR_ACCESS_DENIED == dwError))
{
// OpenPrinter failed because of insufficient permissions.
// in this case we should call credui to obtain credentials.
// then we should call WNetAddConnection2. WNetAddConnection2
// can fail with ERROR_SESSION_CREDENTIAL_CONFLICT if there
// is existing set of credentials which are in conflict with
// ours in which case we should offer the user to overwrite
// the credentials or it can fail with any aother error in the
// case where pszPrinter is not a share name. in this case we
// should try \\server\ipc$ to auth the RPC chanell over named
// pipes and apply the same rules. if WNetAddConnection2 succeeds
// we make another OpenPrinter attempt and if the credentials
// are still not sufficient we inform the user and ask if he
// wants to enter new credentials.
TCHAR szBuffer[PRINTER_MAX_PATH];
LPCTSTR pszServerName = NULL, pszPrinterName = NULL;
// split the full qualified printer name into its components
HRESULT hr = PrinterSplitFullName(pszPrinter, szBuffer, ARRAYSIZE(szBuffer),
&pszServerName, &pszPrinterName);
if (NULL == pszServerName || 0 == pszServerName[0])
{
// this is the case where the user passed a server name (like: '\\servername')
// in this case PrinterSplitFullName succeeds and returns the server name in
// pszPrinterName.
pszServerName = pszPrinterName;
}
if (SUCCEEDED(hr))
{
// load credui.dll
CCredUILoader credUI;
if (credUI)
{
// in case we need to connect through \\server\ipc$
BOOL bAskForCredentials = TRUE;
BOOL bTriedDownlevel = FALSE;
BOOL bSavePassword = TRUE;
BOOL bMustConfirmCredentials = FALSE;
TString strShareIPC;
if (strShareIPC.bFormat(TEXT("%s\\ipc$"), pszServerName))
{
// the server name without slashes
LPCTSTR pszServer = pszServerName + 2;
// credui
DWORD dwCreduiFlags = CREDUI_FLAGS_EXPECT_CONFIRMATION |
CREDUI_FLAGS_SHOW_SAVE_CHECK_BOX |
CREDUI_FLAGS_SERVER_CREDENTIAL;
CREDUI_INFO credUIInfo = { sizeof(CREDUI_INFO), hwnd, NULL, NULL, NULL };
WCHAR szUserName[CREDUI_MAX_USERNAME_LENGTH + 1];
WCHAR szPassword[CREDUI_MAX_PASSWORD_LENGTH + 1];
szUserName[0] = 0;
szPassword[0] = 0;
// NETRESOURCE passed to WNetAddConnection2
NETRESOURCE nr;
ZeroMemory(&nr, sizeof(nr));
nr.dwScope = RESOURCE_GLOBALNET;
nr.dwType = RESOURCETYPE_ANY;
nr.lpRemoteName = const_cast<LPTSTR>(static_cast<LPCTSTR>(strShareIPC));
for (;;)
{
// try to obtain credentials from cred UI. since
// we pass in buffers for the username and password
// credui will *NOT* persist the credentials until
// we confirm they are fine by calling
// CredUIConfirmCredentials
if (bAskForCredentials)
{
if (bMustConfirmCredentials)
{
// if bMustConfirmCredentials is TRUE and we're here this
// means the credentials are fake and we need to call
// CredUIConfirmCredentials(FALSE) to prevent leaking memory.
credUI.m_pfnCredUIConfirmCredentials(pszServer, FALSE);
bMustConfirmCredentials = FALSE;
}
// ask the user for credentails....
dwError = credUI.m_pfnCredUIPromptForCredentials(
&credUIInfo, // PCREDUI_INFOW pUiInfo,
pszServer, // PCWSTR pszTargetName,
NULL, // PCtxtHandle pContext,
dwError, // DWORD dwAuthError,
szUserName, // PWSTR pszUserName,
ARRAYSIZE(szUserName), // ULONG ulUserNameMaxChars,
szPassword, // PWSTR pszPassword,
ARRAYSIZE(szPassword), // ULONG ulPasswordMaxChars,
&bSavePassword, // BOOL *save,
dwCreduiFlags // DWORD dwFlags
);
// any further attemts should not ask for credentials unless
// we explicitly say so!
bMustConfirmCredentials = TRUE;
bAskForCredentials = FALSE;
}
else
{
// assume sucecss if we don't have to ask for credentials
dwError = ERROR_SUCCESS;
}
if (dwError == ERROR_SUCCESS)
{
// try the remote name directly assuming it is a printer share
dwError = WNetAddConnection2(&nr, szPassword, szUserName, 0);
// handle the possible cases:
//
// 1. ERROR_SUCCESS -- the success case. in this case we make another
// OpenPrinter attempt and in the case the credentials are not yet
// suffucient we should offer the user to enter new credentials.
//
// 2. ERROR_SESSION_CREDENTIAL_CONFLICT -- there is an existing
// connection with different set of credentials. offer the user
// to overwrite the credentials with the new ones.
// this may break some running applications, which rely on the
// existing connection but we'll leave that up to the user (to decide).
//
// 3. [any other error] -- nr.lpRemoteName is not a valid net resource.
// try using the \\server\ipc$ if not tried already. if tried then
// just fail silently and propagate the error
if (ERROR_SUCCESS == dwError)
{
// WNetAddConnection2 has succeeded -- let's make
// another OpenPrinter attempt
bRet = OpenPrinter(pszPrinter, phPrinter, pDefault);
dwError = GetLastError();
if (!bRet && ERROR_ACCESS_DENIED == dwError)
{
// got access denied again -- inform the user
// that the supplied credentials are not sufficient
// and ask if he wants to try different credentials
if (IDYES == iMessage(hwnd, IDS_CONNECTTOPRINTER,
IDS_CREDUI_QUESTION_INSUFFICIENT_CREDENTIALS,
MB_YESNO|MB_ICONEXCLAMATION, kMsgNone, NULL))
{
// delete the connection (force) and try again
dwError = WNetCancelConnection2(nr.lpRemoteName,
CONNECT_UPDATE_PROFILE, TRUE);
if (ERROR_SUCCESS == dwError)
{
// the connection was deleted successfuly -- try again
// asking the user for *new* credentials...
bAskForCredentials = TRUE;
continue;
}
else
{
// WNetCancelConnection2 failed -- this could be
// because of lack of permissions or something else
// show UI and cancel the whole operation...
iMessage(hwnd, IDS_CONNECTTOPRINTER, IDS_CREDUI_CANNOT_DELETE_CREDENTIALS,
MB_OK|MB_ICONSTOP, dwError, NULL);
}
}
// if we are here cancel the whole operation
dwError = ERROR_CANCELLED;
}
if (bRet && bMustConfirmCredentials)
{
// we successfully opened the printer -- we need to
// confirm the credentials, so they can be saved.
credUI.m_pfnCredUIConfirmCredentials(pszServer, TRUE);
bMustConfirmCredentials = FALSE;
}
}
else if (ERROR_SESSION_CREDENTIAL_CONFLICT == dwError)
{
// there is a credentials conflict -- ask the user
// if he wants to force and overwrite the existing
// credentials -- this may break some running
// applications but the message box will warn about that.
if (IDYES == iMessage(hwnd, IDS_CONNECTTOPRINTER,
IDS_CREDUI_QUESTION_OWERWRITE_CREDENTIALS,
MB_YESNO|MB_ICONEXCLAMATION, kMsgNone, NULL))
{
// delete the connection (force) and try again
dwError = WNetCancelConnection2(nr.lpRemoteName,
CONNECT_UPDATE_PROFILE, TRUE);
if (ERROR_SUCCESS == dwError)
{
// the previous connection was deleted successfuly --
// try again now.
continue;
}
else
{
// WNetCancelConnection2 failed -- this could be
// because of lack of permissions or something else
// show UI and cancel the whole operation...
iMessage(hwnd, IDS_CONNECTTOPRINTER,
IDS_CREDUI_CANNOT_DELETE_CREDENTIALS,
MB_OK |MB_ICONSTOP, dwError, NULL);
}
}
// if we are here cancel the whole operation
dwError = ERROR_CANCELLED;
}
else if (!bTriedDownlevel)
{
// something else failed. in this case we assume that
// teh remote machine is downlevel server (not NT) and
// the passed in name is a print share name
nr.dwType = RESOURCETYPE_PRINT;
nr.lpRemoteName = pszPrinter;
bTriedDownlevel = TRUE;
// let's try again now
continue;
}
}
if (bMustConfirmCredentials)
{
// if bMustConfirmCredentials is TRUE and we're here this
// means the credentials are fake and we need to call
// CredUIConfirmCredentials(FALSE) to prevent leaking memory.
credUI.m_pfnCredUIConfirmCredentials(pszServer, FALSE);
bMustConfirmCredentials = FALSE;
}
// exit the infinite loop here...
break;
}
}
else
{
// the only possible reason bFormat can fail
dwError = ERROR_OUTOFMEMORY;
}
}
else
{
// just preserve the last error
dwError = GetLastError();
}
}
else
{
// convert to Win32 error (if possible)
dwError = SCODE_CODE(GetScode(hr));
}
}
if (bRet)
{
// make sure the last error is correct
dwError = ERROR_SUCCESS;
}
else
{
// just in case OpenPrinter has trashed
// the handle (unlikely, but...)
*phPrinter = NULL;
}
// make sure the last error is set!
SetLastError(dwError);
return bRet;
}
HANDLE
AddPrinterConnectionUI(
HWND hwnd,
LPCTSTR pszPrinterIn,
PBOOL pbAdded
)
/*++
Routine Description:
Add a printer connection with UI. See AddPrinterConnectionUIQuery.
Arguments:
hwd - Parent window.
pszPrinter - Printer to add.
pbAdded - Indicates whether pszPrinter was added. FALSE = printer
already existed in some form.
Return Value:
HANDLE - hPrinter from pszPrinter.
--*/
{
HANDLE hPrinter = NULL;
HANDLE hServer;
TString strPrinter;
PRINTER_DEFAULTS PrinterDefaults = { NULL, NULL, SERVER_ACCESS_ADMINISTER };
DWORD dwPrinterAttributes = 0;
BOOL bNetConnectionAdded = FALSE;
BOOL bUserDenied = FALSE;
BOOL bMasq = FALSE;
BOOL bUnavailableDriver;
DWORD dwError;
LPWSTR pszArchLocal = NULL;
LPWSTR pszArchRemote = NULL;
LPTSTR pszDriverNew = NULL;
LPTSTR pszDriver = NULL;
LPTSTR pszPrinter = (LPTSTR)pszPrinterIn;
LPTSTR pszPrinterName = NULL;
*pbAdded = FALSE;
if( !OpenPrinter_CredUI(hwnd, pszPrinter, &hPrinter, NULL) ){
DBGMSG( DBG_WARN, ( "OpenPrinter( "TSTR" ) failed: Error = %d\n", pszPrinter, GetLastError( ) ) );
if (GetLastError() != ERROR_CANCELLED){
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
}
goto Fail;
}
if( !PrinterExists( hPrinter, &dwPrinterAttributes, &pszDriver, &pszPrinterName )){
DBGMSG( DBG_WARN, ( "Attempt to connect to a non-existent printer.\n" ) );
//
// Check for an http*:// prefix.
//
if( GetLastError () == ERROR_ACCESS_DENIED &&
(!_tcsnicmp( pszPrinterIn, gszHttpPrefix0, _tcslen(gszHttpPrefix0)) ||
!_tcsnicmp( pszPrinterIn, gszHttpPrefix1, _tcslen(gszHttpPrefix1)))) {
// This is an HTTP printer, we need to give user another chance to enter
// a different username and password
if (!ConfigurePort( NULL, hwnd, (LPTSTR) pszPrinterIn)) {
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
} else {
//
// Retry GetPrinter
//
if( !PrinterExists( hPrinter, &dwPrinterAttributes, &pszDriver, &pszPrinterName )){
DBGMSG( DBG_WARN, ( "2nd Attempt to connect to a non-existent printer.\n" ) );
goto Fail;
}
}
}
else {
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
}
}
if( dwPrinterAttributes & PRINTER_ATTRIBUTE_LOCAL ){
//
// This means the printer is a local pseudo-connection
// probably created when the user tried to connect
// on a previous occasion.
//
goto Done;
}
if (AddPrinterConnectionAndPrompt(hwnd, pszPrinter, &bUserDenied, &strPrinter)) {
//
// This could be refused by the user,
//
if (bUserDenied) {
goto Fail;
}
else {
goto Done;
}
}
dwError = GetLastError();
//
// If this is a KM block issue, display the KM block message and
// go to Fail
//
if( dwError == ERROR_KM_DRIVER_BLOCKED ||
dwError == ERROR_PRINTER_DRIVER_BLOCKED )
{
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
}
if (dwError == ERROR_ACCESS_DISABLED_BY_POLICY)
{
DisplayMessageFromOtherResourceDll(hwnd,
IDS_CONNECTTOPRINTER,
L"xpsp1res.dll",
IDS_TEXT_POINTANDPRINT_POLICY_PRINTUI_DLL,
MB_OK | MB_ICONERROR);
goto Fail;
}
bUnavailableDriver = ( dwError == ERROR_UNKNOWN_PRINTER_DRIVER );
//
// We failed to add the printer connection. This may occur if
//
// 1. This is a mini-print provider, or
// 2. The driver is not installed on the client or server.
//
// In both cases, we need to install the driver locally, so check
// if we have admin privleges.
//
// If the driver was already installed, then AddPrinterConnection
// would have succeeded. If it wasn't installed, then we need
// admin privilege to install it, or create the local printer
// in the masq case, so it's ok to check for admin access here.
//
if( !OpenPrinter( NULL, &hServer, &PrinterDefaults )){
if( GetLastError() == ERROR_ACCESS_DENIED ){
iMessage( hwnd,
IDS_CONNECTTOPRINTER,
IDS_INSUFFPRIV_CREATEPRINTER,
MB_OK | MB_ICONINFORMATION,
kMsgNone,
NULL );
} else {
iMessage( hwnd,
IDS_CONNECTTOPRINTER,
IDS_CANNOTOPENPRINTER,
MB_OK | MB_ICONSTOP,
kMsgNone,
NULL );
}
goto Fail;
}
else
{
pszArchLocal = GetArch(hServer);
}
ClosePrinter( hServer );
{
//
// Create this special scope
// for SplSetupData var
//
SPLSETUP_DATA SplSetupData;
if( !SplSetupData.bValid )
{
DBGMSG( DBG_ERROR, ("AddPrinterConnectionUI: can't initialize SplSetupData -- error %d\n", GetLastError() ) );
goto Fail;
}
//
// If we have the driver name, we don't need to prompt for it.
// We may still have true or false connections.
//
if( pszDriver && pszDriver[0] ){
//
// Check if the reason we failed is because the driver
// isn't available on the client or server.
//
if( bUnavailableDriver ){
BOOL bSamePlatform = TRUE;
DWORD dwNeeded = 0;
LPPRINTER_INFO_2 pPrinterInfo2 = NULL;
if( pszArchLocal )
{
if( !GetPrinter( hPrinter, 2, NULL, 0, &dwNeeded ) &&
GetLastError() == ERROR_INSUFFICIENT_BUFFER &&
NULL != ( pPrinterInfo2 = ( LPPRINTER_INFO_2 ) AllocMem( dwNeeded ) ) &&
GetPrinter( hPrinter, 2, ( LPBYTE ) pPrinterInfo2, dwNeeded, &dwNeeded ) )
{
PRINTER_DEFAULTS PrinterDef = { NULL, NULL, SERVER_READ };
if( OpenPrinter( pPrinterInfo2->pServerName, &hServer, &PrinterDef ) )
{
pszArchRemote = GetArch(hServer);
if( pszArchRemote )
{
bSamePlatform = ( lstrcmpi( pszArchRemote, pszArchLocal ) == 0 );
}
}
ClosePrinter( hServer );
}
if( pPrinterInfo2 )
{
FreeMem( pPrinterInfo2 );
}
}
//
// Add the driver.
//
if( !AddKnownDriver( &SplSetupData, hwnd, pszDriver, bSamePlatform )){
//
// Handles Error UI.
//
SplSetupData.ReportErrorMessage( hwnd );
goto Fail;
}
//
// The only problem was that the driver was not installed, then
// install the driver and try the AddPrinterConnection again.
//
if( !PrintUIAddPrinterConnection( pszPrinter, &strPrinter )){
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
}
} else {
//
// We failed, but not becuase the driver is unavailable.
// It's very likely we have a mini-provider, so
// create the masq case: we have a local printer that
// pretends it's network printer. Of course, it's not
// per-user anymore...
//
// Note that our driver may already be installed.
//
//
// The current hPrinter is a handle to the remote printer.
// We want a handle to the local masq printer instead.
//
ClosePrinter( hPrinter );
bMasq = TRUE;
hPrinter = CreateLocalPrinter( pszPrinterName,
pszDriver,
pszPrinter,
bMasq,
NULL );
if( !hPrinter ){
//
// If we failed because we didn't have the driver,
// then let the user select it and we'll install again.
//
dwError = GetLastError();
if( dwError == ERROR_UNKNOWN_PRINTER_DRIVER ){
//
// Add the driver, but don't prompt user if the printer
// to add is an http printer - just install from local resources.
// Also, don't prompt the user if we were asked not not show UI
// in the first place (obviously).
//
BOOL bPromptUser = _tcsnicmp( pszPrinter, gszHttpPrefix0, _tcslen(gszHttpPrefix0) ) &&
_tcsnicmp( pszPrinter, gszHttpPrefix1, _tcslen(gszHttpPrefix1) );
if( !AddDriver( &SplSetupData,
hwnd,
pszDriver,
bPromptUser,
&pszDriverNew ) ){
//
// Handles Error UI.
//
SplSetupData.ReportErrorMessage( hwnd );
goto Fail;
}
hPrinter = CreateLocalPrinter( pszPrinterName,
pszDriverNew,
pszPrinter,
bMasq,
NULL );
}
if( !hPrinter ){
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
}
}
}
} else {
//
// The driver is not known; we need to prompt the user for it.
//
bMasq = TRUE;
FreeSplStr( pszDriver );
pszDriver = NULL;
if( !AddDriver( &SplSetupData,
hwnd,
NULL,
TRUE,
&pszDriverNew )){
//
// Handles Error UI.
//
SplSetupData.ReportErrorMessage( hwnd );
goto Fail;
}
//
// Create the masq case: we have a local printer that
// pretends it's network printer. Of course, it's not
// per-user anymore...
//
// Close the current handle since it refers to the remote printer,
// and we want a handle to the local masq printer.
//
ClosePrinter( hPrinter );
hPrinter = CreateLocalPrinter(pszPrinterName,
pszDriverNew,
pszPrinter,
bMasq,
NULL);
if( !hPrinter ){
ReportFailure( hwnd,
IDS_CONNECTTOPRINTER,
IDS_COULDNOTCONNECTTOPRINTER );
goto Fail;
}
}
if(hPrinter && SplSetupData.bDriverAdded){
//
// Let the class installer know we have just
// added a local printer drivers.
//
SplSetupData.pfnProcessPrinterAdded(SplSetupData.hDevInfo,
SplSetupData.pSetupLocalData,
pszPrinter,
hwnd);
}
}
Done:
if( bMasq ){
SetDevMode( hPrinter );
}
*pbAdded = TRUE;
//
// If the handle is valid and the connected printer name is not empty
// and it is not equal to the original name then reopen the printer handle
// using the actual printer name. This is necessary for callers to have
// a handle to the printer that was opened with the real printer name.
//
if( hPrinter && !strPrinter.bEmpty() && _tcsicmp( strPrinter, pszPrinter ) )
{
HANDLE hNewPrinter = NULL;
if( OpenPrinter( const_cast<LPTSTR>( static_cast<LPCTSTR>( strPrinter ) ), &hNewPrinter, NULL ) )
{
ClosePrinter( hPrinter );
hPrinter = hNewPrinter;
}
}
Fail:
if( pszArchLocal )
{
FreeMem( pszArchLocal );
}
if( pszArchRemote )
{
FreeMem( pszArchRemote );
}
if( !*pbAdded ){
if( hPrinter ){
ClosePrinter( hPrinter );
hPrinter = NULL;
}
}
//
// Free strings.
//
FreeSplStr(pszDriver);
FreeSplStr(pszDriverNew);
FreeSplStr(pszPrinterName);
return hPrinter;
}
/*++
Routine Name:
AddPrinterConnectionAndPrompt
Routine Description:
This checks to see whether the provider supports opening just the server name
(which implies that it is the win32spl provider),
Arguments:
hWnd - The window handle
pszPrinterName - The name of the printer to which we are adding the connection.
pbUserDenied - If TRUE, the user decided not to connect to the given printer.
pstrNewName - The new, possibly shortened, name of the printer.
Return Value:
BOOL - If TRUE, the connection was added or the user refused the connection.
--*/
BOOL
AddPrinterConnectionAndPrompt(
IN HWND hWnd,
IN PCWSTR pszPrinterName,
OUT BOOL *pbUserDenied,
OUT TString *pstrNewName
)
{
TStatusB bRet;
TString strServerName;
TString strServerUNC;
BOOL bOnDomain = FALSE;
HANDLE hServer = NULL;
if (!hWnd || !pszPrinterName || !pbUserDenied || !pstrNewName)
{
SetLastError(ERROR_INVALID_PARAMETER);
bRet DBGCHK = FALSE;
}
else
{
bRet DBGNOCHK = TRUE;
}
//
// Are we on a Domain?
//
if (bRet)
{
bRet DBGCHK = AreWeOnADomain(&bOnDomain);
}
if (bRet && !bOnDomain)
{
//
// Let's check if this is a remote NT server. For this we need the server
// part of the queue name.
//
if (pszPrinterName[0] == L'\\' && pszPrinterName[1] == L'\\')
{
bRet DBGCHK = strServerName.bUpdate(&pszPrinterName[2]);
}
else
{
SetLastError(ERROR_INVALID_PRINTER_NAME);
bRet DBGCHK = FALSE;
}
if (bRet)
{
PWSTR pszSlash = const_cast<PWSTR>(wcschr(strServerName, L'\\'));
if (pszSlash)
{
*pszSlash = L'\0';
}
else
{
SetLastError(ERROR_INVALID_PRINTER_NAME);
bRet DBGCHK = FALSE;
}
}
//
// OK, we have the server name part, lets get a UNC name.
//
if (bRet)
{
bRet DBGCHK = strServerUNC.bFormat(L"\\\\%s", static_cast<PCWSTR>(strServerName));
}
//
// Bit of a hack here, if we can open the server name, then this implies we
// are talking to an NT Server.
//
if (bRet)
{
bRet DBGCHK = OpenPrinter(const_cast<PWSTR>(static_cast<PCWSTR>(strServerUNC)), &hServer, NULL);
}
if (bRet)
{
int Result = DisplayMessageFromOtherResourceDll(hWnd,
IDS_CONNECTTOPRINTER,
L"xpsp1res.dll",
IDS_TEXT_POINTANDPRINT_WARNING_PRINTUI_DLL,
MB_YESNO | MB_ICONWARNING,
static_cast<PCWSTR>(strServerName));
bRet DBGNOCHK = Result == IDYES || Result == IDNO;
*pbUserDenied = Result == IDNO;
}
}
//
// If we are on a domain, then the user did not deny us the connection (
// although, we didn't ask).
//
if (bRet && bOnDomain)
{
*pbUserDenied = FALSE;
}
if (bRet && !*pbUserDenied)
{
bRet DBGCHK = PrintUIAddPrinterConnection(pszPrinterName, pstrNewName);
}
if (hServer)
{
ClosePrinter(hServer);
}
return bRet;
}
HANDLE
AddPrinterConnectionNoUI(
LPCTSTR pszPrinterIn,
PBOOL pbAdded
)
/*++
Routine Description:
Add a printer connection silently, this routine will never show UI to the user.
Since the installation doesn't show UI, we can't make a normal RPC printer
connection to the remote side since this will refresh the printer driver from
the remote server. So we create a local printer with a redirected port rather.
Arguments:
pszPrinter - Printer to add.
pbAdded - Indicates whether pszPrinter was added.
Return Value:
HANDLE - hPrinter from pszPrinter.
--*/
{
HANDLE hPrinter = NULL;
BOOL bAdded = FALSE;
HANDLE hServer = NULL;
PORT_INFO_1 *pPorts = NULL;
PRINTER_DEFAULTS Defaults = { NULL, NULL, SERVER_ACCESS_ADMINISTER };
DWORD cbPorts = 0;
DWORD cPorts = 0;
TStatusB bContinue;
if (!pszPrinterIn || !pbAdded)
{
SetLastError(ERROR_INVALID_PARAMETER);
bContinue DBGCHK = FALSE;
}
//
// First, does the user have local administrator rights on this machine?
//
if (bContinue)
{
bContinue DBGCHK = OpenPrinter(NULL, &hServer, &Defaults);
}
//
// Get all the port names of the local machine. If this port already exists,
// we just try and open the printer.
//
if (bContinue)
{
bContinue DBGCHK = VDataRefresh::bEnumPorts(NULL, 1, reinterpret_cast<VOID **>(&pPorts), &cbPorts, &cPorts);
}
//
// Check to see if a port with the same name that we are already adding
// exists.
//
for(UINT i = 0; bContinue && i < cPorts; i++)
{
//
// The port names match.
//
if (!_tcsicmp(pPorts[i].pName, pszPrinterIn))
{
//
// Open the printer and return this is the handle. If it can't
// open, this is considered a failure, but we don't add the
// printer.
//
(VOID)OpenPrinter(const_cast<PWSTR>(pszPrinterIn), &hPrinter, NULL);
bContinue DBGNOCHK = FALSE;
SetLastError(ERROR_ALREADY_EXISTS);
}
}
//
// The port name does not exist yet. Create the redirected or Masq printer.
//
if (bContinue)
{
hPrinter = CreateRedirectedPrinter(pszPrinterIn);
bAdded = hPrinter != NULL;
}
//
// Cleanup any resources used.
//
FreeMem(pPorts);
if (hServer)
{
ClosePrinter(hServer);
}
if (pbAdded)
{
*pbAdded = bAdded;
}
return hPrinter;
}
/*++
Routine Name:
CreateRedirectedPrinter
Routine Description:
Create either a local printer with a redirected port to Windows NT or a
masq printer to a 9x box.
Arguments:
pszPrinter - Printer to add.
Return Value:
HANDLE - hPrinter from pszPrinter.
--*/
HANDLE
CreateRedirectedPrinter(
IN PCWSTR pszPrinterIn
)
{
HANDLE hNewPrinter = NULL;
HANDLE hPrinter = NULL;
PRINTER_INFO_2 *pPrinterInfo = NULL;
DWORD cbBuffer = 0;
BOOL bWinNT = FALSE;
PWSTR pszPrinterName = NULL;
PWSTR pszMappedDriver = NULL;
BOOL bDriverMapped = FALSE;
BOOL bPortAdded = FALSE;
DWORD dwLastError = ERROR_SUCCESS;
TStatusB bSucceeded;
{
SPLSETUP_DATA SplSetupData;
bSucceeded DBGCHK = SplSetupData.bValid;
if (!bSucceeded)
{
DBGMSG(DBG_ERROR, ("CreateRedirectedPrinter: can't initialize SplSetupData -- error %d\n", GetLastError()));
}
if (bSucceeded)
{
bSucceeded DBGCHK = OpenPrinter(const_cast<PWSTR>(pszPrinterIn), &hPrinter, NULL);
}
//
// Get the printer info for the remote printer. From this we can determine
// the required driver and we can get the devmode for applying to the printer
// later.
//
if (bSucceeded)
{
bSucceeded DBGCHK = VDataRefresh::bGetPrinter(hPrinter,
2,
reinterpret_cast<VOID **>(&pPrinterInfo),
&cbBuffer);
}
//
// We need the driver name, or we can't add the local printer.
//
if (bSucceeded)
{
bSucceeded DBGNOCHK = pPrinterInfo->pDriverName && *pPrinterInfo->pDriverName;
if (!bSucceeded)
{
SetLastError(ERROR_UNKNOWN_PRINTER_DRIVER);
}
}
if (bSucceeded)
{
//
// If we are being asked to make a connection to ourselves, the server
// name we retrieve from the spooler will be NULL. Since unlike
// AddPrinterConnection this will result in a printer in the desktop,
// kill this attempt now.
//
if (pPrinterInfo->pServerName && *pPrinterInfo->pServerName)
{
//
// Determine whether we are talking to a Win9X or an NT server.
//
if (bSucceeded)
{
HRESULT hr = IsNTServer(pPrinterInfo->pServerName);
if (hr == S_OK)
{
bWinNT = TRUE;
}
else if (FAILED(hr))
{
SetLastError(HRESULT_CODE(hr));
bSucceeded DBGCHK = FALSE;
}
}
//
// If we are talking to an NT server, then create a name rather like a
// connection name for the printer and create a local port to represent
// it.
//
if (bSucceeded)
{
if (bWinNT)
{
bSucceeded DBGCHK = BuildNTPrinterName(pPrinterInfo, &pszPrinterName);
}
else
{
BuildMasqPrinterName(pPrinterInfo, &pszPrinterName);
bSucceeded DBGCHK = pszPrinterName != NULL;
}
}
//
// Check to see whether we have a mapping for the printer driver.
//
if (bSucceeded)
{
bSucceeded DBGCHK = SplSetupData.pfnFindMappedDriver(bWinNT,
pPrinterInfo->pDriverName,
&pszMappedDriver,
&bDriverMapped);
}
//
// If we are talking to an NT Server, create the local redirected port.
//
if (bSucceeded && bWinNT)
{
bSucceeded DBGCHK = CreateLocalPort(pszPrinterIn);
bPortAdded = bSucceeded;
}
//
// Try to Create the local printer.
//
if (bSucceeded)
{
hNewPrinter = CreateLocalPrinter(pszPrinterName,
pszMappedDriver,
pszPrinterIn,
!bWinNT,
pPrinterInfo->pDevMode);
bSucceeded DBGCHK = hNewPrinter != NULL;
//
// We couldn't create the local printer because the driver wasn't there.
// Or, somehow a version 2 driver was put on the machine, and the
// corresponding V3 driver isn't there. In this case, try to add the
// driver ourselves.
//
if (!bSucceeded &&
(GetLastError() == ERROR_UNKNOWN_PRINTER_DRIVER ||
GetLastError() == ERROR_KM_DRIVER_BLOCKED))
{
//
// Add the in-box printer driver silently.
//
DWORD Status = SplSetupData.pfnInstallInboxDriverSilently(pszMappedDriver);
if (Status != ERROR_SUCCESS)
{
SetLastError(Status);
bSucceeded DBGCHK = FALSE;
}
else
{
bSucceeded DBGCHK = TRUE;
}
if (bSucceeded)
{
hNewPrinter = CreateLocalPrinter(pszPrinterName,
pszMappedDriver,
pszPrinterIn,
!bWinNT,
pPrinterInfo->pDevMode);
bSucceeded DBGCHK = hNewPrinter != NULL;
}
}
}
}
}
//
// If we added the port, but failed to add the printer, delete the port.
//
if (!bSucceeded && bPortAdded)
{
DeletePort(NULL, NULL, const_cast<PWSTR>(pszPrinterIn));
}
//
// The SplSetupData class overwrites the last error when it is
// destructed. So, save the last error here,
//
dwLastError = GetLastError();
//
// Clean up all of our local resources.
//
if (hPrinter)
{
ClosePrinter(hPrinter);
}
FreeMem(pPrinterInfo);
FreeSplStr(pszPrinterName);
if (pszMappedDriver)
{
(VOID)SplSetupData.pfnFreeMem(pszMappedDriver);
}
}
SetLastError(dwLastError);
return hNewPrinter;
}
/*++
Routine Name:
GetArch
Routine Description:
Return the architecture of the given server.
Arguments:
hServer - The server whose architecture to retrieve,
Return Value:
An allocated string with the architecture.
--*/
LPWSTR
GetArch(
HANDLE hServer
)
{
LPWSTR pszRetArch = NULL;
WCHAR szArch[kStrMax] = {0};
DWORD dwNeeded = 0;
if( ERROR_SUCCESS == GetPrinterData( hServer,
SPLREG_ARCHITECTURE,
NULL,
(PBYTE)szArch,
sizeof( szArch ),
&dwNeeded ) )
{
if( NULL != ( pszRetArch = (LPWSTR)AllocMem(dwNeeded*sizeof(*szArch)) ) )
{
lstrcpyn( pszRetArch, szArch, dwNeeded );
}
}
return pszRetArch;
}
/* BuildMasqPrinterName
*
* generates a proper printer name for masq printers
*/
VOID
BuildMasqPrinterName(
IN PPRINTER_INFO_2 pPrinter,
OUT PWSTR *ppszPrinterName
)
{
ASSERT(pPrinter);
ASSERT(ppszPrinterName);
// if this is an http printer, we need to take the server name (http://server) from pPrinter->pPortName
// if this is downlevel printer then the pPrinter->pPortName is NULL
TCHAR *p = NULL;
if( pPrinter->pPortName )
{
// check if the server is http://server
if( !p && !_tcsnicmp(pPrinter->pPortName, gszHttpPrefix0, _tcslen(gszHttpPrefix0)) )
{
p = pPrinter->pPortName + _tcslen(gszHttpPrefix0);
}
// check if the server is https://server
if( !p && !_tcsnicmp(pPrinter->pPortName, gszHttpPrefix1, _tcslen(gszHttpPrefix1)) )
{
p = pPrinter->pPortName + _tcslen(gszHttpPrefix1);
}
}
if( p )
{
// this is http printer, we need to build the name in the following format
// \\http://server\printer
LPCTSTR pszServer;
LPCTSTR pszPrinter;
TCHAR szScratch[kPrinterBufMax];
// split the full printer name into its components.
vPrinterSplitFullName(szScratch, pPrinter->pPrinterName, &pszServer, &pszPrinter);
// go to the end of the server name & terminate p
for( ; *p && *p != _T('/'); p++ ) { } *p = 0;
// now pPrinter->pPortName now should be "http://server" or "https://server"
pszServer = pPrinter->pPortName;
if( pszServer && *pszServer )
{
// build the masq printer name here (i.e. "\\http://server\printer")
TString strPrinterName;
if( strPrinterName.bFormat(_T("\\\\%s\\%s"), pszServer, pszPrinter) )
{
*ppszPrinterName = AllocSplStr(strPrinterName);
}
}
}
else
{
// the friendly name is the same as pPrinter->pPrinterName
*ppszPrinterName = AllocSplStr(pPrinter->pPrinterName);
}
}
/* PrinterExists
*
* check if the printer exists & return some of its
* attributes/data.
*/
BOOL
PrinterExists(
HANDLE hPrinter,
PDWORD pAttributes,
LPTSTR *ppszDriver,
LPTSTR *ppszPrinterName
)
{
BOOL bReturn = FALSE;
DWORD cb = 0;
CAutoPtrSpl<PRINTER_INFO_2> spPI2;
ASSERT(ppszDriver);
ASSERT(ppszPrinterName);
*ppszDriver = NULL;
*ppszPrinterName = NULL;
bReturn = VDataRefresh::bGetPrinter(hPrinter, 2, spPI2.GetPPV(), &cb);
if( bReturn )
{
// return the driver & attributes
*pAttributes = spPI2->Attributes;
*ppszDriver = AllocSplStr(spPI2->pDriverName);
// generate a friendly name - i.e.
// "printer or server" or "printer on http://server"
BuildMasqPrinterName(spPI2, ppszPrinterName);
}
else
{
// this is a bit of a hack:
//
// OpenPrinter returns a valid handle if the name of a server is passed in.
// we need to call GetPrinter with that handle to check that it's a printer.
// if this call fails, the error will have been set to ERROR_INVALID_HANDLE,
// whereas it really should be ERROR_INVALID_PRINTER_NAME.
if( ERROR_INVALID_HANDLE == GetLastError() )
{
SetLastError( ERROR_INVALID_PRINTER_NAME );
}
}
return bReturn;
}
/*++
Routine Name:
BuildNTPrinterName
Routine Description:
This builds the name of the NT printer, if the name is \\Foo\Bar, we will
call it Auto Foo on Bar. This will be the real printer name to ensure that the
UI does not confuse it with a printer connection which causes problems in
the shell folder.
Arguments:
pPrinter - The printer info for the printer
ppszPrinterName - The returned
Return Value:
An allocated string with the architecture.
--*/
BOOL
BuildNTPrinterName(
IN PRINTER_INFO_2 *pPrinter,
OUT PWSTR *ppszPrinterName
)
{
PCWSTR pszPrinterName = NULL;
PCWSTR pszServerName = NULL;
WCHAR szScratch[kPrinterBufMax];
TString strNtConnectName;
TStatusB bRet;
if (!pPrinter || !pPrinter->pPrinterName || !ppszPrinterName)
{
SetLastError(ERROR_INVALID_PARAMETER);
bRet DBGCHK = FALSE;
}
else
{
bRet DBGCHK = TRUE;
}
//
// Split the printer name into a server name and printer name. Then we will
// load the resource and format the message string.
//
if (bRet)
{
vPrinterSplitFullName(szScratch, pPrinter->pPrinterName, &pszServerName, &pszPrinterName);
//
// Strip the leading \\ from the server name.
//
if(pszServerName[0] == L'\\' && pszServerName[1] == L'\\')
{
pszServerName = pszServerName + 2;
}
bRet DBGCHK = bConstructMessageString(ghInst, strNtConnectName, IDS_DSPTEMPLATE_NETCRAWLER, pszPrinterName, pszServerName);
}
//
// If this succeeds, allocate a copy a string back. We do this so the caller
// can free the string just as it does with BuildMasqPrinterName.
//
if (bRet)
{
*ppszPrinterName = AllocSplStr(strNtConnectName);
bRet DBGCHK = *ppszPrinterName != NULL;
}
return bRet;
}
/* CreateLocalPrinter
*
* creates a local masq printer with the specified
* name, driver & port.
*/
HANDLE
CreateLocalPrinter(
IN LPCTSTR pPrinterName,
IN LPCTSTR pDriverName,
IN LPCTSTR pPortName,
IN BOOL bMasqPrinter,
IN DEVMODE *pDevMode OPTIONAL
)
{
// make sure the name is unique
TCHAR szPrinterName[kPrinterBufMax];
if( NewFriendlyName(NULL, pPrinterName, szPrinterName) )
{
pPrinterName = szPrinterName;
}
// set the printer info fields to actually create the masq printer.
PRINTER_INFO_2 pi2 = {0};
pi2.pPrinterName = const_cast<PTSTR>(pPrinterName);
pi2.pDriverName = const_cast<PTSTR>(pDriverName);
pi2.pPortName = const_cast<PTSTR>(pPortName);
pi2.pPrintProcessor = (LPTSTR)gszDefaultPrintProcessor;
pi2.pDevMode = pDevMode;
pi2.Attributes = PRINTER_ATTRIBUTE_LOCAL;
if (bMasqPrinter)
{
pi2.Attributes |= PRINTER_ATTRIBUTE_NETWORK;
}
// ask the spooler to create a masq printer for us
// (i.e. attributes should be local + network).
return AddPrinter(NULL, 2, (LPBYTE)&pi2 );
}
/*
*
*/
BOOL ConnectToCancel( HWND hWnd )
{
PBROWSE_DLG_DATA pBrowseDlgData = GET_BROWSE_DLG_DATA( hWnd );
SPLASSERT( pBrowseDlgData );
//
// Check if we are in property page
//
if( pBrowseDlgData->bInPropertyPage )
{
BOOL bEnableClose = TRUE;
//
// Check if there is page switch controller provided
//
if( pBrowseDlgData->pPageSwitchController )
{
if( S_OK == pBrowseDlgData->pPageSwitchController->QueryCancel( ) )
{
//
// We must prevent closing in this case
//
bEnableClose = FALSE;
}
}
if( bEnableClose )
{
//
// Allow closing operation
//
SetWindowLong( hWnd, DWLP_MSGRESULT, FALSE );
}
else
{
//
// Prevent closing operation
//
SetWindowLong( hWnd, DWLP_MSGRESULT, TRUE );
}
}
else
{
//
// We are in dialog box -
// Just close dialog with IDCANCEL
//
EndDialog( hWnd, IDCANCEL );
}
//
// Always processing this message
//
return TRUE;
}
/* GetConnectToObject
*
* Does a recursive search down the ConnectTo object tree to find the Nth
* object, where Index == N.
* On the top-level call, *pObjectsFound must be initialised to zero,
* and this value is incremented each time an object in the tree is encountered.
* On any given level, if *pObjectsFound equals the index being sought,
* then a pointer to the corresponding ConnectTo object is returned.
* If the index hasn't yet been reached, the function is called recursively
* on any subobjects.
*
* Arguments:
*
* pFirstConnectToObject - Pointer to the first ConnectTo object
* in the array of objects at a given level.
*
* cThisLevelObjects - The number of objects in the array at this level.
*
* Index - Which object is requested. E.g. if the top item in the printers
* list box is being drawn, this will be 0.
*
* pObjectsFound - A pointer to the number of objects encountered so far in
* the search. This must be initialised to zero by the top-level caller.
*
* pDepth - A pointer to the depth of the object found in the search.
* This value is zero-based and must be initialised to zero
* by the top-level caller.
*
* Return:
*
* A pointer to the CONNECTTO_OBJECT if found, otherwise NULL.
*
*
* Author: andrewbe July 1992
*
*
*/
PCONNECTTO_OBJECT GetConnectToObject(
IN PCONNECTTO_OBJECT pFirstConnectToObject,
IN DWORD cThisLevelObjects,
IN DWORD Index,
IN PCONNECTTO_OBJECT pFindObject,
OUT PDWORD pObjectsFound,
OUT PDWORD pDepth )
{
PCONNECTTO_OBJECT pConnectToObject = NULL;
DWORD i = 0;
while( !pConnectToObject && ( i < cThisLevelObjects ) )
{
if (&pFirstConnectToObject[i] == pFindObject ||
(!pFindObject && *pObjectsFound == Index))
{
pConnectToObject = &pFirstConnectToObject[i];
}
/* Make a recursive call on any objects which have subobjects:
*/
else if( pFirstConnectToObject[i].pSubObject )
{
(*pObjectsFound)++; // Add the current object to the total count
pConnectToObject = GetConnectToObject(
pFirstConnectToObject[i].pSubObject,
pFirstConnectToObject[i].cSubObjects,
Index,
pFindObject,
pObjectsFound,
pDepth );
if( pConnectToObject )
(*pDepth)++;
}
else
(*pObjectsFound)++; // Add the current object to the total count
i++; // Increment to the next object at this level
}
return pConnectToObject;
}
/* GetDefaultExpand
*
* Searches one level of enumerated objects to find the first one with the
* PRINTER_ENUM_EXPAND flag set.
* This flag should have been set by the spooler to guide us to the user's
* logon domain, so we can show the printers in that domain straight away.
* The user can disable this behaviour by unchecking the box in the ConnectTo
* dialog. If this has been done, this function will return NULL immediately.
*
* Arguments:
*
* pFirstConnectToObject - Pointer to the first ConnectTo object
* in the array of objects at a given level.
*
* cThisLevelObjects - The number of objects in the array at this level.
*
* pIndex - A pointer to a DWORD which will receive the index of the
* object found in the array.
*
* Return:
*
* A pointer to the CONNECTTO_OBJECT if found, otherwise NULL.
*
*
* Author: andrewbe December 1992 (based on GetConnectToObject)
*
*
*/
PCONNECTTO_OBJECT GetDefaultExpand(
IN PCONNECTTO_OBJECT pFirstConnectToObject,
IN DWORD cThisLevelObjects,
OUT PDWORD pIndex )
{
PCONNECTTO_OBJECT pDefaultExpand = NULL;
DWORD i = 0;
while( !pDefaultExpand && ( i < cThisLevelObjects ) )
{
if( pFirstConnectToObject[i].pPrinterInfo->Flags & PRINTER_ENUM_EXPAND )
pDefaultExpand = &pFirstConnectToObject[i];
else
i++; // Increment to the next object at this level
}
*pIndex = i;
return pDefaultExpand;
}
/* FreeConnectToObjects
*
* Frees the array of objects on the current level, after making a recursive
* call on any subobjects of members of the array.
*
* Arguments:
*
* pFirstConnectToObject - Pointer to the first ConnectTo object in the array
* of objects at a given level.
*
* cThisLevelObjects - The number of objects in the array at this level.
*
* cbThisLevelObjects - The size of the the array at this level.
*
* Return:
*
* The number of objects actually removed, regardless of errors.
*
*
* Author: andrewbe July 1992
*/
DWORD FreeConnectToObjects(
IN PCONNECTTO_OBJECT pFirstConnectToObject,
IN DWORD cThisLevelObjects,
IN DWORD cbPrinterInfo )
{
DWORD i;
DWORD SubObjectsFreed = 0;
if( ( cThisLevelObjects > 0 ) && pFirstConnectToObject->pPrinterInfo )
FreeSplMem( pFirstConnectToObject->pPrinterInfo );
for( i = 0; i < cThisLevelObjects; i++ )
{
/* Make a recursive call on any objects which have subobjects:
*/
if( pFirstConnectToObject[i].pSubObject )
{
SubObjectsFreed = FreeConnectToObjects(
pFirstConnectToObject[i].pSubObject,
pFirstConnectToObject[i].cSubObjects,
pFirstConnectToObject[i].cbPrinterInfo );
}
}
if( cThisLevelObjects > 0 )
FreeSplMem( pFirstConnectToObject );
return ( SubObjectsFreed + cThisLevelObjects );
}
/* ToggleExpandConnectToObject
*
* Expands or collapses the node accordingly.
*
* Arguments:
*
* hwndListbox - Handle of the listbox containing the printer info.
*
* pConnectToObject - The node to be expanded or collapsed.
* If it has already been expanded, collapse it, otherwise expand it.
*
* Return:
*
* TRUE if no error occurred.
*
*/
BOOL ToggleExpandConnectToObject(
HWND hwnd,
PCONNECTTO_OBJECT pConnectToObject )
{
PBROWSE_DLG_DATA pBrowseDlgData;
DWORD ObjectsRemoved = 0;
if( !( pBrowseDlgData = GET_BROWSE_DLG_DATA(hwnd) ) )
return FALSE;
ASSERT( pBrowseDlgData->csLock.bInside() );
if( pConnectToObject->pSubObject )
{
ObjectsRemoved = FreeConnectToObjects(
&pConnectToObject->pSubObject[0],
pConnectToObject->cSubObjects,
pConnectToObject->cbPrinterInfo );
pConnectToObject->pSubObject = NULL;
pConnectToObject->cSubObjects = 0;
pConnectToObject->cbPrinterInfo = 0;
UpdateList( hwnd, ( - (INT)ObjectsRemoved ) );
SetCursor( pBrowseDlgData->hcursorArrow );
}
else
{
pBrowseDlgData->Status |= BROWSE_STATUS_EXPAND;
SetCursorShape( hwnd );
DISABLE_LIST(hwnd);
SEND_BROWSE_THREAD_REQUEST( pBrowseDlgData,
BROWSE_THREAD_ENUM_OBJECTS,
pConnectToObject->pPrinterInfo->pName,
pConnectToObject );
}
return TRUE;
}
BOOL UpdateList(
HWND hwnd,
INT Increment )
{
HWND hwndListbox;
LONG_PTR CurSel;
LONG_PTR OldCount;
DWORD ObjectsRemoved = 0;
LONG_PTR NewObjectsOutOfView;
ULONG_PTR TopIndex;
ULONG_PTR BottomIndex;
RECT CurrentSelectionRect;
RECT ListboxRect;
DWORD Error = 0;
ASSERT( GET_BROWSE_DLG_DATA(hwnd)->csLock.bInside() );
hwndListbox = GetDlgItem( hwnd, IDD_BROWSE_SELECT_LB );
CurSel = SendMessage( hwndListbox, LB_GETCURSEL, 0, 0L );
SendMessage( hwndListbox, WM_SETREDRAW, 0, 0L );
TopIndex = SendMessage( hwndListbox, LB_GETTOPINDEX, 0, 0 );
OldCount = SendMessage( hwndListbox, LB_GETCOUNT, 0, 0 );
DBGMSG( DBG_TRACE, ( "Setting list count to %d\n", OldCount + Increment ) );
SendMessage( hwndListbox, LB_SETCOUNT, OldCount + Increment, 0 );
if( Increment > 0 )
{
GetClientRect( hwndListbox, &ListboxRect );
BottomIndex = ( TopIndex +
( ListboxRect.bottom / STATUS_BITMAP_HEIGHT ) - 1 );
NewObjectsOutOfView = ( CurSel + Increment - BottomIndex );
if( NewObjectsOutOfView > 0 )
{
TopIndex = min( CurSel, (LONG_PTR) ( TopIndex + NewObjectsOutOfView ) );
}
}
SendMessage( hwndListbox, LB_SETCURSEL, CurSel, 0L );
SendMessage( hwndListbox, LB_SETTOPINDEX, TopIndex, 0 );
SendMessage( hwndListbox, WM_SETREDRAW, 1, 0L );
SendMessage( hwndListbox, LB_GETITEMRECT, CurSel,
(LPARAM)&CurrentSelectionRect );
InvalidateRect( hwndListbox, NULL, FALSE );
return TRUE;
}
/* GetPrinterStatusString
*
* Loads the resource string corresponding to the supplied status code.
*
* andrewbe wrote it - April 1992
*/
int GetPrinterStatusString( DWORD Status, LPTSTR string )
{
int stringID = -1;
if( Status & PRINTER_STATUS_ERROR )
stringID = IDS_ERROR;
else
if( Status & PRINTER_STATUS_PAUSED )
stringID = IDS_PAUSED;
else
if( Status & PRINTER_STATUS_PENDING_DELETION )
stringID = IDS_PENDING_DELETION;
else
stringID = IDS_READY;
if( stringID != -1 )
{
return LoadString( ghInst, stringID, string, MAX_PATH );
}
return FALSE;
}
/////////////////////////////////////////////////////////////////////////////
//
// SetInfoFields
//
// This routine sets the Printer Information and selected printer textbox
// fields to the currently selected item in the Select Printer listbox.
//
// TO DO:
// error checking for win api calls
// get strings from resource file
//
//
/////////////////////////////////////////////////////////////////////////////
BOOL SetInfoFields (
HWND hWnd,
LPPRINTER_INFO_2 pPrinter
)
{
TCHAR PrinterStatus[MAX_PATH];
BOOL BufferAllocated = FALSE;
ASSERT( GET_BROWSE_DLG_DATA(hWnd)->csLock.bInside() );
if( !pPrinter )
{
SetDlgItemText(hWnd, IDD_BROWSE_DESCRIPTION, TEXT(""));
SetDlgItemText(hWnd, IDD_BROWSE_STATUS, TEXT(""));
SetDlgItemText(hWnd, IDD_BROWSE_DOCUMENTS, TEXT(""));
}
else
{
SetDlgItemText(hWnd, IDD_BROWSE_PRINTER, pPrinter->pPrinterName);
SetDlgItemText(hWnd, IDD_BROWSE_DESCRIPTION, pPrinter->pComment); // !!!???
if(GetPrinterStatusString(pPrinter->Status, PrinterStatus))
SetDlgItemText(hWnd, IDD_BROWSE_STATUS, PrinterStatus);
else
SetDlgItemText(hWnd, IDD_BROWSE_STATUS, TEXT(""));
SetDlgItemInt(hWnd, IDD_BROWSE_DOCUMENTS, (UINT)pPrinter->cJobs, FALSE);
}
return TRUE;
}
/* --- Function: DrawLine() -------------------------------------------------
*
*/
void
DrawLine(
HDC hDC,
LPRECT pRect,
LPTSTR pStr,
BOOL bInvert
)
{
ExtTextOut(hDC, pRect->left, pRect->top, ETO_OPAQUE, (CONST RECT *)pRect,
pStr, _tcslen(pStr), NULL);
}
/* DrawLineWithTabs
*
* Accepts a zero-terminated buffer containing strings delimited by commas
* in the following format: <string> [,<string>[,<string> ... ]]
* where <string> may be zero characters in length,
* e.g.:
* \\ntprint\LASER,HP Laserjet Series II,,other stuff
*
* It takes a copy of the string, and converts any commas into NULLs,
* ensuring that the new buffer has a double NULL termination,
* then steps through calling DrawLine on each NULL-terminated substring.
*/
void
DrawLineWithTabs(
HDC hDC,
LPRECT pRect,
LPTSTR pStr,
BOOL bInvert
)
{
DWORD ColumnWidth = COLUMN_WIDTH; // Arbitrary column width for now
RECT ColumnRect;
TCHAR *pBuffer;
TCHAR *pBufferEnd;
TCHAR OutputBuffer[OUTPUT_BUFFER_LENGTH+2]; // Allow for double null terminator
DWORD StringLength; // Number of TCHARs in string;
DWORD BytesToCopy; // Number of BYTEs in OutputBuffer;
DWORD BufferLength; // NUMBER of TCHARs in OutputBuffer;
#ifdef _HYDRA_
SIZE StrSize;
DWORD ColSize, StrWidth;
#endif
/* Make a copy of the input string so we can mess with it
* without any worries.
* Just in case it's longer than our buffer, copy no more than
* buffer length:
*/
StringLength = _tcslen( pStr );
BytesToCopy = min( ( StringLength * sizeof( TCHAR ) ), OUTPUT_BUFFER_LENGTH );
memcpy( OutputBuffer, pStr, BytesToCopy );
BufferLength = ( BytesToCopy / sizeof( TCHAR ) );
pBufferEnd = &OutputBuffer[BufferLength];
OutputBuffer[BufferLength] = (TCHAR)0; // Ensure double
OutputBuffer[BufferLength+1] = (TCHAR)0; // null terminated
/* Convert commas to nulls:
*/
pBuffer = OutputBuffer;
while( *pBuffer )
{
if( *pBuffer == (TCHAR)',' )
*pBuffer = (TCHAR)0;
pBuffer++;
}
CopyRect( &ColumnRect, (CONST RECT *)pRect );
/* Tokenise the buffer delimited by commas:
*/
pBuffer = OutputBuffer;
while( pBuffer < pBufferEnd )
{
#ifdef _HYDRA_
// For long strings, expand the column size. This prevents
// columns from aligning. But that is better than the alternatives
// which are to truncate the text or to make super wide columns
// that force the user to scroll.
ColSize = ColumnWidth;
if (GetTextExtentPoint32(hDC, pBuffer, _tcslen(pBuffer),
&StrSize) != 0)
{
StrWidth = (DWORD) StrSize.cx;
while (ColSize < StrWidth && ColSize < 8 * ColumnWidth)
ColSize += ColumnWidth / 2;
}
ColumnRect.right = ColumnRect.left + ColSize;
#else
ColumnRect.right = ( ColumnRect.left + ColumnWidth );
#endif
DrawLine( hDC, &ColumnRect, pBuffer, bInvert );
ColumnRect.left = ColumnRect.right;
/* Draw a column separator:
*/
ColumnRect.right = ( ColumnRect.left + COLUMN_SEPARATOR_WIDTH );
DrawLine( hDC, &ColumnRect, TEXT(""), bInvert );
ColumnRect.left = ColumnRect.right;
/* Find and step over the next null:
*/
while( *pBuffer++ )
;
}
ColumnRect.right = pRect->right;
DrawLine( hDC, &ColumnRect, TEXT(""), bInvert );
}
/* DisplayStatusIcon
*
* andrewbe - May 1992
*/
BOOL DisplayStatusIcon( HDC hdc, PRECT prect, int xBase, int yBase, BOOL Highlight )
{
BOOL OK;
int right;
right = prect->right;
if( ( SysColorWindow != GetSysColor(COLOR_WINDOW))
||( SysColorHighlight != GetSysColor(COLOR_HIGHLIGHT)))
FixupBitmapColours( );
OK = BitBlt( hdc, prect->left + STATUS_BITMAP_MARGIN,
prect->top,
STATUS_BITMAP_WIDTH,
STATUS_BITMAP_HEIGHT,
hdcBitmap,
xBase,
Highlight ? ( yBase + STATUS_BITMAP_HEIGHT ) : yBase,
SRCCOPY );
if( OK )
{
/* Draw around it so we don't get a flashing effect on the highlight line:
*/
prect->right = ( prect->left + STATUS_BITMAP_MARGIN );
DrawLine( hdc, prect, TEXT(""), Highlight );
prect->left += STATUS_BITMAP_MARGIN + STATUS_BITMAP_WIDTH;
prect->right = prect->left + STATUS_BITMAP_MARGIN;
DrawLine( hdc, prect, TEXT(""), Highlight );
prect->left += STATUS_BITMAP_MARGIN;
}
else
{
prect->right = STATUS_BITMAP_SPACE;
DrawLine( hdc, prect, TEXT(""), Highlight );
prect->left += STATUS_BITMAP_SPACE;
}
/* Restore the right coordinate (left has now been updated to the new position):
*/
prect->right = right;
return OK;
}
/////////////////////////////////////////////////////////////////////////////
//
// LoadBitmaps
//
// this routine loads DIB bitmaps, and "fixes up" their color tables
// so that we get the desired result for the device we are on.
//
// this routine requires:
// the DIB is a 16 color DIB authored with the standard windows colors
// bright green (00 FF 00) is converted to the background color!
// bright magenta (FF 00 FF) is converted to the background color!
// light grey (C0 C0 C0) is replaced with the button face color
// dark grey (80 80 80) is replaced with the button shadow color
//
// this means you can't have any of these colors in your bitmap
//
/////////////////////////////////////////////////////////////////////////////
DWORD FlipColor(DWORD rgb)
{
return RGB(GetBValue(rgb), GetGValue(rgb), GetRValue(rgb));
}
BOOL LoadBitmaps()
{
HDC hdc;
HANDLE h;
DWORD FAR *pColorTable;
LPBYTE lpBits;
LPBITMAPINFOHEADER lpBitmapInfo;
int i;
UINT cbBitmapSize;
LPBITMAPINFOHEADER lpBitmapData;
h = FindResource(ghInst, MAKEINTRESOURCE(IDB_BROWSE), RT_BITMAP);
if( !h )
return FALSE;
hRes = LoadResource(ghInst, (HRSRC)h);
if( !hRes )
return FALSE;
/* Lock the bitmap and get a pointer to the color table. */
lpBitmapInfo = (LPBITMAPINFOHEADER)LockResource(hRes);
if (!lpBitmapInfo)
return FALSE;
cbBitmapSize = SizeofResource(ghInst, (HRSRC)h);
if (!(lpBitmapData = (LPBITMAPINFOHEADER)LocalAlloc(LMEM_FIXED, cbBitmapSize))) {
FreeResource( hRes );
return FALSE;
}
CopyMemory((PBYTE)lpBitmapData, (PBYTE)lpBitmapInfo, cbBitmapSize);
pColorTable = (DWORD FAR *)((LPBYTE)(lpBitmapData) + lpBitmapData->biSize);
/* Search for the Solid Blue entry and replace it with the current
* background RGB.
*/
if( !ColorIndicesInitialised )
{
for( i = 0; i < 16; i++ )
{
switch( pColorTable[i] )
{
case BACKGROUND:
iBackground = i;
break;
case BACKGROUNDSEL:
iBackgroundSel = i;
break;
case BUTTONFACE:
iButtonFace = i;
break;
case BUTTONSHADOW:
iButtonShadow = i;
break;
}
}
ColorIndicesInitialised = TRUE;
}
pColorTable[iBackground] = FlipColor(GetSysColor(COLOR_WINDOW));
pColorTable[iBackgroundSel] = FlipColor(GetSysColor(COLOR_HIGHLIGHT));
pColorTable[iButtonFace] = FlipColor(GetSysColor(COLOR_BTNFACE));
pColorTable[iButtonShadow] = FlipColor(GetSysColor(COLOR_BTNSHADOW));
UnlockResource(hRes);
/* First skip over the header structure */
lpBits = (LPBYTE)(lpBitmapData + 1);
/* Skip the color table entries, if any */
lpBits += (1 << (lpBitmapData->biBitCount)) * sizeof(RGBQUAD);
/* Create a color bitmap compatible with the display device */
BOOL bResult = FALSE;
hdc = GetDC(NULL);
if( hdc && (hdcBitmap = CreateCompatibleDC(hdc)) )
{
if (hbmBitmap = CreateDIBitmap (hdc, lpBitmapData, (DWORD)CBM_INIT,
lpBits, (LPBITMAPINFO)lpBitmapData, DIB_RGB_COLORS))
{
hbmDefault = (HBITMAP)SelectObject(hdcBitmap, hbmBitmap);
bResult = TRUE;
}
}
if( hdc )
{
ReleaseDC(NULL, hdc);
}
GlobalUnlock(hRes);
FreeResource(hRes);
LocalFree(lpBitmapData);
return bResult;
}
/* I'm sure there's a better way to do this.
* We should be able to modify the colour palette,
* but I haven't managed to make it work...
*/
BOOL FixupBitmapColours( )
{
FreeBitmaps( );
LoadBitmaps( );
return TRUE;
}
INT APIENTRY GetHeightFromPointsString(DWORD Points)
{
HDC hdc;
INT height = Points;
hdc = GetDC(NULL);
if( hdc )
{
height = MulDiv( -(LONG)(Points), GetDeviceCaps(hdc, LOGPIXELSY), 72 );
ReleaseDC(NULL, hdc);
}
return height;
}
VOID FreeBitmaps( )
{
SelectObject( hdcBitmap, hbmDefault );
DeleteObject( hbmBitmap );
DeleteDC( hdcBitmap );
}
/* GetRegShowLogonDomainFlag
*
* Checks to see whether the current user has disabled the ShowLogonDomain
* flag to stop the default domain being expanded.
*
* If the flag is not there or an error occurs, defaults to TRUE.
*
*/
BOOL GetRegShowLogonDomainFlag( )
{
TStatusB bStatus;
BOOL bShowLogonDomain = FALSE;
//
// Read the show logon domain flag from the registry.
//
TPersist Persist( gszRegPrinters, TPersist::kOpen|TPersist::kRead );
bStatus DBGCHK = VALID_OBJ( Persist );
if( bStatus )
{
bStatus DBGCHK = Persist.bRead( gszShowLogonDomain, bShowLogonDomain );
}
return bShowLogonDomain;
}
/* SetRegShowLogonDomainFlag
*
*
*/
BOOL SetRegShowLogonDomainFlag( BOOL bShowLogonDomain )
{
TStatusB bStatus;
//
// Write the show logon domain flag to the registry.
//
TPersist Persist( gszRegPrinters, TPersist::kCreate|TPersist::kWrite );
bStatus DBGCHK = VALID_OBJ( Persist );
if( bStatus )
{
bStatus DBGCHK = Persist.bWrite( gszShowLogonDomain, bShowLogonDomain );
}
return bStatus;
}
/* Strip out carriage return and linefeed characters,
* and convert them to spaces:
*/
VOID RemoveCrLf( LPTSTR pString )
{
while( *pString )
{
if( ( 0x0d == *pString ) || ( 0x0a == *pString ) )
*pString = ' ';
pString++;
}
}
VOID UpdateError( HWND hwnd, DWORD Error )
{
TCHAR ErrorTitle[20] = TEXT("");
TCHAR ErrorText[1048];
LPTSTR pErrorString;
if( Error == NO_ERROR )
{
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DESCRIPTION_TX ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DESCRIPTION ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_STATUS_TX ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_STATUS ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DOCUMENTS_TX ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DOCUMENTS ), SW_SHOW );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_ERROR ), SW_HIDE );
SetDlgItemText(hwnd, IDD_BROWSE_ERROR, TEXT(""));
}
else
{
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DESCRIPTION_TX ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DESCRIPTION ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_STATUS_TX ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_STATUS ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DOCUMENTS_TX ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_DOCUMENTS ), SW_HIDE );
ShowWindow( GetDlgItem( hwnd, IDD_BROWSE_ERROR ), SW_SHOW );
if( !*ErrorTitle )
LoadString( ghInst, IDS_ERROR, ErrorTitle, COUNTOF(ErrorTitle));
if( *ErrorTitle )
{
TString strErrorString;
TResult Result( Error );
Result.bGetErrorString( strErrorString );
pErrorString = (LPTSTR)(LPCTSTR)strErrorString;
if( pErrorString )
{
RemoveCrLf( pErrorString );
_stprintf( ErrorText,
TEXT("%s: %s"),
ErrorTitle,
pErrorString );
SetDlgItemText(hwnd, IDD_BROWSE_ERROR, ErrorText);
}
}
}
}
/*++
Routine Name:
SPLSETUP_DATA
Routine Description:
Constructor/Destructor
Arguments:
Return Value:
--*/
SPLSETUP_DATA::
SPLSETUP_DATA(
VOID
) : hModule(NULL),
hDevInfo(INVALID_HANDLE_VALUE),
pSetupLocalData(NULL),
pfnCreatePrinterDeviceInfoList(NULL),
pfnDestroyPrinterDeviceInfoList(NULL),
pfnSelectDriver(NULL),
pfnGetSelectedDriverInfo(NULL),
pfnDestroySelectedDriverInfo(NULL),
pfnInstallPrinterDriver(NULL),
pfnThisPlatform(NULL),
pfnDriverInfoFromName(NULL),
pfnGetPathToSearch(NULL),
pfnBuildDriversFromPath(NULL),
pfnIsDriverInstalled(NULL),
pfnGetLocalDataField(NULL),
pfnFreeDrvField(NULL),
pfnProcessPrinterAdded(NULL),
pfnFindMappedDriver(NULL),
pfnInstallInboxDriverSilently(NULL),
pfnFreeMem(NULL),
bValid(FALSE),
bDriverAdded(FALSE),
dwLastError(ERROR_SUCCESS)
{
if ( (hModule = LoadLibrary(TEXT("ntprint.dll"))) &&
(pfnCreatePrinterDeviceInfoList
= (pfPSetupCreatePrinterDeviceInfoList)GetProcAddress(hModule, "PSetupCreatePrinterDeviceInfoList")) &&
(pfnDestroyPrinterDeviceInfoList
= (pfPSetupDestroyPrinterDeviceInfoList)GetProcAddress(hModule, "PSetupDestroyPrinterDeviceInfoList")) &&
(pfnSelectDriver
= (pfPSetupSelectDriver)GetProcAddress(hModule, "PSetupSelectDriver")) &&
(pfnGetSelectedDriverInfo
= (pfPSetupGetSelectedDriverInfo)GetProcAddress(hModule, "PSetupGetSelectedDriverInfo")) &&
(pfnDestroySelectedDriverInfo
= (pfPSetupDestroySelectedDriverInfo)GetProcAddress(hModule, "PSetupDestroySelectedDriverInfo")) &&
(pfnInstallPrinterDriver
= (pfPSetupInstallPrinterDriver)GetProcAddress(hModule, "PSetupInstallPrinterDriver")) &&
(pfnThisPlatform
= (pfPSetupThisPlatform)GetProcAddress(hModule, "PSetupThisPlatform")) &&
(pfnDriverInfoFromName
= (pfPSetupDriverInfoFromName)GetProcAddress(hModule, "PSetupDriverInfoFromName")) &&
(pfnGetPathToSearch
= (pfPSetupGetPathToSearch)GetProcAddress(hModule, "PSetupGetPathToSearch")) &&
(pfnBuildDriversFromPath
= (pfPSetupBuildDriversFromPath)GetProcAddress(hModule, "PSetupBuildDriversFromPath")) &&
(pfnIsDriverInstalled
= (pfPSetupIsDriverInstalled)GetProcAddress(hModule, "PSetupIsDriverInstalled")) &&
(pfnGetLocalDataField
= (pfPSetupGetLocalDataField)GetProcAddress(hModule, "PSetupGetLocalDataField")) &&
(pfnFreeDrvField
= (pfPSetupFreeDrvField)GetProcAddress(hModule, "PSetupFreeDrvField")) &&
(pfnProcessPrinterAdded
= (pfPSetupProcessPrinterAdded)GetProcAddress(hModule, "PSetupProcessPrinterAdded")) &&
(pfnFindMappedDriver
= (pfPSetupFindMappedDriver)GetProcAddress(hModule, "PSetupFindMappedDriver")) &&
(pfnInstallInboxDriverSilently
= (pfPSetupInstallInboxDriverSilently)GetProcAddress(hModule, "PSetupInstallInboxDriverSilently")) &&
(pfnFreeMem
= (pfPSetupFreeMem)GetProcAddress(hModule, "PSetupFreeMem"))
)
{
bValid = TRUE;
}
}
SPLSETUP_DATA::
~SPLSETUP_DATA(
VOID
)
{
//
// Free up the alocated resources.
//
FreeDriverInfo( );
if( hModule )
{
FreeLibrary( hModule );
}
}
VOID
SPLSETUP_DATA::
FreeDriverInfo(
VOID
)
{
if( hDevInfo != INVALID_HANDLE_VALUE )
{
pfnDestroyPrinterDeviceInfoList( hDevInfo );
hDevInfo = INVALID_HANDLE_VALUE;
}
if( pSetupLocalData )
{
pfnDestroySelectedDriverInfo( pSetupLocalData );
pSetupLocalData = NULL;
}
}
BOOL
SPLSETUP_DATA::
LoadDriverInfo(
IN HWND hwnd,
IN LPWSTR pszDriver
)
{
FreeDriverInfo( );
//
// Put up the Model/Manf dialog and install the printer driver selected
// by the user
//
hDevInfo = pfnCreatePrinterDeviceInfoList( hwnd );
if ( hDevInfo == INVALID_HANDLE_VALUE )
{
return FALSE;
}
//
// If we know the driver name we will see if it is found in the infs on
// the system
//
if( pszDriver )
{
pSetupLocalData = pfnDriverInfoFromName( hDevInfo, pszDriver );
}
//
// If we do not know the driver name or if the driver is not found in
// the infs we will prompt and let the user select a driver
//
if( !pSetupLocalData )
{
if( !pfnSelectDriver( hDevInfo ) )
{
return FALSE;
}
pSetupLocalData = pfnGetSelectedDriverInfo( hDevInfo );
if( !pSetupLocalData )
{
return FALSE;
}
}
return TRUE;
}
BOOL
SPLSETUP_DATA::
LoadKnownDriverInfo(
IN HWND hwnd,
IN LPWSTR pszDriver
)
{
TString strFormat;
TString strTitle;
TCHAR szInfDir[MAX_PATH];
FreeDriverInfo( );
hDevInfo = pfnCreatePrinterDeviceInfoList( hwnd );
if( hDevInfo == INVALID_HANDLE_VALUE )
{
return FALSE;
}
pSetupLocalData = pfnDriverInfoFromName( hDevInfo, pszDriver );
if( !pSetupLocalData )
{
if( !strFormat.bLoadString( ghInst, IDS_PROMPTFORINF ) ||
!strTitle.bFormat( strFormat, pszDriver ) )
{
return FALSE;
}
szInfDir[0] = TEXT('\0');
if( pfnGetPathToSearch( hwnd, strTitle, NULL, TEXT("*.INF"), FALSE, szInfDir ) &&
pfnBuildDriversFromPath(hDevInfo, szInfDir, FALSE) )
{
pSetupLocalData = pfnDriverInfoFromName( hDevInfo, pszDriver );
}
}
if( !pSetupLocalData )
{
return FALSE;
}
return TRUE;
}
VOID
SPLSETUP_DATA::
ReportErrorMessage(
IN HWND hwnd
)
{
if( dwLastError != ERROR_SUCCESS )
{
switch( dwLastError )
{
case ERROR_UNKNOWN_PRINTER_DRIVER:
{
iMessage( hwnd,
IDS_INSTALLDRIVER,
IDS_ERROR_UNKNOWN_DRIVER,
MB_OK | MB_ICONSTOP,
kMsgNone,
NULL );
}
break;
case ERROR_CANCELLED:
// do nothing
break;
default:
{
iMessage( hwnd,
IDS_INSTALLDRIVER,
IDS_ERRORRUNNINGSPLSETUP,
MB_OK | MB_ICONSTOP,
dwLastError,
NULL );
}
break;
}
}
}
inline static BOOL
IsNULLDriver(LPWSTR pszDriver)
{
// the spooler is setting the driver name to "NULL" for downlevel
// print servers, so applications don't crash. pretty weird, but
// that's how it is.
return (
NULL == pszDriver ||
0 == pszDriver[0] ||
0 == lstrcmpi(pszDriver, TEXT("NULL"))
);
}
BOOL
AddDriver(
IN SPLSETUP_DATA *pSplSetupData,
IN HWND hwnd,
IN LPWSTR pszDriver,
IN BOOL bPromptUser,
OUT LPWSTR *ppszDriverOut
)
/*++
Routine Description:
Verifies the user wants to install a driver, then puts up UI to
select the driver (pre-selects based on pszDriver), then invokes
install code.
Arguments:
pSplSetupData - The setup data used for the driver installation from
ntprint.dll.
hwnd - Parent window.
pszDriver - The driver from the remote printer provider (if known).
bPromptUser - If TRUE, the user should be prompted before installation.
ppszDriverOut - Driver selected by user, must be freed by callee
when this call succeeds.
Return Value:
TRUE = success, FALSE = FAILURE.
Notes:
Doesn't allow third party infs.
--*/
{
BOOL bRet = FALSE;
DRIVER_FIELD DrvField;
// Put up a message box to confirm that the user wants
// to install a driver locally:
//
if (bPromptUser)
{
if( iMessage( hwnd,
IDS_CONNECTTOPRINTER,
IsNULLDriver(pszDriver) ? IDS_CONFIRMINSTALLDRIVER : IDS_CONFIRMINSTALLKNOWNDRIVER,
MB_OKCANCEL | MB_ICONEXCLAMATION,
kMsgNone,
NULL,
pszDriver ) != IDOK )
{
return FALSE;
}
}
DrvField.Index = DRIVER_NAME;
DrvField.pszDriverName = NULL;
SPLASSERT( pSplSetupData );
if ( pSplSetupData->LoadDriverInfo( hwnd, pszDriver ) &&
pSplSetupData->pfnGetLocalDataField(
pSplSetupData->pSetupLocalData,
pSplSetupData->pfnThisPlatform(),
&DrvField) &&
(pSplSetupData->pfnIsDriverInstalled(NULL,
DrvField.pszDriverName,
pSplSetupData->pfnThisPlatform(),
KERNEL_MODE_DRIVER_VERSION) ||
ERROR_SUCCESS == (pSplSetupData->dwLastError = pSplSetupData->pfnInstallPrinterDriver(
pSplSetupData->hDevInfo,
pSplSetupData->pSetupLocalData,
NULL,
pSplSetupData->pfnThisPlatform(),
SPOOLER_VERSION,
NULL,
hwnd,
NULL,
NULL,
0,
APD_COPY_NEW_FILES,
NULL))) )
{
pSplSetupData->bDriverAdded = TRUE;
*ppszDriverOut = AllocSplStr( DrvField.pszDriverName );
if ( *ppszDriverOut )
{
// success here
bRet = TRUE;
}
}
if( DrvField.pszDriverName )
{
pSplSetupData->pfnFreeDrvField(&DrvField);
}
if( !bRet && ERROR_SUCCESS == pSplSetupData->dwLastError )
{
pSplSetupData->dwLastError = GetLastError();
//
// If ntprint.dll fails, but doesn't set the last error then
// we will set a bizarre last error here, so the UI will end
// up displaying a generic error message instead of silence.
//
if( ERROR_SUCCESS == pSplSetupData->dwLastError )
{
pSplSetupData->dwLastError = ERROR_INVALID_FUNCTION;
}
}
return bRet;
}
BOOL
AddKnownDriver(
IN SPLSETUP_DATA *pSplSetupData,
IN HWND hwnd,
IN LPWSTR pszDriver,
IN BOOL bSamePlatform
)
/*++
Routine Description:
Adds a known driver. Doesn't prompt for printer name or driver
selection; calls setup directly.
Arguments:
hwnd - Parent hwnd.
pszDriver - Driver to install.
Return Value:
TRUE = success, FALSE = failure.
--*/
{
ASSERT(pSplSetupData);
BOOL bRet = FALSE;
//
// Put up a message box to confirm that the user wants
// to install a driver locally:
//
if (iMessage(hwnd,
IDS_CONNECTTOPRINTER,
IDS_CONFIRMINSTALLKNOWNDRIVER,
MB_OKCANCEL | MB_ICONEXCLAMATION,
kMsgNone,
NULL,
pszDriver) != IDOK)
{
return FALSE;
}
// don't offer replacement inbox driver in the AddPrinterConnection case
DWORD dwInstallFlags = DRVINST_DONT_OFFER_REPLACEMENT;
if( pSplSetupData->LoadKnownDriverInfo( hwnd, pszDriver ) )
{
pSplSetupData->dwLastError = pSplSetupData->pfnInstallPrinterDriver(
pSplSetupData->hDevInfo,
pSplSetupData->pSetupLocalData,
NULL,
pSplSetupData->pfnThisPlatform(),
SPOOLER_VERSION,
NULL,
hwnd,
NULL,
NULL,
dwInstallFlags,
APD_COPY_NEW_FILES,
NULL );
if ( ERROR_SUCCESS == pSplSetupData->dwLastError )
{
pSplSetupData->bDriverAdded = TRUE;
bRet = TRUE;
}
}
if( !bRet && ERROR_SUCCESS == pSplSetupData->dwLastError )
{
pSplSetupData->dwLastError = GetLastError();
//
// If ntprint.dll fails, but doesn't set the last error then
// we will set a bizarre last error here, so the UI will end
// up displaying a generic error message instead of silence.
//
if( ERROR_SUCCESS == pSplSetupData->dwLastError )
{
pSplSetupData->dwLastError = ERROR_INVALID_FUNCTION;
}
}
return bRet;
}
/*++
Routine Name:
IsNTServer
Routine Description:
Returns whether the given server is a windows nt server or not.
Arguments:
pszServerName -
Return Value:
An HRESULT - S_OK - It is an NT server.
S_FALSE - It is not an NT server.
E_X - There was an error.
--*/
HRESULT
IsNTServer(
IN PCWSTR pszServerName
)
{
HANDLE hServer = NULL;
PRINTER_DEFAULTS PrinterDef = { NULL, NULL, SERVER_READ };
TStatusH hRetval;
//
// If we can open the remote server, then this is an NT box.
//
hRetval DBGCHK = OpenPrinter(const_cast<PWSTR>(pszServerName), &hServer, &PrinterDef) ? S_OK : CreateHRFromWin32();
if (hServer)
{
ClosePrinter(hServer);
}
return SUCCEEDED(hRetval) ? S_OK : S_FALSE;
}
/*++
Routine Name:
CreateLocalPort
Routine Description:
This creates a local port of the given name by communicating with the
local monitor.
Arguments:
pszPortName - The port name to add.
Return Value:
TRUE - Success, FALSE - Failure.
--*/
BOOL
CreateLocalPort(
IN PCWSTR pszPortName
)
{
HANDLE hXcv = NULL;
PRINTER_DEFAULTS Defaults = { NULL, NULL, SERVER_ACCESS_ADMINISTER };
TStatusB bRet;
if (pszPortName == NULL)
{
SetLastError(ERROR_INVALID_PARAMETER);
bRet DBGCHK = FALSE;
}
//
// Open a Xcv handle to the local monitor.
//
if (bRet)
{
bRet DBGCHK = OpenPrinter(const_cast<PWSTR>(gszXvcLocalMonitor), &hXcv, &Defaults);
}
//
// Ask it to add a port for us.
//
if (bRet)
{
DWORD cbOutputNeeded = 0;
DWORD Status = ERROR_SUCCESS;
bRet DBGCHK = XcvData(hXcv,
gszAddPort,
reinterpret_cast<BYTE *>(const_cast<PWSTR>(pszPortName)),
(wcslen(pszPortName) + 1) * sizeof(WCHAR),
NULL,
0,
&cbOutputNeeded,
&Status);
if (bRet)
{
//
// If it succeeded, or the port already exists, we are OK.
//
if (Status != ERROR_SUCCESS && Status != ERROR_ALREADY_EXISTS)
{
SetLastError(Status);
bRet DBGCHK = FALSE;
}
}
}
if (hXcv)
{
ClosePrinter(hXcv);
}
return bRet;
}
LPVOID
AllocSplMem(
DWORD cb
)
{
LPVOID pReturn = AllocMem( cb );
if( pReturn )
{
memset( pReturn, 0, cb );
}
return pReturn;
}
BOOL
FreeSplMem(
LPVOID pMem
)
{
FreeMem( pMem );
return TRUE;
}
LPVOID
ReallocSplMem(
LPVOID pOldMem,
DWORD cbOld,
DWORD cbNew
)
{
LPVOID pNewMem = AllocSplMem( cbNew );
if (pOldMem && pNewMem)
{
if (cbOld)
{
memcpy(pNewMem, pOldMem, min(cbNew, cbOld));
}
FreeSplMem(pOldMem);
}
return pNewMem;
}
LPTSTR
AllocSplStr(
LPCTSTR pStr
)
{
LPTSTR pMem = NULL;
if( pStr )
{
pMem = (LPTSTR)AllocSplMem( _tcslen(pStr) * sizeof(TCHAR) + sizeof(TCHAR) );
if( pMem )
{
_tcscpy( pMem, pStr );
}
}
return pMem;
}
BOOL
FreeSplStr(
LPTSTR pStr
)
{
FreeSplMem( pStr );
return TRUE;
}
static MSG_HLPMAP MsgHelpTable [] =
{
{ ERROR_KM_DRIVER_BLOCKED, IDS_NULLSTR, gszHelpTroubleShooterURL},
{ 0, 0, 0}
};
DWORD
ReportFailure(
HWND hwndParent,
DWORD idTitle,
DWORD idDefaultError
)
{
DWORD dwLastError = GetLastError();
if( dwLastError == ERROR_KM_DRIVER_BLOCKED )
{
iMessageEx ( hwndParent,
idTitle,
IDS_COULDNOTCONNECTTOPRINTER_BLOCKED_HELP,
MB_OK | MB_ICONSTOP,
dwLastError,
NULL,
dwLastError,
MsgHelpTable);
}
else
{
iMessageEx( hwndParent,
idTitle,
idDefaultError,
MB_OK | MB_ICONSTOP,
dwLastError,
NULL,
bGoodLastError(dwLastError) ? -1 : idDefaultError,
NULL );
}
return dwLastError;
}
BROWSE_DLG_DATA::
BROWSE_DLG_DATA(
VOID
) : hPrinter( NULL ),
Request( NULL ),
RequestComplete( NULL ),
pConnectToData( NULL ),
Status( 0 ),
cExpandObjects( 0 ),
ExpandSelection( NULL ),
dwExtent( 0 ),
pEnumerateName( NULL ),
pEnumerateObject( NULL ),
Message( 0 ),
wParam( 0 ),
lParam( 0 ),
pPrinterInfo( NULL ),
cbPrinterInfo( 0 ),
hwndDialog( NULL ),
_bValid( FALSE ),
pPageSwitchController( NULL ),
bInPropertyPage( FALSE )
{
DBGMSG( DBG_TRACE, ( "BROWSE_DLG_DATA::ctor\n") );\
hcursorArrow = LoadCursor(NULL, IDC_ARROW);
hcursorWait = LoadCursor(NULL, IDC_WAIT);
//
// Leave _bValid = FALSE if cursors are *not* loaded properly
//
if( hcursorArrow && hcursorWait )
{
_bValid = MRefCom::bValid();
}
}
BROWSE_DLG_DATA::
~BROWSE_DLG_DATA(
VOID
)
{
DBGMSG( DBG_TRACE, ( "BROWSE_DLG_DATA::dtor\n") );
if( pPrinterInfo )
FreeSplMem( pPrinterInfo );
if (pConnectToData)
FreeSplMem(pConnectToData);
if (RequestComplete)
CloseHandle(RequestComplete);
if (Request)
CloseHandle(Request);
}
VOID
BROWSE_DLG_DATA::
vRefZeroed(
VOID
)
{
delete this;
}
BOOL
BROWSE_DLG_DATA::
bValid(
VOID
)
{
return _bValid;
}
BOOL
BROWSE_DLG_DATA::
bInitializeBrowseThread(
HWND hWnd
)
{
if( !LoadBitmaps() )
{
DBGMSG( DBG_WARN, ( "Bitmaps are not loaded properly" ) );
return FALSE;
}
DWORD ThreadId;
HANDLE hThread;
//
// !! WARNING !!
// Assumes ->Request, ->RequestComplete, ->pConnectToData zero initialized!
//
Request = CreateEvent( NULL,
EVENT_RESET_AUTOMATIC,
EVENT_INITIAL_STATE_NOT_SIGNALED,
NULL );
RequestComplete = CreateEvent( NULL,
EVENT_RESET_AUTOMATIC,
EVENT_INITIAL_STATE_NOT_SIGNALED,
NULL );
if( !RequestComplete || !Request )
{
DBGMSG( DBG_WARN, ( "CreateEvent failed: Error %d\n", GetLastError( ) ) );
return FALSE;
}
if( !( pConnectToData = (PCONNECTTO_OBJECT)AllocSplMem( sizeof( CONNECTTO_OBJECT ) ) ) )
{
return FALSE;
}
vIncRef();
hThread = TSafeThread::Create( NULL,
0,
(LPTHREAD_START_ROUTINE)BrowseThread,
this,
0,
&ThreadId );
if( !hThread )
{
DBGMSG( DBG_WARN, ( "CreateThread of BrowseThread failed: Error %d\n", GetLastError( ) ) );
cDecRef();
return FALSE;
}
CloseHandle( hThread );
if( bInPropertyPage || GetRegShowLogonDomainFlag( ) )
Status |= BROWSE_STATUS_INITIAL | BROWSE_STATUS_EXPAND;
ENTER_CRITICAL( this );
SEND_BROWSE_THREAD_REQUEST( this,
BROWSE_THREAD_ENUM_OBJECTS,
NULL,
pConnectToData );
LEAVE_CRITICAL( this );
hwndDialog = hWnd;
SetCursorShape( hwndDialog );
return TRUE;
}
Reference in a new issue Copy permalink