|
1、要进行点对点聊天，首先需要既能发送，又能接收，如果发送接收放在一个进程有可能导致数据包丢失，因此需要新开一个进程，一个进程在套接字上接收，一个进程在套接字上发送。
2、父进程和子进程是可以共享文件描述符的。
3、fork调用的一个奇妙之处就是它仅仅被调用一次，却能够返回两次，它可能有三种不同的返回值：
1）在父进程中，fork返回新创建子进程的进程ID；
2）在子进程中，fork返回0；
3）如果出现错误，fork返回一个负值；
4、调用exit()退出时可以关闭文件描述符。
5、进程间通信机制：信号
客户端程序：
#include<unistd.h>//read/write
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>//信号，进程间通信
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#define ERR_EXIT(m) \
do\
{\
perror(m);\
exit(EXIT_FAILURE);\
}while (0)
void handler(int sig)
{
printf("recv a sig= %d\n",sig);
exit(EXIT_SUCCESS);
}
int main(void)
{
int sock;//被动套接字
if( (sock = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)//创建套接字小于0表示失败
ERR_EXIT("socket");
struct sockaddr_in servaddr;
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(5188);
servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");//指定服务器端地址
if(connect(sock,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("connect");
pid_t pid;
pid = fork();//父进程发送数据。子进程接收数据
if(pid == -1)
{
ERR_EXIT("fork");
}
else if(pid == 0)
{
char recvbuf[1024] = {0};
while(1)
{
memset(recvbuf,0,sizeof(recvbuf));
int ret = read(sock,recvbuf,sizeof(recvbuf));
if(ret == -1)
ERR_EXIT("read");
else if(ret == 0)
{
printf("peer close\n");
break;//服务器端关闭，则退出子进程。但是父进程没有退出
}
fputs(recvbuf,stdout);
}
close(sock);//关闭端口
kill(getppid(),SIGUSR1);//向父进程发自定义信号
}
else//父进程发送
{
signal(SIGUSR1,handler);//父进程收到自定义信号，执行绑定函数
char sendbuf[1024] = {0};
while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
{
write(sock,sendbuf,strlen(sendbuf));//发送
memset(sendbuf,0,sizeof(sendbuf));
}
close(sock);
}
return 0;
}
服务端：
#include<unistd.h>//read/write
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>//信号，进程间通信
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#define ERR_EXIT(m) \
do\
{\
perror(m);\
exit(EXIT_FAILURE);\
}while (0)
void handler(int sig)
{
printf("recv a sig= %d\n",sig);
exit(EXIT_SUCCESS);
}
int main(void)
{
int listenfd;//被动套接字
if( (listenfd = socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)//创建套接字小于0表示失败
/* if( (listenfd = socket(PF_INET,SOCK_STREAM,0))<0);*///让内核自己选定协议
ERR_EXIT("socket");
struct sockaddr_in servaddr;
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(5188);
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);//表示本机的任意地址
/*servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");*/
/*inet_aton("127.0.0.1",&servaddr.sin_addr);*/
int on = 1;//开启地址重复利用
if(setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on))<0)
ERR_EXIT("setsockopt");
if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("bind");
if(listen(listenfd,SOMAXCONN)<0)//监听后变为被动套接字
ERR_EXIT("listen");
struct sockaddr_in peeraddr;
socklen_t peerlen = sizeof(peeraddr);
int conn;//主动套接字
if((conn = accept(listenfd,(struct sockaddr*)&peeraddr,&peerlen))<0)
ERR_EXIT("accept");
printf("ip=%s,port=%d\n",inet_ntoa(peeraddr.sin_addr),ntohs(peeraddr.sin_port));
pid_t pid;//父进程用来获取数据，子进程用来发送数据
pid = fork();
if(pid == -1)
{
ERR_EXIT("fork");
}
else if(pid == 0)
{
signal(SIGUSR1,handler);
char sendbuf[1024] = {0};
while(fgets(sendbuf,sizeof(sendbuf),stdin)!=NULL)
{
write(conn,sendbuf,strlen(sendbuf));
memset(sendbuf,0,sizeof(sendbuf));
}
printf("child close");
exit(EXIT_SUCCESS);
}
else
{
char recvbuf[1024];
while(1)
{
memset(recvbuf,0,sizeof(recvbuf));
int ret = read(conn,recvbuf,sizeof(recvbuf));
if(ret == -1)
ERR_EXIT("read");
else if(ret == 0)
{
printf("peer close\n");
break;
}
else
fputs(recvbuf,stdout);//打印到标准输出
}
printf("parent close");//父进程退出的时候要通知子进程也退出
kill(pid,SIGUSR1);
exit(EXIT_SUCCESS);
}
return 0;
}
| 
转播
