服务器、客户端需要做到随时收发:多进程多线程实现。
server .c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <strings.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#define PORT 7777
#define IP "192.168.8.68"
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:__%d__\n",__LINE__);\
perror(msg);\
}while(0)
//接收
char buf[128]="";
ssize_t res=0;
struct sockaddr_in cin; //存储获取到的客户端的地址信息
//创建双线程用于收发
pthread_t tid1,tid2;
//收
void *callback1(void *arg)
{
while(1)
{
bzero(buf,sizeof(buf));
res = recv(*(int *)arg, buf,sizeof(buf) , 0);
if(-1 == res)
{
ERR_MSG("recv");
return NULL;
}
else if(0 == res)
{
fprintf(stderr,"[%s:%d] newfd=%d\n 客户端下线\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),*(int *)arg);
break;
}
if(!strcmp(buf,"quit"))
{
printf("客户端主动下线\n");
pthread_cancel(tid2);
break;
}
printf("[%s:%d] newfd=%d: %s\n",\
inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),*(int *)arg,buf);
}
pthread_exit(NULL);
}
//发
void *callback2(void *arg) //void *arg=&newfd
{
while(1)
{
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]='\0';
if(send(*(int *)arg,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return NULL;
}
if(!strcmp(buf,"quit"))
{
printf("服务器退出成功\n");
pthread_cancel(tid1);
break;
}
printf("send success\n");
}
pthread_exit(NULL);
}
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 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);
//绑定服务器的IP和端口
if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
{
ERR_MSG("bind");
return -1;
}
printf("bind success\n");
//将套接字设置为监听状态
if(listen(sfd,5)<0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success\n");
//由于自动获取,只需要定义几个结构体变量进行接受,需要时直接访问里面成员即可
socklen_t addrlen=sizeof(cin);
//从已经完成链接的队列头获取一个客户端的信息,
//生成一个新的文件描述符,新的才是用于和客户端通信的文件描述符
int 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);
//创建callback1线程用于服务器收
if(pthread_create(&tid1,NULL,callback1,&newfd)!=0)
{
fprintf(stderr,"pthread_create error\n");
return -1;
}
pthread_detach(tid1);
//创建callback2线程用于服务器发
if(pthread_create(&tid2,NULL,callback2,&newfd)!=0)
{
fprintf(stderr,"pthread_create error\n");
return -1;
}
//阻塞线程
pthread_join(tid2,NULL);
//关闭文件描述符
close(sfd);
close(newfd);
return 0;
}
client.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <strings.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#define PORT 7777
#define IP "192.168.8.68"
#define ERR_MSG(msg) do{\
fprintf(stderr,"line:__%d__\n",__LINE__);\
perror(msg);\
}while(0)
char buf[128]="";
ssize_t res=0;
//创建双线程用于收发
pthread_t tid1,tid2;
//发
void *callback1(void *arg)//void *arg=&sfd
{
while(1)
{
//发送信息给服务器
bzero(buf,sizeof(buf));
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]='\0';
if(send(*(int *)arg,buf,sizeof(buf),0)<0)
{
ERR_MSG("send");
return NULL;
}
if(!strcmp(buf,"quit"))
{
printf("客户端主动下线\n");
pthread_cancel(tid2);
break;
}
printf("send success\n");
}
pthread_exit(NULL);
}
//收
void *callback2(void *arg) //void *arg=&newfd
{
while(1)
{
//接收服务器的信息
res = recv(*(int *)arg, buf , sizeof(buf) , 0);
if(-1 == res)
{
ERR_MSG("recv");
return NULL;
}
else if(0 == res)
{
fprintf(stderr,"sfd=%d\n 客户端下线\n",*(int *)arg);
break;
}
if(!strcmp(buf,"quit"))
{
printf("服务器退出成功\n");
pthread_cancel(tid1);
break;
}
printf("sfd=%d: %s\n",*(int *)arg,buf);
}
}
int main(int argc, const char *argv[])
{
//创建套接字
int sfd=socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MSG("socket");
return -1;
}
struct sockaddr_in sin; //绑定服务器信息
socklen_t addrlen=sizeof(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,addrlen)<0)
{
ERR_MSG("connect");
return -1;
}
printf("connect success\n");
//创建callback1线程用于服务器发
if(pthread_create(&tid1,NULL,callback1,&sfd)!=0)
{
fprintf(stderr,"pthread_create error\n");
return -1;
}
pthread_detach(tid1);
//创建callback2线程用于服务器收
if(pthread_create(&tid2,NULL,callback2,&sfd)!=0)
{
fprintf(stderr,"pthread_create error\n");
return -1;
}
//阻塞线程
pthread_join(tid2,NULL);
//关闭文件描述符
close(sfd);
return 0;
}
实现结果: