WinPowerOff 

Category: Programming, wikiX,

  1. 윈도우즈 2000,XP 파워 끄기, 리부팅하기

윈도우즈 2000,XP 파워 끄기, 리부팅하기  

2001년도에 짰던 것 같네요. 윈도우즈2000 계열의 파워 끄는 소스코드입니다.
VC6.0으로 짰고, MFC를 사용합니다. 프로그램을 하실 줄 아시면 뭐하는 것인 지 금방 아실 듯.
다음은 빌드한 실행파일입니다.
이 코드가 어디서 본 것으로 짠 거라서(어디였더라..?) 상업적으로 그대로 쓰면 안될 겁니다.
관련 MSDN 링크는 여기 입니다.

설명 파일 size md5sum
파워 끄는 프로그램 poweroff.exe 20480 bytes 4d4b1636cf14891a8ddda52d343a3b39
재부팅하는 프로그램 reboot.exe 20480 bytes 0945bb917736b4718ca26523223bb1f8

PwrTest.h

#if !defined(AFX_PWRTEST_H__149C02D0_B545_4699_A4A5_DFDB719EFBFB__INCLUDED_)
#define AFX_PWRTEST_H__149C02D0_B545_4699_A4A5_DFDB719EFBFB__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "resource.h"

/////////////// Jiseok
int PowerOff();
int Reboot();
int GetWinVersion(OSVERSIONINFO* windowsversion);
int DoPowerOff(OSVERSIONINFO* windowsversion);
int DoReboot(OSVERSIONINFO* windowsversion);
int AquirePrivileges(OSVERSIONINFO* windowsversion);
////////////// end Jiseok

#endif // !defined(AFX_PWRTEST_H__149C02D0_B545_4699_A4A5_DFDB719EFBFB__INCLUDED_)

PwrTest.cpp
// PwrTest.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "PwrTest.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// The one and only application object

CWinApp theApp;

using namespace std;

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
        int nRetCode = 0;

        // initialize MFC and print and error on failure
        if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
        {
                // TODO: change error code to suit your needs
                cerr << _T("Fatal Error: MFC initialization failed") << endl;
                nRetCode = 1;
        }
        else
        {
                // TODO: code your application's behavior here.

                //////////////// Jiseok
                //if (PowerOff()!=0)
                if (Reboot()!=0)
                        return 1;
                //////////////// end Jiseok

                CString strHello;
                strHello.LoadString(IDS_HELLO);
                cout << (LPCTSTR)strHello << endl;
        }

        return nRetCode;
}



///////////////// Jiseok

int PowerOff()
{
        OSVERSIONINFO Version;

        if (GetWinVersion(&Version)!=0)
                return -1;
        if (DoPowerOff(&Version)!=0)
                return -1;
        return 0;
}

int Reboot()
{
        OSVERSIONINFO Version;

        if (GetWinVersion(&Version)!=0)
                return -1;
        if (DoReboot(&Version)!=0)
                return -1;
        return 0;
}

int GetWinVersion(OSVERSIONINFO* windowsversion)
{
        windowsversion->dwOSVersionInfoSize=sizeof(OSVERSIONINFO);
        if (!GetVersionEx(windowsversion))
        {
                printf("Error! getting version info");
                return -1;
        }
        return 0;
}

int DoPowerOff(OSVERSIONINFO* windowsversion)
{
        UINT mode;

        if (windowsversion->dwPlatformId==VER_PLATFORM_WIN32_WINDOWS)
                mode=EWX_SHUTDOWN;
        else
                mode=EWX_POWEROFF;

        if (windowsversion->dwPlatformId==VER_PLATFORM_WIN32_NT)
                {
                // privilege 얻기
                if (AquirePrivileges(windowsversion)!=0)
                {
                        return -1;
                }//if 
        }// if
  
        if (!ExitWindowsEx(mode,0))        // 끄기
        {
                printf("error ExitWindowsEx");
                return -1;
        }
        return 0;
}

      
int DoReboot(OSVERSIONINFO* windowsversion)
{
        UINT mode;

        mode=EWX_REBOOT;

        if (windowsversion->dwPlatformId==VER_PLATFORM_WIN32_NT)
                {
                // privilege 얻기
                if (AquirePrivileges(windowsversion)!=0)
                {
                        return -1;
                }//if 
        }// if
  
        if (!ExitWindowsEx(mode,0))        // 끄기
        {
                printf("error ExitWindowsEx");
                return -1;
        }
        return 0;
}

int AquirePrivileges(OSVERSIONINFO* windowsversion)
{
  HANDLE current_thread,token;
  TOKEN_PRIVILEGES privs;
  LUID luid;

  current_thread=GetCurrentProcess();
  if (!OpenProcessToken(current_thread,TOKEN_ADJUST_PRIVILEGES,&token))
  {
    printf("Error! opening process token");
    return -1;
  }
  if (!LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&luid))
  {
    printf("Error! looking up privilege");
    return -1;
  }
  privs.PrivilegeCount=1;
  privs.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED;
  privs.Privileges[0].Luid=luid;
  if (!AdjustTokenPrivileges(token,FALSE,&privs,0,NULL,NULL))
  {
    printf("Error! getting privileges");
    return -1;
  }
  return 0;
}

//////////// end Jiseok

Category: Programming, wikiX,