创建 Windows API 以从命令行添加、修改、查询、删除注册表项

Question

我正在执行一项任务，需要查询、添加(包含值)、删除和修改注册表项。该程序需要支持所有不同类型的注册表数据，最重要的是，它需要使用命令行参数，与 reg.exe 的方式非常相似。

声明，我不精通低级编程语言(即我是菜鸟)；该任务是更广泛的非编程安全类(class)的一部分，但是我热衷于在我迄今为止所做的基础上继续完成该任务。我在 Visual Studio 中创建了 3 个单独的 .c 文件，它们在 Debug模式下成功运行；但是，我不知道如何创建接受命令行参数的单个代码文件。

到目前为止，我创建的代码是根据我在研究解决方案时挖掘的各种帖子拼凑而成的；此外，我还观看了一堆 YouTube 视频，但遗憾的是，没有任何进展。我也在 Stack Overflow 上进行了尽可能深入的挖掘，但没有发现任何东西可以帮助我理解如何解决主要问题

1. 创建一个 .c 文件，通过单独的调用执行所有必需的功能；和
2. 创建一个调用调度程序来解析来自命令行的用户命令。

下面我放入了 .c 文件，其中包含查询、添加和修改函数，但不包含删除函数。这可以使用 Visual Studio 创建的 .exe 文件从命令行运行，但它只是从上到下运行代码。我不知道如何单独调用程序的每个单独的函数。

#include <stdio.h>
#include <Windows.h>

int main()
{
    LONG lReg;
    HKEY hKey;
    DWORD number = 0x00000001;
    LONG dresult;

    if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"Software\\Test2\\Product\\SmartId", 0, KEY_SET_VALUE | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS)
        {
        printf("Key location open successful \n");
        dresult = RegDeleteKey(HKEY_LOCAL_MACHINE, L"Software\\Test2\\Product\\SmartId");
        RegCloseKey(hKey);
        }
    else 
        {
        lReg = RegCreateKeyEx(
        HKEY_LOCAL_MACHINE,
        L"Software\\Test2\\Product\\SmartId",
        0,
        NULL,
        REG_OPTION_NON_VOLATILE,
        KEY_ALL_ACCESS | KEY_WOW64_64KEY,
        NULL,
        &hKey,
        NULL);

        if (lReg == ERROR_SUCCESS)
            {
            printf("Key successfully added to registry \n");
            }
        else 
            {
            printf("Key not added to registry");
            }   
        }

    if (RegSetValueExW(hKey, L"OEMBackground", 0, REG_DWORD, (LPBYTE)&number, sizeof(DWORD)) == ERROR_SUCCESS)
        {
            printf("Key changed in registry \n");
        }
        else 
        {
            printf("Key not changed in registry \n");
        }

    RegCloseKey(hKey);

    return 0;
}

我希望命令行输入类似于(用于查询)

C:\Windows\system32>regfunction -q hklm\software\test2

Answer 1

您必须解析命令行

其中一种方法是 CommandLineToArgvW

例如(基于 SDK 示例)=>

包含/库:

#include <strsafe.h>
#include <shlwapi.h>
#pragma comment (lib,"Shlwapi.lib")

测试代码(例如 -q(查询)和 -d(删除)=>

BOOL bQuery = FALSE, bDelete = FALSE;
TCHAR wsParameter[MAX_PATH];
int nArgs;
LPWSTR *pszArgs = CommandLineToArgvW(GetCommandLine(), &nArgs);
if (pszArgs)
{       
    for (int iArg = 1; iArg != nArgs; iArg++)
    {
        LPTSTR pszArg = pszArgs[iArg];
        if (!StrCmpNIC(pszArg, TEXT("-q"), 3))
        {
            bQuery = TRUE;
            StringCchCopy(wsParameter, ARRAYSIZE(wsParameter), pszArg + 3);
            break;
        }
        else if (!StrCmpNIC(pszArg, TEXT("-d"), 3))
        {   
            bDelete = TRUE;
            StringCchCopy(wsParameter, ARRAYSIZE(wsParameter), pszArg + 3);
            break;
        }
        // else if ...
    }

    if (bQuery)
    {
        // Query code with wsParameter  
    }
    else if (bDelete)
    {
        // Delete code with wsParameter 
    }
    GlobalFree(pszArgs);        
}