网络编程day2-(搭建TCP服务器&客户端，UDP服务器&客户端)

作业一、TCP服务器&客户端代码

TCP服务器

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define ERR_MSG(msg)                            \
    do                                          \
    {                                           \
        fprintf(stderr, "line:%d", __LINE__); \
        perror(msg);                            \
    } while (0)

#define IP "192.168.5.156"  //本机IP地址
#define PORT 1562           //自定义端口号1023~49151

char buf[128] = "";         //用来存客户端发送的信息

int main(int argc, const char *argv[])
{   
    //创建套字节，在内核中创建两个缓冲区，用户空间可以接收这两个文件描述符
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket success __%d__\n", __LINE__);

    // 允许端口快速重启
    int reuse = 1;
    if ((setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重启成功");

    //填充地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family          = AF_INET;          //必须填写AF_INET
    sin.sin_port            = htons(PORT);      //端口号1023~49151 要转为网络字节序 
    sin.sin_addr.s_addr     = inet_addr(IP);    //本机IP地址 要转为网络字节序

    //将IP和端口绑定到套字节上
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success__%d__\n", __LINE__);

    //将套节字设为监听状态，监听是否有客户端链接成功
    if (listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success__%d__\n", __LINE__);

    struct sockaddr_in cin;           //存储链接成功的客户端信息
    socklen_t addrlen = sizeof(cin);  //计算结构体信息大小

    //重点！！！
    // 阻塞函数，从自己完成连接的队列头中获取一个客户端信息，生成一个新的文件描述符
    // 该文件描述符才是与客户端通信的文件描述符
    int newfd = accept(sfd, (struct sockaddr *)&cin, &addrlen);
    if (newfd < 0)
    {
        ERR_MSG("accept");
        return -1;
    }
    printf("[%s:%d] newfd = %d 连接成功__%d__\n",
           inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);

    ssize_t res = 0;
    while (1)
    {
        bzero(buf, sizeof(buf));
        //接收
        res = recv(newfd, buf, sizeof(buf), 0);
        if (res < 0)
        {
            ERR_MSG("recv");
            return -1;
        }
        else if (0 == res)
        {
            printf("[%s:%d] newfd=%d 客户端下线__%d__\n",
                   inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);
            break;
        }
        printf("[%s:%d] newfd=%d: %s __%d__", 
               inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd, buf, __LINE__);

        //发送
        strcat(buf, "*-*");
        if (send(newfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            return -1;
        }
        printf("发送成功\n");
    }

    //关闭文件
    close(sfd);
    close(newfd);
    return 0;
}

TCP客户端

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define ERR_MSG(msg)                            \
    do                                          \
    {                                           \
        fprintf(stderr, "line:%d", __LINE__); \
        perror(msg);                            \
    } while (0)

#define IP "192.168.5.156"      //本机IP地址
#define PORT 1562               //自定义端口号1023~49151

char buf[128] = "";             //用来给服务器发送数据

int main(int argc, const char *argv[])
{
    //创建套字节，在内核中创建两个缓冲区，用户空间可以接收这两个文件描述符
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if (cfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("cfd = %d\n",cfd);

    int reuse = 1;
    if ((setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重启成功\n");

    // 绑定客户端的地址信息结构体，非必绑定的
    // 如果不绑定，则有操作系统自动选择一个端口号，以及本机可用ip绑定到套字节上

    // 填充服务器的地址信息结构体，提供给connect函数使用
    // 真实的地址信息结构体根据地址族指定 AF_INET：man 7 ip
    
    // 填充地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;
    sin.sin_port = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);

    //连接服务器
    if (connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("connect success__%d__\n", __LINE__);

    ssize_t res = 0;
   
    while (1)
    {
        // 发送
        bzero(buf, sizeof(buf));
        printf("请输入>>>");
        fgets(buf, sizeof(buf), stdin);
        buf[sizeof(buf) - 1] = 0;
        if (send(cfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            return -1;
        }
        printf("发送成功\n");

        // 接收
        bzero(buf, sizeof(buf));
        res = recv(cfd, buf, sizeof(buf), 0);
        if (res < 0)
        {
            ERR_MSG("recv");
            return -1;
        }
        else if (0 == res)
        {
            printf("cfd=%d 客户端下线__%d__\n", cfd, __LINE__);
            break;
        }
        printf("cfd=%d: %s __%d__\n", cfd, buf, __LINE__);
    }

    close(cfd);
    return 0;
}

作业二、UDP服务器&客户端代码

UDP服务器

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define ERR_MSG(msg)                          \
    do                                        \
    {                                         \
        fprintf(stderr, "line:%d", __LINE__); \
        perror(msg);                          \
    } while (0)

#define IP "192.168.5.156" // 本机IP地址
#define PORT 1562          // 自定义端口号1023~49151

char buf[128] = ""; // 接收客户端的消息

int main(int argc, const char *argv[])
{
    // 创建套字节，在内核空间创建两个缓冲区，一个接收，一个发送
    int sfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("sfd success__%d__\n", __LINE__);

    // 填充服务器端口的地址信息
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;            // 必须填写AF_INET
    sin.sin_port = htons(PORT);          // 自定义的端口字节序
    sin.sin_addr.s_addr = inet_addr(IP); // 本机IP

    // 必须绑定，将IP和端口号绑定到套接字
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success__%d__\n", __LINE__);

    struct sockaddr_in cin;          // 存储数据包是从谁那里来的
    socklen_t addrlen = sizeof(cin); // 数据地址的大小

    ssize_t res = 0;
    while (1)
    {
        bzero(buf, sizeof(buf));
        // 接收
        res = recvfrom(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, &addrlen);
        if (res < 0)
        {
            ERR_MSG("rrecvfrom");
            return -1;
        }
        else if (0 == res)
        {
            printf("[%s:%d] 客户端已下线\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
            break;
        }
        printf("[%s:%d]:%s", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), buf);

        // 回应
        strcat(buf,"*-*");
        if (sendto(sfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, addrlen) < 0)
        {
            ERR_MSG("sendto");
            return -1;
        }
        printf("发送成功\n");
    }

    // 关闭
    close(sfd);
    return 0;
}

UDP客户端

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define ERR_MSG(msg)                        \
do                                          \
{                                           \
    fprintf(stderr,"line:%d",__LINE__);     \
    perror(msg);                            \
}while(0)

#define IP "192.168.5.156"       //本机IP第地址
#define PORT 1562               //自定义端口号1023~49151

char buf[128]="";               //给服务器发送数据

int main(int argc,const char * argv[])
{
    //创建套接字，在内核空间创建两个缓冲区，一个用于接收，一个用于发送
    int cfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(cfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket success__%d__\n",__LINE__);

    //填写服务器地址信息，给sendto函数用
    struct sockaddr_in cin;
    cin.sin_family = AF_INET;
    cin.sin_port = ntohs(PORT);
    cin.sin_addr.s_addr = inet_addr(IP);

    //UDP客户端不是必须绑定
    /*
    if (bind(cfd, (struct sockaddr *)&cin, sizeof(cin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    */
    
    //
    struct sockaddr_in recvaddr;               //数据包是从谁那里来
    socklen_t addrlen = sizeof(recvaddr);      //计算地址结构的大小

    ssize_t res = 0;
    while(1)
    {   
        bzero(buf, sizeof(buf));
        //发送
        printf("请输入>>>");
        fgets(buf, sizeof(buf), stdin);
        if(sendto(cfd, buf, sizeof(buf), 0, (struct sockaddr *)&cin, sizeof(cin)) < 0)
        {
            ERR_MSG("sendto");
            return -1;
        }
        printf("发送成功\n");

        //接收，必须接收数据包的发送地址信息，给sendto使用
        res = recvfrom(cfd, buf, sizeof(buf), 0, (struct sockaddr *)&recvaddr, &addrlen);
        if(res < 0)
        {
            ERR_MSG("recvfrom");
            return -1;
        }
        else if(0 == res)
        {
            printf("客户端已下线\n");
            break;
        }
        printf("[%s:%d]:%s\n", inet_ntoa(recvaddr.sin_addr), ntohs(recvaddr.sin_port), buf);
    }

    //关闭文件
    close(cfd);
    return 0;
}