接上篇，既然要在驱动中使用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();elsem_spin_lock->Lock(m_old_irql); } ~CSpinlockGuard() {if (m_current_irql >= DISPATCH_LEVEL)m_spin_lock->UnlockDpc();elsem_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);.....}
【Windows驱动锁的C++简单封装】
推荐阅读
- 一文让你搞懂MYSQL底层原理。-内部结构、索引、锁、集群
- 好用到飞起！微软复活超强Windows工具集，网友：做得好
- 基于Redis实现的分布式锁知识点总结
- 瓜子|阿娇瘦身成功，45度角自拍露小蛮腰，锁骨凹陷：像换了一个人
- 无需重装WINDOWS将计算机恢复到初始状态
- 终于搞懂分布式锁是什么了
- 磁吸锁关不上 磁吸锁锁不住怎么回事
- 我对 MySQL 锁、事务、MVCC 的一些认识
- dead-lock 什么是死锁？如何避免死锁详解
- Kali和Windows用户密码管理的异同