Linux环境下的基于C++的Socket 通信

server:

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

int sockfd;

void fa(int signo)
{
    printf("服务器正在关闭\n");
    sleep(1);
    close(sockfd);
    exit(0);
}

int main()
{
    printf("ctrl+c退出服务器\n");
    signal(SIGINT,fa);
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if (sockfd == -1)
    {
        perror("socket");
        exit(-1);
    }

    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(2222);
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");

    int reuseaddr = 1;//解决地址已被占用问题
    //端口复用
    setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR,
            &reuseaddr,sizeof(reuseaddr));

    int res = bind(sockfd,(struct sockaddr*)&addr,
            sizeof(addr));
    if (res == -1)
    {
        perror("bind");
        exit(-1);
    }
    printf("bind ok\n");
    listen(sockfd,100);//监听
    while (1)
    {
        struct sockaddr_in client;
        socklen_t len = sizeof(client);
        int fd = accept(sockfd,(struct sockaddr*)&client,
                &len);//阻塞函数
        char *from = inet_ntoa(client.sin_addr);
        printf("%s连接成功\n",from);
        pid_t pid = fork();
        if (pid == 0)
        {
        char buf[100] = {};
            while (1)
            {
                res = read(fd,buf,sizeof(buf));
                printf("接受了%d字节，内容:%s\n",res,buf);
                if (res <= 0){//包括0和-1
                    break;
                }
                if (strcmp(buf,"bye") == 0)
                {
                    break;
                }
                write(fd,buf,strlen(buf));
                memset(buf,0,sizeof(buf));
            }
            close(fd);
            exit(0);
        }
    close(fd);
    }
}

client:

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

int main()
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1)
    {
        perror("socket");
        exit(-1);
    }

    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(2222);//连接端口
    addr.sin_addr.s_addr = inet_addr("127.0.0.1");//都是服务器的，改成连接IP
    int res = connect(sockfd,(struct sockaddr*)&addr,sizeof(addr));
    if (res == -1) 
    {
        perror("connect"),exit(-1);
    }
    printf("连接成功\n");

    char buf[100] = {};
    while (1) 
    {
        memset(buf,0,sizeof(buf));//buf清0
        printf("请输入要说的话\n");
        scanf("%s",buf);
        write(sockfd,buf,strlen(buf));
        if (strcmp(buf,"bye") == 0) //退出的合适位置
        {                          
            break;
        }
        memset(buf,0,sizeof(buf));//buf清0
        res = read(sockfd,buf,sizeof(buf));
        printf("读到了%d字节，内容：%s\n",res,buf);
    }

    close(sockfd);
    return 0;
}