大家知道直接使用asp是不能够重启服务器的,这时我们需要制作一个组件来实现功能,ASP通过这个组件调用系统API,然
后按照不同的重启和关机方式进行操作!
下面先说COM的制作,在VB中新建一工程,当然是AceiveX dll的!
1)先修改工程属性,在工程属性窗口将工程名称改为system,在类模块窗口将模块名称改为contral,保存工程;
2)然后添加一个模块,用来声明需要使用的API和常数!下面是模块中的内容!
Declare Function ExitWindowsEx Lib "user32" (ByVal uFlags As Long, ByVal dwReserved As Long) As Long
3)在类模块中输入下列代码:
Public Function reboot(atype As Variant)
t& = ExitWindowsEx(1, atype)
End Function
该函数需要传入一个参数,用来控制启动方式,这个参数是从ASP程序中传入的,等一下就可以看到具体的使用方法!
首先是命名空间,调用Win API的时候,InteropServices不可少:
using System;
using System.Runtime.InteropServices;
然后是一系列的常量声明: protected const int SE_PRIVILEGE_ENABLED = 0x2;
protected const int TOKEN_QUERY = 0x8;
protected const int TOKEN_ADJUST_PRIVILEGES = 0x20;
protected const string SE_SHUTDOWN_NAME = "SeShutdownPrivilege";
protected const int EWX_LOGOFF = 0x0;
protected const int EWX_SHUTDOWN = 0x1;
protected const int EWX_REBOOT = 0x2;
protected const int EWX_FORCE = 0x4;
protected const int EWX_POWEROFF = 0x8;
protected const int EWX_FORCEIFHUNG = 0x10;
定义Luid结构,注意属性:
protected struct LuidStruct {
public int Count;
public long Luid;
public int Attr;
}
外部非托管DLL的声明:
protected static extern IntPtr GetCurrentProcess();
protected static extern bool OpenProcessToken(IntPtr h, int acc, ref IntPtr phtok);
protected static extern bool LookupPrivilegeValue(string host, string name, ref long pluid);
protected static extern bool AdjustTokenPrivileges(IntPtr htok, bool disall, ref LuidStruct newst, int len, IntPtr prev, IntPtr relen);
protected static extern bool ExitWindowsEx(int flg, int rea);
在NT级的操作系统上,需要先通知Windows系统即将关机,并且要获得关机的权限
以下就是关机、重启以及注销的实现: protected static void DoExitWindows(int flg) {
LuidStruct tp;
IntPtr hproc = GetCurrentProcess();
IntPtr htok = IntPtr.Zero;
OpenProcessToken(hproc, TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, ref htok);
tp.Count = 1;
tp.Luid = 0;
tp.Attr = SE_PRIVILEGE_ENABLED;
LookupPrivilegeValue(null, SE_SHUTDOWN_NAME, ref tp.Luid);
AdjustTokenPrivileges(htok, false, ref tp, 0, IntPtr.Zero, IntPtr.Zero);
ExitWindowsEx(flg, 0);
}
public static void Shutdown() {
DoExitWindows(EWX_SHUTDOWN);
}
public static void Reboot() {
DoExitWindows(EWX_REBOOT | EWX_FORCE);
}
public static void Logoff() {
DoExitWindows(EWX_LOGOFF);
}
至此,重启代码结束,这段代码可以很好地工作在交互环境下,也就是在用户已经登录进Windows的情况下
但是ASP.NET是运行在非交互环境下的,查阅MSDN,在ExitWindowsEx函数定义下面发现这样一段话:
The ExitWindowsEx function returns as soon as it has initiated the shutdown process. The shutdown or logoff then proceeds asynchronously. The function is designed to stop all processes in the caller's logon session. Therefore, if you are not the interactive user, the function can succeed without actually shutting down the computer. If you are not the interactive user, use the InitiateSystemShutdown or InitiateSystemShutdownEx function.
于是得到启发,发现非交互下重启服务器不可以用ExitWindowsEx,需要将其替换成InitiateSystemShutdown:
protected static extern bool InitiateSystemShutdown(string name, string msg, int timeout, bool force, bool reboot);
参数解释:
name:机器名,用于重启局域网内的其它机器,如果为 null 则是本机
msg:重启消息,会显示在重启消息框上,在Windows 2003和XP中也会作为消息日志被保存
timeout:如果不是0,那么会显示一个重新消息框,倒计时timeout秒后重启
force:强制重启,不等待应用程序提示是否保存工作,对于服务器来说,应该是true
reboot:是否是重启,如果是false,那么做关机处理,对于服务器,应该是true
先按照文章一开始的方法调用 AdjustTokenPrivileges 得到Privilege,然后在ASP.NET页面里面执行: InitiateSystemShutdown(null,null,0,true,true);
系统就重启了
这里有一点说下,如果重启本机,那么多半会返回Service Unavailable错误,这是因为在ASP.NET执行结束之前系统已经开始结束各个进程了,当然包括ASP.NET进程,算是正常表现,虽然看起来有些不太舒服
另外由于目前我只在自己机器上测试通过,所以没有详细研究权限问题,所以无法确定在一般服务器上是否可以正常运行
初步只想到可以用权限模拟解决,即在web.config文件system.web节写上<identity impersonate="true" userName="Administrator" password="pass">,不过没有经过确认,有时间会尝试一下。web.config不是很安全,所以这里可能要借助于DPAPI,有点扯远了,就先到这里吧。