logging/sqlite: Recognize Log::default_logdir and place files there if set

This commit is contained in:
Arne Welzel 2022-07-06 12:13:32 +02:00
parent aaa47a709c
commit 93584c7c7f
5 changed files with 78 additions and 2 deletions

View file

@ -10,6 +10,7 @@
#include "zeek/logging/writers/sqlite/sqlite.bif.h" #include "zeek/logging/writers/sqlite/sqlite.bif.h"
#include "zeek/threading/SerialTypes.h" #include "zeek/threading/SerialTypes.h"
#include "zeek/util.h"
using namespace std; using namespace std;
using zeek::threading::Field; using zeek::threading::Field;
@ -128,8 +129,11 @@ bool SQLite::DoInit(const WriterInfo& info, int arg_num_fields, const Field* con
num_fields = arg_num_fields; num_fields = arg_num_fields;
fields = arg_fields; fields = arg_fields;
string fullpath(info.path); auto fullpath = zeek::filesystem::path(
fullpath.append(".sqlite"); zeek::id::find_const<StringVal>("Log::default_logdir")->ToStdString());
fullpath /= info.path;
fullpath += ".sqlite";
string tablename; string tablename;
WriterInfo::config_map::const_iterator it = info.config.find("tablename"); WriterInfo::config_map::const_iterator it = info.config.find("tablename");

View file

@ -0,0 +1,3 @@
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
./logs/reporter.sqlite
./logs/test.sqlite

View file

@ -0,0 +1,21 @@
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1024
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|0
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1025
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|1
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1026
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|2
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1027
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|3
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1028
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|4
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1029
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|5
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1030
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|6
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1031
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|7
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1032
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|8
XXXXXXXXXX.XXXXXX|10.0.0.1|20|10.0.0.2|1033
XXXXXXXXXX.XXXXXX|10.0.0.2|20|10.0.0.3|9

View file

@ -0,0 +1,2 @@
### BTest baseline data generated by btest-diff. Do not edit. Use "btest -U/-u" to update. Requires BTest >= 0.63.
XXXXXXXXXX.XXXXXX test/Log::WRITER_SQLITE: tablename configuration option not found. Defaulting to path test

View file

@ -0,0 +1,46 @@
#
# @TEST-REQUIRES: which sqlite3
# @TEST-REQUIRES: has-writer Zeek::SQLiteWriter
# @TEST-GROUP: sqlite
#
# @TEST-EXEC: mkdir logs
# @TEST-EXEC: zeek -b -r ${TRACES}/rotation.trace %INPUT >zeek.out 2>&1
# @TEST-EXEC: ls ./logs/* > ls.logs
# @TEST-EXEC: sqlite3 ./logs/test.sqlite 'select * from test' > test.select
# @TEST-EXEC: btest-diff test.select
# @TEST-EXEC: btest-diff ls.logs
# @TEST-EXEC: btest-diff zeek.out
#
# @TEST-DOC: Configure Log::default_writer, Log::default_logdir and ensure the test.sqlite database is in ./logs
redef Log::default_writer = Log::WRITER_SQLITE;
redef Log::default_logdir = "./logs";
# Also enable log-rotation, but it has no effect on sqlite.
redef Log::default_rotation_interval = 1hr;
redef Log::default_rotation_postprocessor_cmd = "echo";
redef LogSQLite::unset_field = "(unset)";
module Test;
export {
# Create a new ID for our log stream
redef enum Log::ID += { LOG };
# Define a record with all the columns the log file can have.
# (I'm using a subset of fields from ssh-ext for demonstration.)
type Log: record {
t: time;
id: conn_id; # Will be rolled out into individual columns.
} &log;
}
event zeek_init()
{
Log::create_stream(Test::LOG, [$columns=Log]);
}
event new_connection(c: connection)
{
Log::write(Test::LOG, [$t=network_time(), $id=c$id]);
}