1 /* search.c - search devices based on a file or a filesystem label */
3 * GRUB -- GRand Unified Bootloader
4 * Copyright (C) 2005,2007,2008,2009 Free Software Foundation, Inc.
6 * GRUB is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * GRUB is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with GRUB. If not, see <http://www.gnu.org/licenses/>.
20 #include <grub/types.h>
21 #include <grub/misc.h>
25 #include <grub/device.h>
26 #include <grub/file.h>
28 #include <grub/extcmd.h>
30 static const struct grub_arg_option options
[] =
32 {"file", 'f', 0, "search devices by a file", 0, 0},
33 {"label", 'l', 0, "search devices by a filesystem label", 0, 0},
34 {"fs-uuid", 'u', 0, "search devices by a filesystem UUID", 0, 0},
35 {"set", 's', GRUB_ARG_OPTION_OPTIONAL
, "set a variable to the first device found", "VAR", ARG_TYPE_STRING
},
36 {"no-floppy", 'n', 0, "do not probe any floppy drive", 0, 0},
50 search_fs (const char *key
, const char *var
, int no_floppy
, enum options type
)
54 grub_fs_autoload_hook_t saved_autoload
;
56 auto int iterate_device (const char *name
);
57 int iterate_device (const char *name
)
61 /* Skip floppy drives when requested. */
63 name
[0] == 'f' && name
[1] == 'd' && name
[2] >= '0' && name
[2] <= '9')
66 if (type
== SEARCH_FILE
)
72 len
= grub_strlen (name
) + 2 + grub_strlen (key
) + 1;
73 p
= grub_realloc (buf
, len
);
78 grub_sprintf (buf
, "(%s)%s", name
, key
);
80 file
= grub_file_open (buf
);
84 grub_file_close (file
);
89 /* type is SEARCH_FS_UUID or SEARCH_LABEL */
92 int (*compare_fn
) (const char *, const char *);
95 dev
= grub_device_open (name
);
98 fs
= grub_fs_probe (dev
);
100 (type
== SEARCH_FS_UUID
) ? grub_strcasecmp
: grub_strcmp
;
102 if (fs
&& ((type
== SEARCH_FS_UUID
) ? fs
->uuid
: fs
->label
))
104 if (type
== SEARCH_FS_UUID
)
105 fs
->uuid (dev
, &quid
);
107 fs
->label (dev
, &quid
);
109 if (grub_errno
== GRUB_ERR_NONE
&& quid
)
111 if (compare_fn (quid
, key
) == 0)
118 grub_device_close (dev
);
126 grub_env_set (var
, name
);
128 grub_printf (" %s", name
);
131 grub_errno
= GRUB_ERR_NONE
;
132 return (found
&& var
);
135 /* First try without autoloading if we're setting variable. */
138 saved_autoload
= grub_fs_autoload_hook
;
139 grub_fs_autoload_hook
= 0;
140 grub_device_iterate (iterate_device
);
142 /* Restore autoload hook. */
143 grub_fs_autoload_hook
= saved_autoload
;
145 /* Retry with autoload if nothing found. */
146 if (grub_errno
== GRUB_ERR_NONE
&& count
== 0)
147 grub_device_iterate (iterate_device
);
150 grub_device_iterate (iterate_device
);
154 if (grub_errno
== GRUB_ERR_NONE
&& count
== 0)
155 grub_error (GRUB_ERR_FILE_NOT_FOUND
, "no such device: %s", key
);
159 grub_cmd_search (grub_extcmd_t cmd
, int argc
, char **args
)
161 struct grub_arg_list
*state
= cmd
->state
;
165 return grub_error (GRUB_ERR_INVALID_COMMAND
, "no argument specified");
167 if (state
[SEARCH_SET
].set
)
168 var
= state
[SEARCH_SET
].arg
? state
[SEARCH_SET
].arg
: "root";
170 if (state
[SEARCH_LABEL
].set
)
171 search_fs (args
[0], var
, state
[SEARCH_NO_FLOPPY
].set
, SEARCH_LABEL
);
172 else if (state
[SEARCH_FS_UUID
].set
)
173 search_fs (args
[0], var
, state
[SEARCH_NO_FLOPPY
].set
, SEARCH_FS_UUID
);
174 else if (state
[SEARCH_FILE
].set
)
175 search_fs (args
[0], var
, state
[SEARCH_NO_FLOPPY
].set
, SEARCH_FILE
);
177 return grub_error (GRUB_ERR_INVALID_COMMAND
, "unspecified search type");
182 static grub_extcmd_t cmd
;
184 GRUB_MOD_INIT(search
)
187 grub_register_extcmd ("search", grub_cmd_search
,
188 GRUB_COMMAND_FLAG_BOTH
,
189 "search [-f|-l|-u|-s|-n] NAME",
190 "Search devices by file, filesystem label or filesystem UUID."
191 " If --set is specified, the first device found is"
192 " set to a variable. If no variable name is"
193 " specified, \"root\" is used.",
197 GRUB_MOD_FINI(search
)
199 grub_unregister_extcmd (cmd
);