2009-07-05 Bean <bean123ch@gmail.com>

* conf/common.rmk (lua_mode_SOURCES): Add script/lua/lstrlib.c.

	* script/lua/grub_lib.c (grub_lua_run): Check input parameter.
	(grub_lua_getenv): Likewise.
	(grub_lua_setenv): Likewise.
	(save_errno): New function.
	(push_result): Likewise.
	(grub_lua_enum_device): Likewise.
	(grub_lua_enum_file): Likewise.
	(grub_lua_file_open): Likewise.
	(grub_lua_file_close): Likewise.
	(grub_lua_file_seek): Likewise.
	(grub_lua_file_read): Likewise.
	(grub_lua_file_getline): Likewise.
	(grub_lua_file_getsize): Likewise.
	(grub_lua_file_getpos): Likewise.
	(grub_lua_file_eof): Likewise.
	(grub_lua_file_exist): Likewise.
	(grub_lua_add_menu): Likewise.

	* script/lua/grub_lua.h (isupper): New inline function.
	(islower): Likewise.
	(ispunct): Likewise.
	(isxdigit): Likewise.
	(strcspn): Change to normal function.
	(strpbkr): New function declaration.
	(memchr): Likewise.

	* script/lua/grub_main.c (scan_str): New function.
	(strcspn): Likewise.
	(strpbrk): Likewise.
	(memchr): Likewise.

	* script/lua/linit.c (lualibs): Enable the string library.

	* util/osdetect.lua: New file.

util/osdetect.lua

@@ -0,0 +1,111 @@
+#!lua
+--
+-- Copyright (C) 2009  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 <http://www.gnu.org/licenses/>.
+--
+
+function enum_device (device, fs, uuid)
+  local root
+  local title
+  local source
+  local kernels = {}
+  local kernel_num = 0
+
+  local function enum_file (name)
+    local version
+
+    version = string.match (name, "vmlinuz%-(.*)")
+    if (version ~= nil) then
+      table.insert (kernels, version)
+      kernel_num = kernel_num + 1
+    end
+  end
+
+  local function sort_kernel (first, second)
+    local a1, a2, a3, a4, b1, b2, b3, b4
+
+    a1, a2, a3, a4 = string.match (first, "(%d+)%.?(%d*).?(%d*)%-?(%d*)")
+    b1, b2, b3, b4 = string.match (second, "(%d+)%.?(%d*).?(%d*)%-?(%d*)")
+    return (a1 > b1) or (a2 > b2) or (a3 > b3) or (a4 < b4);
+  end
+
+  root = "(" .. device .. ")/"
+  source = "root (" .. device .. ")\nchainloader +1"
+  title = nil
+  if (grub.file_exist (root .. "bootmgr") and
+      grub.file_exist (root .. "boot/bcd")) then
+    title = "Windows Vista bootmgr"
+  elseif (grub.file_exist (root .. "ntldr") and
+	  grub.file_exist (root .. "ntdetect.com") and
+	  grub.file_exist (root .. "boot.ini")) then
+    title = "Windows NT/2000/XP loader"
+  elseif (grub.file_exist (root .. "windows/win.com")) then
+    title = "Windows 98/ME"
+  elseif (grub.file_exist (root .. "io.sys") and
+	  grub.file_exist (root .. "command.com")) then
+    title = "MS-DOS"
+  elseif (grub.file_exist (root .. "kernel.sys")) then
+    title = "FreeDOS"
+  elseif (grub.file_exist (root .. "boot/loader") and
+	  grub.file_exist (root .. "boot/devices.hints")) then
+    source = "root (" .. device .. ")\nfreebsd /boot/loader" ..
+      "\nfreebsd_loadenv /boot/device.hints"
+    title = "FreeBSD"
+  else
+    grub.enum_file (enum_file, root .. "boot")
+    if kernel_num ~= 0 then
+      table.sort (kernels, sort_kernel)
+      for i = 1, kernel_num do
+	local initrd
+
+	title = "Linux " .. kernels[i]
+	source = "root (" .. device ..
+	  ")\nlinux /boot/vmlinuz-" .. kernels[i] ..
+	  " root=UUID=" .. uuid .. " ro"
+
+	if grub.file_exist (root .. "boot/initrd-" ..
+			    kernels[i] .. ".img") then
+	  initrd = "\ninitrd /boot/initrd-" .. kernels[i] .. ".img"
+	elseif grub.file_exist (root .. "boot/initrd.img-" .. kernels[i]) then
+	  initrd = "\ninitrd /boot/initrd.img-" .. kernels[i]
+	elseif grub.file_exist (root .. "boot/initrd-" .. kernels[i]) then
+	  initrd = "\ninitrd /boot/initrd-" .. kernels[i]
+	else
+	  initrd = ""
+	end
+
+	grub.add_menu (source .. initrd, title)
+	grub.add_menu (source .. " single" .. initrd,
+		       title .. " (single-user mode)")
+      end
+      return 0
+    end
+  end
+
+  if title == nil then
+    local partition = string.match (device, ".*,(%d+)")
+
+    if (partition ~= nil) and (tonumber (partition) > 4) then
+      return 0
+    end
+
+    title = "Other OS"
+  end
+
+  grub.add_menu (source, title)
+  return 0
+end
+
+grub.enum_device (enum_device)