/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to initramfs-unpack

  • Committer: Teddy Hogeborn
  • Date: 2021-02-03 23:10:42 UTC
  • Revision ID: teddy@recompile.se-20210203231042-2z3egrvpo1zt7nej
mandos-ctl: Fix bad test for command.Remove and related minor issues

The test for command.Remove removes all clients from the spy server,
and then loops over all clients, looking for the corresponding Remove
command as recorded by the spy server.  But since since there aren't
any clients left after they were removed, no assertions are made, and
the test therefore does nothing.  Fix this.

In tests for command.Approve and command.Deny, add checks that clients
were not somehow removed by the command (in which case, likewise, no
assertions are made).

Add related checks to TestPropertySetterCmd.runTest; i.e. test that a
sequence is not empty before looping over it and making assertions.

* mandos-ctl (TestBaseCommands.test_Remove): Save a copy of the
  original "clients" dict, and loop over those instead.  Add assertion
  that all clients were indeed removed.  Also fix the code which looks
  for the Remove command, which now needs to actually work.
  (TestBaseCommands.test_Approve, TestBaseCommands.test_Deny): Add
  assertion that there are still clients before looping over them.
  (TestPropertySetterCmd.runTest): Add assertion that the list of
  values to get is not empty before looping over them.  Also add check
  that there are still clients before looping over clients.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#!/bin/bash
 
2
 
3
# Initramfs unpacker - unpacks initramfs images into /tmp
 
4
 
5
# Copyright © 2013-2019 Teddy Hogeborn
 
6
# Copyright © 2013-2019 Björn Påhlsson
 
7
 
8
# This file is part of Mandos.
 
9
#
 
10
# Mandos is free software: you can redistribute it and/or modify it
 
11
# under the terms of the GNU General Public License as published by
 
12
# the Free Software Foundation, either version 3 of the License, or
 
13
# (at your option) any later version.
 
14
#
 
15
#     Mandos is distributed in the hope that it will be useful,
 
16
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
 
17
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
18
#     GNU General Public License for more details.
 
19
#
 
20
# You should have received a copy of the GNU General Public License
 
21
# along with Mandos.  If not, see
 
22
# <http://www.gnu.org/licenses/>.
 
23
 
24
# Contact the authors at <mandos@recompile.se>.
 
25
 
 
26
cpio="cpio --extract --make-directories --unconditional --preserve-modification-time"
 
27
 
 
28
if [ -z "$*" ]; then
 
29
    set -- /boot/initrd.img-*
 
30
fi
 
31
 
 
32
for imgfile in "$@"; do
 
33
    if ! [ -f "$imgfile" ]; then
 
34
        echo "Error: Not an existing file: $imgfile" >&2
 
35
        continue
 
36
    fi
 
37
    imgdir="${TMPDIR:-/tmp}/${imgfile##*/}"
 
38
    if [ -d "$imgdir" ]; then
 
39
        rm --recursive -- "$imgdir"
 
40
    fi
 
41
    mkdir --parents "$imgdir"
 
42
    # Does this image contain microcode?
 
43
    if $cpio --quiet --list --file="$imgfile" >/dev/null 2>&1; then
 
44
        # Number of bytes to skip to get to the compressed archive
 
45
        skip=$(($(LANG=C $cpio --io-size=1 --list --file="$imgfile" 2>&1 \
 
46
                      | sed --quiet \
 
47
                            --expression='s/^\([0-9]\+\) blocks$/\1/p')+8))
 
48
        if [ -x /usr/lib/dracut/skipcpio ]; then
 
49
            catimg="/usr/lib/dracut/skipcpio $imgfile"
 
50
        else
 
51
            catimg="dd if=$imgfile bs=$skip skip=1 status=noxfer"
 
52
        fi
 
53
    else
 
54
        echo "No microcode detected"
 
55
        catimg="cat -- $imgfile"
 
56
    fi
 
57
    while :; do
 
58
        # Determine the compression method
 
59
        if { $catimg 2>/dev/null | zcat --test >/dev/null 2>&1;
 
60
             [ ${PIPESTATUS[-1]} -eq 0 ]; }; then
 
61
            decomp="zcat"
 
62
        elif { $catimg 2>/dev/null | bzip2 --test >/dev/null 2>&1;
 
63
               [ ${PIPESTATUS[-1]} -eq 0 ]; }; then
 
64
            decomp="bzip2 --stdout --decompress"
 
65
        elif { $catimg 2>/dev/null | lzop --test >/dev/null 2>&1;
 
66
               [ ${PIPESTATUS[-1]} -eq 0 ]; }; then
 
67
            decomp="lzop --stdout --decompress"
 
68
        else
 
69
            skip=$((${skip}+1))
 
70
            echo "Could not determine compression of ${imgfile}; trying to skip ${skip} bytes" >&2
 
71
            catimg="dd if=$imgfile bs=$skip skip=1 status=noxfer"
 
72
            continue
 
73
        fi
 
74
        break
 
75
    done
 
76
    case "$catimg" in
 
77
        *skipcpio*) echo "Microcode detected, skipping";;
 
78
        *) echo "Microcode detected, skipping ${skip} bytes";;
 
79
    esac
 
80
    $catimg 2>/dev/null | $decomp | ( cd -- "$imgdir" && $cpio --quiet )
 
81
    if [ ${PIPESTATUS[-1]} -eq 0 ]; then
 
82
        echo "$imgfile unpacked into $imgdir"
 
83
    fi
 
84
done