// SPDX-License-Identifier: GPL-2.0 /* Copyright (C) 2022 Linutronix GmbH */ #include <test_progs.h> #include <network_helpers.h> #include "test_time_tai.skel.h" #include <time.h> #include <stdint.h> #define TAI_THRESHOLD 1000000000ULL /* 1s */ #define NSEC_PER_SEC 1000000000ULL static __u64 ts_to_ns(const struct timespec *ts) { return ts->tv_sec * NSEC_PER_SEC + ts->tv_nsec; } void test_time_tai(void) { struct __sk_buff skb = { .cb[0] = 0, .cb[1] = 0, .tstamp = 0, }; LIBBPF_OPTS(bpf_test_run_opts, topts, .data_in = &pkt_v4, .data_size_in = sizeof(pkt_v4), .ctx_in = &skb, .ctx_size_in = sizeof(skb), .ctx_out = &skb, .ctx_size_out = sizeof(skb), ); struct test_time_tai *skel; struct timespec now_tai; __u64 ts1, ts2, now; int ret, prog_fd; /* Open and load */ skel = test_time_tai__open_and_load(); if (!ASSERT_OK_PTR(skel, "tai_open")) return; /* Run test program */ prog_fd = bpf_program__fd(skel->progs.time_tai); ret = bpf_prog_test_run_opts(prog_fd, &topts); ASSERT_OK(ret, "test_run"); /* Retrieve generated TAI timestamps */ ts1 = skb.tstamp; ts2 = skb.cb[0] | ((__u64)skb.cb[1] << 32); /* TAI != 0 */ ASSERT_NEQ(ts1, 0, "tai_ts1"); ASSERT_NEQ(ts2, 0, "tai_ts2"); /* TAI is moving forward only */ ASSERT_GT(ts2, ts1, "tai_forward"); /* Check for future */ ret = clock_gettime(CLOCK_TAI, &now_tai); ASSERT_EQ(ret, 0, "tai_gettime"); now = ts_to_ns(&now_tai); ASSERT_TRUE(now > ts1, "tai_future_ts1"); ASSERT_TRUE(now > ts2, "tai_future_ts2"); /* Check for reasonable range */ ASSERT_TRUE(now - ts1 < TAI_THRESHOLD, "tai_range_ts1"); ASSERT_TRUE(now - ts2 < TAI_THRESHOLD, "tai_range_ts2"); test_time_tai__destroy(skel); }