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

　　接上一小节，这次增加另外的两张表，用于记录用户是保存那些文件。增加传上来的文件的文件指纹，使用MD5表示。

　　两张表如下定义:

create table files(
fid int,
filename varchar(64),
md5 varchar(64)
);

create table relations(
uid int,
fid int
);

　　表与表之间的关系如下:

　　client.cpp 在上一小节基础上增加了一个md5的功能，传输给服务器，用于作为文件的唯一标识

  　　... 
 42 struct File
{
    int uid;
    char filename[64];
    char md5[64];
};
struct FileList
{
    char list[1024];
};

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 md5sum(char *filename,unsigned char *md5);
int file_pull(struct Addr addr,struct User user,char *filenames);


int main(int argc,char *argv[])
{
　　　　...
    return 0;
}

//验证成功时返回大于0的uid号码，错误返回-1
int check_login(struct Addr addr,struct User * user)
{
　　　　...
}

int file_push(struct Addr addr,struct User user,char *filenames)
{
　　　　　　... ...
    struct File file;//文件指纹
    int sockfd;
    FILE *fp;
    char md5[64];
    unsigned char md5tmp[64];

　　　　　　... ...
    //计算MD5
    memset(md5,0,sizeof(md5));
    md5sum(filenames,md5tmp);
    printf("计算得到的MD5:");
    for(int i=0;i<16;i++)
    {
        sprintf(&md5[i*2],"%02X",md5tmp[i]);
    }
    printf("%s\n",md5);

    //打开文件
    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);
    }
    //发送文件指纹
    strcpy(file.filename,filenames);
    strcpy(file.md5,md5);
    file.uid=user.uid;
    if(send(sockfd,(char *)&file,sizeof(struct File),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("传输完毕\n");
    fclose(fp);//关闭文件流
    close(sockfd);//关闭socket连接

    return 0;
}

int md5sum(char *filename,unsigned char *md5)
{
    MD5_CTX ctx;
    char buffer[1024];
    unsigned char outmd[16];
    int len=0;
    int i;
    FILE *fp=NULL;
    memset(outmd,0,sizeof(outmd));
    memset(buffer,0,sizeof(buffer));
    fp=fopen(filename,"rb");
    if(fp==NULL)
    {
        perror("打开文件失败");
    }
    MD5_Init(&ctx);
    while((len=fread(buffer,1,sizeof(buffer),fp))>0)
    {
        MD5_Update(&ctx,buffer,len);
        memset(buffer,0,sizeof(buffer));
    }
    MD5_Final(outmd,&ctx);
    for(i=0;i<16;i++)
    {
        md5[i]=outmd[i];
    }
    fclose(fp);
    return 0;
}

　　server.cpp 

 　　　... ...
struct File
{
    int uid;
    char filename[64];
    char md5[64];
};
struct FileList
{
    char list[1024];
};

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

int mysql_get_max_fid()
{
    MYSQL conn;
    MYSQL_RES *res_ptr;
    MYSQL_ROW result_row;
    int res;int row;int column;int mfid;
    char sql[256];
    strcpy(sql,"select max(fid) from files;");
    mfid=0;
    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;
                if(row<=1)
                {
                    ;//没有数据
                }
                else
                {
                    result_row=mysql_fetch_row(res_ptr);
                    printf("最大的fid是:%s\n",result_row[0]);
                    if(result_row[0]==NULL)
                        mfid=0;
                    else
                        mfid=atoi(result_row[0]);
                }
            }
            else
            {
                printf("没有查询到匹配的数据\n");
            }
        }
    }
    else
    {
        perror("Connect Failed!");
        exit(-1);
    }
    mysql_close(&conn);
    return mfid;
}

int mysql_file_in(struct File file)
{
    int mfid;
    MYSQL conn;
    int res;
    char sql[256];
    char tmp[32];
    mfid=mysql_get_max_fid()+1;
    printf("获取到的最大fid为:%d\n",mfid);
    mysql_init(&conn);
    if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
    {
        //insert into files values(mfid,file.filename,file.md5);
        //insert into files values(file.uid,mfid);
        memset(sql,0,sizeof(sql));
        strcpy(sql,"insert into files values(");
        sprintf(tmp,"%d",mfid);
        strcat(sql,tmp);
        strcat(sql,",'");
        strcat(sql,file.filename);
        strcat(sql,"','");
        strcat(sql,file.md5);
        strcat(sql,"');");
        printf("插入的sql语句: %s\n",sql);
        res=mysql_query(&conn,sql);
        if(res)
            printf("Insert Error!\n");
        else
            printf("Insert Success!\n");

        memset(sql,0,sizeof(sql));
        strcpy(sql,"insert into relations values(");
        sprintf(tmp,"%d",file.uid);
        strcat(sql,tmp);
        strcat(sql,",");
        sprintf(tmp,"%d",mfid);
        strcat(sql,tmp);
        strcat(sql,");");
        printf("插入的sql语句: %s\n",sql);
        res=mysql_query(&conn,sql);
        if(res)
            printf("Insert Error!\n");
        else
            printf("Insert Success!\n");
    }
    else
    {
        perror("Connect Failed!");
        exit(-1);
    }
    return 0;
}


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

        //来一个连接就创建一个进程进行处理
        pid=fork();
        if(pid<0)
        {
            perror("fork error");
        }
        else if(pid==0)
        {
            recv(clientfd,(char *)&control,sizeof(struct Control),0);
            printf("用户 %d 使用命令 %d\n",control.uid,control.control);
            switch(control.control)
            {
                case USER_CHECK_LOGIN:
                    {
                        //身份验证处理
　　　　　　　　　　　　　　　　... ...
                        break;
                    }
                case FILE_PUSH:
                    {
                        char buffer[BUFFER_SIZE];
                        int data_len;
                        FILE * fp=NULL;
                        struct File file;
                        //获取文件指纹
                        recv(clientfd,(char *)&file,sizeof(struct File),0);
                        printf("获取到的用户名ID: %d 文件名:%s  MD5:%s\n",file.uid,file.filename,file.md5);
                        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(size>0)
                        {
                            printf("\n%s:%d的文件传送完毕\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                            //如果文件传输成功那么就可以写入数据库了
                            mysql_file_in(file);
                        }
                        else
                            printf("\n%s:%d的文件传送失败\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));
                        fclose(fp);
                        rename("data",file.md5);//这里可以修改文件的名字
                        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)
{
　　　　... ...
}

　　　　... ...

 　　上面已经介绍了如何计算MD5值并写入数据库，接下来要做的事就是如何判断要上传的文件是否在服务器中已经存在，如果存在就可以不用上传，而是在relations表中记录，写入uid-fid关系，这样就可以实现网盘的秒传功能了。而如何区分文件是否相同，我这里使用的是以MD5作为文件指纹。话说秒传功能也不过如此吧。所以现在应该知道为什么有的文件可以秒传，有的不可以了吧。关于具体的解释可以参考杜鑫先生在知乎中的回到。传送门在第一小节中有。

　　已加入验证和秒传功能的网盘程序

　　client.cpp修改如下

 　　　　... ... 
int file_push(struct Addr addr,struct User user,char *filenames)
{
　　　　　　... ...
    //发送文件指纹
    strcpy(file.filename,filenames);
    strcpy(file.md5,md5);
    file.uid=user.uid;
    if(send(sockfd,(char *)&file,sizeof(struct File),0)<0)
    {
        perror("文件指纹发送失败");
        exit(-1);
    }
    char ch[64];
    if(recv(sockfd,ch,64,0)<0)
    {
        perror("error");
    }
    if(ch[0]=='y')//表示已经存在
    {
        printf("该文件在服务器中存在，正使用秒传功能。\n");
        printf("传输完毕\n");
        return 0;
    }
    //打开文件
    if((fp=fopen(filenames,"rb"))==NULL)
    {
        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("传输完毕\n");
    fclose(fp);//关闭文件流
    close(sockfd);//关闭socket连接

    return 0;
}

　　　　... ...

　　server.cpp修改如下

  　　　　......
int main(int argc,char *argv[])
{
　　　　　　......
        //来一个连接就创建一个进程进行处理
        pid=fork();
        if(pid<0)
        {
            perror("fork error");
        }
        else if(pid==0)
        {
            recv(clientfd,(char *)&control,sizeof(struct Control),0);
            printf("用户 %d 使用命令 %d\n",control.uid,control.control);
            switch(control.control)
            {
                case USER_CHECK_LOGIN:
　　　　　　　　　　　　　　... ...
                case FILE_PUSH:
                    {
                        char buffer[BUFFER_SIZE];
                        int data_len;
                        FILE * fp=NULL;
                        struct File file;
                        //获取文件指纹
                        recv(clientfd,(char *)&file,sizeof(struct File),0);
                        printf("获取到的用户名ID: %d 文件名:%s  MD5:%s\n",file.uid,file.filename,file.md5);
                        //对文件进行验证，如果文件已经存在就不用进行接收了
                        int t=mysql_check_md5(file);
                        char ch[64]={0};
                        printf("t=%d\n",t);
                        if(t!=0)
                        {
                            printf("该文件存在,使用秒传功能\n");
                            strcpy(ch,"yes");
                            send(clientfd,ch,64,0);
                            mysql_file_in(file.uid,t);
                            continue;
                        }
                        strcpy(ch,"no");
                        send(clientfd,ch,64,0);
                        printf("md5验证后得到的fid:%d\n",t);
                        bzero(buffer,BUFFER_SIZE);
　　　　　　　　　　　　　　... ... ...
                        break;
                    }
　　　　　　　　　　　　　　... ...
            }
            close(clientfd);//短连接结束
            exit(0);//退出子进程
        }
    }

    return 0;
}


//函数定义
int mysql_check_md5(struct File file)
{
    MYSQL conn;
    MYSQL_RES * res_ptr;
    MYSQL_ROW result_row;
    int res;int row;int column;int value=0;
    char sql[256]={0};
    strcpy(sql,"select fid from files where md5='");
    strcat(sql,file.md5);
    strcat(sql,"';");
    printf("查询的sql:%s\n",sql);

    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;
                if(row<=1)
                {
                    ;
                }
                else
                {
                    result_row=mysql_fetch_row(res_ptr);
                    value=atoi(result_row[0]);
                }
            }
            else
            {
                printf("没有查询到匹配的数据\n");
            }
        }
    }
    else
    {
        perror("Connect Failed!");
        exit(-1);
    }
    mysql_close(&conn);
    return value;//返回fid
}

int mysql_file_in(int uid,int fid)
{
    MYSQL conn;
    int res;
    char sql[256];
    char tmp[32];
    mysql_init(&conn);
    if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
    {
        //insert into files values(uid,fid);
        memset(sql,0,sizeof(sql));
        strcpy(sql,"insert into relations values(");
        sprintf(tmp,"%d",uid);
        strcat(sql,tmp);
        strcat(sql,",");
        sprintf(tmp,"%d",fid);
        strcat(sql,tmp);
        strcat(sql,");");
        printf("插入的sql语句: %s\n",sql);
        res=mysql_query(&conn,sql);
        if(res)
            printf("Insert Error!\n");
        else
            printf("Insert Success!\n");
    }
    else
    {
        perror("Connect Failed!");
        exit(-1);
    }
    return 0;
}

　　　　... ...

 　　运行时的截图

 　　由上图可以看出如果该文件在服务器中存在的话，那么下一次上传同一个文件的话就会跳过上传的步骤，而是把数据库中的标识号给用户uid即可。具体关系可以看下面数据库数据。

　　好了，现在的上传功能已经很完善了。下一节将实现下载功能了。

 

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