Question

我正在使用套接字和线程处理客户端 - 服务器程序。我有一个客户端菜单，我正在尝试从客户端向服务器发送一个字符，然后调用相应的函数。我正在尝试使用指针和malloc，但我不认为我完全理解如何传递它们并释放它们。我遇到了内存泄漏和错误，如：

./ server'：双重免费或损坏（fasttop）：0x00007f37dc000a50 ***

和

分段错误

这是我的客户菜单：

char *choice = (char*) malloc(sizeof(char));

do {

    // get a string from the server
    get_msg(sockfd);

    printf("\nFile and Information System\n");
    printf("===========================\n");
    printf("1. Show IP and Student ID\n");
    printf("2. Display server time\n");
    printf("3. Display system information\n");
    printf("4. List files on server\n");
    printf("5. File Transfer\n");
    printf("6. Exit\n");
    printf("Enter choice: ");
    scanf("%s", choice);

    //Send input to server
    send_menu_choice(sockfd, choice);

    switch (*choice) 
    {
        case '1':
                get_ip(sockfd);
            break;
        case '2':
                get_time(sockfd);
            break;
        case '3':
                get_and_send_uname(sockfd);
            break;
        case '4':
                get_file_list(sockfd);
            break;
        case '5':
                printf("File transfer not yet implemented\n");;
            break;
        case '6':
                printf("Exiting...\n");
            break;
        default:
                printf("Invalid choice\n");
    }

    //if(menu_choice != NULL)
    //{
    //    free(menu_choice);
    //}

} while (*choice != '6');

if (choice != NULL)
{
    free(choice);
}

将菜单选项发送到服务器：

void send_menu_choice(int socketNumber, char *choice)
{
    printf("Sending menu choice...\n");
    size_t n = strlen(choice) + 1;

    writen(socketNumber, (unsigned char *) &n, sizeof(size_t));
    writen(socketNumber, (unsigned char *) choice, n);

    printf("Sent choice: %s\n\n", choice);
}

服务器端：

char *menu_choice = (char*) malloc(sizeof(char));

do { 

    printf("Waiting for client to select option...\n\n");
    send_msg(connfd);

    get_menu_choice(connfd, menu_choice);
    printf("Client %d choice was: %s\n", connfd, menu_choice);

    switch (*menu_choice) 
    {
        case '1':
                send_ip(connfd);
            break;
        case '2':
                send_time(connfd);
            break;
        case '3':
                get_and_send_uname(connfd, *uts);
            break;
        case '4':
                send_file_list(connfd);
            break;
        case '5':
                printf("File Transfer not implemented\n");
            break;
        case '6': 
            break;
        default:
                printf("Invalid choice\n");
    }

    //if(menu_choice != NULL)
    //{
    //    free(menu_choice);
    //}

} while (*menu_choice != '6');

if (choice != NULL)
{
    free(choice);
}

从客户端获取菜单选项：

void get_menu_choice(int socketNumber, char *choice)
{
    size_t n;

    readn(socketNumber, (unsigned char *) &n, sizeof(size_t));  
    readn(socketNumber, (unsigned char *) choice, n);

    printf("Received: %zu bytes\n\n", n);
}

<小时/> 要求的其他功能：

发送消息（服务器端）：

void send_msg(int socket)
{
    char msg_string[] = "\nPlease enter an option:";

    size_t n = strlen(msg_string) + 1;
    writen(socket, (unsigned char *) &n, sizeof(size_t));   
    writen(socket, (unsigned char *) msg_string, n);
}

获取消息（客户端：

void get_msg(int socket)
{
    char msg_string[32];
    size_t k;

    readn(socket, (unsigned char *) &k, sizeof(size_t));    
    readn(socket, (unsigned char *) msg_string, k);

    printf("%s\n", msg_string);
}

我在这里做错了什么？我不影响指针所指向的值吗？

Answer 1

在do循环中，您反复释放指针

if(menu_choice != NULL)
{
    free(menu_choice);
}

但是你只在循环之前分配一次。

Answer 2

请允许我将循环压缩到以下关键行：

char *choice = (char*) malloc(sizeof(char));

do {

  scanf("%s", choice);
  switch (*choice) {
    case '6':
      printf("Exiting...\n");
      break;
    default:
      printf("Invalid choice\n");
  }

  if (choice != NULL)
  {
    free(choice);
  }

} while (*choice != '6');

这个循环存在严重缺陷。您不仅使用1个字节的内存来存储2（scanf应使用%c而不是%s），但如果您选择6以外的任何内容，则循环不仅会访问释放内存，但它也会导致双重释放（您的错误）。另一方面，如果选择6，则在释放内存后访问内存，这也是一个非常严重的错误。

循环结束后你应该释放内存。

Answer 3

服务器端，

char *menu_choice = (char*) malloc(sizeof(char));

您分配一个字符。

然后在 get_menu_choice（）中使用分配的get_menu_choice(connfd, menu_choice) 1字节调用menu_choice，

void get_menu_choice(int socketNumber, char *choice)
{
   size_t n;

   readn(socketNumber, (unsigned char *) &n, sizeof(size_t));  
   readn(socketNumber, (unsigned char *) choice, n);

将n字节存储到choice ...

这似乎不正确。

一个字节分配足够吗？意思是，n取值＆gt; 1？

Answer 4

这是一个通用答案，以下是我在项目中使用的函数：

/* this is safe free() and no not what you asked, check next one */
void my_free(void *ptr)
{
    if (ptr) {
        free(ptr);
    }
}

/* this is almost same as safe free above but also sets the pointer to NULL */
void my_freep(void **ptr)
{
    if (*ptr != NULL) {
        free(*ptr);
    }

    *ptr = NULL;
}

并像my_freep一样致电my_freep((void**)&buffer); 注意buffer此处被定义为指针（即char *buffer;）希望有所帮助。

C：如何将malloc指针传递给函数并释放它们？

4 个答案: