main.h

好东东

/*++

Copyright (c) 2005  Changzhi Zhou All Rights Reserved

Module Name:

    main.h

Abstract:

   Declaration for all functions.
   Define all structures.

Environment:

    Kernel mode

Revision History:

	Changzhi Zhou Dec 20  2004

--*/
//#if !defined(_FILTER_H_)
//#define _FILTER_H_

#ifndef __MAIN_H_
#define __MAIN_H_


#include <csq.h> // Include this header file and link with csq.lib to use this sample on Win2K.
#include "tdi.h"
#define DRIVERNAME "Sample: "
#define ORIGINALDATA_SIZE 65536

#if DBG
#define DebugPrint(_x_) \
               DbgPrint (DRIVERNAME); \
               DbgPrint _x_;

#define TRAP() DbgBreakPoint()

#else
#define DebugPrint(_x_)
#define TRAP()
#endif

#if DBG
#define CSAMP_KDPRINT(_x_) \
                DbgPrint("CancelIrp.c: ");\
                DbgPrint _x_;
#else
#define CSAMP_KDPRINT(_x_)
#endif

#include "wmilib.h"
#include <ntddser.h>


#define DEVICE_OBJECT_NAME_LENGTH	128
#define DOS_DEVICE_NAME				L"\\DosDevices\\COM"
#define SYMBOLIC_NAME_LENGTH		128
#define SERIAL_DEVICE_MAP			L"SERIALCOMM"
#define DEVICE_NAME					L"\\Device\\Serial"

#define RINGBUFFER_SIZE			65536
#define RINGBUFFER_SIZE_80FULL	52428

#define RECVREMAINDER_BUFFER_SIZE	65536
// read interval timeout 1 second
#define	READINTERVAL_TIMEOUT	10000000

//
// These are the states Filter transition to upon
// receiving a specific PnP Irp. Refer to the PnP Device States
// diagram in DDK documentation for better understanding.
//

typedef enum _DEVICE_PNP_STATE {

    NotStarted,         // not started
    Stopped,            // device stopped
    Working,            // started and working
	//Connecting,
    PendingStop,        // stop pending
    PendingRemove,      // remove pending
    SurpriseRemoved,    // removed by surprise
    Removed             // removed

} DEVICE_PNP_STATE;

#define INITIALIZE_PNP_STATE(_Data_)    \
        (_Data_)->DevicePnPState =  NotStarted;\
        (_Data_)->PreviousPnPState = NotStarted;

#define SET_NEW_PNP_STATE(_Data_, _state_) \
		KeAcquireSpinLock( &(_Data_)->DevStateLock, &oldIrql );\
        (_Data_)->PreviousPnPState =  (_Data_)->DevicePnPState;\
        (_Data_)->DevicePnPState = (_state_);\
		KeReleaseSpinLock( &(_Data_)->DevStateLock, oldIrql );

#define RESTORE_PREVIOUS_PNP_STATE(_Data_)   \
		KeAcquireSpinLock( &(_Data_)->DevStateLock, &oldIrql );\
        (_Data_)->DevicePnPState =   (_Data_)->PreviousPnPState;\
		KeReleaseSpinLock( &(_Data_)->DevStateLock, oldIrql );

typedef struct _RecvContext{
	PIRP	pIrp;
	PMDL	pMdl;
	BOOLEAN	bLocked;
	IO_STATUS_BLOCK		IoStatus;
	KEVENT	Event;
	TDI_CONNECTION_INFORMATION  ReceiveDatagramInfo;
    TDI_CONNECTION_INFORMATION  ReturnInfo;
	PVOID	RemainderBuffer;
}RECV_CONTEXT, *PRECV_CONTEXT;
	
typedef struct _DEVICE_EXTENSION
{
	// regular varialbes
    PDEVICE_OBJECT		Self;
    PDEVICE_OBJECT		NextLowerDriver;

	// PnP state
    DEVICE_PNP_STATE	DevicePnPState;
    DEVICE_PNP_STATE	PreviousPnPState;
    // obtain and hold this lock while changing the device state,
    // the queue state and while processing the queue.	
	KSPIN_LOCK			DevStateLock;

	// relevant to PnP
    KEVENT				RemoveEvent;
    KEVENT				StopEvent;
	ULONG				OutStandingIO;
    KSPIN_LOCK			IOCountLock;
	BOOLEAN				bIsOpen;

	// Names
    UNICODE_STRING      InterfaceName; // The name returned from IoRegisterDeviceInterface,
	UNICODE_STRING		DeviceName;
	UNICODE_STRING		SymbolicLinkName;
	UNICODE_STRING		DosName;
	ULONG				ComX;
	BOOLEAN				CreatedSymbolicLink;
	BOOLEAN				CreatedSerialCommEntry;


	KSPIN_LOCK			ThreadSpinLock;	// Spinlock for threads

	// relevant to Serial Port
	ULONG				CurrentBaud;
	SERIAL_TIMEOUTS		Timeouts;
	SERIAL_LINE_CONTROL	LineControl;
    SERIAL_CHARS		SpecialChars;
	SERIAL_STATUS		SerialStatus;
	ULONG				BufferSize;
    SERIAL_HANDFLOW		HandFlow;
	ULONG				WaitMask;
	PIRP				WaitOnMaskIrp;
	PIRP				PendingReadIrp;
	KTIMER				ReadTimer;
	KDPC				ReadDpc;

	// relevant to read/write buffer
	PUCHAR				RxBuffer;
	PUCHAR				lpRead;
	PUCHAR				lpRx;

    //  SpinLock to protect access to the queue
/*    KSPIN_LOCK			QueueLock;
    IO_CSQ				CancelSafeQueue;
    // Irps waiting to be processed are queued here
    LIST_ENTRY			PendingIrpQueue;
*/

	// variables about TDI client
	HANDLE				hTransAddr;
	PFILE_OBJECT		lpTransAddrFileObject;
	PDEVICE_OBJECT		TDILowerDeviceObject;
	RECV_CONTEXT		recvContext;
	USHORT				RemotePort;
	ULONG				RemoteAddress;

	// WMI
    WMILIB_CONTEXT      WmiLibInfo;     // WMI Information

} DEVICE_EXTENSION, *PDEVICE_EXTENSION;
#if 0
typedef struct _ProcContext{
	PDEVICE_EXTENSION	deviceExtension;
	PIRP				GetTimeIrp;
	KTIMER				Timer;
	KDPC				dpc;
}PROC_CONTEXT, *PPROC_CONTEXT;

typedef struct _ShareMemoryData{
	PFILE_OBJECT		pFileObject;
	PVOID				SystemVirtualAddress;
	PVOID				UserVirtualAddress;
	PMDL				Mdl;
	LIST_ENTRY			Link;
} SHARE_MEMORY_DATA, *PSHARE_MEMORY_DATA;
#endif
PCHAR
PnPMinorFunctionString (
    UCHAR MinorFunction
);

NTSTATUS
AddDevice(
    IN PDRIVER_OBJECT DriverObject,
    IN PDEVICE_OBJECT PhysicalDeviceObject
    );


NTSTATUS
SamplePnpDispatch (
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp
    );

NTSTATUS
SamplePowerDispatch(
    IN PDEVICE_OBJECT    DeviceObject,
    IN PIRP              Irp
    );

VOID
Unload(
    IN PDRIVER_OBJECT DriverObject
    );
NTSTATUS DefaultPnpHandler(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);

NTSTATUS PnpStartDevice(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
NTSTATUS PnpRemoveDevice(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
NTSTATUS ForwardAndWait(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
NTSTATUS SampleCreate (
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);

NTSTATUS SampleClose(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);

NTSTATUS SampleIoControl(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
NTSTATUS SampleCleanup(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
NTSTATUS PnpSurpriseRemoval(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP           Irp
    );
NTSTATUS CompleteRequest ( 
	IN PIRP Irp,
	IN NTSTATUS status,
	IN ULONG info
	);
NTSTATUS OnRequestComplete(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp,
    IN PKEVENT event
	);

NTSTATUS
SampleWrite(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp
	);
NTSTATUS
SampleRead(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp
	);
VOID SampleCancelRoutine(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp
    );
NTSTATUS SampleCompletionRoutine(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP Irp,
    IN PVOID Context
    );
NTSTATUS
ToasterSystemControl (
    IN  PDEVICE_OBJECT  DeviceObject,
    IN  PIRP            Irp
    );
NTSTATUS
PnpQueryStopDevice(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP           Irp
    );
NTSTATUS
PnpCancelStopDevice(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP           Irp
    );
NTSTATUS
PnpQueryRemoveDevice(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP           Irp
    );
NTSTATUS
PnpCancelRemoveDevice(
    IN PDEVICE_OBJECT DeviceObject,
    IN PIRP           Irp
    );
NTSTATUS PnpQueryCapabilities(
	IN PDEVICE_OBJECT DeviceObject,
	IN PIRP Irp
	);
LONG SampleIoIncrement(
    IN OUT PDEVICE_EXTENSION DeviceExtension
    );
LONG SampleIoDecrement(
    IN OUT PDEVICE_EXTENSION DeviceExtension
    );

BOOLEAN InitializeSerialDevName(
	PUNICODE_STRING lpDeviceName
	);
NTSTATUS
SerialDoExternalNaming(
	IN PDEVICE_EXTENSION	deviceExtension,
	IN LONG					ComX
	);
NTSTATUS
SerialRemoveDevObj(
	IN PDEVICE_OBJECT PDevObj
	);
NTSTATUS
SerialUndoExternalNaming(
	IN PDEVICE_EXTENSION	deviceExtension
	);


NTSTATUS CreateDosName( IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);

VOID
ReadDpcRoutine(
    IN PKDPC Dpc,
    IN PVOID DeferredContext,
    IN PVOID SystemArgument1,
    IN PVOID SystemArgument2
    );
ULONG	CopyFromRingBuffer(
	PDEVICE_EXTENSION	deviceExtension,
	PIRP				Irp );
//  CancelIrp 
VOID CsampInsertIrp (
    IN PIO_CSQ   Csq,
    IN PIRP              Irp
    );
VOID CsampRemoveIrp(
    IN  PIO_CSQ Csq,
    IN  PIRP    Irp
    );
PIRP CsampPeekNextIrp(
    IN  PIO_CSQ Csq,
    IN  PIRP    Irp,
    IN  PVOID   PeekContext
    );
VOID CsampAcquireLock(
    IN  PIO_CSQ Csq,
    OUT PKIRQL  Irql
    );
VOID CsampReleaseLock(
    IN PIO_CSQ Csq,
    IN KIRQL   Irql
    );
VOID CsampCompleteCanceledIrp(
    IN  PIO_CSQ             pCsq,
    IN  PIRP                Irp
    );

// TDI
NTSTATUS Disconnection ( PDEVICE_EXTENSION deviceExtension );
NTSTATUS InitializeConnection( PDEVICE_EXTENSION deviceExtension );
NTSTATUS	TDIQueryNetworkInformation( PDEVICE_OBJECT DeviceObject, PIRP Irp );
NTSTATUS	TDIRecvDatagram ( PDEVICE_EXTENSION	deviceExtension );
NTSTATUS	TDISendDatagram( DeviceObject, Irp );
NTSTATUS
DriverEntry(
    IN PDRIVER_OBJECT  DriverObject,
    IN PUNICODE_STRING RegistryPath
    );

#endif