From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from list by lists.gnu.org with archive (Exim 4.71) id 1THOYi-0005Vd-EB for mharc-grub-devel@gnu.org; Thu, 27 Sep 2012 20:36:00 -0400 Received: from eggs.gnu.org ([208.118.235.92]:57646) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1THLqc-0002WJ-2w for grub-devel@gnu.org; Thu, 27 Sep 2012 17:42:19 -0400 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1THLqa-0001EF-Mq for grub-devel@gnu.org; Thu, 27 Sep 2012 17:42:18 -0400 Received: from the.earth.li ([46.43.34.31]:32846) by eggs.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1THLqa-0001E4-Fz for grub-devel@gnu.org; Thu, 27 Sep 2012 17:42:16 -0400 Received: from noodles by the.earth.li with local (Exim 4.72) (envelope-from ) id 1THLqV-0004Kn-Gb for grub-devel@gnu.org; Thu, 27 Sep 2012 22:42:11 +0100 Date: Thu, 27 Sep 2012 14:42:11 -0700 From: Jonathan McDowell To: grub-devel@gnu.org Subject: [PATCH] Add testpci command Message-ID: <20120927214211.GS2289@earth.li> MIME-Version: 1.0 Content-Type: text/plain; charset=us-ascii Content-Disposition: inline User-Agent: Mutt/1.5.20 (2009-06-14) X-detected-operating-system: by eggs.gnu.org: GNU/Linux 2.6 (newer, 3) X-Received-From: 46.43.34.31 X-Mailman-Approved-At: Thu, 27 Sep 2012 20:35:59 -0400 X-BeenThere: grub-devel@gnu.org X-Mailman-Version: 2.1.14 Precedence: list Reply-To: The development of GNU GRUB List-Id: The development of GNU GRUB List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-List-Received-Date: Thu, 27 Sep 2012 21:42:19 -0000 I have a machine with both Linux and Windows installed on the hard drive. Linux runs on the bare metal and I occasionally run the Windows install in a VM using KVM pointed at /dev/sda. However if I'm not quick enough, or Windows decides to reboot when I'm not around to notice, the grub running under KVM will decide to boot Linux and much confusion ensues. I couldn't find an easy way to test what the running environment was, so I knocked up a simple testpci command that allows me to test for the existence of a PCI device and change the default boot option based on that. I have: | if testpci 8086:1237; then | set default="2" | fi in my grub.cfg to correctly select the Windows partition if the qemu provided host bridge is present. Signed-off-by: Jonathan McDowell ----- === modified file 'grub-core/Makefile.core.def' --- grub-core/Makefile.core.def 2012-09-08 07:40:24 +0000 +++ grub-core/Makefile.core.def 2012-09-27 21:00:31 +0000 @@ -837,6 +837,12 @@ }; module = { + name = testpci; + common = commands/testpci.c; + enable = pci; +}; + +module = { name = true; common = commands/true.c; }; === added file 'grub-core/commands/testpci.c' --- grub-core/commands/testpci.c 1970-01-01 00:00:00 +0000 +++ grub-core/commands/testpci.c 2012-09-27 21:15:02 +0000 @@ -0,0 +1,75 @@ +/* testpci.c - Test for PCI device existence. */ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 2012 Free Software Foundation, Inc. + * + * GRUB 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. + * + * GRUB 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 GRUB. If not, see . + */ + +#include +#include +#include +#include +#include + +GRUB_MOD_LICENSE ("GPLv3+"); + +static int pcifound; +static grub_pci_id_t pcisearch; + +static int NESTED_FUNC_ATTR +grub_testpci_iter (grub_pci_device_t dev __attribute__ ((unused)), + grub_pci_id_t pciid) +{ + if (pciid == pcisearch) { + pcifound = 1; + return 1; + } + + return 0; +} + +static grub_err_t +grub_cmd_testpci (grub_command_t ctxt __attribute__ ((unused)), + int argc, + char **args) +{ + pcifound = 0; + + if (argc != 1) + return grub_error (GRUB_ERR_BAD_ARGUMENT, "Device to search for required"); + + if ((grub_strlen(args[0]) != 9) || args[0][4] != ':') + return grub_error (GRUB_ERR_BAD_ARGUMENT, "Device ID must be xxxx:yyyy"); + + pcisearch = grub_strtoul(args[0], 0, 16) + + (grub_strtoul(&args[0][5], 0, 16) << 16); + + grub_pci_iterate (grub_testpci_iter); + + return pcifound ? GRUB_ERR_NONE : grub_error (GRUB_ERR_TEST_FAILURE, "false"); +} + +static grub_command_t cmd; + +GRUB_MOD_INIT(testpci) +{ + cmd = grub_register_command ("testpci", grub_cmd_testpci, "xxxx:yyyy", + N_("Test for PCI device existence.")); +} + +GRUB_MOD_FINI(testpci) +{ + grub_unregister_command (cmd); +} ----- J. -- I'm out of bed and dressed. What more do you want?