完成TCP的服务器,客户端,上交。服务器,客户端需要做到随时收发:多进程多线程实现。
服务器
#include <stdio.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/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <pthread.h>
#define ERR_MSG(msg) do{\
fprintf(stderr, "line:%d\n", __LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.8.125"
#define PORT 8888
char buf[128]="";
ssize_t res=0;
int newfd=0;
struct sockaddr_in cin; //存储获取到的客户端的地址信息
socklen_t addrlen=sizeof(cin); //真实地址信息结构体的大小
void* callBack_r (void* arg)
{
while(1)
{
bzero(buf,sizeof(buf));
//接收
res=recv(newfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MSG("recv");
break;
}
else if(0==res)
{
fprintf(stderr, "[%s:%d] newfd = %d 客户端下线\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd);
break;
}
if(strcmp(buf,"quit")==0)
{
break;
}
printf("[%s:%d] newfd = %d : %s\n", \
inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),newfd, buf);
}
pthread_cancel(*(pthread_t*)arg);
pthread_exit(NULL);
}
void* callBack_w (void* arg)
{
while(1)
{
bzero(buf,sizeof(buf));
printf("请输入>>>");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1];
if(send(newfd, buf, sizeof(buf), 0) < 0)
{
ERR_MSG("send");
break;
}
if(strcmp(buf,"quit")==0)
{
break;
}
printf("send success\n");
}
pthread_cancel(*(pthread_t*)arg);
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;
}
//允许端口快速重启
//必须放在bind之前
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,128)<0)
{
ERR_MSG("listen");
return -1;
}
printf("listen success\n");
//从已经完成连接的队列头中获取一个客户端的信息,生成一个新的文件描述符
//生成的新的文件描述符才是与客户端通信的文件描述符
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);
pthread_t tid_r, tid_w;
//创建一个读线程
if(pthread_create(&tid_r, NULL, callBack_r, &tid_w) !=0)
{
fprintf(stderr, "pthread_create failed __%d__\n", __LINE__);
return -1;
}
//创建一个写线程
if(pthread_create(&tid_w, NULL, callBack_w, &tid_r) !=0)
{
fprintf(stderr, "pthread_create failed __%d__\n", __LINE__);
return -1;
}
pthread_join(tid_r, NULL);
pthread_join(tid_w, NULL);
//关闭文件描述符;
close(newfd);
close(sfd);
return 0;
}
客户端
#include <stdio.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/stat.h>
#include <fcntl.h>
#include <errno.h>
#include <pthread.h>
#define ERR_MAG(msg) do{\
fprintf(stderr,"line:%d\n",__LINE__);\
perror(msg);\
}while(0)
#define IP "192.168.8.125"
#define PORT 8888
char buf[128]="";
ssize_t res=0;
int sfd;
void* callBack_w (void* arg)
{
while(1)
{
bzero(buf,sizeof(buf));
printf("请输入>>>");
fgets(buf,sizeof(buf),stdin);
buf[strlen(buf)-1]=0;
if(send(sfd,buf,sizeof(buf),0)<0)
{
ERR_MAG("send");
break;
}
if(strcmp(buf,"quit")==0){
break;
}
printf("send success\n");
}
pthread_cancel(*(pthread_t*)arg);
pthread_exit(NULL);
}
void* callBack_r (void* arg)
{
while(1)
{
bzero(buf,sizeof(buf));
res=recv(sfd,buf,sizeof(buf),0);
if(res<0)
{
ERR_MAG("recv");
break;
}
else if(0==res)
{
printf("客户端下线\n");
break;
}
if(strcmp(buf,"quit")==0){
break;
}
printf("%s",buf);
}
pthread_cancel(*(pthread_t*)arg);
pthread_exit(NULL);
}
int main(int argc, const char *argv[])
{
sfd=socket(AF_INET,SOCK_STREAM,0);
if(sfd<0)
{
ERR_MAG("socket");
return -1;
}
printf("socket success\n");
int reuse = 1;
if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
{
ERR_MAG("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(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
{
ERR_MAG("connect");
return -1;
}
printf("connect success\n");
pthread_t tid_r, tid_w;
//创建一个读线程
if(pthread_create(&tid_r, NULL, callBack_r, &tid_w) !=0)
{
fprintf(stderr, "pthread_create failed __%d__\n", __LINE__);
return -1;
}
//创建一个写线程
if(pthread_create(&tid_w, NULL, callBack_w, &tid_r) !=0)
{
fprintf(stderr, "pthread_create failed __%d__\n", __LINE__);
return -1;
}
pthread_join(tid_r, NULL);
pthread_join(tid_w, NULL);
close(sfd);
return 0;
}
结果