#!/bin/bash

#

# weak-modules - determine which modules are kABI compatible with installed

#                kernels and set up the symlinks in /lib/*/weak-updates.

#

# This is an updated version of the script which doesn't support

# multiple installation of the same out-of-tree module (stored in the

# 'extra' subdirectory) for multiple kernels. This assumption is

# supposed to be verified at the rpm level of the packages delivering

# these modules.  There are some checks for this assumption, however we

# really don't solve this situation. This limitation allows for a much

# simpler version of the script. Previous version tried to work in this

# case but was incorrect in some cases.

unset LANG LC_ALL LC_COLLATE

tmpdir=$(mktemp -td ${0##*/}.XXXXXX)

trap "rm -rf $tmpdir" EXIT

unset ${!changed_modules_*} ${!changed_initramfs_*}

unset BASEDIR

unset CHECK_INITRAMFS

default_initramfs_prefix="/boot" # will be combined with BASEDIR

dracut="/sbin/dracut"

declare -a modules

declare -A module_krels

declare -A weak_modules_before

# doit:

# A wrapper used whenever we're going to perform a real operation.

doit() {

    [ -n "$verbose" ] && echo "$@"

    [ -n "$dry_run" ] || "$@"

}

# pr_verbose:

# print verbose -- wrapper used to print extra messages if required

pr_verbose() {

    [ -n "$verbose" ] && echo "$@"

}

# pr_warning:

# print warning

pr_warning() {

    echo "WARNING: $*"

}

# rpmsort: The sort in coreutils can't sort the RPM list how we want it so we

# instead transform the list into a form it will sort correctly, then sort.

rpmsort() {

    local IFS=$' '

    REVERSE=""

    rpmlist=($(cat))

    if [ "-r" == "$1" ];

    then

        REVERSE="-r"

    fi

    echo ${rpmlist[@]} | \

        sed -e 's/-/../g' | \

        sort ${REVERSE} -n -t"." -k1,1 -k2,2 -k3,3 -k4,4 -k5,5 -k6,6 -k7,7 \

             -k8,8 -k9,9 -k10,10 | \

        sed -e 's/\.\./-/g'

}

# krel_of_module:

# Compute the kernel release of a module.

krel_of_module() {

    local module="$1"

    if [ x"${module_krels[$module]+set}" = x"set" ]; then

        # version cached in the array already

        echo "${module_krels[$module]}"

    elif [ -f "$module" ]; then

        krel_of_module_modinfo "$module"

    else

        # Try to extract the kernel release from the path

        # delete case, the .ko already deleted

        set -- "${module#*/lib/modules/}"

        echo "${1%%/*}"

    fi

}

# krel_of_module_modinfo:

# Fetches module version from internal module info

krel_of_module_modinfo() {

    local module="$1"

    /sbin/modinfo -F vermagic "$module" | awk '{print $1}'

}

# read_modules_list:

# Read in a list of modules from standard input. Convert the filenames into

# absolute paths and compute the kernel release for each module (either using

# the modinfo section or through the absolute path.

# If used with input redirect, should be used as read_module_list < input,

# not input | read_modules_list, the latter spawns a subshell

# and the arrays are not seen in the caller

read_modules_list() {

    local IFS=$'\n'

    modules=($(cat))

    for ((n = 0; n < ${#modules[@]}; n++)); do

        if [ ${modules[n]:0:1} != '/' ]; then

            modules[n]="$PWD/${modules[n]}"

        fi

        module_krels["${modules[n]}"]=$(krel_of_module ${modules[n]})

    done

}

decompress_initramfs() {

    local input=$1

    local output=$2

    # First, check if this is compressed at all

    if cpio -i -t < "$input" > /dev/null 2>/dev/null; then

        # If this archive contains a file early_cpio, it's a trick. Strip off

        # the early cpio archive and try again.

        if cpio -i -t < "$input" 2>/dev/null | grep -q '^early_cpio$' ; then

            /usr/lib/dracut/skipcpio "$input" > "${tmpdir}/post_early_cpio.img"

            decompress_initramfs "${tmpdir}/post_early_cpio.img" "$output"

            retval="$?"

            rm -f "${tmpdir}/post_early_cpio.img"

            return $retval

        fi

        cp "$input" "$output"

        return 0

    fi

    # Try gzip

    if gzip -cd < "$input" > "$output" 2>/dev/null ; then

        return 0

    fi

    # Next try xz

    if xz -cd < "$input" > "$output" 2>/dev/null ; then

        return 0

    fi

    echo "Unable to decompress $input: Unknown format" >&2

    return 1

}

# List all module files and modprobe configuration that could require a new

# initramfs. The current directory must be the root of the uncompressed

# initramfs. The unsorted list of files is output to stdout.

list_module_files() {

    find . -iname \*.ko 2>/dev/null

    find etc/modprobe.d usr/lib/modprobe.d -name \*.conf 2>/dev/null

}

# read_old_initramfs:

compare_initramfs_modules() {

    local old_initramfs=$1

    local new_initramfs=$2

    rm -rf "$tmpdir/old_initramfs"

    rm -rf "$tmpdir/new_initramfs"

    mkdir "$tmpdir/old_initramfs"

    mkdir "$tmpdir/new_initramfs"

    decompress_initramfs "$old_initramfs" "$tmpdir/old_initramfs.img"

    pushd "$tmpdir/old_initramfs" >/dev/null

    cpio -i < "$tmpdir/old_initramfs.img" 2>/dev/null

    rm "$tmpdir/old_initramfs.img"

    n=0; for i in `list_module_files|sort`; do

        old_initramfs_modules[n]="$i"

        n=$((n+1))

    done

    popd >/dev/null

    decompress_initramfs "$new_initramfs" "$tmpdir/new_initramfs.img"

    pushd "$tmpdir/new_initramfs" >/dev/null

    cpio -i < "$tmpdir/new_initramfs.img" 2>/dev/null

    rm "$tmpdir/new_initramfs.img"

    n=0; for i in `list_module_files|sort`; do

        new_initramfs_modules[n]="$i"

        n=$((n+1))

    done

    popd >/dev/null

    # Compare the length and contents of the arrays

    if [ "${#old_initramfs_modules[@]}" == "${#new_initramfs_modules[@]}" -a \

         "${old_initramfs_modules[*]}" == "${new_initramfs_modules[*]}" ];

    then

        # If the file lists are the same, compare each file to find any that changed

        for ((n = 0; n < ${#old_initramfs_modules[@]}; n++)); do

            if ! cmp "$tmpdir/old_initramfs/${old_initramfs_modules[n]}" \

                     "$tmpdir/new_initramfs/${new_initramfs_modules[n]}" \

                     >/dev/null 2>&1

            then

                return 1

            fi

        done

    else

        return 1

    fi

    return 0

}

# check_initramfs:

# check and possibly also update the initramfs for changed kernels

check_initramfs() {

    local kernel=$1

    # If there is no initramfs already we will not make one here.

    if [ -e "$initramfs_prefix/initramfs-$kernel.img" ];

    then

        old_initramfs="$initramfs_prefix/initramfs-$kernel.img"

        tmp_initramfs="$initramfs_prefix/initramfs-$kernel.tmp"

        new_initramfs="$initramfs_prefix/initramfs-$kernel.img"

        $dracut -f "$tmp_initramfs" "$kernel"

        if ! compare_initramfs_modules "$old_initramfs" "$tmp_initramfs";

        then

            doit mv "$tmp_initramfs" "$new_initramfs"

        else

            rm -f "$tmp_initramfs"

        fi

    fi

}

usage() {

    echo "Usage: ${0##*/} [options] {--add-modules|--remove-modules}"

    echo "${0##*/} [options] {--add-kernel|--remove-kernel} {kernel-release}"

    cat <<'EOF'

--add-modules

        Add a list of modules read from standard input. Create

        symlinks in compatible kernel's weak-updates/ directory.

        The list of modules is read from standard input.

--remove-modules

        Remove compatibility symlinks from weak-updates/ directories

        for a list of modules.  The list of modules is read from

        standard input. Note: it doesn't attempt to locate any

        compatible modules to replace those being removed.

--add-kernel

        Add compatibility symlinks for all compatible modules to the

        specified or running kernel.

--remove-kernel

        Remove all compatibility symlinks for the specified or current

        kernel.

--no-initramfs

        Do not generate an initramfs.

--verbose

        Print the commands executed.

--dry-run

        Do not create/remove any files.

EOF

    exit $1

}

# module_has_changed:

# Mark if an actual change occured that we need to deal with later by calling

# depmod or mkinitramfs against the affected kernel.

module_has_changed() {

    declare module=$1 krel=$2

    module=${module%.ko}

    module=${module##*/}

    eval "changed_modules_${krel//[^a-zA-Z0-9]/_}=$krel"

    eval "changed_initramfs_${krel//[^a-zA-Z0-9]/_}=$krel"

}

# module_weak_link:

# Generate a weak link path for the module.

# Takes module file name and the target kernel release as arguments

# The way of generation intentionally left from the initial version

module_weak_link() {

    local module="$1"

    local krel="$2"

    local module_krel

    local subpath

    module_krel="$(krel_of_module "$module")"

    subpath=$(echo $module | sed -nre "s:$BASEDIR(/usr)?/lib/modules/$module_krel/([^/]*)/(.*):\3:p")

    echo "$BASEDIR/lib/modules/$krel/weak-updates/${subpath#/}"

}

# module_short_name:

# 'basename' version purely in bash, cuts off path from the filename

module_short_name() {

    echo "${1##*/}"

}

#### Helper predicates

# is_weak_for_module_valid:

# Takes real module filename and target kernel as arguments.

# Calculates weak symlink filename for the corresponding module

# for the target kernel,

# returns 'true' if the symlink filename is a symlink

# and the symlink points to a readable file

# EVEN if it points to a different filename

is_weak_for_module_valid() {

    local module="$1"

    local krel="$2"

    local weak_link

    weak_link="$(module_weak_link $module $krel)"

    [[ -L "$weak_link" ]] && [[ -r "$weak_link" ]]

}

# is_weak_link:

# Takes a filename and a kernel release.

# 'true' if the filename is symlink under weak-updates/ for the kernel.

# It doesn't matter, if it's a valid symlink (points to a real file) or not.

is_weak_link() {

    local link="$1"

    local krel="$2"

    echo $link | grep -q "lib/modules/$krel/weak-updates" || return 1

    [[ -L $link ]]

}

# is_extra_exists:

# Takes a module filename, the module's kernel release and target kernel release.

# The module filename should be a real, not a symlink, filename (i.e. in extra/).

# Returns 'true' if the same module exists for the target kernel.

is_extra_exists() {

    local module="$1"

    local module_krel="$2"

    local krel="$3"

    local subpath="${module#*/lib/modules/$module_krel/extra/}"

    [[ -f $BASEDIR/lib/modules/$krel/extra/$subpath ]]

}

is_kernel_installed() {

    local krel="$1"

    [[ -f "$BASEDIR/boot/symvers-$krel.gz" ]]

}

#### Helpers

# find_modules:

# Takes kernel release and a list of subdirectories.

# Produces list of module files in the subdirectories for the kernel

find_modules() {

    local krel="$1"

    shift

    local dirs="$*"

    for dir in $dirs; do

        find $BASEDIR/lib/modules/$krel/$dir -name '*.ko' 2>/dev/null

    done

}

# find_installed_kernels:

# Produces list of kernels, which modules are still installed

find_installed_kernels() {

    ls $BASEDIR/lib/modules/

}

# find_kernels_with_extra:

# Produces list of kernels, where exists extra/ directory

find_kernels_with_extra() {

    local krel

    local extra_dir

    for krel in $(find_installed_kernels); do

        extra_dir="$BASEDIR/lib/modules/$krel/extra"

        [[ -d "$extra_dir" ]] || continue

        echo "$krel"

    done

}

# remove_weak_link_quiet:

# Takes symlink filename and target kernel release.

# Removes the symlink and the directory tree

# if it was the last file in the tree

remove_weak_link_quiet() {

    local link="$1"

    local krel="$2"

    local subpath="${link#*/lib/modules/$krel/weak-updates}"

    doit rm -f $link

    ( cd "$BASEDIR/lib/modules/$krel/weak-updates" && \

          doit rmdir --parents --ignore-fail-on-non-empty "$(dirname "${subpath#/}")" 2>/dev/null )

}

#### Main logic

# update_modules_for_krel:

# Takes kernel release and "action" function name.

# Skips kernel without symvers,

# otherwise triggers the main logic of modules installing/removing

# for the given kernel, which is:

# - save current state of weak modules symlinks

# - install/remove the symlinks for the given (via stdin) list of modules

# - validate the state and remove invalid symlinks

#   (for the modules, which are not compatible (became incompatible) for

#   the given kernel)

# - check the state after validation to produce needed messages

#   and trigger initrd regeneration if the list changed.

update_modules_for_krel() {

    local krel="$1"

    local func="$2"

    local do_fallback="$3"

    [[ -r "$BASEDIR/boot/symvers-$krel.gz" ]] || return

    global_link_state_save $krel

    $func $krel

    if ! validate_weak_links $krel && [[ -n "$do_fallback" ]]; then

        global_link_state_restore $krel

        return;

    fi

    global_link_state_announce_changes $krel

}

# update_modules:

# Common entry point for add/remove modules command

# Takes the "action" function, the module list is supplied via stdin.

# Reads the module list and triggers modules update for all installed

# kernels.

# Triggers initrd rebuild for the kernels, which modules are installed.

update_modules() {

    local func="$1"

    local do_fallback="$2"

    local module_krel

    read_modules_list || exit 1

    [[ ${#modules[@]} -gt 0 ]] || return

    for krel in $(find_installed_kernels); do

        update_modules_for_krel $krel $func $do_fallback

    done

    for module in "${modules[@]}"; do

        # Module was built against this kernel, update initramfs.

        module_krel="${module_krels[$module]}"

        module_has_changed $module $module_krel

    done

}

# add_weak_links:

# Action function for the "add-modules" command

# Takes the kernel release, where the modules are added

# and the modules[] and module_krels[] global arrays.

# Install symlinks for the kernel with minimal checks

# (just filename checks, no symbol checks)

add_weak_links() {

    local krel="$1"

    local module_krel

    local weak_link

    for module in "${modules[@]}"; do

        module_krel="$(krel_of_module $module)"

        [[ "$module_krel" == "$krel" ]] && continue

        if is_extra_exists $module $module_krel $krel; then

            pr_verbose "found $(module_short_name $module) for $krel while installing for $module_krel, update case?"

        fi

        if is_weak_for_module_valid $module $krel; then

            pr_verbose "weak module for $(module_short_name $module) already exists for kernel $krel, update case?"

            # we should update initrd in update case,

            # the change is not seen by the symlink detector

            # (global_link_state_announce_changes())

            module_has_changed $module $krel

        fi

        weak_link="$(module_weak_link $module $krel)"

        doit mkdir -p "$(dirname $weak_link)"

        doit ln -sf $module $weak_link

    done

}

# remove_weak_links:

# Action function for the "remove-modules" command

# Takes the kernel release, where the modules are removed

# and the modules[] and module_krels[] global arrays.

# Removes symlinks from the given kernel if they are installed

# for the modules in the list.

remove_weak_links() {

    local krel="$1"

    local weak_link

    local target

    local module_krel

    for module in "${modules[@]}"; do

        module_krel="$(krel_of_module $module)"

        [[ "$module_krel" == "$krel" ]] && continue

        weak_link="$(module_weak_link $module $krel)"

        target="$(readlink $weak_link)"

        if [[ "$module" != "$target" ]]; then

            pr_verbose "Skipping symlink $weak_link"

            continue

        fi

        # In update case the --remove-modules call is performed

        # after --add-modules (from postuninstall).

        # So, we shouldn't really remove the symlink in this case.

        # But in the remove case the actual target already removed.

        if ! is_weak_for_module_valid "$module" "$krel"; then

            remove_weak_link_quiet "$weak_link" "$krel"

        fi

    done

}

# validate_weak_links:

# Takes kernel release.

# Checks if all the weak symlinks are suitable for the given kernel.

# Uses depmod to perform the actual symbol checks and parses the output.

# Since depmod internally creates the module list in the beginning of its work

# accroding to the priority list in its configuration, but without symbol

# check and doesn't amend the list during the check, the function runs it

# in a loop in which it removes discovered incompatible symlinks

#

# Returns 0 (success) if proposal is fine or

#         1 (false) if some incompatible symlinks were removed

validate_weak_links() {

    local krel="$1"

    local basedir=${BASEDIR:+-b $BASEDIR}

    local tmp

    declare -A symbols

    local is_updates_changed=1

    local module

    local module_krel

    local target

    local modpath

    local symbol

    local weak_link

    # to return to caller that original proposal is not valid

    # here 0 is true, 1 is false, since it will be the return code

    local is_configuration_valid=0

    tmp=$(mktemp -p $tmpdir)

    if ! [[ -e $tmpdir/symvers-$krel ]]; then

        [[ -e $BASEDIR/boot/symvers-$krel.gz ]] || return

        zcat $BASEDIR/boot/symvers-$krel.gz > $tmpdir/symvers-$krel

    fi

    while ((is_updates_changed)); do

        is_updates_changed=0

        # again $tmp because of subshell, see read_modules_list() comment

        # create incompatibility report by depmod

        /sbin/depmod $basedir -aeE $tmpdir/symvers-$krel -e $krel >$tmp 2>&1

        # parse it into symbols[] associative array in form a-la

        #   symbols["/path/to/the/module"]="list of bad symbols"

        while read line; do

            set -- $(echo $line | awk '/needs unknown symbol/{print $3 " " $NF}')

            modpath=$1

            symbol=$2

            if [[ -n "$modpath" ]]; then

                symbols[$modpath]="${symbols[$modpath]} $symbol"

                continue

            fi

            set -- $(echo $line | awk '/disagrees about version of symbol/{print $3 " " $NF}')

            modpath=$1

            symbol=$2

            if [[ -n "$modpath" ]]; then

                symbols[$modpath]="${symbols[$modpath]} $symbol"

                continue

            fi

        done < $tmp

        # loop through all the weak links from the list of incompatible

        # modules and remove them. Skips non-weak incompatibilities

        for modpath in "${!symbols[@]}"; do

            is_weak_link $modpath $krel || continue

            target=$(readlink $modpath)

            module_krel=$(krel_of_module $target)

            remove_weak_link_quiet "$modpath" "$krel"

            pr_verbose "Module $(module_short_name $modpath) from kernel $module_krel is not compatible with kernel $krel in symbols: ${symbols[$modpath]}"

            is_updates_changed=1

            is_configuration_valid=1 # inversed value

        done

    done

    rm -f $tmp

    # this loop is just to produce verbose compatibility messages

    # for the compatible modules

    for module in "${modules[@]}"; do

        is_weak_for_module_valid $module $krel || continue

        weak_link="$(module_weak_link $module $krel)"

        target="$(readlink $weak_link)"

        module_krel=$(krel_of_module $target)

        if [[ "$module" == "$target" ]]; then

            pr_verbose "Module ${module##*/} from kernel $module_krel is compatible with kernel $krel"

        fi

    done

    return $is_configuration_valid

}

# global_link_state_save:

# Takes kernel release

# Saves the given kernel's weak symlinks state into the global array

# weak_modules_before[] for later processing

global_link_state_save() {

    local krel="$1"

    local link

    local target

    weak_modules_before=()

    for link in $(find_modules $krel weak-updates | xargs); do

        target=$(readlink $link)

        weak_modules_before[$link]=$target

    done

}

# global_link_state_restore:

# Takes kernel release

# Restores the previous weak links state

# (for example, if incompatible modules were installed)

global_link_state_restore() {

    local krel="$1"

    local link

    local target

    local weak_modules_dir="$BASEDIR/lib/modules/$krel/weak-updates"

    pr_verbose "Falling back weak-modules state for kernel $krel"

    ( cd "$weak_modules_dir" && doit rm -rf * )

    for link in "${!weak_modules_before[@]}"; do

        target=${weak_modules_before[$link]}

        doit mkdir -p "$(dirname $link)"

        doit ln -sf $target $link

    done

}

# global_link_state_announce_changes:

# Takes kernel release

# Reads the given kernel's weak symlinks state, compares to the saved,

# triggers initrd rebuild if there were changes

# and produces message on symlink removal

global_link_state_announce_changes() {

    local krel="$1"

    local link

    local target

    local new_target

    declare -A weak_modules_after

    for link in $(find_modules $krel weak-updates | xargs); do

        target=${weak_modules_before[$link]}

        new_target=$(readlink $link)

        weak_modules_after[$link]=$new_target

        # report change of existing link and appearing of a new link

        [[ "$target" == "$new_target" ]] || module_has_changed $new_target $krel

    done

    for link in "${!weak_modules_before[@]}"; do

        target=${weak_modules_before[$link]}

        new_target=${weak_modules_after[$link]}

        # report change of existing link and disappearing of an old link

        [[ "$target" == "$new_target" ]] && continue

        module_has_changed $target $krel

        [[ -n "$new_target" ]] ||

            pr_verbose "Removing compatible module $(module_short_name $target) from kernel $krel"

    done

}

# remove_modules:

# Read in a list of modules from stdinput and process them for removal.

# Parameter (noreplace) is deprecated, acts always as "noreplace".

# There is no sense in the "replace" functionality since according

# to the current requirements RPM will track existing of only one version

# of extra/ module (no same extra/ modules for different kernels).

remove_modules() {

    local no_fallback=""

    update_modules remove_weak_links $no_fallback

}

# add_modules:

# Read in a list of modules from stdinput and process them for compatibility

# with installed kernels under /lib/modules.

add_modules() {

    update_modules add_weak_links do_fallback

}

add_kernel() {

    local krel=${1:-$(uname -r)}

    local tmp

    local no_fallback=""

    tmp=$(mktemp -p $tmpdir)

    if [ ! -e "$BASEDIR/boot/symvers-$krel.gz" ]; then

        echo "Symvers dump file $BASEDIR/boot/symvers-$krel.gz" \

             "not found" >&2

        exit 1

    fi

    for k in $(find_kernels_with_extra | rpmsort); do

        [[ "$krel" == "$k" ]] && continue

        find_modules $k extra >> $tmp