tcp一对一双向通信2024-8-20

服务端：

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <unistd.h>
int new_fd;

void * work(void * argv)
{
    struct sockaddr_in *client = (struct sockaddr_in *)argv;
    socklen_t client_len = sizeof(*client);

    char buf[100];
    while(1)
    {
        bzero(buf,100);
        recv(new_fd,buf,100,0);
        if(strcmp(buf,"bye\n") == 0)
        {
            printf("[%s:%d]已退出\n",inet_ntoa(client->sin_addr),ntohs(client->sin_port));
            break;
        }
        printf("[%s:%d]%s",inet_ntoa(client->sin_addr),ntohs(client->sin_port),buf);
    }
}

int main(int argc, const char *argv[])
{
    //创建套接字
    int sock_fd = socket(AF_INET,SOCK_STREAM,0);
    if(sock_fd<0)
    {
        perror("creat socket failed");
        return -1;
    }

    //绑定
    struct sockaddr_in serve,client;
    serve.sin_family = AF_INET;
    serve.sin_port = htons(atoi(argv[2]));
    serve.sin_addr.s_addr = inet_addr(argv[1]);

    socklen_t client_len = sizeof(client);
    socklen_t serve_len = sizeof(serve);
    bind(sock_fd,(struct sockaddr *)&serve,serve_len);

    //监听
    listen(sock_fd,4);
    //等待连接
    new_fd = accept(sock_fd,(struct sockaddr *)&client,&client_len);
    printf("[%s:%d]连接成功\n",inet_ntoa(client.sin_addr),ntohs(client.sin_port));
    pthread_t pid;
    //创建线程（接受消息）
    if(pthread_create(&pid,NULL,work,(void *)&client) != 0)
    {
        perror("creat pthread failed");
        return -1;
    }

    char buf[100];
    while(1)
    {
        bzero(buf,100);
        fgets(buf,100,stdin);
        send(new_fd,buf,100,0);
        if(strcmp(buf,"bye\n") == 0)
            break;
    }

    return 0;
}
                                                                                                          

客户端：

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

int sock_fd;

void* work(void * argv)
{
    struct sockaddr_in * serve=(struct sockaddr_in*)argv;
    socklen_t serve_len = sizeof(*serve);

    char buf[100];
    while(1)
    {
        bzero(buf,100);
        recv(sock_fd,buf,100,0);
        if(strcmp(buf,"bye\n") == 0)
        {
            printf("[%s:%d]已退出\n",inet_ntoa(serve->sin_addr),ntohs(serve->sin_port));
            break;
        }
        printf("[%s:%d]%s\n",inet_ntoa(serve->sin_addr),ntohs(serve->sin_port),buf);
    }
}

int main(int argc, const char *argv[])
{
    //创建套接字
    sock_fd = socket(AF_INET,SOCK_STREAM,0);
    if(sock_fd<0)
    {
        perror("creat sock_fd failed");
        return -1;
    }
    //绑定
    struct sockaddr_in serve;
    serve.sin_family = AF_INET;
    serve.sin_port = htons(atoi(argv[2]));
    serve.sin_addr.s_addr = inet_addr(argv[1]);

    socklen_t serve_len = sizeof(serve);
    bind(sock_fd,(struct sockaddr *)&serve,serve_len);
    //发送连接
    //listen(sock_fd,4);
    connect(sock_fd,(struct sockaddr *)&serve,serve_len);

    pthread_t pid;
    if(pthread_create(&pid,NULL,work,(void *)&serve)!=0)
    {
        perror("creat pthread failed");
        return -1;
    }

    char buf[100];
    while(1)
    {
        bzero(buf,100);
        fgets(buf,100,stdin);
        //发送消息
        send(sock_fd,buf,100,0);
        if(strcmp(buf,"bye\n") == 0)
            break;
    }                                                                                                                        
    return 0;
}