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;
char buff[512];
void sig_handler(int signo)
{
if (signo == SIGINT) {
printf("server ctrl+c,stop server\n");
close(fd);
exit(0);//结束整个进程
}
if(signo == SIGCHLD)
{
printf("child process deaded...\n");
wait(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,10)) == -1) {
perror("server listen error");
close(fd);
return 0;
}
//accept 等待连接
struct sockaddr_in clientaddr;
memset(&clientaddr,0,sizeof(clientaddr));
socklen_t len = sizeof(clientaddr);
while (1) {
int client_fd;
if ((client_fd = accept(fd,(struct sockaddr*)&clientaddr,&len)) == -1) {
perror("server accept error");
continue;
}
//启动子进程调用IO函数和客户端进行通信
pid_t pid = fork();
if (pid < 0) {
perror("fork error");
continue;
} else if (pid == 0) {
//child
//读取客户端发送来的数据,并把它写会客户端
while (1) {
memset(buff,0,sizeof(buff));
// printf("start read and write...\n");
size_t size;
if ((size = read(client_fd,buff,sizeof(buff))) <= 0) {
perror("server read error");
continue;
} else {
printf("> %s\n",buff);
//将读取到的数据在发送回客户端
if (write(client_fd,buff,sizeof(buff)) < 0) {
perror("server write data to client error");
}
}
}
} else {
//parent
//关闭accept建立连接的套接字
//fork()之后会在子进程复制一份accept建立连接的套接子,独立于父进程,此处关闭的是属于父进程的acdept套接字
close(client_fd);
}
}
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;
memset(&seraddr,0,sizeof(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;
}
执行: