commit afc60d017b238d0aadbaf0c77ff844388e70640b · pyrox.dev/nixpkgs

+86

nixos/modules/virtualisation/includes-to-excludes.py

···

       1
       +
       

     

       2
       +
       # Convert a list of strings to a regex that matches everything but those strings

     

       3
       +
       # ... and it had to be a POSIX regex; no negative lookahead :(

     

       4
       +
       # This is a workaround for erofs supporting only exclude regex, not an include list

     

       5
       +
       

     

       6
       +
       import sys

     

       7
       +
       import re

     

       8
       +
       from collections import defaultdict

     

       9
       +
       

     

       10
       +
       # We can configure this script to match in different ways if we need to.

     

       11
       +
       # The regex got too long for the argument list, so we had to truncate the

     

       12
       +
       # hashes and use MATCH_STRING_PREFIX. That's less accurate, and might pick up some

     

       13
       +
       # garbage like .lock files, but only if the sandbox doesn't hide those. Even

     

       14
       +
       # then it should be harmless.

     

       15
       +
       

     

       16
       +
       # Produce the negation of ^a$

     

       17
       +
       MATCH_EXACTLY = ".+"

     

       18
       +
       # Produce the negation of ^a

     

       19
       +
       MATCH_STRING_PREFIX = "//X" # //X should be epsilon regex instead. Not supported??

     

       20
       +
       # Produce the negation of ^a/?

     

       21
       +
       MATCH_SUBPATHS = "[^/].*$"

     

       22
       +
       

     

       23
       +
       # match_end = MATCH_SUBPATHS

     

       24
       +
       match_end = MATCH_STRING_PREFIX

     

       25
       +
       # match_end = MATCH_EXACTLY

     

       26
       +
       

     

       27
       +
       def chars_to_inverted_class(letters):

     

       28
       +
           assert len(letters) > 0

     

       29
       +
           letters = list(letters)

     

       30
       +
       

     

       31
       +
           s = "[^"

     

       32
       +
       

     

       33
       +
           if "]" in letters:

     

       34
       +
               s += "]"

     

       35
       +
               letters.remove("]")

     

       36
       +
       

     

       37
       +
           final = ""

     

       38
       +
           if "-" in letters:

     

       39
       +
               final = "-"

     

       40
       +
               letters.remove("-")

     

       41
       +
       

     

       42
       +
           s += "".join(letters)

     

       43
       +
       

     

       44
       +
           s += final

     

       45
       +
       

     

       46
       +
           s += "]"

     

       47
       +
       

     

       48
       +
           return s

     

       49
       +
       

     

       50
       +
       # There's probably at least one bug in here, but it seems to works well enough

     

       51
       +
       # for filtering store paths.

     

       52
       +
       def strings_to_inverted_regex(strings):

     

       53
       +
           s = "("

     

       54
       +
       

     

       55
       +
           # Match anything that starts with the wrong character

     

       56
       +
       

     

       57
       +
           chars = defaultdict(list)

     

       58
       +
       

     

       59
       +
           for item in strings:

     

       60
       +
               if item != "":

     

       61
       +
                   chars[item[0]].append(item[1:])

     

       62
       +
       

     

       63
       +
           if len(chars) == 0:

     

       64
       +
               s += match_end

     

       65
       +
           else:

     

       66
       +
               s += chars_to_inverted_class(chars)

     

       67
       +
       

     

       68
       +
           # Now match anything that starts with the right char, but then goes wrong

     

       69
       +
       

     

       70
       +
           for char, sub in chars.items():

     

       71
       +
               s += "|(" + re.escape(char) + strings_to_inverted_regex(sub) + ")"

     

       72
       +
       

     

       73
       +
           s += ")"

     

       74
       +
           return s

     

       75
       +
       

     

       76
       +
       if __name__ == "__main__":

     

       77
       +
           stdin_lines = []

     

       78
       +
           for line in sys.stdin:

     

       79
       +
               if line.strip() != "":

     

       80
       +
                   stdin_lines.append(line.strip())

     

       81
       +
       

     

       82
       +
           print("^" + strings_to_inverted_regex(stdin_lines))

     

       83
       +
       

     

       84
       +
       # Test:

     

       85
       +
       # (echo foo; echo fo/; echo foo/; echo foo/ba/r; echo b; echo az; echo az/; echo az/a; echo ab; echo ab/a; echo ab/; echo abc; echo abcde; echo abb; echo ac; echo b) | grep -vE "$((echo ab; echo az; echo foo;) | python includes-to-excludes.py | tee /dev/stderr )"

     

       86
       +
       # should print ab, az, foo and their subpaths

+29 -5

nixos/modules/virtualisation/qemu-vm.nix

···

       122
        
                 TMPDIR=$(mktemp -d nix-vm.XXXXXXXXXX --tmpdir)

     

       123
        
             fi

     

       124
        
       

     

       125
       -
             ${lib.optionalString cfg.useNixStoreImage

     

       126
       -
             ''

     

       127
       -
               # Create a writable copy/snapshot of the store image.

     

       128
       -
               ${qemu}/bin/qemu-img create -f qcow2 -F qcow2 -b ${storeImage}/nixos.qcow2 "$TMPDIR"/store.img

     

       129
       -
             ''}

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       130
        
       

     

       131
        
             # Create a directory for exchanging data with the VM.

     

       132
        
             mkdir -p "$TMPDIR/xchg"

     
···

       769
        
           );

     

       770
        
           boot.loader.grub.gfxmodeBios = with cfg.resolution; "${toString x}x${toString y}";

     

       771
        
       

     

       0
        
       
     

       0
        
       
     

       772
        
           boot.initrd.extraUtilsCommands = lib.mkIf (cfg.useDefaultFilesystems && !config.boot.initrd.systemd.enable)

     

       773
        
             ''

     

       774
        
               # We need mke2fs in the initrd.

     
···

       905
        
               name = "nix-store";

     

       906
        
               file = ''"$TMPDIR"/store.img'';

     

       907
        
               deviceExtraOpts.bootindex = if cfg.useBootLoader then "3" else "2";

     

       0
        
       
     

       908
        
             }])

     

       909
        
             (mkIf cfg.useBootLoader [

     

       910
        
               # The order of this list determines the device names, see

···

       122
        
                 TMPDIR=$(mktemp -d nix-vm.XXXXXXXXXX --tmpdir)

     

       123
        
             fi

     

       124
        
       

     

       125
       +
             ${lib.optionalString (cfg.useNixStoreImage)

     

       126
       +
               (if cfg.writableStore

     

       127
       +
                 then ''

     

       128
       +
                   # Create a writable copy/snapshot of the store image.

     

       129
       +
                   ${qemu}/bin/qemu-img create -f qcow2 -F qcow2 -b ${storeImage}/nixos.qcow2 "$TMPDIR"/store.img

     

       130
       +
                 ''

     

       131
       +
                 else ''

     

       132
       +
                   (

     

       133
       +
                     cd ${builtins.storeDir}

     

       134
       +
                     ${pkgs.erofs-utils}/bin/mkfs.erofs \

     

       135
       +
                       --force-uid=0 \

     

       136
       +
                       --force-gid=0 \

     

       137
       +
                       -U eb176051-bd15-49b7-9e6b-462e0b467019 \

     

       138
       +
                       -T 0 \

     

       139
       +
                       --exclude-regex="$(

     

       140
       +
                         <${pkgs.closureInfo { rootPaths = [ config.system.build.toplevel regInfo ]; }}/store-paths \

     

       141
       +
                           sed -e 's^.*/^^g' \

     

       142
       +
                         | cut -c -10 \

     

       143
       +
                         | ${pkgs.python3}/bin/python ${./includes-to-excludes.py} )" \

     

       144
       +
                       "$TMPDIR"/store.img \

     

       145
       +
                       . \

     

       146
       +
                       </dev/null >/dev/null

     

       147
       +
                   )

     

       148
       +
                 ''

     

       149
       +
               )

     

       150
       +
             }

     

       151
        
       

     

       152
        
             # Create a directory for exchanging data with the VM.

     

       153
        
             mkdir -p "$TMPDIR/xchg"

     
···

       790
        
           );

     

       791
        
           boot.loader.grub.gfxmodeBios = with cfg.resolution; "${toString x}x${toString y}";

     

       792
        
       

     

       793
       +
           boot.initrd.kernelModules = optionals (cfg.useNixStoreImage && !cfg.writableStore) [ "erofs" ];

     

       794
       +
       

     

       795
        
           boot.initrd.extraUtilsCommands = lib.mkIf (cfg.useDefaultFilesystems && !config.boot.initrd.systemd.enable)

     

       796
        
             ''

     

       797
        
               # We need mke2fs in the initrd.

     
···

       928
        
               name = "nix-store";

     

       929
        
               file = ''"$TMPDIR"/store.img'';

     

       930
        
               deviceExtraOpts.bootindex = if cfg.useBootLoader then "3" else "2";

     

       931
       +
               driveExtraOpts.format = if cfg.writableStore then "qcow2" else "raw";

     

       932
        
             }])

     

       933
        
             (mkIf cfg.useBootLoader [

     

       934
        
               # The order of this list determines the device names, see