server.c

#include <stdio.h>
#include <stdlib.h>
#include "sqlite3.h"
#include <sys/socket.h>
#include <sys/types.h>
#include <string.h>

static int callback2(void *NotUsed, int argc, char **argv, char **azColName) 
{ 
     int i=0,k=0; 
     record_num=argc;
     for(i=0; i<argc; i++){
        	strcpy(s_callback2[i],argv[i] ? argv[i] : "NULL");
     } 
     printf("\n");
     return 0; 
}
int record_num=0;
int main(
{
	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("accept client %s/n",inet_ntoa(remote_addr.sin_addr));  
    	len=send(client_sockfd,"Welcome to my server/n",21,0);//发送欢迎信息  
      
    /*接收客户端的数据并将其发送给客户端--recv返回接收到的字节数，send返回发送的字节数*/  
    	while((len=recv(client_sockfd,buf,BUFSIZ,0))>0))  
    	{
		  
        	buf[len]='/0';  
        	printf("%s/n",buf);  
        	/*if(send(client_sockfd,buf,len,0)<0)  
        	{  
            		perror("write");  
            		return 1;  
        	}*/
		char s_callback2[100][1024];
		memset(s_callback2,' ',sizeof(s_callback2));
		char *ErrMsg = 0;
    		int ret = 0;
		int i=0;
   	 	sqlite3 *db = 0;
    		ret = sqlite3_open("mail.db",&db);
    		if(ret != SQLITE_OK){
        		fprintf(stderr,"Cannot open db: %s\n",sqlite3_errmsg(db));
        		return 1;
    		}
		char sql[1024];
		memset(sql,0,sizeof(sql));
		char sender[13];
		memset(sender,0,sizeof(sender));
		char recer[13];
		memset(recer,0,sizeof(recer));
		char date_time[27];
		memset(date_time,0,sizeof(date_time));
		char title[81];
		memset(title,0,sizeof(title));
		char usrname[13];
		memset(usrname,0,sizeof(usrname));
		//char contactor[13];
		//memset(contactor,0,sizeof(contactor));
		switch(buf[0]){
			
			case '2':
				{
					if(buf[1]=='1'){//向客户端发送"已发送"内容
						memset(sender,0,sizeof(sender));
						memcpy(sender,buf+3,(int)buf[2]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));
						sprintf(sql,"SELECT title,date_time FROM send WHERE sender='%s'AND status_sender=2 OR status_sender=5 OR status_sender=8;",sender);
						col_num=2;
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						for(i=0;i<record_num;i++){
							if(send(client_sockfd,s_callback2[i],len,0)<0)  
        						{  
            							perror("write");  
            							return 1;  
        						}
							else{
								printf("send s_callback2[%d] success\n",i);
							}
						}
						record_num=0;
					}
					if(buf[1]=='2'){//向客户端发送一封邮件的具体内容
						memset(sender,0,sizeof(sender));
						memcpy(sender,buf+3,(int)buf[2]);
						memset(date_time,0,sizeof(date_time));
						memcpy(date_time,buf+15,(int)buf[14]);
						memset(recer,0,sizeof(recer));
						memcpy(recer,buf+42,(int)buf[41]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));
						sprintf(sql,"SELECT content FROM send WHERE date_time='%s' AND sender IN (SELECT sender FROM receive WHERE sender='%s' AND date_time='%s' AND recer='%s');",date_time,sender,date_time,recer);
						colnum=1;
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						if(send(client_sockfd,s_callback2[0],len,0)<0)  
        					{  
            						perror("write");  
            						return 1;  
        					}
						else{
							printf("send s_callback2[0] success\n");
						}
						record_num=0;				
					}
					if(buf[1]=='3'){//向客户端发送一封草稿的内容，若有收信人写在content内
						memset(sender,0,sizeof(sender));
						memcpy(sender,buf+3,(int)buf[2]);
						memset(date_time,0,sizeof(date_time));
						memcpy(date_time,buf+15,(int)buf[14]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));
						sprintf(sql,"SELECT content FROM send WHERE sender='%s' AND date_time='%s';",sender,date_time);
						colnum=1;
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						if(send(client_sockfd,s_callback2[0],len,0)<0)  
        					{  
            						perror("write");  
            						return 1;  
        					}
						else{
							printf("send s_callback2[0] success\n");
						}
						record_num=0;	
					}
					break;			
				}
			case '3':
				{
					if(buf[1]=='0'){//向客户端发收信箱内容
						memset(rec,0,sizeof(recer));
						memcpy(recer,buf+3,(int)buf[2]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));						
						sprintf(sql,"SELECT s.sender,s.title,s.date_time FROM send s,receive r WHERE s.sender=r.sender AND s.date_time=r.date_time AND recer='%d' AND status_recer=3;",recer);
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						for(i=0;i<record_num;i++){
							if(send(client_sockfd,s_callback2[i],len,0)<0)  
        						{  
            							perror("write");  
            							return 1;  
        						}
							else{
								printf("send s_callback2[%d] success\n",i);
							}
						}
						record_num=0;
					}
					if(buf[1]=='1'){//向客户端发收件箱内容
						memset(rec,0,sizeof(recer));
						memcpy(recer,buf+3,(int)buf[2]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));						
						sprintf(sql,"SELECT s.sender,s.title,s.date_time FROM send s,receive r WHERE s.sender=r.sender AND s.date_time=r.date_time AND recer='%d' AND status_recer=3 OR status_recer=6;",recer);
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						for(i=0;i<record_num;i++){
							if(send(client_sockfd,s_callback2[i],len,0)<0)  
        						{  
            							perror("write");  
            							return 1;  
        						}
							else{
								printf("send s_callback2[%d] success\n",i);
							}
						}
						record_num=0;
					}
					if(buf[1]=='2'){//向客户端发已删除箱内容
						memset(rec,0,sizeof(recer));
						memcpy(recer,buf+3,(int)buf[2]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));						
						sprintf(sql,"SELECT s.sender,s.title,s.date_time FROM send s,receive r WHERE s.sender=r.sender AND s.date_time=r.date_time AND recer='%d' AND status_recer=4 OR status_recer=7;",recer);
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						for(i=0;i<record_num;i++){
							if(send(client_sockfd,s_callback2[i],len,0)<0)  
        						{  
            							perror("write");  
            							return 1;  
        						}
							else{
								printf("send s_callback2[%d] success\n",i);
							}
						}
						record_num=0;
					}
					break;			
				}
			case '4':
				{
					if(buf[1]=='0'){
						memset(usrname,0,sizeof(usrname));
						memcpy(usrname,buf[3],(int)buf[2]);
						memset(sql,0,sizeof(sql));
						memset(s_callback2,0,sizeof(s_callback2));
						sprintf(sql,"SELECT contactor FROM contact WHERE uarname='%s';",usrname);
						sqlite3_exec(db,sql,callback2,0,&ErrMsg);
						for(i=0;i<record_num;i++){
							if(send(client_sockfd,s_callback2[i],len,0)<0)  
        						{  
            							perror("write");  
            							return 1;  
        						}
							else{
								printf("send s_callback2[%d] success\n",i);
							}
						}
						record_num=0;
					}
					break;
				}
		}  
    	}
	  
    	close(client_sockfd);  
    	close(server_sockfd);  
	return 0;
}