C#不提升自己程式的許可權實現操作登錄檔

Aidan_Lee發表於2022-12-07

1. 緒論

當我們編寫了自己的C#程式,有程式自定義的檔案型別時,通常希望它滿足以下需求:

  • 雙擊自定義檔案開啟自定義程式
  • 自定義檔案有著自己的圖示

此時,在網上檢索可以發現,大多數回答是使用Microsoft.Win32下的CreateSubKey(String)函式,但是很不幸,Win10的登錄檔項受訪問控制列表(ACL)保護。特別是想要實現上述兩個需求時,寫入HKEY_CLASSES_ROOT,程式會報錯。
C#不提升自己程式的許可權實現操作登錄檔

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

受到HandyControl原始碼啟發,可以使用以下的方式,實現上述兩個需求,同時不用提升自己程式的執行許可權,從而免去了一系列的麻煩。


2. 主體思路

檢視它的原始碼,一言以蔽之:利用CMD,執行登錄檔reg檔案,實現讀寫刪登錄檔。具體的實現步驟為:

  1. 獲取當前程式主模組的路徑
  2. 檢測路徑下是否存在reg檔案,如果有,則退出;如果沒有,則認為是第一次啟動,登錄檔內沒有寫入想要的資訊,繼續執行以下步驟
  3. 讀取準備的txt檔案(含操作登錄檔的內容)
  4. 將txt中操作登錄檔內容的引數,根據需要替換賦值
  5. 寫入reg檔案
  6. cmd執行reg檔案,自動彈出管理許可權獲取視窗
  7. “是”實現reg檔案操作,“否”取消操作

但是上述步驟可以考慮以下的最佳化方向:

  • 執行reg檔案前,詢問使用者是否可以寫入登錄檔以實現雙擊開啟檔案功能,變得使用者友好型
  • 若使用者點選否,該功能則再也無法出現。使用者想要實現雙擊開啟檔案功能,無從下手
  • 若目錄下的reg被刪除,該功能再次出現,即使登錄檔中已寫入資訊

因此,我們在此最佳化為如圖所示的步驟:
C#不提升自己程式的許可權實現操作登錄檔

3. C#實現

在程式屬性中,指定好使用的 ico 檔案。
C#不提升自己程式的許可權實現操作登錄檔

假設自定義的檔案字尾名為:.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檔案圖示相同),雙擊後,也可以實現開啟自定義程式。

image


  1. 需要“管理員身份”才能執行的軟體,99% 都是流氓軟體 ↩︎

相關文章