忆杰的博客

忆杰的博客

Win32与内核同步

关于这个线程同步这块, 就比较恶心了, Windows驱动开发技术详解里面 这章的源码都没有. 难道是当时忘记了? 不应该啊!
我倒. 虽然说起来和Win32上面差不多, 但是也不能这样. 连个源码都看不到. 也许是觉得这章并没有什么实际的例子好讲,我看到了第九章的时候, 关于IRP的同步问题, 大部分都用了这个知识, 所以我觉得第8章的话, 先混个眼熟就可以. 回头下面几章的时候再倒腾.

但是第8章, 还是有不少比较有营养的内容的, 首先就是创建线程部分了. 这个必须于要试验下, 还有就是用户层传递一个同步事件对象给内核.内核等待这个对象, 或者是设置这个对象. 真是暴力. 在Win32上面我们拿到的都是句柄, 但是在内核里面就都可以转成指针了.一般是调用ObReferenceObjectByHandle, 这号函数非常好使. 不但可以转这个同步对象, 其他的很多东西都可以转, 只要有句柄就可以. 另外还有个将名称转换成指针, 到时候用到再写!

要说这个同步的话, Win32和内核里面用起来都差不多, 无非是多了几个参数而已. 只是内核里面比Win32多了一个自旋锁. 还有一系列的原子操作函数. 其他的操作起来都差不多, 事件, 信号灯,  互斥量, 还多了一种快速互斥量, 互斥量和快速互斥量之间的区别只是互斥量可以递归的获取, 而快速互斥量不能够递归的获取, 快速互斥量因为不能递归获取, 所以速度上要快一段. 但是就我刚才在虚拟机上面做的实验来看的话, 这个差别太小了. 几乎都是差不多的用. 也许是我实验的比较片面.

但是Win32下面的临界区好像在内核里面没有了. 临界区的话我感觉和自旋锁差不多.当然是用起来. 其他的也没有什么好说的. 我还是说说下面这个代码的逻辑, 首先就是在用户态发送一个IOCTL_CONTROL_CREATETHREAD控制码, 然后内核就是创建两个线程, 在这两个线程中, 我们模拟做一些事情, 然后在主线程中我们等待两个线程的返回, 我死活没有实验出来为什么我调用KeWaitForMultipleObjects会失败. 查了不少资料也没有找到办法, 老是蓝屏.  那我只能分两次来等待对象了. 我倒…

然后在用户态这边又创建了一个时间对象, 就是用来实验用户态的对象可以传递给内核使用的, 在DeviceIoControl函数中, 发送IOCTL_CONTROL_SHOREEVENT控制码, 在此并且将句柄传递给内核. 然后在内核里面将句柄转换成指针, 然后设置事件, 并返回, 如果一切正常那么这时候用户态这边调用WaitForSingleObject去等待这个对象那么就会马上返回了.. 差不多就是这些了!

直接上代码. 这边是用户态的:

/*
	Windows内核与Win32进行同步操作 3环代码
	编译方法参见makefile. TAB = 8
*/
#include <windows.h>
#include <stdio.h>
#include <winioctl.h>

#define  SYS_LINK_NAME	"\\\\.\\SysLinkWaitForEvent"

//		             设备类型             功能号 I/O访问内存使用方式 权限
#define IOCTL_CONTROL_CREATETHREAD CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS )
#define IOCTL_CONTROL_SHOREEVENT CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_BUFFERED, FILE_ANY_ACCESS )

#pragma comment(linker, "/Entry:Jmain")
#pragma comment(linker, "/subsystem:console")
//===========================================================================
	//线程函数.
//===========================================================================
DWORD WINAPI Thread1( PVOID pContext ) {
	HANDLE hEvent;
	DWORD  dwCount;

	hEvent = *(HANDLE*)pContext;
	dwCount = GetTickCount();

	printf( "进入线程函数, 进行等待!\n" );

	//很明显, 如果这里内核不将该事件置位, 那么是不会返回的.
	WaitForSingleObject( hEvent, INFINITE );

	printf( "等待成功, 等待了 %u !\n", GetTickCount() - dwCount );
	ExitThread(0);
}
//===========================================================================
int Jmain() {
	HANDLE	hFile = 0;
	HANDLE	hEvent = 0;
	HANDLE	hThread = 0;
	BOOL	bRet;
	DWORD	dwByteRead;
	DWORD	dwThreadId;
	//打开设备
	do {
		//创建用户层同步对象, 传递进内核进行修改
		hEvent = CreateEvent( NULL, FALSE, FALSE, NULL );
		if ( hEvent == NULL ) {
			printf( "创建同步事件失败!\n" );
			break;
		}

		hFile = CreateFile( SYS_LINK_NAME, GENERIC_READ | GENERIC_WRITE, 0, 0,
		                    OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0  );
		if ( hFile == INVALID_HANDLE_VALUE ) {
			printf( "打开设备错误了!\n" );
			break;
		}

		bRet = DeviceIoControl( hFile, IOCTL_CONTROL_CREATETHREAD, NULL, 0,
		                        NULL, 0, &dwByteRead, NULL );
		if ( !bRet ) {
			printf( "发送IOCTL_CONTROL_CREATETHREAD控制码失败!\n" );
			break;
		}

		//创建一个线程等待. 不然看不出效果
		hThread = CreateThread( NULL, 0, &Thread1, &hEvent, 0, &dwThreadId );
		if (hThread == NULL ) {
			printf( "创建线程失败!\n" );
			break;
		}

		Sleep( 1000 );
		bRet = DeviceIoControl( hFile, IOCTL_CONTROL_SHOREEVENT, &hEvent, sizeof( hEvent ),
		                        NULL, 0, &dwByteRead, NULL );
		if ( !bRet ) {
			printf( "发送IOCTL_CONTROL_SHOREEVENT控制码失败!\n" );
			break;
		}

		//等待线程返回
		WaitForSingleObject( hThread, INFINITE );

		printf( "事件被成功置位, 并返回, 程序退出!\n" );
	} while ( FALSE );
//---------------------------------------------------------------------------
	if ( hFile ) {
		CloseHandle( hFile );
	}

	if ( hEvent ) {
		CloseHandle( hEvent );
	}

	if ( hThread ) {
		CloseHandle( hThread );
	}
	system( "pause" );

	ExitProcess( 0 );
	return 0;
}

这边是内核态的代码:

 

/*
	Windows内核与Win32进行同步操作 0环代码
	编译方法参见makefile. TAB = 8
*/
#include <ntddk.h>

typedef struct tagDeviceExt {
	PDEVICE_OBJECT pDeviceObj;
	UNICODE_STRING uSzDeviceName;
	UNICODE_STRING uSzSysLinkName;
} DEVICE_EXT, *PDEVICE_EXT;

#define IOCTL_CONTROL_CREATETHREAD CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS )
#define IOCTL_CONTROL_SHOREEVENT CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_BUFFERED, FILE_ANY_ACCESS )

#define DEVICE_NAME	L"\\Device\\DevWaitForEvent"
#define SYS_LINK_NAME	L"\\??\\SysLinkWaitForEvent"

#pragma code_seg("PAGE")
//===========================================================================
//	用户线程Routine
VOID SysRoutine( PVOID pContext ) {
	PEPROCESS pEprocess;
	PTSTR pStrProcessName;
	PKEVENT phEvent;

	PAGED_CODE();

	phEvent = ( PKEVENT )pContext;

	//获取进程名称
	pEprocess = IoGetCurrentProcess();
	pStrProcessName = ( PTSTR )( ( ULONG )pEprocess + 0x174 );
	KdPrint( ( "进程名称:%s\n",  pStrProcessName ) );

	//停止50毫秒
	KeStallExecutionProcessor( 50 );
	//置位事件
	KeSetEvent( phEvent, IO_NO_INCREMENT, FALSE );
	//结束本进程
	PsTerminateSystemThread( STATUS_SUCCESS );
}

//===========================================================================
//	系统线程
VOID UserRoutine( PVOID pContext ) {
	PKEVENT	phEvent;
	PEPROCESS pEprocess;
	PTSTR pStrProcessName;

	PAGED_CODE();
	phEvent = ( PKEVENT )pContext;

	//获取进程名称
	pEprocess = IoGetCurrentProcess();
	pStrProcessName = ( PTSTR )( ( ULONG )pEprocess + 0x174 );
	KdPrint( ( "进程名称:%s\n",  pStrProcessName ) );

	//停止50毫秒
	KeStallExecutionProcessor( 50 );
	//置位事件
	KeSetEvent( phEvent, IO_NO_INCREMENT, FALSE );
	//结束本进程
	PsTerminateSystemThread( STATUS_SUCCESS );
}

//===========================================================================
//驱动卸载例程
VOID DriverUnload( PDRIVER_OBJECT pDriverObj ) {
	PDEVICE_OBJECT pNextDeviceObj;
	PDEVICE_EXT pDeviceExt;

	PAGED_CODE();
	pNextDeviceObj = pDriverObj->DeviceObject;

	while( pNextDeviceObj != NULL ) {
		pDeviceExt = ( PDEVICE_EXT )pNextDeviceObj->DeviceExtension;

		IoDeleteDevice( pDeviceExt->pDeviceObj );
		IoDeleteSymbolicLink( &pDeviceExt->uSzSysLinkName );

		KdPrint( ( "删除设备%wZ成功!\n", &pDeviceExt->uSzDeviceName ) );

		pNextDeviceObj = pNextDeviceObj->NextDevice;
	}
}

//===========================================================================
//	所有不关心的例程处理
NTSTATUS DispatchRoutine( PDEVICE_OBJECT pDeviceObj, PIRP pIrp ) {
	pIrp->IoStatus.Information = 0;
	pIrp->IoStatus.Status = STATUS_SUCCESS;
	IoCompleteRequest( pIrp, IO_NO_INCREMENT );
	return STATUS_SUCCESS;
}

//===========================================================================
#pragma code_seg()
//	DeviceIoControl函数IRP处理
NTSTATUS DispatchControl( PDEVICE_OBJECT pDeviceObj, PIRP pIrp ) {
	ULONG Code;
	KIRQL OldIrp;
	HANDLE hThread[2];
	KEVENT hEvent[2];
	HANDLE hUserEvent;
	PKEVENT phEvent;
	NTSTATUS Status;
	PIO_STACK_LOCATION Stack;
	//KWAIT_BLOCK Arrblock[5];

	Stack = IoGetCurrentIrpStackLocation( pIrp );
	//获取IO控制码
	Code = Stack->Parameters.DeviceIoControl.IoControlCode;
	Status = STATUS_UNSUCCESSFUL;

//===========================================================================
	//创建多个线程试验
//===========================================================================
	if ( Code == IOCTL_CONTROL_CREATETHREAD ) {

		//初始化事件对象(初始化为未激发状态)
		KeInitializeEvent( &hEvent[0], NotificationEvent, FALSE );
		KeInitializeEvent( &hEvent[1], NotificationEvent, FALSE );

		//保证这个代码必须在Dispatch级别运行
		ASSERT( KeGetCurrentIrql() <= DISPATCH_LEVEL );

		//提升IRP
		KeRaiseIrql( DISPATCH_LEVEL, &OldIrp );

		//恢复先前的IRP
		KeLowerIrql( OldIrp );

		//创建线程(参数4指定是用户线程函数系统线程, 很明显这里是系统线程)
		Status = PsCreateSystemThread( &hThread[0], 0, NULL, NULL, NULL, &SysRoutine, &hEvent[0] );

		if ( !NT_SUCCESS( Status ) ) {
			KdPrint( ( "创建系统线程失败!\n" ) );
			return Status;
		}

		//创建线程(这里指定的是用户线程)
		Status = PsCreateSystemThread( &hThread[1], 0, NULL,
		                               NtCurrentProcess(), NULL, &UserRoutine, &hEvent[1] );

		if ( !NT_SUCCESS( Status ) ) {
			KdPrint( ( "创建系统线程失败!\n" ) );
			return Status;
		}

		//等待事件对象置位(不等待, ThreadRoutine用了栈上的变量, 要崩溃
		KeWaitForSingleObject( &hEvent[0], Executive, KernelMode, FALSE, NULL );
		KeWaitForSingleObject( &hEvent[1], Executive, KernelMode, FALSE, NULL );
		//等待所有对象位置位才返回
		//KeWaitForMultipleObjects(2, (PVOID)hEvent, WaitAll, Executive, \
		//	KernelMode, FALSE, 0, &Arrblock[0] );

		//清除同步对象
		KeClearEvent( &hEvent[0] );
		KeClearEvent( &hEvent[1] );
		Status = STATUS_SUCCESS;

		KdPrint( ( "IOCTL_CONTROL_CREATETHREAD 控制码成功执行!\n" ) );
//===========================================================================
//		利用用户层创建的同步对象进行同步
//===========================================================================
	} else if ( Code == IOCTL_CONTROL_SHOREEVENT ) {

		hUserEvent = *( HANDLE* )pIrp->AssociatedIrp.SystemBuffer;

		if ( hUserEvent ) {
			//将用户层传入的句柄转换成指针
			Status  = ObReferenceObjectByHandle( hUserEvent, EVENT_MODIFY_STATE,
				*ExEventObjectType, KernelMode,	( PVOID* )&phEvent, NULL );

			if ( !NT_SUCCESS( Status ) ) {
				KdPrint( ( "转换用户层句柄失败!\n" ) );
			} else {

				KeSetEvent( phEvent, IO_NO_INCREMENT, FALSE );
				KdPrint( ( "成功将用户层传入的句柄转换为指针, 并置位" ) );
			}
			Status = STATUS_SUCCESS;
		} else {
			Status = STATUS_UNSUCCESSFUL;
		}

		KdPrint( ( "IOCTL_CONTROL_SHOREEVENT 控制码成功执行!\n" ) );

	}

//---------------------------------------------------------------------------
	pIrp->IoStatus.Information = 0;
	pIrp->IoStatus.Status = Status;
	IoCompleteRequest( pIrp, IO_NO_INCREMENT );
	return Status;
}
//===========================================================================
//	驱动程序的入口
#pragma code_seg( "INIT" )
NTSTATUS DriverEntry( PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pUSzRegPath ) {
	ULONG i;
	NTSTATUS Status;
	PDEVICE_OBJECT pDeviceObj;
	PDEVICE_EXT pDeviceExt;
	UNICODE_STRING uSzDeviceName = RTL_CONSTANT_STRING( DEVICE_NAME );
	UNICODE_STRING uSzSysLinkName = RTL_CONSTANT_STRING( SYS_LINK_NAME );

	//创建设备
	Status = IoCreateDevice( pDriverObj, sizeof( DEVICE_EXT ), &uSzDeviceName,
	                         FILE_DEVICE_UNKNOWN, 0, TRUE, &pDeviceObj );

	if ( !NT_SUCCESS( Status ) ) {
		KdPrint( ( "创建设备失败!\n" ) );
		return Status;
	}

	//创建符号链接
	Status = IoCreateSymbolicLink( &uSzSysLinkName, &uSzDeviceName );

	if ( !NT_SUCCESS( Status ) ) {
		KdPrint( ( "创建符号链接失败!\n" ) );
		return Status;
	}

	//设置设备属性, 和设备扩展内容
	pDeviceExt = ( PDEVICE_EXT )pDeviceObj->DeviceExtension;
	pDeviceExt->pDeviceObj = pDeviceObj;
	pDeviceExt->uSzDeviceName = uSzDeviceName;
	pDeviceExt->uSzSysLinkName = uSzSysLinkName;
	pDeviceObj->Flags |= DO_BUFFERED_IO;

	for( i = 0; i < IRP_MJ_MAXIMUM_FUNCTION; i++ ) {
		pDriverObj->MajorFunction[i] = &DispatchRoutine;
	}

	pDriverObj->DriverUnload = &DriverUnload;
	pDriverObj->MajorFunction[IRP_MJ_DEVICE_CONTROL] = &DispatchControl;
	return Status;
}

 

发表评论

您必须登录 才能进行评论。