summaryrefslogtreecommitdiffstats
path: root/mdk-stage1/probe-modules.c
diff options
context:
space:
mode:
Diffstat (limited to 'mdk-stage1/probe-modules.c')
-rw-r--r--mdk-stage1/probe-modules.c11
1 files changed, 9 insertions, 2 deletions
diff --git a/mdk-stage1/probe-modules.c b/mdk-stage1/probe-modules.c
index 4c440aa16..55ff7df1d 100644
--- a/mdk-stage1/probe-modules.c
+++ b/mdk-stage1/probe-modules.c
@@ -18,6 +18,9 @@
#include "probing.h"
#include "frontend.h"
#include <stdlib.h>
+#include <sys/stat.h>
+#include <string.h>
+#include "utils.h"
void exit_bootsplash(void) {}
void stg1_error_message(char *msg, ...)
@@ -33,11 +36,15 @@ void fatal_error(char *msg)
exit(EXIT_FAILURE);
}
-int main(int argc __attribute__ ((unused)), char **argv __attribute__ ((unused)), char **env)
+int main(int argc, char **argv, char **env)
{
+ enum media_bus bus = BUS_ANY;
+ if (argc > 1 && streq(argv[1], "usb")) {
+ bus = BUS_USB;
+ }
open_log();
init_modules_insmoding();
- find_media(BUS_ANY);
+ find_media(bus);
close_log();
return 0;