path: root/server.c

#include "bank.h"

extern int sockfd;

SHM_t *SHM;
char *fn_passwd = ".PASSWDS";
char flag;

typedef struct map_t {
    char *key;
    int (*fp)(char *);
} map_t;

void sig_handler(int signo){
    pid_t pid;
    while( (pid = waitpid(-1, NULL, WNOHANG)) > 0)
    fprintf(stderr, "Child %d terminated\n", pid);
    exit(0);
}

int list(char *param){
    printf("list: %s\n", param);
    writemsg(SUCCESS, "list success");
    return 1;
}

int query(char *param){
    char buf[STRLEN];
    sprintf(buf, "%s's money is %d", param, bank_moneyof(bank_searchuser(param)));
    writemsg(SUCCESS, buf);
    return 1;
}

int give(char param[STRLEN]){
    char user[STRLEN], money[STRLEN];
    char tmp[STRLEN];
    printf("give: %s\n", param);
    getcmd(param, user, money);
    bank_deumoney(user, atoi(money));
    sprintf(tmp, "give %s %s", user, money);
    writemsg(SUCCESS, tmp);
    return 1;
}

int quit(char *param){
    printf("[server] client quit\n");
    exit(0);
}

map_t map[] = {
    {"list", list},
    {"query", query},
    {"give", give},
    {"quit", quit},
    {"", NULL},
};

void writemsg(int status, char *str){
    //write(sockfd, &status. sizeof(status));
    writeln(str);
}

void serv_work(void){
    int i, status = ERROR;
    char buf[STRLEN], cmd[STRLEN], param[STRLEN];

    fprintf(stderr, "new child\n");

    attach_SHM();

    while(readline(buf, 600) != NULL){
    printf("[server] read: ");
    puts(buf);
    getcmd(buf, cmd, param);
    for (i = 0; map[i].fp; i++){
        if (!strcasecmp(map[i].key, cmd)){
        status = (*(map[i].fp))(param);
        break;
        }
    }
    if(!map[i].fp){
        writemsg(ERROR, "Wrong command");
        strcpy(param, "error");
    }
    printf("[server] complete\n");
    }
}

int check_perm(char *ip){
}

int main(int argc, char **argv){
    int listenfd;
    pid_t pid;
    socklen_t addr_len;
    struct sockaddr_in serv_addr;

    listenfd = socket(PF_INET, SOCK_STREAM, 0);

    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin_family      = AF_INET;
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_addr.sin_port        = htons(PORT);

    bind(listenfd,(struct sockaddr *) &serv_addr, sizeof(serv_addr));
    listen(listenfd, 5);
    signal(SIGCHLD, sig_handler);

    while(1){
    addr_len = sizeof(serv_addr);

    if( (sockfd = accept(listenfd,(struct sockaddr *) &serv_addr, &addr_len)) < 0){
        if(errno == EINTR)
        continue;
        else
        herror("accept");
    }
#define NOFORK 1
#if NOFORK
    if((pid = fork()) < 0)
        perror("fork");
    
    else if(pid == 0){
#endif
        int len;
        struct sockaddr_in name;
        close(listenfd);

        if(getpeername(sockfd,(struct sockaddr *) &name, &len) < 0){
        perror("getpeername");
        exit(-1);
        }
        else{
        printf("address: %s\n", inet_ntoa(((struct sockaddr_in *) &name)->sin_addr));
        }
        serv_work();
        exit(0);
#if NOFORK
    }
#endif
    close(sockfd);
    }
    return 0;
}