c - 微软面试题这个函数有什么错误？

Question

我在 MS writing walkin-interview 中被问到这个问题:

在下面的程序中查找错误，该错误应该返回一个附加有 \n 的新字符串。

char* AddnewlinetoString(char *s)
{
  char buffer[1024];
  strcpy(buffer,s);
  buffer[strlen(s)-1] = '\n';
  return buffer;
}

我曾尝试自己编写代码，并通过将缓冲区变量设置为全局变量并设置 buffer[strlen(s)] = '\n' 使其正常工作。但是不知道里面还有很多其他的bug。

Answer 1

我可以看到一些:

Length of input string not checked.

如果 strlen(s) > 1023 怎么办？您最多可以在缓冲区中容纳长度为 1023 的字符串。

Overwriting the last char with \n

您正在用换行符覆盖最后一个字符。您的 \n 应该位于 \0 曾经所在的位置，您需要在 \n 之后添加一个新的 \0 >

Variable buffer is local to function and you are returning its address.

缓冲区的内存在堆栈上分配，一旦函数返回，该内存就会被释放。

我会这样做:

char* AddnewlinetoString(char *s) {

  size_t buffLen = strlen(s) + 2; // +1 for '\n' +1 for '\0'
  char *buffer = malloc(buffLen); 
  if(!buffer) {
   fprintf(stderr,"Error allocting\n");
   exit(1);
  }
  strcpy(buffer,s);
  buffer[buffLen-2] = '\n';
  buffer[buffLen-1] = 0;
  return buffer;
}