service.c

#include <stdio.h>
#include <syspes.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <arpa/inet.h>

int user_num=0;    //普通用户数量
int vip_num=0;     //VIP用户数量

void doService(char buf[]);

int main(int argc, char *argv[])
{
	int server_sockfd; //服务器端套接字
	int client_sockfd; //客户端套接字
	int len;
	struct sockaddr_in my_addr;   //服务器网络地址结构体
	struct sockaddr_in remote_addr; //客户端网络地址结构体
	int sin_size;
	char buf[BUFSIZ];  //数据传送的缓冲区
	memset(&my_addr,0,sizeof(my_addr)); //数据初始化--清零
	my_addr.sin_family=AF_INET; //设置为IP通信
	my_addr.sin_addr.s_addr=INADDR_ANY;//服务器IP地址--允许连接到所有本地地址上
	my_addr.sin_port=htons(8000); //服务器端口号
	
	/*创建服务器端套接字--IPv4协议，面向连接通信，TCP协议*/
	if((server_sockfd=socket(PF_INET,SOCK_STREAM,0))<0)
	{  
	      perror("socket");
	      return 1;
	}
 
        /*将套接字绑定到服务器的网络地址上*/
	if (bind(server_sockfd,(struct sockaddr *)&my_addr,sizeof(struct sockaddr))<0)
	{
	      perror("bind");
	      return 1;
	}
	
	/*监听连接请求--监听队列长度为5*/
	listen(server_sockfd,5);
	
	sin_size=sizeof(struct sockaddr_in);
	
	/*等待客户端连接请求到达*/
	if((client_sockfd=accept(server_sockfd,(struct sockaddr *)&remote_addr,&sin_size))<0)
	{
	      perror("accept");
	      return 1;
	}
	printf("\n\n\naccept client %s\n",inet_ntoa(remote_addr.sin_addr));
	len=send(client_sockfd,"Welcome to my server",21,0);//发送欢迎信息
	
	/*接收客户端的数据并将其发送给客户端--recv返回接收到的字节数，send返回发送的字节数*/
	while((len=recv(client_sockfd,buf,BUFSIZ,0))>0)
        {  
            // doService(buf);      //将接受到的buf作为参数调用自定义的doService函数
            
	    if(buf[0]=='1')
	    {
 	        user_num++;
	        printf("\nuser_num now is : %d",user_num);
	    }
	    else if(buf[0]=='2')
	    {
	         vip_num++;
                 printf("\nvip_num now is : %d",vip_num);
	    }   
            else
	    {
                 printf("wrong");
	    } 
             
            printf("\nuser : %d; vip :%d",user_num,vip_num);

            if(send(client_sockfd,buf,len,0)<0)  
            {  
                 perror("write");  
                 return 1;  
            }  
        }  
        
	close(client_sockfd);
	close(server_sockfd);
        return 0;
}

void doService(char buf[])
{
 	if(buf[0]=='1')
	{
 	    user_num++;
	    printf("\nuser_num now is : %d",user_num);
	}
	else if(buf[0]=='2')
	{
	    vip_num++;
            printf("\nvip_num now is : %d",vip_num);
	}  
        else
	{
            printf("wrong");
	} 
}