cur 以post的方式进行http上传文件(自定义http head)

最近在项目中需要在嵌入式终端上传录音文件，最后选择了curl来支持该项目，网上现有的一些资源比较简单，并没能满足需求，自己结合curl的example做了改进，具备一下功能：

（1）以post的方式上传文件，并对http head进行了自定义；

（2）文件名封装在附加的文件头里面。

该代码已经测试通过。

/*************************************************************************** *                                  _   _ ____  _ *  Project                     ___| | | |  _ \| | *                             / __| | | | |_) | | *                            | (__| |_| |  _ <| |___ *                             \___|\___/|_| \_\_____| * * Copyright (C) 1998 - 2017, Daniel Stenberg, <[e-mail]>, et al. * * This software is licensed as described in the file COPYING, which * you should have received as part of this distribution. The terms * are also available at https://curl.haxx.se/docs/copyright.html. * * You may opt to use, copy, modify, merge, publish, distribute and/or sell * copies of the Software, and permit persons to whom the Software is * furnished to do so, under the terms of the COPYING file. * * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY * KIND, either express or implied. * ***************************************************************************//* <DESC> * Issue an HTTP POST and provide the data through the read callback. * </DESC> */#include <stdio.h>#include <string.h>#include <curl/curl.h>typedef enum file_part{    FILE_HEAD_PART = 0,    FILE_CONTENT_PART,    FILE_END_PART,    FILE_TRANSFER_FINISH,}E_FilePart;typedef struct http_upload{    FILE* rFile;                // 上传的文件    size_t fileSizeLeft;        // 文件剩余字节    size_t filePos;             // 文件内容位置    char *headerPtr;            // 附加的文件起始头，主要用于(1)告知服务器上传文件的名字;(2)一些自定义参数    size_t  headerSizeLeft;     // 附加的文件起始头剩余字节    char *endPtr;               // 附加的文件结束尾    size_t  endSizeLeft;        // 附加的文件结束尾剩余字节    E_FilePart filePart;        // 0 传输HeaderInfo, 1 传输文件， 2 传输EndInfo, -1则结束    size_t totalSize;           // 总共传输的字节数，附加的文件起始头+文件+附加的文件结束尾}T_HttpUpload;#define HTTP_HEAD_MAX_SIZE  (256)#define HTTP_END_MAX_SIZE   (1024)const char BOUNDARY[] = {"---WebKitFormBoundary"};              // 自定义的附加文件起始头static char sHeaderInfo[HTTP_HEAD_MAX_SIZE];static char sEndInfo[HTTP_END_MAX_SIZE];static size_t read_callback(void *dest, size_t size, size_t nmemb, void *userp){    T_HttpUpload *httpUpload = (T_HttpUpload *)userp;    size_t bufferSize = size*nmemb;    size_t copyThisMuch;    printf("file left %zu, %zu, %zu, size*nmemb = %zu, filePart = %d, totalSize = %zu\n",     httpUpload->headerSizeLeft, httpUpload->fileSizeLeft,    httpUpload->endSizeLeft, bufferSize, httpUpload->filePart,    httpUpload->totalSize);    if(FILE_HEAD_PART == httpUpload->filePart)    {        copyThisMuch = httpUpload->headerSizeLeft;        if(copyThisMuch > bufferSize)            copyThisMuch = bufferSize;        memcpy(dest, httpUpload->headerPtr, copyThisMuch);        httpUpload->headerPtr += copyThisMuch;        httpUpload->headerSizeLeft -= copyThisMuch;        if(httpUpload->headerSizeLeft <= 0)        {            httpUpload->filePart = FILE_CONTENT_PART;        }        httpUpload->totalSize += copyThisMuch;        return copyThisMuch;    }    else if(FILE_CONTENT_PART == httpUpload->filePart)    {        copyThisMuch = httpUpload->fileSizeLeft;        if(copyThisMuch > bufferSize)            copyThisMuch = bufferSize;        copyThisMuch = fread(dest, 1, copyThisMuch, httpUpload->rFile);         httpUpload->filePos += copyThisMuch;        httpUpload->fileSizeLeft -= copyThisMuch;        if(httpUpload->fileSizeLeft <= 0)        {            httpUpload->filePart = FILE_END_PART;        }        httpUpload->totalSize += copyThisMuch;        return copyThisMuch;    }    else if(FILE_END_PART == httpUpload->filePart)    {        copyThisMuch = httpUpload->endSizeLeft;        if(copyThisMuch > bufferSize)          copyThisMuch = bufferSize;        memcpy(dest, httpUpload->endPtr, copyThisMuch);        httpUpload->endPtr += copyThisMuch;        httpUpload->endSizeLeft -= copyThisMuch;        if(httpUpload->endSizeLeft <= 0)        {            httpUpload->filePart = FILE_TRANSFER_FINISH;        }        httpUpload->totalSize += copyThisMuch;        return copyThisMuch;    }    else    {        printf("file transfer finish, totalSize = %zu .............\n", httpUpload->totalSize);    }        return 0;}int HttpUploadFile(const char* url, const char* fileName){    CURL *curl;    CURLcode res;    char chunkString[256];    size_t totalSize;        /* In windows, this will init the winsock stuff */    res = curl_global_init(CURL_GLOBAL_DEFAULT);    /* Check for errors */    if(res != CURLE_OK)     {        fprintf(stderr, "curl_global_init() failed: %s\n",            curl_easy_strerror(res));        return 1;    }    /* get a curl handle */    curl = curl_easy_init();    if(curl)     {        T_HttpUpload httpUpload;    memset(&httpUpload, sizeof(httpUpload), 0);        httpUpload.filePart = FILE_HEAD_PART;        httpUpload.totalSize = 0;            memset(sHeaderInfo, 0 , HTTP_HEAD_MAX_SIZE);        // 设置起始文件头        sprintf(sHeaderInfo, "%s%s%s%s%s%s%s%s%s%s", "--", BOUNDARY, "\r\n",            "Content-Disposition: form-data; name=\"", "file",  "\"; filename=\"",  fileName, "\"", "\r\n",             "\r\n");        httpUpload.headerPtr = sHeaderInfo;        httpUpload.headerSizeLeft = strlen(httpUpload.headerPtr);        // 设置结束文件信息        memset(sEndInfo, 0 , HTTP_END_MAX_SIZE);        sprintf(sEndInfo, "%s%s%s","\r\n--", BOUNDARY, "--\r\n" );        httpUpload.endPtr = sEndInfo;        httpUpload.endSizeLeft = strlen(httpUpload.endPtr);                printf("\n\nHeaderInfo = :%s\n", httpUpload.headerPtr);        printf("HeaderInfo End\n");                printf("\nEndInfo = :%s\n", httpUpload.endPtr);        printf("EndInfo End\n");            //获取要上传的文件指针              httpUpload.rFile = fopen(fileName, "rb");          if (0 == httpUpload.rFile)          {              printf("the file %s that you will read is not exist", fileName);              return -1;          }                  // 获取文件大小          fseek(httpUpload.rFile, 0, 2);          httpUpload.filePos = 0;        httpUpload.fileSizeLeft = ftell(httpUpload.rFile);          rewind(httpUpload.rFile);                  /* First set the URL that is about to receive our POST. */        curl_easy_setopt(curl, CURLOPT_URL, url);        struct curl_slist *chunk=NULL;        /* Remove a header curl would otherwise add by itself */        //chunk = curl_slist_append(chunk, "Accept:");        /* Add a custom header */        //chunk = curl_slist_append(chunk, "Another: yes");        sprintf(chunkString, "%s%s%s", "Content-Type: ", "multipart/form-data; boundary=", BOUNDARY);  // 重新设置Content-Type        chunk = curl_slist_append(chunk, chunkString);                // 上传的文件总长度, 服务器需要剥离附加文件头，以及附加文件尾，中间部分才是文件的实际内容；服务器可以通过文件的文件头解析出文件名字        totalSize = httpUpload.headerSizeLeft + httpUpload.fileSizeLeft + httpUpload.endSizeLeft;        sprintf(chunkString, "%s%zu","Content-Length: ", totalSize);        chunk = curl_slist_append(chunk, chunkString);        /* Modify a header curl otherwise adds differently */        //chunk = curl_slist_append(chunk, "Host: example.com");        /* Add a header with "blank" contents to the right of the colon. Note that           we're then using a semicolon in the string we pass to curl! */        //chunk = curl_slist_append(chunk, "X-silly-header;");        /* set our custom set of headers */        res = curl_easy_setopt(curl, CURLOPT_HTTPHEADER, chunk);                /* Now specify we want to POST data */        curl_easy_setopt(curl, CURLOPT_POST, 1L);        /* we want to use our own read function */        curl_easy_setopt(curl, CURLOPT_READFUNCTION, read_callback);        /* pointer to pass to our read function */        curl_easy_setopt(curl, CURLOPT_READDATA, &httpUpload);        /* get verbose debug output please */        curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);                /*          If you use POST to a HTTP 1.1 server, you can send data without knowing          the size before starting the POST if you use chunked encoding. You          enable this by adding a header like "Transfer-Encoding: chunked" with          CURLOPT_HTTPHEADER. With HTTP 1.0 or without chunked transfer, you must          specify the size in the request.        */        /* Set the expected POST size. If you want to POST large amounts of data,           consider CURLOPT_POSTFIELDSIZE_LARGE */        curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, (long)(httpUpload.headerSizeLeft + httpUpload.fileSizeLeft + httpUpload.endSizeLeft));        /* Perform the request, res will get the return code */        res = curl_easy_perform(curl);        /* Check for errors */        if(res != CURLE_OK)          fprintf(stderr, "curl_easy_perform() failed: %s\n",                  curl_easy_strerror(res));        /* always cleanup */        curl_easy_cleanup(curl);    }    curl_global_cleanup();     return 0;}int main(int argc, char *argv[]){    if(argc < 2)    {        printf("no input file name ...... \n");        return -1;    }    if(HttpUploadFile("换成自己的URL地址", argv[1]) != 0)    {        printf("HttpUploadFile failed\n");    }    return 0;}