远程开关机实现

Server端:


// RemoteShutDownDlg.cpp: 实现文件
//

#include "stdafx.h"
#include "RemoteShutDown.h"
#include "RemoteShutDownDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

// 对话框数据
#ifdef AFX_DESIGN_TIME
    enum { IDD = IDD_ABOUTBOX };
#endif

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

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CRemoteShutDownDlg 对话框

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

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

BEGIN_MESSAGE_MAP(CRemoteShutDownDlg, CDialog)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BTN_SHUTDOWN, &CRemoteShutDownDlg::OnBnClickedBtnShutdown)
END_MESSAGE_MAP()


// CRemoteShutDownDlg 消息处理程序

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

    // 将“关于...”菜单项添加到系统菜单中。

    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

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

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时,框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO: 在此添加额外的初始化代码

    return TRUE;  // 除非将焦点设置到控件,否则返回 TRUE
}

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

// 如果向对话框添加最小化按钮,则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序,
//  这将由框架自动完成。

void CRemoteShutDownDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast(dc.GetSafeHdc()), 0);

        // 使图标在工作区矩形中居中
        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;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialog::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CRemoteShutDownDlg::OnQueryDragIcon()
{
    return static_cast(m_hIcon);
}


void CRemoteShutDownDlg::OnBnClickedBtnShutdown()
{
    // TODO: 在此添加控件通知处理程序代码
    //ShutDown();

    //Socket 
    //1.Windows Socket

    //2.Init Socket
    WSAData wsadata;
    do
    {
        if (0 != WSAStartup(MAKEWORD(2, 2), &wsadata))
            break;
        if (2 != LOBYTE(wsadata.wVersion) || 2 != HIBYTE(wsadata.wVersion))
            break;

        SOCKET sock_server = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    
        sockaddr_in addr_server = { 0 };
        addr_server.sin_family = AF_INET;
        addr_server.sin_port = htons(18080);
        // bind
        bind(sock_server, (SOCKADDR*)&addr_server, sizeof(addr_server));
        // listen
        listen(sock_server, SOMAXCONN);
        // accept()
        SOCKET sock_client =  accept(sock_server,nullptr,nullptr);
        send(sock_client, "Hello", strlen("Hello"), 0);
        MessageBoxW(L"in");
    } while (FALSE);
    
    //3.
}

BOOL CRemoteShutDownDlg::ReSetWindow(DWORD dwFlags, BOOL bForce)
{
    BOOL bRet = FALSE;

    do
    {
        if (EnableShutDownPriv())
        {
            if (ExitWindowsEx(dwFlags, FALSE))
                break;            
        } 
        bRet = TRUE;
    } while (FALSE);                    
    return bRet;
}

BOOL CRemoteShutDownDlg::EnableShutDownPriv()
{
    HANDLE hToken = nullptr;
    TOKEN_PRIVILEGES tkp = {};
    BOOL bRet = FALSE;
    do
    {
        if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
            break;
        if (!LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid))
            break;            
        tkp.PrivilegeCount = 1;
        tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

        AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
        if (GetLastError() != ERROR_SUCCESS)
            break;
        bRet = TRUE;
    } while (false);

    if (FALSE==bRet)
    {
        SetLastError(GetLastError());
    }
    if (hToken)
    {
        CloseHandle(hToken);
    }
    return bRet;
}
void CRemoteShutDownDlg::ShutDown()
{    
    ReSetWindow(EWX_SHUTDOWN, 0);
}

 

Client端:


// ClientDlg.cpp: 实现文件
//

#include "stdafx.h"
#include "Client.h"
#include "ClientDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

// 对话框数据
#ifdef AFX_DESIGN_TIME
    enum { IDD = IDD_ABOUTBOX };
#endif

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

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CClientDlg 对话框

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

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

BEGIN_MESSAGE_MAP(CClientDlg, CDialog)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BTN_OK, &CClientDlg::OnBnClickedBtnOk)
END_MESSAGE_MAP()


// CClientDlg 消息处理程序

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

    // 将“关于...”菜单项添加到系统菜单中。

    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

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

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时,框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO: 在此添加额外的初始化代码

    return TRUE;  // 除非将焦点设置到控件,否则返回 TRUE
}

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

// 如果向对话框添加最小化按钮,则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序,
//  这将由框架自动完成。

void CClientDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast(dc.GetSafeHdc()), 0);

        // 使图标在工作区矩形中居中
        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;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialog::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CClientDlg::OnQueryDragIcon()
{
    return static_cast(m_hIcon);
}

void CClientDlg::OnBnClickedBtnOk()
{
    // TODO: 在此添加控件通知处理程序代码
    
    do
    {
        WSAData wsadata;
        if (0 != WSAStartup(MAKEWORD(2, 2), &wsadata))
            break;
        if (2 != LOBYTE(wsadata.wVersion) || 2 != (HIBYTE(wsadata.wVersion)))
            break;
        SOCKET sock_connect = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
        sockaddr_in addr_server = { 0 };
        addr_server.sin_family = AF_INET;
        addr_server.sin_port = htons(18080);
        addr_server.sin_addr.S_un.S_addr = inet_addr("192.168.80.129");

        connect(sock_connect, (SOCKADDR*)&addr_server, sizeof(addr_server));

        char buffer[MAXBYTE] = { 0 };
        recv(sock_connect, buffer, MAXBYTE, 0);
        MessageBoxA(nullptr, buffer, "receiver from server", MB_OK);
    } while (false);
}
 

你可能感兴趣的:(Windows编程)