diff options
Diffstat (limited to '')
| -rw-r--r-- | src/url_handler.cc | 259 |
1 files changed, 259 insertions, 0 deletions
diff --git a/src/url_handler.cc b/src/url_handler.cc new file mode 100644 index 0000000..a909253 --- /dev/null +++ b/src/url_handler.cc @@ -0,0 +1,259 @@ +/** + * Copyright (c) 2023, 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. + */ + +#include "url_handler.hh" + +#include <curl/curl.h> + +#include "base/fs_util.hh" +#include "base/injector.hh" +#include "base/paths.hh" +#include "lnav.hh" +#include "service_tags.hh" +#include "url_handler.cfg.hh" + +namespace lnav { +namespace url_handler { + +void +looper::handler_looper::loop_body() +{ + pollfd pfd[1]; + + pfd[0].events = POLLIN; + pfd[0].fd = this->hl_line_buffer.get_fd(); + pfd[0].revents = 0; + + log_debug("doing url handler poll"); + auto prc = poll(pfd, 1, 100); + log_debug("poll rc %d", prc); + if (prc > 0) { + auto load_res + = this->hl_line_buffer.load_next_line(this->hl_last_range); + + if (load_res.isErr()) { + log_error("failed to load next line: %s", + load_res.unwrapErr().c_str()); + this->s_looping = false; + } else { + auto li = load_res.unwrap(); + + log_debug("li %d %d:%d", + li.li_partial, + li.li_file_range.fr_offset, + li.li_file_range.fr_size); + if (!li.li_partial && !li.li_file_range.empty()) { + auto read_res + = this->hl_line_buffer.read_range(li.li_file_range); + + if (read_res.isErr()) { + log_error("cannot read line: %s", + read_res.unwrapErr().c_str()); + } else { + auto cmd = trim(to_string(read_res.unwrap())); + log_debug("url handler command: %s", cmd.c_str()); + + isc::to<main_looper&, services::main_t>().send( + [cmd](auto& mlooper) { + auto exec_res + = execute_any(lnav_data.ld_exec_context, cmd); + if (exec_res.isErr()) { + auto um = exec_res.unwrapErr(); + log_error( + "%s", + um.to_attr_line().get_string().c_str()); + } + }); + } + this->hl_last_range = li.li_file_range; + } + } + + if (this->hl_line_buffer.is_pipe_closed()) { + log_info("URL handler finished"); + this->s_looping = false; + } + } +} + +Result<void, lnav::console::user_message> +looper::open(std::string url) +{ + const auto& cfg = injector::get<const config&>(); + + log_info("open request for URL: %s", url.c_str()); + + auto* cu = curl_url(); + auto set_rc = curl_url_set( + cu, CURLUPART_URL, url.c_str(), CURLU_NON_SUPPORT_SCHEME); + if (set_rc != CURLUE_OK) { + return Err( + lnav::console::user_message::error( + attr_line_t("invalid URL: ").append(lnav::roles::file(url))) + .with_reason(curl_url_strerror(set_rc))); + } + + char* scheme_part; + auto get_rc = curl_url_get(cu, CURLUPART_SCHEME, &scheme_part, 0); + if (get_rc != CURLUE_OK) { + return Err(lnav::console::user_message::error( + attr_line_t("cannot get scheme from URL: ") + .append(lnav::roles::file(url))) + .with_reason(curl_url_strerror(set_rc))); + } + + auto proto_iter = cfg.c_schemes.find(scheme_part); + if (proto_iter == cfg.c_schemes.end()) { + return Err(lnav::console::user_message::error( + attr_line_t("no defined handler for URL scheme: ") + .append(lnav::roles::file(scheme_part))) + .with_reason(curl_url_strerror(set_rc))); + } + + log_info("found URL handler: %s", + proto_iter->second.p_handler.pp_value.c_str()); + auto err_pipe_res = auto_pipe::for_child_fd(STDERR_FILENO); + if (err_pipe_res.isErr()) { + return Err( + lnav::console::user_message::error( + attr_line_t("cannot open URL: ").append(lnav::roles::file(url))) + .with_reason(err_pipe_res.unwrapErr())); + } + auto err_pipe = err_pipe_res.unwrap(); + auto out_pipe_res = auto_pipe::for_child_fd(STDOUT_FILENO); + if (out_pipe_res.isErr()) { + return Err( + lnav::console::user_message::error( + attr_line_t("cannot open URL: ").append(lnav::roles::file(url))) + .with_reason(out_pipe_res.unwrapErr())); + } + auto out_pipe = out_pipe_res.unwrap(); + auto child_pid_res = lnav::pid::from_fork(); + if (child_pid_res.isErr()) { + return Err( + lnav::console::user_message::error( + attr_line_t("cannot open URL: ").append(lnav::roles::file(url))) + .with_reason(child_pid_res.unwrapErr())); + } + + auto child_pid = child_pid_res.unwrap(); + + out_pipe.after_fork(child_pid.in()); + err_pipe.after_fork(child_pid.in()); + + auto name = proto_iter->second.p_handler.pp_value; + if (child_pid.in_child()) { + auto dev_null = ::open("/dev/null", O_RDONLY | O_CLOEXEC); + + dup2(dev_null, STDIN_FILENO); + + char* host_part = nullptr; + curl_url_get(cu, CURLUPART_HOST, &host_part, 0); + std::string host_part_str; + if (host_part != nullptr) { + host_part_str = host_part; + } + + auto source_path = ghc::filesystem::path{ + proto_iter->second.p_handler.pp_location.sl_source.get()}; + auto new_path = lnav::filesystem::build_path({ + source_path.parent_path(), + lnav::paths::dotlnav() / "formats/default", + }); + setenv("PATH", new_path.c_str(), 1); + setenv("URL_HOSTNAME", host_part_str.c_str(), 1); + + const char* args[] = { + name.c_str(), + nullptr, + }; + + execvp(name.c_str(), (char**) args); + _exit(EXIT_FAILURE); + } + + auto error_queue = std::make_shared<std::vector<std::string>>(); + std::thread err_reader([err = std::move(err_pipe.read_end()), + name, + error_queue, + child_pid = child_pid.in()]() mutable { + line_buffer lb; + file_range pipe_range; + bool done = false; + + log_debug("error reader"); + lb.set_fd(err); + while (!done) { + auto load_res = lb.load_next_line(pipe_range); + + if (load_res.isErr()) { + done = true; + } else { + auto li = load_res.unwrap(); + + pipe_range = li.li_file_range; + if (li.li_file_range.empty()) { + done = true; + } else { + lb.read_range(li.li_file_range) + .then([name, error_queue, child_pid](auto sbr) { + auto line_str = string_fragment( + sbr.get_data(), 0, sbr.length()) + .trim("\n"); + if (error_queue->size() < 5) { + error_queue->emplace_back(line_str.to_string()); + } + + log_debug("%s[%d]: %.*s", + name.c_str(), + child_pid, + line_str.length(), + line_str.data()); + }); + } + } + } + }); + err_reader.detach(); + + auto child = std::make_shared<handler_looper>( + url, std::move(child_pid), std::move(out_pipe.read_end())); + this->s_children.add_child_service(child); + this->l_children[url] = child; + + return Ok(); +} + +void +looper::close(std::string url) +{ +} + +} // namespace url_handler +} // namespace lnav |