210 lines
8.1 KiB
C#
210 lines
8.1 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Runtime.InteropServices;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
using Microsoft.Win32;
|
||
|
||
namespace PCRobot.Utils
|
||
{
|
||
public class RegistryHelper
|
||
{
|
||
|
||
static readonly IntPtr HKEY_CLASSES_ROOT = new IntPtr(unchecked((int)0x80000000));
|
||
static readonly IntPtr HKEY_CURRENT_USER = new IntPtr(unchecked((int)0x80000001));
|
||
static readonly IntPtr HKEY_LOCAL_MACHINE = new IntPtr(unchecked((int)0x80000002));
|
||
static readonly IntPtr HKEY_USERS = new IntPtr(unchecked((int)0x80000003));
|
||
static readonly IntPtr HKEY_PERFORMANCE_DATA = new IntPtr(unchecked((int)0x80000004));
|
||
static readonly IntPtr HKEY_CURRENT_CONFIG = new IntPtr(unchecked((int)0x80000005));
|
||
static readonly IntPtr HKEY_DYN_DATA = new IntPtr(unchecked((int)0x80000006));
|
||
|
||
// 获取操作Key值句柄
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegOpenKeyEx(IntPtr hKey, string lpSubKey, uint ulOptions, int samDesired, out IntPtr phkResult);
|
||
|
||
//创建或打开Key值
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegCreateKeyEx(IntPtr hKey, string lpSubKey, int reserved, string type, int dwOptions, int REGSAM, IntPtr lpSecurityAttributes, out IntPtr phkResult,
|
||
out int lpdwDisposition);
|
||
|
||
//关闭注册表转向（禁用特定项的注册表反射）
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegDisableReflectionKey(IntPtr hKey);
|
||
|
||
//使能注册表转向（开启特定项的注册表反射）
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegEnableReflectionKey(IntPtr hKey);
|
||
|
||
//获取Key值（即：Key值句柄所标志的Key对象的值）
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegQueryValueEx(IntPtr hKey, string lpValueName, int lpReserved, out uint lpType, System.Text.StringBuilder lpData, ref uint lpcbData);
|
||
|
||
//设置Key值
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegSetValueEx(IntPtr hKey, string lpValueName, uint unReserved, uint unType, byte[] lpData, uint dataCount);
|
||
|
||
//关闭Key值
|
||
[DllImport("Advapi32.dll", CharSet = CharSet.Auto, SetLastError = true)]
|
||
private static extern int RegCloseKey(IntPtr hKey);
|
||
|
||
public static IntPtr TransferKeyName(string keyName)
|
||
{
|
||
IntPtr ret = IntPtr.Zero;
|
||
switch (keyName)
|
||
{
|
||
case "HKEY_CLASSES_ROOT":
|
||
ret = HKEY_CLASSES_ROOT;
|
||
break;
|
||
case "HKEY_CURRENT_USER":
|
||
ret = HKEY_CURRENT_USER;
|
||
break;
|
||
case "HKEY_LOCAL_MACHINE":
|
||
ret = HKEY_LOCAL_MACHINE;
|
||
break;
|
||
case "HKEY_USERS":
|
||
ret = HKEY_USERS;
|
||
break;
|
||
case "HKEY_PERFORMANCE_DATA":
|
||
ret = HKEY_PERFORMANCE_DATA;
|
||
break;
|
||
case "HKEY_CURRENT_CONFIG":
|
||
ret = HKEY_CURRENT_CONFIG;
|
||
break;
|
||
case "HKEY_DYN_DATA":
|
||
ret = HKEY_DYN_DATA;
|
||
break;
|
||
default:
|
||
ret = HKEY_LOCAL_MACHINE;
|
||
break;
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
/// <summary>
|
||
/// 设置64位注册表
|
||
/// </summary>
|
||
/// <param name="key"></param>
|
||
/// <param name="subKey"></param>
|
||
/// <param name="name"></param>
|
||
/// <param name="value"></param>
|
||
/// <returns></returns>
|
||
public static int Set64BitRegistryKey(string key, string subKey, string name, string value)
|
||
{
|
||
int STANDARD_RIGHTS_ALL = (0x001F0000);
|
||
int KEY_QUERY_VALUE = (0x0001);
|
||
int KEY_SET_VALUE = (0x0002);
|
||
int KEY_CREATE_SUB_KEY = (0x0004);
|
||
int KEY_ENUMERATE_SUB_KEYS = (0x0008);
|
||
int KEY_NOTIFY = (0x0010);
|
||
int KEY_CREATE_LINK = (0x0020);
|
||
int SYNCHRONIZE = (0x00100000);
|
||
int KEY_WOW64_64KEY = (0x0100);
|
||
int REG_OPTION_NON_VOLATILE = (0x00000000);
|
||
int KEY_ALL_ACCESS = (STANDARD_RIGHTS_ALL | KEY_QUERY_VALUE | KEY_SET_VALUE | KEY_CREATE_SUB_KEY | KEY_ENUMERATE_SUB_KEYS
|
||
| KEY_NOTIFY | KEY_CREATE_LINK) & (~SYNCHRONIZE);
|
||
|
||
int ret = 0;
|
||
try
|
||
{
|
||
//将Windows注册表主键名转化成为不带正负号的整形句柄（与平台是32或者64位有关）
|
||
IntPtr hKey = TransferKeyName(key);
|
||
|
||
//声明将要获取Key值的句柄
|
||
IntPtr pHKey = IntPtr.Zero;
|
||
|
||
//获得操作Key值的句柄
|
||
int lpdwDisposition = 0;
|
||
ret = RegCreateKeyEx(hKey, subKey, 0, "", REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS | KEY_WOW64_64KEY, IntPtr.Zero, out pHKey, out lpdwDisposition);
|
||
if (ret != 0)
|
||
{
|
||
//Console.WriteLine(string.Format("Unable to create key {0} - {1}: {2}!", key, subKey, ret));
|
||
return ret;
|
||
}
|
||
|
||
//关闭注册表转向（禁止特定项的注册表反射）
|
||
RegDisableReflectionKey(pHKey);
|
||
|
||
//设置访问的Key值
|
||
uint REG_SZ = 1;
|
||
byte[] data = Encoding.Unicode.GetBytes(value);
|
||
|
||
RegSetValueEx(pHKey, name, 0, REG_SZ, data, (uint)data.Length);
|
||
|
||
//打开注册表转向（开启特定项的注册表反射）
|
||
RegEnableReflectionKey(pHKey);
|
||
|
||
RegCloseKey(pHKey);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//Console.WriteLine(ex.ToString());
|
||
return -1;
|
||
}
|
||
|
||
return ret;
|
||
}
|
||
|
||
public static void SetRegistryKey(string key, string subKey, string name, string value)
|
||
{
|
||
//Console.WriteLine("SetRegistryKey start.");
|
||
if (System.IntPtr.Size == 8)
|
||
{
|
||
// 写SOFTWARE\Huawei\VirtualDesktopAgent，需要关闭注册表重定向，再写64位路径的注册表
|
||
int ret = Set64BitRegistryKey(key, subKey, name, value);
|
||
if (ret != 0)
|
||
{
|
||
Console.WriteLine(string.Format("Failed to write Reg {0}\\{1}\\{2},return {3}", key, subKey, name, ret));
|
||
}
|
||
}
|
||
|
||
try
|
||
{
|
||
Microsoft.Win32.Registry.SetValue(key + "\\" + subKey, name, value);
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Console.WriteLine(ex.ToString());
|
||
}
|
||
|
||
//Console.WriteLine("SetRegistryKey exit.");
|
||
}
|
||
|
||
public static string GetRegistryValue(string path, string key)
|
||
{
|
||
RegistryKey regkey = null;
|
||
try
|
||
{
|
||
regkey = Registry.LocalMachine.OpenSubKey(path);
|
||
if (regkey == null)
|
||
{
|
||
//Console.WriteLine("Cannot find Registry path:" + path);
|
||
return null;
|
||
}
|
||
|
||
object val = regkey.GetValue(key);
|
||
if (val == null)
|
||
{
|
||
//Console.WriteLine("Cannot find Registry key:" + key);
|
||
return null;
|
||
}
|
||
|
||
return val.ToString();
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//Console.WriteLine(ex.ToString());
|
||
return null;
|
||
}
|
||
finally
|
||
{
|
||
if (regkey != null)
|
||
{
|
||
regkey.Close();
|
||
}
|
||
}
|
||
}
|
||
|
||
}
|
||
}
|