107 lines
2.7 KiB
C++
107 lines
2.7 KiB
C++
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 2000.
|
|
//
|
|
// File: W S D P S V C . C P P
|
|
//
|
|
// Contents: Start/stop Winsock Direct Path Service.
|
|
//
|
|
// Notes: The service is actually implemented in MS TCP Winsock provider
|
|
//
|
|
// Author: VadimE 24 Jan 2000
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include "pch.h"
|
|
#pragma hdrstop
|
|
#include "wsdpsvc.h"
|
|
|
|
#define MSTCP_PROVIDER_DLL TEXT("mswsock.dll")
|
|
#define START_WSDP_FUNCTION_NAME "StartWsdpService"
|
|
#define STOP_WSDP_FUNCTION_NAME "StopWsdpService"
|
|
|
|
// MS TCP Winsock provider module handle
|
|
HINSTANCE ghMsTcpDll;
|
|
|
|
// Service start function pointer
|
|
typedef INT (WINAPI *PFN_START_WSDP_SVC) (VOID);
|
|
PFN_START_WSDP_SVC gpfnStartWsdpSvc;
|
|
|
|
// Service stop function pointer
|
|
typedef VOID (WINAPI *PFN_STOP_WSDP_SVC) (VOID);
|
|
PFN_STOP_WSDP_SVC gpfnStopWsdpSvc;
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// StartWsdpService - start WSDP service if running on DTC
|
|
//
|
|
//
|
|
VOID
|
|
StartWsdpService (
|
|
VOID
|
|
)
|
|
{
|
|
NTSTATUS status;
|
|
NT_PRODUCT_TYPE product;
|
|
|
|
//
|
|
// First check if we are running Server build
|
|
//
|
|
status = RtlGetNtProductType (&product);
|
|
if (!NT_SUCCESS (status) ||
|
|
(product == NtProductWinNt)) {
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Load MS TCP provider and get WSDP service entry points
|
|
//
|
|
ghMsTcpDll = LoadLibrary (MSTCP_PROVIDER_DLL);
|
|
if (ghMsTcpDll!=NULL) {
|
|
gpfnStartWsdpSvc = (PFN_START_WSDP_SVC) GetProcAddress (
|
|
ghMsTcpDll,
|
|
START_WSDP_FUNCTION_NAME);
|
|
gpfnStopWsdpSvc = (PFN_STOP_WSDP_SVC) GetProcAddress (
|
|
ghMsTcpDll,
|
|
STOP_WSDP_FUNCTION_NAME);
|
|
if (gpfnStartWsdpSvc != NULL && gpfnStopWsdpSvc != NULL) {
|
|
//
|
|
// Launch the service and return if succeded
|
|
//
|
|
INT err = (*gpfnStartWsdpSvc)();
|
|
if (err==0) {
|
|
return;
|
|
}
|
|
}
|
|
//
|
|
// Cleanup if anything fails
|
|
//
|
|
FreeLibrary (ghMsTcpDll);
|
|
ghMsTcpDll = NULL;
|
|
}
|
|
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// StopWsdpService - stop WSDP service if it was started
|
|
//
|
|
//
|
|
VOID
|
|
StopWsdpService (
|
|
VOID
|
|
)
|
|
{
|
|
if (ghMsTcpDll!=NULL) {
|
|
//
|
|
// Tell the service to stop and unload the provider
|
|
//
|
|
(*gpfnStopWsdpSvc)();
|
|
FreeLibrary (ghMsTcpDll);
|
|
ghMsTcpDll = NULL;
|
|
}
|
|
}
|
|
|
|
|
|
|