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.
1514 lines
42 KiB
1514 lines
42 KiB
/*++
|
|
|
|
Copyright (c) 1991-1997 Microsoft Corporation
|
|
Module Name: //KERNEL/RAZZLE3/src/sockets/tcpcmd/icmp/icmp.c
|
|
Abstract: Definitions of the ICMP Echo request API.
|
|
Author: Mike Massa (mikemas) Dec 30, 1993
|
|
Revision History:
|
|
Who When What
|
|
-------- -------- ----------------------------------------------
|
|
mikemas 12-30-93 created
|
|
RameshV 20-Jul-97 new async function IcmpSendEcho2
|
|
|
|
Notes:
|
|
In the functions do_echo_req/do_echo_rep the
|
|
precedence/tos bits are not used as defined RFC 1349.
|
|
-- MohsinA, 30-Jul-97
|
|
--*/
|
|
|
|
#include "inc.h"
|
|
#pragma hdrstop
|
|
|
|
#include <align.h>
|
|
#include <icmp.h>
|
|
#include <icmpapi.h>
|
|
#include <icmpif.h>
|
|
#include <winsock2.h>
|
|
#include <ws2tcpip.h>
|
|
#include <wscntl.h>
|
|
#include <ntddip6.h>
|
|
|
|
//
|
|
// Constants
|
|
//
|
|
#define PLATFORM_NT 0x0
|
|
#define PLATFORM_VXD 0x1
|
|
#define VXD_HANDLE_VALUE 0xDFFFFFFF
|
|
|
|
//
|
|
// Common Global variables
|
|
//
|
|
DWORD Platform = 0xFFFFFFFF;
|
|
|
|
// VxD external function pointers
|
|
//
|
|
LPWSCONTROL wsControl = NULL;
|
|
|
|
|
|
__inline void
|
|
CopyTDIFromSA6(TDI_ADDRESS_IP6 *To, SOCKADDR_IN6 *From)
|
|
{
|
|
memcpy(To, &From->sin6_port, sizeof *To);
|
|
}
|
|
|
|
__inline void
|
|
CopySAFromTDI6(SOCKADDR_IN6 *To, TDI_ADDRESS_IP6 *From)
|
|
{
|
|
To->sin6_family = AF_INET6;
|
|
memcpy(&To->sin6_port, From, sizeof *From);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Public functions
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
HANDLE
|
|
WINAPI
|
|
IcmpCreateFile(
|
|
VOID
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Opens a handle on which ICMP Echo Requests can be issued.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
An open file handle or INVALID_HANDLE_VALUE. Extended error information
|
|
is available by calling GetLastError().
|
|
|
|
Notes:
|
|
|
|
This function is effectively a no-op for the VxD platform.
|
|
|
|
--*/
|
|
|
|
{
|
|
HANDLE IcmpHandle = INVALID_HANDLE_VALUE;
|
|
|
|
|
|
if (Platform == PLATFORM_NT) {
|
|
OBJECT_ATTRIBUTES objectAttributes;
|
|
IO_STATUS_BLOCK ioStatusBlock;
|
|
UNICODE_STRING nameString;
|
|
NTSTATUS status;
|
|
|
|
//
|
|
// Open a Handle to the IP driver.
|
|
//
|
|
RtlInitUnicodeString(&nameString, DD_IP_DEVICE_NAME);
|
|
|
|
InitializeObjectAttributes(
|
|
&objectAttributes,
|
|
&nameString,
|
|
OBJ_CASE_INSENSITIVE,
|
|
(HANDLE) NULL,
|
|
(PSECURITY_DESCRIPTOR) NULL
|
|
);
|
|
|
|
status = NtCreateFile(
|
|
&IcmpHandle,
|
|
GENERIC_EXECUTE,
|
|
&objectAttributes,
|
|
&ioStatusBlock,
|
|
NULL,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
FILE_OPEN_IF,
|
|
0,
|
|
NULL,
|
|
0
|
|
);
|
|
|
|
if (!NT_SUCCESS(status)) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
IcmpHandle = INVALID_HANDLE_VALUE;
|
|
}
|
|
}
|
|
else {
|
|
IcmpHandle = LongToHandle(VXD_HANDLE_VALUE);
|
|
}
|
|
|
|
return(IcmpHandle);
|
|
|
|
} // IcmpCreateFile
|
|
|
|
HANDLE
|
|
WINAPI
|
|
Icmp6CreateFile(
|
|
VOID
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Opens a handle on which ICMPv6 Echo Requests can be issued.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
An open file handle or INVALID_HANDLE_VALUE. Extended error information
|
|
is available by calling GetLastError().
|
|
|
|
--*/
|
|
|
|
{
|
|
HANDLE IcmpHandle = INVALID_HANDLE_VALUE;
|
|
|
|
|
|
if (Platform == PLATFORM_NT) {
|
|
OBJECT_ATTRIBUTES objectAttributes;
|
|
IO_STATUS_BLOCK ioStatusBlock;
|
|
UNICODE_STRING nameString;
|
|
NTSTATUS status;
|
|
|
|
//
|
|
// Open a Handle to the IPv6 driver.
|
|
//
|
|
RtlInitUnicodeString(&nameString, DD_IPV6_DEVICE_NAME);
|
|
|
|
InitializeObjectAttributes(
|
|
&objectAttributes,
|
|
&nameString,
|
|
OBJ_CASE_INSENSITIVE,
|
|
(HANDLE) NULL,
|
|
(PSECURITY_DESCRIPTOR) NULL
|
|
);
|
|
|
|
status = NtCreateFile(
|
|
&IcmpHandle,
|
|
GENERIC_EXECUTE,
|
|
&objectAttributes,
|
|
&ioStatusBlock,
|
|
NULL,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
FILE_OPEN_IF,
|
|
0,
|
|
NULL,
|
|
0
|
|
);
|
|
|
|
if (!NT_SUCCESS(status)) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
IcmpHandle = INVALID_HANDLE_VALUE;
|
|
}
|
|
}
|
|
else {
|
|
SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
|
|
IcmpHandle = INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
return(IcmpHandle);
|
|
|
|
}
|
|
|
|
BOOL
|
|
WINAPI
|
|
IcmpCloseHandle(
|
|
HANDLE IcmpHandle
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Closes a handle opened by IcmpCreateFile.
|
|
|
|
Arguments:
|
|
|
|
IcmpHandle - The handle to close.
|
|
|
|
Return Value:
|
|
|
|
TRUE if the handle was closed successfully, otherwise FALSE. Extended
|
|
error information is available by calling GetLastError().
|
|
|
|
Notes:
|
|
|
|
This function is a no-op for the VxD platform.
|
|
|
|
--*/
|
|
|
|
{
|
|
if (Platform == PLATFORM_NT) {
|
|
NTSTATUS status;
|
|
|
|
|
|
status = NtClose(IcmpHandle);
|
|
|
|
if (!NT_SUCCESS(status)) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
return(FALSE);
|
|
}
|
|
}
|
|
|
|
return(TRUE);
|
|
|
|
} // IcmpCloseHandle
|
|
|
|
|
|
DWORD
|
|
IcmpParseReplies(
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Parses the reply buffer provided and returns the number of ICMP responses found.
|
|
|
|
Arguments:
|
|
|
|
ReplyBuffer - This must be the same buffer that was passed to IcmpSendEcho2
|
|
This is rewritten to hold an array of ICMP_ECHO_REPLY structures.
|
|
(i.e. the type is PICMP_ECHO_REPLY).
|
|
|
|
ReplySize - This must be the size of the above buffer.
|
|
|
|
Return Value:
|
|
Returns the number of ICMP responses found. If there is an errors, return value is
|
|
zero. The error can be determined by a call to GetLastError.
|
|
|
|
--*/
|
|
{
|
|
DWORD numberOfReplies = 0;
|
|
PICMP_ECHO_REPLY reply;
|
|
unsigned short i;
|
|
|
|
reply = ((PICMP_ECHO_REPLY) ReplyBuffer);
|
|
|
|
if( NULL == reply || 0 == ReplySize ) {
|
|
//
|
|
// Invalid parameter passed. But we ignore this and just return # of replies =0
|
|
//
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Convert new IP status IP_NEGOTIATING_IPSEC to IP_DEST_HOST_UNREACHABLE.
|
|
//
|
|
if (reply->Status == IP_NEGOTIATING_IPSEC) {
|
|
reply->Status = IP_DEST_HOST_UNREACHABLE;
|
|
}
|
|
|
|
//
|
|
// The reserved field of the first reply contains the number of replies.
|
|
//
|
|
numberOfReplies = reply->Reserved;
|
|
reply->Reserved = 0;
|
|
|
|
if (numberOfReplies == 0) {
|
|
//
|
|
// Internal IP error. The error code is in the first reply slot.
|
|
//
|
|
SetLastError(reply->Status);
|
|
}
|
|
else {
|
|
//
|
|
// Walk through the replies and convert the data offsets to user mode
|
|
// pointers.
|
|
//
|
|
|
|
for (i=0; i<numberOfReplies; i++, reply++) {
|
|
reply->Data = ((UCHAR *) reply) + ((ULONG_PTR) reply->Data);
|
|
reply->Options.OptionsData =
|
|
((UCHAR FAR *) reply) + ((ULONG_PTR) reply->Options.OptionsData);
|
|
}
|
|
}
|
|
|
|
return(numberOfReplies);
|
|
|
|
} // IcmpParseReplies
|
|
|
|
|
|
DWORD
|
|
IcmpParseReplies2(
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Parses the reply buffer provided and returns the number of ICMP responses found.
|
|
|
|
Arguments:
|
|
|
|
ReplyBuffer - This must be the same buffer that was passed to IcmpSendEcho2
|
|
This is rewritten to hold an array of ICMP_ECHO_REPLY structures.
|
|
(i.e. the type is PICMP_ECHO_REPLY).
|
|
|
|
ReplySize - This must be the size of the above buffer.
|
|
|
|
Return Value:
|
|
Returns the number of ICMP responses found. If there is an errors, return value is
|
|
zero. The error can be determined by a call to GetLastError.
|
|
|
|
--*/
|
|
{
|
|
DWORD numberOfReplies = 0;
|
|
PICMP_ECHO_REPLY reply;
|
|
unsigned short i;
|
|
|
|
reply = ((PICMP_ECHO_REPLY) ReplyBuffer);
|
|
|
|
if( NULL == reply || 0 == ReplySize ) {
|
|
//
|
|
// Invalid parameter passed. But we ignore this and just return # of replies =0
|
|
//
|
|
return 0;
|
|
}
|
|
|
|
|
|
//
|
|
// The reserved field of the first reply contains the number of replies.
|
|
//
|
|
numberOfReplies = reply->Reserved;
|
|
reply->Reserved = 0;
|
|
|
|
if (numberOfReplies == 0) {
|
|
//
|
|
// Internal IP error. The error code is in the first reply slot.
|
|
//
|
|
SetLastError(reply->Status);
|
|
}
|
|
else {
|
|
//
|
|
// Walk through the replies and convert the data offsets to user mode
|
|
// pointers.
|
|
//
|
|
|
|
for (i=0; i<numberOfReplies; i++, reply++) {
|
|
reply->Data = ((UCHAR *) reply) + ((ULONG_PTR) reply->Data);
|
|
reply->Options.OptionsData =
|
|
((UCHAR FAR *) reply) + ((ULONG_PTR) reply->Options.OptionsData);
|
|
}
|
|
}
|
|
|
|
return(numberOfReplies);
|
|
|
|
} // IcmpParseReplies
|
|
|
|
DWORD
|
|
WINAPI
|
|
IcmpSendEcho(
|
|
HANDLE IcmpHandle,
|
|
IPAddr DestinationAddress,
|
|
LPVOID RequestData,
|
|
WORD RequestSize,
|
|
PIP_OPTION_INFORMATION RequestOptions,
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize,
|
|
DWORD Timeout
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Sends an ICMP Echo request and returns one or more replies. The
|
|
call returns when the timeout has expired or the reply buffer
|
|
is filled.
|
|
|
|
Arguments:
|
|
|
|
IcmpHandle - An open handle returned by ICMPCreateFile.
|
|
|
|
DestinationAddress - The destination of the echo request.
|
|
|
|
RequestData - A buffer containing the data to send in the
|
|
request.
|
|
|
|
RequestSize - The number of bytes in the request data buffer.
|
|
|
|
RequestOptions - Pointer to the IP header options for the request.
|
|
May be NULL.
|
|
|
|
ReplyBuffer - A buffer to hold any replies to the request.
|
|
On return, the buffer will contain an array of
|
|
ICMP_ECHO_REPLY structures followed by options
|
|
and data. The buffer must be large enough to
|
|
hold at least one ICMP_ECHO_REPLY structure.
|
|
It should be large enough to also hold
|
|
8 more bytes of data - this is the size of
|
|
an ICMP error message.
|
|
|
|
ReplySize - The size in bytes of the reply buffer.
|
|
|
|
Timeout - The time in milliseconds to wait for replies.
|
|
|
|
Return Value:
|
|
|
|
Returns the number of replies received and stored in ReplyBuffer. If
|
|
the return value is zero, extended error information is available
|
|
via GetLastError().
|
|
|
|
--*/
|
|
|
|
{
|
|
PICMP_ECHO_REQUEST requestBuffer = NULL;
|
|
ULONG requestBufferSize;
|
|
DWORD numberOfReplies = 0;
|
|
|
|
if (ReplySize < sizeof(ICMP_ECHO_REPLY)) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
return(0);
|
|
}
|
|
|
|
requestBufferSize = sizeof(ICMP_ECHO_REQUEST) + RequestSize;
|
|
|
|
if (RequestOptions != NULL) {
|
|
requestBufferSize += RequestOptions->OptionsSize;
|
|
}
|
|
|
|
if (requestBufferSize < ReplySize) {
|
|
requestBufferSize = ReplySize;
|
|
}
|
|
|
|
requestBuffer = LocalAlloc(LMEM_FIXED, requestBufferSize);
|
|
|
|
if (requestBuffer == NULL) {
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return(0);
|
|
}
|
|
|
|
//
|
|
// Initialize the input buffer.
|
|
//
|
|
requestBuffer->Address = DestinationAddress;
|
|
requestBuffer->Timeout = Timeout;
|
|
requestBuffer->DataSize = RequestSize;
|
|
|
|
requestBuffer->OptionsOffset = sizeof(ICMP_ECHO_REQUEST);
|
|
|
|
if (RequestOptions != NULL) {
|
|
requestBuffer->OptionsValid = 1;
|
|
requestBuffer->Ttl = RequestOptions->Ttl;
|
|
requestBuffer->Tos = RequestOptions->Tos;
|
|
requestBuffer->Flags = RequestOptions->Flags;
|
|
requestBuffer->OptionsSize = RequestOptions->OptionsSize;
|
|
|
|
if (RequestOptions->OptionsSize > 0) {
|
|
|
|
CopyMemory(
|
|
((UCHAR *) requestBuffer) + requestBuffer->OptionsOffset,
|
|
RequestOptions->OptionsData,
|
|
RequestOptions->OptionsSize
|
|
);
|
|
}
|
|
}
|
|
else {
|
|
requestBuffer->OptionsValid = 0;
|
|
requestBuffer->OptionsSize = 0;
|
|
}
|
|
|
|
requestBuffer->DataOffset = requestBuffer->OptionsOffset +
|
|
requestBuffer->OptionsSize;
|
|
|
|
if (RequestSize > 0) {
|
|
|
|
CopyMemory(
|
|
((UCHAR *)requestBuffer) + requestBuffer->DataOffset,
|
|
RequestData,
|
|
RequestSize
|
|
);
|
|
}
|
|
|
|
if (Platform == PLATFORM_NT) {
|
|
IO_STATUS_BLOCK ioStatusBlock;
|
|
NTSTATUS status;
|
|
HANDLE eventHandle;
|
|
|
|
eventHandle = CreateEvent(
|
|
NULL, // default security
|
|
FALSE, // auto reset
|
|
FALSE, // initially non-signalled
|
|
NULL // unnamed
|
|
);
|
|
|
|
if (NULL == eventHandle) {
|
|
goto error_exit;
|
|
}
|
|
|
|
status = NtDeviceIoControlFile(
|
|
IcmpHandle, // Driver handle
|
|
eventHandle, // Event
|
|
NULL, // APC Routine
|
|
NULL, // APC context
|
|
&ioStatusBlock, // Status block
|
|
IOCTL_ICMP_ECHO_REQUEST, // Control code
|
|
requestBuffer, // Input buffer
|
|
requestBufferSize, // Input buffer size
|
|
ReplyBuffer, // Output buffer
|
|
ReplySize // Output buffer size
|
|
);
|
|
|
|
if (status == STATUS_PENDING) {
|
|
NtWaitForSingleObject(
|
|
eventHandle,
|
|
FALSE,
|
|
NULL);
|
|
status = ioStatusBlock.Status;
|
|
}
|
|
|
|
CloseHandle(eventHandle);
|
|
|
|
if (status != STATUS_SUCCESS) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
goto error_exit;
|
|
}
|
|
}
|
|
else {
|
|
//
|
|
// VxD Platform
|
|
//
|
|
DWORD status;
|
|
ULONG replyBufferSize = ReplySize;
|
|
|
|
status = (*wsControl)(
|
|
IPPROTO_TCP,
|
|
WSCNTL_TCPIP_ICMP_ECHO,
|
|
requestBuffer,
|
|
&requestBufferSize,
|
|
ReplyBuffer,
|
|
&replyBufferSize
|
|
);
|
|
|
|
if (status != NO_ERROR) {
|
|
SetLastError(status);
|
|
goto error_exit;
|
|
}
|
|
}
|
|
|
|
numberOfReplies = IcmpParseReplies(ReplyBuffer, ReplySize);
|
|
|
|
error_exit:
|
|
|
|
LocalFree(requestBuffer);
|
|
|
|
return(numberOfReplies);
|
|
|
|
} // IcmpSendEcho
|
|
|
|
|
|
DWORD
|
|
WINAPI
|
|
IcmpSendEcho2(
|
|
HANDLE IcmpHandle,
|
|
HANDLE Event,
|
|
PIO_APC_ROUTINE ApcRoutine,
|
|
PVOID ApcContext,
|
|
IPAddr DestinationAddress,
|
|
LPVOID RequestData,
|
|
WORD RequestSize,
|
|
PIP_OPTION_INFORMATION RequestOptions,
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize,
|
|
DWORD Timeout
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Sends an ICMP Echo request and the call returns either immediately
|
|
(if Event or ApcRoutine is NonNULL) or returns after the specified
|
|
timeout. The ReplyBuffer contains the ICMP responses, if any.
|
|
|
|
Arguments:
|
|
|
|
IcmpHandle - An open handle returned by ICMPCreateFile.
|
|
|
|
Event - This is the event to be signalled whenever an IcmpResponse
|
|
comes in.
|
|
|
|
ApcRoutine - This routine would be called when the calling thread
|
|
is in an alertable thread and an ICMP reply comes in.
|
|
|
|
ApcContext - This optional parameter is given to the ApcRoutine when
|
|
this call succeeds.
|
|
|
|
DestinationAddress - The destination of the echo request.
|
|
|
|
RequestData - A buffer containing the data to send in the
|
|
request.
|
|
|
|
RequestSize - The number of bytes in the request data buffer.
|
|
|
|
RequestOptions - Pointer to the IP header options for the request.
|
|
May be NULL.
|
|
|
|
ReplyBuffer - A buffer to hold any replies to the request.
|
|
On return, the buffer will contain an array of
|
|
ICMP_ECHO_REPLY structures followed by options
|
|
and data. The buffer must be large enough to
|
|
hold at least one ICMP_ECHO_REPLY structure.
|
|
It should be large enough to also hold
|
|
8 more bytes of data - this is the size of
|
|
an ICMP error message + this should also have
|
|
space for IO_STATUS_BLOCK which requires 8 or
|
|
16 bytes...
|
|
|
|
ReplySize - The size in bytes of the reply buffer.
|
|
|
|
Timeout - The time in milliseconds to wait for replies.
|
|
This is NOT used if ApcRoutine is not NULL or if Event
|
|
is not NULL.
|
|
|
|
Return Value:
|
|
|
|
Returns the number of replies received and stored in ReplyBuffer. If
|
|
the return value is zero, extended error information is available
|
|
via GetLastError().
|
|
|
|
Remarks:
|
|
|
|
On NT platforms,
|
|
If used Asynchronously (either ApcRoutine or Event is specified), then
|
|
ReplyBuffer and ReplySize are still needed. This is where the response
|
|
comes in.
|
|
ICMP Response data is copied to the ReplyBuffer provided, and the caller of
|
|
this function has to parse it asynchronously. The function IcmpParseReply
|
|
is provided for this purpose.
|
|
|
|
On non-NT platforms,
|
|
Event, ApcRoutine and ApcContext are IGNORED.
|
|
|
|
--*/
|
|
|
|
{
|
|
PICMP_ECHO_REQUEST requestBuffer = NULL;
|
|
ULONG requestBufferSize;
|
|
DWORD numberOfReplies = 0;
|
|
BOOL Asynchronous;
|
|
|
|
Asynchronous = (Platform == PLATFORM_NT && (Event || ApcRoutine));
|
|
|
|
if (ReplySize < sizeof(ICMP_ECHO_REPLY)) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
return(0);
|
|
}
|
|
|
|
requestBufferSize = sizeof(ICMP_ECHO_REQUEST) + RequestSize;
|
|
|
|
if (RequestOptions != NULL) {
|
|
requestBufferSize += RequestOptions->OptionsSize;
|
|
}
|
|
|
|
if (requestBufferSize < ReplySize) {
|
|
requestBufferSize = ReplySize;
|
|
}
|
|
|
|
requestBuffer = LocalAlloc(LMEM_FIXED, requestBufferSize);
|
|
|
|
if (requestBuffer == NULL) {
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return(0);
|
|
}
|
|
|
|
//
|
|
// Initialize the input buffer.
|
|
//
|
|
requestBuffer->Address = DestinationAddress;
|
|
requestBuffer->Timeout = Timeout;
|
|
requestBuffer->DataSize = RequestSize;
|
|
|
|
requestBuffer->OptionsOffset = sizeof(ICMP_ECHO_REQUEST);
|
|
|
|
if (RequestOptions != NULL) {
|
|
requestBuffer->OptionsValid = 1;
|
|
requestBuffer->Ttl = RequestOptions->Ttl;
|
|
requestBuffer->Tos = RequestOptions->Tos;
|
|
requestBuffer->Flags = RequestOptions->Flags;
|
|
requestBuffer->OptionsSize = RequestOptions->OptionsSize;
|
|
|
|
if (RequestOptions->OptionsSize > 0) {
|
|
|
|
CopyMemory(
|
|
((UCHAR *) requestBuffer) + requestBuffer->OptionsOffset,
|
|
RequestOptions->OptionsData,
|
|
RequestOptions->OptionsSize
|
|
);
|
|
}
|
|
}
|
|
else {
|
|
requestBuffer->OptionsValid = 0;
|
|
requestBuffer->OptionsSize = 0;
|
|
}
|
|
|
|
requestBuffer->DataOffset = requestBuffer->OptionsOffset +
|
|
requestBuffer->OptionsSize;
|
|
|
|
if (RequestSize > 0) {
|
|
|
|
CopyMemory(
|
|
((UCHAR *)requestBuffer) + requestBuffer->DataOffset,
|
|
RequestData,
|
|
RequestSize
|
|
);
|
|
}
|
|
|
|
if (Platform == PLATFORM_NT) {
|
|
IO_STATUS_BLOCK *pioStatusBlock;
|
|
NTSTATUS status;
|
|
HANDLE eventHandle;
|
|
|
|
//
|
|
// allocate status block on the reply buffer..
|
|
//
|
|
|
|
pioStatusBlock = (IO_STATUS_BLOCK*)((LPBYTE)ReplyBuffer + ReplySize);
|
|
pioStatusBlock --;
|
|
pioStatusBlock = ROUND_DOWN_POINTER(pioStatusBlock, ALIGN_WORST);
|
|
ReplySize = (ULONG)(((LPBYTE)pioStatusBlock) - (LPBYTE)ReplyBuffer );
|
|
if( (PVOID)pioStatusBlock < ReplyBuffer
|
|
|| ReplySize < sizeof(ICMP_ECHO_REPLY) ) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
goto error_exit;
|
|
}
|
|
|
|
if(!Asynchronous) { // Normal synchronous.
|
|
eventHandle = CreateEvent(
|
|
NULL, // default security
|
|
FALSE, // auto reset
|
|
FALSE, // initially non-signalled
|
|
NULL // unnamed
|
|
);
|
|
|
|
if (NULL == eventHandle) {
|
|
goto error_exit;
|
|
}
|
|
} else { // Asynchronous call.
|
|
eventHandle = Event; // Use specified Event.
|
|
}
|
|
|
|
status = NtDeviceIoControlFile(
|
|
IcmpHandle, // Driver handle
|
|
eventHandle, // Event
|
|
ApcRoutine, // APC Routine
|
|
ApcContext, // APC context
|
|
pioStatusBlock, // Status block
|
|
IOCTL_ICMP_ECHO_REQUEST, // Control code
|
|
requestBuffer, // Input buffer
|
|
requestBufferSize, // Input buffer size
|
|
ReplyBuffer, // Output buffer
|
|
ReplySize // Output buffer size
|
|
);
|
|
|
|
if (Asynchronous) {
|
|
// Asynchronous calls. We cannot give any information.
|
|
// We let the user do the other work.
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
goto error_exit;
|
|
}
|
|
|
|
if (status == STATUS_PENDING) {
|
|
NtWaitForSingleObject(
|
|
eventHandle,
|
|
FALSE,
|
|
NULL);
|
|
status = pioStatusBlock->Status;
|
|
|
|
}
|
|
|
|
CloseHandle(eventHandle);
|
|
|
|
if (status != STATUS_SUCCESS) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
goto error_exit;
|
|
}
|
|
}
|
|
else {
|
|
//
|
|
// VxD Platform
|
|
//
|
|
DWORD status;
|
|
ULONG replyBufferSize = ReplySize;
|
|
|
|
status = (*wsControl)(
|
|
IPPROTO_TCP,
|
|
WSCNTL_TCPIP_ICMP_ECHO,
|
|
requestBuffer,
|
|
&requestBufferSize,
|
|
ReplyBuffer,
|
|
&replyBufferSize
|
|
);
|
|
|
|
if (status != NO_ERROR) {
|
|
SetLastError(status);
|
|
goto error_exit;
|
|
}
|
|
}
|
|
|
|
numberOfReplies = IcmpParseReplies2(ReplyBuffer, ReplySize);
|
|
|
|
error_exit:
|
|
|
|
LocalFree(requestBuffer);
|
|
|
|
return(numberOfReplies);
|
|
|
|
} // IcmpSendEcho2
|
|
|
|
DWORD
|
|
Icmp6ParseReplies(
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Parses the reply buffer provided and returns the number of ICMPv6 responses
|
|
found.
|
|
|
|
Arguments:
|
|
|
|
ReplyBuffer - This must be the same buffer that was passed to
|
|
Icmp6SendEcho2. This is written to hold an array of
|
|
ICMPV6_ECHO_REPLY structures (i.e., the type is
|
|
PICMPV6_ECHO_REPLY).
|
|
|
|
ReplySize - This must be the size of the above buffer.
|
|
|
|
Return Value:
|
|
Returns the number of ICMPv6 responses found. If there is an error,
|
|
return value is zero. The error can be determined by a call to
|
|
GetLastError.
|
|
|
|
--*/
|
|
|
|
{
|
|
PICMPV6_ECHO_REPLY reply;
|
|
|
|
reply = ((PICMPV6_ECHO_REPLY) ReplyBuffer);
|
|
|
|
if( NULL == reply || 0 == ReplySize ) {
|
|
//
|
|
// Invalid parameter passed. But we ignore this and just return # of
|
|
// replies =0
|
|
//
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Convert new IP status IP_NEGOTIATING_IPSEC to IP_DEST_HOST_UNREACHABLE.
|
|
//
|
|
if (reply->Status == IP_NEGOTIATING_IPSEC) {
|
|
reply->Status = IP_DEST_HOST_UNREACHABLE;
|
|
}
|
|
|
|
if ((reply->Status == IP_SUCCESS) || (reply->Status == IP_TTL_EXPIRED_TRANSIT)) {
|
|
return 1;
|
|
} else {
|
|
//
|
|
// Internal IP error. The error code is in the first reply slot.
|
|
//
|
|
SetLastError(reply->Status);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
DWORD
|
|
WINAPI
|
|
Icmp6SendEcho2(
|
|
HANDLE IcmpHandle,
|
|
HANDLE Event,
|
|
PIO_APC_ROUTINE ApcRoutine,
|
|
PVOID ApcContext,
|
|
LPSOCKADDR_IN6 SourceAddress,
|
|
LPSOCKADDR_IN6 DestinationAddress,
|
|
LPVOID RequestData,
|
|
WORD RequestSize,
|
|
PIP_OPTION_INFORMATION RequestOptions,
|
|
LPVOID ReplyBuffer,
|
|
DWORD ReplySize,
|
|
DWORD Timeout
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Sends an ICMPv6 Echo request and the call returns either immediately
|
|
(if Event or ApcRoutine is NonNULL) or returns after the specified
|
|
timeout. The ReplyBuffer contains the ICMPv6 responses, if any.
|
|
|
|
Arguments:
|
|
|
|
IcmpHandle - An open handle returned by ICMP6CreateFile.
|
|
|
|
Event - This is the event to be signalled whenever an
|
|
IcmpResponse comes in.
|
|
|
|
ApcRoutine - This routine would be called when the calling thread
|
|
is in an alertable thread and an ICMPv6 reply comes
|
|
in.
|
|
|
|
ApcContext - This optional parameter is given to the ApcRoutine
|
|
when this call succeeds.
|
|
|
|
DestinationAddress - The destination of the echo request.
|
|
|
|
RequestData - A buffer containing the data to send in the
|
|
request.
|
|
|
|
RequestSize - The number of bytes in the request data buffer.
|
|
|
|
RequestOptions - Pointer to the IPv6 header options for the request.
|
|
May be NULL.
|
|
|
|
ReplyBuffer - A buffer to hold any replies to the request.
|
|
On return, the buffer will contain an array of
|
|
ICMPV6_ECHO_REPLY structures followed by options
|
|
and data. The buffer must be large enough to
|
|
hold at least one ICMPV6_ECHO_REPLY structure.
|
|
It should be large enough to also hold
|
|
8 more bytes of data - this is the size of
|
|
an ICMPv6 error message + this should also have
|
|
space for IO_STATUS_BLOCK which requires 8 or
|
|
16 bytes...
|
|
|
|
ReplySize - The size in bytes of the reply buffer.
|
|
|
|
Timeout - The time in milliseconds to wait for replies.
|
|
This is NOT used if ApcRoutine is not NULL or if
|
|
Event is not NULL.
|
|
|
|
Return Value:
|
|
|
|
Returns the number of replies received and stored in ReplyBuffer. If
|
|
the return value is zero, extended error information is available
|
|
via GetLastError().
|
|
|
|
Remarks:
|
|
|
|
If used Asynchronously (either ApcRoutine or Event is specified), then
|
|
ReplyBuffer and ReplySize are still needed. This is where the response
|
|
comes in.
|
|
ICMP Response data is copied to the ReplyBuffer provided, and the caller of
|
|
this function has to parse it asynchronously. The function Icmp6ParseReply
|
|
is provided for this purpose.
|
|
|
|
--*/
|
|
|
|
{
|
|
PICMPV6_ECHO_REQUEST requestBuffer = NULL;
|
|
ULONG requestBufferSize;
|
|
DWORD numberOfReplies = 0;
|
|
BOOL Asynchronous;
|
|
IO_STATUS_BLOCK *pioStatusBlock;
|
|
NTSTATUS status;
|
|
HANDLE eventHandle;
|
|
|
|
Asynchronous = (Platform == PLATFORM_NT && (Event || ApcRoutine));
|
|
|
|
if (ReplySize < sizeof(ICMPV6_ECHO_REPLY)) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
return(0);
|
|
}
|
|
|
|
requestBufferSize = sizeof(ICMPV6_ECHO_REQUEST) + RequestSize;
|
|
|
|
requestBuffer = LocalAlloc(LMEM_FIXED, requestBufferSize);
|
|
|
|
if (requestBuffer == NULL) {
|
|
SetLastError(ERROR_NOT_ENOUGH_MEMORY);
|
|
return(0);
|
|
}
|
|
|
|
if (Platform != PLATFORM_NT) {
|
|
SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
|
|
goto error_exit;
|
|
}
|
|
|
|
//
|
|
// Initialize the input buffer.
|
|
//
|
|
CopyTDIFromSA6(&requestBuffer->DstAddress, DestinationAddress);
|
|
CopyTDIFromSA6(&requestBuffer->SrcAddress, SourceAddress);
|
|
requestBuffer->Timeout = Timeout;
|
|
requestBuffer->TTL = RequestOptions->Ttl;
|
|
requestBuffer->Flags = RequestOptions->Flags;
|
|
|
|
if (RequestSize > 0) {
|
|
|
|
CopyMemory(
|
|
(UCHAR *)(requestBuffer + 1),
|
|
RequestData,
|
|
RequestSize
|
|
);
|
|
}
|
|
|
|
//
|
|
// allocate status block on the reply buffer..
|
|
//
|
|
|
|
pioStatusBlock = (IO_STATUS_BLOCK*)((LPBYTE)ReplyBuffer + ReplySize);
|
|
pioStatusBlock --;
|
|
pioStatusBlock = ROUND_DOWN_POINTER(pioStatusBlock, ALIGN_WORST);
|
|
ReplySize = (ULONG)(((LPBYTE)pioStatusBlock) - (LPBYTE)ReplyBuffer );
|
|
if( (PVOID)pioStatusBlock < ReplyBuffer
|
|
|| ReplySize < sizeof(ICMPV6_ECHO_REPLY) ) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
goto error_exit;
|
|
}
|
|
|
|
if(!Asynchronous) { // Normal synchronous.
|
|
eventHandle = CreateEvent(
|
|
NULL, // default security
|
|
FALSE, // auto reset
|
|
FALSE, // initially non-signalled
|
|
NULL // unnamed
|
|
);
|
|
|
|
if (NULL == eventHandle) {
|
|
goto error_exit;
|
|
}
|
|
} else { // Asynchronous call.
|
|
eventHandle = Event; // Use specified Event.
|
|
}
|
|
|
|
status = NtDeviceIoControlFile(
|
|
IcmpHandle, // Driver handle
|
|
eventHandle, // Event
|
|
ApcRoutine, // APC Routine
|
|
ApcContext, // APC context
|
|
pioStatusBlock, // Status block
|
|
IOCTL_ICMPV6_ECHO_REQUEST, // Control code
|
|
requestBuffer, // Input buffer
|
|
requestBufferSize, // Input buffer size
|
|
ReplyBuffer, // Output buffer
|
|
ReplySize // Output buffer size
|
|
);
|
|
|
|
if (Asynchronous) {
|
|
// Asynchronous calls. We cannot give any information.
|
|
// We let the user do the other work.
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
goto error_exit;
|
|
}
|
|
|
|
if (status == STATUS_PENDING) {
|
|
NtWaitForSingleObject(
|
|
eventHandle,
|
|
FALSE,
|
|
NULL);
|
|
status = pioStatusBlock->Status;
|
|
|
|
}
|
|
|
|
CloseHandle(eventHandle);
|
|
|
|
if (status != STATUS_SUCCESS) {
|
|
SetLastError(RtlNtStatusToDosError(status));
|
|
goto error_exit;
|
|
}
|
|
|
|
numberOfReplies = Icmp6ParseReplies(ReplyBuffer, ReplySize);
|
|
|
|
error_exit:
|
|
|
|
LocalFree(requestBuffer);
|
|
|
|
return(numberOfReplies);
|
|
|
|
}
|
|
|
|
//
|
|
// Constants
|
|
//
|
|
|
|
#define PING_WAIT 1000
|
|
#define DEFAULT_TTL 32
|
|
|
|
//
|
|
// Local type definitions
|
|
//
|
|
typedef struct icmp_local_storage {
|
|
struct icmp_local_storage *Next;
|
|
HANDLE IcmpHandle;
|
|
LPVOID ReplyBuffer;
|
|
DWORD NumberOfReplies;
|
|
DWORD Status;
|
|
} ICMP_LOCAL_STORAGE, *PICMP_LOCAL_STORAGE;
|
|
|
|
typedef struct status_table {
|
|
IP_STATUS NewStatus;
|
|
int OldStatus;
|
|
} STATUS_TABLE, *PSTATUS_TABLE;
|
|
|
|
|
|
//
|
|
// Global variables
|
|
//
|
|
CRITICAL_SECTION g_IcmpLock;
|
|
PICMP_LOCAL_STORAGE RequestHead = NULL;
|
|
STATUS_TABLE StatusTable[] = {
|
|
{ IP_SUCCESS, ECHO_REPLY },
|
|
{ IP_DEST_NET_UNREACHABLE, DEST_UNR },
|
|
{ IP_DEST_HOST_UNREACHABLE, DEST_UNR },
|
|
{ IP_NEGOTIATING_IPSEC, DEST_UNR },
|
|
{ IP_DEST_PROT_UNREACHABLE, DEST_UNR },
|
|
{ IP_TTL_EXPIRED_TRANSIT, TIME_EXCEEDED },
|
|
{ IP_TTL_EXPIRED_REASSEM, TIME_EXCEEDED },
|
|
{ IP_PARAM_PROBLEM, PARAMETER_ERROR },
|
|
{ IP_BAD_ROUTE, PARAMETER_ERROR },
|
|
{ IP_BAD_OPTION, PARAMETER_ERROR },
|
|
{ IP_BUF_TOO_SMALL, PARAMETER_ERROR },
|
|
{ IP_PACKET_TOO_BIG, PARAMETER_ERROR },
|
|
{ IP_BAD_DESTINATION, PARAMETER_ERROR },
|
|
{ IP_GENERAL_FAILURE, POLL_FAILED }
|
|
};
|
|
|
|
HANDLE
|
|
STRMAPI
|
|
register_icmp(
|
|
void
|
|
)
|
|
{
|
|
HANDLE icmpHandle;
|
|
|
|
icmpHandle = IcmpCreateFile();
|
|
|
|
if (icmpHandle == INVALID_HANDLE_VALUE) {
|
|
SetLastError(ICMP_OPEN_ERROR);
|
|
return(ICMP_ERROR);
|
|
}
|
|
|
|
return(icmpHandle);
|
|
|
|
} // register_icmp
|
|
|
|
|
|
int
|
|
STRMAPI
|
|
do_echo_req(
|
|
HANDLE fd,
|
|
long addr,
|
|
char *data,
|
|
int amount,
|
|
char *optptr,
|
|
int optlen,
|
|
int df,
|
|
int ttl,
|
|
int tos,
|
|
int precedence
|
|
)
|
|
{
|
|
PICMP_LOCAL_STORAGE localStorage;
|
|
DWORD replySize;
|
|
IP_OPTION_INFORMATION options;
|
|
LPVOID replyBuffer;
|
|
|
|
|
|
replySize = sizeof(ICMP_ECHO_REPLY) + amount + optlen;
|
|
|
|
//
|
|
// Allocate a buffer to hold the reply.
|
|
//
|
|
localStorage = (PICMP_LOCAL_STORAGE) LocalAlloc(
|
|
LMEM_FIXED,
|
|
replySize +
|
|
sizeof(ICMP_LOCAL_STORAGE)
|
|
);
|
|
|
|
if (localStorage == NULL) {
|
|
return((int)GetLastError());
|
|
}
|
|
|
|
replyBuffer = ((char *) localStorage) + sizeof(ICMP_LOCAL_STORAGE);
|
|
|
|
if (ttl == 0) {
|
|
options.Ttl = DEFAULT_TTL;
|
|
}
|
|
else {
|
|
options.Ttl = (BYTE)ttl;
|
|
}
|
|
|
|
options.Tos = (UCHAR)((tos << 4) | precedence);
|
|
options.Flags = df ? IP_FLAG_DF : 0;
|
|
options.OptionsSize = (BYTE)optlen;
|
|
options.OptionsData = (PUCHAR)optptr;
|
|
|
|
localStorage->NumberOfReplies = IcmpSendEcho(
|
|
fd,
|
|
(IPAddr) addr,
|
|
data,
|
|
(WORD)amount,
|
|
&options,
|
|
replyBuffer,
|
|
replySize,
|
|
PING_WAIT
|
|
);
|
|
|
|
if (localStorage->NumberOfReplies == 0) {
|
|
localStorage->Status = GetLastError();
|
|
}
|
|
else {
|
|
localStorage->Status = IP_SUCCESS;
|
|
}
|
|
|
|
localStorage->IcmpHandle = fd;
|
|
localStorage->ReplyBuffer = replyBuffer;
|
|
|
|
//
|
|
// Save the reply for later retrieval.
|
|
//
|
|
EnterCriticalSection(&g_IcmpLock);
|
|
localStorage->Next = RequestHead;
|
|
RequestHead = localStorage;
|
|
LeaveCriticalSection(&g_IcmpLock);
|
|
|
|
return(0);
|
|
|
|
} // do_echo_req
|
|
|
|
|
|
int
|
|
STRMAPI
|
|
do_echo_rep(
|
|
HANDLE fd,
|
|
char *data,
|
|
int amount,
|
|
int *rettype,
|
|
int *retttl,
|
|
int *rettos,
|
|
int *retprec,
|
|
int *retdf,
|
|
char *ropt,
|
|
int *roptlen
|
|
)
|
|
{
|
|
PICMP_LOCAL_STORAGE localStorage, tmp;
|
|
PICMP_ECHO_REPLY reply;
|
|
PSTATUS_TABLE entry;
|
|
DWORD status;
|
|
|
|
|
|
//
|
|
// Find the reply.
|
|
//
|
|
EnterCriticalSection(&g_IcmpLock);
|
|
|
|
for ( localStorage = RequestHead, tmp = NULL;
|
|
localStorage != NULL;
|
|
localStorage = localStorage->Next
|
|
) {
|
|
if (localStorage->IcmpHandle == fd) {
|
|
if (RequestHead == localStorage) {
|
|
RequestHead = localStorage->Next;
|
|
}
|
|
else {
|
|
tmp->Next = localStorage->Next;
|
|
}
|
|
break;
|
|
}
|
|
tmp = localStorage;
|
|
}
|
|
|
|
LeaveCriticalSection(&g_IcmpLock);
|
|
|
|
if (localStorage == NULL) {
|
|
SetLastError(POLL_FAILED);
|
|
return(-1);
|
|
}
|
|
|
|
//
|
|
// Process the reply.
|
|
//
|
|
if (localStorage->NumberOfReplies == 0) {
|
|
status = localStorage->Status;
|
|
reply = NULL;
|
|
}
|
|
else {
|
|
reply = (PICMP_ECHO_REPLY) localStorage->ReplyBuffer;
|
|
status = reply->Status;
|
|
}
|
|
|
|
if ((status == IP_SUCCESS) && (reply != NULL)) {
|
|
if (amount < reply->DataSize) {
|
|
status = POLL_FAILED;
|
|
goto der_error_exit;
|
|
}
|
|
|
|
CopyMemory(data, reply->Data, reply->DataSize);
|
|
|
|
*rettype = ECHO_REPLY;
|
|
}
|
|
else {
|
|
//
|
|
// Map to the appropriate old status code & return value.
|
|
//
|
|
if (status < IP_STATUS_BASE) {
|
|
status = POLL_FAILED;
|
|
goto der_error_exit;
|
|
}
|
|
|
|
if (status == IP_REQ_TIMED_OUT) {
|
|
status = POLL_TIMEOUT;
|
|
goto der_error_exit;
|
|
}
|
|
|
|
for ( entry = StatusTable;
|
|
entry->NewStatus != IP_GENERAL_FAILURE;
|
|
entry++
|
|
) {
|
|
if (entry->NewStatus == status) {
|
|
*rettype = entry->OldStatus;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (entry->NewStatus == IP_GENERAL_FAILURE) {
|
|
status = POLL_FAILED;
|
|
goto der_error_exit;
|
|
}
|
|
}
|
|
|
|
if (reply != NULL) {
|
|
*retdf = reply->Options.Flags ? 1 : 0;
|
|
*retttl = reply->Options.Ttl;
|
|
*rettos = (reply->Options.Tos & 0xf0) >> 4;
|
|
*retprec = reply->Options.Tos & 0x0f;
|
|
|
|
if (ropt) {
|
|
if (reply->Options.OptionsSize > *roptlen) {
|
|
reply->Options.OptionsSize = (BYTE)*roptlen;
|
|
}
|
|
|
|
*roptlen = reply->Options.OptionsSize;
|
|
|
|
if (reply->Options.OptionsSize) {
|
|
CopyMemory(
|
|
ropt,
|
|
reply->Options.OptionsData,
|
|
reply->Options.OptionsSize
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
LocalFree(localStorage);
|
|
return(0);
|
|
|
|
der_error_exit:
|
|
LocalFree(localStorage);
|
|
SetLastError(status);
|
|
return(-1);
|
|
|
|
} // do_echo_rep
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// DLL entry point
|
|
//
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
BOOL WINAPI
|
|
IcmpEntryPoint(
|
|
HANDLE hDll,
|
|
DWORD dwReason,
|
|
LPVOID lpReserved
|
|
)
|
|
{
|
|
OSVERSIONINFO versionInfo;
|
|
PICMP_LOCAL_STORAGE entry;
|
|
|
|
UNREFERENCED_PARAMETER(hDll);
|
|
UNREFERENCED_PARAMETER(lpReserved);
|
|
|
|
switch(dwReason) {
|
|
|
|
case DLL_PROCESS_ATTACH:
|
|
|
|
versionInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
|
|
|
|
if (!GetVersionEx(&versionInfo)) {
|
|
return(FALSE);
|
|
}
|
|
|
|
//
|
|
// NT 3.1 interface initialization
|
|
//
|
|
__try {
|
|
InitializeCriticalSection(&g_IcmpLock);
|
|
}
|
|
__except((GetExceptionCode() == STATUS_NO_MEMORY)
|
|
? EXCEPTION_EXECUTE_HANDLER : EXCEPTION_CONTINUE_SEARCH) {
|
|
return(FALSE);
|
|
}
|
|
|
|
if (versionInfo.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS) {
|
|
HINSTANCE WSock32Dll;
|
|
|
|
Platform = PLATFORM_VXD;
|
|
|
|
WSock32Dll = LoadLibrary("wsock32.dll");
|
|
|
|
if (WSock32Dll == NULL) {
|
|
return(FALSE);
|
|
}
|
|
|
|
wsControl = (LPWSCONTROL) GetProcAddress(
|
|
WSock32Dll,
|
|
"WsControl"
|
|
);
|
|
|
|
if (wsControl == NULL) {
|
|
return(FALSE);
|
|
}
|
|
}
|
|
else if (versionInfo.dwPlatformId == VER_PLATFORM_WIN32_NT) {
|
|
|
|
Platform = PLATFORM_NT;
|
|
|
|
}
|
|
else {
|
|
//
|
|
// Unsupported OS Version
|
|
//
|
|
return(FALSE);
|
|
}
|
|
|
|
break;
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
|
|
//
|
|
// NT 3.1 interface cleanup
|
|
//
|
|
DeleteCriticalSection(&g_IcmpLock);
|
|
|
|
while((entry = RequestHead) != NULL) {
|
|
RequestHead = RequestHead->Next;
|
|
LocalFree(entry);
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return(TRUE);
|
|
|
|
} // DllEntryPoint
|
|
|