[ViewVC] Annotation of: cvs/Urlader/bin/urlader-util

#!/opt/bin/perl

use common::sense;
use Getopt::Long;
use Digest::MD5;
use Urlader;

my $verbose = 1;

my $URLADER;
my $OUTPUT;
my $WINDOWS_ICON;
my $MODE = "pack";

Getopt::Long::Configure ("bundling", "no_ignore_case");
GetOptions(
   "verbose|v" => sub {
      $verbose++;
   },
   "quiet|q" => sub {
      $verbose = 0;
   },
   "urlader=s"      => \$URLADER,
   "output|o=s"     => \$OUTPUT,
   "windows-icon=s" => \$WINDOWS_ICON,
   "pack"           => sub { $MODE = "pack" },
   "help|h"         => sub {
      require Pod::Usage;
      Pod::Usage::pod2usage (-verbose => 1, -exitval => 0, -noperldoc => 1);
   },
) or exit 1;

#############################################################################

my $out = *STDOUT;

if (length $OUTPUT) {
   undef $out;
   open $out, ">:perlio", $OUTPUT
      or die "$OUTPUT: $!";
}

#############################################################################
# for now, $MODE is --pack

@ARGV >= 4
   or die "not enough arguments for --pack.\n";

#############################################################################

my $urlader_md5 = "\x00" x 16;

if (length $URLADER) {
   open my $fh, "<:perlio", $URLADER
      or die "$URLADER: $!";
   my $urlader = do { local $/; <$fh> };

   if (length $WINDOWS_ICON) {
      require Win32::Exe;
      if (my $exe = eval { Win32::Exe->new (\$urlader) }) {
         $exe->set_icons ([Win32::Exe::IconFile->new ($WINDOWS_ICON)->icons]);
         $exe->write (\$urlader);
      } else {
         print "unable to set icon ($@)\n" if $verbose >= 2;
      }
   }

   $urlader_md5 = Digest::MD5::md5 $urlader;

   syswrite $out, $urlader;
}

#############################################################################

my $exe_id  = shift;
my $exe_ver = shift;
my $dir     = shift;

my $size;
my $max_uncomp;
my $datasize;

my $md5 = new Digest::MD5;

sub wr {
   syswrite $out, $_[0];
   $size += length $_[0];
   $md5->add ($_[0]);
}

sub ent {
   my ($type, $flags, $name, $data) = @_;

   warn "add record type $type, path $name\n" if $verbose >= 2;

   my $ent = pack "CCnN Z* a*",
      $type,
      $flags,
      (length $name),
      (length $data),
      $name,
      $data;

   wr $ent;
}

sub scandir {
   my ($path, $pfx) = @_;

   $pfx =~ s%^/%%;

   opendir my $fh, $path
      or die "$path: $!";

   for my $file (sort readdir $fh) {
      next if $file eq "." || $file eq "..";

      lstat "$path/$file"
         or die "$path/$file: $!";

      if (-d _) {
         ent Urlader::T_DIR, 0, "$pfx$file";
         &scandir ("$path/$file", "$pfx$file/");
      } elsif (-f _) {
         my $len = -s _;

         $datasize += $len;
         $max_uncomp = $len if $max_uncomp < $len;

         open my $fh2, "<", "$path/$file"
            or die "$path/$file: $!";

         $len == sysread $fh2, my $data, $len
            or die "$path/$file: read error";

         my $flags = (-x _) ? Urlader::F_EXEC : 0;

         if (Urlader::lzf_compress $data) {
            $flags |= Urlader::F_LZF;
         }

         ent Urlader::T_FILE, $flags, "$pfx$file", $data;
      } else {
         warn "$path/$file: unsupported filetype, skipping.\n";
      }
   }
}

ent Urlader::T_META, 0, $exe_id, "$exe_ver\x00";

shift, ent Urlader::T_ENV, 0, $1, "$2\x00"
   while $ARGV[0] =~ /^([^=]+)=(.*)$/s;

ent Urlader::T_ARG, 0, shift
   while @ARGV;

scandir $dir, "";

ent Urlader::T_NULL, 0;

wr pack "NN x8 a16",
   $max_uncomp,
   $size + 4 + 4 + 8 + 16 + 16 + 16, # sizeof this pack + md5s, ugly
   Urlader::TAIL_MAGIC;

wr $urlader_md5;
wr $md5->digest;

printf STDERR "%d bytes written (file data before compression: %d, biggest file %d bytes)\n", $size, $datasize, $max_uncomp
   if $verbose;

exit 0;

=head1 NAME

   urlader-util - generate new urladers, maybe other stuff

=head1 SYNOPSIS

   urlader-util --urlader myurlader.exe --windows-icon myicon.ico \
                --pack exe_id exe_ver directory \
                LD_LIBRARY_PATH=. ./perl run >myprog.exe

=head1 DESCRIPTION

Urlader utility program - see the L<Urlader> module manpage for more info.

=head1 OPTIONS

=head2 SWITCHES

=over 4

=item --help

Display this manual page.

=item -v, --verbose

Increase verbosity (the default verbosity level is 1).

=item -q, --quiet

Set verbosity to zero, that is, only errors and warnings will be printed.

=item -o, --output <path>

Write output (e.g. the generated executable) to the given path instead of
to standard output.

=item --urlader <path>

Use the given file as urlader. What this means depends on the execution mode.

=item --windows-icon <path>

If the urlader executable is a windows executable, patch it with this
icon resource (which must be in .ICO format). For other executables, this
switch is ignored.

=back

=head2 MODES

=item --pack <exe_id> <exe_ver> <path> [env-vars...] <program> [arguments...]

Packs the given directory tree, env-variable assignments, program path
and arguments into an urlader archive and optionally prepend the urlader
binary (see C<--urlader>) and output the whole blob.

See the L<Urlader> manual page for an explanation of C<exe_id>, C<exe_ver>
and other concepts.

=back

=head1 AUTHOR

 Marc Lehmann <schmorp@schmorp.de>
 http://home.schmorp.de/

=head1 SEE ALSO

L<Urlader>, L<PAR> - for more features and easier usage.

=cut
Revision:	1.3
Committed:	Wed Jan 4 15:08:27 2012 UTC (12 years, 4 months ago) by root
Branch:	MAIN
Changes since 1.2:	+1 -1 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	root	1.1	#!/opt/bin/perl
2
3			use common::sense;
4			use Getopt::Long;
5			use Digest::MD5;
6			use Urlader;
7
8			my $verbose = 1;
9
10			my $URLADER;
11	root	1.2	my $OUTPUT;
12	root	1.1	my $WINDOWS_ICON;
13			my $MODE = "pack";
14
15			Getopt::Long::Configure ("bundling", "no_ignore_case");
16			GetOptions(
17			"verbose\|v" => sub {
18			$verbose++;
19			},
20			"quiet\|q" => sub {
21			$verbose = 0;
22			},
23			"urlader=s" => \$URLADER,
24			"output\|o=s" => \$OUTPUT,
25			"windows-icon=s" => \$WINDOWS_ICON,
26			"pack" => sub { $MODE = "pack" },
27			"help\|h" => sub {
28			require Pod::Usage;
29			Pod::Usage::pod2usage (-verbose => 1, -exitval => 0, -noperldoc => 1);
30			},
31			) or exit 1;
32
33			#############################################################################
34
35			my $out = *STDOUT;
36
37			if (length $OUTPUT) {
38			undef $out;
39			open $out, ">:perlio", $OUTPUT
40			or die "$OUTPUT: $!";
41			}
42
43			#############################################################################
44			# for now, $MODE is --pack
45
46			@ARGV >= 4
47			or die "not enough arguments for --pack.\n";
48
49			#############################################################################
50
51			my $urlader_md5 = "\x00" x 16;
52
53			if (length $URLADER) {
54			open my $fh, "<:perlio", $URLADER
55			or die "$URLADER: $!";
56			my $urlader = do { local $/; <$fh> };
57
58			if (length $WINDOWS_ICON) {
59			require Win32::Exe;
60			if (my $exe = eval { Win32::Exe->new (\$urlader) }) {
61	root	1.3	$exe->set_icons ([Win32::Exe::IconFile->new ($WINDOWS_ICON)->icons]);
62	root	1.1	$exe->write (\$urlader);
63			} else {
64			print "unable to set icon ($@)\n" if $verbose >= 2;
65			}
66			}
67
68			$urlader_md5 = Digest::MD5::md5 $urlader;
69
70			syswrite $out, $urlader;
71			}
72
73			#############################################################################
74
75			my $exe_id = shift;
76			my $exe_ver = shift;
77			my $dir = shift;
78
79			my $size;
80			my $max_uncomp;
81			my $datasize;
82
83			my $md5 = new Digest::MD5;
84
85			sub wr {
86			syswrite $out, $_[0];
87			$size += length $_[0];
88			$md5->add ($_[0]);
89			}
90
91			sub ent {
92			my ($type, $flags, $name, $data) = @_;
93
94			warn "add record type $type, path $name\n" if $verbose >= 2;
95
96			my $ent = pack "CCnN Z* a*",
97			$type,
98			$flags,
99			(length $name),
100			(length $data),
101			$name,
102			$data;
103
104			wr $ent;
105			}
106
107			sub scandir {
108			my ($path, $pfx) = @_;
109
110			$pfx =~ s%^/%%;
111
112			opendir my $fh, $path
113			or die "$path: $!";
114
115			for my $file (sort readdir $fh) {
116			next if $file eq "." \|\| $file eq "..";
117
118			lstat "$path/$file"
119			or die "$path/$file: $!";
120
121			if (-d _) {
122			ent Urlader::T_DIR, 0, "$pfx$file";
123			&scandir ("$path/$file", "$pfx$file/");
124			} elsif (-f _) {
125			my $len = -s _;
126
127			$datasize += $len;
128			$max_uncomp = $len if $max_uncomp < $len;
129
130			open my $fh2, "<", "$path/$file"
131			or die "$path/$file: $!";
132
133			$len == sysread $fh2, my $data, $len
134			or die "$path/$file: read error";
135
136			my $flags = (-x _) ? Urlader::F_EXEC : 0;
137
138			if (Urlader::lzf_compress $data) {
139			$flags \|= Urlader::F_LZF;
140			}
141
142			ent Urlader::T_FILE, $flags, "$pfx$file", $data;
143			} else {
144			warn "$path/$file: unsupported filetype, skipping.\n";
145			}
146			}
147			}
148
149			ent Urlader::T_META, 0, $exe_id, "$exe_ver\x00";
150
151			shift, ent Urlader::T_ENV, 0, $1, "$2\x00"
152			while $ARGV[0] =~ /^([^=]+)=(.*)$/s;
153
154			ent Urlader::T_ARG, 0, shift
155			while @ARGV;
156
157			scandir $dir, "";
158
159			ent Urlader::T_NULL, 0;
160
161			wr pack "NN x8 a16",
162			$max_uncomp,
163			$size + 4 + 4 + 8 + 16 + 16 + 16, # sizeof this pack + md5s, ugly
164			Urlader::TAIL_MAGIC;
165
166			wr $urlader_md5;
167			wr $md5->digest;
168
169			printf STDERR "%d bytes written (file data before compression: %d, biggest file %d bytes)\n", $size, $datasize, $max_uncomp
170			if $verbose;
171
172			exit 0;
173
174			=head1 NAME
175
176			urlader-util - generate new urladers, maybe other stuff
177
178			=head1 SYNOPSIS
179
180			urlader-util --urlader myurlader.exe --windows-icon myicon.ico \
181			--pack exe_id exe_ver directory \
182			LD_LIBRARY_PATH=. ./perl run >myprog.exe
183
184			=head1 DESCRIPTION
185
186			Urlader utility program - see the L<Urlader> module manpage for more info.
187
188			=head1 OPTIONS
189
190			=head2 SWITCHES
191
192			=over 4
193
194			=item --help
195
196			Display this manual page.
197
198			=item -v, --verbose
199
200			Increase verbosity (the default verbosity level is 1).
201
202			=item -q, --quiet
203
204			Set verbosity to zero, that is, only errors and warnings will be printed.
205
206			=item -o, --output <path>
207
208			Write output (e.g. the generated executable) to the given path instead of
209			to standard output.
210
211			=item --urlader <path>
212
213			Use the given file as urlader. What this means depends on the execution mode.
214
215			=item --windows-icon <path>
216
217			If the urlader executable is a windows executable, patch it with this
218			icon resource (which must be in .ICO format). For other executables, this
219			switch is ignored.
220
221			=back
222
223			=head2 MODES
224
225			=item --pack <exe_id> <exe_ver> <path> [env-vars...] <program> [arguments...]
226
227			Packs the given directory tree, env-variable assignments, program path
228			and arguments into an urlader archive and optionally prepend the urlader
229			binary (see C<--urlader>) and output the whole blob.
230
231			See the L<Urlader> manual page for an explanation of C<exe_id>, C<exe_ver>
232			and other concepts.
233
234			=back
235
236			=head1 AUTHOR
237
238			Marc Lehmann <schmorp@schmorp.de>
239			http://home.schmorp.de/
240
241			=head1 SEE ALSO
242
243			L<Urlader>, L<PAR> - for more features and easier usage.
244
245			=cut