|
| 1 | +# number: 35 |
| 2 | +# tmt: |
| 3 | +# summary: Test composefs garbage collection with same and different kernel+initrd |
| 4 | +# duration: 30m |
| 5 | + |
| 6 | +use std assert |
| 7 | +use tap.nu |
| 8 | + |
| 9 | +# bootc status |
| 10 | +let st = bootc status --json | from json |
| 11 | +let booted = $st.status.booted.image |
| 12 | + |
| 13 | +if not (tap is_composefs) or ($st.status.booted.composefs.bootType | str downcase) == "uki" { |
| 14 | + exit 0 |
| 15 | +} |
| 16 | + |
| 17 | +# Create a large file in a new container image, then bootc switch to the image |
| 18 | +def first_boot [] { |
| 19 | + bootc image copy-to-storage |
| 20 | + |
| 21 | + echo $" |
| 22 | + FROM localhost/bootc |
| 23 | + RUN dd if=/dev/zero of=/usr/share/large-test-file bs=1k count=1337 |
| 24 | + RUN echo 'large-file-marker' | dd of=/usr/share/large-test-file conv=notrunc |
| 25 | + " | podman build -t localhost/bootc-derived . -f - |
| 26 | + |
| 27 | + let current_time = (date now) |
| 28 | + |
| 29 | + bootc switch --transport containers-storage localhost/bootc-derived |
| 30 | + |
| 31 | + # Find the large file's verity and save it |
| 32 | + # nu has its own built in find which sucks, so we use the other one |
| 33 | + # TODO: Replace this with some concrete API |
| 34 | + # See: https://github.com/composefs/composefs-rs/pull/236 |
| 35 | + let file_path = ( |
| 36 | + /usr/bin/find /sysroot/composefs/objects -type f -size 1337k -newermt ($current_time | format date "%Y-%m-%d %H:%M:%S") |
| 37 | + | xargs grep -lx "large-file-marker" |
| 38 | + ) |
| 39 | + |
| 40 | + echo $file_path | save /var/large-file-marker-objpath |
| 41 | + cat /var/large-file-marker-objpath |
| 42 | + |
| 43 | + echo $st.status.booted.composefs.verity | save /var/first-verity |
| 44 | + |
| 45 | + tmt-reboot |
| 46 | +} |
| 47 | + |
| 48 | +# Create a container image derived from the first boot image, but update the initrd |
| 49 | +def second_boot [] { |
| 50 | + assert equal $booted.image.image "localhost/bootc-derived" |
| 51 | + |
| 52 | + let path = cat /var/large-file-marker-objpath |
| 53 | + |
| 54 | + echo "\$path" |
| 55 | + echo $path |
| 56 | + |
| 57 | + assert ($path | path exists) |
| 58 | + |
| 59 | + # Create another image with a different initrd so we can test kernel + initrd cleanup |
| 60 | + |
| 61 | + echo " |
| 62 | + FROM localhost/bootc |
| 63 | +
|
| 64 | + RUN echo 'echo hello' > /usr/bin/hello |
| 65 | + RUN chmod +x /usr/bin/hello |
| 66 | +
|
| 67 | + RUN mkdir /usr/lib/dracut/modules.d/99something |
| 68 | +
|
| 69 | + RUN cat <<-EOF > /usr/lib/dracut/modules.d/99something/module-setup.sh |
| 70 | + #!/usr/bin/bash |
| 71 | +
|
| 72 | + check() { |
| 73 | + return 0 |
| 74 | + } |
| 75 | +
|
| 76 | + depends() { |
| 77 | + return 0 |
| 78 | + } |
| 79 | +
|
| 80 | + install() { |
| 81 | + inst '/usr/bin/hello' /bin/hello |
| 82 | + } |
| 83 | + EOF |
| 84 | +
|
| 85 | + RUN set -x; kver=$(cd /usr/lib/modules && echo *); dracut -vf --add bootc /usr/lib/modules/$kver/initramfs.img $kver; |
| 86 | + " | lines | each { str trim } | str join "\n" | podman build -t localhost/bootc-derived-initrd . -f - |
| 87 | + |
| 88 | + bootc switch --transport containers-storage localhost/bootc-derived-initrd |
| 89 | + |
| 90 | + tmt-reboot |
| 91 | +} |
| 92 | + |
| 93 | +# The large file should've been GC'd as we switched to an image derived from the original one |
| 94 | +def third_boot [] { |
| 95 | + assert equal $booted.image.image "localhost/bootc-derived-initrd" |
| 96 | + |
| 97 | + let path = cat /var/large-file-marker-objpath |
| 98 | + assert (not ($"/sysroot/composefs/objects/($path)" | path exists)) |
| 99 | + |
| 100 | + # Also assert we have two different kernel + initrd pairs |
| 101 | + let booted_verity = (bootc status --json | from json).status.booted.composefs.verity |
| 102 | + |
| 103 | + let bootloader = (bootc status --json | from json).status.booted.composefs.bootloader |
| 104 | + |
| 105 | + let boot_dir = if ($bootloader | str downcase) == "systemd" { |
| 106 | + # TODO: Some concrete API for this would be great |
| 107 | + mkdir /var/tmp/efi |
| 108 | + mount /dev/vda2 /var/tmp/efi |
| 109 | + "/var/tmp/efi/EFI/Linux" |
| 110 | + } else { |
| 111 | + "/sysroot/boot" |
| 112 | + } |
| 113 | + |
| 114 | + assert ($"($boot_dir)/($booted_verity)" | path exists) |
| 115 | + |
| 116 | + # This is for the rollback, but since the rollback and the very |
| 117 | + # first boot have the same kernel + initrd pair, and this rollback |
| 118 | + # was deployed after the first boot, we will still be using the very |
| 119 | + # first verity for the boot binary name |
| 120 | + assert ($"($boot_dir)/(cat /var/first-verity)" | path exists) |
| 121 | + |
| 122 | + echo $"($boot_dir)/($booted_verity)" | save /var/to-be-deleted-kernel |
| 123 | + |
| 124 | + # Now we create a new image derived from the current kernel + initrd |
| 125 | + # Switching to this and rebooting should remove the old kernel + initrd |
| 126 | + echo " |
| 127 | + FROM localhost/bootc-derived-initrd |
| 128 | + RUN echo 'another file' > /usr/share/another-one |
| 129 | + " | podman build -t localhost/bootc-final . -f - |
| 130 | + |
| 131 | + |
| 132 | + bootc switch --transport containers-storage localhost/bootc-final |
| 133 | + |
| 134 | + tmt-reboot |
| 135 | +} |
| 136 | + |
| 137 | +def fourth_boot [] { |
| 138 | + assert equal $booted.image.image "localhost/bootc-final" |
| 139 | + assert (not ((cat /var/to-be-deleted-kernel | path exists))) |
| 140 | + |
| 141 | + tap ok |
| 142 | +} |
| 143 | + |
| 144 | +def main [] { |
| 145 | + match $env.TMT_REBOOT_COUNT? { |
| 146 | + null | "0" => first_boot, |
| 147 | + "1" => second_boot, |
| 148 | + "2" => third_boot, |
| 149 | + "3" => fourth_boot, |
| 150 | + $o => { error make { msg: $"Invalid TMT_REBOOT_COUNT ($o)" } }, |
| 151 | + } |
| 152 | +} |
| 153 | + |
0 commit comments