TCP text communication simulation (C)

Transfer from: https://blog.csdn.net/jinmie0193/article/details/79913462 

    

server.c 

//============================================================================
// Name        : livePlayer.cpp
// Author      : ffy
// Version     :
// Copyright   : Your copyright notice
// Description : Hello World in C++, Ansi-style
//============================================================================
#include <sys/types.h>
#include <sys/socket.h> //包含套接字的函数库
#include <netinet/in.h> //包含AF_INET的相关结构
#include <arpa/inet.h>  //包含AF_INET的操作函数
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

/*
 监听后,一直处于accept阻塞状态, 直到有客户端连接,
输入EOF后,断开与客户端的连接,输入quit关闭服务器
读取EOF后,断开和此客户端连接,读取quit关闭服务器
 */

#define PORT 3333
int main(void)
{
    printf("程序开始\n");

    int s_fd,c_fd;                  //服务器和客户端的套接字标识符
    int s_len,c_len;                //服务器和客户端的消息长度
    struct sockaddr_in s_addr;    //服务器套接字地址
    struct sockaddr_in c_addr;    //客户端套接字地
    int dataBytes=0;//读取消息长度
    //消息
    char sendbuf[BUFSIZ];
    char recvbuf[BUFSIZ];

    /***************************创建套接字**************************/
    //socket函数,失败返回-1
    //int socket(int domain, int type, int protocol);
    //第一个参数表示使用的地址类型,一般都是ipv4即AF_INET
    //第二个参数表示套接字类型:tcp一般用SOCK_STREAM数据流传输
    //第三个参数设置为0
    s_fd = socket(AF_INET,SOCK_STREAM,0);
    if(s_fd < 0){
        perror("创建套接字失败");
        return 0;
    }
    printf("创建套接字成功\n");
    /****************************开启地址复用*******************/
    int optval = 1;//1允许地址重用0禁止
    int optlen = sizeof(optval);
    setsockopt(s_fd,SOL_SOCKET,SO_REUSEADDR,&optval,optlen);
    /********************初始化服务器套接字*****************/
    //htons和htonl将端口和地址转成网络字节序
    //定义服务器中地址中的域,AF_INET指IPv4
    s_addr.sin_family = AF_INET;
    //定义服务端的套接字端口
    s_addr.sin_port = htons(PORT);
    //定义套接字地址
    //ip可是本服务器的ip,用宏INADDR_ANY代替,代表0.0.0.0,表明所有地址
    s_addr.sin_addr.s_addr = htonl(INADDR_ANY);

    /****************绑定套接字设置的端口号和IP***********************/
    //对于bind,accept之类的函数,里面套接字参数都是需要强制转换成(struct sockaddr *)
    //bind三个参数:服务器端的套接字的文件描述符
    s_len = sizeof(s_addr);//设置发送消息长度
    if( bind(s_fd,(struct sockaddr*)&s_addr,s_len) < 0){
        perror("绑定套接字失败");
        return 0;
    }
    printf("绑定套接字成功\n");
    /*****************设置服务器上的socket为监听状态*******************/
    //监听,最大连接数10
    if(listen(s_fd,10) < 0){
        perror("监听失败");
        return 0;
    }
    printf("监听成功端口: %d\n", PORT);

    while(1){

        printf("等待连接...\n");
        //fflush(stdin)刷新标准输入缓冲区,fflush(stdout)刷新标准输出缓冲区
        fflush(stdout);
        //设置接收消息长度
        c_len = sizeof(c_addr);
        /********************接收客户端连接请求*************************/
        //调用accept函数后,会进入阻塞状态
        //accept返回一个套接字的文件描述符,这样服务器端便有两个套接字的文件描述符,
        //s_fd和c_fd
        //s_fd仍然继续在监听状态,c_fd则负责接收和发送数据
        //c_addr是一个传出参数,accept返回时,传出客户端的地址和端口号
        //c_len是一个传入-传出参数,传入的是调用者提供的缓冲区的c_addr的长度,以避免缓冲区溢出。
        //传出的是客户端地址结构体的实际长度。
        //出错返回-1
        c_fd = accept(s_fd,(struct sockaddr*)&c_addr,(socklen_t*)&c_len);
        if(c_fd < 0){
            perror("accept失败");
            continue;
        }

        printf("新连接:\n");
        //inet_ntoa   ip地址转换函数,将网络字节序IP转换为点分十进制IP
        //表达式:char *inet_ntoa (struct in_addr);
        printf("\tIP is %s\n", inet_ntoa(c_addr.sin_addr));
        printf("\tPort is %d\n", htons(c_addr.sin_port));
        printf("等待消息...\n");
        while(1){
            dataBytes=0;
            /*************************接收数据***********************/
            printf("--------------------读取:");
            fflush(stdout);
            dataBytes = recv(c_fd,recvbuf,BUFSIZ,0);
            if( dataBytes < 0){
                perror("读取失败");
                continue;
            }else if(dataBytes == 0){
                printf("无消息\n");
            }else
                printf("%s\n",recvbuf);

            //判断退出,quit,断开连接,关闭客户端
            if(strncmp(recvbuf,"quit",4) == 0){
                sprintf(recvbuf,"%s","EOF");
                send(c_fd,recvbuf,sizeof(recvbuf)+1,0);
                //关闭连接
                close(s_fd);
                printf("关闭服务器\n");
                printf("程序结束\n");
                return 0;
            }
            //EOF,断开连接
            if(strncmp(recvbuf, "EOF",3) == 0){
                break;
            }

            /************************发送数据*************************/
            printf("--------------------发送:");
            scanf("%s",sendbuf);
            send(c_fd,sendbuf,strlen(sendbuf)+1,0);

            //判断退出,quit,断开连接,关闭客户端
            if(strncmp(sendbuf,"quit",4) == 0){
                sprintf(sendbuf,"%s","EOF");
                send(c_fd,sendbuf,sizeof(sendbuf)+1,0);
                //关闭连接
                close(s_fd);
                printf("程序结束\n");
                return 0;
            }
            //EOF,断开连接
            if(strncmp(sendbuf,"EOF",3) == 0){
                break;
            }
        }//while 收发消息
        printf("断开连接\n\n");
    }//while accept

    //关闭连接
    close(s_fd);
    printf("程序结束\n");
    return 0;
}

client.c 

//============================================================================
// Name        : liveSource.cpp
// Author      : ffy
// Version     :
// Copyright   : Your copyright notice
// Description : Hello World in C++, Ansi-style
//============================================================================
#include <sys/types.h>
#include <sys/socket.h> //包含套接字的函数库
#include <netinet/in.h> //包含AF_INET的相关结构
#include <arpa/inet.h>  //包含AF_INET的操作函数
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

/*
 连接到服务器后,会不停循环,等待输入,
输入EOF后,断开连接,输入quit,断开连接并关闭服务器
读取EOF后,断开连接,读取quit,断开连接并关闭服务器
 */
#define PORT 3333
int main(void)
{
    printf("程序开始\n");

    //客户端套接字标识符,只需要一个套接字文件描述符,用于和客户端通信
    int sockfd;
    //描述客户端的socket
    struct sockaddr_in addr;
    //消息长度
    int dataBytes = 0;
    //消息
    char sendbuf[BUFSIZ];
    char recvbuf[BUFSIZ];

    //创建套接字
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0){
        perror("创建套接字失败");
        return 0;
    }
    printf("创建套接字成功\n");
    /********************初始化客户端套接字*****************/
    //htons和htonl将端口和地址转成网络字节序
    //定义客户端中地址中的域,AF_INET指IPv4
    addr.sin_family = AF_INET;
    //定义服务端的套接字端口
    addr.sin_port = htons(PORT);
    //指定服务器端的ip,本地测试:127.0.0.1
    //inet_addr()函数,将点分十进制IP转换成网络字节序IP
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    /**********************连接服务器**********************/
    if(connect(sockfd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
    {
        perror("连接服务器失败");
        return 0;
    }
    printf("连接到服务器...\n");
    while(1){
        /*******************发送消息************************/
        printf("--------------------发送:");
        scanf("%s", sendbuf);
        send(sockfd,sendbuf,strlen(sendbuf)+1,0);

        if(strcmp(sendbuf, "quit") == 0){
            //关闭服务器
            //关闭客户端
            break;
        }
        if(strcmp(sendbuf, "EOF") == 0){
            //关闭客户端
            break;
        }

        //读取消息
        printf("--------------------读取:");
        fflush(stdout);
        dataBytes = recv(sockfd, recvbuf, 200, 0);
        printf("%s\n", recvbuf);
        if(strncmp(recvbuf, "quit",4) == 0){
            //关闭服务器
            //关闭客户端
            break;
        }
        if(strncmp(recvbuf, "EOF",3) == 0){
            //关闭客户端
           break;
        }
    }

    //关闭连接
    close(sockfd);
    printf("程序结束\n");
    return 0;
}

 

Guess you like

Origin blog.csdn.net/qq_37437983/article/details/108256146