编写屏幕保护程序

参考网址:

https://blog.csdn.net/xylary/article/details/1737627

https://www.windows-commandline.com/configure-screensaver-command-line/

https://docs.microsoft.com/en-us/windows/desktop/api/winuser/nf-winuser-systemparametersinfow

一 编写屏幕保护程序

微软已经写了一个静态类库ScrnSav.lib, 只要包含头文件, 实现几个接口就可以做出一个屏保来, 相关代码如下:

注意: 屏幕保护程序实际是个exe, 但是使用的时候要把exe改成scr, 并且放在 windowssystem32 或者windowssyswow64目录下

#include <Windows.h>
#include <ScrnSave.h>

#ifdef UNICODE
#pragma comment(lib, "ScrnSavw.lib")
#else
#pragma comment(lib, "ScrnSave.lib")
#endif
#pragma comment(lib,"comctl32.lib")

extern "C" LRESULT WINAPI ScreenSaverProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch (message)
    {// 省略的代码
    default:
        return DefScreenSaverProc(hWnd, message, wParam, lParam);
    }
    return 0;
}

extern "C" BOOL WINAPI ScreenSaverConfigureDialog(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{//暂时不需要配置对话框，仅返回FALSE
    return FALSE;
}

extern "C" BOOL WINAPI RegisterDialogClasses(HANDLE hInstance)
{//一般不需要，仅返回TRUE 
    return TRUE;
}

二  设置屏保

 1 拷贝文件到系统目录

 2 设置注册表, 这个大多数是给系统重启后初始化屏保逻辑的初始参数

3 调用SystemParametersInfo api给各窗口发消息, 因为屏保的部分逻辑已经运行在当前已经运行的窗口程序中了, 需要调用此api给窗口发消息更新这些参数

代码如下:

#include <windows.h>
#include <strsafe.h>
#include <iostream>

bool GetScreenSaverRunning(bool &isRunning)
{
    bool result = false;
    try
    {
        BOOL BRun = TRUE;
        BOOL ret = SystemParametersInfoW(SPI_GETSCREENSAVERRUNNING, NULL, &BRun, 0);
        if (ret)
        {
            isRunning = BRun ? true : false;
            result = true;
        }
    }
    catch (...)
    {
        result = false;
    }
    return result;
}
bool GetScreenSaverActive(bool &active)
{
    bool result = false;
    try
    {
        BOOL bActive = TRUE;
        BOOL ret = SystemParametersInfoW(SPI_GETSCREENSAVEACTIVE, NULL, &bActive, 0);
        if (ret)
        {
            active = bActive ? true : false;
            result = true;
        }
    }
    catch (...)
    {
        result = false;
    }
    return result;
}
bool SetScreenSaverActive(bool active)
{
    bool result = false;
    try
    {
        BOOL ret = FALSE;
        if (active)
            ret = SystemParametersInfoW(SPI_SETSCREENSAVEACTIVE, TRUE, NULL, SPIF_SENDWININICHANGE);
        else
            ret = SystemParametersInfoW(SPI_SETSCREENSAVEACTIVE, FALSE, NULL, SPIF_SENDWININICHANGE);

        result = ret ? true : false;
    }
    catch (...)
    {
        result = false;
    }
    return result;
}

//screenSaveTimeoutSeconds, 单位是秒
bool SetScreenSaver(const wchar_t *scrFile, UINT screenSaveTimeoutSeconds)
{
    if (!scrFile || !screenSaveTimeoutSeconds)
        return false;

    bool result = false;
    HRESULT hr = S_OK;
    LSTATUS status = ERROR_SUCCESS;
    BOOL ret2 = FALSE;
    wchar_t buffer[MAX_PATH] = { 0 };
    DWORD accessmask = KEY_SET_VALUE;
    HKEY root = NULL;
    DWORD charLen = 0;
    do
    {
        // copy到系统路径
        BOOL isWOW64 = FALSE;
        IsWow64Process(GetCurrentProcess(), &isWOW64);

        UINT ret = 0;
        if (isWOW64)
            ret = GetSystemWow64DirectoryW(buffer, MAX_PATH);
        else
            ret = GetSystemDirectoryW(buffer, MAX_PATH);
        if(ret ==0 || ret >MAX_PATH)
            break;
        
        hr = StringCchCatW(buffer, MAX_PATH, L"\aaa.scr");
        if(hr != S_OK)
            break;

        ret2 = CopyFile(scrFile, buffer, FALSE);
        if(!ret2)
        {
            wchar_t buffer2[MAX_PATH] = { 0 };
            hr =StringCchPrintfW(buffer2, MAX_PATH, L"%s1",buffer);
            if(hr != S_OK)
                break;

            ret2 =MoveFileExW(buffer, buffer2, MOVEFILE_REPLACE_EXISTING);
            if (!ret2)
                break;

            ret2 = MoveFileExW(buffer2, NULL, MOVEFILE_DELAY_UNTIL_REBOOT);
            if(!ret2)
                break;

            ret2 = CopyFile(scrFile, buffer, FALSE);
            if (!ret2)
                break;
        }

        status = RegOpenKeyExW(HKEY_CURRENT_USER, L"Control Panel", 0, accessmask, &root);
        if(status != ERROR_SUCCESS)
            break;

        // 设置屏保程序
        charLen = (DWORD)((wcslen(buffer) + 1) * sizeof(wchar_t));
        status = RegSetKeyValueW(root, L"Desktop", L"SCRNSAVE.EXE", REG_SZ, buffer, charLen);
        if (status != ERROR_SUCCESS)
            break;

        // 设置超时时间
        // 仅设置注册表不会立即生效, 要给其他程序发消息才能立即生效
        hr = StringCchPrintfW(buffer, MAX_PATH, L"%u", screenSaveTimeoutSeconds);
        if(hr !=S_OK)
            break;

        charLen = (DWORD)((wcslen(buffer) + 1) * sizeof(wchar_t));
        status = RegSetKeyValueW(root, L"Desktop", L"ScreenSaveTimeOut", REG_SZ, buffer, charLen);
        if(status != ERROR_SUCCESS)
            break;

        try
        {
            ret2 = SystemParametersInfoW(SPI_SETSCREENSAVETIMEOUT, screenSaveTimeoutSeconds, NULL, SPIF_SENDWININICHANGE);
            if (!ret2)
                break;
        }
        catch (...)
        {
            break;
        }

        // 设置恢复时显示登录界面
        // 仅设置注册表会立即生效, 但是设置界面跟注册表里的值不一致
        hr = StringCchPrintfW(buffer, MAX_PATH, L"%u", 1);
        if(hr != S_OK)
            break;

        charLen = (DWORD)((wcslen(buffer) + 1) * sizeof(wchar_t));
        status = RegSetKeyValueW(root, L"Desktop", L"ScreenSaverIsSecure", REG_SZ, buffer, charLen);
        if(status !=ERROR_SUCCESS)
            break;

        try
        {
            BOOL bSecurite = TRUE;
            ret2 = SystemParametersInfoW(SPI_SETSCREENSAVESECURE, bSecurite, NULL, SPIF_SENDWININICHANGE);
            if (!ret2)
                break;
        }
        catch (...)
        {
            break;
        }

        // 设置屏保开关
        // 仅设置注册表不会立即生效, 要给其他程序发消息才能立即生效
        hr = StringCchPrintfW(buffer, MAX_PATH, L"%u", 1);
        if(hr != S_OK)
            break;

        charLen = (DWORD)((wcslen(buffer) + 1) * sizeof(wchar_t));
        status = RegSetKeyValueW(root, L"Desktop", L"ScreenSaveActive", REG_SZ, buffer, charLen);
        if (status != ERROR_SUCCESS)
            break;

        SetScreenSaverActive(true);

        result = true;
    } while (false);


    return result;
}
int main(int argc, char **argv, char **env)
{
     SetScreenSaver(L"aaa.exe",60);
    system("pause");
    return 0;
}