35 #include <XrdVersion.hh>
37 #include <curl/curl.h>
38 #include <openssl/bio.h>
39 #include <openssl/evp.h>
46 #include <sys/syscall.h>
47 #include <sys/types.h>
58 thread_local std::vector<CURL*> HandlerQueue::m_handles;
59 std::atomic<unsigned> CurlWorker::m_maintenance_period = 5;
60 std::vector<std::unique_ptr<XrdClHttp::CurlWorker>> CurlWorker::m_workers;
61 std::mutex CurlWorker::m_workers_mutex;
64 std::atomic<uint64_t> CurlWorker::m_conncall_errors = 0;
65 std::atomic<uint64_t> CurlWorker::m_conncall_req = 0;
66 std::atomic<uint64_t> CurlWorker::m_conncall_success = 0;
67 std::atomic<uint64_t> CurlWorker::m_conncall_timeout = 0;
68 decltype(CurlWorker::m_ops) CurlWorker::m_ops = {};
69 std::vector<std::atomic<std::chrono::system_clock::rep>*> CurlWorker::m_workers_last_completed_cycle;
70 std::vector<std::atomic<std::chrono::system_clock::rep>*> CurlWorker::m_workers_oldest_op;
71 std::mutex CurlWorker::m_worker_stats_mutex;
74 std::atomic<uint64_t> HandlerQueue::m_ops_consumed = 0;
75 std::atomic<uint64_t> HandlerQueue::m_ops_produced = 0;
76 std::atomic<uint64_t> HandlerQueue::m_ops_rejected = 0;
79 CurlWorker::initcontrol CurlWorker::m_initcontrol;
89 #if defined(__APPLE__)
90 uint64_t pth_threadid;
91 pthread_threadid_np(pthread_self(), &pth_threadid);
93 #elif defined(__linux__)
98 return syscall(SYS_gettid);
107 return (status < 100) || (status >= 400);
182 case CURLE_COULDNT_RESOLVE_PROXY:
183 case CURLE_COULDNT_RESOLVE_HOST:
185 case CURLE_LOGIN_DENIED:
190 case CURLE_REMOTE_ACCESS_DENIED:
192 case CURLE_SSL_CONNECT_ERROR:
193 case CURLE_SSL_ENGINE_NOTFOUND:
194 case CURLE_SSL_ENGINE_SETFAILED:
195 case CURLE_SSL_CERTPROBLEM:
196 case CURLE_SSL_CIPHER:
198 case CURLE_SSL_SHUTDOWN_FAILED:
199 case CURLE_SSL_CRL_BADFILE:
200 case CURLE_SSL_ISSUER_ERROR:
201 case CURLE_SSL_CACERT:
205 case CURLE_SEND_ERROR:
206 case CURLE_RECV_ERROR:
208 case CURLE_COULDNT_CONNECT:
209 case CURLE_GOT_NOTHING:
211 case CURLE_OPERATION_TIMEDOUT:
212 #ifdef HAVE_XPROTOCOL_TIMEREXPIRED
217 case CURLE_UNSUPPORTED_PROTOCOL:
218 case CURLE_NOT_BUILT_IN:
220 case CURLE_FAILED_INIT:
222 case CURLE_URL_MALFORMAT:
228 case CURLE_PARTIAL_FILE:
233 case CURLE_READ_ERROR:
234 case CURLE_WRITE_ERROR:
236 case CURLE_RANGE_ERROR:
237 case CURLE_BAD_CONTENT_ENCODING:
239 case CURLE_TOO_MANY_REDIRECTS:
247 if (input.size() > 44 || input.size() % 4 != 0)
return false;
248 if (input.size() == 0)
return true;
250 std::unique_ptr<BIO, decltype(&BIO_free_all)> b64(BIO_new(BIO_f_base64()), &BIO_free_all);
251 BIO_set_flags(b64.get(), BIO_FLAGS_BASE64_NO_NL);
252 std::unique_ptr<BIO, decltype(&BIO_free_all)> bmem(
253 BIO_new_mem_buf(
const_cast<char *
>(input.data()), input.size()), &BIO_free_all);
254 bmem.reset(BIO_push(b64.release(), bmem.release()));
257 size_t expectedLen =
static_cast<size_t>(input.size() * 0.75);
258 if (input[input.size() - 1] ==
'=') {
260 if (input[input.size() - 2] ==
'=') {
265 auto len = BIO_read(bmem.get(), &output[0], output.size());
267 if (len == -1 ||
static_cast<size_t>(len) != expectedLen)
return false;
279 if (m_recv_all_headers) {
280 m_recv_all_headers =
false;
281 m_recv_status_line =
false;
284 if (!m_recv_status_line) {
285 m_recv_status_line =
true;
287 std::stringstream ss(header_line);
290 m_resp_protocol = item;
293 m_status_code = std::stol(item);
297 if (m_status_code < 100 || m_status_code >= 600) {
301 auto cr_loc = item.find(
'\r');
302 if (cr_loc != std::string::npos) {
303 m_resp_message = item.substr(0, cr_loc);
305 m_resp_message = item;
310 if (header_line.empty() || header_line ==
"\n" || header_line ==
"\r\n") {
311 m_recv_all_headers =
true;
315 auto found = header_line.find(
":");
316 if (found == std::string::npos) {
320 std::string header_name = header_line.substr(0, found);
326 while (found < header_line.size()) {
327 if (header_line[found] !=
' ') {
break;}
330 std::string header_value = header_line.substr(found);
333 header_value.erase(header_value.find_last_not_of(
" \r\n\t") + 1);
337 auto iter = m_headers.find(header_name);
338 if (iter == m_headers.end()) {
339 m_headers.insert(iter, {header_name, {header_value}});
341 iter->second.push_back(header_value);
344 if (header_name ==
"Allow") {
345 std::string_view val(header_value);
346 while (!val.empty()) {
347 auto found = val.find(
',');
348 auto method = val.substr(0, found);
349 if (method ==
"PROPFIND") {
353 if (found == std::string_view::npos)
break;
354 val = val.substr(found + 1);
359 }
else if (header_name ==
"Content-Length") {
361 m_content_length = std::stoll(header_value);
366 else if (header_name ==
"Content-Type") {
367 std::string_view val(header_value);
368 auto found = val.find(
";");
369 auto first_type = val.substr(0, found);
370 m_multipart_byteranges = first_type ==
"multipart/byteranges";
371 if (m_multipart_byteranges) {
372 auto remainder = val.substr(found + 1);
373 found = remainder.find(
"boundary=");
374 if (found != std::string_view::npos) {
379 else if (header_name ==
"Content-Range") {
380 auto found = header_value.find(
" ");
381 if (found == std::string::npos) {
384 std::string range_unit = header_value.substr(0, found);
385 if (range_unit !=
"bytes") {
388 auto range_resp = header_value.substr(found + 1);
389 found = range_resp.find(
"/");
390 if (found == std::string::npos) {
393 auto incl_range = range_resp.substr(0, found);
394 found = incl_range.find(
"-");
395 if (found == std::string::npos) {
398 auto first_pos = incl_range.substr(0, found);
400 m_response_offset = std::stoll(first_pos);
404 auto last_pos = incl_range.substr(found + 1);
407 last_byte = std::stoll(last_pos);
411 m_content_length = last_byte - m_response_offset + 1;
413 else if (header_name ==
"Location") {
414 m_location = header_value;
415 }
else if (header_name ==
"Digest") {
418 else if (header_name ==
"Etag")
422 m_etag = header_value;
423 m_etag.erase(remove(m_etag.begin(), m_etag.end(),
'\"'), m_etag.end());
425 else if (header_name ==
"Cache-Control")
427 m_cache_control = header_value;
437 std::string_view view(digest);
438 std::array<unsigned char, 32> checksum_value;
439 std::string digest_lower;
440 while (!view.empty()) {
441 auto nextsep = view.find(
',');
442 auto entry = view.substr(0, nextsep);
443 if (nextsep == std::string_view::npos) {
446 view = view.substr(nextsep + 1);
448 nextsep = entry.find(
'=');
449 auto name = entry.substr(0, nextsep);
450 auto value = entry.substr(nextsep + 1);
451 digest_lower.clear();
452 digest_lower.resize(name.size());
453 std::transform(name.begin(), name.end(), digest_lower.begin(), [](
unsigned char c) {
454 return std::tolower(c);
456 if (digest_lower ==
"md5") {
457 if (value.size() != 24) {
461 info.
Set(XrdClHttp::ChecksumType::kMD5, checksum_value);
463 }
else if (digest_lower ==
"crc32c") {
468 if (value.size() == 8 && value[6] ==
'=' && value[7] ==
'=') {
470 info.
Set(XrdClHttp::ChecksumType::kCRC32C, checksum_value);
477 val = std::stoul(value.data(), &pos, 16);
481 if (pos == value.size()) {
482 checksum_value[0] = (val >> 24) & 0xFF;
483 checksum_value[1] = (val >> 16) & 0xFF;
484 checksum_value[2] = (val >> 8) & 0xFF;
485 checksum_value[3] = val & 0xFF;
486 info.
Set(XrdClHttp::ChecksumType::kCRC32C, checksum_value);
496 case XrdClHttp::ChecksumType::kMD5:
498 case XrdClHttp::ChecksumType::kCRC32C:
500 case XrdClHttp::ChecksumType::kSHA1:
502 case XrdClHttp::ChecksumType::kSHA256:
510 bool HeaderParser::validHeaderByte(
unsigned char c)
512 const static uint64_t mask_lower = 0 |
513 uint64_t((1<<10)-1) <<
'0' |
519 uint64_t(1) <<
'\'' |
525 const static uint64_t mask_upper = 0 |
526 uint64_t((1<<26)-1) << (
'a'-64) |
527 uint64_t((1<<26)-1) << (
'A'-64) |
528 uint64_t(1) << (
'^'-64) |
529 uint64_t(1) << (
'_'-64) |
530 uint64_t(1) << (
'`'-64) |
531 uint64_t(1) << (
'|'-64) |
532 uint64_t(1) << (
'~'-64);
534 if (c >= 128)
return false;
535 if (c >= 64)
return (uint64_t(1)<<(c-64)) & mask_upper;
536 return (uint64_t(1) << c) & mask_lower;
542 const static int toLower =
'a' -
'A';
543 for (
size_t idx=0; idx<headerName.size(); idx++) {
544 char c = headerName[idx];
545 if (!validHeaderByte(c)) {
548 if (upper &&
'a' <= c && c <=
'z') {
550 }
else if (!upper &&
'A' <= c && c <=
'Z') {
560 m_max_pending_ops(max_pending_ops)
563 auto result = pipe(filedes);
565 throw std::runtime_error(strerror(errno));
567 if (fcntl(filedes[0], F_SETFL, O_NONBLOCK | O_CLOEXEC) == -1 || fcntl(filedes[1], F_SETFL, O_NONBLOCK | O_CLOEXEC) == -1) {
570 throw std::runtime_error(strerror(errno));
572 m_read_fd = filedes[0];
573 m_write_fd = filedes[1];
578 bool EnableCurlHeaderDump() {
587 int DumpHeader(
CURL *handle, curl_infotype type,
char *data,
size_t size,
void *clientp) {
589 auto *logger =
static_cast<XrdCl::Log *
>(clientp);
590 if (!logger || !data || size == 0) {
594 const char *direction =
nullptr;
596 case CURLINFO_HEADER_OUT:
599 case CURLINFO_HEADER_IN:
606 const std::string redacted =
obfuscateAuth(std::string(data, size));
607 logger->Debug(
kLogXrdClHttp,
"%s %s", direction, redacted.c_str());
616 for (
size_t idx = 0; idx < input_view.size(); idx++) {
617 if (!isspace(view[view.size() - 1 - idx])) {
618 return view.substr(0, view.size() - idx);
626 for (
size_t idx = 0; idx < input_view.size(); idx++) {
627 if (!isspace(input_view[idx])) {
628 return input_view.substr(idx);
636 auto result = curl_easy_init();
637 if (result ==
nullptr) {
641 curl_easy_setopt(result, CURLOPT_USERAGENT,
"xrdcl-http/" XrdVERSION);
642 curl_easy_setopt(result, CURLOPT_DEBUGFUNCTION, DumpHeader);
645 curl_easy_setopt(result, CURLOPT_VERBOSE, 1L);
649 if (!env->GetString(
"HttpCertFile", ca_file) || ca_file.empty()) {
650 char *x509_ca_file = getenv(
"X509_CERT_FILE");
652 ca_file = std::string(x509_ca_file);
655 if (!ca_file.empty()) {
656 curl_easy_setopt(result, CURLOPT_CAINFO, ca_file.c_str());
659 if (!env->GetString(
"HttpCertDir", ca_dir) || ca_dir.empty()) {
660 char *x509_ca_dir = getenv(
"X509_CERT_DIR");
662 ca_dir = std::string(x509_ca_dir);
665 if (!ca_dir.empty()) {
666 curl_easy_setopt(result, CURLOPT_CAPATH, ca_dir.c_str());
669 curl_easy_setopt(result, CURLOPT_BUFFERSIZE, 32*1024);
676 if (m_handles.size()) {
677 auto result = m_handles.back();
678 m_handles.pop_back();
687 m_handles.push_back(curl);
693 std::unique_lock<std::mutex> lk(m_mutex);
694 auto now = std::chrono::steady_clock::now();
697 for (
auto &op : m_ops) {
698 if (!op->IsPaused())
continue;
700 if (op->TransferStalled(0, now)) {
701 op->ContinueHandle();
705 std::vector<decltype(m_ops)::value_type> expired_ops;
706 unsigned expired_count = 0;
707 auto it = std::remove_if(m_ops.begin(), m_ops.end(),
708 [&](
const std::shared_ptr<CurlOperation> &handler) {
709 auto expired = handler->GetOperationExpiry() < now;
711 expired_ops.push_back(handler);
716 m_ops.erase(it, m_ops.end());
721 unsigned bytes_to_read = expired_count;
722 while (bytes_to_read > 0) {
723 size_t chunk = std::min<size_t>(
sizeof(throwaway), bytes_to_read);
724 ssize_t n =
read(m_read_fd, throwaway, chunk);
727 }
else if (n == -1) {
728 if (errno == EINTR) {
748 for (
auto &handler : expired_ops) {
754 HandlerQueue::Produce(std::shared_ptr<CurlOperation> handler)
756 auto handler_expiry = handler->GetOperationExpiry();
757 std::unique_lock<std::mutex> lk{m_mutex};
758 m_producer_cv.wait_until(lk,
760 [&]{
return m_ops.size() < m_max_pending_ops;}
762 if (std::chrono::steady_clock::now() > handler_expiry) {
765 m_ops_rejected.fetch_add(1, std::memory_order_relaxed);
769 m_ops.push_back(handler);
772 auto result =
write(m_write_fd, ready, 1);
774 if (errno == EINTR) {
776 }
else if (errno == EAGAIN || errno == EWOULDBLOCK) {
781 throw std::runtime_error(strerror(errno));
787 m_consumer_cv.notify_one();
788 m_ops_produced.fetch_add(1, std::memory_order_relaxed);
791 std::shared_ptr<CurlOperation>
792 HandlerQueue::Consume(std::chrono::steady_clock::duration dur)
794 std::unique_lock<std::mutex> lk(m_mutex);
795 m_consumer_cv.wait_for(lk, dur, [&]{
return m_ops.size() > 0 || m_shutdown;});
796 if (m_shutdown || m_ops.empty()) {
800 std::shared_ptr<CurlOperation> result = m_ops.front();
805 auto result =
read(m_read_fd, ready, 1);
807 if (errno == EINTR) {
809 }
else if (errno == EAGAIN || errno == EWOULDBLOCK) {
814 throw std::runtime_error(strerror(errno));
820 m_producer_cv.notify_one();
821 m_ops_consumed.fetch_add(1, std::memory_order_relaxed);
827 HandlerQueue::GetMonitoringJson()
829 auto consumed = m_ops_consumed.load(std::memory_order_relaxed);
830 auto produced = m_ops_produced.load(std::memory_order_relaxed);
832 "\"produced\":" + std::to_string(produced) +
","
833 "\"consumed\":" + std::to_string(consumed) +
","
834 "\"pending\":" + std::to_string(produced - consumed) +
","
835 "\"rejected\":" + std::to_string(m_ops_rejected.load(std::memory_order_relaxed)) +
839 std::shared_ptr<CurlOperation>
840 HandlerQueue::TryConsume()
842 std::unique_lock<std::mutex> lk(m_mutex);
843 if (m_ops.size() == 0) {
844 std::shared_ptr<CurlOperation> result;
848 std::shared_ptr<CurlOperation> result = m_ops.front();
853 auto result =
read(m_read_fd, ready, 1);
855 if (errno == EINTR) {
857 }
else if (errno == EAGAIN || errno == EWOULDBLOCK) {
862 throw std::runtime_error(strerror(errno));
868 m_producer_cv.notify_one();
869 m_ops_consumed.fetch_add(1, std::memory_order_relaxed);
875 HandlerQueue::Shutdown()
877 std::unique_lock lock(m_mutex);
879 m_consumer_cv.notify_all();
883 HandlerQueue::ReleaseHandles()
885 for (
auto handle : m_handles) {
886 curl_easy_cleanup(handle);
897 std::unique_lock lk(m_worker_stats_mutex);
898 m_stats_offset = m_workers_last_completed_cycle.size();
899 m_workers_last_completed_cycle.push_back(&m_last_completed_cycle);
900 m_workers_oldest_op.push_back(&m_oldest_op);
903 if ((pipe(pipeInfo) == -1) || (fcntl(pipeInfo[0], F_SETFD, FD_CLOEXEC)) || (fcntl(pipeInfo[1], F_SETFD, FD_CLOEXEC))) {
904 throw std::runtime_error(
"Failed to create shutdown monitoring pipe for curl worker");
906 m_shutdown_pipe_r = pipeInfo[0];
907 m_shutdown_pipe_w = pipeInfo[1];
911 env->GetString(
"HttpClientCertFile", m_x509_client_cert_file);
912 env->GetString(
"HttpClientKeyFile", m_x509_client_key_file);
917 return std::make_tuple(m_x509_client_cert_file, m_x509_client_key_file);
923 auto now = std::chrono::system_clock::now().time_since_epoch().count();
924 auto oldest_op = now;
925 auto oldest_cycle = now;
927 std::unique_lock lk(m_worker_stats_mutex);
928 for (
const auto &entry : m_workers_last_completed_cycle) {
929 if (!entry) {
continue;}
930 auto cycle = entry->load(std::memory_order_relaxed);
931 if (cycle < oldest_cycle) oldest_cycle = cycle;
933 for (
const auto &entry : m_workers_oldest_op) {
934 if (!entry) {
continue;}
935 auto op = entry->load(std::memory_order_relaxed);
936 if (op < oldest_op) oldest_op = op;
939 auto oldest_op_dbl = std::chrono::duration<double>(std::chrono::system_clock::time_point(std::chrono::system_clock::duration(oldest_op)).time_since_epoch()).count();
940 auto oldest_cycle_dbl = std::chrono::duration<double>(std::chrono::system_clock::time_point(std::chrono::system_clock::duration(oldest_cycle)).time_since_epoch()).count();
941 std::string retval =
"{"
942 "\"oldest_op\":" + std::to_string(oldest_op_dbl) +
","
943 "\"oldest_cycle\":" + std::to_string(oldest_cycle_dbl) +
","
946 for (
size_t verb_idx = 0; verb_idx < static_cast<int>(XrdClHttp::CurlOperation::HttpVerb::Count); verb_idx++) {
948 for (
size_t op_idx = 0; op_idx < 402; op_idx++) {
949 if (op_idx == 401)
continue;
951 auto &op_stats = m_ops[verb_idx][op_idx];
952 auto duration = op_stats.m_duration.load(std::memory_order_relaxed);
953 if (duration == 0)
continue;
955 std::string prefix =
"http_" + verb_str +
"_" + ((op_idx == 402) ?
"invalid" : std::to_string(200 + op_idx)) +
"_";
957 auto duration_dbl = std::chrono::duration<double>(std::chrono::steady_clock::duration(duration)).count();
958 retval +=
"\"" + prefix +
"duration\":" + std::to_string(duration_dbl) +
",";
960 duration = op_stats.m_pause_duration.load(std::memory_order_relaxed);
962 duration_dbl = std::chrono::duration<double>(std::chrono::steady_clock::duration(duration)).count();
963 retval +=
"\"" + prefix +
"pause_duration\":" + std::to_string(duration_dbl) +
",";
966 auto count = op_stats.m_bytes.load(std::memory_order_relaxed);
967 if (count) retval +=
"\"" + prefix +
"bytes\":" + std::to_string(count) +
",";
968 count = op_stats.m_error.load(std::memory_order_relaxed);
969 if (count) retval +=
"\"" + prefix +
"error\":" + std::to_string(count) +
",";
970 count = op_stats.m_finished.load(std::memory_order_relaxed);
971 if (count) retval +=
"\"" + prefix +
"finished\":" + std::to_string(count) +
",";
972 count = op_stats.m_client_timeout.load(std::memory_order_relaxed);
973 if (count) retval +=
"\"" + prefix +
"client_timeout\":" + std::to_string(count) +
",";
974 count = op_stats.m_server_timeout.load(std::memory_order_relaxed);
975 if (count) retval +=
"\"" + prefix +
"server_timeout\":" + std::to_string(count) +
",";
978 auto &op_stats = m_ops[verb_idx][401];
979 auto duration = op_stats.m_duration.load(std::memory_order_relaxed);
980 if (duration == 0)
continue;
982 std::string prefix =
"http_" + verb_str +
"_";
984 auto duration_dbl = std::chrono::duration<double>(std::chrono::steady_clock::duration(duration)).count();
985 retval +=
"\"" + prefix +
"preheader_duration\":" + std::to_string(duration_dbl) +
",";
987 auto count = op_stats.m_started.load(std::memory_order_relaxed);
988 if (count) retval +=
"\"" + prefix +
"started\":" + std::to_string(count) +
",";
989 count = op_stats.m_error.load(std::memory_order_relaxed);
990 if (count) retval +=
"\"" + prefix +
"preheader_error\":" + std::to_string(count) +
",";
991 count = op_stats.m_finished.load(std::memory_order_relaxed);
992 if (count) retval +=
"\"" + prefix +
"preheader_finished\":" + std::to_string(count) +
",";
993 count = op_stats.m_server_timeout.load(std::memory_order_relaxed);
994 if (count) retval +=
"\"" + prefix +
"preheader_timeout\":" + std::to_string(count) +
",";
995 count = op_stats.m_conncall_timeout.load(std::memory_order_relaxed);
996 if (count) retval +=
"\"" + prefix +
"conncall_timeout\":" + std::to_string(count) +
",";
1001 "\"conncall_error\":" + std::to_string(m_conncall_errors.load(std::memory_order_relaxed)) +
","
1002 "\"conncall_started\":" + std::to_string(m_conncall_req.load(std::memory_order_relaxed)) +
","
1003 "\"conncall_success\":" + std::to_string(m_conncall_success.load(std::memory_order_relaxed)) +
","
1004 "\"conncall_timeout\":" + std::to_string(m_conncall_timeout.load(std::memory_order_relaxed)) +
1017 if (sc < 0 || kind == OpKind::Start || sc == 100) {
1019 }
else if (sc < 200 || sc >= 600) {
1024 auto [bytes, pre_headers, post_headers, pause_duration] = op.
StatisticsReset();
1025 auto &op_stats = m_ops[
static_cast<int>(op.
GetVerb())][sc];
1026 op_stats.m_bytes.fetch_add(bytes, std::memory_order_relaxed);
1027 op_stats.m_duration.fetch_add((sc == 401) ? pre_headers.count() : post_headers.count(), std::memory_order_relaxed);
1028 op_stats.m_pause_duration.fetch_add(pause_duration.count(), std::memory_order_relaxed);
1029 if (pre_headers != std::chrono::steady_clock::duration::zero() && sc != 401) {
1030 auto &old_stats = m_ops[
static_cast<int>(op.
GetVerb())][401];
1031 old_stats.m_duration.fetch_add(pre_headers.count(), std::memory_order_relaxed);
1034 case OpKind::ConncallTimeout:
1035 op_stats.m_conncall_timeout.fetch_add(1, std::memory_order_relaxed);
1037 case OpKind::ClientTimeout:
1038 op_stats.m_client_timeout.fetch_add(1, std::memory_order_relaxed);
1041 op_stats.m_error.fetch_add(1, std::memory_order_relaxed);
1043 case OpKind::Finish:
1044 op_stats.m_finished.fetch_add(1, std::memory_order_relaxed);
1047 op_stats.m_started.fetch_add(1, std::memory_order_relaxed);
1049 case OpKind::ServerTimeout:
1050 op_stats.m_server_timeout.fetch_add(1, std::memory_order_relaxed);
1052 case OpKind::Update:
1061 std::unique_lock lock(m_workers_mutex);
1062 m_workers.emplace_back(std::move(
self));
1063 m_self_tid = std::move(tid);
1065 std::unique_lock lock(m_start_lock);
1066 m_start_complete =
true;
1067 m_start_complete_cv.notify_one();
1074 std::unique_lock lock(myself->m_start_lock);
1075 myself->m_start_complete_cv.wait(lock, [&]{
return myself->m_start_complete;});
1082 std::unique_lock lock(m_workers_mutex);
1083 auto iter = std::remove_if(m_workers.begin(), m_workers.end(), [&](std::unique_ptr<XrdClHttp::CurlWorker> &worker){return worker.get() == myself;});
1084 m_workers.erase(iter);
1091 int max_pending = 50;
1094 auto &queue = *m_queue.get();
1097 CURLM *multi_handle = curl_multi_init();
1098 if (multi_handle ==
nullptr) {
1099 throw std::runtime_error(
"Failed to create curl multi-handle");
1102 int running_handles = 0;
1103 time_t last_maintenance = time(NULL);
1104 CURLMcode mres = CURLM_OK;
1108 std::unordered_map<int, WaitingForBroker> broker_reqs;
1109 std::vector<struct curl_waitfd> waitfds;
1111 bool want_shutdown =
false;
1112 while (!want_shutdown) {
1113 m_last_completed_cycle.store(std::chrono::system_clock::now().time_since_epoch().count());
1114 auto oldest_op = std::chrono::system_clock::now();
1115 for (
const auto &entry : m_op_map) {
1116 OpRecord(*entry.second.first, OpKind::Update);
1117 if (entry.second.second < oldest_op) {
1118 oldest_op = entry.second.second;
1121 m_oldest_op.store(oldest_op.time_since_epoch().count());
1125 auto op = m_continue_queue->TryConsume();
1132 m_logger->
Debug(
kLogXrdClHttp,
"Ignoring continuation of operation that has already completed");
1135 m_logger->
Debug(
kLogXrdClHttp,
"Continuing the curl handle from op %p on thread %d", op.get(), getthreadid());
1142 curl_multi_remove_handle(multi_handle, curl);
1143 curl_easy_cleanup(curl);
1144 m_op_map.erase(curl);
1146 running_handles -= 1;
1149 auto iter = m_op_map.find(curl);
1150 if (iter != m_op_map.end()) iter->second.second = std::chrono::system_clock::now();
1154 while (running_handles <
static_cast<int>(m_max_ops)) {
1155 auto op = running_handles == 0 ? queue.Consume(std::chrono::seconds(1)) : queue.TryConsume();
1159 auto curl = queue.GetHandle();
1160 if (curl ==
nullptr) {
1166 auto rv = op->
Setup(curl, *
this);
1187 m_op_map[curl] = {op, std::chrono::system_clock::now()};
1192 std::string modified_url;
1193 std::shared_ptr<CurlOptionsOp> options_op(
1205 curl = queue.GetHandle();
1206 if (curl ==
nullptr) {
1212 auto rv = options_op->Setup(curl, *
this);
1217 m_op_map[curl] = {options_op, std::chrono::system_clock::now()};
1218 OpRecord(*options_op, OpKind::Start);
1219 running_handles += 1;
1221 OpRecord(*op, OpKind::Start);
1224 auto mres = curl_multi_add_handle(multi_handle, curl);
1225 if (mres != CURLM_OK) {
1232 running_handles += 1;
1237 time_t now = time(NULL);
1238 time_t next_maintenance = last_maintenance + m_maintenance_period.load(std::memory_order_relaxed);
1239 if (now >= next_maintenance) {
1241 m_continue_queue->Expire();
1243 getthreadid(), running_handles);
1244 last_maintenance = now;
1247 std::vector<std::pair<int, CURL *>> expired_ops;
1248 for (
const auto &entry : broker_reqs) {
1249 if (entry.second.expiry < now) {
1250 expired_ops.emplace_back(entry.first, entry.second.curl);
1253 for (
const auto &entry : expired_ops) {
1254 auto iter = m_op_map.find(entry.second);
1255 if (iter == m_op_map.end()) {
1256 m_logger->
Warning(
kLogXrdClHttp,
"Found an expired curl handle with no corresponding operation!");
1260 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(iter->second.first.get())) !=
nullptr) {
1262 bool parent_op_failed =
false;
1263 if (parent_op->IsRedirect()) {
1265 if (parent_op->Redirect(target) == CurlOperation::RedirectAction::Fail) {
1267 if (iter != m_op_map.end()) {
1270 m_op_map.erase(iter);
1271 running_handles -= 1;
1273 parent_op_failed =
true;
1275 OpRecord(*parent_op, OpKind::Start);
1278 OpRecord(*parent_op, OpKind::Start);
1280 if (!parent_op_failed){
1286 iter->second.first->ReleaseHandle();
1287 OpRecord(*(iter->second.first), OpKind::ConncallTimeout);
1288 m_op_map.erase(entry.second);
1289 curl_easy_cleanup(entry.second);
1290 running_handles -= 1;
1292 broker_reqs.erase(entry.first);
1293 m_conncall_timeout.fetch_add(1, std::memory_order_relaxed);
1301 waitfds.resize(3 + broker_reqs.size());
1303 waitfds[0].fd = queue.PollFD();
1304 waitfds[0].events = CURL_WAIT_POLLIN;
1305 waitfds[0].revents = 0;
1306 waitfds[1].fd = m_continue_queue->PollFD();
1307 waitfds[1].events = CURL_WAIT_POLLIN;
1308 waitfds[1].revents = 0;
1309 waitfds[2].fd = m_shutdown_pipe_r;
1310 waitfds[2].revents = 0;
1311 waitfds[2].events = CURL_WAIT_POLLIN | CURL_WAIT_POLLPRI;
1314 for (
const auto &entry : broker_reqs) {
1315 waitfds[idx].fd = entry.first;
1316 waitfds[idx].events = CURL_WAIT_POLLIN|CURL_WAIT_POLLPRI;
1317 waitfds[idx].revents = 0;
1322 curl_multi_timeout(multi_handle, &timeo);
1326 if (running_handles && timeo == -1) {
1331 mres = curl_multi_wait(multi_handle, &waitfds[0], waitfds.size(), 50,
nullptr);
1337 mres = curl_multi_wait(multi_handle, &waitfds[0], waitfds.size(), 50,
nullptr);
1339 if (mres != CURLM_OK) {
1344 for (
const auto &entry : waitfds) {
1346 if (waitfds[0].fd == entry.fd || waitfds[1].fd == entry.fd) {
1350 if ((waitfds[2].fd == entry.fd) && entry.revents) {
1351 want_shutdown =
true;
1354 if ((entry.revents & CURL_WAIT_POLLIN) != CURL_WAIT_POLLIN) {
1357 auto handle = broker_reqs[entry.fd].curl;
1358 auto iter = m_op_map.find(handle);
1359 if (iter == m_op_map.end()) {
1360 m_logger->
Warning(
kLogXrdClHttp,
"Internal error: broker responded on FD %d but no corresponding curl operation", entry.fd);
1361 broker_reqs.erase(entry.fd);
1362 m_conncall_errors.fetch_add(1, std::memory_order_relaxed);
1366 auto result = iter->second.first->WaitSocketCallback(err);
1371 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(iter->second.first.get())) !=
nullptr) {
1373 bool parent_op_failed =
false;
1374 if (parent_op->IsRedirect()) {
1376 if (parent_op->Redirect(target) == CurlOperation::RedirectAction::Fail) {
1378 if (iter != m_op_map.end()) {
1381 m_op_map.erase(iter);
1382 running_handles -= 1;
1384 parent_op_failed =
true;
1386 OpRecord(*parent_op, OpKind::Start);
1389 OpRecord(*parent_op, OpKind::Start);
1391 if (!parent_op_failed){
1398 m_op_map.erase(handle);
1399 broker_reqs.erase(entry.fd);
1400 m_conncall_errors.fetch_add(1, std::memory_order_relaxed);
1401 running_handles -= 1;
1403 broker_reqs.erase(entry.fd);
1404 curl_multi_add_handle(multi_handle, handle);
1405 m_conncall_success.fetch_add(1, std::memory_order_relaxed);
1411 auto mres = curl_multi_perform(multi_handle, &still_running);
1412 if (mres == CURLM_CALL_MULTI_PERFORM) {
1414 }
else if (mres != CURLM_OK) {
1422 msg = curl_multi_info_read(multi_handle, &msgq);
1423 if (msg && (msg->msg == CURLMSG_DONE)) {
1424 if (!msg->easy_handle) {
1426 mres = CURLM_BAD_EASY_HANDLE;
1429 auto iter = m_op_map.find(msg->easy_handle);
1430 if (iter == m_op_map.end()) {
1431 m_logger->
Error(
kLogXrdClHttp,
"Logic error: got a callback for an entry that doesn't exist");
1432 mres = CURLM_BAD_EASY_HANDLE;
1435 auto op = iter->second.first;
1436 auto res = msg->data.result;
1437 bool keep_handle =
false;
1438 bool waiting_on_callout =
false;
1439 if (res == CURLE_OK) {
1441 OpRecord(*op, OpKind::Finish);
1450 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(op.get())) !=
nullptr) {
1452 bool parent_op_failed =
false;
1453 if (parent_op->IsRedirect()) {
1455 if (parent_op->Redirect(target) == CurlOperation::RedirectAction::Fail) {
1458 running_handles -= 1;
1459 parent_op_failed =
true;
1461 OpRecord(*parent_op, OpKind::Start);
1464 OpRecord(*parent_op, OpKind::Start);
1467 if (!parent_op_failed) {
1472 queue.RecycleHandle(iter->first);
1476 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(op.get()))) {
1482 OpRecord(*op, OpKind::Start);
1484 curl_multi_remove_handle(multi_handle, iter->first);
1485 queue.RecycleHandle(iter->first);
1493 case CurlOperation::RedirectAction::Fail:
1502 keep_handle =
false;
1504 case CurlOperation::RedirectAction::Reinvoke:
1510 OpRecord(*op, OpKind::Start);
1513 case CurlOperation::RedirectAction::ReinvokeAfterAllow:
1520 std::string modified_url;
1523 std::shared_ptr<CurlOperation> new_op(options_op);
1524 auto curl = queue.GetHandle();
1525 if (curl ==
nullptr) {
1528 keep_handle =
false;
1529 options_op =
nullptr;
1532 OpRecord(*new_op, OpKind::Start);
1534 auto rv = new_op->Setup(curl, *
this);
1537 keep_handle =
false;
1538 options_op =
nullptr;
1542 m_logger->
Debug(
kLogXrdClHttp,
"Unable to setup the curl handle for the OPTIONS operation");
1543 new_op->Fail(
XrdCl::errInternal, ENOMEM,
"Failed to setup the curl handle for the OPTIONS operation");
1545 keep_handle =
false;
1548 new_op->SetContinueQueue(m_continue_queue);
1549 m_op_map[curl] = {new_op, std::chrono::system_clock::now()};
1550 auto mres = curl_multi_add_handle(multi_handle, curl);
1551 if (mres != CURLM_OK) {
1552 m_logger->
Debug(
kLogXrdClHttp,
"Unable to add OPTIONS operation to the curl multi-handle: %s", curl_multi_strerror(mres));
1557 running_handles += 1;
1558 m_logger->
Debug(
kLogXrdClHttp,
"Invoking the OPTIONS operation before redirect to %s", target.c_str());
1565 if ((waiting_on_callout = callout_socket >= 0)) {
1566 auto expiry = time(
nullptr) + 20;
1567 m_logger->
Debug(
kLogXrdClHttp,
"Creating a callout wait request on socket %d", callout_socket);
1568 broker_reqs[callout_socket] = {iter->first, expiry};
1569 m_conncall_req.fetch_add(1, std::memory_order_relaxed);
1571 }
else if (options_op) {
1576 curl_multi_remove_handle(multi_handle, iter->first);
1577 if (!waiting_on_callout && !options_op) {
1578 curl_multi_add_handle(multi_handle, iter->first);
1580 }
else if (!options_op) {
1584 queue.RecycleHandle(iter->first);
1593 int wait_socket = -1;
1595 m_logger->
Error(
kLogXrdClHttp,
"Failed to start broker-based connection: %s", err.c_str());
1597 keep_handle =
false;
1599 curl_multi_remove_handle(multi_handle, iter->first);
1600 auto expiry = time(
nullptr) + 20;
1601 m_logger->
Debug(
kLogXrdClHttp,
"Curl operation requires a new TCP socket; waiting for callout to respond on socket %d", wait_socket);
1602 broker_reqs[wait_socket] = {iter->first, expiry};
1603 m_conncall_req.fetch_add(1, std::memory_order_relaxed);
1606 if (res == CURLE_ABORTED_BY_CALLBACK || res == CURLE_WRITE_ERROR) {
1610 case CurlOperation::OpError::ErrHeaderTimeout:
1611 #ifdef HAVE_XPROTOCOL_TIMEREXPIRED
1618 case CurlOperation::OpError::ErrCallback: {
1624 case CurlOperation::OpError::ErrOperationTimeout:
1626 OpRecord(*op, op->
IsPaused() ? OpKind::ClientTimeout : OpKind::ServerTimeout);
1628 case CurlOperation::OpError::ErrTransferSlow:
1630 OpRecord(*op, OpKind::ServerTimeout);
1632 case CurlOperation::OpError::ErrTransferClientStall:
1634 OpRecord(*op, OpKind::ClientTimeout);
1636 case CurlOperation::OpError::ErrTransferStall:
1638 OpRecord(*op, OpKind::ServerTimeout);
1640 case CurlOperation::OpError::ErrNone:
1646 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(op.get())) !=
nullptr) {
1648 bool parent_op_failed =
false;
1649 if (parent_op->IsRedirect()) {
1651 if (parent_op->Redirect(target) == CurlOperation::RedirectAction::Fail) {
1653 if (iter != m_op_map.end()) {
1656 m_op_map.erase(iter);
1657 running_handles -= 1;
1659 parent_op_failed =
true;
1661 OpRecord(*parent_op, OpKind::Start);
1664 OpRecord(*parent_op, OpKind::Start);
1666 if (!parent_op_failed){
1673 const char *curl_easy_err = curl_easy_strerror(res);
1674 const std::string fail_err = !curl_err.empty() ? curl_err : curl_easy_err;
1675 m_logger->
Debug(
kLogXrdClHttp,
"Curl generated an error: %s (%d)", fail_err.c_str(), res);
1676 op->
Fail(xrdCode.first, xrdCode.second, fail_err);
1679 if ((options_op =
dynamic_cast<CurlOptionsOp*
>(op.get())) !=
nullptr) {
1681 bool parent_op_failed =
false;
1682 if (parent_op->IsRedirect()) {
1684 if (parent_op->Redirect(target) == CurlOperation::RedirectAction::Fail) {
1686 if (iter != m_op_map.end()) {
1689 m_op_map.erase(iter);
1690 running_handles -= 1;
1692 parent_op_failed =
true;
1695 if (!parent_op_failed){
1703 curl_multi_remove_handle(multi_handle, iter->first);
1704 if (res != CURLE_OK) {
1705 curl_easy_cleanup(iter->first);
1707 for (
auto &req : broker_reqs) {
1708 if (req.second.curl == iter->first) {
1709 m_logger->
Warning(
kLogXrdClHttp,
"Curl handle finished while a broker operation was outstanding");
1710 m_conncall_errors.fetch_add(1, std::memory_order_relaxed);
1713 m_op_map.erase(iter);
1714 running_handles -= 1;
1720 for (
auto map_entry : m_op_map) {
1725 if (multi_handle && map_entry.first) curl_multi_remove_handle(multi_handle, map_entry.first);
1728 m_queue->ReleaseHandles();
1729 curl_multi_cleanup(multi_handle);
1733 CurlWorker::Shutdown()
1735 m_queue->Shutdown();
1736 if (m_shutdown_pipe_w == -1) {
1740 close(m_shutdown_pipe_w);
1741 m_shutdown_pipe_w = -1;
1747 std::unique_lock lk(m_worker_stats_mutex);
1748 m_workers_last_completed_cycle[m_stats_offset] =
nullptr;
1749 m_workers_oldest_op[m_stats_offset] =
nullptr;
1755 CurlWorker::ShutdownAll()
1757 std::unique_lock lock(m_workers_mutex);
1758 for (
auto &worker : m_workers) {
1763 CurlWorker::initcontrol::initcontrol()
1765 curl_global_init(CURL_GLOBAL_DEFAULT);
1768 CurlWorker::initcontrol::~initcontrol()
1771 curl_global_cleanup();
std::pair< uint16_t, uint32_t > CurlCodeConvert(CURLcode res)
std::string obfuscateAuth(const std::string &input)
ssize_t read(int fildes, void *buf, size_t nbyte)
void getline(uchar *buff, int blen)
int emsg(int rc, char *msg)
bool Set(ChecksumType ctype, const std::array< unsigned char, g_max_checksum_length > &value)
virtual void OptionsDone()
int GetStatusCode() const
bool FinishSetup(CURL *curl)
const std::string & GetUrl() const
std::pair< XErrorCode, std::string > GetCallbackError() const
virtual void Fail(uint16_t errCode, uint32_t errNum, const std::string &)
static const std::string GetVerbString(HttpVerb)
bool UseConnectionCallout()
virtual HttpVerb GetVerb() const =0
std::string GetCurlErrorMessage() const
virtual void ReleaseHandle()
virtual bool RequiresOptions() const
static void CleanupDnsCache()
bool GetTriedBoker() const
std::tuple< uint64_t, std::chrono::steady_clock::duration, std::chrono::steady_clock::duration, std::chrono::steady_clock::duration > StatisticsReset()
virtual bool ContinueHandle()
std::string GetStatusMessage() const
CURL * GetCurlHandle() const
CreateConnCalloutType GetConnCalloutFunc() const
virtual RedirectAction Redirect(std::string &target)
virtual void SetContinueQueue(std::shared_ptr< XrdClHttp::HandlerQueue > queue)
bool StartConnectionCallout(std::string &err)
virtual bool Setup(CURL *curl, CurlWorker &)
CURL * GetParentCurlHandle() const
void ReleaseHandle() override
std::shared_ptr< CurlOperation > GetOperation() const
std::tuple< std::string, std::string > ClientX509CertKeyFile() const
static void RunStatic(CurlWorker *myself)
void Start(std::unique_ptr< XrdClHttp::CurlWorker > self, std::thread tid)
static std::string GetMonitoringJson()
HandlerQueue(unsigned max_pending_ops)
void RecycleHandle(CURL *)
static std::string_view GetUrlKey(const std::string &url, std::string &modified_url)
static Log * GetLog()
Get default log.
static Env * GetEnv()
Get default client environment.
bool GetInt(const std::string &key, int &value)
@ DumpMsg
print details of the request and responses
void Error(uint64_t topic, const char *format,...)
Report an error.
void Warning(uint64_t topic, const char *format,...)
Report a warning.
void Debug(uint64_t topic, const char *format,...)
Print a debug message.
std::pair< uint16_t, uint32_t > HTTPStatusConvert(unsigned status)
CURL * GetHandle(bool verbose)
bool HTTPStatusIsError(unsigned status)
std::string_view ltrim_view(const std::string_view &input_view)
std::string_view trim_view(const std::string_view &input_view)
const uint16_t errUnknown
Unknown error.
const uint16_t errInvalidAddr
const uint16_t errRedirectLimit
const uint16_t errErrorResponse
const uint16_t errTlsError
const uint16_t errOperationExpired
const uint16_t errLoginFailed
const uint16_t errDataError
data is corrupted
const uint16_t errInternal
Internal error.
const uint16_t errInvalidArgs
const uint16_t errConnectionError
const uint16_t errNotSupported
const uint16_t errSocketError
const uint16_t errCorruptedHeader
const uint16_t errNone
No error.
const uint64_t kLogXrdClHttp