fix bug (update vmlinuz and initrd files)

1 files changed, 20 insertions, 7 deletions

diff --git a/drakpxelinux.pl b/drakpxelinux.pl
index 75b523e..b87ebc1 100644
--- a/drakpxelinux.pl
+++ b/drakpxelinux.pl
@@ -40,6 +40,15 @@ use ugtk2 qw(:ask :helpers :wrappers :create :dialogs);
 use constant FALSE => 0;
 use constant TRUE => 1;
 
+
+sub AmIRoot() {
+  unless ($> == 0) {
+  err_dialog(N("Error!"), N("You are not root. Exiting...")) and !$::testing && die;
+  }
+}
+
+AmIRoot();
+
 # ie of entry menu in PXE:
 #label linux
 #        KERNEL images/vmlinuz
@@ -510,7 +519,7 @@ sub edit_box_item {
 
     # create file dialog widget, with file or directory selection
     my $fdwidget = sub {
-      my ($data, $test, $filetotest) = @_;
+      my ($data, $test, $filetotest, $label) = @_;
       chdir($X86);
       my $fd = new Gtk2::FileSelection(N("Selection"));
       $fd->set_modal(TRUE);
@@ -525,13 +534,17 @@ sub edit_box_item {
 				       }
 				       if ($filetotest eq "kernel") {
 					 run_program::get_stdout("file $file") =~ /boot sector/ or err_dialog(N("Error!"), N("Should be a boot sector file")) and return;
+					 run_program::get_stdout("cp -avf $file $IMGPATH/vmlinuz-$label");
+					 $data->set_text("images/vmlinuz-$label");
 				       } elsif ($filetotest eq "initrd") {
 					 run_program::get_stdout("file $file") =~ /initrd/ or err_dialog(N("Error!"), N("Should be an initrd file")) and return;
+					 system("cp -avf $file $IMGPATH/$label.rdz");
+					 $data->set_text("images/$label.rdz");
 				       }
-				       my $d = dirname($file);
-				       if ($d =~ /images/) { $filesel = "images/" . $filesel }
-				       print "$filesel\n";
-				       $data->set_text($filesel);
+#				       my $d = dirname($file);
+#				       if ($d =~ /images/) { $filesel = "images/" . $filesel }
+#				       print "$filesel\n";
+#				       $data->set_text($filesel);
 				       $fd->hide;
 				     }, $fd);
       $fd->cancel_button->signal_connect(clicked => sub { $fd->hide });
@@ -543,7 +556,7 @@ sub edit_box_item {
     $kernel->set_text($listpxe[$i]{kernel});
     set_help_tip($kernel, 'kernel');
 
-    my $file_dialogk = $fdwidget->($kernel, "", "kernel");
+    my $file_dialogk = $fdwidget->($kernel, "", "kernel", $listpxe[$i]{label});
     # button kernel
     my $buttonkernel = Gtk2::Button->new(N("Select kernel to boot"));
     $buttonkernel->signal_connect(clicked => sub { $file_dialogk->show });
@@ -552,7 +565,7 @@ sub edit_box_item {
     $initrd->set_text($listpxe[$i]{initrd});
     set_help_tip($initrd, 'initrd');
 
-    my $file_dialog = $fdwidget->($initrd, "", "initrd");
+    my $file_dialog = $fdwidget->($initrd, "", "initrd", $listpxe[$i]{label});
     my $buttoninitrd = Gtk2::Button->new(N("Select associated initrd"));
     $buttoninitrd->signal_connect(clicked => sub { $file_dialog->show });