改写文件上传支持断点续传-后端C代码

后端代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#include "qsi.h"
#include "cJSON.h"

#define UPLOAD_LOG_MODULE "cgi"
#define UPLOAD_LOG_PROC   "upload"

#define TEMP_BUF_MAX_LEN 51200
#define FILE_NAME_LEN 256
#define USER_NAME_LEN 256

#define THREAD_COUNT 20  
static int counts[THREAD_COUNT];

/*
    功能:获取已经上传过后的文件列表,(文件名,MD5值)
    参数:
        filePath    文件路径
        folderPath  文件夹路径
    返回值:成功0 失败返回-1
*/
int getChunkList(char *filePath, char *folderPath, FCGX_Request request)
{
    /*
        将目录下的所用文件组成一个文件列表,返回到前端 
    */
    char fullpath[256];
    memset(fullpath, 0x00, 256);
    sprintf(fullpath, "%s/%s", filePath, folderPath);
        
    //创建一个文件jSON对象
    cJSON * allfile = cJSON_CreateArray();    
    cJSON * fullnode= cJSON_CreateObject();

    char * out = NULL;
    if(NULL == fullpath)
    {       
        FCGX_FPrintF(request.out,"%s", "enter path null!\n");
        return -1;
    }

    DIR *dirp = opendir(fullpath);  //打开目录
    if(dirp == NULL)
    {
        FCGX_FPrintF(request.out,"%s", "open dir err");
        return -1;
    }
        
    struct dirent * dentp = NULL;
    
    //读取目录下文件
    while((dentp = readdir(dirp)))
    {           
        if((strcmp(".", dentp->d_name)) == 0 || (strcmp("..", dentp->d_name)) == 0)
        {
            continue;
        }
        
        //不是目录文件
        if(dentp->d_type == DT_REG)
        {           
            struct stat filestruct;           
            char newfilepath[256];
            sprintf(newfilepath, "%s/%s", fullpath, dentp->d_name);
            stat(newfilepath, &filestruct);
            
            //创建一个jSON对象
            cJSON *chunkList = NULL;            
            if(strcmp(folderPath,dentp->d_name ) == 0)
            {
                chunkList = cJSON_CreateObject();
                cJSON_AddStringToObject(chunkList, "filename", dentp->d_name);
                FCGX_FPrintF(request.out,"%s", cJSON_Print(chunkList));
                return 0;
            }
            else
            {
                cJSON_AddItemToArray(allfile, chunkList = cJSON_CreateObject());
                cJSON_AddStringToObject(chunkList, "filename", dentp->d_name); 
                cJSON_AddNumberToObject(chunkList, "indexof", atoi(dentp->d_name));                 
            }                                                                                     
        }   
        else
        {
            return 0;               
        }                  
    }   
    cJSON_AddItemToObject(fullnode, "chunkList", allfile);          
    FCGX_FPrintF(request.out,"%s", cJSON_Print(fullnode));
    return 0;   
}

/*
    功能:判断文件夹是否存在,不存在创建一个文件夹
    
    参数:
        filepath   文件夹路径
        filename   文件夹名称
    
    返回值:成功返回0 失败返回-1
*/
int folderIsExit(char *filepath, char *filename, FCGX_Request request)
{       
    char fullpath[256];
    memset(fullpath, 0x00, sizeof(fullpath));
    
    sprintf(fullpath, "%s/%s", filepath, filename);
    
    if(access(fullpath, W_OK) == 0)
    {       
        //目录存在且有写的权限    遍历目录下所有文件组织文件链表
        getChunkList(filepath, filename,request);               
    }
    else
    {
        //目录不存在或没有写的权限  
         mkdir(fullpath, S_IRWXU|S_IRWXG|S_IRWXO); //0777
         cJSON *fileinfo = cJSON_CreateObject();
         cJSON_AddStringToObject(fileinfo, "fileName", filename);       
         FCGX_FPrintF(request.out,"%s", cJSON_Print(fileinfo));
    }   
    return 0;   
}

/*
    功能:在指定长度的字符串中查找字符串
    参数:
        (in)    full_data            查找字符串
        (in)    full_data_len        查找长度长度
        (in)    substr               要查找的字符串
    返回值:成功返回要查找的字符串的初始位置 失败返回NULL   
*/
char* memstr(char* full_data, int full_data_len, char* substr)
{
    if (full_data == NULL || full_data_len <= 0 || substr == NULL)
    {
        return NULL;
    }

    if (*substr == '\0')
    {
        return NULL;
    }

    int sublen = strlen(substr);
    char* cur = full_data;
    int last_possible = full_data_len - sublen + 1;
    int i = 0;
    for (i = 0; i < last_possible; i++)
    {
        if (*cur == *substr)
        {
            if (memcmp(cur, substr, sublen) == 0)
            {
                // found
                return cur;
            }

        }
        cur++;
    }
    return NULL;
}


/**
 * @brief  解析上传的post数据 保存到本地临时路径
 *         同时得到文件上传者、文件名称、文件大小
 *
 * @param len       (in)    post数据的长度
 * @param user      (out)   文件上传者
 * @param file_name (out)   文件的文件名
 * @param p_size    (out)   文件大小
 *
 * @returns
 *          0 succ, -1 fail
 */
int recv_save_file(long len, char *user, char *filename, long *p_size, FCGX_Request request, char *filepath)
{   
    char log[526] = {0};
    memset(log,0x00, 526);
    int ret = 0;
    char *file_buf = NULL;
    char *begin = NULL;
    char *p, *q, *k;
    
     int fd = 0;
    int ret2 = 0;
    
    char *heto= NULL;
    char *to = NULL;
         
    int totallen = 0;
    char * toen = NULL;
    
    int num = 0;
     
    int FragmentName = 0;   
    char FragmentNum[50];
    
    char FPATH[256] = {0};
    
    char content_text[TEMP_BUF_MAX_LEN] = {0};      //文件头部信息
    char boundary[TEMP_BUF_MAX_LEN] = {0};          //分界线信息

    //==========> 开辟存放文件的 内存 <===========
    file_buf = (char *)malloc(len);
    memset(file_buf,0x00, sizeof(file_buf));
    
    if (file_buf == NULL)
    {     
        goto END;
    }
        
      for(int i=0; i 开始处理前端发送过来的post数据格式 <============
    begin = file_buf;    //内存起点
    p = begin;

    p = strstr(begin, "\r\n");
    if (p == NULL)
    {
        ret = -1;
        goto END;
    }
        
    //拷贝分界线
    strncpy(boundary, begin, p-begin);
    boundary[p-begin] = '\0';   //字符串结束符
       
    p += 2;
    //已经处理了p-begin的长度
    len -= (p-begin);
    begin = p;

    p = strstr(begin, "\r\n");
    if(p == NULL)
    {      
        ret = -1;
       goto END;
    }
    strncpy(content_text, begin, p-begin);              //拷贝Content-Disposition这一行内容
    content_text[p-begin] = '\0';
   
    p += 2;
    len -= (p-begin);

    q = begin;
    q = strstr(begin, "name=");
    
    q += strlen("name=");
    q++;    //跳过第一个"

    k = strchr(q, '"');
    strncpy(user, q, k-q);  //拷贝用户名
    user[k-q] = '\0';
    
    begin = k;
    q = begin;
    q = strstr(begin, "filename=");

    q += strlen("filename=");
    q++;    //跳过第一个"

    k = strchr(q, '"');
    strncpy(filename, q, k-q);  //拷贝文件名
    filename[k-q] = '\0';
        
    begin = p;
    p = strstr(begin, "\r\n");
    p += 4;//\r\n\r\n
    len -= (p-begin);

    //下面才是文件的真正内容 
    begin = p;
    
    //find file's end
    p = memstr(begin, len, boundary);//util_cgi.h, 找文件结尾
    if (p == NULL)
    {
        ret = -1;
        goto END;
    }
    else
    {
        p = p - 2;//\r\n
    }
    
    //解析获的文件编号 获取总片数
     heto = p + 2;
     to = NULL;
     to = strstr(heto, "\r\n");
     to += 4;
     
     totallen = 0;
     toen = strstr(to, "\r\n");
         
     toen += 4;
     to = toen;
     toen = strstr(to, "\r\n");
     
     totallen = toen - to;
         
     memset(FragmentNum, 0x00, sizeof(FragmentNum));
     strncpy(FragmentNum, to, toen - to);               //获取片段总数
    
     num = atoi(FragmentNum);
     toen+= 2;
     to = toen;
     
     toen = strstr(to, "\r\n");
     toen += 2;
     
     to = toen;
     toen = strstr(to, "\r\n");
     
     toen += 4;
     to = toen;
     toen = strstr(to, "\r\n");
         
     memset(FragmentNum, 0x00, sizeof(FragmentNum));
     strncpy(FragmentNum, to, toen - to);  
     FragmentName = atoi(FragmentNum);
         
     sprintf(FPATH, "%s/%s", filepath, FragmentNum);
          
      fd = open(FPATH, O_CREAT|O_WRONLY, 0644);
      
      if (fd < 0)
      {
          ret = -1;
          goto END;
      }
    
      //ftruncate会将参数fd指定的文件大小改为参数length指定的大小
      ftruncate(fd, (p-begin));
      write(fd, begin, (p-begin));
      close(fd);
    
END:
    free(file_buf);
    FCGX_FFlush(request.in);  //刷新标准输入缓冲区
    return ret;
}

/*
    功能:获取文件大小
    参数:
        path  文件路径
    返回值:成功返回文件大小 失败返回-1
*/
int getfilesize(char *path)
{
    FILE *pf = fopen(path, "rb");
    if (pf==NULL)
    {
        return -1;
    } 
    else
    {
        fseek(pf, 0, SEEK_END);
        int size = ftell(pf);
        fclose(pf);
        return size;//返回文件大小
    }
}

/*
    功能:将多个文件合并为一个文件, 同时删除片段文件
    参数:
        part_f_path  片段文件路径
        filepath     合并后文件路径
        filename     合并后文件名
    返回值: 成功返回 0  失败返回-1

*/
int  merge(char *part_f_path , int part_num, char *filepath, char *filename, FCGX_Request request)
{
    char newpath[part_num][600]; //代表路径 
    for(int i=0; i

这个是依赖于nginx、Fastcgi的程序,还需要CJson库,供参考。
以上的解析post数据特定的,仅供参考。

你可能感兴趣的:(改写文件上传支持断点续传-后端C代码)