Chinaunix首页 | 论坛 | 博客
  • 博客访问: 196137
  • 博文数量: 106
  • 博客积分: 3810
  • 博客等级: 中校
  • 技术积分: 1007
  • 用 户 组: 普通用户
  • 注册时间: 2009-03-18 13:35
文章分类

全部博文(106)

文章存档

2014年(17)

2011年(5)

2010年(75)

2009年(9)

我的朋友

分类:

2010-04-29 22:30:46

'*************************************************************************
'
**模 块 名:ModRW_Reg
'
**创 建 人:嗷嗷叫的老马
'
**日    期:2003年11月17日
'
**描    述:本模块是有关注册表操作的
'
**版    本:V1.0
'
*************************************************************************
'
'
使用示例:

'新建串值
'
SetStringValue "HKEY_LOCAL_MACHINE", "String Value", "Hello Visual Basic programmer"

'SetStringValue "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\Run", "Internat", "internat.exe"

'新建二进制值
'
SetBinaryValue "HKEY_LOCAL_MACHINE", "Binary Value", Chr$(&H1) + Chr$(&H2) + Chr$(&H3) + Chr$(&H4)

'新建 DWORD 值
'
SetDWORDValue "HKEY_LOCAL_MACHINE", "DWORD Value", "1"

'读取串值
'
GetStringValue("HKEY_LOCAL_MACHINE", "String Value")

'读取二进制值
'
GetBinaryValue("HKEY_LOCAL_MACHINE", "Binary Value")
'
If rtn = Chr$(&H1) + Chr$(&H2) + Chr$(&H3) + Chr$(&H4) Then

'读取 DWORD 值
'
GetDWORDValue("HKEY_LOCAL_MACHINE", "DWORD Value")

'删除键值
'
DelValue("HKEY_LOCAL_MACHINE", "String Value")

'新建主键
'
CreateKey "HKEY_LOCAL_MACHINE\Registry Editor"

'删除主键
'
DeleteKey "HKEY_LOCAL_MACHINE\Registry Editor"

Type FILETIME
    lLowDateTime
As Long
    lHighDateTime
As Long
End Type

Private Declare Function RegOpenKeyEx Lib "advapi32.dll" Alias "RegOpenKeyExA" _
        (ByVal hKey
As Long, _
        ByVal lpSubKey
As String, _
        ByVal ulOptions
As Long, _
        ByVal samDesired
As Long, phkResult As Long) As Long
Private Declare Function RegCloseKey Lib "advapi32.dll" _
        (ByVal hKey
As Long) As Long
Private Declare Function RegCreateKey Lib "advapi32.dll" Alias "RegCreateKeyA" _
        (ByVal hKey
As Long, _
        ByVal lpSubKey
As String, phkResult As Long) As Long
Private Declare Function RegDeleteKey Lib "advapi32.dll" Alias "RegDeleteKeyA" _
        (ByVal hKey
As Long, _
        ByVal lpSubKey
As String) As Long
Private Declare Function RegQueryValueEx Lib "advapi32.dll" Alias "RegQueryValueExA" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String, _
        ByVal lpReserved
As Long, lpType As Long, _
        ByVal lpData
As String, lpcbData As Long) As Long
Private Declare Function RegQueryValueExA Lib "advapi32.dll" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String, _
        ByVal lpReserved
As Long, lpType As Long, _
        ByRef lpData
As Long, lpcbData As Long) As Long
Private Declare Function RegSetValueEx Lib "advapi32.dll" Alias "RegSetValueExA" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String, _
        ByVal Reserved
As Long, _
        ByVal dwType
As Long, _
        ByVal lpData
As String, _
        ByVal cbData
As Long) As Long
Private Declare Function RegSetValueExA Lib "advapi32.dll" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String, _
        ByVal Reserved
As Long, _
        ByVal dwType
As Long, _
        ByRef lpData
As Long, _
        ByVal cbData
As Long) As Long
Private Declare Function RegSetValueExB Lib "advapi32.dll" Alias "RegSetValueExA" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String, _
        ByVal Reserved
As Long, _
        ByVal dwType
As Long, _
        ByRef lpData
As Byte, _
        ByVal cbData
As Long) As Long
Private Declare Function RegDeleteValue Lib "advapi32.dll" Alias "RegDeleteValueA" _
        (ByVal hKey
As Long, _
        ByVal lpValueName
As String) As Long

Const ERROR_SUCCESS = 0&
Const ERROR_BADDB = 1009&
Const ERROR_BADKEY = 1010&
Const ERROR_CANTOPEN = 1011&
Const ERROR_CANTREAD = 1012&
Const ERROR_CANTWRITE = 1013&
Const ERROR_OUTOFMEMORY = 14&
Const ERROR_INVALID_PARAMETER = 87&
Const ERROR_ACCESS_DENIED = 5&
Const ERROR_NO_MORE_ITEMS = 259&
Const ERROR_MORE_DATA = 234&

Const REG_NONE = 0&
Const REG_SZ = 1&
Const REG_EXPAND_SZ = 2&
Const REG_BINARY = 3&
Const REG_DWORD = 4&
Const REG_DWORD_LITTLE_ENDIAN = 4&
Const REG_DWORD_BIG_ENDIAN = 5&
Const REG_LINK = 6&
Const REG_MULTI_SZ = 7&
Const REG_RESOURCE_LIST = 8&
Const REG_FULL_RESOURCE_DESCRIPTOR = 9&
Const REG_RESOURCE_REQUIREMENTS_LIST = 10&

Const KEY_QUERY_VALUE = &H1&
Const KEY_SET_VALUE = &H2&
Const KEY_CREATE_SUB_KEY = &H4&
Const KEY_ENUMERATE_SUB_KEYS = &H8&
Const KEY_NOTIFY = &H10&
Const KEY_CREATE_LINK = &H20&
Const READ_CONTROL = &H20000
Const WRITE_DAC = &H40000
Const WRITE_OWNER = &H80000
Const SYNCHRONIZE = &H100000
Const STANDARD_RIGHTS_REQUIRED = &HF0000
Const STANDARD_RIGHTS_READ = READ_CONTROL
Const STANDARD_RIGHTS_WRITE = READ_CONTROL
Const STANDARD_RIGHTS_EXECUTE = READ_CONTROL
Const KEY_READ = STANDARD_RIGHTS_READ Or KEY_QUERY_VALUE Or KEY_ENUMERATE_SUB_KEYS Or KEY_NOTIFY
Const KEY_WRITE = STANDARD_RIGHTS_WRITE Or KEY_SET_VALUE Or KEY_CREATE_SUB_KEY
Const KEY_EXECUTE = KEY_READ

Dim hKey As Long, MainKeyHandle As Long
Dim rtn As Long, lBuffer As Long, sBuffer As String
Dim lBufferSize As Long
Dim lDataSize As Long
Dim ByteArray() As Byte

'This constant determins wether or not to display error messages to the
'
user. I have set the default value to False as an error message can and
'
does become irritating after a while. Turn this value to true if you want
'
to debug your programming code when reading and writing to your system
'
registry, as any errors will be displayed in a message box.

Const DisplayErrorMsg = False


Function SetDWORDValue(SubKey As String, Entry As String, Value As Long)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key was open successfully then
            rtn = RegSetValueExA(hKey, Entry, 0, REG_DWORD, Value, 4)    'write the value
            If Not rtn = ERROR_SUCCESS Then    'if there was an error writting the value
                If DisplayErrorMsg = True Then    'if the user want errors displayed
                    MsgBox ErrorMsg(rtn)    'display the error
                End If
           
End If
            rtn
= RegCloseKey(hKey)    'close the key
        Else    'if there was an error opening the key
            If DisplayErrorMsg = True Then    'if the user want errors displayed
                MsgBox ErrorMsg(rtn)    'display the error
            End If
       
End If
   
End If

End Function
Function GetDWORDValue(SubKey As String, Entry As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key could be opened then
            rtn = RegQueryValueExA(hKey, Entry, 0, REG_DWORD, lBuffer, 4)    'get the value from the registry
            If rtn = ERROR_SUCCESS Then    'if the value could be retreived then
                rtn = RegCloseKey(hKey)    'close the key
                GetDWORDValue = lBuffer    'return the value
            Else    'otherwise, if the value couldnt be retreived
                GetDWORDValue = "Error"    'return Error to the user
                If DisplayErrorMsg = True Then    'if the user wants errors displayed
                    MsgBox ErrorMsg(rtn)    'tell the user what was wrong
                End If
           
End If
       
Else    'otherwise, if the key couldnt be opened
            GetDWORDValue = "Error"    'return Error to the user
            If DisplayErrorMsg = True Then    'if the user wants errors displayed
                MsgBox ErrorMsg(rtn)    'tell the user what was wrong
            End If
       
End If
   
End If

End Function
'第二部分
Function SetBinaryValue(SubKey As String, Entry As String, Value As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key was open successfully then
            lDataSize = LenB(StrConv(Value, vbFromUnicode))
           
ReDim ByteArray(lDataSize)
           
For I = 1 To lDataSize
                ByteArray(I)
= Asc(Mid$(Value, I, 1))
           
Next
            rtn
= RegSetValueExB(hKey, Entry, 0, REG_BINARY, ByteArray(1), lDataSize)    'write the value
            If Not rtn = ERROR_SUCCESS Then    'if the was an error writting the value
                If DisplayErrorMsg = True Then    'if the user want errors displayed
                    MsgBox ErrorMsg(rtn)    'display the error
                End If
           
End If
            rtn
= RegCloseKey(hKey)    'close the key
        Else    'if there was an error opening the key
            If DisplayErrorMsg = True Then    'if the user wants errors displayed
                MsgBox ErrorMsg(rtn)    'display the error
            End If
       
End If
   
End If

End Function


Function GetBinaryValue(SubKey As String, Entry As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key could be opened
            lBufferSize = 1
            rtn
= RegQueryValueEx(hKey, Entry, 0, REG_BINARY, 0, lBufferSize)    'get the value from the registry
            sBuffer = Space(lBufferSize)
            rtn
= RegQueryValueEx(hKey, Entry, 0, REG_BINARY, sBuffer, lBufferSize)    'get the value from the registry
            If rtn = ERROR_SUCCESS Then    'if the value could be retreived then
                rtn = RegCloseKey(hKey)    'close the key
                GetBinaryValue = sBuffer    'return the value to the user
            Else    'otherwise, if the value couldnt be retreived
                GetBinaryValue = "Error"    'return Error to the user
                If DisplayErrorMsg = True Then    'if the user wants to errors displayed
                    MsgBox ErrorMsg(rtn)    'display the error to the user
                End If
           
End If
       
Else    'otherwise, if the key couldnt be opened
            GetBinaryValue = "Error"    'return Error to the user
            If DisplayErrorMsg = True Then    'if the user wants to errors displayed
                MsgBox ErrorMsg(rtn)    'display the error to the user
            End If
       
End If
   
End If

End Function
Function DeleteKey(Keyname As String)

   
Call ParseKey(Keyname, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, Keyname, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key could be opened then
            rtn = RegDeleteKey(hKey, Keyname)    'delete the key
            rtn = RegCloseKey(hKey)    'close the key
        End If
   
End If

End Function

Function GetMainKeyHandle(MainKeyName As String) As Long

Const HKEY_CLASSES_ROOT = &H80000000
Const HKEY_CURRENT_USER = &H80000001
Const HKEY_LOCAL_MACHINE = &H80000002
Const HKEY_USERS = &H80000003
Const HKEY_PERFORMANCE_DATA = &H80000004
Const HKEY_CURRENT_CONFIG = &H80000005
Const HKEY_DYN_DATA = &H80000006

   
Select Case MainKeyName
   
Case "HKEY_CLASSES_ROOT"
        GetMainKeyHandle
= HKEY_CLASSES_ROOT
   
Case "HKEY_CURRENT_USER"
        GetMainKeyHandle
= HKEY_CURRENT_USER
   
Case "HKEY_LOCAL_MACHINE"
        GetMainKeyHandle
= HKEY_LOCAL_MACHINE
   
Case "HKEY_USERS"
        GetMainKeyHandle
= HKEY_USERS
   
Case "HKEY_PERFORMANCE_DATA"
        GetMainKeyHandle
= HKEY_PERFORMANCE_DATA
   
Case "HKEY_CURRENT_CONFIG"
        GetMainKeyHandle
= HKEY_CURRENT_CONFIG
   
Case "HKEY_DYN_DATA"
        GetMainKeyHandle
= HKEY_DYN_DATA
   
End Select

End Function

Function ErrorMsg(lErrorCode As Long) As String

   
'If an error does accurr, and the user wants error messages displayed, then
    'display one of the following error messages

   
Select Case lErrorCode
   
Case 1009, 1015
        GetErrorMsg
= "The Registry Database is corrupt!"
   
Case 2, 1010
        GetErrorMsg
= "Bad Key Name"
   
Case 1011
        GetErrorMsg
= "Can't Open Key"
   
Case 4, 1012
        GetErrorMsg
= "Can't Read Key"
   
Case 5
        GetErrorMsg
= "Access to this key is denied"
   
Case 1013
        GetErrorMsg
= "Can't Write Key"
   
Case 8, 14
        GetErrorMsg
= "Out of memory"
   
Case 87
        GetErrorMsg
= "Invalid Parameter"
   
Case 234
        GetErrorMsg
= "There is more data than the buffer has been allocated to hold."
   
Case Else
        GetErrorMsg
= "Undefined Error Code: " & Str$(lErrorCode)
   
End Select

End Function

Function ClearKK(Str As String)

Dim I As Long

Dim K As String

   
For I = 1 To Len(Str)

        K
= Mid(Str, I, 1)

       
If K = Chr(0) Then

            Str
= Mid(Str, 1, I)

           
Exit For

       
End If

   
Next

End Function

Function GetStringValue(SubKey As String, Entry As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key could be opened then
            sBuffer = Space(255)    'make a buffer
            lBufferSize = LenB(StrConv(sBuffer, vbFromUnicode))
            rtn
= RegQueryValueEx(hKey, Entry, 0, REG_SZ, sBuffer, lBufferSize)    'get the value from the registry
            If rtn = ERROR_SUCCESS Then    'if the value could be retreived then
                rtn = RegCloseKey(hKey)    'close the key
                sBuffer = Trim(sBuffer)
                ClearKK sBuffer
                GetStringValue
= Trim(Left(sBuffer, LenB(StrConv(sBuffer, vbFromUnicode)) - 1))    'return the value to the user
            Else    'otherwise, if the value couldnt be retreived
                GetStringValue = "Error"    'return Error to the user
                If DisplayErrorMsg = True Then    'if the user wants errors displayed then
                    MsgBox ErrorMsg(rtn)    'tell the user what was wrong
                End If
           
End If
       
Else    'otherwise, if the key couldnt be opened
            GetStringValue = "Error"    'return Error to the user
            If DisplayErrorMsg = True Then    'if the user wants errors displayed then
                MsgBox ErrorMsg(rtn)    'tell the user what was wrong
            End If
       
End If
   
End If

End Function

Private Sub ParseKey(Keyname As String, Keyhandle As Long)

    rtn
= InStr(Keyname, "\")    'return if "\" is contained in the Keyname

   
If Left(Keyname, 5) <> "HKEY_" Or Right(Keyname, 1) = "\" Then    'if the is a "\" at the end of the Keyname then
        MsgBox "Incorrect Format:" + Chr(10) + Chr(10) + Keyname    'display error to the user
        Exit Sub    'exit the procedure
    ElseIf rtn = 0 Then    'if the Keyname contains no "\"
        Keyhandle = GetMainKeyHandle(Keyname)
        Keyname
= ""    'leave Keyname blank
    Else    'otherwise, Keyname contains "\"
        Keyhandle = GetMainKeyHandle(Left(Keyname, rtn - 1))    'seperate the Keyname
        Keyname = Right(Keyname, LenB(StrConv(Keyname, vbFromUnicode)) - rtn)
   
End If

End Sub
Function CreateKey(SubKey As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'create the key
        If rtn = ERROR_SUCCESS Then    'if the key was created then
            rtn = RegCloseKey(hKey)    'close the key
        End If
   
End If

End Function
'第三部分,完结.

Function SetStringValue(SubKey As String, Entry As String, Value As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key was open successfully then
            rtn = RegSetValueEx(hKey, Entry, 0, REG_SZ, ByVal Value, LenB(StrConv(Value, vbFromUnicode)))    'write the value
            If Not rtn = ERROR_SUCCESS Then    'if there was an error writting the value
                If DisplayErrorMsg = True Then    'if the user wants errors displayed
                    MsgBox ErrorMsg(rtn)    'display the error
                End If
           
End If
            rtn
= RegCloseKey(hKey)    'close the key
        Else    'if there was an error opening the key
            If DisplayErrorMsg = True Then    'if the user wants errors displayed
                MsgBox ErrorMsg(rtn)    'display the error
            End If
       
End If
   
End If

End Function
Function DelValue(SubKey As String, Entry As String)

   
Call ParseKey(SubKey, MainKeyHandle)

   
If MainKeyHandle Then
        rtn
= RegCreateKey(MainKeyHandle, SubKey, hKey)    'open the key
        If rtn = ERROR_SUCCESS Then    'if the key could be opened then
            rtn = RegDeleteValue(hKey, Entry)
       
End If
   
End If
End Function
阅读(457) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~