简单TCP服务器和客户端

简易tcp server和client  不支持客户端并发连接

server.c

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

#define PRINT_ERR(msg)  \
    do {                \
        printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
        perror(msg);    \
        exit(-1);        \
    } while (0)

int main(int argc, const char * argv[]) {

    if (argc != 3) {
        printf("Usage: %s <IP> <port>\n", argv[0]);
        exit(-1);
    }
    // 创建套接字
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        PRINT_ERR("socket error");
    }
    // 填充服务器网络信息结构体
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);

    socklen_t serveraddr_len = sizeof(serveraddr);
    // 将套接字和服务器网络信息结构体绑定
    if (bind(sockfd, (struct sockaddr*)&serveraddr, serveraddr_len) == -1) {
        PRINT_ERR("bind error");
    }
    // 将套接字设置成被动监听状态
    if (listen(sockfd, 5) == -1) {
        PRINT_ERR("listen error");
    }
    // 定义结构体保存客户端信息
    struct sockaddr_in client_addr;
    memset(&client_addr, 0, sizeof(client_addr));
    socklen_t clientaddr_len = sizeof(client_addr);

    int acceptfd = 0;
    char buf[128];
    int nbytes = 0;

    while (1) {
        if ((acceptfd = accept(sockfd, (struct sockaddr *)&client_addr, &clientaddr_len)) == -1) {
            PRINT_ERR("accept error");
        }
        printf("client [%s:%d] connect!!!..\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
        while (1) {
            memset(buf, 0 , sizeof(buf));
            if(-1 == (nbytes = recv(acceptfd, buf, sizeof(buf), 0))){
                PRINT_ERR("recv error");
            } else if(0 == nbytes){
                printf("client[%s:%d] disconnect!!!..\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
                break;
            }
            printf("receive data [%s]\n", buf);
            if (!strncmp(buf, "quit", 4)) {
                break;
            }
            strcat(buf, "---hi!!!");
            write(acceptfd, buf, sizeof(buf));
        }
        close(acceptfd);
    }
    return 0;
}

client.c

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

#define PRINT_ERR(msg)  \
    do {                \
        printf("%s:%s:%d\n", __FILE__, __func__, __LINE__); \
        perror(msg);    \
        exit(-1);        \
    } while (0)

int main(int argc, const char *argv[]) {
    if (argc != 3) {
        printf("Usage: %s <IP> <port>\n", argv[0]);
        exit(-1);
    }
    // 创建套接字
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        PRINT_ERR("sock error");
    }
    // 填充服务器信息结构体
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
    socklen_t serveraddr_len = sizeof(serveraddr);
    // 建立连接
    if (connect(sockfd, (struct sockaddr *)&serveraddr, serveraddr_len) == -1) {
        PRINT_ERR("connect error");
    }
    printf("connect success!!!\n");
    while (1) {
        char buf[128];
        fgets(buf, sizeof(buf), stdin);
        buf[strlen(buf) - 1] = '\0';
        write(sockfd, buf, sizeof(buf));
        if (!strncmp(buf, "quit", 4)) {
            break;
        }
        read(sockfd, buf, sizeof(buf));
        printf("reponse ----> [%s]\n", buf);
    }
    close(sockfd);
    return 0;
}

执行结果

server

ubuntu:ubuntu $ ./server 127.0.0.1 8888
client [127.0.0.1:54857] connect!!!..
receive data [nihao]
receive data [hello world]
receive data [quit]
^C
ubuntu:ubuntu $

client

ubuntu:ubuntu $ ./client 127.0.0.1 8888
connect success!!!
nihao
reponse ----> [nihao---hi!!!]
hello world
reponse ----> [hello world---hi!!!]
quit
ubuntu:ubuntu $