cachepc-linux

Fork of AMDESE/linux with modifications for CachePC side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-linux
Log | Files | Refs | README | LICENSE | sfeed.txt

subprogs.c (2050B)

      1// SPDX-License-Identifier: GPL-2.0
      2/* Copyright (c) 2020 Facebook */
      3#include <test_progs.h>
      4#include "test_subprogs.skel.h"
      5#include "test_subprogs_unused.skel.h"
      6
      7struct toggler_ctx {
      8	int fd;
      9	bool stop;
     10};
     11
     12static void *toggle_jit_harden(void *arg)
     13{
     14	struct toggler_ctx *ctx = arg;
     15	char two = '2';
     16	char zero = '0';
     17
     18	while (!ctx->stop) {
     19		lseek(ctx->fd, SEEK_SET, 0);
     20		write(ctx->fd, &two, sizeof(two));
     21		lseek(ctx->fd, SEEK_SET, 0);
     22		write(ctx->fd, &zero, sizeof(zero));
     23	}
     24
     25	return NULL;
     26}
     27
     28static void test_subprogs_with_jit_harden_toggling(void)
     29{
     30	struct toggler_ctx ctx;
     31	pthread_t toggler;
     32	int err;
     33	unsigned int i, loop = 10;
     34
     35	ctx.fd = open("/proc/sys/net/core/bpf_jit_harden", O_RDWR);
     36	if (!ASSERT_GE(ctx.fd, 0, "open bpf_jit_harden"))
     37		return;
     38
     39	ctx.stop = false;
     40	err = pthread_create(&toggler, NULL, toggle_jit_harden, &ctx);
     41	if (!ASSERT_OK(err, "new toggler"))
     42		goto out;
     43
     44	/* Make toggler thread to run */
     45	usleep(1);
     46
     47	for (i = 0; i < loop; i++) {
     48		struct test_subprogs *skel = test_subprogs__open_and_load();
     49
     50		if (!ASSERT_OK_PTR(skel, "skel open"))
     51			break;
     52		test_subprogs__destroy(skel);
     53	}
     54
     55	ctx.stop = true;
     56	pthread_join(toggler, NULL);
     57out:
     58	close(ctx.fd);
     59}
     60
     61static void test_subprogs_alone(void)
     62{
     63	struct test_subprogs *skel;
     64	struct test_subprogs_unused *skel2;
     65	int err;
     66
     67	skel = test_subprogs__open_and_load();
     68	if (!ASSERT_OK_PTR(skel, "skel_open"))
     69		return;
     70
     71	err = test_subprogs__attach(skel);
     72	if (!ASSERT_OK(err, "skel attach"))
     73		goto cleanup;
     74
     75	usleep(1);
     76
     77	ASSERT_EQ(skel->bss->res1, 12, "res1");
     78	ASSERT_EQ(skel->bss->res2, 17, "res2");
     79	ASSERT_EQ(skel->bss->res3, 19, "res3");
     80	ASSERT_EQ(skel->bss->res4, 36, "res4");
     81
     82	skel2 = test_subprogs_unused__open_and_load();
     83	ASSERT_OK_PTR(skel2, "unused_progs_skel");
     84	test_subprogs_unused__destroy(skel2);
     85
     86cleanup:
     87	test_subprogs__destroy(skel);
     88}
     89
     90void test_subprogs(void)
     91{
     92	if (test__start_subtest("subprogs_alone"))
     93		test_subprogs_alone();
     94	if (test__start_subtest("subprogs_and_jit_harden"))
     95		test_subprogs_with_jit_harden_toggling();
     96}