You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.4 KiB
C++
118 lines
3.4 KiB
C++
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
// Tester application for testing the http/https/ftp/ssh/sftp connection
|
|
|
|
#include <exiv2/exiv2.hpp>
|
|
#ifdef EXV_USE_CURL
|
|
#include <curl/curl.h>
|
|
#endif
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
|
|
void httpcon(const std::string& url, bool useHttp1_0 = false) {
|
|
Exiv2::Dictionary response;
|
|
Exiv2::Dictionary request;
|
|
std::string errors;
|
|
|
|
Exiv2::Uri uri = Exiv2::Uri::Parse(url);
|
|
Exiv2::Uri::Decode(uri);
|
|
|
|
request["server"] = uri.Host;
|
|
request["page"] = uri.Path;
|
|
request["port"] = uri.Port;
|
|
if (!useHttp1_0)
|
|
request["version"] = "1.1";
|
|
|
|
int serverCode = Exiv2::http(request, response, errors);
|
|
if (serverCode < 0 || serverCode >= 400 || !errors.empty()) {
|
|
throw Exiv2::Error(Exiv2::ErrorCode::kerTiffDirectoryTooLarge, "Server", serverCode);
|
|
}
|
|
}
|
|
|
|
#ifdef EXV_USE_CURL
|
|
void curlcon(const std::string& url, bool useHttp1_0 = false) {
|
|
CURL* curl = curl_easy_init();
|
|
if (!curl) {
|
|
throw Exiv2::Error(Exiv2::ErrorCode::kerErrorMessage, "Unable to init libcurl.");
|
|
}
|
|
|
|
// get the timeout value
|
|
std::string timeoutStr = Exiv2::getEnv(Exiv2::envTIMEOUT);
|
|
long timeout = atol(timeoutStr.c_str());
|
|
if (timeout == 0) {
|
|
throw Exiv2::Error(Exiv2::ErrorCode::kerErrorMessage, "Timeout Environmental Variable must be a positive integer.");
|
|
}
|
|
|
|
std::string response;
|
|
curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, Exiv2::curlWriter);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L);
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, timeout);
|
|
// curl_easy_setopt(curl, CURLOPT_VERBOSE, 1); // debug
|
|
if (useHttp1_0)
|
|
curl_easy_setopt(curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
|
|
else
|
|
curl_easy_setopt(curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_1);
|
|
|
|
/* Perform the request, res will get the return code */
|
|
CURLcode res = curl_easy_perform(curl);
|
|
if (res != CURLE_OK) { // error happened
|
|
throw Exiv2::Error(Exiv2::ErrorCode::kerErrorMessage, curl_easy_strerror(res));
|
|
}
|
|
|
|
// get return code
|
|
long returnCode;
|
|
curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &returnCode); // get code
|
|
curl_easy_cleanup(curl);
|
|
|
|
if (returnCode >= 400 || returnCode < 0) {
|
|
throw Exiv2::Error(Exiv2::ErrorCode::kerTiffDirectoryTooLarge, "Server", returnCode);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
int main(int argc, const char** argv) {
|
|
Exiv2::XmpParser::initialize();
|
|
::atexit(Exiv2::XmpParser::terminate);
|
|
|
|
if (argc < 2) {
|
|
std::cout << "Usage: " << argv[0] << " url {-http1_0}" << '\n';
|
|
return EXIT_FAILURE;
|
|
}
|
|
std::string url(argv[1]);
|
|
Exiv2::Protocol prot = Exiv2::fileProtocol(url);
|
|
|
|
bool useHttp1_0 = false;
|
|
for (int a = 1; a < argc; a++) {
|
|
std::string arg(argv[a]);
|
|
if (arg == "-http1_0")
|
|
useHttp1_0 = true;
|
|
}
|
|
|
|
bool isOk = false;
|
|
try {
|
|
#ifdef EXV_USE_CURL
|
|
if (prot == Exiv2::pHttp || prot == Exiv2::pHttps || prot == Exiv2::pFtp) {
|
|
curlcon(url, useHttp1_0);
|
|
isOk = true;
|
|
}
|
|
#endif
|
|
if (!isOk && prot == Exiv2::pHttp) {
|
|
httpcon(url, useHttp1_0);
|
|
isOk = true;
|
|
}
|
|
} catch (const Exiv2::Error& e) {
|
|
std::cout << "Error: '" << e << "'" << '\n';
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!isOk)
|
|
std::cout << "The protocol is unsupported." << '\n';
|
|
else
|
|
std::cout << "OK." << '\n';
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
// That's all Folks!
|