Annotation of /drakx-kbd-mouse-x11/trunk/tools/display_driver_helper

#!/bin/sh
#
# Display driver helper
#
# Copyright (c) 2010, 2011, 2012 Anssi Hannula <anssi.hannula@iki.fi>
#
# - Load drivers for specified modaliases, skipping disabled display drivers
#   that would cause conflicts (KMS vs. vesa, KMS vs. proprietary).
# - Get information about enabled driver modules
# - Check that the loaded modules are correct
#
# Note that drivers not known by this script are handled normally, i.e.
# loaded automatically as udev normally would've.
#
# Licensed under terms of GPLv2 or later.
#
# When updating, check:
# - the variables below
# - check_driver function
# - check_dkms_status function
#

# DEBUG_DISPLAY_DRIVER_HELPER=yes

if [ -n "$DEBUG_DISPLAY_DRIVER_HELPER" ]; then
        echo "$(date) $*" >> /dev/ddh_debug
        exec 2>>/dev/ddh_debug
        set -x
fi

export LC_ALL=C

KMS_DRIVERS="i915 radeon nouveau"
# module names at run-time (hence nvidia instead of nvidia*):
KNOWN_MODULES="i915|radeon|nouveau|fglrx|nvidia"

XORG_i915="intel"
CONFLICTS_i915=""

XORG_nouveau="nouveau"
CONFLICTS_nouveau="nv nvidia"

XORG_radeon="ati radeon"
CONFLICTS_radeon="fglrx"

# Note: no /usr
# See end of script for descriptions of global variables.
check_driver() {
        local name="$1"
        # modalias is optional, only needed to set MODOPTIONS for radeon
        # this should not be set when not intending to load the module
        local modalias="$2"

        MODOPTIONS=

        case "$name" in
        i915)
                # implicitely loaded by X.org
                check_xorg $name 0 || return 1
                IS_KMS=1
                ;;
        radeon)
                # implicitely loaded by X.org
                check_xorg $name 0 || return 1
                # Do not load if the proprietary driver is temporarily disabled
                # on a PowerXpress system.
                # TODO: this check could be omitted if radeon was explicitely
                # specified in xorg.conf - but that is only known by check_xorg.
                check_gl /etc/fglrx/pxpress-free.ld.so.conf && return 1

                IS_KMS=1
                # radeon KMS needs to be loaded before X server
                NEEDS_LOAD_NOW=1

                # check if firmware is needed and present (random fw checked)
                if [ -n "$modalias" ] && [ ! -e "/lib/firmware/radeon/R700_rlc.bin" ]; then
                        # no firmware, lets check if it is needed
                        pciid="${modalias#pci:v00001002d0000}"
                        pciid="${pciid%sv*}"
                        case "$pciid" in
                        # All Radeons up to PALM (see radeon_probe.c)
                        31??|3E??|4???|5???|7???|68??|94??|95??|961?|971?|98??)
                                # The radeon driver reportedly has some issues
                                # when loaded without firmware - it doesn't fallback to noaccel
                                # nicely as it should, instead the framebuffer gets messed up.
                                # However, these cards are still supported by usermode modesetting,
                                # so just disable KMS. (Mageia bug #3466)
                                # TODO: actually fix the driver to not mess up framebuffer when no firmware
                                # is present and disable accel cleanly so that this workaround
                                # wouldn't be needed -Anssi
                                MODOPTIONS="$MODOPTIONS modeset=0"
                                IS_KMS=
                                # needs to be loaded now in order to set the module option
                                NEEDS_LOAD_NOW=1
                                ;;
                        ????)
                                # Cards newer than PALM don't support UMS, so don't load the driver
                                # at all (it might still get done by X server, though).
                                return 1
                                ;;
                        *)
                                # Modalias parse error, strange. Load normally...
                                ;;
                        esac
                fi
                ;;
        nouveau)
                # implicitely loaded by X.org
                check_xorg $name 0 || return 1
                IS_KMS=1
                ;;
        fglrx)
                check_xorg fglrx 1 || return 1
                check_dkms fglrx || return 1
                ;;
        nvidia)
                # manually installed driver or a call from check_loaded()
                UNSURE=1
                check_xorg nvidia 1 || return 1
                ;;
        nvidiafb)
                # this is only reached if nvidiafb is manually unblacklisted
                return 2
                ;;
        nvidia*)
                [ "$name" = "nvidia_current" ] && name=nvidia-current
                # there are multiple co-installable driver versions, so check
                # the active alternative as well
                check_gl /etc/$name/ld.so.conf || return 1
                check_xorg nvidia 1 || return 1
                check_dkms $name || return 1
                ;;
        *)
                # unknown, will be loaded only if no known drivers were found
                return 2
                ;;
        esac
        return 0
}

# Return success if there is no new pending DKMS build (needed to disallow
# speedboot on very early boot).
# Previously failed build or a missing module is counted as no pending build.
# Note: no /usr
check_dkms_status() {
        [ -e /etc/alternatives/gl_conf ] || return 0
        # return fast for non-DKMS
        check_gl /etc/ld.so.conf.d/GL/standard.conf && return 0

        local active="$(ls -l /etc/alternatives/gl_conf | awk '{ print $NF }')"

        local modname=

        case $active in
        /etc/nvidia*/ld.so.conf)
                modname="nvidia${active#*nvidia}"
                modname="${modname%/*}"
                ;;
        /etc/ld.so.conf.d/GL/ati.conf)
                modname="fglrx"
                ;;
        *)
                # Unknown DKMS-looking driver,
                # allow speedboot.
                return 0
                ;;
        esac
        check_dkms "$modname" 1
}

# Check if all loaded kernel modules have correct xorg.conf
check_loaded() {
        for module in $(grep -oE "^($KNOWN_MODULES) " /proc/modules); do
                # try to unload the driver in case it is not in use before bailing
                check_driver "$module" || rmmod "$module" &>/dev/null || return 1
        done
        return 0
}

# Check that specified DKMS driver is not queued for build for the current
# kernel. Used to check if we 1) should disable speedboot for this boot
# (--check-dkms-status), and 2) if should should load the currently
# existing driver (--load). Doing otherwise might cause us to load a wrong old
# version of the driver that had been installed using e.g. binary DKMS
# packages.
# Note: no /usr
check_dkms() {
        local driver="$1"
        local force="$2"

        # If called from DKMS itself or we are not in rc.sysinit anymore,
        # there are no pending builds.
        if [ -z "$force" ]; then
                [ "$DKMS_AUTOLOAD_MODULE" = "$driver" ] && return 0
                [ -z "$STARTUP" ] && [ ! -f "/dev/.in_sysinit" ] && return 0
        fi

        local found=
        local uname_r="$(uname -r)"

        for dir in /var/lib/dkms/$driver/*; do
                [ -e "$dir" ] || return 0 # no module, no build; or no /var
                [ -L "$dir" ] && continue # not a module version
                found=1 # module version found
                [ -e "$dir/$uname_r" ] && return 0
                [ -e "/var/lib/dkms-binary/$driver/$(basename "$dir")/$uname_r" ] && return 0

                if [ -e "$dir/build/make.log" ]; then
                        # Build has failed for some kernel, check if it is this one.
                        # If so, there is no point in returning 1.
                        grep -q "^DKMS make\.log.* $uname_r " && return 0
                fi
        done

        # if module versions were found but none were built for this kernel, return 1
        [ -n "$found" ] && return 1 || return 0
}

# Note: no /usr
check_gl() {
        local alt_inode="$(stat -L -c%i $1 2>/dev/null)"
        [ -n "$alt_inode" ] || return 1
        [ -n "$GL_INODE" ] || GL_INODE="$(stat -L -c%i /etc/alternatives/gl_conf 2>/dev/null)"
        [ "$alt_inode" = "$GL_INODE" ] || return 1
        return 0
}

# Note: no /usr
get_xorg_drivers() {
        if [ -z "$XORG_DRIVERS" ]; then
                XORG_DRIVERS="$(cat /etc/X11/xorg.conf /etc/X11/xorg.conf.d/*.conf 2>/dev/null |
                        awk -F'"' -vORS=' ' -vIGNORECASE=1 '
                        /^[[:space:]]+*section[[:space:]]+"device"/     { device=1 }
                        /endsection/                                    { device=0 }
                        /^[[:space:]]*driver[[:space:]]*".*"/           { if (device) drivers[$2]=$2 }
                        END                                             { for (driver in drivers) print driver }
                        ')"
                [ -n "$XORG_DRIVERS" ] || XORG_DRIVERS="-"
        fi
}

# Note: no /usr
# parameter 1: KMS module or xorg driver
# parameter 2: 1 - check if the driver is explicitely enabled
#              0 - check only for conflicts
check_xorg() {
        local driver="$1"
        local explicit_only="$2"

        eval local xorg_drivers=\"\$XORG_$driver\"
        [ -n "$xorg_drivers" ] || xorg_drivers="$driver"
        eval local conflicts=\"\$CONFLICTS_$driver\"

        get_xorg_drivers

        conflict_found=
        for enabled_driver in $XORG_DRIVERS; do
                for xorg_driver in $xorg_drivers; do
                        [ "$enabled_driver" = "$xorg_driver" ] && return 0
                done

                # if the X.org driver can be loaded implicitely, check that
                # there are no conflicting drivers that override the driver
                if [ "$explicit_only" = "0" -a -z "$conflict_found" ]; then
                        for conflict in vesa $conflicts; do
                                if [ "$enabled_driver" = "$conflict" ]; then
                                       conflict_found=1
                                       continue 2
                                       # continue loop to check for an explicit load
                                fi
                        done
                fi
        done

        # in case of a conflict, do not load the module
        [ -n "$conflict_found" ] && return 1

        # no driver is selected - don't load if explicit_only is 1
        [ "$explicit_only" = "1" ] && return 1

        # implicit load allowed; only load if there is evidence that this is
        # not a live cd or similar with automatic configuration occurring later
        # in the boot process (which might configure a driver conflicting with
        # the implicit driver, e.g. a proprietary one)
        # TODO: Could this be replaced with a more robust check?
        [ -e "/etc/X11/xorg.conf" ] || [ -e "/etc/sysconfig/harddrake2/kernels" ] ||
                [ -e "/etc/sysconfig/harddrake2/xorg" ] || [ -e "/boot/grub/menu.lst" ]
}

# Load the driver for the specified modalias, if configured.
# Note: no /usr
load_driver() {
        local modalias="$1"
        local modulename
        local load_default=1

        for modulename in $(/sbin/modprobe -Rq "$modalias"); do
                check_driver "$modulename" "$modalias"
                case $? in
                1)      # a driver which needs handling by this script matches
                        # the modalias, but was not configured - do not run
                        # the generic modprobe if no other drivers are
                        # configured either
                        load_default=
                        continue
                        ;;
                2)      continue
                        ;;
                esac

                if [ -n "$IS_KMS" ]; then
                        grep -q "^$modulename " /proc/modules && return 0
                        echo "$modulename" > /dev/.late_kms 2>/dev/null
                        # If NEEDS_LOAD_NOW is not set and plymouth is running,
                        # skip loading the driver to avoid quitting plymouth.
                        # The driver will be loaded later by X server itself.
                        [ -z "$NEEDS_LOAD_NOW" ] && /bin/plymouth --ping 2>/dev/null && return 0
                        /bin/plymouth quit 2>/dev/null
                fi
                /sbin/modprobe -b "$modulename" $MODOPTIONS && return 0
        done

        # no specially handled modules were loaded, so load all modules normally
        # unless $load_default was set above
        [ -z "$load_default" ] || /sbin/modprobe -b "$modalias"
}

is_kms_allowed() {
        for driver in $KMS_DRIVERS; do
                # Check all drivers for conflicts only.
                check_xorg $driver 0 || return 1
        done
        return 0
}

get_initrd_kms_drivers() {
        local initrd="$1"

        local kms_drivers="$(echo "$KMS_DRIVERS" | tr " " "|")"
        zcat "$initrd" | cpio -t --quiet | sed -nr "s,.*/($kms_drivers)\.ko.*$,\1,p"
}

get_active_kms_drivers() {
        local kms_drivers=
        for device in $(grep -l 0x03 /sys/bus/pci/devices/0000\:0*/class); do
                [ -e "$device" ] || continue
                device="$(dirname $device)"
                [ -f "$device/modalias" ] || continue
                modalias="$(cat "$device/modalias")"
                for modulename in $(/sbin/modprobe -Rq "$modalias"); do
                        IS_KMS=
                        check_driver "$modulename" "$modalias" || continue
                        [ -n "$IS_KMS" ] && echo $modulename
                done
        done
}

usage() {
        cat <<EOF
Usage: $0 action [arguments]

Known actions:

  --load MODALIAS
        Load drivers matching MODALIAS, checking that they are enabled and
        configured.

  --load-dkms-autoload MODNAME MODALIAS
        Same as --load, but assume MODNAME is built and correct so that
        checking dkms status is unnecessary.

  --is-disabled MODNAME
        Checks whether the driver corresponding to MODNAME is disabled (e.g.
        a conflicting driver is configured, etc.). Unknown MODNAMEs are
        considered not disabled.

  --is-enabled-kms MODNAME
        Checks whether the driver corresponding to MODNAME is enabled and
        MODNAME is a known KMS module. Note that drivers may be enabled even
        if there is no such hardware. This just checks that there are
        no conflicting drivers in use etc.

  --is-kms-allowed
        Checks whether it is ok to load KMS drivers in initrd. This returns
        a failure when a conflicting driver is set up (vesa or a proprietary
        one).

  --get-all-kms-drivers
        Get a list of the known KMS drivers.

  --get-active-kms-drivers
        Get a list of the known KMS drivers which are enabled and the hardware
        is present.

  --get-initrd-kms-drivers INITRD
        Get a list of the known KMS drivers in initrd INITRD.

  --check-dkms-status
        Checks if there are no pending DKMS builds for the currently enabled
        drivers.

  --check-loaded
        Checks that there are no disabled drivers loaded.

  --check-speedboot
        Does --check-dkms-status and --check-loaded.

  --check-loaded-strict
        As --check-loaded, and consider ambiguous cases (e.g. nvidia where
        we can't detect if the loaded driver has the correct version) as
        failure.
EOF
}

# clear global variables

# cache for check_gl()
GL_INODE=

# cache for check_xorg()
XORG_DRIVERS=

# The driver is a KMS enabled driver. This will cause the script to quit
# plymouth when a driver is loaded by --load and NEEDS_LOAD_NOW below is set.
# This is done as plymouth is still attached to the default framebuffer (the
# issues caused by not doing this don't seem to be fatal, though, but the
# display may be lost completely until plymouth eventually stops).
# There is no option in plymouth to "reload" a driver, it expects any KMS
# driver to be loaded be before starting it.
IS_KMS=

# This KMS driver needs to be loaded before X server starts, so load it now
# even if we have to shut down plymouth (see above).
NEEDS_LOAD_NOW=

# dkms module that was built when calling from DKMS
DKMS_AUTOLOAD_MODULE=

# Set by check_loaded() when it can't be sure that the correct driver is loaded
# (e.g. in case of the multiple proprietary nvidia drivers which all identify as
# "nvidia" in loaded modules list).
UNSURE=

# Extra module options - used to disable radeon modesetting when no firmware
# is present and it is needed.
MODOPTIONS=

case "$1" in
--load)
        load_driver "$2"
        ;;
--load-dkms-autoload)
        DKMS_AUTOLOAD_MODULE="$2"
        load_driver "$3"
        ;;
--is-disabled)
        check_driver "$2"
        [ $? -eq 1 ]
        # unknown (2) are not considered disabled :)
        ;;
--is-enabled-kms)
        check_driver "$2" && [ -n "$IS_KMS" ]
        ;;
--is-kms-allowed)
        is_kms_allowed
        ;;
--check-dkms-status)
        check_dkms_status
        ;;
--get-all-kms-drivers)
        echo $KMS_DRIVERS
        ;;
--get-active-kms-drivers)
        get_active_kms_drivers
        ;;
--get-initrd-kms-drivers)
        get_initrd_kms_drivers "$2"
        ;;
--check-loaded)
        check_loaded
        ;;
--check-loaded-strict)
        check_loaded && [ -z "$UNSURE" ]
        ;;
--check-speedboot)
        check_dkms_status && check_loaded
        ;;
*)
        usage
        ;;
esac
1	anssi	912	#!/bin/sh
2			#
3			# Display driver helper
4			#
5	anssi	3921	# Copyright (c) 2010, 2011, 2012 Anssi Hannula <anssi.hannula@iki.fi>
6	anssi	912	#
7			# - Load drivers for specified modaliases, skipping disabled display drivers
8			# that would cause conflicts (KMS vs. vesa, KMS vs. proprietary).
9			# - Get information about enabled driver modules
10			# - Check that the loaded modules are correct
11			#
12	anssi	3918	# Note that drivers not known by this script are handled normally, i.e.
13			# loaded automatically as udev normally would've.
14			#
15	anssi	912	# Licensed under terms of GPLv2 or later.
16			#
17			# When updating, check:
18			# - the variables below
19			# - check_driver function
20			# - check_dkms_status function
21			#
22
23	anssi	3918	# DEBUG_DISPLAY_DRIVER_HELPER=yes
24
25	anssi	958	if [ -n "$DEBUG_DISPLAY_DRIVER_HELPER" ]; then
26			echo "$(date) $*" >> /dev/ddh_debug
27			exec 2>>/dev/ddh_debug
28			set -x
29			fi
30	anssi	912
31			export LC_ALL=C
32
33			KMS_DRIVERS="i915 radeon nouveau"
34			# module names at run-time (hence nvidia instead of nvidia*):
35			KNOWN_MODULES="i915\|radeon\|nouveau\|fglrx\|nvidia"
36
37			XORG_i915="intel"
38			CONFLICTS_i915=""
39
40			XORG_nouveau="nouveau"
41			CONFLICTS_nouveau="nv nvidia"
42
43			XORG_radeon="ati radeon"
44			CONFLICTS_radeon="fglrx"
45
46			# Note: no /usr
47			# See end of script for descriptions of global variables.
48			check_driver() {
49			local name="$1"
50	anssi	3921	# modalias is optional, only needed to set MODOPTIONS for radeon
51			# this should not be set when not intending to load the module
52			local modalias="$2"
53
54			MODOPTIONS=
55
56	anssi	912	case "$name" in
57			i915)
58			# implicitely loaded by X.org
59			check_xorg $name 0 \|\| return 1
60			IS_KMS=1
61			;;
62			radeon)
63			# implicitely loaded by X.org
64			check_xorg $name 0 \|\| return 1
65	anssi	997	# Do not load if the proprietary driver is temporarily disabled
66			# on a PowerXpress system.
67			# TODO: this check could be omitted if radeon was explicitely
68			# specified in xorg.conf - but that is only known by check_xorg.
69			check_gl /etc/fglrx/pxpress-free.ld.so.conf && return 1
70	anssi	3921
71	anssi	912	IS_KMS=1
72	anssi	3921	# radeon KMS needs to be loaded before X server
73	anssi	912	NEEDS_LOAD_NOW=1
74	anssi	3921
75			# check if firmware is needed and present (random fw checked)
76			if [ -n "$modalias" ] && [ ! -e "/lib/firmware/radeon/R700_rlc.bin" ]; then
77			# no firmware, lets check if it is needed
78			pciid="${modalias#pci:v00001002d0000}"
79			pciid="${pciid%sv*}"
80			case "$pciid" in
81			# All Radeons up to PALM (see radeon_probe.c)
82			31??\|3E??\|4???\|5???\|7???\|68??\|94??\|95??\|961?\|971?\|98??)
83			# The radeon driver reportedly has some issues
84			# when loaded without firmware - it doesn't fallback to noaccel
85			# nicely as it should, instead the framebuffer gets messed up.
86			# However, these cards are still supported by usermode modesetting,
87			# so just disable KMS. (Mageia bug #3466)
88			# TODO: actually fix the driver to not mess up framebuffer when no firmware
89			# is present and disable accel cleanly so that this workaround
90			# wouldn't be needed -Anssi
91			MODOPTIONS="$MODOPTIONS modeset=0"
92			IS_KMS=
93			# needs to be loaded now in order to set the module option
94			NEEDS_LOAD_NOW=1
95			;;
96			????)
97			# Cards newer than PALM don't support UMS, so don't load the driver
98			# at all (it might still get done by X server, though).
99			return 1
100			;;
101			*)
102			# Modalias parse error, strange. Load normally...
103			;;
104			esac
105			fi
106	anssi	912	;;
107			nouveau)
108	anssi	3920	# implicitely loaded by X.org
109			check_xorg $name 0 \|\| return 1
110	anssi	912	IS_KMS=1
111			;;
112			fglrx)
113			check_xorg fglrx 1 \|\| return 1
114			check_dkms fglrx \|\| return 1
115			;;
116			nvidia)
117			# manually installed driver or a call from check_loaded()
118			UNSURE=1
119			check_xorg nvidia 1 \|\| return 1
120			;;
121			nvidiafb)
122			# this is only reached if nvidiafb is manually unblacklisted
123			return 2
124			;;
125			nvidia*)
126			[ "$name" = "nvidia_current" ] && name=nvidia-current
127			# there are multiple co-installable driver versions, so check
128			# the active alternative as well
129			check_gl /etc/$name/ld.so.conf \|\| return 1
130			check_xorg nvidia 1 \|\| return 1
131			check_dkms $name \|\| return 1
132			;;
133			*)
134			# unknown, will be loaded only if no known drivers were found
135			return 2
136			;;
137			esac
138			return 0
139			}
140
141			# Return success if there is no new pending DKMS build (needed to disallow
142			# speedboot on very early boot).
143			# Previously failed build or a missing module is counted as no pending build.
144			# Note: no /usr
145			check_dkms_status() {
146			[ -e /etc/alternatives/gl_conf ] \|\| return 0
147			# return fast for non-DKMS
148			check_gl /etc/ld.so.conf.d/GL/standard.conf && return 0
149
150			local active="$(ls -l /etc/alternatives/gl_conf \| awk '{ print $NF }')"
151
152			local modname=
153
154			case $active in
155			/etc/nvidia*/ld.so.conf)
156			modname="nvidia${active#*nvidia}"
157			modname="${modname%/*}"
158			;;
159			/etc/ld.so.conf.d/GL/ati.conf)
160			modname="fglrx"
161			;;
162			*)
163			# Unknown DKMS-looking driver,
164			# allow speedboot.
165			return 0
166			;;
167			esac
168			check_dkms "$modname" 1
169			}
170
171			# Check if all loaded kernel modules have correct xorg.conf
172			check_loaded() {
173			for module in $(grep -oE "^($KNOWN_MODULES) " /proc/modules); do
174	anssi	961	# try to unload the driver in case it is not in use before bailing
175			check_driver "$module" \|\| rmmod "$module" &>/dev/null \|\| return 1
176	anssi	912	done
177			return 0
178			}
179
180			# Check that specified DKMS driver is not queued for build for the current
181			# kernel. Used to check if we 1) should disable speedboot for this boot
182			# (--check-dkms-status), and 2) if should should load the currently
183			# existing driver (--load). Doing otherwise might cause us to load a wrong old
184			# version of the driver that had been installed using e.g. binary DKMS
185			# packages.
186			# Note: no /usr
187			check_dkms() {
188			local driver="$1"
189			local force="$2"
190
191			# If called from DKMS itself or we are not in rc.sysinit anymore,
192			# there are no pending builds.
193			if [ -z "$force" ]; then
194			[ "$DKMS_AUTOLOAD_MODULE" = "$driver" ] && return 0
195			[ -z "$STARTUP" ] && [ ! -f "/dev/.in_sysinit" ] && return 0
196			fi
197
198			local found=
199			local uname_r="$(uname -r)"
200
201			for dir in /var/lib/dkms/$driver/*; do
202			[ -e "$dir" ] \|\| return 0 # no module, no build; or no /var
203			[ -L "$dir" ] && continue # not a module version
204			found=1 # module version found
205			[ -e "$dir/$uname_r" ] && return 0
206			[ -e "/var/lib/dkms-binary/$driver/$(basename "$dir")/$uname_r" ] && return 0
207
208			if [ -e "$dir/build/make.log" ]; then
209			# Build has failed for some kernel, check if it is this one.
210			# If so, there is no point in returning 1.
211			grep -q "^DKMS make\.log.* $uname_r " && return 0
212			fi
213			done
214
215			# if module versions were found but none were built for this kernel, return 1
216			[ -n "$found" ] && return 1 \|\| return 0
217			}
218
219			# Note: no /usr
220			check_gl() {
221			local alt_inode="$(stat -L -c%i $1 2>/dev/null)"
222			[ -n "$alt_inode" ] \|\| return 1
223			[ -n "$GL_INODE" ] \|\| GL_INODE="$(stat -L -c%i /etc/alternatives/gl_conf 2>/dev/null)"
224			[ "$alt_inode" = "$GL_INODE" ] \|\| return 1
225			return 0
226			}
227
228			# Note: no /usr
229			get_xorg_drivers() {
230			if [ -z "$XORG_DRIVERS" ]; then
231			XORG_DRIVERS="$(cat /etc/X11/xorg.conf /etc/X11/xorg.conf.d/*.conf 2>/dev/null \|
232			awk -F'"' -vORS=' ' -vIGNORECASE=1 '
233			/^[[:space:]]+*section[[:space:]]+"device"/ { device=1 }
234			/endsection/ { device=0 }
235			/^[[:space:]]driver[[:space:]]".*"/ { if (device) drivers[$2]=$2 }
236			END { for (driver in drivers) print driver }
237			')"
238			[ -n "$XORG_DRIVERS" ] \|\| XORG_DRIVERS="-"
239			fi
240			}
241
242			# Note: no /usr
243	anssi	3918	# parameter 1: KMS module or xorg driver
244			# parameter 2: 1 - check if the driver is explicitely enabled
245			# 0 - check only for conflicts
246	anssi	912	check_xorg() {
247			local driver="$1"
248			local explicit_only="$2"
249
250			eval local xorg_drivers=\"\$XORG_$driver\"
251	anssi	959	[ -n "$xorg_drivers" ] \|\| xorg_drivers="$driver"
252	anssi	912	eval local conflicts=\"\$CONFLICTS_$driver\"
253
254			get_xorg_drivers
255
256			conflict_found=
257			for enabled_driver in $XORG_DRIVERS; do
258			for xorg_driver in $xorg_drivers; do
259			[ "$enabled_driver" = "$xorg_driver" ] && return 0
260			done
261
262			# if the X.org driver can be loaded implicitely, check that
263			# there are no conflicting drivers that override the driver
264			if [ "$explicit_only" = "0" -a -z "$conflict_found" ]; then
265			for conflict in vesa $conflicts; do
266			if [ "$enabled_driver" = "$conflict" ]; then
267			conflict_found=1
268			continue 2
269			# continue loop to check for an explicit load
270			fi
271			done
272			fi
273			done
274
275			# in case of a conflict, do not load the module
276			[ -n "$conflict_found" ] && return 1
277
278			# no driver is selected - don't load if explicit_only is 1
279			[ "$explicit_only" = "1" ] && return 1
280
281	anssi	1681	# implicit load allowed; only load if there is evidence that this is
282			# not a live cd or similar with automatic configuration occurring later
283			# in the boot process (which might configure a driver conflicting with
284			# the implicit driver, e.g. a proprietary one)
285			# TODO: Could this be replaced with a more robust check?
286			[ -e "/etc/X11/xorg.conf" ] \|\| [ -e "/etc/sysconfig/harddrake2/kernels" ] \|\|
287			[ -e "/etc/sysconfig/harddrake2/xorg" ] \|\| [ -e "/boot/grub/menu.lst" ]
288	anssi	912	}
289
290			# Load the driver for the specified modalias, if configured.
291			# Note: no /usr
292			load_driver() {
293	anssi	3918	local modalias="$1"
294	anssi	912	local modulename
295			local load_default=1
296
297	anssi	3918	for modulename in $(/sbin/modprobe -Rq "$modalias"); do
298	anssi	3921	check_driver "$modulename" "$modalias"
299	anssi	912	case $? in
300			1) # a driver which needs handling by this script matches
301			# the modalias, but was not configured - do not run
302			# the generic modprobe if no other drivers are
303			# configured either
304			load_default=
305			continue
306			;;
307			2) continue
308			;;
309			esac
310
311			if [ -n "$IS_KMS" ]; then
312			grep -q "^$modulename " /proc/modules && return 0
313			echo "$modulename" > /dev/.late_kms 2>/dev/null
314			# If NEEDS_LOAD_NOW is not set and plymouth is running,
315			# skip loading the driver to avoid quitting plymouth.
316			# The driver will be loaded later by X server itself.
317			[ -z "$NEEDS_LOAD_NOW" ] && /bin/plymouth --ping 2>/dev/null && return 0
318			/bin/plymouth quit 2>/dev/null
319			fi
320	anssi	3921	/sbin/modprobe -b "$modulename" $MODOPTIONS && return 0
321	anssi	912	done
322
323			# no specially handled modules were loaded, so load all modules normally
324			# unless $load_default was set above
325	anssi	3918	[ -z "$load_default" ] \|\| /sbin/modprobe -b "$modalias"
326	anssi	912	}
327
328			is_kms_allowed() {
329			for driver in $KMS_DRIVERS; do
330			# Check all drivers for conflicts only.
331			check_xorg $driver 0 \|\| return 1
332			done
333			return 0
334			}
335
336			get_initrd_kms_drivers() {
337			local initrd="$1"
338
339			local kms_drivers="$(echo "$KMS_DRIVERS" \| tr " " "\|")"
340			zcat "$initrd" \| cpio -t --quiet \| sed -nr "s,./($kms_drivers)\.ko.$,\1,p"
341			}
342
343			get_active_kms_drivers() {
344			local kms_drivers=
345			for device in $(grep -l 0x03 /sys/bus/pci/devices/0000\:0*/class); do
346			[ -e "$device" ] \|\| continue
347			device="$(dirname $device)"
348			[ -f "$device/modalias" ] \|\| continue
349			modalias="$(cat "$device/modalias")"
350	anssi	2245	for modulename in $(/sbin/modprobe -Rq "$modalias"); do
351	anssi	912	IS_KMS=
352	anssi	3921	check_driver "$modulename" "$modalias" \|\| continue
353	anssi	912	[ -n "$IS_KMS" ] && echo $modulename
354			done
355			done
356			}
357
358			usage() {
359			cat <<EOF
360			Usage: $0 action [arguments]
361
362			Known actions:
363
364			--load MODALIAS
365			Load drivers matching MODALIAS, checking that they are enabled and
366			configured.
367
368			--load-dkms-autoload MODNAME MODALIAS
369			Same as --load, but assume MODNAME is built and correct so that
370			checking dkms status is unnecessary.
371
372			--is-disabled MODNAME
373			Checks whether the driver corresponding to MODNAME is disabled (e.g.
374			a conflicting driver is configured, etc.). Unknown MODNAMEs are
375			considered not disabled.
376
377			--is-enabled-kms MODNAME
378			Checks whether the driver corresponding to MODNAME is enabled and
379			MODNAME is a known KMS module. Note that drivers may be enabled even
380			if there is no such hardware. This just checks that there are
381			no conflicting drivers in use etc.
382
383			--is-kms-allowed
384			Checks whether it is ok to load KMS drivers in initrd. This returns
385			a failure when a conflicting driver is set up (vesa or a proprietary
386			one).
387
388			--get-all-kms-drivers
389			Get a list of the known KMS drivers.
390
391			--get-active-kms-drivers
392			Get a list of the known KMS drivers which are enabled and the hardware
393			is present.
394
395			--get-initrd-kms-drivers INITRD
396			Get a list of the known KMS drivers in initrd INITRD.
397
398			--check-dkms-status
399			Checks if there are no pending DKMS builds for the currently enabled
400			drivers.
401
402			--check-loaded
403			Checks that there are no disabled drivers loaded.
404
405			--check-speedboot
406			Does --check-dkms-status and --check-loaded.
407
408			--check-loaded-strict
409	anssi	3918	As --check-loaded, and consider ambiguous cases (e.g. nvidia where
410	anssi	912	we can't detect if the loaded driver has the correct version) as
411			failure.
412			EOF
413			}
414
415			# clear global variables
416
417			# cache for check_gl()
418			GL_INODE=
419
420			# cache for check_xorg()
421			XORG_DRIVERS=
422
423			# The driver is a KMS enabled driver. This will cause the script to quit
424			# plymouth when a driver is loaded by --load and NEEDS_LOAD_NOW below is set.
425			# This is done as plymouth is still attached to the default framebuffer (the
426			# issues caused by not doing this don't seem to be fatal, though, but the
427			# display may be lost completely until plymouth eventually stops).
428			# There is no option in plymouth to "reload" a driver, it expects any KMS
429			# driver to be loaded be before starting it.
430			IS_KMS=
431
432			# This KMS driver needs to be loaded before X server starts, so load it now
433			# even if we have to shut down plymouth (see above).
434			NEEDS_LOAD_NOW=
435
436			# dkms module that was built when calling from DKMS
437			DKMS_AUTOLOAD_MODULE=
438
439			# Set by check_loaded() when it can't be sure that the correct driver is loaded
440			# (e.g. in case of the multiple proprietary nvidia drivers which all identify as
441			# "nvidia" in loaded modules list).
442			UNSURE=
443
444	anssi	3921	# Extra module options - used to disable radeon modesetting when no firmware
445			# is present and it is needed.
446			MODOPTIONS=
447
448	anssi	912	case "$1" in
449			--load)
450			load_driver "$2"
451			;;
452			--load-dkms-autoload)
453			DKMS_AUTOLOAD_MODULE="$2"
454			load_driver "$3"
455			;;
456			--is-disabled)
457			check_driver "$2"
458			[ $? -eq 1 ]
459			# unknown (2) are not considered disabled :)
460			;;
461			--is-enabled-kms)
462			check_driver "$2" && [ -n "$IS_KMS" ]
463			;;
464			--is-kms-allowed)
465			is_kms_allowed
466			;;
467			--check-dkms-status)
468			check_dkms_status
469			;;
470			--get-all-kms-drivers)
471			echo $KMS_DRIVERS
472			;;
473			--get-active-kms-drivers)
474			get_active_kms_drivers
475			;;
476			--get-initrd-kms-drivers)
477			get_initrd_kms_drivers "$2"
478			;;
479			--check-loaded)
480			check_loaded
481			;;
482			--check-loaded-strict)
483			check_loaded && [ -z "$UNSURE" ]
484			;;
485			--check-speedboot)
486			check_dkms_status && check_loaded
487			;;
488			*)
489			usage
490			;;
491			esac