pass media to grub functions

1 files changed, 3 insertions, 3 deletions

diff --git a/draklive b/draklive
index 86f4fe0..433595c 100755
--- a/draklive
+++ b/draklive
@@ -747,11 +747,11 @@ sub build_grub_cfg {
     #- FIXME? first partition is hardcoded for loopback (master images)
     my ($part_nb) = $device =~ m!/dev/loop! ? 1 : $device =~ /(\d+)$/;
     my $initrd = get_initrd_path($media);
-    build_grub_cfg_raw($live, $initrd, $opts, $part_nb);
+    build_grub_cfg_raw($live, $media, $initrd, $opts, $part_nb);
 }
 
 sub build_grub_cfg_raw {
-    my ($live, $initrd, $opts, $part_nb) = @_;
+    my ($live, $media, $initrd, $opts, $part_nb) = @_;
     #- FIXME: use the bootloader module from drakx
     my $grub_part = "(hd0" . (defined $part_nb ? "," . ($part_nb-1) : "") . ")";
     my $boot = get_media_prefix($live, 'boot'); #- FIXME handle boot media
@@ -1199,7 +1199,7 @@ $live->{media}{title} <kernel options>
 sub create_classical_bootloader {
     my ($live) = @_;
     my $part_nb = 1; #- FIXME: get from partition layout config
-    output_p(get_system_root($live) . '/boot/grub/menu.lst', build_grub_cfg_raw($live, "/initrd.img", {}, $part_nb));
+    output_p(get_system_root($live) . '/boot/grub/menu.lst', build_grub_cfg_raw($live, $live->{media}, "/initrd.img", {}, $part_nb));
 }
 
 sub device_allocate_file {