mirror of https://github.com/tstack/lnav
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.
121 lines
3.9 KiB
C++
121 lines
3.9 KiB
C++
/**
|
|
* 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 "piper.file.hh"
|
|
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
|
|
#include "base/injector.hh"
|
|
#include "base/lnav_log.hh"
|
|
#include "base/paths.hh"
|
|
#include "piper.looper.cfg.hh"
|
|
|
|
namespace lnav {
|
|
namespace piper {
|
|
|
|
const char HEADER_MAGIC[4] = {'L', 0, 'N', 1};
|
|
|
|
const ghc::filesystem::path&
|
|
storage_path()
|
|
{
|
|
static auto INSTANCE = lnav::paths::workdir() / "piper";
|
|
|
|
return INSTANCE;
|
|
}
|
|
|
|
std::optional<auto_buffer>
|
|
read_header(int fd, const char* first8)
|
|
{
|
|
if (memcmp(first8, HEADER_MAGIC, sizeof(HEADER_MAGIC)) != 0) {
|
|
log_trace("first 4 bytes are not a piper header: %02x%02x%02x%02x",
|
|
first8[0],
|
|
first8[1],
|
|
first8[2],
|
|
first8[3]);
|
|
return std::nullopt;
|
|
}
|
|
|
|
uint32_t meta_size = ntohl(*((uint32_t*) &first8[4]));
|
|
|
|
auto meta_buf = auto_buffer::alloc(meta_size);
|
|
if (meta_buf.in() == nullptr) {
|
|
log_error("failed to alloc %d bytes for header", meta_size);
|
|
return std::nullopt;
|
|
}
|
|
auto meta_prc = pread(fd, meta_buf.in(), meta_size, 8);
|
|
if (meta_prc != meta_size) {
|
|
log_error("failed to read piper header: %s", strerror(errno));
|
|
return std::nullopt;
|
|
}
|
|
meta_buf.resize(meta_size);
|
|
|
|
return meta_buf;
|
|
}
|
|
|
|
std::optional<std::string>
|
|
multiplex_id_for_line(string_fragment line)
|
|
{
|
|
const auto& cfg = injector::get<const config&>();
|
|
auto md = lnav::pcre2pp::match_data::unitialized();
|
|
|
|
for (const auto& demux_pair : cfg.c_demux_definitions) {
|
|
const auto& df = demux_pair.second;
|
|
|
|
if (!df.dd_enabled) {
|
|
continue;
|
|
}
|
|
|
|
log_info("attempting to demux using: %s", demux_pair.first.c_str());
|
|
md = df.dd_pattern.pp_value->create_match_data();
|
|
if (df.dd_pattern.pp_value->capture_from(line)
|
|
.into(md)
|
|
.matches()
|
|
.ignore_error())
|
|
{
|
|
log_info(" demuxer pattern matched");
|
|
if (!md[df.dd_muxid_capture_index].has_value()) {
|
|
log_info(" however, mux_id was not captured");
|
|
continue;
|
|
}
|
|
if (!md[df.dd_body_capture_index].has_value()) {
|
|
log_info(" however, body was not captured");
|
|
continue;
|
|
}
|
|
log_info(" and required captures were found, using demuxer");
|
|
return demux_pair.first;
|
|
}
|
|
}
|
|
|
|
return std::nullopt;
|
|
}
|
|
|
|
} // namespace piper
|
|
} // namespace lnav
|