C++ libcurl获取网络文件大小

最新推荐文章于 2024-07-30 09:03:28 发布

原创最新推荐文章于 2024-07-30 09:03:28 发布 · 903 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #开发语言 #java

本文主要介绍如何利用C++的libcurl库来获取网络文件的大小，包括获取本地文件大小、解析URL获取文件名、检查文件存在性、创建文件夹以及获取应用程序数据目录的操作。

1.获取网络文件大小：

#include <curl/curl.h>

size_t get_size_struct(void* ptr, size_t size, size_t nmemb, void* data) {
    return (size_t)(size * nmemb);
}

double GetRemoteFileSize(const char* url) {
    CURL* curl;
    CURLcode res;
    double size = 0.0;

    curl = curl_easy_init();
    curl_easy_setopt(curl, CURLOPT_URL, url);
    curl_easy_setopt(curl, CURLOPT_NOBODY, 1L);
    curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, get_size_struct);
    curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);

    res = curl_easy_perform(curl);
    res = curl_easy_getinfo(curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &size);
    if (res != CURLE_OK) {
        fprintf(stderr, "curl_easy_getinfo() failed: %s\n", curl_easy_strerror(res));
    }
    curl_easy_cleanup(curl);

    return size;
}

2.获取本地文件大小

#include <fstream>
double GetLocalFileSize(const char* filePath)
{
    FILE* fp = fopen(filePath, "rb");

    if (NULL != fp)
    {
        fseek(fp, 0, SEEK_END);
        long fSize =  ftell(fp);
        fclose(fp);
        return fSize;
    }
    return 0;
}

3.从下载文件url中分割出文件名称

string GetImageNameFromUrl(string url)
{
    string::size_type pos = url.find_last_of("/");
    string::size_type len = url.length();
    string name = url.substr(pos + 1,len - pos - 1);
    return name;
}

4.判断文件是否存在

bool IsFileExist(string filePath)
{
    FILE* fp = fopen(filePath.c_str(), "rb");
    
    if (NULL != fp)
    {
        fclose(fp);
        return true;
    }
    return false;
}

5.创建文件夹

bool CreateDir(string dirPath)
{
    BOOL bOk = CreateDirectoryA(dirPath.c_str(), NULL);
    if (TRUE != bOk)
    {
        DWORD errCode = GetLastError();
        // 如果是已经存在就没关系
        if (errCode != 183)
        {
            LOG(INF, "create dir fail,errCode = [%d].", errCode);
        }
    }
    return bOk;
}

6.获取appdata目录

std::string GetAppDataDirPath()
{
    wchar_t buffer[MAX_PATH];
    SHGetSpecialFolderPath(0, buffer, CSIDL_APPDATA, false);
    wstring wstr = buffer;
    std::string str = UnicodeToANSI(wstr);
    return str;
}