infra
/
init-wrapper
1
0
Fork 0
Code Packages Projects Releases Activity

arg0 replace

This commit is contained in:
Sangbum Kim 2024-03-06 18:08:21 +09:00
parent 3f5fc7d16f
commit 81fd9bb9cf
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/main.rs
View File

@ -12,7 +12,7 @@ use core::ptr::null;
use core::time::Duration; use core::time::Duration;
use errno::{errno, Errno}; use errno::{errno, Errno};
use libc::{ use libc::{
c_int, c_void, chdir, clock_gettime, execvp, mkdir, mode_t, mount, rmdir, syscall, timespec, c_int, c_void, chdir, clock_gettime, execv, mkdir, mode_t, mount, rmdir, syscall, timespec,
umount2, SYS_pivot_root, CLOCK_BOOTTIME, EXIT_FAILURE, EXIT_SUCCESS, MNT_DETACH, MS_BIND, umount2, SYS_pivot_root, CLOCK_BOOTTIME, EXIT_FAILURE, EXIT_SUCCESS, MNT_DETACH, MS_BIND,
MS_LAZYTIME, MS_MOVE, MS_NODEV, MS_NOSUID, MS_PRIVATE, MS_RDONLY, MS_REC, MS_RELATIME, MS_LAZYTIME, MS_MOVE, MS_NODEV, MS_NOSUID, MS_PRIVATE, MS_RDONLY, MS_REC, MS_RELATIME,
}; };
@ -180,14 +180,14 @@ fn replace_root() -> Result<(), ()> {
}; };
let ovr_dir_structures = [ let ovr_dir_structures = [
"/run/overlay", ("/run/overlay", 0o0700),
"/run/overlay/lower", ("/run/overlay/lower", 0o0700),
"/run/overlay/upper", ("/run/overlay/upper", 0o0755),
"/run/overlay/work", ("/run/overlay/work", 0o0755),
"/run/overlay/merged", ("/run/overlay/merged", 0o0755),
]; ];
for path in ovr_dir_structures { for (path, mode) in ovr_dir_structures {
match do_mkdir(path, 0o0700) { match do_mkdir(path, mode) {
Err(e) => { Err(e) => {
eprintln!("mkdir \"{}\" failed: {}", path, e); eprintln!("mkdir \"{}\" failed: {}", path, e);
return Err(()); return Err(());
@ -283,7 +283,7 @@ fn replace_root() -> Result<(), ()> {
} }
#[no_mangle] #[no_mangle]
unsafe extern "C" fn main(_argc: c_int, argv: *const *const u8) -> c_int { unsafe extern "C" fn main(_argc: c_int, argv: *mut *const u8) -> c_int {
let start = match do_gettime() { let start = match do_gettime() {
Err(err) => { Err(err) => {
eprintln!("{}", err); eprintln!("{}", err);
@ -307,8 +307,10 @@ unsafe extern "C" fn main(_argc: c_int, argv: *const *const u8) -> c_int {
let elapsed = end - start; let elapsed = end - start;
eprintln!("processed in {:?}", elapsed); eprintln!("processed in {:?}", elapsed);
// overwrite arg0
let init_path = CStr::from_bytes_with_nul_unchecked(b"/sbin/init\0"); let init_path = CStr::from_bytes_with_nul_unchecked(b"/sbin/init\0");
execvp(init_path.as_ptr(), argv); *argv = init_path.as_ptr();
execv(init_path.as_ptr(), argv);
EXIT_SUCCESS EXIT_SUCCESS
} }