您当前的位置:首页 > 电脑百科 > 程序开发 > 语言 > C/C++/C#

Windows驱动锁的C++简单封装

时间:2020-11-13 10:14:30  来源:  作者:

接上篇,既然要在驱动中使用C++,可以把内核中的锁操作封闭一下,通常不封闭的情况下,锁的获取和释放要成对出现,通过封装可以实现一行语句解决获取和释放的问题(类似应用层的lock_guard),代码奉上,有兴趣的可以自行扩展:

#pragma once

#include <ntddk.h>

class CMutexLock
{
public:
	CMutexLock()
		: m_inited(FALSE)
	{
	}
	~CMutexLock() {}

	void Init()
	{
		ExInitializeFastMutex(&m_mutex);
		m_inited = TRUE;
	}

	void Lock()
	{
		if (m_inited)
			ExAcquireFastMutex(&m_mutex);
	}

	void Unlock()
	{
		if (m_inited)
			ExReleaseFastMutex(&m_mutex);
	}

private:
	BOOLEAN m_inited;
	FAST_MUTEX m_mutex;
};

class CMutexGuard
{
public:
	CMutexGuard(CMutexLock* lock)
		: m_mutex_lock(lock)
	{
		m_mutex_lock->Lock();
	}

	~CMutexGuard()
	{
		m_mutex_lock->Unlock();
	}
private:
	CMutexLock* m_mutex_lock;
};

class CSpinLock
{
public:
	CSpinLock()
		: m_inited(FALSE)
	{
	}
	~CSpinLock() {}

	void Init()
	{
		KeInitializeSpinLock(&m_spin_lock);
		m_inited = TRUE;
	}

	void Lock(KIRQL& old_irql)
	{
		if (m_inited)
		{
			KeAcquireSpinLock(&m_spin_lock, &old_irql);
		}
	}

	void LockDpc()
	{
		if (m_inited)
		{
			KeAcquireSpinLockAtDpcLevel(&m_spin_lock);
		}
	}

	void Unlock(KIRQL prev_irql)
	{
		if (m_inited)
		{
			KeReleaseSpinLock(&m_spin_lock, prev_irql);
		}
	}

	void UnlockDpc()
	{
		if (m_inited)
		{
			KeReleaseSpinLockFromDpcLevel(&m_spin_lock);
		}
	}

private:
	BOOLEAN m_inited;
	KSPIN_LOCK m_spin_lock;
};

class CSpinlockGuard
{
public:
	CSpinlockGuard(CSpinLock* lock)
		: m_spin_lock(lock)
		, m_old_irql(PASSIVE_LEVEL)
	{
		m_current_irql = KeGetCurrentIrql();
		if (m_old_irql >= DISPATCH_LEVEL)
			m_spin_lock->LockDpc();
		else
			m_spin_lock->Lock(m_old_irql);
	}
	~CSpinlockGuard()
	{
		if (m_current_irql >= DISPATCH_LEVEL)
			m_spin_lock->UnlockDpc();
		else
			m_spin_lock->Unlock(m_old_irql);
	}

private:
	KIRQL m_old_irql;
	KIRQL m_current_irql;
	CSpinLock* m_spin_lock;
};

在代码中使用时只需类似如下操作即可:

CSpinLock g_test_lock;

g_test_lock.Init();

void TestFunc()

{
		CSpinlockGuard guard(&g_test_lock);
  
  	.....
}


Tags:C++   点击:()  评论:()
声明:本站部分内容及图片来自互联网,转载是出于传递更多信息之目的,内容观点仅代表作者本人,如有任何标注错误或版权侵犯请与我们联系(Email:2595517585@qq.com),我们将及时更正、删除,谢谢。
▌相关推荐
一、编程语言1.根据熟悉的语言,谈谈两种语言的区别?主要浅谈下C/C++和PHP语言的区别:1)PHP弱类型语言,一种脚本语言,对数据的类型不要求过多,较多的应用于Web应用开发,现在好多互...【详细内容】
2021-12-15  Tags: C++  点击:(17)  评论:(0)  加入收藏
函数调用约定(Calling Convention),是一个重要的基础概念,用来规定调用者和被调用者是如何传递参数的,既调用者如何将参数按照什么样的规范传递给被调用者。在参数传递中,有两个很...【详细内容】
2021-11-30  Tags: C++  点击:(19)  评论:(0)  加入收藏
一、为什么需要使用内存池在C/C++中我们通常使用malloc,free或new,delete来动态分配内存。一方面,因为这些函数涉及到了系统调用,所以频繁的调用必然会导致程序性能的损耗;另一...【详细内容】
2021-11-17  Tags: C++  点击:(37)  评论:(0)  加入收藏
C++编程中,你是否有为 我到底该写个struct还是class 而苦恼过?如果你到现在还不知道该如何选择,那么请求继续阅读,下文或许能给你些建议。问题的产生C++语言继承了 C语言的 stru...【详细内容】
2021-10-18  Tags: C++  点击:(61)  评论:(0)  加入收藏
C++在C的面向过程概念的基础上提供了面向对象和模板(泛型编程)的语法功能。下面以一个简单实例(动态数组的简单封装,包括下标的值可以是任意正数值,并提供边界检查)来说明C++是如...【详细内容】
2021-10-18  Tags: C++  点击:(49)  评论:(0)  加入收藏
0 前言Hello,大家好,欢迎来到『自由技艺』的 C++ 系列专题。代码重用,尽可能避免冗余代码是程序员的一项必备技能,今天就来给大家介绍其中一种:函数装饰器。在设计模式中,与它对应...【详细内容】
2021-09-28  Tags: C++  点击:(75)  评论:(0)  加入收藏
今天我们就来聊一聊 C++ 中的异常机制吧。在学校期间,我们很少会用得上异常机制。然而,工作之后,很多时候却不得不引入异常机制。因为一般情况下,使用函数的返回值来确定函数的...【详细内容】
2021-09-26  Tags: C++  点击:(181)  评论:(0)  加入收藏
一、内存泄漏(memory leak)内存泄漏(memory leak)是指由于疏忽或错误造成了程序未能释放掉不再使用的内存的情况。内存泄漏并非指内存在物理上的消失,而是应用程序分配某段内存...【详细内容】
2021-09-03  Tags: C++  点击:(104)  评论:(0)  加入收藏
stack容器#include <iostream>using namespace std;#include <stack>//容器头文件void test(){stack<int>p;p.push(100);p.push(1000);p.push(100);while(!p.empty()){cout<...【详细内容】
2021-08-17  Tags: C++  点击:(81)  评论:(0)  加入收藏
stl 常用遍历算法(for_each transform)示例代码(结论在结尾!!!!)#include<iostream>using namespace std;#include"vector"#include"map"#include"string"#include"list"#in...【详细内容】
2021-08-13  Tags: C++  点击:(89)  评论:(0)  加入收藏
▌简易百科推荐
一、简介很多时候我们都需要用到一些验证的方法,有时候需要用正则表达式校验数据时,往往需要到网上找很久,结果找到的还不是很符合自己想要的。所以我把自己整理的校验帮助类分...【详细内容】
2021-12-27  中年农码工    Tags:C#   点击:(1)  评论:(0)  加入收藏
引言在学习C语言或者其他编程语言的时候,我们编写的一个程序代码,基本都是在屏幕上打印出 hello world ,开始步入编程世(深)界(坑)的。C 语言版本的 hello world 代码:#include <std...【详细内容】
2021-12-21  一起学嵌入式    Tags:C 语言   点击:(10)  评论:(0)  加入收藏
读取SQLite数据库,就是读取一个路径\\192.168.100.**\position\db.sqlite下的文件<startup useLegacyV2RuntimeActivationPolicy="true"> <supportedRuntime version="v4.0"/...【详细内容】
2021-12-16  今朝我的奋斗    Tags:c#   点击:(21)  评论:(0)  加入收藏
什么是shellshell是c语言编写的程序,它在用户和操作系统之间架起了一座桥梁,用户可以通过这个桥梁访问操作系统内核服务。 它既是一种命令语言,同时也是一种程序设计语言,你可以...【详细内容】
2021-12-16  梦回故里归来    Tags:shell脚本   点击:(16)  评论:(0)  加入收藏
一、编程语言1.根据熟悉的语言,谈谈两种语言的区别?主要浅谈下C/C++和PHP语言的区别:1)PHP弱类型语言,一种脚本语言,对数据的类型不要求过多,较多的应用于Web应用开发,现在好多互...【详细内容】
2021-12-15  linux上的码农    Tags:c/c++   点击:(17)  评论:(0)  加入收藏
1.字符串数组+初始化char s1[]="array"; //字符数组char s2[6]="array"; //数组长度=字符串长度+1,因为字符串末尾会自动添&lsquo;\0&lsquo;printf("%s,%c\n",s1,s2[2]);...【详细内容】
2021-12-08  灯-灯灯    Tags:C语言   点击:(46)  评论:(0)  加入收藏
函数调用约定(Calling Convention),是一个重要的基础概念,用来规定调用者和被调用者是如何传递参数的,既调用者如何将参数按照什么样的规范传递给被调用者。在参数传递中,有两个很...【详细内容】
2021-11-30  小智雅汇    Tags:函数   点击:(19)  评论:(0)  加入收藏
一、问题提出问题:把m个苹果放入n个盘子中,允许有的盘子为空,共有多少种方法?注:5,1,1和1 5 1属同一种方法m,n均小于10二、算法分析设f(m,n) 为m个苹果,n个盘子的放法数目,则先对...【详细内容】
2021-11-17  C语言编程    Tags:C语言   点击:(46)  评论:(0)  加入收藏
一、为什么需要使用内存池在C/C++中我们通常使用malloc,free或new,delete来动态分配内存。一方面,因为这些函数涉及到了系统调用,所以频繁的调用必然会导致程序性能的损耗;另一...【详细内容】
2021-11-17  深度Linux    Tags:C++   点击:(37)  评论:(0)  加入收藏
OpenCV(Open Source Computer Vision Library)是一个(开源免费)发行的跨平台计算机视觉库,可以运行在Linux、Windows、Android、ios等操作系统上,它轻量级而且高效---由一系列...【详细内容】
2021-11-11  zls315    Tags:C#   点击:(50)  评论:(0)  加入收藏
最新更新
栏目热门
栏目头条