Socket网络编程--小小网盘程序(2)

　　这一节将不会介绍太多的技术的问题，这节主要是搭建一个小小的框架，为了方便接下来的继续编写扩展程序。本次会在上一小节的基础上加上一个身份验证的功能。

　　因为网盘程序不像聊天程序，网盘是属于主动向服务器拉取信息，而聊天程序有可能要被动获取信息，所以为了减轻服务器压力，每次要向服务器获取服务就建立一个短连接，而不像聊天程序一样的长连接，微信的公众平台，输入指令获取服务，就是这个样子了。具体看一下代码就知道了。

　　还有为了方便处理，我增加了一个控制信号，这个控制信号在以前的聊天程序中讲到过，但是当时为了简单就没有实现，现在就简单实现一些，方便以后可以参考。

　　还有就是这次的网盘程序我将尽量实现下面的这些功能:

　　(1) file push filename   　　　　　　　　//用户上传文件到服务器上

　　(2) file pull filename    　　　　　　　　//用户下载文件到本地上

　　(3) file list　　　　　　　　　　　　　　//列出用户在服务器中的所有文件

　　(4) file sendto  filename　username　//共享文件给其他用户

　　(5) file delete filename　　　　　　　　//删除服务器中的文件

　　(6) 用户登录，自动完成

　　本次数据库名为filetranslate,目前有表user

create table user
(
    uid int;
    username varchar(64);
    password varchar(64);  
);

　　修改后的client.cpp

#include <netinet/in.h> // sockaddr_in
#include <sys/types.h> //socket
#include <sys/socket.h> //socket
#include <netdb.h> //gethostbyname
#include <unistd.h> //close
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <arpa/inet.h> //inet_addr

#define SERVVER_PORT 12138
#define LISTEN_QUEUE 20
#define BUFFER_SIZE 1024

//传输控制信号宏定义
//struct Control中control的取值
#define USER_CHECK_LOGIN 1
#define FILE_PUSH 2
#define FILE_PULL 3
#define FILE_LIST 4
#define FILE_SENDTO 5
#define FILE_DELECT 6

struct Addr
{
    char host[64];
    int port;
};
struct User
{
    int uid;
    char username[64];
    char password[64];
};
struct Control
{
    int uid;
    int control;
};

void print_time(char *ch);//打印时间
int file_push(struct Addr addr,struct User user,char *filenames);
int check_login(struct Addr addr,struct User * user);


int main(int argc,char *argv[])
{
    char orderbuf[BUFFER_SIZE];
    struct Addr addr;
    char arg1[32],arg2[32],arg3[32],arg4[32];
    struct User user;

    if(argc!=5)
    {
        perror("usage: ./client [serverhost] [serverport] [username] [password]");
        exit(-1);
    }
    strcpy(addr.host,argv[1]);
    addr.port=atoi(argv[2]);
    strcpy(user.username,argv[3]);
    strcpy(user.password,argv[4]);
    int uid=check_login(addr,&user);
    if(uid<=0)
    {
        perror("用户验证失败");
        exit(-1);
    }
    printf("验证登陆成功
");
    while(1)
    {
        printf("
请输入操作指令:");
        fgets(orderbuf,BUFFER_SIZE,stdin);
        memset(arg1,0,sizeof(arg1));
        memset(arg2,0,sizeof(arg2));
        memset(arg3,0,sizeof(arg3));
        memset(arg4,0,sizeof(arg4));
        sscanf(orderbuf,"%s%s%s%s",arg1,arg2,arg3,arg4);
        if(strcmp(arg1,"file")==0)
        {
            if(strcmp(arg2,"push")==0)
            {
                strcpy(orderbuf,arg3);
                file_push(addr,user,orderbuf);
            }
            else if(strcmp(arg2,"pull")==0)
            {
                ;
            }
            else if(strcmp(arg2,"list")==0)
            {
                ;
            }
            else if(strcmp(arg2,"sendto")==0)
            {
                ;
            }
            else if(strcmp(arg2,"delect")==0)
            {
                ;
            }
            else
            {
                printf("该命令不支持
");
            }
        }
        else
        {
            printf("该命令不支持
");
        }
    }


    return 0;
}

//验证成功时返回大于0的uid号码，错误返回-1
int check_login(struct Addr addr,struct User * user)
{
    struct sockaddr_in servAddr;
    struct hostent * host;
    struct Control control;
    int sockfd;

    host=gethostbyname(addr.host);
    servAddr.sin_family=AF_INET;
    servAddr.sin_addr=*((struct in_addr *)host->h_addr);
    //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    servAddr.sin_port=htons(addr.port);
    if(host==NULL)
    {
        perror("获取IP地址失败");
        exit(-1);
    }
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket创建失败");
        exit(-1);
    }
    if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
    {
        perror("connect 失败");
        exit(-1);
    }

    control.control=USER_CHECK_LOGIN;
    control.uid=0;
    if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)
    {
        perror("发送数据失败");
        exit(-1);
    }

    if(send(sockfd,(char *)user,sizeof(struct User),0)<0)
    {
        perror("发送数据失败");
        exit(-1);
    }
    bzero(user,sizeof(struct User));
    if(recv(sockfd,(char *)user,sizeof(struct User),0)<0)
    {
        perror("接收数据失败");
        exit(-1);
    }
    printf("获取后用户名:%s 密码:%s ID号:%d
",user->username,user->password,user->uid);

    close(sockfd);//关闭socket连接
    return user->uid;
}

int file_push(struct Addr addr,struct User user,char *filenames)
{
    struct sockaddr_in servAddr;
    struct hostent * host;
    struct Control control;
    int sockfd;
    FILE *fp;

    host=gethostbyname(addr.host);
    servAddr.sin_family=AF_INET;
    servAddr.sin_addr=*((struct in_addr *)host->h_addr);
    //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1");
    servAddr.sin_port=htons(addr.port);
    if(host==NULL)
    {
        perror("获取IP地址失败");
        exit(-1);
    }
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
    {
        perror("socket创建失败");
        exit(-1);
    }

    if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
    {
        perror("connect 失败");
        exit(-1);
    }

    //打开文件
    if((fp=fopen(filenames,"rb"))==NULL)
    {
        perror("文件打开失败");
        exit(-1);
    }
    //这里传输控制信号
    control.control=FILE_PUSH;
    control.uid=user.uid;
    if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)
    {
        perror("控制信号发送失败");
        exit(-1);
    }
    char buffer[BUFFER_SIZE];
    bzero(buffer,BUFFER_SIZE);
    printf("正在传输文件");
    int len=0;
    //不断的读取文件直到文件结束
    while((len=fread(buffer,1,BUFFER_SIZE,fp))>0)
    {
        if(send(sockfd,buffer,len,0)<0)
        {
            perror("发送数据失败");
            exit(-1);
        }
        bzero(buffer,BUFFER_SIZE);
        printf(".");//1K打印一个点//如果要实现百分比，就要计算文件大小，然后再处理即可
    }

    printf("传输完毕
");
    fclose(fp);//关闭文件流
    close(sockfd);//关闭socket连接

    return 0;
}

void print_time(char *ch)
{
    time_t now;
    struct tm * stm;
    time(&now);
    stm=localtime(&now);
    sprintf(ch,"%02d:%02d:%02d
",stm->tm_hour,stm->tm_min,stm->tm_sec);
    return ;
}

　　下面这个是server.cpp

#include <netinet/in.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <netdb.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>
#include <arpa/inet.h> //inet_ntoa
#include <mysql.h>

#define SERVER_PORT 12138
#define LISTEN_QUEUE 20
#define BACKLOG 200
#define BUFFER_SIZE 1024

//传输控制信号宏定义
//struct Control中control的取值
#define USER_CHECK_LOGIN 1
#define FILE_PUSH 2
#define FILE_PULL 3
#define FILE_LIST 4
#define FILE_SENDTO 5
#define FILE_DELECT 6

struct User
{
    int uid;
    char username[64];
    char password[64];
};

struct Control
{
    int uid;
    int control;
};

void print_time(char *ch);//打印时间
int MAX(int a,int b);
int mysql_check_login(struct User user);


int main(int argc,char *argv[])
{
    struct sockaddr_in server_addr;
    struct sockaddr_in client_addr;
    struct User user;
    struct Control control;
    char ch[64];
    int clientfd;
    pid_t pid;
    socklen_t length;
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family=AF_INET;
    server_addr.sin_addr.s_addr=htons(INADDR_ANY);
    server_addr.sin_port=htons(SERVER_PORT);

    //创建套接字
    int sockfd=socket(AF_INET,SOCK_STREAM,0);
    if(sockfd<0)
    {
        perror("创建套接字失败");
        exit(-1);
    }

    if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr))==-1)
    {
        perror("bind 失败");
        exit(-1);
    }

    if(listen(sockfd,LISTEN_QUEUE))
    {
        perror("listen 失败");
        exit(-1);
    }

    length=sizeof(struct sockaddr);

    while(1)
    {
        clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&length);
        if(clientfd==-1)
        {
            perror("accept 失败");
            continue;
        }
        printf(">>>>>%s:%d 连接成功，当前所在的ID(fd)号: %d 
",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),clientfd);
        print_time(ch);
        printf("加入的时间是:%s
",ch);

        //来一个连接就创建一个进程进行处理
        pid=fork();
        if(pid<0)
        {
            perror("fork error");
        }
        else if(pid==0)
        {
            recv(clientfd,(char *)&control,sizeof(struct Control),0);
            printf("用户 %d 使用命令 %d
",control.uid,control.control);
            switch(control.control)
            {
                case USER_CHECK_LOGIN:
                    {
                        //身份验证处理
                        recv(clientfd,(char *)&user,sizeof(struct User),0);
                        printf("客户端发送过来的用户名是:%s,密码:%s
",user.username,user.password);
                        if((user.uid=mysql_check_login(user))>0)
                        {
                            printf("验证成功
");
                        }
                        else
                        {
                            printf("验证失败
");
                        }
                        send(clientfd,(char *)&user,sizeof(struct User),0);
                        break;
                    }
                case FILE_PUSH:
                    {
                        char buffer[BUFFER_SIZE];
                        int data_len;
                        FILE * fp=NULL;
                        bzero(buffer,BUFFER_SIZE);
                        if((fp=fopen("data","wb"))==NULL)
                        {
                            perror("文件打开失败");
                            exit(-1);
                        }
                        //循环接收数据
                        int size=0;//表示有多少个块
                        while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0))
                        {
                            if(data_len<0)
                            {
                                perror("接收数据错误");
                                exit(-1);
                            }
                            size++;
                            if(size==1)
                                printf("正在接收来自%s:%d的文件
",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                            else
                                printf(".");
                            //向文件中写入
                            int write_len=fwrite(buffer,sizeof(char),data_len,fp);
                            if(write_len>data_len)
                            {
                                perror("写入数据错误");
                                exit(-1);
                            }
                            bzero(buffer,BUFFER_SIZE);
                        }
                        if(size>0)
                            printf("
%s:%d的文件传送完毕
",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                        else
                            printf("
%s:%d的文件传送失败
",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                        fclose(fp);
                        //rename("data","asdf");//这里可以修改文件的名字
                        exit(0);
                        break;
                    }
                case FILE_PULL:
                    {
                        break;
                    }
                case FILE_LIST:
                    {
                        break;
                    }
                case FILE_DELECT:
                    {
                        break;
                    }
                default:
                    {
                        break;
                    }
            }
            close(clientfd);//短连接结束
            exit(0);//退出子进程
        }
    }

    return 0;
}


//函数定义
int mysql_check_login(struct User user)
{
    MYSQL conn;
    MYSQL_RES *res_ptr;
    MYSQL_ROW result_row;
    int res;
    int row;
    int column;
    int uid;
    char sql[256]={0};
    strcpy(sql,"select uid from users where username="");
    strcat(sql,user.username);
    strcat(sql,"" and password="");
    strcat(sql,user.password);
    strcat(sql,"";");
    printf("查询的sql:%s
",sql);
    uid=-1;
    mysql_init(&conn);
    if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
    {
        res=mysql_query(&conn,sql);
        if(res)
        {
            perror("Select SQL ERROR!");
            exit(-1);
        }
        else
        {
            res_ptr=mysql_store_result(&conn);
            if(res_ptr)
            {
                column=mysql_num_fields(res_ptr);//获取列数
                row=mysql_num_rows(res_ptr)+1;//获取行数，加1表示还有第一行字段名
                if(row<=1)
                {
                    ;//验证失败
                }
                else
                {
                    //这里先假定username是唯一
                    result_row=mysql_fetch_row(res_ptr);
                    printf("获取到的uid是:%s
",result_row[0]);
                    uid=atoi(result_row[0]);
                }
            }
            else
            {
                printf("没有查询到匹配的数据
");
            }
        }
    }
    else
    {
        perror("Connect Failed!
");
        exit(-1);
    }
    mysql_close(&conn);
    return uid;
}

void print_time(char *ch)
{
    time_t now;
    struct tm * stm;
    time(&now);
    stm=localtime(&now);
    sprintf(ch,"%02d:%02d:%02d
",stm->tm_hour,stm->tm_min,stm->tm_sec);
    return ;
}

int MAX(int a,int b)
{
    if(a>b)
        return a;
    return b;
}

　　下面给张运行时的截图

　　关于c语言调用mysql: http://www.cnblogs.com/wunaozai/p/3876134.html

　　本文地址: http://www.cnblogs.com/wunaozai/p/3887728.html