TCP客户端和服务器

#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include <time.h>
#include <semaphore.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/wait.h>
#include <signal.h>
#include <fcntl.h>
#include <sys/stat.h>

void *recv_thread(void *arg); // 接收数据的线程
void *send_thread(void *arg); // 发送数据的线程

int main(int argc, char const *argv[])
{
    // 判断参数个数是否正确
    if (argc != 3)
    {
        printf("Usage: %s <port> <ip>\n", argv[0]);
        exit(1);
    }
    printf("port: %s\n", argv[1]);

    // 创建发送数据的线程
    pthread_t send_tid;
    pthread_create(&send_tid, NULL, send_thread, argv);

    printf("ip: %s\n", argv[2]);
    // 创建接收数据的线程
    pthread_t recv_tid;
    pthread_create(&recv_tid, NULL, recv_thread, argv);
    printf("fdfdfdfd\n");

    // 等待发送线程结束
    pthread_join(send_tid, NULL);
    // 等待发送线程结束
    pthread_join(recv_tid, NULL);

    // 关闭线程
    pthread_exit(NULL);

    return 0;
}

// 发送数据的线程
void *send_thread(void *arg)
{
    char **argv = (char **)arg;

    int udp_socket_pid = socket(AF_INET, SOCK_STREAM, 0);
    if (udp_socket_pid < 0)
    {
        perror("socket creation failed");
        pthread_exit(NULL);
    }

    struct sockaddr_in send_addr;
    send_addr.sin_family = AF_INET;            // 协议族，是固定的
    send_addr.sin_port = htons(atoi(argv[1])); // 服务器端口，必须转换为网络字节序
    send_addr.sin_addr.s_addr = inet_addr(argv[2]);

    // 发送请求connect
    int connect_result = connect(udp_socket_pid, (struct sockaddr *)&send_addr, sizeof(send_addr));
    if (connect_result < 0)
    {
        perror("connect failed");
        pthread_exit(NULL);
    }

    int send_len;

    // 发送数据send
    char send_buf[512];

    while (1)
    {

        scanf("%s", send_buf); // 输入要发送的数据

        send_len = send(udp_socket_pid, send_buf, strlen(send_buf), 0);
        if (send_len < 0)
        {
            perror("send failed");
            pthread_exit(NULL);
        }
        else
        {
            printf("send: %s\n", send_buf);
        }
        // 数组清零
        memset(send_buf, 0, sizeof(send_buf));

        sleep(1);
    }

    // 关闭socket
    close(udp_socket_pid);

    return 0;
}

// 接收数据的线程
void *recv_thread(void *arg)
{
    char **argv = (char **)arg;

    // 1.创建TCP套接字
    int tcp_socket = socket(AF_INET, SOCK_STREAM, 0);
    if (tcp_socket == -1)
    {
        fprintf(stderr, "tcp socket error,errno:%d,%s\n", errno, strerror(errno));
        exit(1);
    }

    // 2.绑定自身的IP地址和端口
    struct sockaddr_in host_addr;

    host_addr.sin_family = AF_INET;                // 协议族，是固定的
    host_addr.sin_port = htons(atoi(argv[1]));     // 目标端口，必须转换为网络字节序
    host_addr.sin_addr.s_addr = htonl(INADDR_ANY); // 目标地址  INADDR_ANY 这个宏是一个整数，所以需要使用htonl转换为网络字节序

    bind(tcp_socket, (struct sockaddr *)&host_addr, sizeof(host_addr));

    // 3.设置监听  队列最大容量是5
    listen(tcp_socket, 5);

    // 4.等待接受客户端的连接请求
    struct sockaddr_in client;
    socklen_t client_len = sizeof(client);

    int connect_fd = accept(tcp_socket, (struct sockaddr *)&client, &client_len); // 会阻塞
    char buf[128] = {0};

    // 5.说明双方建立连接，此时可以接收数据
    while (1)
    {

        read(connect_fd, buf, sizeof(buf));
        printf("recv from [%s],data is = %s\n", inet_ntoa(client.sin_addr), buf);
        bzero(buf, sizeof(buf));
    }

    return 0;
}