mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
468 lines
12 KiB
468 lines
12 KiB
/*++
|
|
|
|
Copyright (c) 1990 - 1995 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
server.c
|
|
|
|
Abstract:
|
|
|
|
Browsing
|
|
|
|
This module contains the thread for notifying all Printer Servers
|
|
|
|
Author:
|
|
|
|
Dave Snipp (DaveSn) 2-Aug-1992
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include <precomp.h>
|
|
#include <lm.h>
|
|
|
|
DWORD ServerThreadRunning = FALSE;
|
|
HANDLE ServerThreadSemaphore = NULL;
|
|
DWORD ServerThreadTimeout = TEN_MINUTES;
|
|
DWORD RefreshTimesPerDecayPeriod = DEFAULT_REFRESH_TIMES_PER_DECAY_PERIOD;
|
|
DWORD BrowsePrintWorkstations = DEFAULT_NUMBER_BROWSE_WORKSTATIONS;
|
|
BOOL bNetInfoReady = FALSE; // TRUE when the browse list is "valid"
|
|
#define NT_SERVER ( SV_TYPE_SERVER_NT | SV_TYPE_DOMAIN_CTRL | SV_TYPE_DOMAIN_BAKCTRL )
|
|
|
|
extern FARPROC pfnNetServerEnum;
|
|
extern FARPROC pfnNetApiBufferFree;
|
|
|
|
DWORD
|
|
ServerThread(
|
|
PVOID
|
|
);
|
|
|
|
BOOL
|
|
UpdateServer(
|
|
LPCTSTR pszServer
|
|
);
|
|
|
|
|
|
BOOL
|
|
CreateServerThread(
|
|
VOID
|
|
)
|
|
{
|
|
HANDLE ThreadHandle;
|
|
DWORD ThreadId;
|
|
|
|
SplInSem();
|
|
|
|
if (!ServerThreadRunning) {
|
|
|
|
ServerThreadSemaphore = CreateEvent( NULL, FALSE, FALSE, NULL );
|
|
|
|
ThreadHandle = CreateThread( NULL, INITIAL_STACK_COMMIT,
|
|
(LPTHREAD_START_ROUTINE)ServerThread,
|
|
NULL,
|
|
0, &ThreadId );
|
|
|
|
if (!SetThreadPriority(ThreadHandle,
|
|
dwServerThreadPriority))
|
|
DBGMSG(DBG_WARNING, ("Setting thread priority failed %d\n",
|
|
GetLastError()));
|
|
|
|
ServerThreadRunning = TRUE;
|
|
|
|
CloseHandle( ThreadHandle );
|
|
}
|
|
|
|
if( ServerThreadSemaphore != NULL ){
|
|
|
|
// CreateServerThread is called each time a printer is shared out
|
|
// see net.c ShareThisPrinter.
|
|
// So if the ServerThread is sleeping wake prematurely so it can start
|
|
// to tell the world about this new shared printed.
|
|
|
|
SetEvent( ServerThreadSemaphore );
|
|
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
DWORD
|
|
ServerThread(
|
|
PVOID pv
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Notify other machines in our domain about our shared printers.
|
|
|
|
We are going to have to enter and leave, revalidate, enter and leave our
|
|
semaphore inside the loop.
|
|
|
|
Arguments:
|
|
|
|
Return Value:
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD NoReturned, i, Total;
|
|
PSERVER_INFO_101 pserver_info_101;
|
|
PINIPRINTER pIniPrinter;
|
|
PINISPOOLER pIniSpooler;
|
|
DWORD ReturnValue=FALSE;
|
|
WCHAR ServerName[128];
|
|
DWORD StartTickCount;
|
|
DWORD TimeForAllServers;
|
|
DWORD dwActualWaitTime = ServerThreadTimeout;
|
|
UINT cPrintWorkstations;
|
|
UINT cPrintServers;
|
|
UINT cServersToInform;
|
|
UINT cWorkStationsToInform;
|
|
|
|
ServerName[0] = ServerName[1] = '\\';
|
|
|
|
while (TRUE) {
|
|
|
|
SplOutSem();
|
|
|
|
DBGMSG( DBG_TRACE, ("ServerThread sleeping for %d\n", dwActualWaitTime));
|
|
|
|
WaitForSingleObject( ServerThreadSemaphore, dwActualWaitTime );
|
|
|
|
// Wait for a couple of minutes more to avoid the boot time crunch
|
|
Sleep(TWO_MINUTES);
|
|
|
|
if ( !ServerThreadRunning ) {
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
SPLASSERT( pfnNetServerEnum != NULL );
|
|
|
|
if (!(*pfnNetServerEnum)(NULL, 101, (LPBYTE *)&pserver_info_101, -1,
|
|
&NoReturned, &Total, SV_TYPE_PRINTQ_SERVER,
|
|
NULL, NULL)) {
|
|
EnterSplSem();
|
|
|
|
StartTickCount = GetTickCount();
|
|
|
|
//
|
|
// 1 Master + 3 Backup + 1 Backup per 32 Printer Servers.
|
|
//
|
|
|
|
cServersToInform = DEFAULT_NUMBER_MASTER_AND_BACKUP + NoReturned/32 ;
|
|
cWorkStationsToInform = BrowsePrintWorkstations;
|
|
|
|
//
|
|
// Count the NT Server and Workstation machines ( which have a printq )
|
|
//
|
|
|
|
for ( i = 0, cPrintServers = 0, cPrintWorkstations = 0;
|
|
i < NoReturned;
|
|
i++ ) {
|
|
|
|
if ( pserver_info_101[i].sv101_type & NT_SERVER ) {
|
|
|
|
cPrintServers++;
|
|
|
|
} else if ( pserver_info_101[i].sv101_type & SV_TYPE_NT ) {
|
|
|
|
cPrintWorkstations++;
|
|
}
|
|
}
|
|
|
|
//
|
|
// If there are no NT Servers to inform then up the number of Workstations
|
|
//
|
|
|
|
if ( cPrintServers == 0 ) {
|
|
|
|
cWorkStationsToInform = max( cWorkStationsToInform, cServersToInform );
|
|
cServersToInform = 0;
|
|
|
|
} else if ( cPrintServers < cServersToInform ) {
|
|
|
|
cWorkStationsToInform = max( cWorkStationsToInform, cServersToInform - cPrintServers );
|
|
}
|
|
|
|
|
|
DBGMSG( DBG_TRACE, ("ServerThread NetServerEnum returned %d printer servers will inform %d, workstations %d\n", NoReturned, cServersToInform, cWorkStationsToInform ));
|
|
|
|
//
|
|
// Loop Until we have informed the correct Number of WorkStations and Servers
|
|
//
|
|
|
|
for ( i = 0,
|
|
cPrintServers = 0,
|
|
cPrintWorkstations = 0;
|
|
|
|
i < NoReturned &&
|
|
( cPrintServers < cServersToInform || cPrintWorkstations < cWorkStationsToInform );
|
|
|
|
i++ ) {
|
|
|
|
DBGMSG( DBG_TRACE, ("ServerThread Loop Count %d cPrintServer %d cServersToInform %d cPrintWorkstations %d cWorkStationsToInform %d\n",
|
|
i, cPrintServers, cServersToInform, cPrintWorkstations, cWorkStationsToInform ));
|
|
|
|
|
|
DBGMSG( DBG_TRACE, ("ServerThread %ws type %x\n",pserver_info_101[i].sv101_name, pserver_info_101[i].sv101_type ));
|
|
|
|
if (( pserver_info_101[i].sv101_type & NT_SERVER ) ||
|
|
( pserver_info_101[i].sv101_type & SV_TYPE_NT && cPrintWorkstations < cWorkStationsToInform )) {
|
|
|
|
wcscpy(&ServerName[2], pserver_info_101[i].sv101_name);
|
|
|
|
if( UpdateServer( ServerName )){
|
|
|
|
// Servers are also counted as WorkStations
|
|
|
|
cPrintWorkstations++;
|
|
|
|
if ( pserver_info_101[i].sv101_type & NT_SERVER ) {
|
|
|
|
cPrintServers++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TimeForAllServers = GetTickCount() - StartTickCount;
|
|
|
|
DBGMSG( DBG_TRACE, ("ServerThread took %d milliseconds for %d Workstations %d Servers\n",
|
|
TimeForAllServers, cPrintWorkstations, cPrintServers ));
|
|
|
|
//
|
|
// Calculate time to wait before we try again.
|
|
//
|
|
|
|
if ( NetPrinterDecayPeriod > TimeForAllServers ) {
|
|
|
|
dwActualWaitTime = max( ServerThreadTimeout, ( NetPrinterDecayPeriod - TimeForAllServers ) / RefreshTimesPerDecayPeriod );
|
|
|
|
} else {
|
|
|
|
dwActualWaitTime = ServerThreadTimeout;
|
|
}
|
|
|
|
//
|
|
// Remove WAS Shared Bits
|
|
//
|
|
|
|
//
|
|
// Do this for all spoolers.
|
|
//
|
|
for( pIniSpooler = pLocalIniSpooler;
|
|
pIniSpooler;
|
|
pIniSpooler = pIniSpooler->pIniNextSpooler ){
|
|
|
|
for ( pIniPrinter = pIniSpooler->pIniPrinter;
|
|
pIniPrinter != NULL;
|
|
pIniPrinter = pIniPrinter->pNext ) {
|
|
|
|
SplInSem();
|
|
pIniPrinter->Status &= ~PRINTER_WAS_SHARED;
|
|
}
|
|
}
|
|
|
|
LeaveSplSem();
|
|
|
|
(*pfnNetApiBufferFree)((LPVOID)pserver_info_101);
|
|
}
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
typedef struct _UPDATE_SERVER_MAP_DATA {
|
|
LPCWSTR pszServer;
|
|
BOOL bSuccessfulAdd;
|
|
} UPDATE_SERVER_MAP_DATA, *PUPDATE_SERVER_MAP_DATA;
|
|
|
|
BOOL
|
|
UpdateServerPrinterMap(
|
|
HANDLE h,
|
|
PINIPRINTER pIniPrinter
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Update the a browser server with one pIniPrinter.
|
|
|
|
Leaves Spooler Section--pIniPrinter may be invalid on return
|
|
unless explicitly refcounted by callee.
|
|
|
|
Arguments:
|
|
|
|
pIniPrinter - Printer that should be sent to the server.
|
|
|
|
pszServer - Server that needs to be updated.
|
|
|
|
pbSuccessfulAdd - Indicates whether the add was successful.
|
|
|
|
Return Value:
|
|
|
|
Succes or failure?
|
|
|
|
--*/
|
|
|
|
{
|
|
PUPDATE_SERVER_MAP_DATA pData = (PUPDATE_SERVER_MAP_DATA)h;
|
|
|
|
WCHAR string[MAX_PRINTER_BROWSE_NAME];
|
|
WCHAR Name[MAX_UNC_PRINTER_NAME];
|
|
PRINTER_INFO_1 Printer1;
|
|
HANDLE hPrinter;
|
|
PINISPOOLER pIniSpooler;
|
|
DWORD dwLastError;
|
|
|
|
Printer1.Flags = 0;
|
|
|
|
if (( pIniPrinter->Attributes & PRINTER_ATTRIBUTE_SHARED ) ||
|
|
( pIniPrinter->Status & PRINTER_WAS_SHARED )) {
|
|
|
|
//
|
|
// Pass our Printer Attributes so that AddNetPrinter can remove
|
|
// this printer from the browse list if it is not shared.
|
|
//
|
|
|
|
Printer1.Flags = pIniPrinter->Attributes | PRINTER_ATTRIBUTE_NETWORK;
|
|
|
|
wsprintf(string, L"%ws\\%ws,%ws,%ws",
|
|
pIniPrinter->pIniSpooler->pMachineName,
|
|
pIniPrinter->pName,
|
|
pIniPrinter->pIniDriver->pName,
|
|
pIniPrinter->pLocation ?
|
|
pIniPrinter->pLocation :
|
|
L"");
|
|
|
|
Printer1.pDescription = string;
|
|
|
|
wsprintf(Name, L"%ws\\%ws", pIniPrinter->pIniSpooler->pMachineName,
|
|
pIniPrinter->pName);
|
|
|
|
Printer1.pName = Name;
|
|
|
|
Printer1.pComment = AllocSplStr(pIniPrinter->pComment);
|
|
|
|
SplInSem();
|
|
|
|
LeaveSplSem();
|
|
|
|
//
|
|
// Keep trying until the server is not Too Busy.
|
|
//
|
|
|
|
for ( hPrinter = NULL;
|
|
hPrinter == NULL;
|
|
Sleep( GetTickCount() & 0xfff ) ) {
|
|
|
|
hPrinter = AddPrinter( (LPTSTR)pData->pszServer, 1, (LPBYTE)&Printer1 );
|
|
|
|
dwLastError = GetLastError();
|
|
|
|
if ( hPrinter == NULL && dwLastError != RPC_S_SERVER_TOO_BUSY ) {
|
|
|
|
if ( dwLastError != ERROR_PRINTER_ALREADY_EXISTS ) {
|
|
|
|
pData->bSuccessfulAdd = FALSE;
|
|
}
|
|
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
FreeSplStr(Printer1.pComment);
|
|
|
|
if ( hPrinter != NULL ) {
|
|
|
|
DBGMSG( DBG_TRACE,
|
|
( "ServerThread AddPrinter(%ws, %ws) hPrinter %x Flags %x OK\n",
|
|
pData->pszServer, Printer1.pName, hPrinter, Printer1.Flags));
|
|
|
|
ClosePrinter( hPrinter );
|
|
}
|
|
|
|
EnterSplSem();
|
|
|
|
if ( hPrinter == NULL ) {
|
|
|
|
|
|
if ( GetLastError() != ERROR_PRINTER_ALREADY_EXISTS ) {
|
|
|
|
DBGMSG( DBG_TRACE,
|
|
( "ServerThread AddPrinter(%ws, 1) Flags %x failed %d\n",
|
|
pData->pszServer, Printer1.Flags, GetLastError()));
|
|
|
|
// Don't bother with this server if we get an error
|
|
return FALSE;
|
|
|
|
} else {
|
|
|
|
//
|
|
// 3.51 will return a NULL handle ( so it doesn't need closing
|
|
// and ERROR_PRINTER_ALREADY_EXISTS on success ( see printer.c addnetprinter )
|
|
//
|
|
DBGMSG( DBG_TRACE,
|
|
( "pszServerThread AddPrinter(%ws, %ws) hPrinter %x Flags %x OK\n",
|
|
pData->pszServer, Printer1.pName, hPrinter, Printer1.Flags));
|
|
}
|
|
}
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
UpdateServerSpoolerMap(
|
|
HANDLE h,
|
|
PINISPOOLER pIniSpooler
|
|
)
|
|
{
|
|
//
|
|
// Do this only for spoolers that want this "feature."
|
|
//
|
|
if( pIniSpooler->SpoolerFlags & SPL_SERVER_THREAD ){
|
|
RunForEachPrinter( pIniSpooler, h, UpdateServerPrinterMap );
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
UpdateServer(
|
|
LPCTSTR pszServer
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Update a server about all the printers on this node.
|
|
|
|
Arguments:
|
|
|
|
pszServer - Server to update in the form "\\server."
|
|
|
|
Return Value:
|
|
|
|
TRUE - Successfully added.
|
|
FALSE - Not.
|
|
|
|
--*/
|
|
|
|
{
|
|
UPDATE_SERVER_MAP_DATA Data;
|
|
Data.bSuccessfulAdd = TRUE;
|
|
Data.pszServer = pszServer;
|
|
|
|
RunForEachSpooler( &Data, UpdateServerSpoolerMap );
|
|
|
|
return Data.bSuccessfulAdd;
|
|
}
|
|
|