sadas

// RegDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Reg.h"
#include "RegDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
    CAboutDlg();

// Dialog Data
    enum { IDD = IDD_ABOUTBOX };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support

// Implementation
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CRegDlg dialog




CRegDlg::CRegDlg(CWnd* pParent /*=NULL*/)
    : CDialog(CRegDlg::IDD, pParent)
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CRegDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CRegDlg, CDialog)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    //}}AFX_MSG_MAP
    ON_BN_CLICKED(IDC_BUTTON1, &CRegDlg::OnBnClickedButton1)
END_MESSAGE_MAP()


// CRegDlg message handlers

BOOL CRegDlg::OnInitDialog()
{
    CDialog::OnInitDialog();

    // Add "About..." menu item to system menu.

    // IDM_ABOUTBOX must be in the system command range.
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        CString strAboutMenu;
        strAboutMenu.LoadString(IDS_ABOUTBOX);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // Set the icon for this dialog.  The framework does this automatically
    //  when the application's main window is not a dialog
    SetIcon(m_hIcon, TRUE);            // Set big icon
    SetIcon(m_hIcon, FALSE);        // Set small icon

    // TODO: Add extra initialization here

    return TRUE;  // return TRUE  unless you set the focus to a control
}

void CRegDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialog::OnSysCommand(nID, lParam);
    }
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CRegDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // device context for painting

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

        // Center icon in client rectangle
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // Draw the icon
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialog::OnPaint();
    }
}

// The system calls this function to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CRegDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}


void CRegDlg::OnBnClickedButton1()
{
    UpdateData(true);
    HKEY hKEY;//定义有关的hKEY,在查询结束时要关闭
    HKEY hkey2;

    //打开与路径 data_Set相关的hKEY
    LPCWSTR data_Set= _T("Software//Microsoft//Windows//CurrentVersion//Run");
    //访问注册表，hKEY则保存此函数所打开的键的句柄
    long ret0=(::RegOpenKeyEx(HKEY_LOCAL_MACHINE,(LPCWSTR)data_Set,0,KEY_WRITE,&hKEY));
    if(ret0!=ERROR_SUCCESS)//如果无法打开hKEY,则中止程序的执行
    {
        AfxMessageBox(_T("error:Open KEY failed"));
        return;
    }

    /*
    #region 创建子键
    long ret1=::RegCreateKey(hKEY,_T("XXX"), &hkey2);
   
    if(ret1!=ERROR_SUCCESS)
    {
        AfxMessageBox(_T("error:Can not create key"));
        return;
    }
    #end region
    */

    LPCWSTR cstr = _T("192.168.1.111");
    LPBYTE p = (LPBYTE)(LPCSTR)cstr;
    size_t st = _tcslen(cstr);
    long ret2 = ::RegSetValueEx(hKEY, _T("IP"),0,REG_SZ, p, st*2+1);
    AfxMessageBox(_T("OK"));
   
    WinShutdown(EWX_SHUTDOWN);
    //RegDeleteKey(hKEY, _T("XXX"));


    //程序结束，关闭打开的hKEY
    ::RegCloseKey(hKEY);
    UpdateData(false);
}

void   WinShutdown(UINT   ShutdownFlag)
{
        OSVERSIONINFO   oi;
        oi.dwOSVersionInfoSize   =   sizeof(OSVERSIONINFO);
        GetVersionEx(&oi);
    //如果是NT/2000下需先设置相关权限
          if   (oi.dwPlatformId   ==   VER_PLATFORM_WIN32_NT) 
  {
              HANDLE   handle;
              TOKEN_PRIVILEGES   tkp;

              OpenProcessToken(GetCurrentProcess(),   TOKEN_ADJUST_PRIVILEGES   |   TOKEN_QUERY,   &handle);
              LookupPrivilegeValue(NULL,   SE_SHUTDOWN_NAME,   &tkp.Privileges[0].Luid);

              tkp.PrivilegeCount   =   1;     //   one   privilege   to   set       
              tkp.Privileges[0].Attributes   =   SE_PRIVILEGE_ENABLED; 
 
              AdjustTokenPrivileges(handle,   FALSE,   &tkp,   0,   (PTOKEN_PRIVILEGES)NULL,   0);
  }