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.
1267 lines
35 KiB
1267 lines
35 KiB
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c) 2001 Microsoft Corporation
|
|
//
|
|
// Module Name:
|
|
// tdisample.cpp
|
|
//
|
|
// Abstract:
|
|
// test executable--demonstrates the tdi client by calling it via the library
|
|
//
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "libbase.h"
|
|
|
|
const ULONG MAX_ADDRESS_SIZE = sizeof(TRANSPORT_ADDRESS) + TDI_ADDRESS_LENGTH_OSI_TSAP;
|
|
const USHORT DGRAM_SERVER_PORT = 0x5555;
|
|
const USHORT DGRAM_CLIENT_PORT = 0x4444;
|
|
const USHORT CONN_SERVER_PORT = 0x5656;
|
|
const USHORT CONN_CLIENT_PORT = 0x4545;
|
|
|
|
VOID
|
|
ServerTest(ULONG ulType,
|
|
ULONG ulNumDevices);
|
|
|
|
|
|
VOID
|
|
ClientTest(ULONG ulType,
|
|
ULONG ulNumDevices);
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: main
|
|
//
|
|
// Descript: parse the arguments to the program, initialize library and driver,
|
|
// then call client or server side of test
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
|
|
int __cdecl main(ULONG argc, TCHAR *argv[])
|
|
{
|
|
//
|
|
// first step: get the arguments for this run
|
|
//
|
|
BOOLEAN fIsServer = FALSE;
|
|
ULONG ulType = TDI_ADDRESS_TYPE_UNSPEC;
|
|
BOOLEAN fArgsOk = FALSE;
|
|
TCHAR *pArgStr;
|
|
|
|
if(argc > 1)
|
|
{
|
|
for(ULONG ulArg = 1; ulArg < argc; ulArg++)
|
|
{
|
|
pArgStr = argv[ulArg];
|
|
if ((*pArgStr == TEXT('/')) || (*pArgStr == TEXT('-')))
|
|
{
|
|
pArgStr++;
|
|
}
|
|
if (_tcsicmp(pArgStr, TEXT("server")) == 0)
|
|
{
|
|
if (fIsServer)
|
|
{
|
|
fArgsOk = FALSE;
|
|
break;
|
|
}
|
|
fIsServer = TRUE;
|
|
}
|
|
else
|
|
{
|
|
fArgsOk = FALSE;
|
|
if (ulType)
|
|
{
|
|
break;
|
|
}
|
|
if (_tcsicmp(pArgStr, TEXT("ipx")) == 0)
|
|
{
|
|
fArgsOk = TRUE;
|
|
ulType = TDI_ADDRESS_TYPE_IPX;
|
|
}
|
|
else if (_tcsicmp(pArgStr, TEXT("ipv4")) == 0)
|
|
{
|
|
fArgsOk = TRUE;
|
|
ulType = TDI_ADDRESS_TYPE_IP;
|
|
}
|
|
else if (_tcsicmp(pArgStr, TEXT("netbt")) == 0)
|
|
{
|
|
fArgsOk = TRUE;
|
|
ulType = TDI_ADDRESS_TYPE_NETBIOS;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if (!fArgsOk)
|
|
{
|
|
_putts(TEXT("Usage: tdisample [/server] [/ipx | /ipv4 | /netbt] \n"));
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// ready to go. Initialize the library, connect to the driver, etc
|
|
//
|
|
if (TdiLibInit())
|
|
{
|
|
//
|
|
// change this to limit debug output for kernel mode driver
|
|
// 0 = none, 1 = commands, 2 = handlers, 3 = both
|
|
//
|
|
DoDebugLevel(0x03);
|
|
ULONG ulNumDevices = DoGetNumDevices(ulType);
|
|
if (ulNumDevices)
|
|
{
|
|
if (fIsServer)
|
|
{
|
|
ServerTest(ulType, ulNumDevices);
|
|
}
|
|
else
|
|
{
|
|
ClientTest(ulType, ulNumDevices);
|
|
}
|
|
}
|
|
TdiLibClose();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// server-side test functions
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
ULONG_PTR
|
|
__cdecl
|
|
ServerThread(LPVOID pvDummy);
|
|
|
|
|
|
BOOLEAN WaitForClient(
|
|
ULONG TdiHandle,
|
|
PTRANSPORT_ADDRESS pRemoteAddr
|
|
);
|
|
|
|
CRITICAL_SECTION CriticalSection;
|
|
HANDLE hEvent;
|
|
ULONG ulServerCount;
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: IncServerCount
|
|
//
|
|
// Descript: multi-thread safe incrementing of server count
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
VOID IncServerCount()
|
|
{
|
|
EnterCriticalSection(&CriticalSection);
|
|
++ulServerCount;
|
|
LeaveCriticalSection(&CriticalSection);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: DecServerCount
|
|
//
|
|
// Descript: multi-thread safe decrementing of server count
|
|
// when last one done, sets event
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
VOID DecServerCount()
|
|
{
|
|
BOOLEAN fDone = FALSE;
|
|
|
|
EnterCriticalSection(&CriticalSection);
|
|
--ulServerCount;
|
|
if (!ulServerCount)
|
|
{
|
|
fDone = TRUE;
|
|
}
|
|
LeaveCriticalSection(&CriticalSection);
|
|
if (fDone)
|
|
{
|
|
SetEvent(hEvent);
|
|
}
|
|
}
|
|
|
|
struct THREAD_DATA
|
|
{
|
|
ULONG ulType;
|
|
ULONG ulSlot;
|
|
};
|
|
typedef THREAD_DATA *PTHREAD_DATA;
|
|
|
|
// ----------------------------------------------
|
|
//
|
|
// Function: ServerTest
|
|
//
|
|
// Arguments: ulType -- protocol type to use
|
|
// NumDevices -- number of devices of this protocol type
|
|
//
|
|
// Descript: this function controls the server side of the test
|
|
//
|
|
// ----------------------------------------------
|
|
|
|
|
|
VOID
|
|
ServerTest(ULONG ulType, ULONG ulNumDevices)
|
|
{
|
|
//
|
|
// initialize globals
|
|
//
|
|
try
|
|
{
|
|
InitializeCriticalSection(&CriticalSection);
|
|
}
|
|
catch(...)
|
|
{
|
|
return;
|
|
}
|
|
|
|
|
|
hEvent = CreateEvent(NULL,
|
|
TRUE, // manual reset
|
|
FALSE, // starts reset
|
|
NULL);
|
|
ulServerCount = 1; // a single bogus reference, so event doesn't fire prematurely
|
|
|
|
|
|
|
|
//
|
|
// go thru our list of nodes, starting a thread for each one
|
|
//
|
|
for(ULONG ulCount = 0; ulCount < ulNumDevices; ulCount++)
|
|
{
|
|
ULONG ulThreadId;
|
|
PTHREAD_DATA pThreadData = (PTHREAD_DATA)LocalAllocateMemory(sizeof(THREAD_DATA));
|
|
if (!pThreadData)
|
|
{
|
|
_putts(TEXT("ServerTest: unable to allocate memory for pThreadData\n"));
|
|
break;
|
|
}
|
|
|
|
pThreadData->ulType = ulType;
|
|
pThreadData->ulSlot = ulCount;
|
|
|
|
//
|
|
// reference for one starting now
|
|
//
|
|
IncServerCount();
|
|
|
|
HANDLE hThread = CreateThread((LPSECURITY_ATTRIBUTES)NULL,
|
|
0,
|
|
(LPTHREAD_START_ROUTINE)ServerThread,
|
|
(LPVOID)pThreadData,
|
|
0,
|
|
&ulThreadId);
|
|
|
|
if (!hThread)
|
|
{
|
|
_putts(TEXT("ServerTest: failed starting server thread\n"));
|
|
DecServerCount();
|
|
}
|
|
Sleep(100);
|
|
}
|
|
|
|
//
|
|
// get rid of bogus reference
|
|
//
|
|
DecServerCount();
|
|
|
|
//
|
|
// wait until all the threads have completed
|
|
//
|
|
WaitForSingleObject(hEvent, INFINITE);
|
|
|
|
//
|
|
// cleanup
|
|
//
|
|
CloseHandle(hEvent);
|
|
DeleteCriticalSection(&CriticalSection);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: ServerThread
|
|
//
|
|
// Arguments: pvData -- actually pThreadData for this server
|
|
//
|
|
// Descript: This is the thread that runs for each server instance
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
ULONG_PTR
|
|
__cdecl
|
|
ServerThread(LPVOID pvData)
|
|
{
|
|
PTHREAD_DATA pThreadData = (PTHREAD_DATA)pvData;
|
|
PTRANSPORT_ADDRESS pTransAddr = NULL;
|
|
PTRANSPORT_ADDRESS pRemoteAddr = NULL;
|
|
TCHAR *pDeviceName = NULL;
|
|
BOOLEAN fSuccessful = FALSE;
|
|
|
|
while (TRUE)
|
|
{
|
|
//
|
|
// stores local interface address (server)
|
|
//
|
|
pTransAddr = (PTRANSPORT_ADDRESS)LocalAllocateMemory(MAX_ADDRESS_SIZE);
|
|
if (!pTransAddr)
|
|
{
|
|
_putts(TEXT("ServerThread: unable to allocate memory for pTransAddr\n"));
|
|
break;
|
|
}
|
|
pTransAddr->TAAddressCount = 1;
|
|
|
|
//
|
|
// stores remote interface address (client)
|
|
//
|
|
pRemoteAddr = (PTRANSPORT_ADDRESS)LocalAllocateMemory(MAX_ADDRESS_SIZE);
|
|
if (!pRemoteAddr)
|
|
{
|
|
_putts(TEXT("ServerThread: unable to allocate memory for pRemoteAddr\n"));
|
|
break;
|
|
}
|
|
pRemoteAddr->TAAddressCount = 1;
|
|
|
|
//
|
|
// stores local interface name (server)
|
|
//
|
|
pDeviceName = (TCHAR *)LocalAllocateMemory(256 * sizeof(TCHAR));
|
|
if (!pDeviceName)
|
|
{
|
|
_putts(TEXT("ServerThread: unable to allocate memory for pDeviceName\n"));
|
|
break;
|
|
}
|
|
|
|
//
|
|
// get name of local device
|
|
//
|
|
if (DoGetDeviceName(pThreadData->ulType, pThreadData->ulSlot, pDeviceName) != STATUS_SUCCESS)
|
|
{
|
|
break;
|
|
}
|
|
|
|
TCHAR *pDataDeviceName = NULL;
|
|
TCHAR *pConnDeviceName = NULL;
|
|
|
|
//
|
|
// for netbios, each "address" has its own name. You open a device based mostly on the name
|
|
//
|
|
if (pThreadData->ulType == TDI_ADDRESS_TYPE_NETBIOS)
|
|
{
|
|
pDataDeviceName = pDeviceName;
|
|
pConnDeviceName = pDeviceName;
|
|
|
|
PTA_NETBIOS_ADDRESS pTaAddr = (PTA_NETBIOS_ADDRESS)pTransAddr;
|
|
|
|
pTaAddr->Address[0].AddressType = TDI_ADDRESS_TYPE_NETBIOS;
|
|
pTaAddr->Address[0].AddressLength = TDI_ADDRESS_LENGTH_NETBIOS;
|
|
pTaAddr->Address[0].Address[0].NetbiosNameType = TDI_ADDRESS_NETBIOS_TYPE_UNIQUE;
|
|
memcpy(pTaAddr->Address[0].Address[0].NetbiosName, "SERVER", 7); // NOTE: ascii string
|
|
}
|
|
|
|
//
|
|
// for others, there is one name for the datagram device and another for the "connected"
|
|
// device. You open an interface based largely on the address
|
|
//
|
|
else
|
|
{
|
|
if (DoGetAddress(pThreadData->ulType, pThreadData->ulSlot, pTransAddr) != STATUS_SUCCESS)
|
|
{
|
|
break;
|
|
}
|
|
|
|
switch (pThreadData->ulType)
|
|
{
|
|
case TDI_ADDRESS_TYPE_IPX:
|
|
{
|
|
PTA_IPX_ADDRESS pAddr = (PTA_IPX_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].Socket = DGRAM_SERVER_PORT;
|
|
|
|
pDataDeviceName = TEXT("\\device\\nwlnkipx");
|
|
pConnDeviceName = TEXT("\\device\\nwlnkspx");
|
|
break;
|
|
}
|
|
case TDI_ADDRESS_TYPE_IP:
|
|
{
|
|
PTA_IP_ADDRESS pAddr = (PTA_IP_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].sin_port = DGRAM_SERVER_PORT;
|
|
|
|
pDataDeviceName = TEXT("\\device\\udp");
|
|
pConnDeviceName = TEXT("\\device\\tcp");
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
_tprintf(TEXT("ServerThread: DeviceName: %s\n"), pDeviceName);
|
|
_putts(TEXT("Device Address:\n"));
|
|
DoPrintAddress(pTransAddr);
|
|
|
|
//
|
|
// wait for a client to contact us
|
|
//
|
|
ULONG TdiHandle = DoOpenAddress(pDataDeviceName, pTransAddr);
|
|
if (!TdiHandle)
|
|
{
|
|
_putts(TEXT("ServerThread: failed to open address object\n"));
|
|
break;
|
|
}
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_ERROR);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_RECEIVE_DATAGRAM);
|
|
|
|
if (!WaitForClient(TdiHandle, pRemoteAddr))
|
|
{
|
|
_putts(TEXT("ServerThread: Timed out waiting for client\n"));
|
|
DoCloseAddress(TdiHandle);
|
|
break;
|
|
}
|
|
_putts(TEXT("ServerThread: Found by client. Client address:\n"));
|
|
DoPrintAddress(pTransAddr);
|
|
|
|
//
|
|
// echo datagram packets until we get one that is TEXT("Last Packet"), or until we time out
|
|
//
|
|
for (ULONG ulCount = 1; ulCount < 60000; ulCount++)
|
|
{
|
|
ULONG ulNumBytes;
|
|
PUCHAR pucData;
|
|
|
|
Sleep(10);
|
|
ulNumBytes = DoReceiveDatagram(TdiHandle, NULL, pRemoteAddr, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, pucData, ulNumBytes);
|
|
|
|
TCHAR *pString = (TCHAR *)pucData;
|
|
_tprintf(TEXT("ServerThread: Packet Received: %s\n"), pString);
|
|
if (_tcscmp(pString, TEXT("Last Packet")))
|
|
{
|
|
ulCount = 0;
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
if (ulCount)
|
|
{
|
|
_putts(TEXT("ServerThread: Exitting datagram receive loop\n"));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
Sleep(50);
|
|
DoCloseAddress(TdiHandle);
|
|
|
|
//
|
|
// now, open an endpoint, and wait for a connection request
|
|
//
|
|
switch (pThreadData->ulType)
|
|
{
|
|
case TDI_ADDRESS_TYPE_IPX:
|
|
{
|
|
PTA_IPX_ADDRESS pAddr = (PTA_IPX_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].Socket = CONN_SERVER_PORT;
|
|
break;
|
|
}
|
|
case TDI_ADDRESS_TYPE_IP:
|
|
{
|
|
PTA_IP_ADDRESS pAddr = (PTA_IP_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].sin_port = CONN_SERVER_PORT;
|
|
break;
|
|
}
|
|
}
|
|
TdiHandle = DoOpenEndpoint(pConnDeviceName, pTransAddr);
|
|
if (!TdiHandle)
|
|
{
|
|
_putts(TEXT("ServerThread: unable to open endpoint\n"));
|
|
break;
|
|
}
|
|
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_CONNECT);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_DISCONNECT);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_ERROR);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_RECEIVE);
|
|
|
|
fSuccessful = FALSE;
|
|
for (ULONG ulCount = 0; ulCount < 100; ulCount++)
|
|
{
|
|
if (DoIsConnected(TdiHandle))
|
|
{
|
|
_putts(TEXT("ServerThread: connect successful\n"));
|
|
fSuccessful = TRUE;
|
|
break;
|
|
}
|
|
Sleep(20);
|
|
}
|
|
if (!fSuccessful)
|
|
{
|
|
_putts(TEXT("ServerThread: timed out waiting for connect\n"));
|
|
DoCloseEndpoint(TdiHandle);
|
|
break;
|
|
}
|
|
|
|
//
|
|
// echo packets until we get one that is TEXT("Last Packet")
|
|
//
|
|
|
|
for (ULONG ulCount = 0; ulCount < 60000; ulCount++)
|
|
{
|
|
ULONG ulNumBytes;
|
|
PUCHAR pucData;
|
|
|
|
Sleep(10);
|
|
ulNumBytes = DoReceive(TdiHandle, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
DoSend(TdiHandle, pucData, ulNumBytes, 0);
|
|
|
|
TCHAR *pString = (TCHAR *)pucData;
|
|
_tprintf(TEXT("ServerThread: Packet received: %s\n"), pString);
|
|
if (_tcscmp(pString, TEXT("Last Packet")))
|
|
{
|
|
ulCount = 0;
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
if (ulCount)
|
|
{
|
|
_putts(TEXT("ServerThread: Exitting connected receive loop\n"));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
for (ulCount = 0; ulCount < 1000; ulCount++)
|
|
{
|
|
if (!DoIsConnected(TdiHandle))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
DoCloseEndpoint(TdiHandle);
|
|
break;
|
|
}
|
|
|
|
//
|
|
// cleanup
|
|
//
|
|
if (pTransAddr)
|
|
{
|
|
LocalFreeMemory(pTransAddr);
|
|
}
|
|
if (pRemoteAddr)
|
|
{
|
|
LocalFreeMemory(pTransAddr);
|
|
}
|
|
if (pDeviceName)
|
|
{
|
|
LocalFreeMemory(pDeviceName);
|
|
}
|
|
LocalFreeMemory(pvData);
|
|
DecServerCount();
|
|
_putts(TEXT("ServerThread: exitting\n"));
|
|
|
|
return 0;
|
|
}
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: WaitForClient
|
|
//
|
|
// Arguments: TdiHandle -- address object handle for calling driver
|
|
// pRemoteAddr -- returns address received from
|
|
//
|
|
// Returns: TRUE if hear from client before timeout
|
|
//
|
|
// Descript: This function is used by the server side of the test to
|
|
// wait for contact with the client side.
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
BOOLEAN WaitForClient(ULONG TdiHandle,
|
|
PTRANSPORT_ADDRESS pRemoteAddr)
|
|
{
|
|
while(TRUE)
|
|
{
|
|
//
|
|
// wait for up to a 2 minutes for first packet (broadcast)
|
|
//
|
|
BOOLEAN fSuccessful = FALSE;
|
|
for (ULONG ulCount = 0; ulCount < 6000; ulCount++)
|
|
{
|
|
ULONG ulNumBytes;
|
|
PUCHAR pucData;
|
|
|
|
Sleep(20);
|
|
ulNumBytes = DoReceiveDatagram(TdiHandle, NULL, pRemoteAddr, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
if (ulNumBytes == 4)
|
|
{
|
|
PULONG pulValue = (PULONG)pucData;
|
|
if (*pulValue == 0x12345678)
|
|
{
|
|
_putts(TEXT("WaitForClient: first packet received\n"));
|
|
fSuccessful = TRUE;
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("WaitForClient: unexpected packet received\n"));
|
|
}
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
//
|
|
// break out of wait loop if successful
|
|
//
|
|
if (fSuccessful)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// check for timed out
|
|
//
|
|
if (!fSuccessful)
|
|
{
|
|
_putts(TEXT("WaitForClient: timed out waiting for first packet\n"));
|
|
break;
|
|
}
|
|
|
|
|
|
//
|
|
// send 1st response
|
|
//
|
|
ULONG ulBuffer = 0x98765432;
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)&ulBuffer, sizeof(ULONG));
|
|
|
|
//
|
|
// wait for second response (directed)
|
|
//
|
|
fSuccessful = FALSE;
|
|
for (ULONG ulCount = 0; ulCount < 1000; ulCount++)
|
|
{
|
|
ULONG ulNumBytes;
|
|
PUCHAR pucData;
|
|
|
|
Sleep(10);
|
|
ulNumBytes = DoReceiveDatagram(TdiHandle, NULL, NULL, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
if (ulNumBytes == 4)
|
|
{
|
|
PULONG pulValue = (PULONG)pucData;
|
|
if (*pulValue == 0x22222222)
|
|
{
|
|
_putts(TEXT("WaitForClient: Second packet received\n"));
|
|
fSuccessful = TRUE;
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("WaitForClient: unexpected packet received\n"));
|
|
}
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
//
|
|
// break out if recieved
|
|
//
|
|
if (fSuccessful)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// if received second packet, send second response
|
|
//
|
|
if (fSuccessful)
|
|
{
|
|
ulBuffer = 0x33333333;
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)&ulBuffer, sizeof(ULONG));
|
|
return TRUE;
|
|
}
|
|
//
|
|
// else reloop and wait again for broadcast
|
|
//
|
|
_putts(TEXT("WaitForClient: timed out waiting for second packet\n"));
|
|
}
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// client-side test functions
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
BOOLEAN
|
|
FindServer(TCHAR *pDataDeviceName,
|
|
PTRANSPORT_ADDRESS pTransAddr,
|
|
PTRANSPORT_ADDRESS pRemoteAddr);
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: ClientTest
|
|
//
|
|
// Arguments: ulType -- protocol type to use
|
|
// NumDevices -- number of devices of this protocol type
|
|
//
|
|
// Descript: this function controls the client side of the test
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
VOID
|
|
ClientTest(ULONG ulType, ULONG ulNumDevices)
|
|
{
|
|
//
|
|
// address of local interface
|
|
//
|
|
PTRANSPORT_ADDRESS pTransAddr = (PTRANSPORT_ADDRESS)LocalAllocateMemory(MAX_ADDRESS_SIZE);
|
|
if (!pTransAddr)
|
|
{
|
|
_putts(TEXT("ClientTest: unable to allocate memory for pTransAddr\n"));
|
|
return;
|
|
}
|
|
pTransAddr->TAAddressCount = 1;
|
|
|
|
//
|
|
// address of remote interface
|
|
//
|
|
PTRANSPORT_ADDRESS pRemoteAddr = (PTRANSPORT_ADDRESS)LocalAllocateMemory(MAX_ADDRESS_SIZE);
|
|
if (!pRemoteAddr)
|
|
{
|
|
_putts(TEXT("ClientTest: unable to allocate memory for pRemoteAddr\n"));
|
|
LocalFreeMemory(pTransAddr);
|
|
return;
|
|
}
|
|
pRemoteAddr->TAAddressCount = 1;
|
|
|
|
//
|
|
// name of device (from driver)
|
|
//
|
|
TCHAR *pDeviceName = (TCHAR *)LocalAllocateMemory(256 * sizeof(TCHAR));
|
|
if (!pDeviceName)
|
|
{
|
|
_putts(TEXT("ClientTest: unable to allocate memory for pDeviceNameAddr\n"));
|
|
LocalFreeMemory(pTransAddr);
|
|
LocalFreeMemory(pRemoteAddr);
|
|
return;
|
|
}
|
|
|
|
//
|
|
// name of tdi datagram interface to open
|
|
//
|
|
TCHAR *pDataDeviceName = NULL;
|
|
//
|
|
// name of tdi connection endpoint interface to open
|
|
//
|
|
TCHAR *pConnDeviceName = NULL;
|
|
//
|
|
// Stores handle used by driver to access interface
|
|
//
|
|
ULONG TdiHandle;
|
|
|
|
//
|
|
// for netbios, each "address" has its own name. You open a device based on the name
|
|
//
|
|
if (ulType == TDI_ADDRESS_TYPE_NETBIOS)
|
|
{
|
|
PTA_NETBIOS_ADDRESS pTaAddr = (PTA_NETBIOS_ADDRESS)pTransAddr;
|
|
|
|
pTaAddr->Address[0].AddressType = TDI_ADDRESS_TYPE_NETBIOS;
|
|
pTaAddr->Address[0].AddressLength = TDI_ADDRESS_LENGTH_NETBIOS;
|
|
pTaAddr->Address[0].Address[0].NetbiosNameType = TDI_ADDRESS_NETBIOS_TYPE_UNIQUE;
|
|
memcpy(pTaAddr->Address[0].Address[0].NetbiosName, "CLIENT", 7); // NOTE: ascii string
|
|
}
|
|
|
|
//
|
|
// for others, there is one name for the datagram device and another for the "connected"
|
|
// device. You open an interface based on the address
|
|
//
|
|
else
|
|
{
|
|
switch (ulType)
|
|
{
|
|
case TDI_ADDRESS_TYPE_IPX:
|
|
pDataDeviceName = TEXT("\\device\\nwlnkipx");
|
|
pConnDeviceName = TEXT("\\device\\nwlnkspx");
|
|
break;
|
|
case TDI_ADDRESS_TYPE_IP:
|
|
pDataDeviceName = TEXT("\\device\\udp");
|
|
pConnDeviceName = TEXT("\\device\\tcp");
|
|
break;
|
|
}
|
|
_putts(TEXT("ClientTest: get provider information\n"));
|
|
TdiHandle = DoOpenControl(pDataDeviceName);
|
|
|
|
if (TdiHandle)
|
|
{
|
|
PTDI_PROVIDER_INFO pInfo = (PTDI_PROVIDER_INFO)DoTdiQuery(TdiHandle, TDI_QUERY_PROVIDER_INFO);
|
|
if (pInfo)
|
|
{
|
|
DoPrintProviderInfo(pInfo);
|
|
LocalFreeMemory(pInfo);
|
|
}
|
|
DoCloseControl(TdiHandle);
|
|
}
|
|
}
|
|
|
|
//
|
|
// loop thru the available devices, trying each one in turn..
|
|
//
|
|
for (ULONG ulCount = 0; ulCount < ulNumDevices; ulCount++)
|
|
{
|
|
//
|
|
// collect necessary information
|
|
//
|
|
if (DoGetDeviceName(ulType, ulCount, pDeviceName) != STATUS_SUCCESS)
|
|
{
|
|
continue;
|
|
}
|
|
_tprintf(TEXT("ClientTest: LocalDeviceName = %s\n"), pDeviceName);
|
|
|
|
if (ulType == TDI_ADDRESS_TYPE_NETBIOS)
|
|
{
|
|
pDataDeviceName = pDeviceName;
|
|
pConnDeviceName = pDeviceName;
|
|
|
|
_putts(TEXT("ClientTest: get provider information\n"));
|
|
TdiHandle = DoOpenControl(pDataDeviceName);
|
|
if (TdiHandle)
|
|
{
|
|
PTDI_PROVIDER_INFO pInfo = (PTDI_PROVIDER_INFO)DoTdiQuery(TdiHandle, TDI_QUERY_PROVIDER_INFO);
|
|
if (pInfo)
|
|
{
|
|
DoPrintProviderInfo(pInfo);
|
|
LocalFreeMemory(pInfo);
|
|
}
|
|
DoCloseControl(TdiHandle);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (DoGetAddress(ulType, ulCount, pTransAddr) != STATUS_SUCCESS)
|
|
{
|
|
continue;
|
|
}
|
|
}
|
|
|
|
_putts(TEXT("ClientTest: Local device address:\n"));
|
|
DoPrintAddress(pTransAddr);
|
|
|
|
//
|
|
// try to contact server
|
|
//
|
|
if (FindServer(pDataDeviceName, pTransAddr, pRemoteAddr))
|
|
{
|
|
_putts(TEXT("Remote interface found:\n"));
|
|
DoPrintAddress(pRemoteAddr);
|
|
//
|
|
// do a datagram send/receive test
|
|
//
|
|
TdiHandle = DoOpenAddress(pDataDeviceName, pTransAddr);
|
|
if (TdiHandle)
|
|
{
|
|
_putts(TEXT("ClientTest: Sending first test packet\n"));
|
|
|
|
TCHAR *strBuffer = TEXT("This is only a test");
|
|
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_ERROR);
|
|
DoPostReceiveBuffer(TdiHandle, 128);
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)strBuffer, sizeof(TCHAR) * (1 + _tcslen(strBuffer)));
|
|
Sleep(300);
|
|
|
|
PUCHAR pucData;
|
|
ULONG ulNumBytes = DoFetchReceiveBuffer(TdiHandle, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
strBuffer = (TCHAR *)pucData;
|
|
_tprintf(TEXT("ClientTest: Response received: %s\n"), strBuffer);
|
|
LocalFreeMemory(pucData);
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("ClientTest: Response packet not received\n"));
|
|
}
|
|
|
|
_putts(TEXT("ClientTest: Sending second test packet\n"));
|
|
|
|
DoPostReceiveBuffer(TdiHandle, 128);
|
|
strBuffer = TEXT("Last Packet");
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)strBuffer, sizeof(TCHAR) * (1 + _tcslen(strBuffer)));
|
|
Sleep(300);
|
|
ulNumBytes = DoFetchReceiveBuffer(TdiHandle, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
strBuffer = (TCHAR *)pucData;
|
|
_tprintf(TEXT("ClientTest: Response received: %s\n"), strBuffer);
|
|
LocalFreeMemory(pucData);
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("ClientTest: Response packet not received\n"));
|
|
}
|
|
Sleep(50);
|
|
DoCloseAddress(TdiHandle);
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("ClientTest: unable to open address object\n"));
|
|
}
|
|
|
|
//
|
|
// adjust addresses...
|
|
//
|
|
switch (ulType)
|
|
{
|
|
case TDI_ADDRESS_TYPE_IPX:
|
|
{
|
|
PTA_IPX_ADDRESS pAddr = (PTA_IPX_ADDRESS)pRemoteAddr;
|
|
pAddr->Address[0].Address[0].Socket = CONN_SERVER_PORT;
|
|
|
|
pAddr = (PTA_IPX_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].Socket = CONN_CLIENT_PORT;
|
|
break;
|
|
}
|
|
case TDI_ADDRESS_TYPE_IP:
|
|
{
|
|
PTA_IP_ADDRESS pAddr = (PTA_IP_ADDRESS)pRemoteAddr;
|
|
pAddr->Address[0].Address[0].sin_port = CONN_SERVER_PORT;
|
|
|
|
pAddr = (PTA_IP_ADDRESS)pTransAddr;
|
|
pAddr->Address[0].Address[0].sin_port = CONN_CLIENT_PORT;
|
|
break;
|
|
}
|
|
}
|
|
|
|
//
|
|
// establish a connection
|
|
//
|
|
_putts(TEXT("ClientTest: Attempt to establish a connection\n"));
|
|
TdiHandle = DoOpenEndpoint(pConnDeviceName, pTransAddr);
|
|
if (TdiHandle)
|
|
{
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_CONNECT);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_DISCONNECT);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_ERROR);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_RECEIVE);
|
|
|
|
if (DoConnect(TdiHandle, pRemoteAddr, 20) == STATUS_SUCCESS)
|
|
{
|
|
_putts(TEXT("ClientTest: Sending first packet over connection\n"));
|
|
|
|
//
|
|
// do a connected send/receive test
|
|
//
|
|
TCHAR *strBuffer = TEXT("This is only a test");
|
|
|
|
DoSend(TdiHandle, (PUCHAR)strBuffer, sizeof(TCHAR) * (1 + _tcslen(strBuffer)), 0);
|
|
|
|
//
|
|
// wait for response
|
|
//
|
|
for (ULONG ulWait = 0; ulWait < 100; ulWait++)
|
|
{
|
|
Sleep(10);
|
|
PUCHAR pucData;
|
|
ULONG ulNumBytes = DoReceive(TdiHandle, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
_tprintf(TEXT("ClientTest: Response received: %s\n"), (TCHAR *)pucData);
|
|
LocalFreeMemory(pucData);
|
|
break;
|
|
}
|
|
}
|
|
_putts(TEXT("ClientTest: Sending second packet over connection\n"));
|
|
|
|
strBuffer = TEXT("Last Packet");
|
|
|
|
DoSend(TdiHandle, (PUCHAR)strBuffer, sizeof(TCHAR) * (1 + _tcslen(strBuffer)), 0);
|
|
|
|
//
|
|
// wait for response
|
|
//
|
|
for (ULONG ulWait = 0; ulWait < 100; ulWait++)
|
|
{
|
|
Sleep(10);
|
|
PUCHAR pucData;
|
|
ULONG ulNumBytes = DoReceive(TdiHandle, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
_tprintf(TEXT("ClientTest: Response received: %s\n"), (TCHAR *)pucData);
|
|
LocalFreeMemory(pucData);
|
|
break;
|
|
}
|
|
}
|
|
|
|
//
|
|
// shut down the connection
|
|
//
|
|
_putts(TEXT("ClientTest: closing connection\n"));
|
|
|
|
DoDisconnect(TdiHandle, TDI_DISCONNECT_RELEASE);
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("ClientTest: failed to establish connection\n"));
|
|
}
|
|
DoCloseEndpoint(TdiHandle);
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("ClientTest: failed to open endpoint\n"));
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
_putts(TEXT("Unable to find remote server"));
|
|
}
|
|
|
|
if (ulType == TDI_ADDRESS_TYPE_NETBIOS)
|
|
{
|
|
_putts(TEXT("ClientTest: get provider status\n"));
|
|
TdiHandle = DoOpenControl(pDataDeviceName);
|
|
if (TdiHandle)
|
|
{
|
|
PADAPTER_STATUS pStatus = (PADAPTER_STATUS)DoTdiQuery(TdiHandle, TDI_QUERY_ADAPTER_STATUS);
|
|
if (pStatus)
|
|
{
|
|
DoPrintAdapterStatus(pStatus);
|
|
LocalFreeMemory(pStatus);
|
|
}
|
|
|
|
DoCloseControl(TdiHandle);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ulType != TDI_ADDRESS_TYPE_NETBIOS)
|
|
{
|
|
_putts(TEXT("ClientTest: get provider statistics\n"));
|
|
|
|
TdiHandle = DoOpenControl(pDataDeviceName);
|
|
if (TdiHandle)
|
|
{
|
|
|
|
PTDI_PROVIDER_STATISTICS pStats
|
|
= (PTDI_PROVIDER_STATISTICS)DoTdiQuery(TdiHandle,
|
|
TDI_QUERY_PROVIDER_STATISTICS);
|
|
if (pStats)
|
|
{
|
|
DoPrintProviderStats(pStats);
|
|
LocalFreeMemory(pStats);
|
|
}
|
|
DoCloseControl(TdiHandle);
|
|
}
|
|
}
|
|
LocalFreeMemory(pDeviceName);
|
|
LocalFreeMemory(pTransAddr);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////
|
|
//
|
|
// Function: FindServer
|
|
//
|
|
// Arguments: pDataDeviceName -- name of data device to open
|
|
// pTransAddr -- address of data device to open
|
|
// pRemoteAddr -- on return, address of remote device
|
|
//
|
|
// Returns: TRUE if able to establish communication with server,
|
|
// FALSE if it times out
|
|
//
|
|
// Descript: This function is called by the client to find a server
|
|
// to participate with it in the tests.
|
|
//
|
|
/////////////////////////////////////////////////
|
|
|
|
BOOLEAN
|
|
FindServer(TCHAR *pDataDeviceName,
|
|
PTRANSPORT_ADDRESS pTransAddr,
|
|
PTRANSPORT_ADDRESS pRemoteAddr)
|
|
{
|
|
//
|
|
// set up remote and local address for broadcast/multicast search for server
|
|
//
|
|
pRemoteAddr->Address[0].AddressLength = pTransAddr->Address[0].AddressLength;
|
|
pRemoteAddr->Address[0].AddressType = pTransAddr->Address[0].AddressType;
|
|
|
|
switch (pTransAddr->Address[0].AddressType)
|
|
{
|
|
case TDI_ADDRESS_TYPE_IP:
|
|
{
|
|
PTDI_ADDRESS_IP pTdiAddressIp
|
|
= (PTDI_ADDRESS_IP)pTransAddr->Address[0].Address;
|
|
ULONG ulAddr = pTdiAddressIp->in_addr;
|
|
|
|
pTdiAddressIp->sin_port = DGRAM_CLIENT_PORT;
|
|
|
|
|
|
pTdiAddressIp = (PTDI_ADDRESS_IP)pRemoteAddr->Address[0].Address;
|
|
pTdiAddressIp->in_addr = 0xFFFF0000 | ulAddr;
|
|
pTdiAddressIp->sin_port = DGRAM_SERVER_PORT;
|
|
}
|
|
break;
|
|
|
|
case TDI_ADDRESS_TYPE_IPX:
|
|
{
|
|
PTDI_ADDRESS_IPX pTdiAddressIpx
|
|
= (PTDI_ADDRESS_IPX)pTransAddr->Address[0].Address;
|
|
ULONG TempNetwork = pTdiAddressIpx->NetworkAddress;
|
|
pTdiAddressIpx->Socket = DGRAM_CLIENT_PORT;
|
|
|
|
pTdiAddressIpx = (PTDI_ADDRESS_IPX)pRemoteAddr->Address[0].Address;
|
|
pTdiAddressIpx->NetworkAddress = TempNetwork;
|
|
pTdiAddressIpx->Socket = DGRAM_SERVER_PORT;
|
|
memset(pTdiAddressIpx->NodeAddress, 0xFF, 6);
|
|
}
|
|
break;
|
|
|
|
case TDI_ADDRESS_TYPE_NETBIOS:
|
|
{
|
|
PTDI_ADDRESS_NETBIOS pTdiAddressNetbios
|
|
= (PTDI_ADDRESS_NETBIOS)pRemoteAddr->Address[0].Address;
|
|
|
|
pTdiAddressNetbios->NetbiosNameType = TDI_ADDRESS_NETBIOS_TYPE_UNIQUE;
|
|
memcpy(pTdiAddressNetbios->NetbiosName, "SERVER", 7 ); // NOTE: ascii string
|
|
}
|
|
break;
|
|
|
|
|
|
default:
|
|
_putts(TEXT("FindServer: invalid address type\n"));
|
|
return FALSE;
|
|
}
|
|
|
|
//
|
|
// try to find server program to test against
|
|
//
|
|
BOOLEAN fSuccessful = FALSE;
|
|
ULONG TdiHandle;
|
|
|
|
_putts(TEXT("FindServer: try to find remote test server\n"));
|
|
|
|
while (TRUE)
|
|
{
|
|
TdiHandle = DoOpenAddress(pDataDeviceName, pTransAddr);
|
|
if (!TdiHandle)
|
|
{
|
|
_putts(TEXT("FindServer: unable to open address object\n"));
|
|
break;
|
|
}
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_ERROR);
|
|
DoEnableEventHandler(TdiHandle, TDI_EVENT_RECEIVE_DATAGRAM);
|
|
|
|
//
|
|
// send broadcast query
|
|
//
|
|
_putts(TEXT("FindServer: send first packet (broadcast)\n"));
|
|
|
|
ULONG ulBuffer = 0x12345678;
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)&ulBuffer, sizeof(ULONG));
|
|
|
|
//
|
|
// wait for first response
|
|
//
|
|
for (ULONG ulCount = 0; ulCount < 100; ulCount++)
|
|
{
|
|
Sleep(10);
|
|
PUCHAR pucData;
|
|
ULONG ulNumBytes = DoReceiveDatagram(TdiHandle, NULL, pRemoteAddr, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
if (ulNumBytes == 4)
|
|
{
|
|
PULONG pulValue = (PULONG)pucData;
|
|
if (*pulValue == 0x98765432)
|
|
{
|
|
_putts(TEXT("FindServer: first response received\n"));
|
|
fSuccessful = TRUE;
|
|
}
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
//
|
|
// break out of loop if received response
|
|
//
|
|
if (fSuccessful)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// timed out -- no response
|
|
//
|
|
if (!fSuccessful)
|
|
{
|
|
_putts(TEXT("FindServer: did not receive first response\n"));
|
|
break;
|
|
}
|
|
|
|
//
|
|
// send second message
|
|
//
|
|
fSuccessful = FALSE;
|
|
ulBuffer = 0x22222222;
|
|
_putts(TEXT("FindServer: send second packet (directed)\n"));
|
|
|
|
DoSendDatagram(TdiHandle, pRemoteAddr, (PUCHAR)&ulBuffer, sizeof(ULONG));
|
|
|
|
//
|
|
// wait for second response
|
|
//
|
|
for (ULONG ulCount = 0; ulCount < 50; ulCount++)
|
|
{
|
|
Sleep(10);
|
|
PUCHAR pucData;
|
|
ULONG ulNumBytes = DoReceiveDatagram(TdiHandle, NULL, NULL, &pucData);
|
|
if (ulNumBytes)
|
|
{
|
|
if (ulNumBytes == 4)
|
|
{
|
|
PULONG pulValue = (PULONG)pucData;
|
|
if (*pulValue == 0x33333333)
|
|
{
|
|
_putts(TEXT("FindServer: second response received\n"));
|
|
fSuccessful = TRUE;
|
|
}
|
|
}
|
|
LocalFreeMemory(pucData);
|
|
//
|
|
// break out if got response
|
|
//
|
|
if (fSuccessful)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (!fSuccessful)
|
|
{
|
|
_putts(TEXT("FindServer: second response not received\n"));
|
|
|
|
}
|
|
if (TdiHandle)
|
|
{
|
|
DoCloseAddress(TdiHandle);
|
|
}
|
|
return fSuccessful;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// end of file tdisample.cpp
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|