- 论坛徽章:
- 0
|
#include<stdlib.h>
#include<stdio.h>
#include<unistd.h>
#include<string.h>
#include<sys/socket.h>
#include<sys/types.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<pthread.h>
#define PORT 1234
#define MAXDATASIZE 100
#define BACKLOG 5
void process_cli(int connectfd,struct sockaddr_in client);
void savedata_r(char* recvbuf,int len,char*client);
void* start_routine(void* arg);
struct ARG{
int connfd;
struct sockaddr_in client;
};
static pthread_key_t key;
static pthread_once_t once=PTHREAD_ONCE_INIT;
static void destructor(void *pt)
{
free(pt);
}
static void getkey_once(void)
{
pthread_key_creat((&key),destructor);
}
struct DATA_THR
{
int index;
};
main()
{
int listenfd,connectfd;
pthread_t thread;
struct ARG* arg;
struct sockaddr_in server;
struct sockaddr_in client;
int sin_size;
if((listenfd=socket(AF_INET,SOCK_STREAM,0))==-1)
{
perror("Creating socket failed.");
exit(1);
}
int opt=SO_REUSEADDR;
setsockopt(listenfd,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
bzero(&server,sizeof(server));
server.sin_family=AF_INET;
server.sin_port=htons(PORT);
server.sin_addr.s_addr=htonl(INADDR_ANY);
if(bind(listenfd,(struct sockaddr*)&server,sizeof(struct sockaddr))==-1)
{
perror("bind error.");
exit(1);
}
if(listen(listenfd,BACKLOG)==-1)
{
perror("listen() error\n");
exit(1);
}
sin_size=sizeof(struct sockaddr_in);
while(1)
{
if((connectfd=accept(listenfd,(struct sockaddr*)&client,&sin_size))==-1)
{
perror("accept() error\n");
exit(1);
}
arg=(struct ARG*)malloc(sizeof(struct ARG));
arg->connfd=connectfd;
memcpy((void*)&arg->client,&client,sizeof(client));
if(pthread_create(&thread,NULL,start_routine,(void*)arg))
{
perror("pthread_create() error");
exit(1);
}
}
close(listenfd);
}
void process_cli(int connectfd,struct sockaddr_in client)
{
int num;
char cli_data[5000];
char recvbuf[MAXDATASIZE],sendbuf[MAXDATASIZE],cli_name[MAXDATASIZE];
printf("You got connection from %s. ",inet_ntoa(client.sin_addr));
num=recv(connectfd,cli_name,MAXDATASIZE,0);
if(num==0)
{
close(connectfd);
printf("Client disconnected.\n");
return;
}
cli_name[num-1]='\0';
printf("Client's name is %s.\n",cli_name);
while(num=recv(connectfd,recvbuf,MAXDATASIZE,0))
{
recvbuf[num]='\0';
printf("Recive client(%s) message: %s",cli_name,recvbuf);
savedata_r(recvbuf,num,cli_data);
fgets(sendbuf,MAXDATASIZE,stdin);
send(connectfd,sendbuf,strlen(sendbuf),0);
}
close(connectfd);
printf("Client (%s) close connection. User's data: %s\n",cli_name,cli_data);
}
void* start_routine(void* arg)
{
struct ARG *info;
info=(void *)arg;
process_cli(info->connfd,info->client);
free(info);
pthread_exit(NULL);
}
void savedata_r(char* recvbuf,int len,char* cli_data)
{
struct DATA_THR* data;
pthread_once(&once,getkey_once);
if((data=(struct DATA_THR*)pthread_getspecific(key))==NULL)
{
data=(struct DATA_THR*)calloc(1,sizeof(struct DATA_THR));
pthread_setspecific(key,data);
data->index=0;
}
for(int i=0;i<len-1;i++)
{
cli_data[data->index++]=recvbuf[i];
}
cli_data[data->index]='\0';
} |
|