summaryrefslogtreecommitdiffstats
path: root/src/curl_looper.hh
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/curl_looper.hh226
1 files changed, 226 insertions, 0 deletions
diff --git a/src/curl_looper.hh b/src/curl_looper.hh
new file mode 100644
index 0000000..2b350ad
--- /dev/null
+++ b/src/curl_looper.hh
@@ -0,0 +1,226 @@
+/**
+ * Copyright (c) 2015, Timothy Stack
+ *
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ * * Neither the name of Timothy Stack nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * @file curl_looper.hh
+ */
+
+#ifndef curl_looper_hh
+#define curl_looper_hh
+
+#include <atomic>
+#include <map>
+#include <string>
+#include <utility>
+#include <vector>
+
+#include "base/isc.hh"
+#include "base/lnav.console.hh"
+#include "base/result.h"
+#include "config.h"
+
+#if !defined(HAVE_LIBCURL)
+
+typedef int CURLcode;
+
+class curl_request {
+public:
+ curl_request(const std::string& name){};
+};
+
+class curl_looper : public isc::service<curl_looper> {
+public:
+ void start(){};
+ void stop(){};
+ void add_request(std::shared_ptr<curl_request> cr){};
+ void close_request(const std::string& name){};
+ void process_all(){};
+};
+
+#else
+# include <condition_variable>
+# include <mutex>
+# include <thread>
+
+# include <curl/curl.h>
+
+# include "base/auto_mem.hh"
+# include "base/lnav_log.hh"
+# include "base/time_util.hh"
+
+class curl_request {
+public:
+ curl_request(std::string name)
+ : cr_name(std::move(name)), cr_handle(curl_easy_cleanup)
+ {
+ this->cr_handle.reset(curl_easy_init());
+ curl_easy_setopt(this->cr_handle, CURLOPT_NOSIGNAL, 1);
+ curl_easy_setopt(
+ this->cr_handle, CURLOPT_ERRORBUFFER, this->cr_error_buffer);
+ curl_easy_setopt(this->cr_handle, CURLOPT_DEBUGFUNCTION, debug_cb);
+ curl_easy_setopt(this->cr_handle, CURLOPT_DEBUGDATA, this);
+ curl_easy_setopt(this->cr_handle, CURLOPT_VERBOSE, 1);
+ if (getenv("SSH_AUTH_SOCK") != nullptr) {
+ curl_easy_setopt(this->cr_handle,
+ CURLOPT_SSH_AUTH_TYPES,
+# ifdef CURLSSH_AUTH_AGENT
+ CURLSSH_AUTH_AGENT |
+# endif
+ CURLSSH_AUTH_PASSWORD);
+ }
+ }
+
+ virtual ~curl_request() = default;
+
+ const std::string& get_name() const
+ {
+ return this->cr_name;
+ }
+
+ virtual void close()
+ {
+ this->cr_open = false;
+ }
+
+ bool is_open() const
+ {
+ return this->cr_open;
+ }
+
+ CURL* get_handle() const
+ {
+ return this->cr_handle;
+ }
+
+ operator CURL*() const
+ {
+ return this->cr_handle;
+ }
+
+ int get_completions() const
+ {
+ return this->cr_completions;
+ }
+
+ virtual long complete(CURLcode result)
+ {
+ double total_time = 0, download_size = 0, download_speed = 0;
+
+ this->cr_completions += 1;
+ curl_easy_getinfo(this->cr_handle, CURLINFO_TOTAL_TIME, &total_time);
+ log_debug("%s: total_time=%f", this->cr_name.c_str(), total_time);
+ curl_easy_getinfo(
+ this->cr_handle, CURLINFO_SIZE_DOWNLOAD, &download_size);
+ log_debug("%s: download_size=%f", this->cr_name.c_str(), download_size);
+ curl_easy_getinfo(
+ this->cr_handle, CURLINFO_SPEED_DOWNLOAD, &download_speed);
+ log_debug(
+ "%s: download_speed=%f", this->cr_name.c_str(), download_speed);
+
+ return -1;
+ }
+
+ Result<std::string, CURLcode> perform()
+ {
+ std::string response;
+
+ curl_easy_setopt(this->get_handle(), CURLOPT_WRITEFUNCTION, string_cb);
+ curl_easy_setopt(this->get_handle(), CURLOPT_WRITEDATA, &response);
+
+ auto rc = curl_easy_perform(this->get_handle());
+ if (rc == CURLE_OK) {
+ return Ok(response);
+ }
+
+ return Err(rc);
+ }
+
+ long get_response_code() const
+ {
+ long retval;
+
+ curl_easy_getinfo(this->get_handle(), CURLINFO_RESPONSE_CODE, &retval);
+ return retval;
+ }
+
+protected:
+ static int debug_cb(
+ CURL* handle, curl_infotype type, char* data, size_t size, void* userp);
+
+ static size_t string_cb(void* data, size_t size, size_t nmemb, void* userp);
+
+ const std::string cr_name;
+ bool cr_open{true};
+ auto_mem<CURL> cr_handle;
+ char cr_error_buffer[CURL_ERROR_SIZE];
+ int cr_completions{0};
+};
+
+class curl_looper : public isc::service<curl_looper> {
+public:
+ curl_looper() : cl_curl_multi(curl_multi_cleanup)
+ {
+ this->cl_curl_multi.reset(curl_multi_init());
+ }
+
+ void process_all();
+
+ void add_request(const std::shared_ptr<curl_request>& cr)
+ {
+ require(cr != nullptr);
+
+ this->cl_all_requests.emplace_back(cr);
+ this->cl_new_requests.emplace_back(cr);
+ }
+
+ void close_request(const std::string& name)
+ {
+ this->cl_close_requests.emplace_back(name);
+ }
+
+protected:
+ void loop_body() override;
+
+private:
+ void perform_io();
+ void check_for_new_requests();
+ void check_for_finished_requests();
+ void requeue_requests(mstime_t up_to_time);
+ std::chrono::milliseconds compute_timeout(
+ mstime_t current_time) const override;
+
+ auto_mem<CURLM> cl_curl_multi;
+ std::vector<std::shared_ptr<curl_request> > cl_all_requests;
+ std::vector<std::shared_ptr<curl_request> > cl_new_requests;
+ std::vector<std::string> cl_close_requests;
+ std::map<CURL*, std::shared_ptr<curl_request> > cl_handle_to_request;
+ std::vector<std::pair<mstime_t, std::shared_ptr<curl_request> > >
+ cl_poll_queue;
+};
+#endif
+
+#endif