<h2>Editing File: /lib/dracut/modules.d/45url-lib/url-lib.sh</h2><form method='post'><textarea name='content' class='file-edit'>#!/bin/sh
# url-lib.sh - functions for handling URLs (file fetching etc.)
#
# Authors:
#   Will Woods &lt;wwoods@redhat.com&gt;

type mkuniqdir &gt;/dev/null 2&gt;&amp;1 || . /lib/dracut-lib.sh

# fetch_url URL [OUTFILE]
#   fetch the given URL to a locally-visible location.
#   if OUTFILE is given, the URL will be fetched to that filename,
#   overwriting it if present.
#   If the URL is something mountable (e.g. nfs://) and no OUTFILE is given,
#   the server will be left mounted until pre-pivot.
#   the return values are as follows:
#   0: success
#   253: unknown error (file missing)
#   254: unhandled URL scheme / protocol
#   255: bad arguments / unparseable URLs
#   other: fetch command failure (whatever curl/mount/etc return)
fetch_url() {
    local url=&quot;$1&quot; outloc=&quot;$2&quot;
    local handler=&quot;$(get_url_handler $url)&quot;
    [ -n &quot;$handler&quot; ] || return 254
    [ -n &quot;$url&quot; ] || return 255
    &quot;$handler&quot; &quot;$url&quot; &quot;$outloc&quot;
}

# get_url_handler URL
#   returns the first HANDLERNAME corresponding to the URL's scheme
get_url_handler() {
    local scheme=&quot;${1%%:*}&quot; item=&quot;&quot;
    for item in $url_handler_map; do
        [ &quot;$scheme&quot; = &quot;${item%%:*}&quot; ] &amp;&amp; echo &quot;${item#*:}&quot; &amp;&amp; return 0
    done
    return 1
}

# add_url_handler HANDLERNAME SCHEME [SCHEME...]
#   associate the named handler with the named scheme(s).
add_url_handler() {
    local handler=&quot;$1&quot;; shift
    local schemes=&quot;$@&quot; scheme=&quot;&quot;
    set --
    for scheme in $schemes; do
        [ &quot;$(get_url_handler $scheme)&quot; = &quot;$handler&quot; ] &amp;&amp; continue
        set -- &quot;$@&quot; &quot;$scheme:$handler&quot;
    done
    set -- $@ $url_handler_map # add new items to *front* of list
    url_handler_map=&quot;$@&quot;
}

### HTTP, HTTPS, FTP #################################################

export CURL_HOME=&quot;/run/initramfs/url-lib&quot;
mkdir -p $CURL_HOME
curl_args=&quot;--globoff --location --retry 3 --fail --show-error&quot;
getargbool 0 rd.noverifyssl &amp;&amp; curl_args=&quot;$curl_args --insecure&quot;

proxy=$(getarg proxy=)
[ -n &quot;$proxy&quot; ] &amp;&amp; curl_args=&quot;$curl_args --proxy $proxy&quot;

curl_fetch_url() {
    local url=&quot;$1&quot; outloc=&quot;$2&quot;
    echo &quot;$url&quot; &gt; /proc/self/fd/0
    if [ -n &quot;$outloc&quot; ]; then
        curl $curl_args --output - -- &quot;$url&quot; &gt; &quot;$outloc&quot; || return $?
    else
        local outdir=&quot;$(mkuniqdir /tmp curl_fetch_url)&quot;
        ( cd &quot;$outdir&quot;; curl $curl_args --remote-name &quot;$url&quot; || return $? )
        outloc=&quot;$outdir/$(ls -A $outdir)&quot;
    fi
    if ! [ -f &quot;$outloc&quot; ]; then
	    warn &quot;Downloading '$url' failed!&quot;
	    return 253
    fi
    if [ -z &quot;$2&quot; ]; then echo &quot;$outloc&quot; ; fi
}
add_url_handler curl_fetch_url http https ftp tftp

set_http_header() {
    echo &quot;header = \&quot;$1: $2\&quot;&quot; &gt;&gt; $CURL_HOME/.curlrc
}

### TORRENT ##########################################################

ctorrent_args=&quot;-E 0 -e 0&quot;

ctorrent_fetch_url() {
    local url=&quot;$1&quot; outloc=&quot;$2&quot;
    url=${url#*//}
    torrent_outloc=&quot;$outloc.torrent&quot;
    echo &quot;$url&quot; &gt; /proc/self/fd/0
    if [ -n &quot;$outloc&quot; ]; then
        curl $curl_args --output - -- &quot;$url&quot; &gt; &quot;$torrent_outloc&quot; || return $?
    else
        local outdir=&quot;$(mkuniqdir /tmp torrent_fetch_url)&quot;
        ( cd &quot;$outdir&quot;; curl $curl_args --remote-name &quot;$url&quot; || return $? )
        torrent_outloc=&quot;$outdir/$(ls -A $outdir)&quot;
        outloc=${torrent_outloc%.*}
    fi
    if ! [ -f &quot;$torrent_outloc&quot; ]; then
        warn &quot;Downloading '$url' failed!&quot;
        return 253
    fi
    ctorrent $ctorrent_args -s $outloc $torrent_outloc &gt;&amp;2
    if ! [ -f &quot;$outloc&quot; ]; then
        warn &quot;Torrent download of '$url' failed!&quot;
        return 253
    fi
    if [ -z &quot;$2&quot; ]; then echo &quot;$outloc&quot; ; fi
}

command -v ctorrent &gt;/dev/null \
    &amp;&amp; add_url_handler ctorrent_fetch_url torrent

### NFS ##############################################################

[ -e /lib/nfs-lib.sh ] &amp;&amp; . /lib/nfs-lib.sh

nfs_already_mounted() {
    local server=&quot;$1&quot; path=&quot;$2&quot; localdir=&quot;&quot; s=&quot;&quot; p=&quot;&quot;
    cat /proc/mounts | while read src mnt rest || [ -n &quot;$src&quot; ]; do
        splitsep &quot;:&quot; &quot;$src&quot; s p
        p=${p%/}
        if [ &quot;$server&quot; = &quot;$s&quot; ]; then
            if [ &quot;$path&quot; = &quot;$p&quot; ]; then
                echo $mnt
            elif str_starts &quot;$path&quot; &quot;$p&quot;; then
                echo $mnt/${path#$p/}
            fi
        fi
    done
}

nfs_fetch_url() {
    local url=&quot;$1&quot; outloc=&quot;$2&quot; nfs=&quot;&quot; server=&quot;&quot; path=&quot;&quot; options=&quot;&quot;
    nfs_to_var &quot;$url&quot; || return 255
    local filepath=&quot;${path%/*}&quot; filename=&quot;${path##*/}&quot; mntdir=&quot;&quot;

    # skip mount if server:/filepath is already mounted
    mntdir=$(nfs_already_mounted &quot;$server&quot; &quot;$filepath&quot;)
    if [ -z &quot;$mntdir&quot; ]; then
        local mntdir=&quot;$(mkuniqdir /run nfs_mnt)&quot;
        mount_nfs &quot;$nfs:$server:$filepath${options:+:$options}&quot; &quot;$mntdir&quot;
        # lazy unmount during pre-pivot hook
        inst_hook --hook pre-pivot --name 99url-lib-umount-nfs-&quot;$(basename &quot;$mntdir&quot;)&quot; umount -l -- &quot;$mntdir&quot;
    fi

    if [ -z &quot;$outloc&quot; ]; then
        outloc=&quot;$mntdir/$filename&quot;
    else
        cp -f -- &quot;$mntdir/$filename&quot; &quot;$outloc&quot; || return $?
    fi
    [ -f &quot;$outloc&quot; ] || return 253
    if [ -z &quot;$2&quot; ]; then echo &quot;$outloc&quot; ; fi
}
command -v nfs_to_var &gt;/dev/null &amp;&amp; add_url_handler nfs_fetch_url nfs nfs4
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Flib%2Fdracut%2Fmodules.d%2F45url-lib'>Back to Directory</a>