#include
#define ERR_(err) {perror(err);\
fprintf(stderr,"__%d__\n",__LINE__);}
#define PORT 8888
#define IP "192.168.122.50"
int main(int argc, const char *argv[])
{
//IO多路复用
//创建套接字
int sfd=socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_("socket");
return -1;
}
puts("创建成功");
//允许端口快速重用
int reuse=1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_("setsockopt");
return -1;
}
puts("允许端口快速被重用成功");
//填充地址结构体
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
struct sockaddr_in cin;
socklen_t len = sizeof(cin);
//绑定套接字
if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
{
ERR_("bind");
return -1;
}
//监听
if(listen(sfd,128)<0)
{
ERR_("listen");
return -1;
}
puts("listen success");
//创建集合
fd_set readfds,tempfds;
//将集合初始化
FD_ZERO(&readfds);
FD_ZERO(&tempfds);
//将需要监测的文件描述符放入集合中
FD_SET(0,&tempfds);
FD_SET(sfd,&tempfds);
int newfd=-1;
int maxfd = sfd;
ssize_t res=0;
char buf[128]="";
char buf1[128]="";
struct sockaddr_in saveCin[1024-3];//newfd对应的客户端信息存储在newfd-3下标位置
while(1)
{
//让内核监测集合中的文件描述符是否准备就绪
readfds=tempfds;
int rea=select(maxfd+1,&readfds,NULL,NULL,NULL);
if(rea==-1)
{
ERR_("select");
return -1;
}
// printf("maxfd:%d%d\n",maxfd,__LINE__);
else if(rea==0)
{
printf("time out...");
break;
}
printf("触发了:%d个文件描述符\n",rea);
for(int i=0;i<=maxfd;i++)
{
if(FD_ISSET(i,&readfds)==0)
{
continue;
}
if(i==0)
{
//键盘输入事件
printf("触发键盘输入事件\n");
int sndfd;
res = scanf("%d %s",&sndfd,buf1);
while(getchar()!=10);
if(res!=2)
{
//判断是不是非法的文件描述符
fprintf(stderr,"输入的数据格式错误\n");
continue;
}
if(sndfd<=2||sndfd>=1024||!FD_ISSET(sndfd,&tempfds))
{
fprintf(stderr,"sndfd=%d 文件描述符错误\n",sndfd);
continue;
}
if(send(sndfd,buf1,sizeof(buf1),0)<0)
{
ERR_("send");
continue;
}
printf("发送成功");
}
else if(sfd==i)
{
//客户端接入事件
printf("触发客户端连接事件\n");
newfd = accept(sfd,(struct sockaddr *)&cin,&len);
if(newfd<0)
{
ERR_("accept");
continue;
}
printf("[%s:%d]newfd=%d 客户端连接成功\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
//将客户端信息另存
saveCin[newfd-3] = cin;
//将newfd添加到集合readfds中
FD_SET(newfd,&tempfds);
//更新maxfd
maxfd = maxfd>newfd ? maxfd:newfd;
}
else
{
//客户端交互事件
printf("触发客户端交互信息\n");
bzero(buf,sizeof(buf));
//接收数据
res=recv(i,buf,sizeof(buf),0);
if(res<0)
{
ERR_("recv");
continue;
}
else if(0==res)
{
printf("[%s:%d]newfd=%d 客户端下线\n",\
inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i-3].sin_port),i);
close(i);
FD_CLR(i,&tempfds);
while(!FD_ISSET(maxfd,&readfds)&&maxfd-->=0);
continue;
}
if(send(i,buf1,sizeof(buf1),0)<0)
{
ERR_("send");
return -1;
}
puts("send success");
printf("[%s:%d]newfd=%d发来了: %s\n",\
inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i3].sin_port),i,buf);
}
}
}
//关闭套接字
if(close(sfd))
{
ERR_("close");
return -1;
}
return 0;
}
客户端
#include
#define ERR_MSG(msg) do{\
fprintf(stderr, "_%d_", __LINE__);\
perror("msg");\
}while(0)
#define PORT 8888
#define IP "192.168.122.50"
int main(int argc, const char *argv[])
{
//创建套接字
int sfd = socket(AF_INET, SOCK_STREAM, 0);
if (sfd < 0)
{
ERR_MSG("socket");
return -1;
}
printf("socket success\n");
//端口快速重用
int rouse = 1;
if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &rouse, sizeof(rouse)) < 0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("setsockopt success\n");
//绑定服务器信息
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_port = htons(PORT);
sin.sin_addr.s_addr = inet_addr(IP);
//连接服务器
if (connect(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
{
ERR_MSG("connect");
return -1;
}
printf("connect success\n");
char buf[128] = "";
ssize_t res = 0;
int s_res = 0;
//创建读集合
fd_set readfds, tempfds;
//清空集合
FD_ZERO(&readfds);
//将文件描述符填入集合
FD_SET(0, &readfds);
FD_SET(sfd, &readfds);
//接收发送
while (1)
{
tempfds = readfds;
//阻塞等待集合的文件描述符就绪
s_res = select(sfd+1, &tempfds, NULL, NULL, NULL);
if (s_res < 0)
{
ERR_MSG("select");
return -1;
}
else if (0 == s_res)
{
printf("timeout...");
break;
}
printf("__%d__\n", __LINE__);
if (FD_ISSET(0, &tempfds))
{
printf("键盘输入\n");
res = scanf("%s", buf);
while(getchar() != '\n');
//发送
if(send(sfd, buf, sizeof(buf), 0) < 0)
{
ERR_MSG("send");
continue;
}
printf("send success\n");
}
if (FD_ISSET(sfd, &tempfds))
{
printf("服务器交互\n");
bzero(buf, sizeof(buf));
//接收
res = recv(sfd, buf, sizeof(buf), 0);
if (res < 0)
{
ERR_MSG("recv");
continue;
}
else if(0 == res)
{
printf("服务器下线\n");
break;
}
printf(" : %s\n", buf);
bzero(buf, sizeof(buf));
}
}
close(sfd);
return 0;
}