zeek/src/logging/writers/ascii/Ascii.h
Tim Wojtulewicz a1d121e5aa Use std::string/string_view versions of starts_with/ends_with where appropriate
The util:: versions of these methods remain as a thin wrapper around them so
they can be used with const char* arguments. Otherwise callers have to manually
make string_view objects from the input.
s Please enter the commit message for your changes. Lines starting
2025-07-17 09:08:54 -07:00

83 lines
2.4 KiB
C++

// See the file "COPYING" in the main distribution directory for copyright.
//
// Log writer for delimiter-separated ASCII logs.
#pragma once
#include <zlib.h>
#include <string>
#include "zeek/Desc.h"
#include "zeek/logging/WriterBackend.h"
#include "zeek/threading/Formatter.h"
namespace zeek::plugin::detail::Zeek_AsciiWriter {
class Plugin;
}
namespace zeek::logging::writer::detail {
class Ascii : public WriterBackend {
public:
explicit Ascii(WriterFrontend* frontend);
~Ascii() override;
static std::string LogExt();
static WriterBackend* Instantiate(WriterFrontend* frontend) { return new Ascii(frontend); }
protected:
bool DoInit(const WriterInfo& info, int num_fields, const threading::Field* const* fields) override;
bool DoWrite(int num_fields, const threading::Field* const* fields, threading::Value** vals) override;
bool DoSetBuf(bool enabled) override;
bool DoRotate(const char* rotated_path, double open, double close, bool terminating) override;
bool DoFlush(double network_time) override;
bool DoFinish(double network_time) override;
bool DoHeartbeat(double network_time, double current_time) override;
private:
friend class plugin::detail::Zeek_AsciiWriter::Plugin;
static void RotateLeftoverLogs();
bool IsSpecial(const std::string& path) { return path.starts_with("/dev/"); }
bool WriteHeader(const std::string& path);
bool WriteHeaderField(const std::string& key, const std::string& value);
void CloseFile(double t);
std::string Timestamp(double t); // Uses current time if t is zero.
void InitConfigOptions();
bool InitFilterOptions();
bool InitFormatter();
bool InternalWrite(int fd, const char* data, int len);
bool InternalClose(int fd);
int fd;
gzFile gzfile;
std::string fname;
ODesc desc;
bool ascii_done;
// Options set from the script-level.
bool output_to_stdout;
bool include_meta;
bool tsv;
std::string separator;
std::string set_separator;
std::string empty_field;
std::string unset_field;
std::string meta_prefix;
int gzip_level; // level > 0 enables gzip compression
std::string gzip_file_extension;
bool use_json;
bool enable_utf_8;
std::string json_timestamps;
bool json_include_unset_fields;
std::string logdir;
threading::Formatter* formatter;
bool init_options;
};
} // namespace zeek::logging::writer::detail