1*df1a0d06SVipin Sharma /* SPDX-License-Identifier: GPL-2.0 */ 2*df1a0d06SVipin Sharma 3*df1a0d06SVipin Sharma /* 4*df1a0d06SVipin Sharma * Copyright (c) 2025, Google LLC. 5*df1a0d06SVipin Sharma * Pasha Tatashin <pasha.tatashin@soleen.com> 6*df1a0d06SVipin Sharma * 7*df1a0d06SVipin Sharma * Utility functions for LUO kselftests. 8*df1a0d06SVipin Sharma */ 9*df1a0d06SVipin Sharma 10*df1a0d06SVipin Sharma #ifndef SELFTESTS_LIVEUPDATE_LIB_LIVEUPDATE_H 11*df1a0d06SVipin Sharma #define SELFTESTS_LIVEUPDATE_LIB_LIVEUPDATE_H 12*df1a0d06SVipin Sharma 13*df1a0d06SVipin Sharma #include <errno.h> 14*df1a0d06SVipin Sharma #include <string.h> 15*df1a0d06SVipin Sharma #include <linux/liveupdate.h> 16*df1a0d06SVipin Sharma #include "../../../kselftest.h" 17*df1a0d06SVipin Sharma 18*df1a0d06SVipin Sharma #define LUO_DEVICE "/dev/liveupdate" 19*df1a0d06SVipin Sharma 20*df1a0d06SVipin Sharma #define fail_exit(fmt, ...) \ 21*df1a0d06SVipin Sharma ksft_exit_fail_msg("[%s:%d] " fmt " (errno: %s)\n", \ 22*df1a0d06SVipin Sharma __func__, __LINE__, ##__VA_ARGS__, strerror(errno)) 23*df1a0d06SVipin Sharma 24*df1a0d06SVipin Sharma int luo_open_device(void); 25*df1a0d06SVipin Sharma int luo_create_session(int luo_fd, const char *name); 26*df1a0d06SVipin Sharma int luo_retrieve_session(int luo_fd, const char *name); 27*df1a0d06SVipin Sharma int luo_session_finish(int session_fd); 28*df1a0d06SVipin Sharma int luo_get_session_name(int session_fd, char *name, size_t name_len); 29*df1a0d06SVipin Sharma 30*df1a0d06SVipin Sharma int luo_ensure_nofile_limit(long min_limit); 31*df1a0d06SVipin Sharma int luo_session_preserve_fd(int session_fd, int fd, __u64 token); 32*df1a0d06SVipin Sharma int luo_session_retrieve_fd(int session_fd, __u64 token); 33*df1a0d06SVipin Sharma 34*df1a0d06SVipin Sharma int create_and_preserve_memfd(int session_fd, int token, const char *data); 35*df1a0d06SVipin Sharma int restore_and_verify_memfd(int session_fd, int token, const char *expected_data); 36*df1a0d06SVipin Sharma 37*df1a0d06SVipin Sharma void create_state_file(int luo_fd, const char *session_name, int token, 38*df1a0d06SVipin Sharma int next_stage); 39*df1a0d06SVipin Sharma void restore_and_read_stage(int state_session_fd, int token, int *stage); 40*df1a0d06SVipin Sharma 41*df1a0d06SVipin Sharma void daemonize_and_wait(void); 42*df1a0d06SVipin Sharma 43*df1a0d06SVipin Sharma typedef void (*luo_test_stage1_fn)(int luo_fd); 44*df1a0d06SVipin Sharma typedef void (*luo_test_stage2_fn)(int luo_fd, int state_session_fd); 45*df1a0d06SVipin Sharma 46*df1a0d06SVipin Sharma int luo_test(int argc, char *argv[], const char *state_session_name, 47*df1a0d06SVipin Sharma luo_test_stage1_fn stage1, luo_test_stage2_fn stage2); 48*df1a0d06SVipin Sharma 49*df1a0d06SVipin Sharma #endif /* SELFTESTS_LIVEUPDATE_LIB_LIVEUPDATE_H */ 50