Linux网络编程示例

Linux网络编程示例

服务端代码 server.c

#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <errno.h>
#include <signal.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>

int fd;
int client_fd;
char buff[512];
void sig_handler(int signo)
{
    if (signo == SIGINT) {
        printf("server ctrl+c,stop server\n");
        close(fd);
        close(client_fd);
        exit(0);//结束整个进程
    }
}

int main(int argc,char* argv[])
{
    if (argc < 2) {
        perror("usage port");
        return 0;
    }

    //接收信号，ctrl+z结束程序
    if (signal(SIGINT,sig_handler) == SIG_ERR) {
        perror("server signal error");
        return 0;
    }

    int err = 0;
    //创建socket fd
    if ((fd = socket(AF_INET,SOCK_STREAM,0)) == -1) {
        perror("server socket error");
        return 0;
    }

    //绑定ip和端口号
    struct sockaddr_in serveraddr;
	memset(&serveraddr,0,sizeof(serveraddr));
	serveraddr.sin_family = AF_INET;
	serveraddr.sin_port = htons(atoi(argv[1]));//port
	serveraddr.sin_addr.s_addr = INADDR_ANY;
    if ((err = bind(fd,(struct sockaddr*)&serveraddr,sizeof(serveraddr))) == -1) {
        perror("server bind error");
        close(fd);
        return 0;
    }

    //listen 监听
    if ((err = listen(fd,1)) == -1) {
        perror("server listen error");
        close(fd);
        return 0;
    }

    //accept 等待连接
    struct sockaddr_in clientaddr;
    memset(&clientaddr,0,sizeof(clientaddr));
    socklen_t len = sizeof(clientaddr);
    if ((client_fd = accept(fd,(struct sockaddr*)&clientaddr,&len)) == -1) {
        perror("server accept error");
        close(fd);
        return 0;
    }

    //开始读写
    while(1) {
        //读取客户端发送来的数据,并回显到终端
        memset(buff,0,sizeof(buff));
        if (read(client_fd,buff,sizeof(buff)) < 0) {
            perror("server read error");
            continue;
        }
        write(STDOUT_FILENO,buff,sizeof(buff));

        //将读取到的数据在发送回客户端
        if (write(client_fd,buff,sizeof(buff)) < 0) {
            perror("server write data to client error");
        }
    }
    
    return 0;
}

客户端代码 client.c

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <errno.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>

int fd;
char buff[512];
static char serversay[] = "                              server say:";
static char clientsay[] = "client say:";

int main(int argc,char* argv[])
{
    if (argc < 3) {
		perror("usage error ip & port");
		return 0;
	}

	if ((fd = socket(AF_INET,SOCK_STREAM,0)) == -1) {
		perror("socket error");
		return 0;
	}

	//填充服务器地址，进行连接请求
	struct sockaddr_in seraddr;
	seraddr.sin_family = AF_INET;
	seraddr.sin_port = htons(atoi(argv[2]));
	inet_pton(AF_INET,argv[1],&seraddr.sin_addr.s_addr);
	if (connect(fd,(struct sockaddr*)&seraddr,sizeof(seraddr)) == -1) {
		perror("connect error");
		close(fd);
		return 0;
	}

	//连接成功，与服务端通信，进行读写
	while (1) {
		write(STDOUT_FILENO,clientsay,sizeof(clientsay));
		//从终端读取数据
		memset(buff,0,sizeof(buff));
		if (read(STDIN_FILENO,buff,sizeof(buff)) > 0) {
			//读取成功，将数据发往服务端
			if (write(fd,buff,sizeof(buff)) > 0) {
				memset(buff,0,sizeof(buff));
				if (read(fd,buff,sizeof(buff)) > 0) {
					write(STDOUT_FILENO,serversay,sizeof(serversay));
					write(STDOUT_FILENO,buff,sizeof(buff));
				}
			}
		}
	}

    return 0;
}

运行：
gcc server.c -o server
gcc client.c -o client