网络字典服务器端

#include <stdio.h>

#include <sys/wait.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <sqlite3.h>

#include <signal.h>

#include <time.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <arpa/inet.h>

 

#define  N  256

#define  R  1   //  user register

#define  L  2   //  user login

#define  Q  3   //  query word

#define  H  4   //  history record

 

#define DATABASE "my.db"

 

typedef struct sockaddr SA;

 

typedef struct

{

    int type;

    char name[N];

    char data[N];   // password or word

} MSG;

 

void handler(int signo)

{

    int a;

    pid_t pid;

    while ((pid = waitpid(-1, &a, WNOHANG)) > 0)

        printf("child %d terminate:%#x\n", pid, a);

}

 

void do_register(int connfd, MSG *pbuf, sqlite3 *db)

{

    char sqlstr[N];

    char *errmsg;

   

    sprintf(sqlstr, "insert into usr values ('%s', '%s')", pbuf->name, pbuf->data);

    printf("%s\n", sqlstr);

 

    bzero(pbuf->data, sizeof(pbuf->data));

    if (sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg) != SQLITE_OK)

    {

        sqlite3_free(errmsg);

        sprintf(pbuf->data, "user %s already exist!!!", pbuf->name);

    }

    else

        strcpy(pbuf->data, "OK");

    send(connfd, pbuf, sizeof(MSG), 0);

}

 

 

void do_login(int connfd, MSG *pbuf, sqlite3 *db)

{

    char sqlstr[N];

    char *errmsg, **result;

    int nrow, ncolumn;

   

    sprintf(sqlstr, "select * from usr where name = '%s' and pass = '%s'", pbuf->name, pbuf->data);

    if (sqlite3_get_table(db, sqlstr, &result, &nrow, &ncolumn, &errmsg) != SQLITE_OK)

    {

        printf("error : %s\n", errmsg);

        sqlite3_free(errmsg);

    }

       

    bzero(pbuf->data, sizeof(pbuf->data));

    if (nrow == 0)

        strcpy(pbuf->data, "name or password is wrong!!!");

    else

        strcpy(pbuf->data, "OK");

 

    send(connfd, pbuf, sizeof(MSG), 0);

 

    sqlite3_free_table(result);

}

 

void get_date(char date[])

{

    time_t t;

    struct tm *tp;

 

    time(&t);

   

    tp = localtime(&t);

    /*

    sprintf(date, "%d-%02d-%02d %02d:%02d:%02d", tp->tm_year+1900,

            tp->tm_mon+1, tp->tm_mday, tp->tm_hour, tp->tm_min, tp->tm_sec);

    */

    strftime(date, 64, "%Y-%m-%d %H:%M:%S", tp);

 

    return;

}

 

int do_searchword(int connfd, MSG *pbuf)

{

    FILE *fp;

    char line[300]={0};

    char *p;

    int len, result;

        

    /***************pbuf->data 传入单词*****************/

    len = strlen(pbuf->data);//pbuf->data --- about

    if ((fp = fopen("dict.txt", "r")) == NULL)

        return -1;

 

    printf("query word is %s\n", pbuf->data);

    /***************fgets 函数读到’\n’结束 **********************/

    while (fgets(line, 300, fp) != NULL)

    {

       /****************和单词比较******************/

        result = strncmp(pbuf->data, line, len);

        if (result > 0) continue;

             

        if (result < 0 || line[len] != ' ') break;

                       

        p = line + len;

        /*******取单词解释段*************/

        while (*p == ' ') p++;

        strcpy(pbuf->data, p);

        fclose(fp);

        return 1;

    }

    fclose(fp);

 

    return 0;

}

 

void do_query(int connfd, MSG *pbuf, sqlite3 *db)

{

    char sqlstr[128], *errmsg;

    int found = 0;

    char date[64]={0}, word[64] = {0};

           

    strcpy(word, pbuf->data);

    found = do_searchword(connfd, pbuf);

 

    if (found == -1)

        strcpy(pbuf->data, "server error\n");

    else if ( found )

    {

        get_date(date);

        sprintf(sqlstr, "insert into record values ('%s', '%s', '%s')", pbuf->name, date, word);

        if (sqlite3_exec(db, sqlstr, NULL, NULL, &errmsg) != SQLITE_OK)

        {

            printf("error : %s\n", errmsg);

            sqlite3_free(errmsg);

        }

    }

    else

        strcpy(pbuf->data, "not found\n");

    send(connfd, pbuf, sizeof(MSG), 0);

}

 

 

int history_callback(void *arg, int f_num, char **f_value, char **f_name)

{

    int connfd;

    MSG buf;

 

    connfd = *(int *)arg;

    sprintf(buf.data, "%s : %s", f_value[1], f_value[2]);

    send(connfd, &buf, sizeof(buf), 0);

 

    return 0;

}

 

void do_history(int connfd, MSG *pbuf, sqlite3 *db)

{

    char sqlstr[128], *errmsg;

 

    sprintf(sqlstr, "select * from record where name = '%s'", pbuf->name);

    if (sqlite3_exec(db, sqlstr, history_callback, (void *)&connfd, &errmsg) != SQLITE_OK)

    {

        printf("error : %s\n", errmsg);

        sqlite3_free(errmsg);

    }

    pbuf->data[0] = '\0';

    send(connfd, pbuf, sizeof(MSG), 0);

   

    return;

}

 

int main(int argc, char *argv[])

{

    int listenfd, connfd;

    struct sockaddr_in myaddr;

    pid_t pid;

    MSG buf;

    sqlite3 *db;

 

    if (argc < 3)

    {

        printf("Usage : %s <ip> <port>\n", argv[0]);

        exit(-1);

    }

 

    if (sqlite3_open(DATABASE, &db) != SQLITE_OK)

    {

        printf("error : %s\n", sqlite3_errmsg(db));

        exit(-1);

    }

 

    if ((listenfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)

    {

        perror("fail to socket");

        exit(-1);

    }

 

    bzero(&myaddr, sizeof(myaddr));

    myaddr.sin_family = PF_INET;

    myaddr.sin_port = htons(atoi(argv[2]));

    myaddr.sin_addr.s_addr = inet_addr(argv[1]);

    if (bind(listenfd, (SA *)&myaddr, sizeof(myaddr)) < 0)

    {

        perror("fail to bind");

        exit(-1);

    }

 

    if (listen(listenfd, 5) < 0)

    {

        perror("fail to listen");

        exit(-1);

    }

 

    signal(SIGCHLD, handler);

    while ( 1 )

    {

        if ((connfd = accept(listenfd, NULL, NULL)) < 0)

        {

            perror("fail to accept");

            exit(-1);

        }

        if ((pid = fork()) < 0)

        {

            perror("fail to fork");

            exit(-1);

        }

        else if (pid == 0)

        {

            close(listenfd);

            while (recv(connfd, &buf, sizeof(buf), 0) > 0)  // receive request

        {

                switch ( buf.type )

             {

                    case R :

                      do_register(connfd, &buf, db);

                      break;

                    case L :

                      do_login(connfd, &buf, db);

                      break;

                  case Q :

                        do_query(connfd, &buf, db);

                      break;

                  case H :

                        do_history(connfd, &buf, db);

                      break;

                }

            }

            printf("client quit\n");

            close(connfd);

            exit(0);

        }

        close(connfd);

    }

 

    exit(0);

}