📜  暂停 powershell 脚本直到按键 - Shell-Bash (1)

📅  最后修改于: 2023-12-03 15:26:20.800000             🧑  作者: Mango

暂停 PowerShell 脚本直到按键 - Shell-Bash

有时候我们需要在 PowerShell 脚本中暂停执行直到用户按下某个键。这在调试或需要手动干预的场景下非常有用。在本教程中,我们将学习如何使用 PowerShell 在 Windows 上实现这一目标。

实现过程

在 PowerShell 中,我们可以使用 Read-Host 命令来等待用户的输入。但是,如果我们需要在未按下任何键的情况下阻塞脚本的执行,我们就需要一种不阻塞电脑资源的方法。

下面是一个基于 Windows API 的解决方案:

Add-Type -TypeDefinition @"
    using System;
    using System.Runtime.InteropServices;
    public class KbHook {
        const int WH_KEYBOARD_LL = 13;
        const int WM_KEYDOWN = 0x0100;
        const int WM_SYSKEYDOWN = 0x0104;
        private static LowLevelKeyboardProc _proc = HookCallback;
        private static IntPtr _hookID = IntPtr.Zero;
        public static void SetHook()
        {
            using (Process curProcess = Process.GetCurrentProcess())
            using (ProcessModule curModule = curProcess.MainModule)
            {
                _hookID = SetWindowsHookEx(WH_KEYBOARD_LL, _proc,
                    GetModuleHandle(curModule.ModuleName), 0);
            }
        }
        public static void Unhook()
        {
            UnhookWindowsHookEx(_hookID);
        }
        private delegate IntPtr LowLevelKeyboardProc(
            int nCode, IntPtr wParam, IntPtr lParam);
        private static IntPtr HookCallback(
            int nCode, IntPtr wParam, IntPtr lParam)
        {
            if (nCode >= 0 &&
                (wParam == (IntPtr)WM_KEYDOWN || wParam == (IntPtr)WM_SYSKEYDOWN))
            {
                Console.WriteLine("Key pressed");
                Unhook();
            }
            return CallNextHookEx(_hookID, nCode, wParam, lParam);
        }
        [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern IntPtr SetWindowsHookEx(int idHook,
            LowLevelKeyboardProc lpfn, IntPtr hMod, uint dwThreadId);
        [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
        [return: MarshalAs(UnmanagedType.Bool)]
        private static extern bool UnhookWindowsHookEx(IntPtr hhk);
        [DllImport("user32.dll", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern IntPtr CallNextHookEx(IntPtr hhk, int nCode,
            IntPtr wParam, IntPtr lParam);
        [DllImport("kernel32.dll", CharSet = CharSet.Auto, SetLastError = true)]
        private static extern IntPtr GetModuleHandle(string lpModuleName);
    }
"@

这段代码定义了一个 KbHook 类,它使用 SetWindowsHookEx 函数来安装键盘钩子,等待用户按下任意键,并在用户按下了键之后卸载钩子。你只需要调用 KbHook::SetHook() 就可以让它开始工作了。

为了测试这个方法,我们可以添加以下代码:

Write-Host "Waiting for key press..."
KbHook::SetHook()
while ($true) {
    Start-Sleep -Milliseconds 10
}

这会让脚本持续运行,直到用户按下任意键。在用户按下键之后,脚本会打印出 "Key pressed",然后退出。

结论

在 PowerShell 中暂停执行直到按下按键是一个非常有用的技能,特别是在调试和手动干预的情况下。通过使用 Windows API 中的键盘钩子,我们可以实现一种不阻塞电脑资源的方法来等待键盘输入。关于 PowerShell 的更多内容,请参阅 官方文档