#include <stdlib.h>
#include <sys/types.h>
#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>
//server
int main()
{
int sfp, nfp,flag=0;
struct sockaddr_in s_add,c_add;
int sin_size;
unsigned short portnum=0x8888;
char buffer[100] = {0};
pid_t pid;
printf("Hello,welcome to my server !\r\n");
//create socket
sfp = socket(AF_INET, SOCK_STREAM, 0);
if(-1 == sfp)
{
printf("socket fail ! \r\n");
return -1;
}
printf("socket ok !\r\n");
//clean struct sockaddr and init
bzero(&s_add,sizeof(struct sockaddr_in));
s_add.sin_family=AF_INET;
s_add.sin_addr.s_addr=htonl(INADDR_ANY);
s_add.sin_port=htons(portnum);
//bind s_add to sfp
if(-1 == bind(sfp,(struct sockaddr *)(&s_add), sizeof(struct sockaddr)))
{
printf("bind fail !\r\n");
return -1;
}
printf("bind ok !\r\n");
//wait connect
if(-1 == listen(sfp,5))
{
printf("listen fail !\r\n");
return -1;
}
printf("listen ok\r\n");
sin_size = sizeof(struct sockaddr_in);
//accept connect request
nfp = accept(sfp, (struct sockaddr *)(&c_add), &sin_size);
if(-1 == nfp)
{
printf("accept fail !\r\n");
return -1;
}
printf("accept ok!\nServer start get connect from %#x : %#x\n",
ntohl(c_add.sin_addr.s_addr), ntohs(c_add.sin_port));
/************ massage **************************/
pid = fork();
if(pid==-1)
{
printf("pid create failed!!\n");
exit(1);
}
else if(pid == 0)//accept
{
printf("pid_s = %d\n",getpid());
int recbyte=0;
while(1)
{
memset(buffer,0,100);
if(-1 == (recbyte = read(nfp,buffer,1024)))
{
printf("read data failed!\n");
return -1;
}
if(0==strncmp(buffer,"quit",4))
{
printf("client quit OK!\n");
memset(buffer,0,100);
strcpy(buffer,"quit");
printf("quit OK!\n");
flag=1;
send(nfp,buffer,sizeof(buffer),0);
//return -1;
}
if(flag != 1)
printf("client: %s\n",buffer);
sleep(1);
}
}
else//send
{
printf("pid_f = %d\n",getppid());
memset(buffer, 0, 100);//clean buffer
sleep(1);
while(fgets(buffer,sizeof(buffer),stdin))//wait stdin
{
if(strncmp(buffer,"quit",4) == 0)
{
memset(buffer,0,100);
strcpy(buffer,"quit");
printf("quit OK!\n");
send(nfp,buffer,sizeof(buffer),0);
break;
}
send(nfp,buffer,sizeof(buffer),0);
memset(buffer,0,100);
}
}
/********************************************************************/
close(nfp);
close(sfp);
return 0;
}
