server.c 端
#include<stdio.h>
#include<stdlib.h>#include<unistd.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<pthread.h>
#include<string.h>
#define PORT 1234
#define BACKLOG 5
#define MAXDATASIZE 1000
typedef struct _ARG
{
int connfd;
struct sockaddr_in client;
}ARG;
void process_cli(int connectfd,struct sockaddr_in client);
void* start_routine(void* arg);
int main()
{
int listenfd,connectfd;
pthread_t thread;
ARG *arg;
struct sockaddr_in server;
struct sockaddr_in client;
int sin_size;
if((listenfd=socket(AF_INET,SOCK_STREAM,0))==-1)
{
perror("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");
exit(1);
}
sin_size =sizeof(struct sockaddr_in);
while(1)
{
if((connectfd=accept(listenfd,(struct sockaddr*)&client,&sin_size))==-1)
{
perror("accept failed");
exit(1);
}
// ARG arg;
arg=(void*)malloc(sizeof(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,i;
char recvbuf[MAXDATASIZE],sendbuf[MAXDATASIZE],cli_name[MAXDATASIZE];
printf("you got a connection from %s.\n",inet_ntoa(client.sin_addr));
send(connectfd,"welcome to my server.\n",22,0);
num=recv(connectfd,cli_name,MAXDATASIZE,0);
if(num==0)
{
close(connectfd);
perror("client connected failed.");
exit(1);
}
cli_name[num]='\0';
printf("client'name is %s.\n",cli_name);
while(num=recv(connectfd,recvbuf,MAXDATASIZE,0))
{
recvbuf[num]='\0';
printf("recv client(%s)message:%s\n",cli_name,recvbuf);
for(i=0;i<num;i++)
{
sendbuf[i]=recvbuf[i];
}
sendbuf[num]='\0';
send(connectfd,sendbuf,strlen(sendbuf),0);
}
close(connectfd);
}
void* start_routine(void *arg)
{
ARG *info;
info=(ARG*)arg;
process_cli(info->connfd,info->client);
free(info);
pthread_exit(NULL);
}
client.c 端
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/socket.h>
#include<netinet/in.h>
#include<netdb.h>
#define PORT 1234
#define MAXDATASIZE 1000
int main(int argc,char *argv[])
{
int fd,numbytes;
char buf[MAXDATASIZE];
struct hostent *he;
struct sockaddr_in server;
int i=1;
if((he=gethostbyname(argv[1]))==NULL)
{
perror("gethostbyname error");
exit(1);
}
if((fd=socket(AF_INET,SOCK_STREAM,0))==-1)
{
perror("socket failed");
exit(1);
}
bzero(&server,sizeof(server));
server.sin_family =AF_INET;
server.sin_port =htons(PORT);
server.sin_addr =*((struct in_addr*)he->h_addr);
if(connect(fd,(struct sockaddr*)&server,sizeof(struct sockaddr))==-1)
{
perror("bind error");
exit(1);
}
if((numbytes=recv(fd,buf,MAXDATASIZE,0))==-1)
{
perror("recv error");
exit(1);
}
buf[numbytes]='\0';
printf("%s\n",buf);
printf("input your name:");
scanf("%s",buf);
if((numbytes=send(fd,buf,strlen(buf),0))==-1)
{
perror("send failed");
exit(1);
}
while(i)
{
printf("input me message(max char:%d\n.):",MAXDATASIZE);
scanf("%s",buf);
if(strlen(buf)<1)
{
i=0;
}
if((numbytes=send(fd,buf,strlen(buf),0))==-1)
{
perror("send failed");
exit(1);
}
if((numbytes=recv(fd,buf,MAXDATASIZE,0))==-1)
{
perror("recv error");
exit(1);
}
buf[numbytes]='\0';
printf("server message:%s\n",buf);
printf("\n");
}
close(fd);
return 0;
}