c++ - 无缝知道包含空字符的字符串的大小

Question

对于一个非常简单和新手的问题来说，这是一个很长的问题，但我需要一些建议。

背景

所以我有一个需要解析的二进制文件。该文件以一些包含空字符 (\0) 的神奇字符串开头。让我们定义为 ab\0cd。

我正在编写一个方法，如果某个文件以魔术字符串开头，则返回 true。

尝试 1

#define MAGIC_STRING "ab\0cd"

bool IsMagicFile(const wpath& pathFile)
{
    string strData;
    if (!ReadFile(pathFile, strData))
        return false;

    if (strData.size() < 5)
        return false;

    string strPrefix = strData.substr(0, 5);

    if (strcmp(strPrefix.c_str(), MAGIC_STRING) != 0)
        return false;

    return true;
}

问题1

上面的代码让我感到困扰的是，我“硬编码”假设魔术字符串的大小是 5。

如果明天魔法弦变了怎么办？说:

#define MAGIC_STRING "abe\0fcd"

字符串宏已更改，代码不再正常工作。

尝试 2

#define MAGIC_STRING "ab\0cd"

bool IsMagicFile(const wpath& pathFile)
{
    string strMagic = MAGIC_STRING;

    string strData;
    if (!ReadFile(pathFile, strData))
        return false;

    if (strData.size() < strMagic.size())
        return false;

    string strPrefix = strData.substr(0, strMagic.size());

    if (strcmp(strPrefix.c_str(), MAGIC_STRING) != 0)
        return false;

    return true;
}

问题2

我应该摆脱了硬编码的大小问题，但 strMagic 的大小实际上不是 5，而是 2。字符串以 \0

结尾

尝试 3

#define MAGIC_STRING        "ab\0cd"    // CAUTION - MAGIC_STRING & MAGIC_STRING_SIZE must be changes together 
#define MAGIC_STRING_SIZE   5           // CAUTION - MAGIC_STRING & MAGIC_STRING_SIZE must be changes together

bool IsMagicFile(const wpath& pathFile)
{
    string strData;
    if (!ReadFile(pathFile, strData))
        return false;

    if (strData.size() < MAGIC_STRING_SIZE)
        return false;

    string strPrefix = strData.substr(0, MAGIC_STRING_SIZE);

    if (strcmp(strPrefix.c_str(), MAGIC_STRING) != 0)
        return false;

    return true;
}

问题3

这解决了第一个问题，但我仍然没有得到我想要的无缝魔术字符串更改。

问题

尝试 3 是否足够好？你有更好的方法吗？

Answer 1

您可以定义常量字符数组，而不是使用宏定义。例如

const char MAGIC_STRING[] = "abe\0fcd";

在这种情况下，不包括终止零的字符数等于

sizeof( MAGIC_STRING ) - 1

要比较原始字节，您可以使用标准 C 函数 memcmp 提供等于上述表达式的比较字节数。

这是一个演示程序

#include <iostream>
#include <string>
#include <cstring>
#include <iterator>

const char MAGIC_STRING[] = "abe\0fcd";

int main() 
{
    std::string s( std::begin( MAGIC_STRING ), std::prev( std::end( MAGIC_STRING ) )  );

    if ( memcmp( s.c_str(), MAGIC_STRING, sizeof( MAGIC_STRING ) - 1 ) == 0 )
    {
        std::cout << "The string starts with the MAGIC_STRING" << std::endl;
    }

    return 0;
}

它的输出是

The string starts with the MAGIC_STRING