forked from mirrors/nixpkgs
* Rewrote the hardware scanner in Perl.
svn path=/nixos/trunk/; revision=10405
This commit is contained in:
Eelco Dolstra
parent
c1f619a087
commit
7363895f6d
174
installer/nixos-hardware-scan.pl
Normal file
174
installer/nixos-hardware-scan.pl
Normal file
|
|
@ -0,0 +1,174 @@
|
|||
#! @perl@ -w
|
||||
|
||||
use File::Spec;
|
||||
use File::Basename;
|
||||
|
||||
|
||||
my @kernelModules = ();
|
||||
my @initrdKernelModules = ();
|
||||
|
||||
|
||||
# Read a file, returning undef if the file cannot be opened.
|
||||
sub readFile {
|
||||
my $filename = shift;
|
||||
my $res;
|
||||
if (open FILE, "<$filename") {
|
||||
my $prev = $/;
|
||||
undef $/;
|
||||
$res = <FILE>;
|
||||
$/ = $prev;
|
||||
close FILE;
|
||||
chomp $res;
|
||||
}
|
||||
return $res;
|
||||
}
|
||||
|
||||
|
||||
my $cpuinfo = readFile "/proc/cpuinfo";
|
||||
|
||||
|
||||
sub hasCPUFeature {
|
||||
my $feature = shift;
|
||||
return $cpuinfo =~ /^flags\s*:.* $feature( |$)/m;
|
||||
}
|
||||
|
||||
|
||||
# Detect the number of CPU cores.
|
||||
my $cpus = scalar (grep {/^processor\s*:/} (split '\n', $cpuinfo));
|
||||
|
||||
|
||||
# CPU frequency scaling. Not sure about this test.
|
||||
push @kernelModules, "acpi-cpufreq" if hasCPUFeature "acpi";
|
||||
|
||||
|
||||
# Virtualization support?
|
||||
push @kernelModules, "kvm-intel" if hasCPUFeature "vmx";
|
||||
push @kernelModules, "kvm-amd" if hasCPUFeature "svm";
|
||||
|
||||
|
||||
# Look at the PCI devices and add necessary modules. Note that most
|
||||
# modules are auto-detected so we don't need to list them here.
|
||||
# However, some are needed in the initrd to boot the system.
|
||||
|
||||
sub pciCheck {
|
||||
my $path = shift;
|
||||
my $vendor = readFile "$path/vendor";
|
||||
my $device = readFile "$path/device";
|
||||
my $class = readFile "$path/class";
|
||||
|
||||
my $module;
|
||||
if (-e "$path/driver/module") {
|
||||
$module = basename `readlink -f $path/driver/module`;
|
||||
chomp $module;
|
||||
}
|
||||
|
||||
print "$path: $vendor $device $class";
|
||||
print " $module" if defined $module;
|
||||
print "\n";
|
||||
|
||||
if (defined $module) {
|
||||
# See the bottom of http://pciids.sourceforge.net/pci.ids for
|
||||
# device classes.
|
||||
if (# Mass-storage controller. Definitely important.
|
||||
$class =~ /^0x01/ ||
|
||||
|
||||
# Firewire controller. A disk might be attached.
|
||||
$class =~ /^0x0c00/ ||
|
||||
|
||||
# USB controller. Needed if we want to use the
|
||||
# keyboard when things go wrong in the initrd.
|
||||
$class =~ /^0x0c03/
|
||||
)
|
||||
{
|
||||
push @initrdKernelModules, $module;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
foreach my $path (glob "/sys/bus/pci/devices/*") {
|
||||
pciCheck $path;
|
||||
}
|
||||
|
||||
|
||||
# Idem for USB devices.
|
||||
|
||||
sub usbCheck {
|
||||
my $path = shift;
|
||||
my $class = readFile "$path/bInterfaceClass";
|
||||
my $subclass = readFile "$path/bInterfaceSubClass";
|
||||
my $protocol = readFile "$path/bInterfaceProtocol";
|
||||
|
||||
my $module;
|
||||
if (-e "$path/driver/module") {
|
||||
$module = basename `readlink -f $path/driver/module`;
|
||||
chomp $module;
|
||||
}
|
||||
|
||||
print "$path: $class $subclass $protocol";
|
||||
print " $module" if defined $module;
|
||||
print "\n";
|
||||
|
||||
if (defined $module) {
|
||||
if (# Mass-storage controller. Definitely important.
|
||||
$class eq "08" ||
|
||||
|
||||
# Keyboard. Needed if we want to use the
|
||||
# keyboard when things go wrong in the initrd.
|
||||
($class eq "03" && $protocol eq "01")
|
||||
)
|
||||
{
|
||||
push @initrdKernelModules, $module;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
foreach my $path (glob "/sys/bus/usb/devices/*") {
|
||||
if (-e "$path/bInterfaceClass") {
|
||||
usbCheck $path;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
# Generate the configuration file.
|
||||
|
||||
sub removeDups {
|
||||
my %seen;
|
||||
my @res = ();
|
||||
foreach my $s (@_) {
|
||||
if (!defined $seen{$s}) {
|
||||
$seen{$s} = "";
|
||||
push @res, $s;
|
||||
}
|
||||
}
|
||||
return @res;
|
||||
}
|
||||
|
||||
sub toNixExpr {
|
||||
my $res = "";
|
||||
foreach my $s (@_) {
|
||||
$res .= " \"$s\"";
|
||||
}
|
||||
return $res;
|
||||
}
|
||||
|
||||
my $initrdKernelModules = toNixExpr(removeDups @initrdKernelModules);
|
||||
my $kernelModules = toNixExpr(removeDups @kernelModules);
|
||||
|
||||
|
||||
print <<EOF ;
|
||||
# This is a generated file. Do not modify!
|
||||
# Make changes to /etc/nixos/configuration.nix instead.
|
||||
{
|
||||
boot = {
|
||||
initrd = {
|
||||
extraKernelModules = [ $initrdKernelModules ];
|
||||
};
|
||||
kernelModules = [ $kernelModules ];
|
||||
};
|
||||
|
||||
nix = {
|
||||
maxJobs = $cpus;
|
||||
};
|
||||
}
|
||||
EOF
|
||||
|
|
@ -1,144 +0,0 @@
|
|||
#! @shell@ -e
|
||||
|
||||
|
||||
kernelModules=
|
||||
initrdKernelModules=
|
||||
|
||||
hasCPUFeature() {
|
||||
local feature="$1"
|
||||
cat /proc/cpuinfo | grep -q "^flags.* $feature\( \|$\)"
|
||||
}
|
||||
|
||||
needKernelModule() {
|
||||
kernelModules="$kernelModules \"$1\""
|
||||
}
|
||||
|
||||
needInitrdKernelModule() {
|
||||
initrdKernelModules="$initrdKernelModules \"$1\""
|
||||
}
|
||||
|
||||
|
||||
# Detect the number of CPU cores.
|
||||
cpus="$(cat /proc/cpuinfo | grep '^processor' | wc -l)"
|
||||
|
||||
|
||||
# CPU frequency scaling. Not sure about this test.
|
||||
if hasCPUFeature "acpi"; then
|
||||
needKernelModule "acpi-cpufreq"
|
||||
fi
|
||||
|
||||
|
||||
# Virtualization support?
|
||||
if hasCPUFeature "vmx"; then
|
||||
needKernelModule "kvm-intel"
|
||||
fi
|
||||
|
||||
if hasCPUFeature "svm"; then
|
||||
needKernelModule "kvm-amd"
|
||||
fi
|
||||
|
||||
|
||||
# Look at the PCI devices and add necessary modules. Note that most
|
||||
# modules are auto-detected so we don't need to list them here.
|
||||
# However, some are needed in the initrd to boot the system.
|
||||
|
||||
pciCheck() {
|
||||
local path="$1"
|
||||
local vendor="$(cat $path/vendor)"
|
||||
local device="$(cat $path/device)"
|
||||
local class="$(cat $path/class)"
|
||||
|
||||
local module
|
||||
if test -e "$path/driver/module"; then
|
||||
module=$(basename $(readlink -f $path/driver/module))
|
||||
fi
|
||||
|
||||
echo "$path: $vendor $device $class $module"
|
||||
|
||||
if test -n "$module"; then
|
||||
# See the bottom of http://pciids.sourceforge.net/pci.ids for
|
||||
# device classes.
|
||||
case $class in
|
||||
0x01*)
|
||||
# Mass-storage controller. Definitely important.
|
||||
needInitrdKernelModule $module
|
||||
;;
|
||||
0x0c00*)
|
||||
# Firewire controller. A disk might be attached.
|
||||
needInitrdKernelModule $module
|
||||
;;
|
||||
0x0c03*)
|
||||
# USB controller. Needed if we want to use the
|
||||
# keyboard when things go wrong in the initrd.
|
||||
needInitrdKernelModule $module
|
||||
;;
|
||||
esac
|
||||
fi
|
||||
}
|
||||
|
||||
for path in /sys/bus/pci/devices/*; do
|
||||
pciCheck "$path"
|
||||
done
|
||||
|
||||
|
||||
# Idem for USB devices.
|
||||
|
||||
usbCheck() {
|
||||
local path="$1"
|
||||
local class="$(cat $path/bInterfaceClass)"
|
||||
local subclass="$(cat $path/bInterfaceSubClass)"
|
||||
local protocol="$(cat $path/bInterfaceProtocol)"
|
||||
|
||||
local module
|
||||
if test -e "$path/driver/module"; then
|
||||
module=$(basename $(readlink -f $path/driver/module))
|
||||
fi
|
||||
|
||||
echo "$path: $class $subclass $protocol $module"
|
||||
|
||||
if test -n "$module"; then
|
||||
# See http://www.linux-usb.org/usb.ids for
|
||||
# classes/subclasses/protocols.
|
||||
case $class:$subclass:$protocol in
|
||||
08:*)
|
||||
# Mass-storage controller. Definitely important.
|
||||
needInitrdKernelModule $module
|
||||
;;
|
||||
03:*:01)
|
||||
# Keyboard. Needed if we want to use the
|
||||
# keyboard when things go wrong in the initrd.
|
||||
needInitrdKernelModule $module
|
||||
;;
|
||||
esac
|
||||
fi
|
||||
}
|
||||
|
||||
for path in /sys/bus/usb/devices/*; do
|
||||
if test -e "$path/bInterfaceClass"; then
|
||||
usbCheck "$path"
|
||||
fi
|
||||
done
|
||||
|
||||
|
||||
# Remove duplicate modules. !!! preserve order
|
||||
kernelModules=$(for i in $kernelModules; do echo $i; done | sort | uniq)
|
||||
initrdKernelModules=$(for i in $initrdKernelModules; do echo $i; done | sort | uniq)
|
||||
|
||||
|
||||
# Generation the configuration file.
|
||||
cat <<EOF
|
||||
# This is a generated file. Do not modify!
|
||||
# Make changes to /etc/nixos/configuration.nix instead.
|
||||
{
|
||||
boot = {
|
||||
initrd = {
|
||||
extraKernelModules = [ $(echo $initrdKernelModules) ];
|
||||
};
|
||||
kernelModules = [ $(echo $kernelModules) ];
|
||||
};
|
||||
|
||||
nix = {
|
||||
maxJobs = $cpus;
|
||||
};
|
||||
}
|
||||
EOF
|
||||
Loading…
Reference in a new issue