简单的socket编程

服务端tcp_server.c：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <linux/in.h>
#define PORT 8888  //端口地址
#define BACKLOG 2   //监听队列长度
int main(int argc, char* argv[])
{
    int ss,sc;
    struct sockaddr_in server_addr;
    struct sockaddr_in client_addr;
    int err;
    pid_t pid;

    ss = socket(AF_INET, SOCK_STREAM, 0);
    if (ss < 0)
    {
        printf("socket error\n");
        return -1;
    }

    memset(&server_addr,0,sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_port = htons(PORT);

    err = bind(ss, (struct sockaddr*)&server_addr, sizeof(server_addr));
    if (err < 0)
    {
        printf("bind error\n");
        return -1;
    }

    err = listen(ss, BACKLOG);
    if (err < 0)
    {
        printf("listen error\n");
        return -1;
    }

    while(1)
    {
        int addrlen = sizeof(struct sockaddr);
        sc = accept(ss, (struct sockaddr*)&client_addr,&addrlen);
        if (sc < 0)
          continue;
        pid = fork();
        if (pid == 0) //子进程
        {
            close(ss); //关闭服务器监听
            process_conn_server(sc);
        }
        else
        {
            close(sc);
        }
    }
}

客户端 tcp_client.c：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <linux/in.h>
#define PORT 8888
int main(int argc, char* argv[])
{
    int s; 
    struct sockaddr_in server_addr;
    int err;

    s = socket(AF_INET, SOCK_STREAM, 0);
    if (s < 0)
    {
        printf("socket error\n");
        return -1;
    }

    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY); //本地地址
    server_addr.sin_port = htons(PORT); //服务器端口
    //将用户输入的字符串类型的IP地址转为整形
    inet_pton(AF_INET,argv[1],&server_addr.sin_addr);

    connect(s, (struct sockaddr*)&server_addr, sizeof(struct sockaddr));
    process_conn_client(s); //客户端处理过程`这里写代码片`
    close(s);
}

tcp_process.c

#include <sys/types.h>

void process_conn_server(int s)
{
    ssize_t size = 0;
    char buffer[1024];
    while(1)
    {
        size = read(s,buffer,1024);
        if (size == 0)
            return;
        sprintf(buffer, "%d bytes altogether\n", size);
        write(s,buffer,strlen(buffer)+1);
    }
}

void process_conn_client(int s)
{
    ssize_t size = 0;
    char buffer[1024];
    while (1)
    {
        //从标准输入中读取数据数据放到缓冲区buffer中
        size = read(0,buffer,1024);
        if (size > 0)
        {
            write(s,buffer,size);
            size = read(s,buffer,1024);//从服务器读取数据
            write(1,buffer,size); //写到标准输出
        }
    }

}

先运行server端， ./server
然后客户端 ./client 127.0.0.1