powershell_samples/close_handler.ps1 at master · sergueik/powershell_samples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# origin: https://stackoverflow.com/questions/8353581/how-to-handle-close-event-of-powershell-window-if-user-clicks-on-closex-butt

$code = @"

using System;
using System.Runtime.InteropServices;
using System.Management.Automation;
using System.Management.Automation.Runspaces;

namespace MyNamespace {
	public static class MyClass	{
		public static void SetHandler()	{
			SetConsoleCtrlHandler(new HandlerRoutine(ConsoleCtrlCheck), true);
		}

		private static bool ConsoleCtrlCheck(CtrlTypes ctrlType) {
			switch (ctrlType) {
				case CtrlTypes.CTRL_C_EVENT:
					Console.WriteLine("CTRL+C received!");
					return false;

				case CtrlTypes.CTRL_CLOSE_EVENT:
					Console.WriteLine("CTRL_CLOSE_EVENT received!");
					return true;

				case CtrlTypes.CTRL_BREAK_EVENT:
					Console.WriteLine("CTRL+BREAK received!");
					return false;

				case CtrlTypes.CTRL_LOGOFF_EVENT:
					Console.WriteLine("User is logging off!");
					return false;

				case CtrlTypes.CTRL_SHUTDOWN_EVENT:
					Console.WriteLine("User is shutting down!");
					return false;
			}
			return false;
		}

		[DllImport("Kernel32")]
		public static extern bool SetConsoleCtrlHandler(HandlerRoutine Handler, bool Add);

		public delegate bool HandlerRoutine(CtrlTypes CtrlType);

		public enum CtrlTypes {
			CTRL_C_EVENT = 0,
			CTRL_BREAK_EVENT,
			CTRL_CLOSE_EVENT,
			CTRL_LOGOFF_EVENT = 5,
			CTRL_SHUTDOWN_EVENT
		}
	}
}
"@

$text = Add-Type  -TypeDefinition $code -Language CSharp
$rs = [System.Management.Automation.Runspaces.Runspace]::DefaultRunspace
[MyNamespace.MyClass]::SetHandler()
start-sleep 10000