如何修复将文字字符串附加到 C 字符串的错误？

Question

我的最后一个功能不起作用。此函数将文字字符串附加到 C 字符串。它检查 C 字符串中是否有足够的空间来附加文字字符串。如果没有足够的空间，则必须将 C 字符串长度扩展为（字面字符串长度 + C 字符串长度）的两倍大小。然后它可以将文字字符串附加到 C 字符串中。在我运行程序并输入文本字符串后，会显示第一个输出语句，然后在我不断收到“在抛出 std::bad_alloc 的实例后调用终止”错误并且程序停止工作之后。所有其他功能在此附加功能之前工作。有没有办法修复最后一个附加功能工作？

int main()
{
char* s1 = assign();            
char* s2 = assign(" C++ "); 



char* s3 = add(s1, s2);
cout << "length of \"" << s3 << "\" is " << strlen(s3) << endl;


 append(s3, "programming language");   // This function doesn't work
 cout << "length of \"" << s3 << "\" is " << strlen(s3) << endl;

return 0;
}


char* assign()
{
const int SIZE = 100;
char temp[SIZE];
int length;
int twicelen;

cout << "Enter a text string which will be used to append literal strings to it: ";
cin.getline(temp, SIZE);
length = strlen(temp);
twicelen = length * 2;


char* newCstring = new char[twicelen];
strcpy(newCstring, temp);

return newCstring;
}



char* assign(string str)
{
int len = strlen(str.c_str());
int newlen = len * 2;
char* newStr = new char[newlen];

strcpy(newStr, str.c_str());;

return newStr;
}


char* add(char* s1, char* s2)
{
strcat(s1, s2);
return s1;
}


void append(char* s3, string literalStr)  // Every function before this works and this is where the program gives an error and closes.
{


if (sizeof(s3) < (strlen(s3) + strlen(literalStr.c_str()) + 1))
{
    int expandLength = (strlen(s3) + strlen(literalStr.c_str())) * 2;
    char* s3 = new char[expandLength];
    strcat(s3, literalStr.c_str());

}
else
    strcat(s3, literalStr.c_str());

}

Answer 1

问题 1：

您对add() 的实现可能会导致缓冲区溢出：

这是你的主要内容：

char* s1 = assign();  // here memory is allocated for the string you input          
char* s2 = assign(" C++ ");  // here memory is allocated again 

char* s3 = add(s1, s2);  // <====  ouch 

不幸的是，add() 只是在没有确保目标字符串有足够内存的情况下创建了一个 strcat()。从那时起，你就进入了 UB 的恐怖世界。任何事情都有可能发生。例如，字符串的结尾 null 可能会丢失，导致 strlen() 找到一个巨大的长度，并在您尝试分配两次如此巨大的数字时导致错误的内存异常。

问题 2：

您的append() 函数本身存在缺陷。

首先，sizeof(s3) 是指针 s3 的大小，所以是一个很小的数字。这不是分配字节的大小。因此，您很有可能会进入if 块（但出于错误的原因）。

接下来，您分配一个新的s3。问题是您在函数中存储在s3 中的值是函数的本地值。 main中的指针s3什么都不会发生，仍然指向原来的地方。

要纠正第二个问题，您需要通过引用传递指针。或者更改函数的签名以返回 s3 指针。在这种情况下，你会在 main 中写：s3 = append (s3, ...);