Something went wrong. Try again.
Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144// SPDX-License-Identifier: GPL-2.0
/* * Copyright (c) 2025, Google LLC. * Pasha Tatashin <pasha.tatashin@soleen.com> */
#define pr_fmt(fmt) KBUILD_MODNAME " test: " fmt
#include <linux/cleanup.h>#include <linux/errno.h>#include <linux/init.h>#include <linux/liveupdate.h>#include <linux/module.h>#include "../../kernel/liveupdate/luo_internal.h"
static const struct liveupdate_flb_ops test_flb_ops;#define DEFINE_TEST_FLB(i) { \ .ops = &test_flb_ops, \ .compatible = LIVEUPDATE_TEST_FLB_COMPATIBLE(i), \}
/* Number of Test FLBs to register with every file handler */#define TEST_NFLBS 3static struct liveupdate_flb test_flbs[TEST_NFLBS] = { DEFINE_TEST_FLB(0), DEFINE_TEST_FLB(1), DEFINE_TEST_FLB(2),};
#define TEST_FLB_MAGIC_BASE 0xFEEDF00DCAFEBEE0ULL
static int test_flb_preserve(struct liveupdate_flb_op_args *argp){ ptrdiff_t index = argp->flb - test_flbs;
pr_info("%s: preserve was triggered\n", argp->flb->compatible); argp->data = TEST_FLB_MAGIC_BASE + index;
return 0;}
static void test_flb_unpreserve(struct liveupdate_flb_op_args *argp){ pr_info("%s: unpreserve was triggered\n", argp->flb->compatible);}
static int test_flb_retrieve(struct liveupdate_flb_op_args *argp){ ptrdiff_t index = argp->flb - test_flbs; u64 expected_data = TEST_FLB_MAGIC_BASE + index;
if (argp->data == expected_data) { pr_info("%s: found flb data from the previous boot\n", argp->flb->compatible); argp->obj = (void *)argp->data; } else { pr_err("%s: ERROR - incorrect data handle: %llx, expected %llx\n", argp->flb->compatible, argp->data, expected_data); return -EINVAL; }
return 0;}
static void test_flb_finish(struct liveupdate_flb_op_args *argp){ ptrdiff_t index = argp->flb - test_flbs; void *expected_obj = (void *)(TEST_FLB_MAGIC_BASE + index);
if (argp->obj == expected_obj) { pr_info("%s: finish was triggered\n", argp->flb->compatible); } else { pr_err("%s: ERROR - finish called with invalid object\n", argp->flb->compatible); }}
static const struct liveupdate_flb_ops test_flb_ops = { .preserve = test_flb_preserve, .unpreserve = test_flb_unpreserve, .retrieve = test_flb_retrieve, .finish = test_flb_finish, .owner = THIS_MODULE,};
static void liveupdate_test_init(void){ static DEFINE_MUTEX(init_lock); static bool initialized; int i;
guard(mutex)(&init_lock);
if (initialized) return;
for (i = 0; i < TEST_NFLBS; i++) { struct liveupdate_flb *flb = &test_flbs[i]; void *obj; int err;
err = liveupdate_flb_get_incoming(flb, &obj); if (err && err != -ENODATA && err != -ENOENT) { pr_err("liveupdate_flb_get_incoming for %s failed: %pe\n", flb->compatible, ERR_PTR(err)); }
if (!err) liveupdate_flb_put_incoming(flb); } initialized = true;}
void liveupdate_test_register(struct liveupdate_file_handler *fh){ int err, i;
liveupdate_test_init();
for (i = 0; i < TEST_NFLBS; i++) { struct liveupdate_flb *flb = &test_flbs[i];
err = liveupdate_register_flb(fh, flb); if (err) { pr_err("Failed to register %s %pe\n", flb->compatible, ERR_PTR(err)); } }
err = liveupdate_register_flb(fh, &test_flbs[0]); if (!err || err != -EEXIST) { pr_err("Failed: %s should be already registered, but got err: %pe\n", test_flbs[0].compatible, ERR_PTR(err)); }
pr_info("Registered %d FLBs with file handler: [%s]\n", TEST_NFLBS, fh->compatible);}
MODULE_LICENSE("GPL");MODULE_AUTHOR("Pasha Tatashin <pasha.tatashin@soleen.com>");MODULE_DESCRIPTION("In-kernel test for LUO mechanism");