mirror of
https://gitlab.nic.cz/knot/knot-dns.git
synced 2026-04-27 17:18:52 -04:00
117 lines
3.5 KiB
C
117 lines
3.5 KiB
C
/* Copyright (C) 2023 CZ.NIC, z.s.p.o. <knot-dns@labs.nic.cz>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <tap/basic.h>
|
|
#include <tap/files.h>
|
|
|
|
#include "knot/zone/timers.h"
|
|
#include "libknot/db/db_lmdb.h"
|
|
#include "libknot/dname.h"
|
|
#include "libknot/error.h"
|
|
|
|
static const zone_timers_t MOCK_TIMERS = {
|
|
.last_flush = 1474559960,
|
|
.next_refresh = 1474559961,
|
|
.last_refresh_ok = true,
|
|
.last_notified_serial = 123456,
|
|
.next_ds_check = 1474559962,
|
|
.next_ds_push = 1474559963,
|
|
.catalog_member = 1474559964,
|
|
.next_expire = 1474559965,
|
|
.last_master = { .sin6_family = AF_INET, .sin6_port = 53 },
|
|
.master_pin_hit = 1474559966,
|
|
};
|
|
|
|
static bool timers_eq(const zone_timers_t *val, const zone_timers_t *ref)
|
|
{
|
|
return val->last_flush == ref->last_flush &&
|
|
val->next_refresh == ref->next_refresh &&
|
|
val->last_refresh_ok == ref->last_refresh_ok &&
|
|
val->last_notified_serial == ref->last_notified_serial &&
|
|
val->next_ds_check == ref->next_ds_check &&
|
|
val->next_ds_push == ref->next_ds_push &&
|
|
val->catalog_member == ref->catalog_member &&
|
|
val->next_expire == ref->next_expire &&
|
|
sockaddr_cmp((struct sockaddr_storage *)&val->last_master,
|
|
(struct sockaddr_storage *)&ref->last_master, false) == 0 &&
|
|
val->master_pin_hit == ref->master_pin_hit;
|
|
}
|
|
|
|
static bool keep_all(const knot_dname_t *zone, void *data)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
static bool remove_all(const knot_dname_t *zone, void *data)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
plan_lazy();
|
|
assert(knot_db_lmdb_api());
|
|
|
|
char *dbid = test_mkdtemp();
|
|
if (!dbid) {
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
const knot_dname_t *zone = (uint8_t *)"\x7""example""\x3""com";
|
|
struct zone_timers timers = MOCK_TIMERS;
|
|
|
|
// Create database
|
|
knot_lmdb_db_t _db = { 0 }, *db = &_db;
|
|
knot_lmdb_init(db, dbid, 1024 * 1024, 0, NULL);
|
|
int ret = knot_lmdb_open(db);
|
|
ok(ret == KNOT_EOK && db != NULL, "open timers");
|
|
|
|
// Lookup nonexistent
|
|
ret = zone_timers_read(db, zone, &timers);
|
|
is_int(KNOT_ENOENT, ret, "zone_timer_read() nonexistent");
|
|
|
|
// Write timers
|
|
ret = zone_timers_write(db, zone, &timers);
|
|
is_int(KNOT_EOK, ret, "zone_timers_write()");
|
|
|
|
// Read timers
|
|
memset(&timers, 0, sizeof(timers));
|
|
ret = zone_timers_read(db, zone, &timers);
|
|
ok(ret == KNOT_EOK, "zone_timers_read()");
|
|
ok(timers_eq(&timers, &MOCK_TIMERS), "inconsistent timers");
|
|
|
|
// Sweep none
|
|
ret = zone_timers_sweep(db, keep_all, NULL);
|
|
is_int(KNOT_EOK, ret, "zone_timers_sweep() none");
|
|
ret = zone_timers_read(db, zone, &timers);
|
|
is_int(KNOT_EOK, ret, "zone_timers_read()");
|
|
|
|
// Sweep all
|
|
ret = zone_timers_sweep(db, remove_all, NULL);
|
|
is_int(KNOT_EOK, ret, "zone_timers_sweep() all");
|
|
ret = zone_timers_read(db, zone, &timers);
|
|
is_int(KNOT_ENOENT, ret, "zone_timers_read() nonexistent");
|
|
|
|
// Clean up.
|
|
knot_lmdb_deinit(db);
|
|
test_rm_rf(dbid);
|
|
free(dbid);
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|