在Windows/Linux下调用API函数实现重启系统

一、Linux下重启系统

linux下很简单，直接看代码：

#include <unistd.h>
#include <sys/reboot.h>

bool rebootSystem()
{
    sync();
    if (reboot(RB_AUTOBOOT) < 0)
    {
        return false;
    }
    return true;
}

sync函数：将缓存中的数据写入磁盘，防止数据丢失。
reboot函数：对于关机或重启，该函数若成功，则无返回值；若失败，则返回-1。

reboot(RB_AUTOBOOT)  // 重启
reboot(RB_POWER_OFF) // 关机

二、Windows下重启系统

win下也不难，稍麻烦，看代码如下：

#include <Windows.h>

// for OpenProcessToken()、LookupPrivilegeValueA()、AdjustTokenPrivileges()
#pragma comment (lib, "Advapi32.lib")
// for ExitWindowsEx()
#pragma comment (lib, "user32.lib")

bool rebootSystem()
{
    HANDLE hToken;
    TOKEN_PRIVILEGES tkp;

    // 获取进程标志
    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
    {
        return false;
    }

    // 获取关机特权的LUID
    LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);
    tkp.PrivilegeCount = 1;
    tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;

    // 获取这个进程的关机特权
    AdjustTokenPrivileges(hToken, false, &tkp, 0, (PTOKEN_PRIVILEGES)NULL, 0);
    if (GetLastError() != ERROR_SUCCESS)
    {
        return false;
    }

    // 强制重启计算机
    if ( !ExitWindowsEx(EWX_REBOOT | EWX_FORCE, 0))
    {
        return false;
    }
    return true;
}

注释写的够清楚了，不多说。
另外特别注意的是，链接时需要Advapi32.lib、user32.lib，已直接通过代码指定链接这2个库了。

ExitWindowsEx(EWX_REBOOT | EWX_FORCE, 0)  // 重启
ExitWindowsEx(EWX_SHUTDOWN | EWX_FORCE, 0) // 关机

---

若对你有帮助，欢迎点赞、收藏、评论，你的支持就是我的最大动力！！！

同时，阿超为大家准备了丰富的学习资料，欢迎关注公众号“超哥学编程”，即可领取。