1.TCP多线程并发服务器
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#define PORT 6666
#define IP "192.168.124.124"
#define ERR_MSG(err) do{\
fprintf(stderr,"%d",__LINE__);\
perror(err);\
}while(0)
struct msg
{
int newsfd;
struct sockaddr_in cin;
};
void* recv_send_pthread(void* arg);
int main(int argc, const char *argv[])
{
//创建报式套接字
int sfd=socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("报式套接字创建成功\n");
//允许端口快速被复用
int reuse=1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口快速被复用\n");
//补参数
struct sockaddr_in sin;
sin.sin_family =AF_INET;
sin.sin_port =htons(PORT);
sin.sin_addr.s_addr=inet_addr(IP);
//bind
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("绑定成功\n");
//将套接字文件设置为监听
if(listen(sfd,128)<0)
{
ERR_MSG("listen");
return -1;
}
printf("监听成功\n");
struct sockaddr_in cin; //获取客户端信息
socklen_t size=sizeof(cin); //客户端信息,真实结构体大小
int newsfd;
pthread_t tid;
struct msg clinfo;
while(1){
//accept 阻塞函数
//当客户端请求连接时,创建新的套接字文件
newsfd= accept(sfd,(struct sockaddr*)&cin,&size);
if(newsfd<0)
{
ERR_MSG("accept");
return -1;
}
printf("[%s : %d] newsfd=%d 连接成功\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port),newsfd);
clinfo.newsfd=newsfd;
clinfo.cin=cin;
if(pthread_create(&tid,NULL,recv_send_pthread,(void*)&clinfo)!=0)
{
fprintf(stderr,"pthread_create failed __%d__",__LINE__);
break;
}
pthread_detach(tid); //分离线程
}
//关闭文件描述符
if(close(sfd)<0)
{
ERR_MSG("close");
return -1;
}
return 0;
}
void* recv_send_pthread(void* arg)
{
int newsfd=((struct msg*)arg)->newsfd;
struct sockaddr_in cin = ((struct msg*)arg)->cin;
char buf[128]="";
ssize_t res;
while(1)
{
//接收数据
bzero(buf,sizeof(buf));
res=recv(newsfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("recv");
break;
}
else if(0==res)
{
printf("[%s : %d]客户端关闭\n",inet_ntoa(cin.sin_addr),htons(cin.sin_port));
break;
}
printf("客户端[%s : %d] 发送的数据:%s\n",inet_ntoa(cin.sin_addr),\
htons(cin.sin_port),buf);
//发送数据
strcat(buf ,"*_**_*");
if(send(newsfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
break;
}
printf("[%s : %d]服务器发送数据成功\n",IP,PORT);
}
close(newsfd);
pthread_exit(NULL);
}
2.select的TCP服务端
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define PORT 6666
#define IP "192.168.124.124"
#define ERR_MSG(err) do{\
fprintf(stderr,"%d",__LINE__);\
perror(err);\
}while(0)
int deal_key_send(fd_set readfds);
int deal_CliConnect(int sfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd);
int deal_CliRecvSend(int newfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd);
int main(int argc, const char *argv[])
{
int sfd=socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("流式套接字创建成功\n");
int reuse=1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_MSG("setsockopt");
return -1;
}
printf("允许端口被快速复用\n");
struct sockaddr_in sin;
sin.sin_family=AF_INET;
sin.sin_port=htons(PORT);
sin.sin_addr.s_addr=inet_addr(IP);
if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("绑定成功\n");
if(listen(sfd,128)<0)
{
ERR_MSG("listen");
return -1;
}
printf("监听成功\n");
//添加到select集合
fd_set readfds,tempfds;
FD_ZERO(&readfds); //将集合清空
FD_ZERO(&tempfds);
FD_SET(0,&readfds);
FD_SET(sfd,&readfds);
struct sockaddr_in clinfo[1024];//定义数组存储每个客户端的消息
int maxfd=sfd;
int num;
while(1)
{
tempfds=readfds;//readfds不变,每次重新赋值给tempfds
//
num=select(maxfd+1,&tempfds,NULL,NULL,NULL);
if(num<0)
{
ERR_MSG("select");
return -1;
}
else if(num==0)
{
printf("timeout....\n");
break;
}
//判断select中哪个资源准备就绪
//准备就绪的资源可解除select阻塞,执行对应的功能
for(int i=0;i<=maxfd;i++)
{
if(FD_ISSET(i,&tempfds)==0)
{
continue;//不在集合中,就不是外面监听的文件描述符
//直接跳出,不做后续的判断
}
if(i==0)
{
printf("执行键盘输入事件,向客户端发送消息\n");
deal_key_send(readfds);
}
else if(i==sfd)
{
printf("执行连接客户端事件\n");
deal_CliConnect(sfd,clinfo,&readfds,&maxfd);
}
else
{
printf("执行与客户端的交互事件\n");
deal_CliRecvSend(i,clinfo,&readfds,&maxfd);
}
}
}
if(close(sfd)<0)
{
ERR_MSG("close");
return -1;
}
return 0;
}
//执行键盘输入事件,向客户端发送消息
int deal_key_send(fd_set readfds)
{
char buf[128];
int size;
int newfd;
size=scanf("%d %s",&newfd,buf);//服务端指明要向哪个服务器对应的文件描述符发送数据,发送数据内容
while(getchar()!='\n');
if(size!=2)
{
printf("输入数据格式错误,请正确输入格式:fd string\n");
return -1;
}
if(newfd<=3 || FD_ISSET(newfd,&readfds)==0)
{
printf("请输入合法的文件描述符\n");
return -1;
}
if(send(newfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return -1;
}
return 0;
}
//执行连接客户端事件
int deal_CliConnect(int sfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd)
{
int newfd;
struct sockaddr_in cin;
socklen_t addrlen=sizeof(cin);
newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
if(newfd<0)
{
ERR_MSG("accept");
return -1;
}
printf("[%s %d] newfd=%d客户端连接成功\n",inet_ntoa(cin.sin_addr),\
ntohs(cin.sin_port),newfd);
clinfo[newfd]=cin;//将获取到的客户端信息存储在对应文件描述符的下标中
//将连接到的文件描述符添加到select的集合中
FD_SET(newfd,preadfds);
//更新maxfd
*pmaxfd = *pmaxfd>newfd?*pmaxfd:newfd;
return 0;
}
//执行与客户端的交互事件
int deal_CliRecvSend(int newfd,struct sockaddr_in clinfo[],fd_set *preadfds,int *pmaxfd)
{
char buf[128]="";
ssize_t res;
//接收
bzero(buf,sizeof(buf));
res=recv(newfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("recv");
return -1;
}
else if(0==res)
{
printf("[%s %d] newfd=%d客户端已下线\n",inet_ntoa(clinfo[newfd].sin_addr),\
ntohs(clinfo[newfd].sin_port),newfd);
close(newfd);//下线则关闭对应的文件描述符
//将文件描述符从select集合中删除
FD_CLR(newfd,preadfds);
//更新maxfd
//删除的文件描述符如果不是最大文件描述符,则不需要更新maxfd
//
//删除的文件描述符可能是最大文件描述符
/*
*如果是最大文件描述符,则最大文件描述符不在集合中,
*依次判断最大文件描述符的前一位是否存在(如删除10,则依次判断9,8,7.....)
*/
while(FD_ISSET(*pmaxfd,preadfds)==0 && *pmaxfd-- >0);
return 0;
}
printf("[%s %d] newfd=%d客户端发送的数据:%s\n",inet_ntoa(clinfo[newfd].sin_addr),\
ntohs(clinfo[newfd].sin_port),newfd,buf);
/* //发送数据
strcat(buf,"*-*");
if(send(newfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return -1;
}
printf("发送数据成功\n");
*/
return 0;
}
3.select的TCP客户端
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#define PORT 6666
#define IP "192.168.124.124"
#define ERR_MSG(err) do{\
fprintf(stderr,"%d",__LINE__);\
perror(err);\
}while(0);
int deal_Send(int cfd);
int deal_Recv(int cfd);
int main(int argc, const char *argv[])
{
//创建套接字文件
int cfd=socket(AF_INET,SOCK_STREAM,0);
if(cfd<0)
{
ERR_MSG("socket");
return -1;
}
printf("流式套接字文件创建成功\n");
//绑定bind,非必须绑定
//手动补充服务器的地址信息结构体
//ip,端口号,为connect函数使用
struct sockaddr_in sin;
sin.sin_family =AF_INET;
sin.sin_port =htons(PORT);
sin.sin_addr.s_addr=inet_addr(IP);
//连接服务器
if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MSG("connect");
return -1;
}
printf("连接服务器成功[%s : %d]\n",IP,PORT);
//发送数据
fd_set readfd,tempfd;
FD_ZERO(&readfd);
FD_ZERO(&tempfd);
FD_SET(0,&readfd);
FD_SET(cfd,&readfd);
char buf[128]="";
while(1)
{
tempfd=readfd;
int num=select(cfd+1,&tempfd,NULL,NULL,NULL);
if(num<0)
{
ERR_MSG("select");
return -1;
}
else if(num==0)
{
printf("timeout.....\n");
break;
}
if(FD_ISSET(0,&tempfd))
{
printf("触发键盘输入事件,向服务器发送消息\n");
deal_Send(cfd);
}
if(FD_ISSET(cfd,&tempfd))
{
printf("接收服务器的数据\n");
deal_Recv(cfd);
}
}
if(close(cfd)<0)
{
ERR_MSG("close");
return -1;
}
return 0;
}
int deal_Recv(int cfd)
{
char buf[128]="";
bzero(buf,sizeof(buf));
int res=recv(cfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("recv");
return -1;
}else if(res==0)
{
printf("服务器[%s : %d]已关闭\n",IP,PORT);
return 0;
}
printf("接收服务器[%s: %d]发送的数据:%s\n",IP,PORT,buf);
return 0;
}
int deal_Send(int cfd)
{
//发送数据
char buf[128]="";
ssize_t res;
scanf("%s",buf);
while(getchar()!='\n');
if(send(cfd,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return -1;
}
return 0;
}