获取套接字协议地址函数

根据套接字获取本地协议地址(getsockname)

【头文件】

#include <sys/socket.h>

【函数原型】

int getsockname(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

【函数功能】

        该函数可以获取 sockfd 套接字的协议地址信息，放到 addr 中，特别适合未调用 bind()就

调用了 connect()，这时唯有 getsockname()调用可以获知系统内定的本地地址。

【参数含义】

        [sockfd]: 通信套接字

        [addr]:保存获取到的协议地址

        [addrlen]:保存获取到的协议地址的长度

【返回值】

        如果函数调用成功，则返回 0 ，如果调用出错，则返回 -1 。

【示例】

#define PORT 6563
#include<stdio.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<unistd.h>
#include<string.h>
#include<arpa/inet.h>

int main(int argc, char **argv) 
{
    struct sockaddr_in servaddr;//服务器端地址
    struct sockaddr_in clientAddr;//客户端地址
    int sockfd; 
    int clientAddrLen = sizeof(clientAddr);
    char ipAddress[INET_ADDRSTRLEN];//保存点分十进制的 ip 地址
    if(argc < 2) 
    {
        printf("parameter error");
        return -1;
    }
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(PORT); 
    if(inet_pton(AF_INET, argv[1], &servaddr.sin_addr) <= 0) 
    {
        printf("server address error\n");//地址参数不合法
    }
    int r = connect(sockfd, (struct sockaddr *)&servaddr, 
    sizeof(servaddr));//向服务器端发起连接请求
    if(r == -1) //连接失败
    {
        perror("connect error");
        return -1;
    }
    getsockname(sockfd, (struct sockaddr*)&clientAddr, &clientAddrLen);//获取 sockfd 表示的连接上的本地地址
    printf("client:client ddress = %s:%d\n", inet_ntop(AF_INET, &clientAddr.sin_addr, ipAddress, sizeof(ipAddress)), ntohs(clientAddr.sin_port));
    return 0;
}

获取与套接字关联的外地协议地址(getpeername)

【头文件】

#include <sys/socket.h>

【函数原型】

int getpeername(int sockfd, struct sockaddr *addr, socklen_t *addrlen);

【函数功能】

        该函数可以获取 listen 监听的 sockfd 套接字的外地协议地址信息。适用与当服务器调用 accept 函数连接客户端后进入其他线程或进程，就只能使用 getpeername 获取客户端的协议地址信息了。

【参数含义】

        [sockfd]: 通信套接字

        [addr]:保存获取到的协议地址

        [addrlen]:保存获取到的协议地址的长度

【返回值】

        如果函数调用成功，则返回 0 ，如果调用出错，则返回 -1 。

【示例】

void server_run(struct GLZ *pool) //建立线程去处理客户端的连接后的操作
{
    while (1)
    {
        //阻塞等待连接
        struct sockaddr_in caddr; //保存客户端的地址(IP+port)
        socklen_t len = sizeof(caddr);
        int *confp = (int *)malloc(sizeof(int));
        *confp = accept(ssocket,(struct sockaddr *)&caddr,&len);
        if(*confp > 0) //客户端连接成功,返回一个连接套接字专门用来和客户端通信
        {
            //一个客户端连接成功.开一个进程/线程去处理这个连接
            printf("客户端 IP:%s,客户端Port:%d\n",inet_ntoa(caddr.sin_addr),ntohs(caddr.sin_port));
            bool add=add_task(pool,client_handle,(void*)confp);
            if(add==false)
            {
                printf("添加任务失败\n");
                return;
            }
        }
    }
}

void* client_handle(void* arg)
{
    int client = *(int*)arg; 
    struct sockaddr_in addr; //地址信息结构体
    socklen_t len = sizeof(addr);
    getpeername(client, (struct sockaddr*)&addr, &len); //获取已连接套接口的地址
    printf("[%s] online...\n", inet_ntoa(addr.sin_addr)); //输出连接客户端的 ip 地址
    while (1)
    {
        char buf[1024];
        memset(buf,0,sizeof(buf));
        int ret = read(client, buf, sizeof(buf)); //读取 client 内容到 buf中
        if (ret <= 0) //客户端断开
        {
            printf("[%s] offline...\n", inet_ntoa(addr.sin_addr)); //输出客户端脱离连接
            close(client); //关闭套接字
            free(arg);
            return NULL;
        }
        printf("%s",buf);
    }
}

【备注】

        UDP 没法使用