Chinaunix首页 | 论坛 | 博客
  • 博客访问: 1072980
  • 博文数量: 77
  • 博客积分: 11498
  • 博客等级: 上将
  • 技术积分: 1840
  • 用 户 组: 普通用户
  • 注册时间: 2006-05-04 11:10
文章分类

全部博文(77)

文章存档

2011年(1)

2010年(16)

2009年(5)

2008年(55)

分类: C/C++

2008-12-01 20:44:32

利用Windows SDK提供的虚存技术和SEH实现的稀疏数组。
类SparseArray是在《Windows核心编程(第5版)》P678页上的VMArray基础上修改而得,并提供了线程安全特性。

/**
* SparseArray.h
* @Author   Tu Yongce
* @Created  2008-11-27
* @Modified 2008-12-10
* @Version  0.2
* @Note: SparseArray类是在《Windows核心编程(第5版)》P678页上的VMArray基础上修改而得
*/

#ifndef SPARSE_ARRAY_H_INCLUDED
#define SPARSE_ARRAY_H_INCLUDED

#ifdef _MSC_VER
#pragma once
#endif

#include
#include
#include
#include "Noncopyable.h"
#include "LightWeightLock.h"

/**
* 稀疏数组,利用虚拟内存和SEH实现
* Note: 该类是线程安全的,但最好在构造第一个SparseArray对象之前(一般在主线程中进行)调用类的
*           静态函数EnsureThreadSecurity来确保内部锁被构造;
*           如果不调用,则在极小概率下可能会出现线程不安全问题,详见EnsureThreadSecurity函数说明
*/
template
class SparseArray: private Noncopyable
{
private:
    // 前一个未处理异常过滤函数指针
    static PTOP_LEVEL_EXCEPTION_FILTER sm_preFilter;

    static SparseArray *sm_head;    // SparseArray对象链表头指针
    SparseArray *m_next;            // 下一个SparseArray对象指针

    ElemType *m_array;    // 稀疏数组
    SIZE_T m_reserved;    // 数组预订区域大小(字节)

    // 获取用于在多线程中进行数据保护的锁
    static LightWeightLock& GetLock();

public:
    // 确保SparseArray类的线程安全
    static void EnsureThreadSecurity();

public:
    /**
    * 构造稀疏数组
    * @param size: 数组大小(元素个数)
    * @param failedThrow: 如果为true,则构造函数会在预定区域失败时抛出异常;
    *                                       否则需要用户自己检测SparseArray对象是否为NULL(是否失败)
    */
    explicit SparseArray(SIZE_T size, BOOL failedThrow = false);
    virtual ~SparseArray();

    // 使SparseArray可以像普通数组一样使用
    operator ElemType* ();
    operator const ElemType* () const;

    // SparseArray对象的异常过滤函数
    LONG ExceptionFilter(PEXCEPTION_POINTERS pep,
        BOOL bRetryUntilSuccessful = FALSE);

protected:
    // 访问异常的默认处理函数(派生类可重载此函数)
    virtual LONG OnAccessViolation(PVOID addr, BOOL read,
        PEXCEPTION_POINTERS pep, BOOL retryUntilSuccessful);

private:
    // SparseArray类的未处理异常过滤函数
    static LONG WINAPI MyUnhandledExceptionFilter(PEXCEPTION_POINTERS pep);
};

template
PTOP_LEVEL_EXCEPTION_FILTER SparseArray::sm_preFilter = NULL;

template
SparseArray* SparseArray::sm_head = NULL;

template
LightWeightLock& SparseArray::GetLock()
{
    // 用于在多线程中进行数据保护的锁
    static LightWeightLock lock;
    return lock;
}

/**
* 确保SparseArray类的线程安全
* 在构造SparseArray类对象之前,最好先调用此函数对类的内部锁进行构造(一般在主线程中进行)
* @Note: 如果不调用此函数,则在第一次构造SparseArray对象时可能出现线程安全问题(极小概率):
*               如果此时有多个SparseArray对象被同时构造,那么有可能内部锁在构造好之前被使用或被多次构造
*               但如果已经有SparseArray对象被成功构造后,则不再会有线程安全问题
* @Note: 不需要调用此函数的情况:
*              (1)有非局部静态对象的SparseArray静态对象(此时内部锁会被正确构造)
*              (2)能够确保第一个SparseArray对象被正确构造
*/
template
void SparseArray::EnsureThreadSecurity()
{
    // 保证内部锁被构造
    GetLock();
}

/**
* 构造稀疏数组
* @param size: 数组大小(元素个数)
* @param failedThrow: 如果为true,则构造函数会在预定区域失败时抛出异常;
*                                       否则需要用户自己检测SparseArray对象是否为NULL(是否失败)
*/
template
SparseArray::SparseArray(SIZE_T size, BOOL failedThrow)
{
    // 预订一块虚拟内存区域
    m_reserved = sizeof(ElemType) * size;
    m_array = (ElemType*) ::VirtualAlloc(NULL, m_reserved,
        MEM_RESERVE | MEM_TOP_DOWN, PAGE_READWRITE);

    if (m_array == NULL) {
        // 预订区域失败
        if (failedThrow)
            throw std::bad_alloc("SparseArray failed to reserve the virtual memory");
        return ;
    }

    // 把该SparseArray对象加入到链表中
    GetLock().Lock();  // 互斥访问SparseArray对象链表
    if (sm_head == NULL) {
        // 设置SparseArray使用的未处理异常过滤函数
        sm_preFilter = ::SetUnhandledExceptionFilter(MyUnhandledExceptionFilter);
    }
    m_next = sm_head;
    sm_head = this;
    GetLock().Unlock();
}

template
SparseArray::~SparseArray()
{
    if (m_array == NULL) {
        // 构造时没有成功预订区域
        return;
    }

    // 释放预订的虚拟内存区域
    ::VirtualFree(m_array, 0, MEM_RELEASE);

    // 从链表中移除该SparseArray对象
    GetLock().Lock();  // 互斥访问SparseArray对象链表
    SparseArray *p = sm_head;
    if (p == this) {
        sm_head = p->m_next;
    } else {
        for (; p->m_next != NULL; p = p->m_next) {
            if (p->m_next == this) {
                p->m_next = p->m_next->m_next;
                break;
            }
        }
    }
    GetLock().Unlock();
}

template
SparseArray::operator ElemType* ()
{
    return m_array;
}

template
SparseArray::operator const ElemType* () const
{
    return m_array;
}

// SparseArray类的未处理异常过滤函数
template
LONG WINAPI SparseArray::MyUnhandledExceptionFilter(PEXCEPTION_POINTERS pep)
{
    // 只处理访问异常
    if (pep->ExceptionRecord->ExceptionCode != EXCEPTION_ACCESS_VIOLATION)
        return EXCEPTION_CONTINUE_SEARCH;

    // 遍历SparseArray对象链表,尝试用每一个SparseArray对象来处理访问异常
    GetLock().Lock();  // 互斥访问SparseArray对象链表
    for (SparseArray* p = sm_head; p != NULL; p = p->m_next) {
        LONG retCode = p->ExceptionFilter(pep, TRUE);
        if (retCode != EXCEPTION_CONTINUE_SEARCH) {
            // 当前SparseArray对象对访问异常进行了处理
            return retCode;
        }
    }
    GetLock().Unlock();

    // 没有SparseArray对象对此访问异常进行处理,尝试用先前的未处理异常过滤程序
    return sm_preFilter(pep);
}

// SparseArray对象的异常过滤函数
template
LONG SparseArray::ExceptionFilter(PEXCEPTION_POINTERS pep, BOOL retry)
{
    // 只处理访问异常
    if (pep->ExceptionRecord->ExceptionCode != EXCEPTION_ACCESS_VIOLATION)
        return EXCEPTION_CONTINUE_SEARCH;

    // 获得访问异常发生位置和是否为“读”操作
    PVOID addr = (PVOID) pep->ExceptionRecord->ExceptionInformation[1];
    BOOL read = (pep->ExceptionRecord->ExceptionInformation[0] == 0);

    // 如果发生非法访问的内存地址在数组范围内,则尝试对其进行处理
    LONG retCode = EXCEPTION_CONTINUE_SEARCH;

    if (m_array <= addr && addr < (PBYTE)m_array + m_reserved) {
        retCode = OnAccessViolation(addr, read, pep, retry);
    }

    return retCode;
}

// 访问异常的默认处理函数(派生类可重载此函数)
template
LONG SparseArray::OnAccessViolation(PVOID addr, BOOL /*read*/,
                                              PEXCEPTION_POINTERS /*pep*/, BOOL retry)
{
    BOOL ok;
    do {
        // 尝试调拨内存
        ok = (::VirtualAlloc(addr, sizeof(ElemType), MEM_COMMIT,
            PAGE_READWRITE) != NULL);

        // 如果调拨内存失败且需要重试,则显示对话框提示用户释放内存
        if (!ok && retry) {
            ::MessageBox(NULL,
                _T("Please close some other applications and Press OK."),
                _T("Insufficient Memory Available"),
                MB_ICONWARNING | MB_OK);
        }
    } while (!ok && retry);

    // 如果内存调拨成功,则回到发生访问异常的地方重新执行;
    // 否则,返回EXCEPTION_EXECUTE_HANDLER让异常处理程序得到以执行
    return (ok ? EXCEPTION_CONTINUE_EXECUTION : EXCEPTION_EXECUTE_HANDLER);
}

#endif // SPARSE_ARRAY_H_INCLUDED

/**
* LightWeightLock.h
* @Author   Tu Yongce
* @Created  2008-11-28
* @Modified 2008-11-28
* @Version  0.1
*/

#ifndef LIGHT_WEIGHT_LOCK_H_INCLUDED
#define LIGHT_WEIGHT_LOCK_H_INCLUDED

#ifdef _MSC_VER
#pragma once
#endif

#include
#include "Noncopyable.h"

// 轻量级互斥锁(功能等同于MFC中的CCriticalSection)
class LightWeightLock: private Noncopyable
{
private:
    CRITICAL_SECTION m_cs;

public:
    LightWeightLock()
    {
        ::InitializeCriticalSection(&m_cs);
    }

    ~LightWeightLock()
    {
        ::DeleteCriticalSection(&m_cs);
    }

    void Lock()
    {
        ::EnterCriticalSection(&m_cs);
    }

    void Unlock()
    {
        ::LeaveCriticalSection(&m_cs);
    }
};

#endif // LIGHT_WEIGHT_LOCK_H_INCLUDED

阅读(2012) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~