1.select服务器客户端代码
//服务器
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define MSG_ERR(msg) do{\
fprintf(stderr,"line:%d",__LINE__);\
perror(msg);\
}while(0)
#define PORT 8888 //1024~49151
#define IP "192.168.52.184" //ifconfig查看自己的IP
int updatemax(int maxfd,fd_set readfds);
int main(int argc, const char *argv[])
{
//创建流式(TCP)套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
MSG_ERR("socket");
return -1;
}
//允许端口快速重用
int reuse=1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
MSG_ERR("setsockopt");
return -1;
}
printf("允许端口快速重用成功\n");
//填充服务器的IP和端口到地址信息结构体上
//地址信息结构体根据地址族指定 man 7 IP
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必须填AF_INET
sin.sin_port = htons(PORT);//网络字节序的端口号1024~49151
sin.sin_addr.s_addr = inet_addr(IP);//网络字节序的IP地址
//将服务器的IP和端口绑定到套接字上;必须绑定
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
MSG_ERR("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd,10)<0)
{
MSG_ERR("listen");
return -1;
}
printf("listen success\n");
//创建读集合
fd_set readfds,tempfds;
//清空集合
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//将需要监测读事件的文件描述符添加到集合
FD_SET(0,&readfds);//将0号文件描述符添加到集合中
FD_SET(sfd,&readfds);//将sfd文件描述符添加到集合中
int maxfd=sfd;//定义最大的文件描述符
int res=0;
char buf[128]="";
ssize_t ret=0;
struct sockaddr_in cin; //存储客户端的信息结构体
socklen_t addrlen=sizeof(cin); //地址信息结构体的大小
struct sockaddr_in save_cin[1024-4];//存储连接成功的所有客户端信息的结构体数组
while(1)
{
//调用select
tempfds=readfds;
res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(res<0)
{
MSG_ERR("select");
return -1;
}
if(res==0)
{
printf("超时\n");
break;
}
for(int i=0;i<=maxfd;i++)
{
//若i对应的文件描述符不集合中则继续遍历下一个文件描述符
if(FD_ISSET(i,&tempfds)==0)
{
continue;
}
//运行到当前位置,则说明i所代表的文件描述符在集合中
//只需要判断i所代表的文件描述符是什么,就走什么步骤即可
if(i==0)
{
//触发键盘输入事件
printf("键盘触发\n");
int sndfd;
bzero(buf,sizeof(buf));
ret=scanf("%d %s",&sndfd,buf);
while(getchar()!=10);
if(ret!=2)
{
fprintf(stderr,"请输入正确格式:整型 字符串\n");
continue;
}
//能运行到当前位置,则代表格式输入正确
//可以成功到那一个sndfd以及一个字符串
if(FD_ISSET(sndfd,&readfds)&&sndfd>sfd)
{
if(send(sndfd,buf,sizeof(buf),0)<0)
{
MSG_ERR("send");
continue;
}
else
{
printf("发送成功\n");
}
}
else
{
fprintf(stderr,"sndfd=%d 文件描述符输入错误\n",sndfd);
}
}
else if(sfd==i)
{
int newfd=accept(sfd,(struct sockaddr *)&cin,&addrlen);
if(newfd<0)
{
MSG_ERR("accept");
return -1;
}
printf("[%s:%d] newfd=%d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
save_cin[newfd-4]=cin;
//将newfd添加到集合中让内核监测是否有读事件
FD_SET(newfd,&readfds);
maxfd=maxfd>newfd ? maxfd:newfd;//更新maxfd
}
else//除了0和sfd,剩下的都是新加入的newfd
{
bzero(buf,sizeof(buf));
printf("触发客户端交互\n");
//循环接收数据,阻塞方式接收
ret=recv(i,buf,sizeof(buf),0);
if(ret<0)
{
MSG_ERR("recv");
return -1;
}
else if(ret==0)
{
printf("[%s:%d] newfd=%d %s\n",\
inet_ntoa(save_cin[i-4].sin_addr),ntohs(save_cin[i-4].sin_port),i,buf);
//将退出客户端对应的文件描述符从集合中剔除
FD_CLR(i,&readfds);
close(i);
//更新maxfd
maxfd=updatemax(maxfd,readfds);
}
else
{
printf("[%s:%d] newfd=%d %s\n",\
inet_ntoa(save_cin[i-4].sin_addr),ntohs(save_cin[i-4].sin_port),i,buf);
}
}
}
}
close(sfd);
return 0;
}
//更新maxfd
int updatemax(int maxfd,fd_set readfds)
{
for(int j=maxfd;j>0;j--)
{
if(FD_ISSET(j,&readfds))
{
return j;
}
}
return 0;
}
//客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <poll.h>
#define MSG_ERR(msg) do{\
fprintf(stderr,"line:%d",__LINE__);\
perror(msg);\
}while(0)
#define PORT 8888 //1024~49151
#define IP "192.168.52.184" //ifconfig查看自己的IP
int updatemax(int maxfd,fd_set readfds);
int main(int argc, const char *argv[])
{
//创建流式(TCP)套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
MSG_ERR("socket");
return -1;
}
//填充要连接的服务器的地址信息结构体
//因为只有知道了服务器的IP和端口才能连接服务器
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必须填AF_INET
sin.sin_port = htons(PORT);//网络字节序的端口号1024~49151
sin.sin_addr.s_addr = inet_addr(IP);//网络字节序的IP地址
//连接服务器
if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
MSG_ERR("connect");
return -1;
}
printf("connect success\n");
//创建select集合
fd_set readfds,tempfds;
//清空集合
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//将需要监测读事件的文件描述符添加到集合
FD_SET(0,&readfds);//将0号文件描述符添加到集合中
FD_SET(sfd,&readfds);//将sfd文件描述符添加到集合中
int maxfd=sfd;
int res=0;
char buf[128]="";
ssize_t ret=0;
while(1)
{
//调用select
tempfds=readfds;
res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(res<0)
{
MSG_ERR("select");
return -1;
}
if(res==0)
{
printf("超时\n");
break;
}
for(int i=0;i<=maxfd;i++)
{
//若i对应的文件描述符不在集合中则继续遍历下一个文件描述符
if(FD_ISSET(i,&tempfds)==0)
{
continue;
}
//运行到当前位置,则说明i所代表的文件描述符在集合中
//只需要判断i所代表的文件描述符是什么,就走什么步骤即可
if(i==0)
{
//触发键盘输入事件
printf("键盘触发\n");
// int sndfd;
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
if(send(sfd,buf,sizeof(buf),0)<0)
{
MSG_ERR("send");
continue;
}
printf("发送成功\n");
}
else
{
bzero(buf,sizeof(buf));
printf("触发客户端交互\n");
//循环接收数据,阻塞方式接收
ret=recv(i,buf,sizeof(buf),0);
if(ret<0)
{
MSG_ERR("recv");
return -1;
}
else if(ret==0)
{
printf("接收完毕\n");
//将退出客户端对应的文件描述符从集合中剔除
FD_CLR(i,&readfds);
close(i);
//更新maxfd
maxfd=updatemax(maxfd,readfds);
}
else
{
printf("%s\n",buf);
}
}
}
}
close(sfd);
return 0;
}
//更新maxfd
int updatemax(int maxfd,fd_set readfds)
{
for(int j=maxfd;j>0;j--)
{
if(FD_ISSET(j,&readfds))
{
return j;
}
}
return 0;
}
2.poll客户端代码
include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <poll.h>
#define MSG_ERR(msg) do{\
fprintf(stderr,"line:%d",__LINE__);\
perror(msg);\
}while(0)
#define PORT 8888 //1024~49151
#define IP "192.168.52.184" //ifconfig查看自己的IP
int main(int argc, const char *argv[])
{
//创建流式(TCP)套接字
int sfd = socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
MSG_ERR("socket");
return -1;
}
//填充要连接的服务器的地址信息结构体
//因为只有知道了服务器的IP和端口才能连接服务器
struct sockaddr_in sin;
sin.sin_family = AF_INET;//必须填AF_INET
sin.sin_port = htons(PORT);//网络字节序的端口号1024~49151
sin.sin_addr.s_addr = inet_addr(IP);//网络字节序的IP地址
//连接服务器
if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
MSG_ERR("connect");
return -1;
}
printf("connect success\n");
//创建poll集合
struct pollfd fds[2];
//将要监测的文件描述符和要监测的事件填入数组中
fds[0].fd=0; //监测0号
fds[0].events=POLLIN; //监测读事件
fds[1].fd=sfd; //监测sfd文件描述符
fds[1].events=POLLIN; //监测读事件
int p_res=0;
char buf[128]="";
ssize_t res=0;
while(1)
{
//阻塞等待文件描述符产生事件
p_res=poll(fds,2,-1);
if(p_res<0)
{
MSG_ERR("poll");
return -1;
}
else if(p_res==0)
{
printf("超时了\n");
break;
}
//能运行到当前位置说明集合中有文件描述符准备就绪
//只需要判断实际产生的事件中是否有需要的事件即可
//POLLIN事件在revents中只占了一个bit
//所以最终判断是否产生POLLIN事件,需要将POLLIN那1bit提取出来
if((fds[0].revents & POLLIN)!=0)
{
bzero(buf,sizeof(buf));
//从终端获取数据,进行发送
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
//循环发送数据,阻塞方式接收
if(send(sfd,buf,sizeof(buf),0)<0)
{
MSG_ERR("send");
return -1;
}
printf("send success\n");
}
if((fds[1].revents &POLLIN))
{
//接收服务器的信息
bzero(buf,sizeof(buf));
res=recv(sfd,buf,sizeof(buf),0);
if(res<0)
{
MSG_ERR("recv");
return -1;
}
else if(res==0)
{
printf("服务器退出\n");
break;
}
printf("%s\n",buf);
}
}
close(sfd);
return 0;
}
3.TCP域套接字服务器客户端代码
//服务器
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/un.h>
#define MSG_ERR(msg) do{\
fprintf(stderr,"line:%d",__LINE__);\
perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{
//创建流式域(TCP)套接字
int sfd = socket(AF_UNIX,SOCK_STREAM,0);
if(sfd<0)
{
MSG_ERR("socket");
return -1;
}
//判断文件是否存在
if(access("./myunix",F_OK)==0)
{
if(unlink("./myunix")<0)
{
MSG_ERR("unlink");
return -1;
}
}
//填充地址信息结构体上
//地址信息结构体根据地址族指定 man 7 unix
struct sockaddr_un sin;
sin.sun_family = AF_UNIX;//必须填AF_UNIX
strcpy(sin.sun_path,"./myunix");//套接字路径名
//将地址信息结构体绑定到套接字上;必须绑定
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
MSG_ERR("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为被动监听状态
if(listen(sfd,10)<0)
{
MSG_ERR("listen");
return -1;
}
printf("listen success\n");
struct sockaddr_un cin; //存储客户端的信息结构体
socklen_t addrlen=sizeof(cin); //地址信息结构体的大小
//获取完成连接的客户信息,生成一个新的文件描述符
int newfd=accept(sfd,(struct sockaddr *)&cin,&addrlen);
if(newfd<0)
{
MSG_ERR("accept");
return -1;
}
printf(" newfd=%d\n",newfd);
char buf[128]="";
ssize_t ret=0;
while(1)
{
bzero(buf,sizeof(buf));
//循环接收数据,阻塞方式接收
ret=recv(newfd,buf,sizeof(buf),0);
if(ret<0)
{
MSG_ERR("recv");
return -1;
}
else if(ret==0)
{
printf("newfd=%d 对应的客户端退出\n",newfd);
break;
}
printf("ret=%ld:buf=%s\n",ret,buf);
//发送数据,将对方发送过来的数据发送回去
strcat(buf,"*_*");
if(send(newfd,buf,sizeof(buf),0)<0)
{
MSG_ERR("send");
return -1;
}
printf("send success\n");
}
close(newfd);
close(sfd);
return 0;
}
//客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#define MSG_ERR(msg) do{\
fprintf(stderr,"line:%d",__LINE__);\
perror(msg);\
}while(0)
int main(int argc, const char *argv[])
{
//创建流式域(TCP)套接字
int sfd = socket(AF_UNIX,SOCK_STREAM,0);
if(sfd<0)
{
MSG_ERR("socket");
return -1;
}
//填充地址信息结构体
struct sockaddr_un sin;
sin.sun_family = AF_UNIX;//必须填AF_UNIX
strcpy(sin.sun_path,"./myunix");
//连接服务器
if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
MSG_ERR("connect");
return -1;
}
printf("connect success\n");
//发送信息给服务器
char buf[128]="";
ssize_t ret=0;
ssize_t res=0;
while(1)
{
//从终端获取数据,进行发送
printf("please input:");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
//循环发送数据,阻塞方式接收
ret=send(sfd,buf,sizeof(buf),0);
if(ret<0)
{
MSG_ERR("send");
return -1;
}
printf("send success\n");
//接收服务器的信息
bzero(buf,sizeof(buf));
ssize_t res=recv(sfd,buf,sizeof(buf),0);
if(res<0)
{
MSG_ERR("recv");
return -1;
}
else if(res==0)
{
printf("服务器退出\n");
break;
}
printf("%s\n",buf);
}
close(sfd);
return 0;
}