examples/chrootfuse.py: add support for erofsHEADmain

1 files changed, 4 insertions, 2 deletions

diff --git a/examples/chrootfuse.py b/examples/chrootfuse.py
index 93a9da0..7e77384 100755
--- a/examples/chrootfuse.py
+++ b/examples/chrootfuse.py
@@ -24,13 +24,13 @@ import linuxnamespaces
 
 def main() -> None:
     parser = argparse.ArgumentParser()
-    parser.add_argument("--fstype", choices=["ext4", "squashfs"])
+    parser.add_argument("--fstype", choices=["erofs", "ext4", "squashfs"])
     parser.add_argument("fsimage", type=pathlib.Path)
     args = parser.parse_args()
     assert args.fsimage.exists()
     if args.fstype is None:
         args.fstype = args.fsimage.suffix.removeprefix(".")
-    if args.fstype not in ("ext4", "squashfs"):
+    if args.fstype not in ("erofs", "ext4", "squashfs"):
         print("Cannot determine filesystem type for image.")
         sys.exit(1)
     uidmap = linuxnamespaces.IDAllocation.loadsubid("uid").allocatemap(65536)
@@ -46,6 +46,7 @@ def main() -> None:
         childsock.close()
         os.set_inheritable(fds[0], True)
         driver = {
+            "erofs": "erofsfuse",
             "ext4": "fuse2fs",
             "squashfs": "squashfuse",
         }[args.fstype]
@@ -62,6 +63,7 @@ def main() -> None:
     socket.send_fds(mainsock, [b"\0"], [fusefd])
     mainsock.close()
     readonly = {
+        "erofs": True,
         "ext4": False,
         "squashfs": True,
     }[args.fstype]