利用多进程实现,一个进程负责发送,一个进程负责接收!毕竟文件描述符共享(实际上是两个进程空间中不相关的文件描述符指向同一个文件)
服务器端代码:
/*server03*/
#include /* See NOTES */
#include
#include
#include /* superset of previous */
#include
#include
#include
#include
#include
#include
#define MAX_CLIENT 10
#define MAX_READ 1024
void handle(int sig)//辅助杀死子进程或者父进程
{
printf("recv : %d\n",sig);
exit(0);
}
int main()
{
int serv_fd,con_fd;//服务器端至少要有两个套接字文件描述符--一个用来监听,一个/其余多个用来和客户端通信
struct sockaddr_in serv_addr;//IPV4套接字结构体--服务器
struct sockaddr_in clt_addr;//IPV4套接字结构体--客户端
char addr_dst[INET_ADDRSTRLEN] = {0};//存储客户端IP地址
char buf[MAX_READ]={0};//应用程序自己的缓冲区
int optvar;//地址复用使用的参数
pid_t pid;//子进程PID
socklen_t addr_len;
ssize_t ret ;
signal(SIGUSR1,handle);//注册新号和处理函数
serv_fd = socket(AF_INET,SOCK_STREAM,0);//建立套接字
if(-1 == serv_fd)
{
perror("socket");
exit(-1);
}
if(setsockopt(serv_fd, SOL_SOCKET,SO_REUSEADDR,&optvar,sizeof(optvar)) == -1 )//地址复用
{
perror("setsockopt");
exit(-1);
}
/*设置地址*/
bzero(&serv_addr,sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(8001);
serv_addr.sin_addr.s_addr = htons(INADDR_ANY);
if(bind(serv_fd,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) == -1)//绑定端口
{
perror("bind");
exit(-1);
}
/*一旦调用listen函数--套接字就会变成被动套接字--用来监听客户端,让客户端连接他
被动套接字--只能接受连接,不能主动发送连接
做了两个队列:
一个已经完成三次握手,建立连接的队列--客户端发connect请求被响应,已经成功完成连接
一个是未完成成三次握手的队列--正在握手
*/
if(listen(serv_fd,MAX_CLIENT)== -1)//开始监听
{
perror("listen");
exit(-1);
}
addr_len = sizeof(clt_addr);
printf("Accepting connections ...\n");
if((con_fd = accept(serv_fd,(struct sockaddr *)&clt_addr,&addr_len)) == -1)//由于是点对点--所以只能处理一个客户端的连接请求
{
perror("accept");
exit(-1);
}
printf("received from %s at PORT %d\n",inet_ntop(AF_INET,&clt_addr.sin_addr,addr_dst,sizeof(addr_dst)),ntohs(clt_addr.sin_port));
pid = fork();//创建子进程
if(pid > 0)//父进程负责读取客户端数据
{
while(1)
{
ret = read(con_fd,buf,sizeof(buf));//读取数据
if(ret == 0)
{
/*如果在读的过程中,对方已关闭,TCP/IP协议会返回一个0数据包*/
printf("client close\n");
break;
}
else if(ret < 0)
{
perror("read");
exit(-1);
}
fputs(buf,stdout);//打印内容
memset(buf,0,sizeof(buf));
}
close(serv_fd);//关闭套接字
close(con_fd);
kill(pid,SIGUSR1);//发送信号给子进程--通知他死期已到
}
else if(pid == 0)//子进程负责向客户端写入数据
{
printf("Please input alphabet:");
while(fgets(buf,sizeof(buf),stdin) != NULL)//从标准输入获取数据
{
write(con_fd,buf,strlen(buf));//给客户端发送数据
memset(buf,0,sizeof(buf));//缓冲区清理
printf("Please input alphabet:");
}
}
else if(pid == -1)//创建子进程错误
{
perror("fork");
close(serv_fd);
close(con_fd);
exit(-1);
}
return 0;
}
客户端代码
#include /* See NOTES */
#include
#include
#include /* superset of previous */
#include
#include
#include
#include
#include
#include
#define MAX_BUF 1024
void handle(int sig)
{
printf("recv : %d\n",sig);
exit(0);
}
int main()
{
int clt_fd;
struct sockaddr_in serv_addr;
char buf[MAX_BUF]={0};
char addr_dst[INET_ADDRSTRLEN] = {0};
ssize_t ret;
pid_t pid;
signal(SIGUSR1,handle);
clt_fd = socket(AF_INET,SOCK_STREAM,0);
if(-1 == clt_fd)
{
perror("socket");
exit(-1);
}
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(8001);
serv_addr.sin_addr.s_addr = inet_addr("192.168.1.110");
if(connect(clt_fd,(struct sockaddr*)&serv_addr,sizeof(serv_addr)) == -1)
{
perror("connect");
exit(-1);
}
printf("Connect successfully\t%s at PORT %d\n",inet_ntop(AF_INET,&serv_addr.sin_addr,addr_dst,sizeof(addr_dst)),ntohs(serv_addr.sin_port));
pid = fork();
if(pid > 0)//父进程负责给服务器端发送数据
{
printf("Please input the alphabet:");
while(fgets(buf,sizeof(buf),stdin) != NULL)//从标准输入获取数据
{
write(clt_fd,buf,strlen(buf));//发送数据
memset(buf,0,sizeof(buf));//缓冲器清理
printf("Please input the alphabet:");
}
}
else if(pid == 0)//子进程负责从服务器端获取数据
{
while(1)
{
ret = read(clt_fd,buf,sizeof(buf));//读数据
if(ret == 0)
{
/*如果在读的过程中,对方已关闭,TCP/IP协议会返回一个0数据包*/
printf("server close\n");
break;
}
else if(ret < 0)
{
perror("read");
exit(-1);
}
fputs(buf,stdout);//打印内容
memset(buf,0,sizeof(buf));
}
close(clt_fd);
kill(getppid(),SIGUSR1);//发送信号给父进程--通知他死期已到
}
else if(pid == -1)//创建子进程错误
{
perror("fork");
close(clt_fd);
exit(-1);
}
return 0;
}