socketServer

无人久伴 提交于 2020-08-04 11:09:24
#include<stdio.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<ctype.h>
#include<arpa/inet.h>

// =================================================
// =================================================
//                  宏定义
// =================================================
// =================================================
#define RECE_BUF_LEN 128 // receive buffer length
#define SEND_BUF_LEN 128 // send buffer length
#define PORT      0xa5a6 // port
#define SERVER_INIT_IP      "192.168.50.130"
#define CLIENT_INIT_IP      "192.168.50.129"


// =================================================
// =================================================
//                  全局变量
// =================================================
// =================================================
int g_uDebugFlag = 0;


// 服务器端文件描述符
int server_fd;

// 发送数据的客户端地址信息的结构体
struct sockaddr_in client_addr_from;
socklen_t socklen = sizeof(client_addr_from);

// 用于存储接收到的数据
char receBuf[RECE_BUF_LEN] = {'0'};
// int receLen;

// 用于存储发送的数据
char sendBuf[SEND_BUF_LEN] = {'s','e','r','v','e','r',' ',' ','s','e','n','d','!','!','!'};


struct sockaddr_in client_addr;
// sockaddr数据结构struct sockaddr_in中包含了IP地址协议类型、IP地址和端口号
struct sockaddr_in server_addr;

// =================================================
// =================================================
//                  函数
// =================================================
// =================================================
// 对struct sockaddr_in进行设置
int setSockaddr_in(struct sockaddr_in *pAddr, sa_family_t sin_family, const char *strptr, uint16_t hostshort){
    // 对socket数据结构内存初始化
    memset(pAddr, 0, sizeof(struct sockaddr_in));

    if(g_uDebugFlag){
        printf("%s at %d : hostlong = %s\n", __FUNCTION__, __LINE__, strptr);
    }
    
    (*pAddr).sin_family = sin_family;
    // 两种配置绑定IP的方式
    // 方式一
    inet_pton(sin_family, strptr, &(*pAddr).sin_addr.s_addr);
    // 方式二

    // (*pAddr).sin_port = htons(hostshort);
    (*pAddr).sin_port = htons(PORT); // 这里写死,统一使用同一个端口号

    // =============================
    // 打印查看一下
    // =============================
    // inet_ntop调用成功时,这个指针就是inet_ntop函数的返回值
    char str[INET_ADDRSTRLEN] = {0};
    printf("set struct sockaddr_in IP: %s; port %d.\n", inet_ntop(AF_INET, &(*pAddr).sin_addr, str, sizeof(str)), ntohs((*pAddr).sin_port));
    // printf("str = %s\n", str);

    return 0;
}

int getSockaddr_in(struct sockaddr_in* pAddr){
    // inet_ntop调用成功时,这个指针就是inet_ntop函数的返回值
    char str[INET_ADDRSTRLEN] = {0};
    printf("get struct sockaddr_in IP: %s; port %d.\n", inet_ntop(AF_INET, &(*pAddr).sin_addr, str, sizeof(str)), ntohs((*pAddr).sin_port));
    // printf("str = %s\n", str);

    return 0;
}

// 默认初始化服务端
int serverInit(){
    // ==========================================================
    // 1. 创建socket
    // ==========================================================
    if((server_fd = socket(AF_INET, SOCK_DGRAM, 0)) == -1){
        printf("%s at %d : server_fd created fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
    }
    // ==========================================================
    // 2. 通过struct sockaddr_in 结构体设置服务器地址和监听端口
    // ==========================================================
    if(setSockaddr_in(&server_addr, AF_INET, SERVER_INIT_IP, PORT) != 0){
        printf("%s at %d : setSockaddr_in set fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
    }

    // ==========================================================
    // 3. 使用bind()函数绑定监听端口,将套接字文件描述符和地址类型
    //    变量(struct sockaddr_in)进行绑定
    // ==========================================================
    if(bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1){
        printf("%s at %d : bind fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
    }
    
    return 0;
}

int serverSend(char* sendBuf, int sendBufLen){
    printf("=================== begin to send message =====================\n");
    // ==========================================================
    // 5. 向客户端发送数据,是用sendto()函数向服务器主机发送数据
    // ==========================================================
    int sendLen = 0;
    sendLen = sendto(server_fd, sendBuf, sendBufLen, 0, (struct sockaddr*)&client_addr, sizeof(client_addr));

    // inet_ntop调用成功时,这个指针就是inet_ntop函数的返回值
    char str[INET_ADDRSTRLEN] = {0};
    printf("send to %s at port %d.\n", inet_ntop(AF_INET, &client_addr.sin_addr, str, sizeof(str)), ntohs(client_addr.sin_port));
    // printf("str = %s\n", str);

    printf("sendLen = %d\n", sendLen);

    if(sendLen <= 0){
        printf("%s at %d : sendLen  0 fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
    }

    // =========================================================
    // =========================================================
    // 打印发送内容
    // =========================================================
    // =========================================================
    printf("sendBuf: %s\n", sendBuf);
    printf("=================== now send message over =====================\n");
    printf("\n");

    // =========================================================
    // =========================================================
    // 发送结束之后,需要把发送缓存清空
    // 如果不清空,如果再下次接收的个数少于之前的内容,
    // 那么再次发送的时候,会把上次保留的也发出来
    // =========================================================
    // =========================================================
    memset(sendBuf, 0, sendBufLen);
    // ==========================================================
    // 6. 关闭套接字,是用close()函数释放资源
    // 如果在这里关闭套接字,那么,server_fd就被释放了,又因为server_fd
    // 是全局变量,server_fd释放后,就不再是socket了,recfrom就不会再
    // 阻塞了,就导致了无限循环
    // ==========================================================
    // close(server_fd);
    return 0;
}

int serverReceive(){
    // ==========================================================
    // 4. 接收客户端的数据,使用recvfrom()函数接收客户端的网络数据
    // ==========================================================
    while(1){
        printf("=================== begin to receive message =====================\n");
        int receLen;
        receLen = recvfrom(server_fd, receBuf, sizeof(receBuf), 0, (struct sockaddr*)&client_addr_from, &socklen);
        
        // inet_ntop调用成功时,这个指针就是inet_ntop函数的返回值
        char str[INET_ADDRSTRLEN] = {0};
        printf("receive from %s at port %d.\n", inet_ntop(AF_INET, &client_addr_from.sin_addr, str, sizeof(str)), ntohs(client_addr_from.sin_port));
        // printf("str = %s\n", str);

        printf("receLen = %d\n", receLen);
        if(receLen <= 0){
            printf("%s at %d : receLen  0 fail !!!\n", __FUNCTION__,__LINE__);
            continue;
        }

        // =========================================================
        // =========================================================
        // 打印接收内容
        // =========================================================
        // =========================================================
        printf("receBuf = %s\n", receBuf);
        printf("=================== now receive message over =====================\n");
        printf("\n");

        // 正常功能用这个
        // if(setSockaddr_in(&client_addr, AF_INET, inet_ntop(AF_INET, &client_addr_from.sin_addr, str, sizeof(str)), ntohs(client_addr_from.sin_port)) != 0){
        //     printf("%s at %d : setSockaddr_in set fail !!!\n", __FUNCTION__,__LINE__);
        //     continue;
        // }

        // 调试用,正常功能用上面的
        if(setSockaddr_in(&client_addr, AF_INET, CLIENT_INIT_IP, PORT) != 0){
            printf("%s at %d : setSockaddr_in set fail !!!\n", __FUNCTION__,__LINE__);
            continue;
        }

        if(serverSend(receBuf, receLen) != 0){
            printf("%s at %d : serverSend fail !!!\n", __FUNCTION__,__LINE__);
            continue;
        }
    }
    // ==========================================================
    // 5. 关闭套接字,是用close()函数释放资源
    // ==========================================================
    close(server_fd);
    return 0;
}


int main(){

   if(serverInit() != 0){
        printf("%s at %d : serverInit fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
   }

   if(getSockaddr_in(&server_addr) != 0){
        printf("%s at %d : getSockaddr_in server_addr fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
   }

   if(serverReceive() != 0){
        printf("%s at %d : serverReceive fail !!!\n", __FUNCTION__,__LINE__);
        return -1;
   }
   
}

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!