[c语言]-注意创建指针时的内存区域是栈内存 or 堆内存

   之前用c语言写了个udp_socket的代码，但是一直报错，花了好长时间才找到原因，特此记录。

   上代码：

#include <stdio.h>
#include <stdlib.h>
#include "string.h"
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <arpa/inet.h>


void createSockAddr(char *ip,int port ,struct sockaddr * tmp){
    //创建sockaddr_in结构体变量
    struct sockaddr_in addr;
    //每个字节都用0填充
    memset(&addr,0,sizeof(addr));

    //使用IPv4地址
    addr.sin_family=AF_INET;
    //具体的IP地址
    addr.sin_addr.s_addr=inet_addr(ip);
     //端口
    addr.sin_port=htons(port);

    tmp=(struct sockaddr *)&addr;
}


int main(void)
{
    /* code */

    // 1. 声明 udp socket，
    int udp_sock=socket(AF_INET,SOCK_DGRAM,0);
    if(udp_sock<0){
        perror("get socket fail \n");
        exit(EXIT_FAILURE);
    }

    // 2. 创建本地sockaddr
    struct sockaddr * local_addr;

    createSockAddr("127.0.0.1",9002,local_addr);

    if(local_addr==NULL){
        perror("get local_addr fail \n");
        exit(EXIT_FAILURE);
    }

    // 3. 监听端口
    int bind_result=bind(udp_sock,local_addr,sizeof(struct sockaddr_in ));

    if(bind_result == -1){
        printf("bind fail!\n");
        return -1;
    }


    // 2. 创建server addr
    struct sockaddr * server_addr;

    createSockAddr("127.0.0.1",2000,server_addr);

    if(server_addr==NULL){
        perror("get server_addr fail \n");
        exit(EXIT_FAILURE);
    }

    // 5. 连接server
    int conn_result=connect(udp_sock,server_addr,sizeof(struct sockaddr ));

    if(conn_result < 0 ){
        perror("connect fail \n");
        exit(EXIT_FAILURE);
    }

    // 6, 发送消息
    char * msg="hello i'm c_client";
    size_t msg_len=strlen(msg);
    ssize_t  w_result=sendto(udp_sock,msg,msg_len,0,server_addr,sizeof(struct sockaddr ));

    if(w_result==-1){
        perror("send fail \n");
        exit(EXIT_FAILURE);
    }

    close(udp_sock);

    return 0;
}

  有问题的就是struct sockaddr 指针的创建方式：

 

  首先在 createSocketAddr函数的外部声明了一个 struct socketaddr  的指针变量 local_addr，然后想的是复用代码，就写了个 createSocketAddr 函数，将指针当做参数传进去以便赋值。

 

 

  有经验的一定会发现这种写法存在问题，并且编译器也会发出一个类似 “返回了栈内存的变量地址”的警告。

  错误的原因正是由于指针指向的数据结构是在函数createSocketAddr ( ) 内部直接创建的，所以该地址是在栈内存中，当退出函数时由于出栈会导致该地址其实属于一种“无效内存”。这种"无效内存"中的数据可能会被分配给其他数据结构从而改变了数据，或者直接产生警告或者报错。

 正确的写法应该是在堆内存中申请一块区域用来存放数据，通过指针将该结构传入函数内部中进行处理，并且执行完后由于是申请的堆内存，一定要手动free回收；

 正确的代码

#include <stdio.h>
#include <stdlib.h>
#include "string.h"
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <arpa/inet.h>


struct sockaddr_in * createSockAddr(char *ip,int port ){
    // 使用malloc从堆内存中分配一块 struct sockaddr_in 的区域
    struct sockaddr_in * tmp = (struct sockaddr_in *)malloc(sizeof(struct sockaddr_in));

    // 填充 tmp 的内存区域
    //每个字节都用0填充
    memset(tmp,0,sizeof(struct sockaddr_in));

    //使用IPv4地址
    (*tmp).sin_family=AF_INET;
    //具体的IP地址
    (*tmp).sin_addr.s_addr=inet_addr(ip);
     //端口
    (*tmp).sin_port=htons(port);

    return tmp;
}


int main(void)
{
    /* code */

    // 1. 声明 udp socket，
    int udp_sock=socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP);
    if(udp_sock<0){
        perror("get socket fail \n");
        exit(EXIT_FAILURE);
    }

    // 2. 创建本地sockaddr, 初始化指针==NULL，表示指针指向未分配的地址0x00,不能对该地址进行 (*操作)，相当于指针未分配空间进行初始化；

    struct sockaddr_in * local_addr=createSockAddr("127.0.0.1",9002);;

    if(local_addr==NULL){
        perror("get local_addr fail \n");
        exit(EXIT_FAILURE);
    }

    // 3. 监听端口
    int bind_result=bind(udp_sock,(const struct sockaddr *)local_addr,sizeof(struct sockaddr_in ));

    if(bind_result == -1){
        perror("bind fail \n");
        exit(EXIT_FAILURE);
    }


    // 2. 创建server addr
    struct sockaddr_in * server_addr =  createSockAddr("10.108.132.161",2000);;


    if(server_addr==NULL){
        perror("get server_addr fail \n");
        exit(EXIT_FAILURE);
    }

    // 5. 连接server
    int conn_result=connect(udp_sock,(const struct sockaddr *)server_addr,sizeof(struct sockaddr_in ));

    if(conn_result < 0 ){
        perror("connect fail \n");
        exit(EXIT_FAILURE);
    }

    // 6, 发送消息
    char * msg="hello i'm c_client\n";
    
    ssize_t  w_result=write(udp_sock,msg,strlen(msg));

    if(w_result<0){
        perror("send fail \n");
        exit(EXIT_FAILURE);
    }

    free(server_addr);
    free(local_addr);
    close(udp_sock);

    return 0;
}