#!/usr/bin/perl
use strict;
use Data::Dumper;
use Getopt::Std;
use IO::File;
our %opts = (d => 'size-tests');
# default arguments go here
my @sets =
({
"with-defaults" => '',
});
my %base_args = %{$sets[0]};
my ($basename, $basenum);
getopts('t:hlcD:nj:o:eb:T', \%opts) || usage();
$opts{'j'} = "-j$opts{j}" if (exists($opts{'j'}));
usage() if ($opts{'h'});
my @argumentsets =
(
{
title => 'security-types',
arguments =>
{
usm => {
'with-security-modules' => 'usm',
},
dtlsandtls => {
'with-security-modules' => 'tsm',
'with-out-security-modules' => 'usm',
'with-transports' => 'DTLSUDP TLSTCP',
'with-mib-modules' => 'tlstm-mib tsm-mib'
},
usmanddtlsandtls => {
'with-security-modules' => 'tsm usm',
'with-transports' => 'DTLSUDP TLSTCP',
'with-mib-modules' => 'tlstm-mib tsm-mib'
}
}
},
{
title => 'minimalist',
arguments =>
{
minimal => {
'enable-minimalist' => '',
},
'not-minimal' => {
}
}
},
{
title => 'mib-loading',
arguments =>
{
'without-mib-loading' => {
'disable-mib-loading' => '',
},
'with-mib-loading' => {
}
}
},
{
title => 'debugging',
arguments =>
{
'without-debugging' => {
'disable-debugging' => '',
},
'with-debugging' => {
}
}
},
{
title => 'logging',
arguments =>
{
'without-logging' => {
'with-out-features' => 'logging',
},
'with-logging' => {
}
}
},
{
title => 'agent-mibs',
arguments =>
{
'full-agent' => {
},
'mini-agent' => {
'enable-mini-agent' => '',
}
}
},
{
title => 'protocol-support',
arguments =>
{
'everything' => {
},
'read-only' => {
'enable-read-only' => '',
},
'notify-only' => {
'enable-notify-only' => '',
}
}
},
{
title => 'perl',
arguments =>
{
'without-perl-modules' => {
'without-perl-modules' => '',
'disable-embedded-perl' => '',
},
'with-perl-no-embedding' => {
'with-perl-modules' => '',
'disable-embedded-perl' => '',
},
'with-perl-and-embedding' => {
'with-perl-modules' => '',
'enable-embedded-perl' => '',
}
}
},
);
# map these to a referencable lookup hash
my %argumentsets;
foreach my $set (@argumentsets) {
$argumentsets{$set->{'title'}} = $set;
}
if ($opts{'l'}) {
print "Types available:\n";
printf(" %-40.40s %s\n", "Test Title", "Number of subtests");
printf(" %-40.40s %s\n", "-" x 39, "-" x length("Number of subtests"));
foreach my $type (@argumentsets) {
my @keys = keys(%{$type->{'arguments'}});
printf(" %-40.40s %s\n", $type->{'title'}, $#keys+1);
}
exit;
}
my %types;
if ($opts{'t'}) {
my @types = split(/,\s*/, $opts{'t'});
foreach my $type (@types) {
$types{$type} = 1;
}
}
if ($opts{'b'}) {
# use this set as the base default set of arguments
($basename, $basenum) = ($opts{'b'} =~ /(.*):(\d+)/);
if (!exists($argumentsets{$basename})) {
printf STDERR "failed to find set for -b argument: %s\n", $basename;
exit(1);
}
@sets = add_in_flags($argumentsets{$basename}, \%base_args, ());
@sets = @sets[$basenum];
%base_args = %{$sets[0]};
}
foreach my $set (@argumentsets) {
if (!$opts{'t'} || exists($types{$set->{'title'}})) {
if ($basename && $set->{'title'} eq $basename) {
next;
}
@sets = add_in_flags($set, \%base_args, @sets);
}
}
if ($opts{'c'}) {
# print the configure flags
foreach my $set (@sets) {
print "$set->{'title'}:\n";
print " ", generate_configure_flags($set), "\n";
}
exit;
}
my $summaryfile;
if ($opts{'o'} && !$opts{'n'}) {
$summaryfile = new IO::File;
$summaryfile->open(">$opts{o}");
}
my $count = 0;
foreach my $set (@sets) {
$count++;
build_set($count, $set);
}
sub add_in_flags {
my ($argumentset, $base_flags, @current_flags) = @_;
my @new_flags;
# do a linear search of the options
if (! $opts{'e'}) {
@new_flags = @current_flags;
foreach my $newargs (keys(%{$argumentset->{'arguments'}})) {
my %flags = %$base_flags;
$flags{'title'} = "$flags{'title'}:$newargs";
foreach my $newflag (keys(%{$argumentset->{'arguments'}{$newargs}})) {
$flags{$newflag} .= " $argumentset->{'arguments'}{$newargs}{$newflag}";
}
push @new_flags, \%flags;
}
return @new_flags;
}
# or an exponential search
foreach my $flags (@current_flags) {
foreach my $newargs (keys(%{$argumentset->{'arguments'}})) {
my %flags = %{$flags}; # copy the existing hash-set of flags
if (exists($flags{'title'})) {
$flags{'title'} .= ", $newargs";
} else {
$flags{'title'} .= "$newargs";
}
foreach my $newflag (keys(%{$argumentset->{'arguments'}{$newargs}})) {
$flags{$newflag} .= " $argumentset->{'arguments'}{$newargs}{$newflag}";
}
push @new_flags, \%flags;
}
}
return @new_flags;
}
sub generate_configure_flags {
my ($arguments) = @_;
my $line = "";
foreach my $arg (keys(%$arguments)) {
next if ($arg eq 'title');
if ($arguments->{$arg} =~ /^\s*$/) {
$line .= " --$arg";
} else {
$line .= " --$arg=\"$arguments->{$arg}\"";
}
}
return $line;
}
sub build_set {
my ($num, $arguments) = @_;
$num = sprintf("%03d", $num);
my $file;
if (!$opts{'n'}) {
mkdir $opts{'d'} if (! -d $opts{'d'});
die "failed to create the $opts{'d'} directory" if (! -d $opts{'d'});
$file = new IO::File;
$file->open(">$opts{'d'}/$num-0-cmd-summary");
$file->print("Creating output for: $arguments->{'title'}\n");
}
print "==================== $arguments->{'title'}\n";
System ($file, $num, "1-distclean", "make distclean");
System ($file, $num, "2-configure",
"./configure " . generate_configure_flags($arguments));
System ($file, $num, "3-features", "make features");
System ($file, $num, "4-make", "make $opts{'j'}");
System ($file, $num, "5-unused-code", "perl local/minimalist/find-unused-code -g");
System ($file, $num, "6-testing", "make -C testing test") if ($opts{'T'});
if (!$opts{'n'}) {
analyze_size($arguments->{'title'}, "$opts{'d'}/$num-9-results");
}
}
sub analyze_size {
my ($title, $filename) = @_;
my $outfile = new IO::File;
$outfile->open(">$filename");
print "Results for: $title\n";
printf $outfile "Results for: $title\n";
printf ("%-50.50s %10s\n", "-" x 49, "-" x 10);
printf $outfile ("%-50.50s %10s\n", "-" x 49, "-" x 10);
my $totalsize = 0;
foreach my $buildfile (glob("snmplib/.libs/lib*.so.*.0.0"),
glob("agent/.libs/lib*.so.*.0.0"),
"agent/.libs/snmpd") {
my @info = stat($buildfile);
printf $outfile ("%-50.50s %10s\n", $buildfile, $info[7]);
printf("%-50.50s %10s\n", $buildfile, $info[7]);
$totalsize += $info[7];
}
printf $outfile ("%-50.50s %10s\n", "-" x 49, "-" x 10);
printf $outfile ("%-50.50s %10s bytes\n", "TOTAL", $totalsize);
printf("%-50.50s %10s\n", "-" x 49, "-" x 10);
printf("%-50.50s %10s bytes\n", "TOTAL", $totalsize);
if (defined($summaryfile)) {
my $restricted_title = $title;
$restricted_title =~ s/[^a-zA-Z]/_/g;
printf $summaryfile "%10s %s \n", $totalsize, $title;
}
return $totalsize;
}
sub usage {
print "Usage: $0 [FLAGS]\n\n";
print "FLAGS:\n";
print " -h\t\thelp\n";
print " -t TYPES\tSelect types to analyze (default = all)\n";
print " -l\t\tList available types\n";
print " -c\t\tPrint the configuration flags that would be used\n";
print " -D DIR\tSave data results to this directory\n";
print " -o FILE\tSave a complete summary chart to FILE\n";
print " -n\t\tDry run -- only show the commands to execute\n";
print " -j NUM\tExecute make with parallel building (-jNUM)\n";
print " -e\t\tUse exponential expansion (all combos) of the requested options\n";
print " -b NAME:NUM\tUse NAME and the NUM'th set as the base arguments to start from\n";
print " -T\t\tRun 'make test' at each step as well\n";
exit;
}
sub System {
my $file = shift;
my $num = shift;
my $title = shift;
my $pipe = " 2>&1 | tee $opts{'d'}/$num-$title.out\n";
print "### ", join(" ", @_), $pipe;
if (!$opts{'n'} && $file) {
print $file "### ", join(" ", @_), "\n";
}
if (!$opts{'n'}) {
system(join(" ", @_) . $pipe);
}
}