如何替换c中的子串？

Question

这个例子有效,但我认为内存泄漏.如果使用此功能,则在简单Web服务器模块中使用的函数会因共享内存而增加.

    char *str_replace ( const char *string, const char *substr, const char *replacement ){
      char *tok = NULL;
      char *newstr = NULL;
      char *oldstr = NULL;
      if ( substr == NULL || replacement == NULL ) return strdup (string);
      newstr = strdup (string);
      while ( (tok = strstr ( newstr, substr ))){
        oldstr = newstr;
        newstr = malloc ( strlen ( oldstr ) - strlen ( substr ) + strlen ( replacement ) + 1 );
        memset(newstr,0,strlen ( oldstr ) - strlen ( substr ) + strlen ( replacement ) + 1);
        if ( newstr == NULL ){
          free (oldstr);
          return NULL;
        }
        memcpy ( newstr, oldstr, tok - oldstr );
        memcpy ( newstr + (tok - oldstr), replacement, strlen ( replacement ) );
        memcpy ( newstr + (tok - oldstr) + strlen( replacement ), tok + strlen ( substr ), strlen ( oldstr ) - strlen ( substr ) - ( tok - oldstr ) );
        memset ( newstr + strlen ( oldstr ) - strlen ( substr ) + strlen ( replacement ) , 0, 1 );
        free (oldstr);
      }
      return newstr;
    }

Answer 1

我可以看到的一个问题是,如果替换字符串包含搜索字符串,您将永远循环(直到内存不足).

例如:

char *result = str_replace("abc", "a", "aa");

另外,每次更换一个实例时再做一次malloc/free是非常昂贵的.

更好的方法是在输入字符串上完成2次传递:

• 第一遍,计算搜索字符串的实例数

• 现在您知道了多少匹配,计算结果和malloc的长度一次:

  strlen(string)+ matches*(strlen(replacement)-strlen(substr))+ 1

• 第二次通过源字符串,复制/替换