Server


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <config.h>
#include <fcntl.h>

int get_cmd_type(char *cmd)
{
        if(! strcmp("ls",cmd))        return LS;
        if(! strcmp("quit",cmd))      return QUIT;
        if(! strcmp("pwd",cmd))       return PWD;

        if(strstr(cmd,"cd") != NULL)  return CD;
        if(strstr(cmd,"get") != NULL) return GET;
        if(strstr(cmd,"put") != NULL) return PUT;

        return 100;
}

char *getDesDir(char *cmsg)
{
        char *p = NULL;

        p = strtok(cmsg," ");
        p = strtok(NULL," ");

        return p;
}

void msg_handler(struct Msg msg,int fd)
{
        char dataBuf[1024] = {0};
        char *file = NULL;
        int fdfile;

        printf("cmd : %s\n",msg.data);
        int ret = get_cmd_type(msg.data);

        switch(ret){
                case LS:
                case PWD:
                        msg.type = 0;

                        FILE *r = popen(msg.data,"r");

                        fread(msg.data,sizeof(msg.data),1,r);

                        write(fd,&msg,sizeof(msg));
                        break;

                case CD:
                        msg.type = 1;

                        char *dir = getDesDir(msg.data);

                        printf("dir : %s\n",dir);

                        chdir(dir);
                        break;

                case GET:
                        file = getDesDir(msg.data);

                        if(access(file,F_OK) == -1){
                                strcpy(msg.data,"No This File !");

                                write(fd,&msg,sizeof(msg));
                        }
                        else{
                                msg.type = DOFILE;

                                fdfile = open(file,O_RDWR);

                                read(fdfile,dataBuf,sizeof(dataBuf));

                                close(fdfile);

                                strcpy(msg.data,dataBuf);

                                write(fd,&msg,sizeof(msg));
                        }
                        break;

                case PUT:
                        fdfile = open(getDesDir(msg.data),O_RDWR | O_CREAT,0666);

                        write(fdfile,msg.secondBuf,strlen(msg.secondBuf));

                        close(fdfile);
                        break;

                case QUIT:
                        print("client quit\n");
                        exit(-1);
        }
}

int main(int argc,char **argv)
{
        int s_fd;
        int c_fd;
        int n_read;
        char readBuf[128] = {0};

        struct sockaddr_in s_addr;
        struct sockaddr_in c_addr;
        struct Msg msg;

        if(argc != 3){
                printf("param is not good\n");
                exit(-1);
        }

        memset(&s_addr,0,sizeof(struct sockaddr_in));
        memset(&c_addr,0,sizeof(struct sockaddr_in));

        //1. socket
        s_fd = socket(AF_INET,SOCK_STREAM,0);
        if(s_fd == -1){
                perror("socket");
                exit(-1);
        }

        s_addr.sin_family = AF_INET;
        s_addr.sin_port = htons(atoi(argv[2]));
        inet_aton(argv[1],&s_addr.sin_addr);

        //2. bind
        bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr_in));

        //3. listen
        listen(s_fd,9);

        //4. accept
        int clen = sizeof(struct sockaddr_in);


        while(1){
                c_fd = accept(s_fd,(struct sockaddr *)&c_addr,&clen);
                if(c_fd == -1){
                        perror("accept");
                }

                printf("get connect : %s\n",inet_ntoa(c_addr.sin_addr));

                if(fork() == 0){
                        while(1){
                                memset(msg.data,0,sizeof(msg.data));

                                n_read = read(c_fd,&msg,sizeof(msg));
                                if(n_read == 0){
                                        printf("client out\n");
                                        break;
                                }
                                else if(n_read > 0){
                                        msg_handler(msg,c_fd);
                                }
                        }
                }
        }

        close(c_fd);
        close(s_fd);

        return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值