连接字符串函数中的内存泄漏

问题描述

我正在尝试创建一个函数,将2个字符串(str1和str2)连接到由char分隔符分隔的新字符串(str3)中。不幸的是,此功能导致内存泄漏,我不知道为什么,因为我最后释放了str3。

示例:str_join_string(“ ABC”,“ DEF”,'|')--->“ ABC | DEF”

这是代码

char *str_join_string(const char *str1,const char *str2,char separator) {
  char *str3;
  size_t len = str_length(str1)+ str_length(str2)+1;
  size_t i = 0;
  size_t j = 0;

  str3 = (char * )calloc(len,sizeof(char));
  if(str3 == NULL){
    printf("Impossible d'allouer la mémoire");
    return NULL;
  }
  

  while(str1[i] != '\0' && str1 != NULL){
    str3[i] = str1[i];
    i++;
  }
  str3[i] = separator;
  i+=1;

  while(str2[j] != '\0' && str2 != NULL){
    str3[i+j] = str2[j];
    j++;
  }
  str3[len] = '\0';
  
  return str3;
} 

我要补充一点,我不能使用任何函数,例如strcat()或来自string.h的任何东西。

Valgrind显示内容

==4300== Searching for pointers to 3 not-freed blocks
==4300== Checked 131,560 bytes
==4300== 
==4300== 4 bytes in 1 blocks are definitely lost in loss record 1 of 3
==4300==    at 0x4C31B25: calloc (in /usr/lib/valgrind/vgpreload_memcheck-amd64-linux.so)
==4300==    by 0x13E3B2: str_join_string (stringslib.c:238)
==4300==    by 0x13E545: str_join_array (stringslib.c:283)
==4300==    by 0x137065: JoinArrayTest_OneEmpty_Test::TestBody() (stringslib_test.cc:779)
==4300==    by 0x1652A9: HandleSehExceptionsInMethodIfSupported<testing::Test,void> (gtest.cc:2611)
==4300==    by 0x1652A9: void testing::internal::HandleExceptionsInMethodIfSupported<testing::Test,void>(testing::Test*,void (testing::Test::*)(),char const*) (gtest.cc:2647)
==4300==    by 0x15A9DE: testing::Test::Run() [clone .part.658] (gtest.cc:2686)
==4300==    by 0x15AC61: Run (gtest.cc:2677)
==4300==    by 0x15AC61: testing::TestInfo::Run() [clone .part.659] (gtest.cc:2863)
==4300==    by 0x15B350: Run (gtest.cc:2837)
==4300==    by 0x15B350: testing::TestSuite::Run() [clone .part.660] (gtest.cc:3017)
==4300==    by 0x15BAF4: Run (gtest.cc:2997)
==4300==    by 0x15BAF4: testing::internal::UnitTestImpl::RunAllTests() (gtest.cc:5709)
==4300==    by 0x165769: HandleSehExceptionsInMethodIfSupported<testing::internal::UnitTestImpl,bool> (gtest.cc:2611)
==4300==    by 0x165769: bool testing::internal::HandleExceptionsInMethodIfSupported<testing::internal::UnitTestImpl,bool>(testing::internal::UnitTestImpl*,bool (testing::internal::UnitTestImpl::*)(),char const*) (gtest.cc:2647)
==4300==    by 0x15AD82: testing::UnitTest::Run() (gtest.cc:5292)
==4300==    by 0x11C08E: RUN_ALL_TESTS (gtest.h:2485)
==4300==    by 0x11C08E: main (stringslib_test.cc:799)
==4300== 

我希望你能帮助我,因为我现在真的迷路了。

------编辑------ 是的,我完全忘记添加调用者,这是释放内存的地方:

TEST(JoinStringTest,Simple) {
  char *buf = str_join_string("ABC","XYZ",'|');
  ASSERT_TRUE(buf != NULL);
  EXPECT_EQ(buf[0],'A');
  EXPECT_EQ(buf[1],'B');
  EXPECT_EQ(buf[2],'C');
  EXPECT_EQ(buf[3],'|');
  EXPECT_EQ(buf[4],'X');
  EXPECT_EQ(buf[5],'Y');
  EXPECT_EQ(buf[6],'Z');
  EXPECT_EQ(buf[7],'\0');
  free(buf);
}

解决方法

对于初学者,该函数将调用未定义的行为,因为没有足够的内存分配给结果字符串。

代替

size_t len = str_length(str1)+ str_length(str2)+1;

你必须写

size_t len = str_length(str1)+ str_length(str2)+2;

此外,这句话

str3[len] = '\0';

还尝试写入分配的数组之外的内存。

看来你的意思

str3[i + j] = '\0';

尽管您可以删除此语句,因为您正在使用函数calloc,该函数将分配的内存设置为零。另一方面,在功能上下文中使用calloc效率低下。

返回语句后的语句

//...
return str3;
free(str3);
str3 = NULL;

从不执行。

请注意诸如此类的for循环条件

while(str1[i] != '\0' && str1 != NULL){

没有任何意义。至少逻辑AND运算符的操作数应该像这样交换

while( str1 != NULL && str1[i] != '\0' ){

尽管在任何情况下条件str1 != NULL都是多余的,或者您可以在if语句中的循环之前检查条件。

这里是一个演示程序,显示了如何定义该函数(不使用标准字符串函数)和调用该函数。

#include <stdio.h>
#include <stdlib.h>

size_t str_length( const char *s )
{
    size_t n = 0;
    
    while ( *s++ ) ++n;
    
    return n;
}

char * str_join_string( const char *s1,const char *s2,char separator ) 
{
    size_t n = str_length( s1 ) + str_length( s2 ) + sizeof( separator ) + 1;
    
    char *s3 = malloc( n );
    
    if ( s3 )
    {
        char *p = s3;
        
        for ( ; *s1; ++s1 ) *p++ = *s1;
        
        *p++ = separator;
        
        for ( ; *s2; ++s2 ) *p++ = *s2;
        
        *p = '\0';
    }
    
    
    return s3;
}

int main(void) 
{
    char *s = str_join_string( "ABC","DEF",'|' );
    
    if ( s ) puts( s );
    
    free( s );
    
    return 0;
}

程序输出为

ABC|DEF

该函数的用户应提供不等于NULL的参数。

,
return str3;
free(str3);

看看这个片段,您是否认为free()会被打电话?

,

也许是因为您先返回了函数,然后释放了缓冲区!?

https://docs.microsoft.com/en-us/cpp/c-language/return-statement-c?view=vs-2019

return语句结束函数的执行并返回 控制调用功能