-1

我有这样的代码：

char buf[128];
char *data = NULL;
char *temp = NULL;

int full = 0;

int n = 0;

do {
     bzero(buf, sizeof(buf));
     n = recv(newsock, buf, sizeof(buf), 0);

     full += sizeof(buf);

     if(n != 0)
          temp = realloc(data, sizeof(buf));

     if(temp != NULL) {
          data = temp;
     } else {
         // error
     }

     memcpy(data + (full - sizeof(buf)), buf, sizeof(buf));

     printf("%s\n",data);
} while(n > 0);

在这段代码中，我尝试从套接字获取一些数据到缓冲区并将这些数据放入内存中。但我有问题。在 while 循环的第三次迭代中，我收到如下消息：

*** glibc detected *** ./server: realloc(): invalid next size: 0x09a4c008 ***

当我删除 memcpy() 函数时一切正常，但我怎样才能将数据放入内存？怎么了？

4

2 回答 2

1

问题是你不断地添加数据，但你没有扩展它的内存。线

temp = realloc(data, sizeof(buf));

总是分配 128 个字节。您需要更新它在每次迭代中将分配更多。也许你打算这样做：

temp = realloc(data, full);

?

于 2013-09-23T18:55:50.907 回答
1

我会使用更像这样的东西：

char buf[128];
char *data = NULL;
char *temp = NULL;
int n, datalen = 0, datacap = 0;

do
{
    n = recv(newsock, buf, sizeof(buf), 0);
    if (n < 0)
    {
        if ((errno != EAGAIN) && (errno != EWOULDBLOCK))
        {
            // error
            break;
        }

        fd_set fdr;
        FD_ZERO(&fdr);
        FD_SET(nwsock, &fdr);

        struct timeval to;
        to.tv_sec = 5;
        to.tv_usec = 0;

        n = select(newsock+1, &fdr, NULL, NULL, &to);
        if (n <= 0)
        {
            // error/timeout
            break;
        }

        continue;
    }

    if (n == 0)
    {
        // disconnected
        break;
    }

    if ((datalen + n) > datacap)
    {
        datacap = ((datalen + n) + (sizeof(buf)-1)) & ~(sizeof(buf)-1);
        // or:
        // datacap = (((datalen + n) + sizeof(buf)) / sizeof(buf)) * sizeof(buf);

        temp = realloc(data, datacap);
        if (temp == NULL)
        {
            // error
            break;
        }

        data = temp;
    }

    memcpy(data + datalen, buf, n);
    datalen += n;

    printf("%*.*s", n, n, buf);
}
while (1);
...
if (data) free(data);
于 2013-09-23T21:59:13.403 回答