C#不提升自己程序的權限實現操作註冊表

Posted on by

1. 緒論

當我們編寫瞭自己的C#程序，有程序自定義的文件類型時，通常希望它滿足以下需求：

雙擊自定義文件打開自定義程序 自定義文件有著自己的圖標

此時，在網上檢索可以發現，大多數回答是使用Microsoft.Win32下的CreateSubKey(String)函數，但是很不幸，Win10的註冊表項受訪問控制列表(ACL)保護。特別是想要實現上述兩個需求時，寫入HKEY_CLASSES_ROOT，程序會報錯。

這時，我們可能會進一步提高程序運行時的權限，例如，使用管理員權限啟動程序，並且有各種各樣的方式。雖然可行，但是某些情況下，是需要以非管理員權限執行的，這時又要降級，沒有必要的反反復復，著實累！而且會使得用戶心存疑慮。

受到HandyControl的源碼啟發，可以使用以下的方式，實現上述兩個需求，同時不用提升自己程序的運行權限，從而免去瞭一系列的麻煩。

2. 主體思路

查看它的源碼，一言以蔽之：利用CMD，執行註冊表reg文件，實現讀寫刪註冊表。具體的實現步驟為：

獲取當前程序主模塊的路徑 檢測路徑下是否存在reg文件，如果有，則退出；如果沒有，則認為是第一次啟動，註冊表內沒有寫入想要的信息，繼續執行以下步驟 讀取準備的txt文件（含操作註冊表的內容） 將txt中操作註冊表內容的參數，根據需要替換賦值 寫入reg文件 cmd執行reg文件，自動彈出管理權限獲取窗口 “是”實現reg文件操作，“否”取消操作

但是上述步驟可以考慮以下的優化方向：

執行reg文件前，詢問用戶是否可以寫入註冊表以實現雙擊打開文件功能，變得用戶友好型 若用戶點擊否，該功能則再也無法出現。用戶想要實現雙擊打開文件功能，無從下手 若目錄下的reg被刪除，該功能再次出現，即使註冊表中已寫入信息

因此，我們在此優化為如圖所示的步驟：

3. C#實現

在程序屬性中，指定好使用的 ico 文件。

假設自定義的文件後綴名為：.mySuffix。那麼實現上述兩個需求，按照Saito Asuka的步驟可以手動實現。結合註冊表文件編寫方法，利用優化後的流程，即可程序實現。

3.1 檢測是否註冊

函數Registry.ClassesRoot.OpenSubKey(".mySuffix")可以讀取其中的名稱，如果沒有，返回null，實現變相的判斷是否存在。對於同一個後綴名，可能有著不同的程序實現，需要遍歷所有的值。在關聯的值中，查看是否有 open/command 的值。Registry.ClassesRoot.OpenSubKey(path).GetValue(null)可以返回名稱對應的值。

        private bool IsRegistryExist(string suffix, string path)
        {
            try
            {
                using RegistryKey hkSoftWare = Registry.ClassesRoot.OpenSubKey(suffix);
                if (hkSoftWare == null) return false;

                // 獲取到該項下所有的名稱
                string[] sValueNameColl = hkSoftWare.GetValueNames();

                int len = sValueNameColl.Length;

                // 獲取到所有名稱對應的數據
                for (int i = 0; i < len; i++)
                {
                    string data = hkSoftWare.GetValue(sValueNameColl[i]).ToString();

                    if (string.Equals(data, string.Empty)) continue;

                    RegistryKey rk = Registry.ClassesRoot.OpenSubKey($"{data}\\shell\\open\\command");

                    if (rk == null) continue;

                    var commandData = rk.GetValue(null)?.ToString();

                    rk.Close();

                    if (commandData == null) return false;

                    if (string.Equals(commandData, string.Empty)) return false;

                    if (string.Equals(commandData, path)) return true;

                }
            }
            finally
            {

            }
            return false;
        }

3.2 替換參數寫入reg文件並執行

本程序基於 WPF ，因此獲取程序所在路徑使用的是Process.GetCurrentProcess().MainModule。在替換時，務必註意先後順序，想知道不按順序的後果，自己可以試驗一下😀。

        private void UpdateRegistry()
        {
			// 獲取程序運行路徑
            var processModule = Process.GetCurrentProcess().MainModule;

            if (processModule == null) return;

            if (IsRegistryExist(".mySuffix", processModule.FileName)) return;

            var processWithSuffix = processModule.ModuleName.Split('.')[0] + ".mySuffix";

            var registryFilePath = $"{Path.GetDirectoryName(processModule.FileName)}\\Registry.reg";
            if (!File.Exists(registryFilePath))
            {
                string registryStr =
                    "Windows Registry Editor Version 5.00\r\n" +
                    "\r\n" +
                    "[HKEY_CLASSES_ROOT\\##]\r\n" +
                    "@=\"###\"\r\n" +
                    "\r\n" +
                    "[HKEY_CLASSES_ROOT\\###\\DefaultIcon]\r\n" +
                    "@=\"#\"\r\n" +
                    "\r\n" +
                    "[HKEY_CLASSES_ROOT\\###\\shell]\r\n" +
                    "\r\n" +
                    "[HKEY_CLASSES_ROOT\\###\\shell\\open]\r\n" +
                    "\r\n" +
                    "[HKEY_CLASSES_ROOT\\###\\shell\\open\\command]\r\n" +
                    "@=\"#\"" + "\r\n";


                // 替換
                var newRegistryStr = registryStr.Replace("###", processWithSuffix).Replace("##", ".mySuffix").Replace("#", processModule.FileName.Replace("\\", "\\\\"));

                File.WriteAllText(registryFilePath, newRegistryStr);

            }

            Process.Start(new ProcessStartInfo("cmd", $"/c {registryFilePath}")
            {
                UseShellExecute = false,
                CreateNoWindow = true
            });

        }

3.3 更新註冊表

想要實現程序第一次啟動後，僅註冊寫入一次，將上述函數，放置在OnStartup(StartupEventArgs e)函數中。

protected override void OnStartup(StartupEventArgs e)
        {
            base.OnStartup(e);


            UpdateRegistry();

        }

4. 總結

在桌面新建自定義文件後，可以實現自定義的 ico 圖標（與第3節中，程序指定的ico文件圖標相同），雙擊後，也可以實現打開自定義程序。

需要“管理員身份”才能運行的軟件，99％ 都是流氓軟件

到此這篇關於C#不提升自己程序的權限實現操作註冊表的文章就介紹到這瞭,更多相關C#不提權實現操作註冊表內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：