registryaccess.h

完整的基于Conxant平台的USB电视棒的WIN驱动程序。

/*+++ *******************************************************************\ 
* 
*  Copyright and Disclaimer: 
*  
*     --------------------------------------------------------------- 
*     This software is provided "AS IS" without warranty of any kind, 
*     either expressed or implied, including but not limited to the 
*     implied warranties of noninfringement, merchantability and/or 
*     fitness for a particular purpose.
*     --------------------------------------------------------------- 
*   
*     Copyright (c) 2008 Conexant Systems, Inc. 
*     All rights reserved. 
*
\******************************************************************* ---*/ 

#ifndef _REGISTRYACCESS_H_
#define _REGISTRYACCESS_H_

extern "C"
{
    #include <wdm.h>
}

#include <windef.h>

class RegistryAccess
{
public:
    RegistryAccess(PDEVICE_OBJECT pdo);
    ~RegistryAccess();

    // p_value receives the DWORD value, remains 
    // unchanged if the value didn't exist.
    NTSTATUS readDword(PCHAR name, PDWORD p_value);
    NTSTATUS writeDword(PCHAR name, DWORD value);

    NTSTATUS readDword(PWCHAR keyname, PCHAR name, PDWORD p_value);

protected:
    HANDLE openDriverKey(
        PWCHAR key_name, 
        ACCESS_MASK desired_access, 
        PDEVICE_OBJECT pdo);

    NTSTATUS getValue(
        const PUNICODE_STRING key_name,
        PCHAR data,
        ULONG data_length);

private:
    PDEVICE_OBJECT _pdo;
    HANDLE _key_handle;
};



#endif