//ShutDown.cs
using System;
using System.Collections.Generic;
using System.Text;
using System.Runtime.InteropServices;

namespace Walter.K.Wang
{
    
/// <summary>
    
/// 
    
/// </summary>
    public class ShutDown
    {
        
/// <summary>
        
/// 
        
/// </summary>
        public enum Operat
        {
            
/// <summary>
            
/// 关闭计算机
            
/// </summary>
            ShutDown,
            
/// <summary>
            
/// 登出計算機
            
/// </summary>
            Logoff,
            
/// <summary>
            
/// 重起計算機
            
/// </summary>
            Reboot,
            
/// <summary>
            
/// 關閉計算機
            
/// </summary>
            Remind
        };
        [StructLayout(LayoutKind.Sequential, Pack 
= 1)]
        
internal struct TokPrivlLuid
        {
            
public int Count;
            
public long Luid;
            
public int Attr;
        }

        [DllImport(
"kernel32.dll", ExactSpelling = true)]
        
internal static extern IntPtr GetCurrentProcess();

        [DllImport(
"advapi32.dll", ExactSpelling = true, SetLastError = true)]
        
internal static extern bool OpenProcessToken(IntPtr h, int acc, ref IntPtr phtok);

        [DllImport(
"advapi32.dll", SetLastError = true)]
        
internal static extern bool LookupPrivilegeValue(string host, string name, ref long pluid);

        [DllImport(
"advapi32.dll", ExactSpelling = true, SetLastError = true)]
        
internal static extern bool AdjustTokenPrivileges(IntPtr htok, bool disall, ref TokPrivlLuid newst, int len, IntPtr prev, IntPtr relen);

        [DllImport(
"user32.dll", ExactSpelling = true, SetLastError = true)]
        
internal static extern bool ExitWindowsEx(int flg, int rea);

        
internal const int SE_PRIVILEGE_ENABLED = 0x00000002;
        
internal const int TOKEN_QUERY = 0x00000008;
        
internal const int TOKEN_ADJUST_PRIVILEGES = 0x00000020;
        
internal const string SE_SHUTDOWN_NAME = "SeShutdownPrivilege";
        
internal const int EWX_LOGOFF = 0x00000000;
        
internal const int EWX_SHUTDOWN = 0x00000001;
        
internal const int EWX_REBOOT = 0x00000002;
        
internal const int EWX_FORCE = 0x00000004;
        
internal const int EWX_POWEROFF = 0x00000005;
        
internal const int EWX_FORCEIFHUNG = 0x00000010;

        
/// <summary>
        
/// 
        
/// </summary>
        
/// <param name="op"></param>
        public static void DoExitWin(Operat op)
        {
            
int flg;
            
switch (op)
            {
                
case Operat.ShutDown:
                    flg 
= EWX_SHUTDOWN;
                    
break;
                
case Operat.Logoff:
                    flg 
= EWX_LOGOFF;
                    
break;
                
case Operat.Reboot:
                    flg 
= EWX_REBOOT;
                    
break;
                
default:
                    flg 
= EWX_SHUTDOWN;
                    
break;
            }
            
bool ok;
            TokPrivlLuid tp;
            IntPtr hproc 
= GetCurrentProcess();
            IntPtr htok 
= IntPtr.Zero;
            ok 
= OpenProcessToken(hproc, TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, ref htok);
            tp.Count 
= 1;
            tp.Luid 
= 0;
            tp.Attr 
= SE_PRIVILEGE_ENABLED;
            ok 
= LookupPrivilegeValue(null, SE_SHUTDOWN_NAME, ref tp.Luid);
            ok 
= AdjustTokenPrivileges(htok, falseref tp, 0, IntPtr.Zero, IntPtr.Zero);
            ok 
= ExitWindowsEx(flg, 0);
        }
    }
}

Walter.K.Wang.ShutDown.DoExitWin(Walter.K.Wang.ShutDown.Operat.ShutDown);

posted on 2007-10-26 10:12  wkjs  阅读(302)  评论(0编辑  收藏  举报