initial commit

IMG/cpio/ventoy/hook/ventoy-os-lib.sh

@@ -0,0 +1,98 @@
+#!/ventoy/busybox/sh
+#************************************************************************************
+# Copyright (c) 2020, longpanda <admin@ventoy.net>
+# 
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation; either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, see <http://www.gnu.org/licenses/>.
+# 
+#************************************************************************************
+
+VT_RULE_DIR_PREFIX=""
+VT_PRINTK_LEVEL=0
+VT_UDEV_RULE_FILE_NAME="99-ventoy.rules"
+VT_UDEV_RULE_PREFIX="ACTION==\"add\", SUBSYSTEM==\"block\","
+
+ventoy_close_printk() {
+    VT_PRINTK_LEVEL=$($CAT /proc/sys/kernel/printk | $AWK '{print $1}')
+    if [ -e /proc/sys/kernel/printk ]; then
+        echo 0 > /proc/sys/kernel/printk
+    fi
+}
+
+ventoy_restore_printk() {
+    if [ -e /proc/sys/kernel/printk ]; then
+        echo $VT_PRINTK_LEVEL > /proc/sys/kernel/printk
+    fi
+}
+
+ventoy_set_rule_dir_prefix() {
+    VT_RULE_DIR_PREFIX=$1
+}
+
+ventoy_get_udev_conf_dir() {
+    if [ -d $VT_RULE_DIR_PREFIX/etc/udev/rules.d ]; then
+        VT_RULE_PATH=$VT_RULE_DIR_PREFIX/etc/udev/rules.d
+    elif [ -d $VT_RULE_DIR_PREFIX/lib/udev/rules.d ]; then
+        VT_RULE_PATH=$VT_RULE_DIR_PREFIX/lib/udev/rules.d
+    else
+        $BUSYBOX_PATH/mkdir -p $VT_RULE_DIR_PREFIX/etc/udev/rules.d
+        VT_RULE_PATH=$VT_RULE_DIR_PREFIX/etc/udev/rules.d
+    fi
+    echo -n "$VT_RULE_PATH"
+}
+
+ventoy_get_udev_conf_path() {
+    VT_RULE_DIR=$(ventoy_get_udev_conf_dir)
+    echo "$VT_RULE_DIR/$VT_UDEV_RULE_FILE_NAME"
+}
+
+ventoy_add_kernel_udev_rule() {
+    VT_UDEV_RULE_PATH=$(ventoy_get_udev_conf_path)
+    echo "KERNEL==\"$1\", $VT_UDEV_RULE_PREFIX RUN+=\"$2\"" >> $VT_UDEV_RULE_PATH
+}
+
+ventoy_add_udev_rule_with_name() {
+    VT_UDEV_RULE_DIR=$(ventoy_get_udev_conf_dir)
+    echo "KERNEL==\"*2\", $VT_UDEV_RULE_PREFIX RUN+=\"$1\"" >> $VT_UDEV_RULE_DIR/$2
+}
+
+ventoy_add_udev_rule_with_path() {
+    echo "KERNEL==\"*2\", $VT_UDEV_RULE_PREFIX RUN+=\"$1\"" >> $2
+}
+
+ventoy_add_udev_rule() {
+    VT_UDEV_RULE_PATH=$(ventoy_get_udev_conf_path)
+    echo "KERNEL==\"*2\", $VT_UDEV_RULE_PREFIX RUN+=\"$1\"" >> $VT_UDEV_RULE_PATH
+}
+
+#
+# It seems there is a bug in somw version of systemd-udevd
+# https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=869719
+#
+ventoy_systemd_udevd_work_around() {
+    VTSYSTEMUDEV="$VT_RULE_DIR_PREFIX/lib/systemd/system/systemd-udevd.service"
+    if [ -e $VTSYSTEMUDEV ]; then
+        if $GREP -q 'SystemCallArchitectures.*native' $VTSYSTEMUDEV; then
+            $SED "s/.*\(SystemCallArchitectures.*native\)/#\1/g"  -i $VTSYSTEMUDEV
+        fi
+    fi
+}
+
+ventoy_print_yum_repo() {
+    echo "[$1]"
+    echo "name=$1"
+    echo "baseurl=$2"
+    echo "enabled=1"
+    echo "gpgcheck=0"
+    echo "priority=0"
+}