Linux TCP协议使用线程实现并发服务器

来源：互联网发布：张松献图知乎编辑：程序博客网时间：2024/06/12 02:08

直接放代码：

服务器：

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>

#define PORTNUM 3333

//char send_buf[1024];
#if 0
void write(void *sock_fd)
{
int *sock_fd_new = (int *)sock_fd;
printf("send:");
scanf("%s",send_buf);
if(send((*sock_fd_new),send_buf,1024,0) < 0)
{
printf("send error\n");
}
pthread_exit(0);
}
#endif
void* read(void *sock_fd)
{
int sock_fd_new = (*((int *)sock_fd));
char buf[1024];

while(1)
{
bzero(buf,sizeof(buf));
//5.recv
if(recv((sock_fd_new),(void *)buf,1024,0) > 0)
{
printf("buf = %s\n",buf);
}

if(send((sock_fd_new),"ok!",1024,0) < 0)
{
printf("send error\n");
}

if(strncmp(buf,"quit",4) == 0)
{
close(sock_fd_new);
break;
}
}
pthread_exit(0);
}
int main()
{
pthread_t ip;
int sock_fd;
int sock_fd_new;
struct sockaddr_in my_addr;
int addrlen = sizeof(struct sockaddr);
struct sockaddr_in addr_client;
//1.socket
sock_fd = socket(AF_INET,SOCK_STREAM,0);
if(sock_fd < 0)
{
printf("socket error\n");
exit(1);
}
//2.bind
bzero(&my_addr,sizeof(struct sockaddr_in));
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(PORTNUM);
//my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
my_addr.sin_addr.s_addr = inet_addr("192.168.1.11");
if(bind(sock_fd,(struct sockaddr *)(&my_addr),addrlen) < 0)
{
printf("bind error\n");
close(sock_fd);
exit(1);
}
//3.listen
if(listen(sock_fd,10) < 0)
{
printf("listen error\n");
}
while(1)
{
//4.accept
bzero(&addr_client,sizeof(struct sockaddr_in));
sock_fd_new = accept(sock_fd,(struct sockaddr *)(&addr_client),&addrlen);
if(sock_fd_new < 0)
{
printf("accept error\n");
}
pthread_create((&ip),NULL,(void *)read,(void *)(&sock_fd_new));//有客户发送就新建一个线程，不需要两个线程分别读写，服务器写不知道写哪里。
//pthread_create(&ip,NULL,(void *)write,(void *)(&sock_fd_new));
}
//6.close
close(sock_fd);

return 0;
}

客户：

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <arpa/inet.h>

#define PORTNUM 3333

char buf_read[1024];
char buf_send[1024];

void write(void *sock_fd)
{
int *sock_fd_new = (int *)sock_fd;

while(1)
{
bzero(buf_read,sizeof(buf_read));
printf("send:");
scanf("%s",buf_send);
if(send((*sock_fd_new),buf_send,1024,0) < 0)
{
printf("send error\n");
}
if(strncmp(buf_send,"quit",4) == 0)
{
break;
}
}
pthread_exit(0);
}

void read(void *sock_fd)
{
int *sock_fd_new = (int *)sock_fd;

while(1)
{
bzero(buf_read,sizeof(buf_read));
//5.recv
if(recv((*sock_fd_new),(void *)buf_read,1024,0) > 0)
{
printf("buf = %s\n",buf_read);
}
if(strncmp(buf_send,"quit",4) == 0)
{
break;
}
}
pthread_exit(0);
}
int main()
{
pthread_t ip1;
pthread_t ip2;
int sock_fd;
struct sockaddr_in addr_serv;
int addrlen = sizeof(struct sockaddr);
//1.socket
sock_fd = socket(AF_INET,SOCK_STREAM,0);
if(sock_fd < 0)
{
printf("socket error\n");
exit(1);
}
//2.connet
bzero(&addr_serv,sizeof(struct sockaddr_in));
addr_serv.sin_family = AF_INET;
addr_serv.sin_port = htons(PORTNUM);
//addr_serv.sin_addr.s_addr = htonl(INADDR_ANY);
addr_serv.sin_addr.s_addr = inet_addr("192.168.1.11");
if(connect(sock_fd,(struct sockaddr *)(&addr_serv),addrlen) < 0)
{
printf("connect error\n");
}

pthread_create(&ip1,NULL,(void *)read,(void *)(&sock_fd));//收到信息是一个线程
pthread_create(&ip2,NULL,(void *)write,(void *)(&sock_fd)); //发送信息是一个线程

pthread_join(ip1,NULL);
pthread_join(ip2,NULL);
//6.close
close(sock_fd);

return 0;
}

一个在努力中的未来程序员，如果有更好的算法，欢迎评论。

0 0