ftp_socket模型

#include "myhead.h"
#include "sem.h"
#include "socket.h"
#include "read_write.h"
#include "myhash.h"

#define ERROR_EXIT(m) (perror(m), exit(-1))

int                    listenfd, localfd;
int                    connsockpair[2];
int                    semid;
int                    shmid;

char                sendbuf[1024];
char                recvbuf[1024];
MYSHM               *shm;
MYSHM                connctrl;

struct sockaddr_in    seraddr, cliaddr, localaddr;

int size = sizeof(seraddr);

   //四个哈希表分别存放4对数据
HASHNODE *hash_conf[HASH_MAX] = {NULL};//配置文件
HASHNODE *hash_pid_ip[HASH_MAX] = {NULL};//客户端进程号与对应的IP地址
HASHNODE *hash_ip_count[HASH_MAX] = {NULL};//同一IP的连接数
HASHNODE *hash_comm[HASH_MAX] = {NULL};//命令对应的执行函数




void my_socket()
{
    char ppid[10] = "";
    char tempip[30] = "";
    pid_t pid;
    int msgid;
    int port = 0;
    int peripcount = 0;
    int totalcount = 0;
    int nowperipcount = 0;
    struct sigaction myact;
    struct msgtype msg;

    listenfd = socket(AF_INET,SOCK_STREAM, 0);
    if(listenfd<0)
        ERROR_EXIT("socket");

    memset(&seraddr, 0, sizeof(seraddr));
    seraddr.sin_family  = AF_INET;
    port = atoi((char *)hash_getval(hash_conf,"serverport",1));
    seraddr.sin_port        = htons(port);
    seraddr.sin_addr.s_addr = inet_addr((char *)hash_getval(hash_conf,"serverip",1));

/*设置地址重用*/
    int opt = 1;
    int res = setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, 4);
    if(res<0)
        ERROR_EXIT("setsockopt");

/*绑定侦听套接口*/
    res = bind(listenfd, (Sockad *)&seraddr, sizeof(seraddr));
    if(res<0)
        ERROR_EXIT("bind");
/*设置侦听口*/
    res = listen(listenfd, 5);
    if(res<0)
        ERROR_EXIT("listen");

    memset(&myact, 0, sizeof(myact));
    myact.sa_handler = myquit;
    sigaction(SIGCHLD, &myact, NULL);

    while(1) {
        printf("=========wait client connect=========\n");

        memset(&cliaddr, 0, sizeof(cliaddr));

        localfd = accept(listenfd, (Sockad *)&cliaddr, &size);
        printf("localfd = %d, %d\n", localfd, EINTR);
    /*accetp()是慢系统调用，在信号产生时会中断accept其调用,
    使accept返回-1，并将errno变量设置为EINTR，此时应重新调用accept()。*/
        if(localfd < 0) {                
            if(errno == EINTR)
                continue;
            else
                ERROR_EXIT("accept");
        }

        /*连接数控制*/
        totalcount = atoi((char *)hash_getval(hash_conf, "maxip", 1));
        peripcount = atoi((char *)hash_getval(hash_conf, "perip", 1));

        strcpy(tempip,inet_ntoa(cliaddr.sin_addr));
        if (NULL == hash_getval(hash_ip_count, tempip, 1))
            nowperipcount = 0;
        else
            nowperipcount = atoi((char *)hash_getval(hash_ip_count, tempip, 1));
        
        printf("totalcount=%d\n", totalcount);
        printf("peripcount=%d\n\n", peripcount);

        if (connctrl.conncount >= totalcount || nowperipcount >= peripcount)
        {
            write_loop(localfd, "450 connect max\r\n", strlen("450 connect max\r\n"));
            close(localfd);
            continue;
        }
        else
            printf("nowperipcount=%d\n\n", nowperipcount+1); //该IP连接数

        memset(&localaddr, 0, sizeof(localaddr));
        getsockname(localfd, (Sockad *)&localaddr, &size);

        printf("local_addr = %s\n", inet_ntoa(localaddr.sin_addr));
        printf("local_port = %d\n", ntohs(localaddr.sin_port));
        printf("client_addr = %s\n", inet_ntoa(cliaddr.sin_addr));
        printf("client_port = %d\n\n", ntohs(cliaddr.sin_port));    
        
        write_loop(localfd, "220 WELCOM\r\n", strlen("220 WELCOM\r\n"));
        int pid = fork();
        switch(pid) {
            case -1:
            //    close(localfd);
            //    close(listenfd);
                exit(-1);
            case(0):
                close(listenfd);
                clisession();
                exit(0);
                break;
            default:
                close(localfd);
                sprintf(ppid, "%d", pid);
                printf("ip=[%s] pid=[%s]\n", tempip, ppid);
                hash_insert(hash_pid_ip, ppid, tempip, 0, NULL);
                printf("ip=[%s]\n",(char *)hash_getval(hash_pid_ip, ppid, 1));
                hash_insert(hash_ip_count, tempip, NULL, 1, NULL);

                sem_p(semid);
                msgid = open_msgqid((key_t)2222);
                msgrcv(msgid, &msg, sizeof(struct msgtype), 2, 0);
                msg.exitpid++;
                msg.mtype = 2;
                printf("nowtotalcount = %d\n", msg.exitpid);
                msgsnd(msgid, &msg, sizeof(struct msgtype), 0);
                sem_v(semid);
                
        }
    }
}

void clisession()
{
    char buf[1024] = "";
    int res, n;
    res = socketpair(PF_UNIX, SOCK_STREAM, 0, connsockpair);
    if (-1 == res)
        ERR_EXIT("socketpair");

    switch (fork())
    {
    case -1:
        ERR_EXIT("datafork");
        break;
    case 0:
        close(connsockpair[0]);
        while (1)
        {
            memset(buf, 0, sizeof(buf));
            read_line(connsockpair[1], buf, sizeof(buf));
            exec_com(buf);
        }
        break;
    default :
        close(connsockpair[1]);
        while (1)
        {
            memset(buf, 0, sizeof(buf));
            n = read_line(localfd, buf, sizeof(buf));
            if (0 == n)
                break;
            check_com(buf);
        }
    }
}

/*执行命令对应函数*/
int exec_com(char *buf)
{
    SESSION nowsession;
    char user[20] = "";
    char dir[200] = "";
    char *q = user;
    FUN p;
    int i;

    
    memset(&nowsession, 0, sizeof(nowsession));
    get_com(buf, nowsession.com);
    get_charm(buf, nowsession.charm);
    getcwd(nowsession.dir, sizeof(nowsession.dir));
    if (0 == strncmp(nowsession.com, "LIST", 4))
    {
        get_com(nowsession.charm, user);
        for (i = 0; i < strlen(user); i++)
        {
            *(q+i) = tolower(*(q+i));
        }
        get_charm(nowsession.charm, dir);
        strcpy(connuser, user);
        strcpy(nowsession.dir, dir);
    }
    printf("com=[%s]charm=[%s]dir=[%s]\n",
        nowsession.com, nowsession.charm, nowsession.dir);
/*    q=(FUN)hash_getval(hash_comm,nowsession.com,2);
    if(q!=NULL)
        q(&nowsession);
    else
    {
        write_loop(localfd,REPLY_UNKNOWN_COMM,strlen(REPLY_UNKNOWN_COMM));
        return -1;
    } */
    
    p = (FUN )hash_getval(hash_comm, nowsession.com, 2);
    if (NULL != p)
        p(&nowsession);
    else
        write_loop(localfd, "500 Unknown command!\r\n", strlen("500 Unknown command!\r\n"));
}

/*检查命令*/
int check_com(char *buf)
{
    char temp[1024] = "";
    char dir[1024] = "";
    if (0 == strncmp(buf, "LIST", 4)
        || 0 == strncmp(buf, "PORT", 4)
        || 0 == strncmp(buf, "PASV", 4)
        || 0 == strncmp(buf, "REST", 4)
        || 0 == strncmp(buf, "STOR", 4)
        || 0 == strncmp(buf, "RETR", 4))
    {
        if (0 == strncmp(buf, "LIST", 4))
        {
            getcwd(dir, sizeof(dir));
            sprintf(temp, "%s %s %s\r\n", "LIST", connuser, dir);
        }
        else
        {
            sprintf(temp, "%s", buf);
        }
        write_loop(connsockpair[0], temp, strlen(temp));
    }
    else
        exec_com(buf);
}