75 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			75 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// 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_GE(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);
 | 
						|
}
 |