1#!/usr/local/bin/perl
2# download.cgi
3# Save a download or start it right now
4
5require './updown-lib.pl';
6use Time::Local;
7&ReadParse();
8&error_setup($text{'download_err'});
9
10# Validate and store inputs
11$i = 0;
12@urls = split(/[\r\n]+/, $in{'urls'});
13@urls || &error($text{'download_enone'});
14foreach $u (@urls) {
15	local ($proto, $host, $port, $page, $ssl);
16	if ($u =~ /^http/) {
17		($host, $port, $page, $ssl, $user, $pass) = &parse_http_url($u);
18		$proto = $ssl ? "https" : "http";
19		}
20	elsif ($u =~ /^ftp:\/\/([^\/]+)(:21)?(\/.*)$/) {
21		$proto = "ftp";
22		$host = $1; $page = $3;
23		}
24	else {
25		&error(&text('download_eurl', $u));
26		}
27	if ($host =~ /^([^:\@]+):([^:\@]+)\@(\S+)/) {
28		$user = $1;
29		$pass = $2;
30		$host = $3;
31		}
32	$download{"url_$i"} = $u;
33	$download{"proto_$i"} = $proto;
34	$download{"host_$i"} = $host;
35	$download{"user_$i"} = $user;
36	$download{"pass_$i"} = $pass;
37	$download{"port_$i"} = $port;
38	$download{"page_$i"} = $page;
39	$download{"ssl_$i"} = $ssl;
40	$i++;
41	}
42$in{'dir'} || &error($text{'upload_edir'});
43&can_write_file($in{'dir'}) ||
44	&error(&text('download_eaccess', "<tt>$in{'dir'}</tt>", $!));
45$download{'dir'} = $in{'dir'};
46if ($can_mode != 3) {
47	# User can be entered
48	scalar(@uinfo = getpwnam($in{'user'})) || &error($text{'upload_euser'});
49	&can_as_user($in{'user'}) ||
50		&error(&text('download_eucannot', $in{'user'}));
51	$download{'uid'} = $uinfo[2];
52	$in{'group_def'} || scalar(@ginfo = getgrnam($in{'group'})) ||
53		&error($text{'upload_egroup'});
54	$can_mode == 0 || $in{'group_def'} || &in_group(\@uinfo, \@ginfo) ||
55		&error($text{'download_egcannot'});
56	$download{'gid'} = scalar(@ginfo) ? $ginfo[2] : $uinfo[3];
57	}
58else {
59	# User is fixed
60	if (&supports_users()) {
61		@uinfo = getpwnam($remote_user);
62		$download{'uid'} = $uinfo[2];
63		$download{'gid'} = $uinfo[3];
64		}
65	}
66if ($in{'bg'} && $can_schedule) {
67	# Validate time
68	$in{'hour'} =~ /^\d+$/ && $in{'min'} =~ /^\d+$/ &&
69		$in{'day'} =~ /^\d+$/ && $in{'year'} =~ /^\d+$/ ||
70			&error($text{'download_edate'});
71	eval { $download{'time'} = timelocal(0, $in{'min'}, $in{'hour'},
72			 $in{'day'}, $in{'month'}, $in{'year'}-1900) };
73	$@ && &error($text{'download_edate2'});
74	}
75if (defined($in{'email_def'}) && !$in{'email_def'}) {
76	# Validate email
77	$in{'email'} =~ /\S/ || &error($text{'upload_eemail'});
78	$download{'email'} = $in{'email'};
79	}
80
81# Create the directory if needed
82if (!-d $download{'dir'} && $in{'mkdir'}) {
83	&switch_uid_to($download{'uid'}, $download{'gid'});
84	mkdir($download{'dir'}, 0755) || &error(&text('upload_emkdir', $!));
85	&switch_uid_back();
86	}
87
88# Save the settings
89if ($module_info{'usermin'}) {
90	&lock_file("$user_module_config_directory/config");
91	$userconfig{'ddir'} = $in{'dir'};
92	&write_file("$user_module_config_directory/config", \%userconfig);
93	&unlock_file("$user_module_config_directory/config");
94	}
95else {
96	&lock_file("$module_config_directory/config");
97	$config{'ddir_'.$remote_user} = $in{'dir'};
98	$config{'duser_'.$remote_user} = $in{'user'};
99	$config{'dgroup_'.$remote_user} = $in{'group_def'} ? undef
100							   : $in{'group'};
101	&write_file("$module_config_directory/config", \%config);
102	&unlock_file("$module_config_directory/config");
103	}
104
105if ($in{'bg'} && $can_background) {
106	# Create a script to be called by At
107	&foreign_require("cron", "cron-lib.pl");
108	&lock_file($atjob_cmd);
109	&cron::create_wrapper($atjob_cmd, $module_name, "download.pl");
110	&unlock_file($atjob_cmd);
111	&save_download(\%download);
112
113	if (!$can_schedule) {
114		# Just run this script right now
115		&execute_command("$atjob_cmd $download{'id'} &");
116		}
117	else {
118		# Create an At job to do the download
119		&foreign_require("at", "at-lib.pl");
120			{
121			local %ENV;
122			delete($ENV{'FOREIGN_MODULE_NAME'});
123			delete($ENV{'FOREIGN_ROOT_DIRECTORY'});
124			&clean_environment();
125			$ENV{'REMOTE_USER'} = $remote_user;	# For usermin
126			$ENV{'BASE_REMOTE_USER'} = $base_remote_user;
127			&at::create_atjob(
128				$module_info{'usermin'} ? $remote_user : "root",
129				$download{'time'},
130				"$atjob_cmd $download{'id'}",
131				"/");
132			&reset_environment();
133			}
134		}
135
136	&redirect("index.cgi?mode=download");
137	}
138else {
139	# Download it now, and show the results
140	&ui_print_unbuffered_header(undef, $text{'download_title'}, "");
141
142	$error = &do_download(\%download, \&progress_callback, \@paths);
143	@paths = grep { $_ } @paths;
144	if (@paths) {
145		print "<p>$text{'download_done'}<p>\n";
146		foreach $p (@paths) {
147			@st = stat($p);
148			print "<tt>$p</tt> ($st[7] bytes)<br>\n";
149			}
150		}
151	if ($error) {
152		print "<p><b>",&text('download_failed', $error),"</b><p>\n";
153		}
154
155	&ui_print_footer("index.cgi?mode=download", $text{'index_return'});
156	}
157&webmin_log("download", undef, undef, { 'urls' => \@urls,
158					'time' => $download{'time'} });
159
160