fix a race condition with dynamic interfaces and firewalling

[openwrt.git] / scripts / download.pl

diff --git a/scripts/download.pl b/scripts/download.pl
index a3a5f09..543dcc1 100755 (executable)
--- a/scripts/download.pl
+++ b/scripts/download.pl
@@ -1,22 +1,71 @@
 #!/usr/bin/perl
+# 
+# Copyright (C) 2006 OpenWrt.org
+#
+# This is free software, licensed under the GNU General Public License v2.
+# See /LICENSE for more information.
+#
+
 use strict;
 use warnings;
+use File::Basename;
 
 my $target = shift @ARGV;
 my $filename = shift @ARGV;
 my $md5sum = shift @ARGV;
-my @mirrors = @ARGV;
+my $scriptdir = dirname($0);
+my @mirrors;
 
 my $ok;
 
-@mirrors > 0 or die "Syntax: $0 <target dir> <filename> <md5sum> <mirror> [<mirror> ...]\n";
+@ARGV > 0 or die "Syntax: $0 <target dir> <filename> <md5sum> <mirror> [<mirror> ...]\n";
+
+sub localmirrors {
+    my @mlist;
+    open LM, "$scriptdir/localmirrors" and do {
+           while (<LM>) {
+                       chomp $_;
+                       push @mlist, $_;
+               }
+               close LM;
+       };
+       open CONFIG, "<".$ENV{'TOPDIR'}."/.config" and do {
+               while (<CONFIG>) {
+                       /^CONFIG_LOCALMIRROR="(.+)"/ and do {
+                               chomp;
+                               push @mlist, $1;
+                       };
+               }
+               close CONFIG;
+       };
+       
+
+    return @mlist;
+}
+
+sub which($) {
+       my $prog = shift;
+       my $res = `which $prog`;
+       $res or return undef;
+       $res =~ /^no / and return undef;
+       $res =~ /not found/ and return undef;
+       return $res;
+}
+
+my $md5cmd = which("md5sum");
+$md5cmd or $md5cmd = which("md5");
+$md5cmd or die 'no md5 checksum program found, please install md5 or md5sum';
+chomp $md5cmd;
 
 sub download
 {
        my $mirror = shift;
+       my $options = $ENV{WGET_OPTIONS};
+       $options or $options = "";
        
-       open WGET, "wget -t1 --connect-timeout=20 --read-timeout=15 -O- \"$mirror/$filename\" |" or die "Cannot launch wget.\n";
-       open MD5SUM, "| md5sum > \"$target/$filename.md5sum\"" or die "Cannot launch md5sum.\n";
+       $mirror =~ s/\/$//;
+       open WGET, "wget -t1 --timeout=20 $options -O- \"$mirror/$filename\" |" or die "Cannot launch wget.\n";
+       open MD5SUM, "| $md5cmd > \"$target/$filename.md5sum\"" or die "Cannot launch md5sum.\n";
        open OUTPUT, "> $target/$filename.dl" or die "Cannot create file $target/$filename.dl: $!\n";
        my $buffer;
        while (read WGET, $buffer, 1048576) {
@@ -34,7 +83,7 @@ sub download
        }
        
        my $sum = `cat "$target/$filename.md5sum"`;
-       $sum =~ /^(\w+)\s+/ or die "Could not generate md5sum\n";
+       $sum =~ /^(\w+)\s*/ or die "Could not generate md5sum\n";
        $sum = $1;
        
        if (($md5sum =~ /\w{32}/) and ($sum ne $md5sum)) {
@@ -45,6 +94,7 @@ sub download
        
        unlink "$target/$filename";
        system("mv \"$target/$filename.dl\" \"$target/$filename\"");
+       cleanup();
 }
 
 sub cleanup
@@ -53,22 +103,38 @@ sub cleanup
        unlink "$target/$filename.md5sum";
 }
 
-while (!$ok) {
-       my $mirror = shift @mirrors;
-       $mirror or die "No more mirrors to try - giving up.\n";
-       
+@mirrors = localmirrors();
+
+foreach my $mirror (@ARGV) {
        if ($mirror =~ /^\@SF\/(.+)$/) {
-               my $sfpath = $1;
-               open SF, "wget -t1 -q -O- 'http://prdownloads.sf.net/$sfpath/$filename' |";
-               while (<SF>) {
-                       /RADIO NAME=use_default VALUE=(\w+) OnClick="form\.submit\(\)">/ and do {
-                               push @mirrors, "http://$1.dl.sourceforge.net/sourceforge/$sfpath";
-                       };
+               # give sourceforge a few more tries, because it redirects to different mirrors
+               for (1 .. 5) {
+                       push @mirrors, "http://downloads.sourceforge.net/$1";
                }
-               close SF;
+       } elsif ($mirror =~ /^\@GNU\/(.+)$/) {
+               my $gnupath = $1;
+               push @mirrors, "ftp://ftp.gnu.org/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.belnet.be/mirror/ftp.gnu.org/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.mirror.nl/pub/mirror/gnu/$gnupath";
+               push @mirrors, "http://mirror.switch.ch/ftp/mirror/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.uu.net/archive/systems/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.eu.uu.net/pub/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.leo.org/pub/comp/os/unix/gnu/$gnupath";
+               push @mirrors, "ftp://ftp.digex.net/pub/gnu/$gnupath";
        } else {
-               download($mirror);
+               push @mirrors, $mirror;
        }
+}
+
+#push @mirrors, 'http://mirror1.openwrt.org';
+push @mirrors, 'http://mirror2.openwrt.org/sources';
+push @mirrors, 'http://downloads.openwrt.org/sources';
+
+while (!$ok) {
+       my $mirror = shift @mirrors;
+       $mirror or die "No more mirrors to try - giving up.\n";
+       
+       download($mirror);
        -f "$target/$filename" and $ok = 1;
 }