动态内存分配 +截断字符串问题

Question

为了编写一些基本函数来操作 C 字符串（char*），我一直在使用 malloc、realloc 和 free。从字符串中删除最后一个字符时，我遇到了这个奇怪的问题。我写了一个具有这样原型的函数：

int string_erase_end (char ** dst, size_t size);

它应该将“dst”字符串缩短一个字符。到目前为止，我已经想出了这段代码：

int string_erase_end (char ** dst, size_t size)
{
    size_t s = strlen(*dst) - size;
    char * tmp = NULL;
    if (s < 0) return (-1);
    if (size == 0) return 0;
    tmp = (char*)malloc(s);
    if (tmp == NULL) return (-1);
    strncpy(tmp,*dst,s);
    free(*dst);
    *dst = (char*)malloc(s+1);
    if (*dst == NULL) return (-1);
    strncpy(*dst,tmp,s);
    *dst[s] = '\0';
    free(tmp);
    return 0;
}

在 main() 中，当我截断字符串时（是的，我之前对它们调用了 malloc），我得到了奇怪的结果。根据我想要截断的字符数，它要么工作正常，要么截断错误数量的字符，要么引发分段错误。

我没有动态内存分配的经验，一直使用 C++ 及其 std::string 来完成所有这些肮脏的工作，但这次我需要在 C 中完成这项工作。如果有人帮助我找到并纠正我的问题，我将不胜感激。这里有错误。提前致谢。

Answer 1

第一个 strncpy() 不会在 tmp 末尾放置 '\0'。

另外，您可以避免重复复制： *dst = tmp;

Answer 2

根据您的描述，您的函数应该删除字符串中的最后 n 个字符：

/* Assumes passed string is zero terminated... */
void string_erase_last_char(char * src, int num_chars_to_erase)
{
    size_t len = strlen(src);

    if (num_chars_to_erase > len)
    {
        num_chars_to_erase = len;
    }

    src[len - num_chars_to_erase] = '\0';
} 

Answer 3

我不明白 size 参数的用途。

如果您的字符串最初是使用 malloc() 分配的，那么您应该只使用 realloc() 来更改它们的大小。这将自动保留内容，并且需要更少的操作：

int string_erase_end (char ** dst)
{
  size_t len;
  char *ns;

  if (dst == NULL || *dst == NULL)
   return -1;

  len = strlen(*dst);
  if (len == 0)
    return -1;

  ns = realloc(*dst, len - 1);
  if (ns == NULL)
   return -1;
  ns[len - 1] = '\0';
  *dst = ns;

  return 0;
}

在“现实世界”中，您通常不会更改 1 字符截断的分配大小；效率太低了。相反，您可以分别跟踪字符串的长度及其分配的大小。这使得弦更容易生长；只要已经分配了空间，追加字符就非常快。

此外，在 C 中，您永远不需要转换 malloc() 的返回值；它没有任何作用并且可以隐藏错误，所以不要这样做。