C++怎么用libcurl獲取下載文件名稱及大小

今天小編給大家分享一下C++怎么用libcurl獲取下載文件名稱及大小的相關知識點，內容詳細，邏輯清晰，相信大部分人都還太了解這方面的知識，所以分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后有所收獲，下面我們一起來了解一下吧。

顯示效果

客戶端利用libcurl庫下載文件，一般需要預先知道文件名稱及大小以及下載進度、下載速度等，以便用戶通過界面顯示實時下載狀態及信息。

1、從下載url鏈接獲取文件名字

如何通過一個文件的下載鏈接（url）獲取到需要下載的文件名稱是我們經常要遇到的一個問題。通常是通過獲取header中的Content-Disposition字段來進行解析獲取。

Content-disposition 是 MIME 協議的擴展，MIME 協議指示 MIME 用戶代理如何顯示附加的文件。Content-disposition其實可以控制用戶請求所得的內容存為一個文件的時候提供一個默認的文件名，文件直接在瀏覽器上顯示或者在訪問時彈出文件下載對話框。

該字段包含了兩種返回值：

1、inline：將文件內容直接顯示在頁面

Content-Disposition: inline;filename=hello.jpg

2、attachment：彈出對話框讓用戶下載

Content-Disposition: attachment;filename=hello.jpg

可以看出無論是哪種返回值，其中都包括了filename，“=”后面就是我們需要的文件名稱。但是有時會出現獲取到的文件名稱存在中文亂碼問題（可能包含非 ASCII 字符）。

2、LibCurl官方示例

Example

static size_t header_callback(char *buffer, size_t size,
                              size_t nitems, void *userdata)
{
  /* received header is nitems * size long in 'buffer' NOT ZERO TERMINATED */
  /* 'userdata' is set with CURLOPT_HEADERDATA */
  return nitems * size;
}
 
CURL *curl = curl_easy_init();
if(curl) 
{
  curl_easy_setopt(curl, CURLOPT_URL, "https://example.com");
 
  curl_easy_setopt(curl, CURLOPT_HEADERFUNCTION, header_callback);
 
  curl_easy_perform(curl);
}

3、實例

// Get the file size and name on the server

size_t header_callback(void *buffer, size_t size, size_t nitems, void *userdata)
{
	string sHeaderData = (char*)buffer;
	string sKey = "Content-Disposition";
	std::size_t nFound = sHeaderData.find(sKey);
	if (nFound != std::string::npos)
	{
		printf("Header data: %s \n", (char*)buffer);
		stringstream ss(sHeaderData);     
		char cSplit = ';'; // 設定好分隔符號
		sKey = "filename=";
		vector<string> results; // 用來存儲結果
		string strResult;
		while (getline(ss, strResult, cSplit))
		{
			nFound = strResult.find(sKey);
			if (nFound != std::string::npos)
			{
				string sFileName = strResult.substr(nFound + sKey.length());
				sKey = '"';
				nFound = sFileName.find(sKey);
				if (nFound != std::string::npos)
				{
					sFileName = sFileName.substr(1, sFileName.length()-2);
				}
				m_sFileName = sFileName;
				printf("sFileName: %s \n", m_sFileName.c_str());
				break;
			}
		}
	}

	return nitems * size;
}

double getDownloadFileInfo(const string url)
{
    CURL *easy_handle = NULL;
	int ret = CURLE_OK;
	double size = -1;

	do
	{
		easy_handle = curl_easy_init();
		if (!easy_handle)
		{
            ZLOG("curl_easy_init error");
			break;
		}

		// Only get the header data
		curl_easy_setopt(easy_handle, CURLOPT_URL, url.c_str());
		curl_easy_setopt(easy_handle, CURLOPT_CUSTOMREQUEST, "GET");    //使用CURLOPT_CUSTOMREQUEST
		//ret |= curl_easy_setopt(easy_handle, CURLOPT_HEADER, 1L);
		curl_easy_setopt(easy_handle, CURLOPT_NOBODY, 1L);
        //ret |= curl_easy_setopt(easy_handle, CURLOPT_WRITEFUNCTION, nousecb);	

		curl_easy_setopt(easy_handle, CURLOPT_HEADERFUNCTION, header_callback);

		ret = curl_easy_perform(easy_handle);
		if (ret != CURLE_OK)
		{
			char s[100] = {0};
			sprintf_s(s, sizeof(s), "error:%d:%s", ret, curl_easy_strerror(static_cast<CURLcode>(ret)));

            ZLOG(s);
			break;
		}

		// size = -1 if no Content-Length return or Content-Length=0
		ret = curl_easy_getinfo(easy_handle, CURLINFO_CONTENT_LENGTH_DOWNLOAD, &size);

		if (ret != CURLE_OK)
		{
            ZLOG("curl_easy_getinfo error");
			break;
		}

	} while (0);

	curl_easy_cleanup(easy_handle);

	if (size > 0.0)
	{
		downloadFileLength = size;
	}

	return size;
}

以上就是“C++怎么用libcurl獲取下載文件名稱及大小”這篇文章的所有內容，感謝各位的閱讀！相信大家閱讀完這篇文章都有很大的收獲，小編每天都會為大家更新不同的知識，如果還想學習更多的知識，請關注億速云行業資訊頻道。