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.
393 lines
10 KiB
393 lines
10 KiB
/********************************************************************/
|
|
/** Copyright(c) 1989 Microsoft Corporation. **/
|
|
/********************************************************************/
|
|
|
|
//***
|
|
//
|
|
// Filename: main.c
|
|
//
|
|
// Description: This module contains the main procedure of the AFP server
|
|
// service. It will contain code to initialize and install
|
|
// itself and the kernel-mode AFP Server. It also contains
|
|
// code to respond to the server controller. It will also
|
|
// handle service shutdown.
|
|
//
|
|
// ??? Does the service controller log start/stop events etc ??
|
|
// if not log it.
|
|
//
|
|
// History:
|
|
// May 11,1990. NarenG Created original version.
|
|
//
|
|
#define DEFINE_AFP_GLOBALS // This will cause AfpGlobals to be defined.
|
|
#include "afpsvcp.h"
|
|
|
|
// Prototypes of functions used only within this module.
|
|
//
|
|
VOID
|
|
AfpMain(
|
|
IN DWORD argc,
|
|
IN LPWSTR * lpwsServiceArgs
|
|
);
|
|
|
|
VOID
|
|
AfpCleanupAndExit(
|
|
IN DWORD dwError
|
|
);
|
|
|
|
VOID
|
|
AfpControlResponse(
|
|
IN DWORD dwControlCode
|
|
);
|
|
|
|
|
|
//**
|
|
//
|
|
// Call: main.c
|
|
//
|
|
// Returns: none.
|
|
//
|
|
// Description: Will simply register the entry point of the AFP server
|
|
// service with the service controller. The service controller
|
|
// will capture this thread. It will be freed only when
|
|
// the service is shutdown. At that point we will simply exit
|
|
// the process.
|
|
//
|
|
void
|
|
_cdecl
|
|
main( int argc, unsigned char * argv[] )
|
|
{
|
|
SERVICE_TABLE_ENTRY AfpServiceDispatchTable[2];
|
|
|
|
#ifdef DBG
|
|
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
|
COORD coord;
|
|
(VOID)AllocConsole( );
|
|
(VOID)GetConsoleScreenBufferInfo( GetStdHandle(STD_OUTPUT_HANDLE),
|
|
&csbi
|
|
);
|
|
coord.X = (SHORT)(csbi.srWindow.Right - csbi.srWindow.Left + 1);
|
|
coord.Y = (SHORT)((csbi.srWindow.Bottom - csbi.srWindow.Top + 1) * 20);
|
|
(VOID)SetConsoleScreenBufferSize( GetStdHandle(STD_OUTPUT_HANDLE),
|
|
coord
|
|
);
|
|
#endif
|
|
|
|
AFP_UNREFERENCED( argc );
|
|
AFP_UNREFERENCED( argv );
|
|
|
|
|
|
AfpServiceDispatchTable[0].lpServiceName = AFP_SERVICE_NAME;
|
|
AfpServiceDispatchTable[0].lpServiceProc = AfpMain;
|
|
AfpServiceDispatchTable[1].lpServiceName = NULL;
|
|
AfpServiceDispatchTable[1].lpServiceProc = NULL;
|
|
|
|
if ( !StartServiceCtrlDispatcher( AfpServiceDispatchTable ) )
|
|
AfpLogEvent( AFPLOG_CANT_START, 0, NULL,
|
|
GetLastError(), EVENTLOG_ERROR_TYPE );
|
|
|
|
ExitProcess(0);
|
|
|
|
}
|
|
|
|
//**
|
|
//
|
|
// Call: AfpMain
|
|
//
|
|
// Returns: none.
|
|
//
|
|
// Description: This is the main procedure for the Afp Server Service. It
|
|
// will be called when the service is supposed to start itself.
|
|
// It will do all service wide initialization.
|
|
//
|
|
VOID
|
|
AfpMain( DWORD argc, // Command line arguments. Will be ignored.
|
|
LPWSTR * lpwsServiceArgs
|
|
)
|
|
{
|
|
DWORD dwRetCode;
|
|
|
|
|
|
AFP_UNREFERENCED( argc );
|
|
AFP_UNREFERENCED( lpwsServiceArgs );
|
|
|
|
// NULL out all the globals
|
|
//
|
|
ZeroMemory( (LPBYTE)&AfpGlobals, sizeof(AfpGlobals) );
|
|
|
|
// Register the service control handler with the service controller
|
|
//
|
|
AfpGlobals.hServiceStatus = RegisterServiceCtrlHandler(AFP_SERVICE_NAME,
|
|
AfpControlResponse );
|
|
|
|
if ( AfpGlobals.hServiceStatus == (SERVICE_STATUS_HANDLE)0 ) {
|
|
AfpLogEvent( AFPLOG_CANT_START, 0, NULL,
|
|
GetLastError(), EVENTLOG_ERROR_TYPE );
|
|
AfpCleanupAndExit( GetLastError() );
|
|
return;
|
|
}
|
|
|
|
AfpGlobals.ServiceStatus.dwServiceType = SERVICE_WIN32;
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_START_PENDING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = 0;
|
|
AfpGlobals.ServiceStatus.dwWin32ExitCode = NO_ERROR;
|
|
AfpGlobals.ServiceStatus.dwServiceSpecificExitCode= 0;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 1;
|
|
AfpGlobals.ServiceStatus.dwWaitHint =AFP_SERVICE_INSTALL_TIME;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
|
|
// Read in registry information and initialize the kernel-mode
|
|
// server. Initialize the server to accept RPC calls. Initialize
|
|
// all global vriables etc.
|
|
//
|
|
if ( dwRetCode = AfpInitialize() )
|
|
{
|
|
if (AfpGlobals.dwServerState & AFPSTATE_BLOCKED_ON_DOMINFO)
|
|
{
|
|
AfpCleanupAndExit( NO_ERROR );
|
|
}
|
|
else
|
|
{
|
|
AfpCleanupAndExit( dwRetCode );
|
|
}
|
|
return;
|
|
}
|
|
|
|
|
|
// Set the MAC bit for NetServerEnum
|
|
//
|
|
if ( !I_ScSetServiceBits( AfpGlobals.hServiceStatus,
|
|
SV_TYPE_AFP,
|
|
TRUE,
|
|
TRUE,
|
|
NULL ))
|
|
{
|
|
|
|
dwRetCode = GetLastError();
|
|
AfpLogEvent( AFPLOG_CANT_START, 0, NULL,
|
|
GetLastError(), EVENTLOG_ERROR_TYPE );
|
|
AfpCleanupAndExit( dwRetCode );
|
|
return;
|
|
}
|
|
|
|
// now tell the service controller that we are up
|
|
//
|
|
if (AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_START_PENDING)
|
|
{
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_RUNNING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
|
|
SERVICE_ACCEPT_PAUSE_CONTINUE;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = 0;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
}
|
|
|
|
|
|
// Start listening for RPC admin client calls. This will block
|
|
// until RpcMgmtStopServerListening is called while processing a
|
|
// STOP_SERVICE control request.
|
|
//
|
|
if ( dwRetCode = RpcServerListen( 1,
|
|
RPC_C_LISTEN_MAX_CALLS_DEFAULT,
|
|
0 ) ) // Blocking mode
|
|
{
|
|
|
|
AfpLogEvent( AFPLOG_CANT_INIT_RPC, 0, NULL,
|
|
dwRetCode, EVENTLOG_ERROR_TYPE );
|
|
}
|
|
|
|
AfpCleanupAndExit( dwRetCode );
|
|
|
|
}
|
|
|
|
//**
|
|
//
|
|
// Call: AfpCleanupAndExit
|
|
//
|
|
// Returns: none
|
|
//
|
|
// Description: Will free any allocated memory, deinitialize RPC, deinitialize
|
|
// the kernel-mode server and unload it if it was loaded.
|
|
// This could have been called due to an error on SERVICE_START
|
|
// or normal termination.
|
|
//
|
|
VOID
|
|
AfpCleanupAndExit(
|
|
IN DWORD dwError
|
|
)
|
|
{
|
|
|
|
AFP_PRINT( ("AFPSVC_main: Cleaning up and exiting Code = %d\n", dwError));
|
|
|
|
// Tear down and free everything
|
|
//
|
|
AfpTerminate();
|
|
|
|
if ( dwError == NO_ERROR )
|
|
AfpGlobals.ServiceStatus.dwWin32ExitCode = NO_ERROR;
|
|
else {
|
|
AfpGlobals.ServiceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;
|
|
//AFP_ASSERT(0);
|
|
}
|
|
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_STOPPED;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = 0;
|
|
AfpGlobals.ServiceStatus.dwServiceSpecificExitCode = dwError;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
|
|
return;
|
|
}
|
|
|
|
//**
|
|
//
|
|
// Call: AfpControlResponse
|
|
//
|
|
// Returns: none
|
|
//
|
|
// Description: Will respond to control requests from the service controller.
|
|
//
|
|
VOID
|
|
AfpControlResponse( IN DWORD dwControlCode )
|
|
{
|
|
AFP_REQUEST_PACKET AfpRequestPkt;
|
|
DWORD dwRetCode;
|
|
|
|
switch( dwControlCode ) {
|
|
|
|
case SERVICE_CONTROL_STOP:
|
|
|
|
if ( (AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_STOP_PENDING)
|
|
||
|
|
(AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_STOPPED ))
|
|
break;
|
|
|
|
// Announce that we are stopping
|
|
//
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_STOP_PENDING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = 0;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 1;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = AFP_SERVICE_STOP_TIME;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
|
|
// if srvrhlpr thread is blocked retrying to get domain info, unblock it
|
|
SetEvent(AfpGlobals.heventSrvrHlprSpecial);
|
|
|
|
// if srvrhlpr thread was blocked, no more init was done, so we're done
|
|
if (AfpGlobals.dwServerState & AFPSTATE_BLOCKED_ON_DOMINFO)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// This call will unblock the main thread that had called
|
|
// RpcServerListen. We let that thread do the announcing
|
|
// while cleaning up.
|
|
//
|
|
if ( (dwRetCode =
|
|
RpcMgmtStopServerListening( (RPC_BINDING_HANDLE)NULL ))
|
|
!= RPC_S_OK )
|
|
{
|
|
ASSERT (0);
|
|
}
|
|
|
|
|
|
return;
|
|
|
|
case SERVICE_CONTROL_PAUSE:
|
|
|
|
if ( (AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_PAUSE_PENDING)
|
|
||
|
|
(AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_PAUSED ))
|
|
break;
|
|
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_PAUSE_PENDING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = 0;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = AFP_SERVICE_PAUSE_TIME;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
|
|
|
|
// Tell the kernel-mode that we want to pause.
|
|
//
|
|
AfpRequestPkt.dwRequestCode = OP_SERVICE_PAUSE;
|
|
AfpRequestPkt.dwApiType = AFP_API_TYPE_COMMAND;
|
|
|
|
dwRetCode = AfpServerIOCtrl( &AfpRequestPkt );
|
|
|
|
AFP_ASSERT( dwRetCode == NO_ERROR );
|
|
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = 0;
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_PAUSED;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
|
|
SERVICE_ACCEPT_PAUSE_CONTINUE;
|
|
break;
|
|
|
|
case SERVICE_CONTROL_CONTINUE:
|
|
|
|
if ( (AfpGlobals.ServiceStatus.dwCurrentState==SERVICE_CONTINUE_PENDING)
|
|
||
|
|
(AfpGlobals.ServiceStatus.dwCurrentState == SERVICE_RUNNING ))
|
|
break;
|
|
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_CONTINUE_PENDING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = 0;
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = AFP_SERVICE_CONTINUE_TIME;
|
|
|
|
AfpAnnounceServiceStatus();
|
|
|
|
// Tell the kernel-mode that we want to continue.
|
|
//
|
|
AfpRequestPkt.dwRequestCode = OP_SERVICE_CONTINUE;
|
|
AfpRequestPkt.dwApiType = AFP_API_TYPE_COMMAND;
|
|
|
|
dwRetCode = AfpServerIOCtrl( &AfpRequestPkt );
|
|
|
|
AFP_ASSERT( dwRetCode == NO_ERROR );
|
|
|
|
AfpGlobals.ServiceStatus.dwCheckPoint = 0;
|
|
AfpGlobals.ServiceStatus.dwWaitHint = 0;
|
|
AfpGlobals.ServiceStatus.dwCurrentState = SERVICE_RUNNING;
|
|
AfpGlobals.ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP |
|
|
SERVICE_ACCEPT_PAUSE_CONTINUE;
|
|
|
|
break;
|
|
|
|
case SERVICE_CONTROL_INTERROGATE:
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
|
|
}
|
|
|
|
AfpAnnounceServiceStatus();
|
|
}
|
|
|
|
//**
|
|
//
|
|
// Call: AfpAnnounceServiceStatus
|
|
//
|
|
// Returns: none
|
|
//
|
|
// Description: Will simly call SetServiceStatus to inform the service
|
|
// control manager of this service's current status.
|
|
//
|
|
VOID
|
|
AfpAnnounceServiceStatus( VOID )
|
|
{
|
|
BOOL dwRetCode;
|
|
|
|
|
|
dwRetCode = SetServiceStatus( AfpGlobals.hServiceStatus,
|
|
&(AfpGlobals.ServiceStatus) );
|
|
|
|
AFP_ASSERT( dwRetCode == TRUE );
|
|
|
|
}
|