Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-17 14:08:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add basic SQLite storage backend

Browse source
This commit is contained in:
Tim Wojtulewicz 2024-01-27 13:34:20 -07:00
parent 7ad6a05f5b
commit 9d1eef3fbc
15 changed files with 388 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

56
testing/btest/scripts/base/frameworks/storage/sqlite-basic.zeek Normal file
View file

@ -0,0 +1,56 @@
# @TEST-DOC: Basic functionality for storage: opening/closing an sqlite backend, storing/retrieving/erasing basic data
# @TEST-EXEC: zeek -b %INPUT > out
# @TEST-EXEC: btest-diff out
# @TEST-EXEC: btest-diff .stderr
@load base/frameworks/storage
@load policy/frameworks/storage/backend/sqlite
# Create a typename here that can be passed down into open_backend.
type str: string;
event zeek_init() {
# Create a database file in the .tmp directory with a 'testing' table
local opts : Storage::Backend::SQLite::Options;
opts$database_path = "test.sqlite";
opts$table_name = "testing";
local key = "key1111";
local value = "value";
# Test inserting/retrieving a key/value pair that we know won't be in
# the backend yet.
local b = Storage::open_backend(Storage::SQLITE, opts, str, str);
local res = Storage::put(b, [$key=key, $value=value, $overwrite=T, $async_mode=F]);
print res;
local res2 = Storage::get(b, key, F);
print res2;
# Test overwriting a value with put()
local value2 = "value2";
local res3 = Storage::put(b, [$key=key, $value=value2, $overwrite=T, $async_mode=F]);
print res3;
local res4 = Storage::get(b, key, F);
print res4;
# Test erasing a key and getting a "false" result
local res5 = Storage::erase(b, key, F);
print res5;
local res6 = Storage::get(b, key, F);
if ( ! res6 as bool ) {
print "got empty result";
}
# Insert something back into the database to test reopening
Storage::put(b, [$key=key, $value=value2, $overwrite=T, $async_mode=F]);
Storage::close_backend(b);
# Test reopening the same database and getting the data back out of it
local b2 = Storage::open_backend(Storage::SQLITE, opts, str, str);
local res7 = Storage::get(b2, key, F);
print res7;
}

21
testing/btest/scripts/base/frameworks/storage/sqlite-error-handling.zeek Normal file
View file

@ -0,0 +1,21 @@
# @TEST-DOC: Tests various error handling scenarios for the storage framework
# @TEST-EXEC: zeek -b %INPUT > out
# @TEST-EXEC: TEST_DIFF_CANONIFIER=$SCRIPTS/diff-remove-abspath btest-diff out
# @TEST-EXEC: TEST_DIFF_CANONIFIER=$SCRIPTS/diff-remove-abspath btest-diff .stderr
@load base/frameworks/storage
@load base/frameworks/reporter
@load policy/frameworks/storage/backend/sqlite
# Create a typename here that can be passed down into open_backend.
type str: string;
event zeek_init() {
# Test opening a database with an invalid path
local opts : Storage::Backend::SQLite::Options;
opts$database_path = "/this/path/should/not/exist/test.sqlite";
opts$table_name = "testing";
# This should report an error in .stderr and reporter.log
local b = Storage::open_backend(Storage::SQLITE, opts, str, str);
}