1cbf18d4a7
this allows us to add a valid Authorization header to each but the first GET request, saving us from performing two GET requests every time we want to perform the GET request. we still need a new client nonce and we need to increase the nonce counter, so we also need to calculate a whole new response, as we cannot just reuse the previous Authorization header (that would be a replay attack).
91 lines
2.5 KiB
C++
91 lines
2.5 KiB
C++
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
#pragma once
|
|
|
|
#include "Configuration.h"
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <utility>
|
|
#include <string>
|
|
#include <HTTPClient.h>
|
|
#include <WiFiClient.h>
|
|
|
|
class HttpGetterClient : public HTTPClient {
|
|
public:
|
|
void restartTCP() {
|
|
// keeps the NetworkClient, and closes the TCP connections (as we
|
|
// effectively do not support keep-alive with HTTP 1.0).
|
|
HTTPClient::disconnect(true);
|
|
HTTPClient::connect();
|
|
}
|
|
};
|
|
|
|
using up_http_client_t = std::unique_ptr<HttpGetterClient>;
|
|
using sp_wifi_client_t = std::shared_ptr<WiFiClient>;
|
|
|
|
class HttpRequestResult {
|
|
public:
|
|
HttpRequestResult(bool success,
|
|
up_http_client_t upHttpClient = nullptr,
|
|
sp_wifi_client_t spWiFiClient = nullptr)
|
|
: _success(success)
|
|
, _upHttpClient(std::move(upHttpClient))
|
|
, _spWiFiClient(std::move(spWiFiClient)) { }
|
|
|
|
~HttpRequestResult() {
|
|
// the wifi client *must* die *after* the http client, as the http
|
|
// client uses the wifi client in its destructor.
|
|
if (_upHttpClient) { _upHttpClient->end(); }
|
|
_upHttpClient = nullptr;
|
|
_spWiFiClient = nullptr;
|
|
}
|
|
|
|
HttpRequestResult(HttpRequestResult const&) = delete;
|
|
HttpRequestResult(HttpRequestResult&&) = delete;
|
|
HttpRequestResult& operator=(HttpRequestResult const&) = delete;
|
|
HttpRequestResult& operator=(HttpRequestResult&&) = delete;
|
|
|
|
operator bool() const { return _success; }
|
|
|
|
Stream* getStream() {
|
|
if(!_upHttpClient) { return nullptr; }
|
|
return _upHttpClient->getStreamPtr();
|
|
}
|
|
|
|
private:
|
|
bool _success;
|
|
up_http_client_t _upHttpClient;
|
|
sp_wifi_client_t _spWiFiClient;
|
|
};
|
|
|
|
class HttpGetter {
|
|
public:
|
|
explicit HttpGetter(HttpRequestConfig const& cfg)
|
|
: _config(cfg) { }
|
|
|
|
bool init();
|
|
void addHeader(char const* key, char const* value);
|
|
HttpRequestResult performGetRequest();
|
|
|
|
char const* getErrorText() const { return _errBuffer; }
|
|
|
|
private:
|
|
std::pair<bool, String> getAuthDigest();
|
|
HttpRequestConfig const& _config;
|
|
|
|
template<typename... Args>
|
|
void logError(char const* format, Args... args);
|
|
char _errBuffer[256];
|
|
|
|
bool _useHttps;
|
|
String _host;
|
|
String _uri;
|
|
uint16_t _port;
|
|
|
|
String _wwwAuthenticate = "";
|
|
unsigned _nonceCounter = 0;
|
|
|
|
sp_wifi_client_t _spWiFiClient; // reused for multiple HTTP requests
|
|
|
|
std::vector<std::pair<std::string, std::string>> _additionalHeaders;
|
|
};
|