学步园

控制台程序作为简捷的人机对话方式，我们通常让用户输入类似exit的命令来退出程序。然而，用些用户更喜欢直接点击控制台窗口右上角的关闭按钮退出，此时，我们需要捕获这个操作，方便执行我们的退出代码。

要实现这样的功能需要使用系统的SetConsoleCtrlHandler API函数。

例子和说明见代码部分，直接用csc.exe编译即可：

 using System;<br /> using System.Collections;<br /> using System.Runtime.InteropServices;<br /> using System.Text.RegularExpressions;<br /> using System.Threading;</p> <p>class Program<br /> {<br /> //存储命令行参数<br /> public static Hashtable CommandLineArgs = new Hashtable();</p> <p> static void Main(string[] args)<br /> {<br /> ParseCommandLineArgs(args); //解释命令行参数</p> <p> //声明委托方法，如果在运行时产生异常，可以在后面告诉GC保留它(用GC.KeepAlive方法)，防止被回收掉。<br /> PHANDLER_ROUTINE handlerRoutine = HandlerRoutine;<br /> //注册handlerRoutine为监听回调函数。这里没有检查返回值是否成功，假设它是成功的。<br /> NativeMethods.SetConsoleCtrlHandler(handlerRoutine, true);</p> <p> //显示些数据<br /> Console.WriteLine("SetConsoleCtrlHandler Demo by Oyi319@csdn");<br /> Console.WriteLine(new string('-',10));<br /> if (Information.ShowTime) Console.WriteLine(DateTime.Now);<br /> if (Information.UserName.Length>0) Console.WriteLine("Hi, "+Information.UserName);<br /> Console.WriteLine("输入'exit'正常退出，或者测试<Ctrl+C>，<Ctrl+Break>，或者直接点击关闭按钮。。。");<br /> for(;;)<br /> {<br /> var input = Console.ReadLine();<br /> if(input!=null && input.Equals("exit", StringComparison.OrdinalIgnoreCase))<br /> {<br /> CloseConsole("正常退出",0); //退出控制台<br /> }<br /> }</p> <p> //GC.KeepAlive(handlerRoutine); //使GC不要回收委托函数<br /> }</p> <p> private static int HandlerRoutine(uint ctrltype)<br /> {<br /> string outputText = "";<br /> switch (ctrltype)<br /> {<br /> case NativeConstants.CTRL_C_EVENT:<br /> outputText = "按下了 Ctrl+C";<br /> break;<br /> case NativeConstants.CTRL_BREAK_EVENT:<br /> outputText = "按下了 Ctrl+Break";<br /> break;<br /> case NativeConstants.CTRL_CLOSE_EVENT:<br /> outputText = "收到关闭窗口的消息(按下了窗口关闭按钮 或者 被任务管理器结束任务)";<br /> break;<br /> case NativeConstants.CTRL_LOGOFF_EVENT:<br /> outputText = "注销系统";<br /> break;<br /> case NativeConstants.CTRL_SHUTDOWN_EVENT:<br /> outputText = "关机";<br /> break;<br /> }<br /> var r = (int)ctrltype; //作为程序的返回值<br /> CloseConsole(outputText, r); //调用退出方法<br /> return 1; //返回1 (TRUE)。 另说明一下：由于CloseConsole方法使用调用退出函数，此句代码不会被执行。<br /> }</p> <p> /// <summary><br /> /// 退出控制台<br /> /// </summary><br /> /// <param name="outputText">输出字符串</param><br /> /// <param name="r">退出代码</param><br /> static void CloseConsole(string outputText, int r)<br /> {<br /> Console.WriteLine("程序结束："+outputText);<br /> Console.WriteLine("退出代码："+r);<br /> Thread.Sleep(3000); //等待3秒钟，让用户观察到退出信息。<br /> Environment.Exit(r); //退出代码，通常用0表示程序的正常退出，大于0表示非正常退出的代号。<br /> }</p> <p> static void ParseCommandLineArgs(string[] args)<br /> {<br /> //读取命令行参数<br /> if(args.Length != 0)<br /> {<br /> //命令行参数格式：/argname:argvalue<br /> string pattern = @"(?<argname>//w+)(:(?<argvalue>/w+))?";<br /> foreach(var arg in args)<br /> {<br /> var match = Regex.Match(arg, pattern);<br /> if (!match.Success)<br /> throw new ArgumentException(<br /> "The command line arguments are improperly formed. Use /argname:argvalue.");</p> <p> CommandLineArgs[match.Groups["argname"].Value] = match.Groups["argvalue"].Value;<br /> }<br /> }</p> <p> if (CommandLineArgs.ContainsKey("/name"))<br /> {<br /> Information.UserName = CommandLineArgs["/name"].ToString();<br /> }</p> <p> if (CommandLineArgs.ContainsKey("/showtime"))<br /> {<br /> Information.ShowTime = true;<br /> }<br /> }</p> <p> #region SetConsoleCtrlHandler</p> <p> /// Return Type: BOOL->int<br /> ///CtrlType: DWORD->unsigned int<br /> public delegate int PHANDLER_ROUTINE(uint CtrlType);<br /> //~ 说明：PHANDLER_ROUTINE被声明为委托，它是SetConsoleCtrlHandler的回调函数指针，<br /> //~ 当控制台退出事件发生时，我们利用此委托执行回调函数。<br /> //~ 参数CtrlType由系统传入，通过它能判断出控制台退出事件的原因，它会是下面NativeConstants类<br /> //~ 中CTRL_开头的常量的值，意义见后面的解释。<br /> //~ 返回值为BOOL类型，这里声明为int，我们可以返回0或者1。用0表示false，用1表示true。<br /> //~ 当我们返回false时，如果我们用SetConsoleCtrlHandler注册了多个监听，将会继续完成其他的回调函数；<br /> //~ 而我们返回true时，假设我们用SetConsoleCtrlHandler注册了多个监听，其他的回调函数将不再执行，<br /> //~ 起一个拦截的作用。<br /> //~ 更多资料请参考：http://msdn.microsoft.com/en-us/library/ms683242(VS.85).aspx</p> <p> public class NativeMethods<br /> {<br /> /// Return Type: BOOL->int<br /> ///HandlerRoutine: PHANDLER_ROUTINE<br /> ///Add: BOOL->int<br /> [DllImport("kernel32.dll", EntryPoint = "SetConsoleCtrlHandler")]<br /> [return: MarshalAs(UnmanagedType.Bool)]<br /> public static extern bool SetConsoleCtrlHandler(PHANDLER_ROUTINE HandlerRoutine,<br /> [MarshalAs(UnmanagedType.Bool)] bool Add);</p> <p> //~ 说明：SetConsoleCtrlHandler API函数，当控制台程序并非从主函数正常退出的情况进行监视回调。<br /> //~ 被监视的操作包括：用户按Ctrl+C、按Ctrl+Break、收到关闭控制台窗口消息（点击控制台窗口的关闭按钮<br /> //~ 或者在任务管理器里“结束任务”）、注销Windows系统、或者关机。<br /> //~ 它的第一个参数HandlerRoutine为回调函数，就是处理此类操作时我们让程序执行的代码；<br /> //~ 第二个参数设定为true时，注册HandlerRoutine回调函数，而为false时，则取消HandlerRoutine回调函数，<br /> //~ 因此，我们可以注册一系列处理方法，类似事件的注册和移除。<br /> //~ 执行SetConsoleCtrlHandler函数会立即返回结果，当返回0时，表示函数执行成功，返回非0值，表示失败，<br /> //~ 使用GetLastError API函数得到失败消息。<br /> //~ 更多材料参考：http://msdn.microsoft.com/en-us/library/ms686016(VS.85).aspx<br /> }</p> <p> public class NativeConstants<br /> {<br /> /// CTRL_SHUTDOWN_EVENT -> 6<br /> public const int CTRL_SHUTDOWN_EVENT = 6; //关机</p> <p> /// CTRL_LOGOFF_EVENT -> 5<br /> public const int CTRL_LOGOFF_EVENT = 5; //注销</p> <p> /// CTRL_CLOSE_EVENT -> 2<br /> public const int CTRL_CLOSE_EVENT = 2; //收到关闭消息</p> <p> /// CTRL_BREAK_EVENT -> 1<br /> public const int CTRL_BREAK_EVENT = 1; //Ctrl+Break 中断</p> <p> /// CTRL_C_EVENT -> 0<br /> public const int CTRL_C_EVENT = 0; //Ctrl+C 中断<br /> }</p> <p> #endregion<br /> }</p> <p>//保存信息的静态类，这是为了演示主程序命令行参数而定义的<br /> public static class Information<br /> {<br /> public static string UserName = "";<br /> public static bool ShowTime;<br /> }<br />

另外，示例还演示了命令行参数的执行方法，供大家参考，测试方法如下图：