📅  最后修改于: 2023-12-03 15:26:20.800000             🧑  作者: Mango
有时候我们需要在 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 的更多内容,请参阅 官方文档。