mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Support passing command line arguments to nix-shell #! scripts

Browse source
This commit is contained in:
Eelco Dolstra 2015-02-18 15:55:18 +01:00
parent bb10010582
commit dc7e8fae48
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
scripts/nix-build.in
View file

@ -30,6 +30,7 @@ my $myName = $runEnv ? "nix-shell" : "nix-build";
my $inShebang = 0; my $inShebang = 0;
my $script; my $script;
my @savedArgs;
my $tmpDir = mkTempDir($myName); my $tmpDir = mkTempDir($myName);
@ -43,13 +44,14 @@ $SIG{'INT'} = sub { exit 1 };
# Heuristic to see if we're invoked as a shebang script, namely, if we # Heuristic to see if we're invoked as a shebang script, namely, if we
# have a single argument, it's the name of an executable file, and it # have a single argument, it's the name of an executable file, and it
# starts with "#!". # starts with "#!".
if ($runEnv && scalar @ARGV == 1) { if ($runEnv && $ARGV[0] !~ /nix-shell/) {
$script = $ARGV[0]; $script = $ARGV[0];
if (-f $script && -x $script) { if (-f $script && -x $script) {
open SCRIPT, "<$script" or die "$0: cannot open ‘$script’: $!\n"; open SCRIPT, "<$script" or die "$0: cannot open ‘$script’: $!\n";
my $first = <SCRIPT>; my $first = <SCRIPT>;
if ($first =~ /^\#\!/) { if ($first =~ /^\#\!/) {
$inShebang = 1; $inShebang = 1;
@savedArgs = @ARGV; shift @savedArgs;
@ARGV = (); @ARGV = ();
while (<SCRIPT>) { while (<SCRIPT>) {
chomp; chomp;
@ -193,7 +195,8 @@ for (my $n = 0; $n < scalar @ARGV; $n++) {
# or (undocumented) argv[0] does not contain "perl". Exploit # or (undocumented) argv[0] does not contain "perl". Exploit
# the latter by doing "exec -a". # the latter by doing "exec -a".
my $execArgs = $interpreter =~ /perl/ ? "-a PERL" : ""; my $execArgs = $interpreter =~ /perl/ ? "-a PERL" : "";
$envCommand = "exec $execArgs $interpreter $script"; # FIXME: escape arguments.
$envCommand = "exec $execArgs $interpreter $script @savedArgs";
} }
elsif (substr($arg, 0, 1) eq "-") { elsif (substr($arg, 0, 1) eq "-") {