Something went wrong. Try again.
Something went wrong. Try again.
5.8 kB · 194 lines
C++
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195#include "HttpDownloader.h"
#include <HTTPClient.h>#include <Logging.h>#include <NetworkClient.h>#include <NetworkClientSecure.h>#include <StreamString.h>#include <base64.h>
#include <cstring>#include <memory>#include <utility>
#include "util/UrlUtils.h"
namespace {class FileWriteStream final : public Stream { public: FileWriteStream(FsFile& file, size_t total, HttpDownloader::ProgressCallback progress) : file_(file), total_(total), progress_(std::move(progress)) {}
size_t write(uint8_t byte) override { return write(&byte, 1); }
size_t write(const uint8_t* buffer, size_t size) override { // Write-through stream for HTTPClient::writeToStream with progress tracking. const size_t written = file_.write(buffer, size); if (written != size) { writeOk_ = false; } downloaded_ += written; if (progress_ && total_ > 0) { progress_(downloaded_, total_); } return written; }
int available() override { return 0; } int read() override { return -1; } int peek() override { return -1; } void flush() override { file_.flush(); }
size_t downloaded() const { return downloaded_; } bool ok() const { return writeOk_; }
private: FsFile& file_; size_t total_; size_t downloaded_ = 0; bool writeOk_ = true; HttpDownloader::ProgressCallback progress_;};} // namespace
bool HttpDownloader::fetchUrl(const std::string& url, Stream& outContent, const std::string& username, const std::string& password) { std::unique_ptr<NetworkClient> client; if (UrlUtils::isHttpsUrl(url)) { auto* secureClient = new NetworkClientSecure(); secureClient->setInsecure(); client.reset(secureClient); } else { client.reset(new NetworkClient()); } HTTPClient http;
LOG_DBG("HTTP", "Fetching: %s", url.c_str());
http.begin(*client, url.c_str()); http.setFollowRedirects(HTTPC_STRICT_FOLLOW_REDIRECTS); http.addHeader("User-Agent", "CrossPoint-ESP32-" CROSSPOINT_VERSION);
if (!username.empty() && !password.empty()) { std::string credentials = username + ":" + password; String encoded = base64::encode(credentials.c_str()); http.addHeader("Authorization", "Basic " + encoded); }
const int httpCode = http.GET(); if (httpCode != HTTP_CODE_OK) { LOG_ERR("HTTP", "Fetch failed: %d", httpCode); http.end(); return false; }
http.writeToStream(&outContent);
http.end();
LOG_DBG("HTTP", "Fetch success"); return true;}
bool HttpDownloader::fetchUrl(const std::string& url, std::string& outContent, const std::string& username, const std::string& password) { StreamString stream; if (!fetchUrl(url, stream, username, password)) { return false; } outContent = stream.c_str(); return true;}
HttpDownloader::DownloadError HttpDownloader::downloadToFile(const std::string& url, const std::string& destPath, ProgressCallback progress, const std::string& username, const std::string& password) { std::unique_ptr<NetworkClient> client; if (UrlUtils::isHttpsUrl(url)) { auto* secureClient = new NetworkClientSecure(); secureClient->setInsecure(); client.reset(secureClient); } else { client.reset(new NetworkClient()); } HTTPClient http;
LOG_DBG("HTTP", "Downloading: %s", url.c_str()); LOG_DBG("HTTP", "Destination: %s", destPath.c_str());
http.begin(*client, url.c_str()); http.setFollowRedirects(HTTPC_STRICT_FOLLOW_REDIRECTS); http.addHeader("User-Agent", "CrossPoint-ESP32-" CROSSPOINT_VERSION);
if (!username.empty() && !password.empty()) { std::string credentials = username + ":" + password; String encoded = base64::encode(credentials.c_str()); http.addHeader("Authorization", "Basic " + encoded); }
const int httpCode = http.GET(); if (httpCode != HTTP_CODE_OK) { LOG_ERR("HTTP", "Download failed: %d", httpCode); http.end(); return HTTP_ERROR; }
const int64_t reportedLength = http.getSize(); const size_t contentLength = reportedLength > 0 ? static_cast<size_t>(reportedLength) : 0; if (contentLength > 0) { LOG_DBG("HTTP", "Content-Length: %zu", contentLength); } else { LOG_DBG("HTTP", "Content-Length: unknown"); }
// Remove existing file if present if (Storage.exists(destPath.c_str())) { Storage.remove(destPath.c_str()); }
// Open file for writing FsFile file; if (!Storage.openFileForWrite("HTTP", destPath.c_str(), file)) { LOG_ERR("HTTP", "Failed to open file for writing"); http.end(); return FILE_ERROR; }
// Let HTTPClient handle chunked decoding and stream body bytes into the file. FileWriteStream fileStream(file, contentLength, progress); const int writeResult = http.writeToStream(&fileStream);
file.close(); http.end();
if (writeResult < 0) { LOG_ERR("HTTP", "writeToStream error: %d", writeResult); Storage.remove(destPath.c_str()); return HTTP_ERROR; }
const size_t downloaded = fileStream.downloaded(); LOG_DBG("HTTP", "Downloaded %zu bytes", downloaded);
// Guard against partial writes even if HTTPClient completes. if (!fileStream.ok()) { LOG_ERR("HTTP", "Write failed during download"); Storage.remove(destPath.c_str()); return FILE_ERROR; }
if (contentLength == 0 && downloaded == 0) { LOG_ERR("HTTP", "Download failed: no data received"); Storage.remove(destPath.c_str()); return HTTP_ERROR; }
// Verify download size if known if (contentLength > 0 && downloaded != contentLength) { LOG_ERR("HTTP", "Size mismatch: got %zu, expected %zu", downloaded, contentLength); Storage.remove(destPath.c_str()); return HTTP_ERROR; }
return OK;}