openclos.c

用来监视网络通信数据的源代码和应用程序,方便网络程序底层开发.

/*
 * Copyright (c) 1999, 2000
 *	Politecnico di Torino.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that: (1) source code distributions
 * retain the above copyright notice and this paragraph in its entirety, (2)
 * distributions including binary code include the above copyright notice and
 * this paragraph in its entirety in the documentation or other materials
 * provided with the distribution, and (3) all advertising materials mentioning
 * features or use of this software display the following acknowledgement:
 * ``This product includes software developed by the Politecnico
 * di Torino, and its contributors.'' Neither the name of
 * the University nor the names of its contributors may be used to endorse
 * or promote products derived from this software without specific prior
 * written permission.
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 */

#include <basedef.h>
#include <vmm.h>
#include <ndis.h>
#include <vwin32.h>
#include "debug.h"
#include "packet.h"
#pragma VxD_LOCKED_CODE_SEG
#pragma VxD_LOCKED_DATA_SEG


void YieldExecution( void )
{
	VMMCall(Release_Time_Slice);
	VMMCall(Begin_Nest_Exec);
	VMMCall(Resume_Exec);
	VMMCall(End_Nest_Exec);
}

/************************************************************
Function called when the user level application performs a open
IOCTL. Opens the adapter.
************************************************************/

static NDIS_MEDIUM MediumArray[] = {
    NdisMedium802_3,
    NdisMediumWan,
    NdisMediumFddi,
    NdisMediumArcnet878_2,
    NdisMedium802_5
};

#define NUM_NDIS_MEDIA  (sizeof MediumArray / sizeof MediumArray[0])

DWORD PacketOpen(PNDIS_STRING AdapterName,DWORD dwDDB,DWORD hDevice,PDIOCPARAMETERS pDiocParms)
{

	LARGE_INTEGER		SystemTime;
	__int64				ltime1;
	PDEVICE_EXTENSION	pde;
	POPEN_INSTANCE 		oiNew;
	NDIS_STATUS			nsErrorStatus, nsOpenStatus;
	UINT           	i;
	UINT           	uiMedium;
	NDIS_STRING		NameStr;
	NDIS_STATUS	Status;


	pde = GlobalDeviceExtension;
	/*Allocate an element that describe an adapter*/
	NdisAllocateMemory( (PVOID *)&oiNew, sizeof( OPEN_INSTANCE ), 0, -1 );
	if ( oiNew == NULL ) 
	{
		return NDIS_STATUS_FAILURE;
	}
	NdisZeroMemory( (PVOID)oiNew, sizeof( OPEN_INSTANCE ) );
	/*allocate a pool for the packet headers*/

	NdisAllocatePacketPool( &nsErrorStatus,
							&(oiNew->PacketPool),
							TRANSMIT_PACKETS,
							sizeof(PACKET_RESERVED) );

	IF_TRACE_MSG( "PACKET_RESERVED_a :%lx",sizeof(PACKET_RESERVED));
	if ( nsErrorStatus != NDIS_STATUS_SUCCESS ) 
	{
		IF_TRACE_MSG( "Failed to allocate packet pool AllocStatus=%x", nsErrorStatus );
		NdisFreeMemory( oiNew, sizeof( OPEN_INSTANCE ) ,  0 );
		TRACE_LEAVE( "BindAdapter" );
		return NDIS_STATUS_FAILURE;
	}


	/*allocate a buffer pool for the packet data*/
	NdisAllocateBufferPool( &nsErrorStatus,
							&(oiNew->BufferPool),
							TRANSMIT_PACKETS );
	if ( nsErrorStatus != NDIS_STATUS_SUCCESS )
	{
		IF_TRACE_MSG( "Failed to allocate packet pool AllocStatus=%x", nsErrorStatus );
		NdisFreePacketPool( oiNew->PacketPool );
		NdisFreeMemory( oiNew, sizeof( OPEN_INSTANCE ) ,  0 );
		TRACE_LEAVE( "BindAdapter" );
		return NDIS_STATUS_FAILURE;
	}
	NdisAllocateSpinLock( &(oiNew->ResetSpinLock) );
	InitializeListHead( &(oiNew->ResetIrpList) );
	NdisAllocateSpinLock( &(oiNew->RcvQSpinLock) );
	InitializeListHead( &(oiNew->RcvList) );
	NdisAllocateSpinLock( &(oiNew->RequestSpinLock) );
	InitializeListHead( &(oiNew->RequestList) );

	for ( i=0;i<MAX_REQUESTS;i++ ) 
	{
		InsertTailList( &(oiNew->RequestList), &(oiNew->Requests[i].Reserved.ListElement) );
	}

	oiNew->Status = NDIS_STATUS_PENDING;

	/*initialize the timer variables for this session*/

	SystemTime=GetDate();	

	ltime1=((__int64)SystemTime.HighPart*86400);
	ltime1+=(__int64)(SystemTime.LowPart/1000);	//current time from 1980 in seconds
	ltime1+=(__int64)315532800;	//current time from 1970 (Unix format) in seconds
	ltime1*=1193182;
	ltime1+=(SystemTime.LowPart%1000)*1193182/1000; //current time from 1970 in ticks
	ltime1-=QuerySystemTime();	//boot time from 1970 in ticks
	oiNew->StartTime=ltime1;


	oiNew->Dropped=0;		//reset the dropped packets counter
	oiNew->Received=0;		//reset the received packets counter
	oiNew->bpfprogram=NULL;	//set an accept-all filter
	oiNew->bpfprogramlen=0;
	oiNew->BufSize=0;		//set an empty buffer
	oiNew->Buffer=NULL;		//reset the buffer
	oiNew->Bhead=0;
	oiNew->Btail=0;
	oiNew->BLastByte=0;
	oiNew->TimeOut=0;		//reset the timeouts
	oiNew->ReadTimeoutTimer=0;
	oiNew->mode=0;			//set capture mode
	oiNew->Nbytes=0;		//reset the counters
	oiNew->Npackets=0;
	oiNew->hDevice=hDevice;
	oiNew->tagProcess=pDiocParms->tagProcess;
	oiNew->ReadEvent=0;		//reset the read event

	NdisAllocateSpinLock( &(oiNew->CountersLock) );
	/*open the MAC driver calling NDIS*/
	NdisOpenAdapter( &nsOpenStatus,
					 &nsErrorStatus,
					 &oiNew->AdapterHandle,
					 &uiMedium,
					 MediumArray,
					 NUM_NDIS_MEDIA,
					 pde->NdisProtocolHandle,
					 oiNew,
					 AdapterName,
					 0,
					 NULL );

	IF_TRACE_MSG( "Open Status                   : %lx", nsOpenStatus );
	IF_TRACE_MSG( "Error Status                  : %lx", nsErrorStatus );
	IF_TRACE_MSG( "Completion Status             : %lx", oiNew->Status );

	if ( nsOpenStatus == NDIS_STATUS_PENDING )
	{
		while ( oiNew->Status == NDIS_STATUS_PENDING )
			YieldExecution();
	}
	else
	{
		PacketOpenAdapterComplete( oiNew, nsOpenStatus, nsErrorStatus );
	}

	Status = oiNew->Status;
	if ( Status != NDIS_STATUS_SUCCESS ) 
	{
		NdisFreeMemory( oiNew, sizeof( OPEN_INSTANCE ) ,  0 );
		return NDIS_STATUS_FAILURE;
	}
	else
	{

	}

	TRACE_LEAVE( "BindAdapter" );

	/*return succesfully*/
	return STATUS_SUCCESS;

}

/************************************************************
Function called when the user level application performs a Close
IOCTL. Closes the adapter and free the reources associated with it
************************************************************/

DWORD PacketClose(POPEN_INSTANCE Open,DWORD dwDDB,DWORD hDevice,PDIOCPARAMETERS pDiocParms)
{
	
	NDIS_STATUS			Status;
	NDIS_STATUS			nsErrorStatus;
	UINT				to;
	DWORD				TEvent;

	TRACE_ENTER( "PacketClose" );

	Open->BufSize=0;
	
	to=Open->ReadTimeoutTimer;
	Open->ReadTimeoutTimer=0;
	if(to!=0){
		_asm push esi;
		_asm mov esi,to;
		CancelReadTimeOut();
		_asm pop esi;
	}
	
	// Free the read event
	TEvent=Open->ReadEvent;
	_asm mov eax,TEvent;
	VxDCall(_VWIN32_CloseVxDHandle);
	
	//close the adapter
	NdisCloseAdapter(&nsErrorStatus,Open->AdapterHandle);
	if ( nsErrorStatus == NDIS_STATUS_PENDING )
	{
		while ( Open->Status == NDIS_STATUS_PENDING )
			YieldExecution();
		
		if(Open->Status!=NDIS_STATUS_SUCCESS){
			TRACE_LEAVE( "PacketClose" );
			return NDIS_STATUS_FAILURE;
		}
	}
	else
	{
		PacketUnbindAdapterComplete( Open, nsErrorStatus );
		if(nsErrorStatus!=NDIS_STATUS_SUCCESS){
			TRACE_LEAVE( "PacketClose" );
			return NDIS_STATUS_FAILURE;
		}
	}

	Status = Open->Status;
	
	if(Open->Buffer!=NULL)NdisFreeMemory(Open->Buffer,Open->BufSize,0);
	Open->Buffer=NULL;
	if(Open->bpfprogram!=NULL)NdisFreeMemory(Open->bpfprogram,Open->bpfprogramlen,0);
	
	//remove this adapter from the list of open adapters
	NdisAcquireSpinLock( &GlobalDeviceExtension->OpenSpinLock );
	RemoveEntryList(&(Open->ListElement));
	NdisReleaseSpinLock( &GlobalDeviceExtension->OpenSpinLock );

	NdisFreeMemory( Open, sizeof( OPEN_INSTANCE ) ,  0 );

	if(pDiocParms!=NULL)
		*(DWORD *)(pDiocParms->lpcbBytesReturned) = 0;
	
	TRACE_LEAVE( "PacketClose" );
	return Status;
	
}

/************************************************************
Function used by NDIS to update the VXD when a new MAC driver
is added
************************************************************/
VOID NDIS_API PacketBindAdapter( OUT PNDIS_STATUS Status,
						 IN  NDIS_HANDLE  BindAdapterContext,
						 IN  PNDIS_STRING AdapterName,
						 IN  PVOID        SystemSpecific1,
						 IN  PVOID        SystemSpecific2 )
{
	PDEVICE_EXTENSION	pde;
	POPEN_INSTANCE		oiNew;
	NDIS_STATUS			nsErrorStatus, nsOpenStatus;
	UINT           		uiMedium;
	UINT           		i;
	PWRAPPER_PROTOCOL_BLOCK				pWPBlock;
	PNDIS_PROTOCOL_CHARACTERISTICS	pNPChar;
	PADAPTER_NAME		AName;
	PWRAPPER_MAC_BLOCK	pWMBlock;
	PNDIS_MAC_CHARACTERISTICS	  pNMChar;
    BYTE                *lpzName;


	TRACE_ENTER( "BindAdapter" );
	pde = GlobalDeviceExtension;
	/*Allocate an element that describe an adapter*/
	NdisAllocateMemory( (PVOID *)&AName, sizeof(ADAPTER_NAME), 0, -1 );
	if ( AName == NULL ) 
	{
		*Status = NDIS_STATUS_RESOURCES;
		return;
	}

	NdisAllocateMemory( (PVOID *)&oiNew, sizeof( OPEN_INSTANCE ), 0, -1 );
	if ( oiNew == NULL ) 
	{
		*Status = NDIS_STATUS_RESOURCES;
		return;
	}
	NdisZeroMemory( (PVOID)oiNew, sizeof( OPEN_INSTANCE ) );

	/*Save Binding Context*/
	oiNew->BindAdapterContext = BindAdapterContext;

    /*Save the device handle*/
    
    oiNew->hDevice = (DWORD) SystemSpecific1;

	/*allocate a pool for the packet headers*/

	NdisAllocatePacketPool( &nsErrorStatus,
							&(oiNew->PacketPool),
							TRANSMIT_PACKETS,
							sizeof(PACKET_RESERVED) );

	IF_TRACE_MSG( "PACKET_RESERVED_b :%lx",sizeof(PACKET_RESERVED));
	if ( nsErrorStatus != NDIS_STATUS_SUCCESS ) 
	{
		IF_TRACE_MSG( "Failed to allocate packet pool AllocStatus=%x", nsErrorStatus );
		NdisFreeMemory( oiNew, sizeof( OPEN_INSTANCE ) ,  0 );