package common; use diagnostics; use strict; use vars qw(@ISA %EXPORT_TAGS @EXPORT_OK $printable_chars $sizeof_int $bitof_int $cancel $SECTORSIZE); @ISA = qw(Exporter); %EXPORT_TAGS = ( common => [ qw(__ even odd min max sqr sum and_ or_ sign product bool invbool listlength bool2text text2bool to_int to_float ikeys member divide is_empty_array_ref is_empty_hash_ref add2hash add2hash_ set_new set_add round round_up round_down first second top uniq translate untranslate warp_text formatAlaTeX formatLines deref) ], functional => [ qw(fold_left compose map_index grep_index map_each grep_each map_tab_hash mapn mapn_ difference2 before_leaving catch_cdie cdie) ], file => [ qw(dirname basename touch all glob_ cat_ output symlinkf chop_ mode typeFromMagic) ], system => [ qw(sync makedev unmakedev psizeof strcpy gettimeofday syscall_ salt getVarsFromSh setVarsInSh setVarsInCsh substInFile availableRam availableMemory removeXiBSuffix template2file formatTime) ], constant => [ qw($printable_chars $sizeof_int $bitof_int $SECTORSIZE) ], ); @EXPORT_OK = map { @$_ } values %EXPORT_TAGS; #-##################################################################################### #- Globals #-##################################################################################### $printable_chars = "\x20-\x7E"; $sizeof_int = psizeof("i"); $bitof_int = $sizeof_int * 8; $SECTORSIZE = 512; #-##################################################################################### #- Functions #-##################################################################################### sub fold_left(&@) { my $f = shift; local $a = shift; foreach $b (@_) { $a = &$f() } $a } sub _ { my $s = shift @_; my $t = translate($s); $t && ref $t or return sprintf $t, @_; my ($T, @p) = @$t; sprintf $T, @_[@p]; } #-delete $main::{'_'}; sub __ { $_[0] } sub even($) { $_[0] % 2 == 0 } sub odd($) { $_[0] % 2 == 1 } sub min { fold_left { $a < $b ? $a : $b } @_ } sub max { fold_left { $a > $b ? $a : $b } @_ } sub sum { fold_left { $a + $b } @_ } sub and_{ fold_left { $a && $b } @_ } sub or_ { fold_left { $a || $b } @_ } sub sqr { $_[0] * $_[0] } sub sign { $_[0] <=> 0 } sub product { fold_left { $a * $b } @_ } sub first { $_[0] } sub second { $_[1] } sub top { $_[-1] } sub uniq { my %l; @l{@_} = (); keys %l } sub to_int { $_[0] =~ /(\d*)/; $1 } sub to_float { $_[0] =~ /(\d*(\.\d*)?)/; $1 } sub ikeys { my %l = @_; sort { $a <=> $b } keys %l } sub add2hash($$) { my ($a, $b) = @_; while (my ($k, $v) = each %{$b || {}}) { $a->{$k} ||= $v } } sub add2hash_($$) { my ($a, $b) = @_; while (my ($k, $v) = each %{$b || {}}) { exists $a->{$k} or $a->{$k} = $v } } sub member { my $e = shift; foreach (@_) { $e eq $_ and return 1 } 0 } sub dirname { @_ == 1 or die "usage: dirname <name>\n"; local $_ = shift; s|[^/]*/*\s*$||; s|(.)/*$|$1|; $_ || '.' } sub basename { @_ == 1 or die "usage: basename <name>\n"; local $_ = shift; s|/*\s*$||; s|.*/||; $_ } sub bool($) { $_[0] ? 1 : 0 } sub invbool { my $a = shift; $$a = !$$a; $$a } sub listlength { scalar @_ } sub bool2text { $_[0] ? "true" : "false" } sub text2bool { my $t = lc($_[0]); $t eq "true" || $t eq "yes" ? 1 : 0 } sub strcpy { substr($_[0], $_[2] || 0, length $_[1]) = $_[1] } sub cat_ { local *F; open F, $_[0] or $_[1] ? die "cat of file $_[0] failed: $!\n" : return; my @l = <F>; wantarray ? @l : join '', @l } sub output { my $f = shift; local *F; open F, ">$f" or die "output in file $f failed: $!\n"; print F foreach @_; } sub deref { ref $_[0] eq "ARRAY" ? @{$_[0]} : ref $_[0] eq "HASH" ? %{$_[0]} : $_[0] } sub linkf { unlink $_[1]; link $_[0], $_[1] } sub symlinkf { unlink $_[1]; symlink $_[0], $_[1] } sub chop_ { map { my $l = $_; chomp $l; $l } @_ } sub divide { my $d = int $_[0] / $_[1]; wantarray ? ($d, $_[0] % $_[1]) : $d } sub round { int ($_[0] + 0.5) } sub round_up { my ($i, $r) = @_; $i += $r - ($i + $r - 1) % $r - 1; } sub round_down { my ($i, $r) = @_; $i -= $i % $r; } sub is_empty_array_ref { my $a = shift; !defined $a || @$a == 0 } sub is_empty_hash_ref { my $a = shift; !defined $a || keys(%$a) == 0 } sub difference2 { my %l; @l{@{$_[1]}} = (); grep { !exists $l{$_} } @{$_[0]} } sub intersection { my (%l, @m); @l{@{shift @_}} = (); foreach (@_) { @m = grep { exists $l{$_} } @$_; %l = (); @l{@m} = (); } keys %l } sub set_new(@) { my %l; @l{@_} = undef; { list => [ @_ ], hash => \%l } } sub set_add($@) { my $o = shift; foreach (@_) { exists $o->{hash}{$_} and next; push @{$o->{list}}, $_; $o->{hash}{$_} = undef } } sub sync { syscall_('sync') } sub gettimeofday { my $t = pack "LL"; syscall_('gettimeofday', $t, 0) or die "gettimeofday failed: $!\n"; unpack("LL", $t) } sub remove_spaces { local $_ = shift; s/^ +//; s/ +$//; $_ } sub mode { my @l = stat $_[0] or die "unable to get mode of file $_[0]: $!\n"; $l[2] } sub psizeof { length pack $_[0] } sub touch { my ($f) = @_; unless (-e $f) { local *F; open F, ">$f"; } my $now = time; utime $now, $now, $f; } sub map_index(&@) { my $f = shift; my $v; local $::i = 0; map { $v = &$f($::i); $::i++; $v } @_; } sub grep_index(&@) { my $f = shift; my $v; local $::i = 0; grep { $v = &$f($::i); $::i++; $v } @_; } sub map_each(&%) { my ($f, %h) = @_; my @l; local ($::a, $::b); while (($::a, $::b) = each %h) { push @l, &$f($::a, $::b) } @l; } sub grep_each(&%) { my ($f, %h) = @_; my %l; local ($::a, $::b); while (($::a, $::b) = each %h) { $l{$::a} = $::b if &$f($::a, $::b) } %l; } #- pseudo-array-hash :) sub map_tab_hash(&$@) { my ($f, $fields, @tab_hash) = @_; my %hash; my $key = { map_index {($_, $::i + 1)} @{$fields} }; for (my $i = 0; $i < @tab_hash; $i += 2) { my $h = [$key, @{$tab_hash[$i + 1]}]; &$f($i, $h) if $f; $hash{ $tab_hash[$i] } = $h; } %hash; } sub smapn { my $f = shift; my $n = shift; my @r = (); for (my $i = 0; $i < $n; $i++) { push @r, &$f(map { $_->[$i] } @_); } @r } sub mapn(&@) { my $f = shift; smapn($f, min(map { scalar @$_ } @_), @_); } sub mapn_(&@) { my $f = shift; smapn($f, max(map { scalar @$_ } @_), @_); } sub add_f4before_leaving { my ($f, $b, $name) = @_; unless ($common::before_leaving::{$name}) { no strict 'refs'; ${"common::before_leaving::$name"} = 1; ${"common::before_leaving::list"} = 1; } local *N = *{$common::before_leaving::{$name}}; my $list = *common::before_leaving::list; $list->{$b}{$name} = $f; *N = sub { my $f = $list->{$_[0]}{$name} or die ''; $name eq 'DESTROY' and delete $list->{$_[0]}; goto $f; } unless defined &{*N}; } #- ! the functions are not called in the order wanted, in case of multiple before_leaving :( sub before_leaving(&) { my ($f) = @_; my $b = bless {}, 'common::before_leaving'; add_f4before_leaving($f, $b, 'DESTROY'); $b; } sub catch_cdie(&&) { my ($f, $catch) = @_; local @common::cdie_catches; unshift @common::cdie_catches, $catch; &$f(); } sub cdie($;&) { my ($err, $f) = @_; foreach (@common::cdie_catches) { $@ = $err; &{$_}(\$err) and return; } die $err; } sub all { my $d = shift; local *F; opendir F, $d or die "all: can't open dir $d: $!\n"; my @l = grep { $_ ne '.' && $_ ne '..' } readdir F; closedir F; @l; } sub glob_ { my ($d, $f) = ($_[0] =~ /\*/) ? (dirname($_[0]), basename($_[0])) : ($_[0], '*'); $d =~ /\*/ and die "glob_: wildcard in directory not handled ($_[0])\n"; ($f = quotemeta $f) =~ s/\\\*/.*/g; $d =~ m|/$| or $d .= '/'; map { $d eq './' ? $_ : "$d$_" } grep { /^$f$/ } all($d); } sub syscall_ { my $f = shift; require 'syscall.ph'; syscall(&{$common::{"SYS_$f"}}, @_) == 0; } sub salt($) { my ($nb) = @_; require 'devices.pm'; open F, devices::make("random") or die "missing random"; my $s; read F, $s, $nb; local $_ = pack "b8" x $nb, unpack "b6" x $nb, $s; tr [\0-\x3f] [0-9a-zA-Z./]; $_; } sub makedev { ($_[0] << 8) | $_[1] } sub unmakedev { $_[0] >> 8, $_[0] & 0xff } sub translate { my ($s) = @_; my ($lang) = $ENV{LANGUAGE} || $ENV{LC_MESSAGES} || $ENV{LC_ALL} || $ENV{LANG} || 'en'; require lang; foreach (split ':', $lang) { lang::load_po($_) unless defined $po::I18N::{$_}; return ${$po::I18N::{$_}}{$s} || $s if %{$po::I18N::{$_}}; } $s; } sub untranslate($@) { my $s = shift || return; foreach (@_) { translate($_) eq $s and return $_ } die "untranslate failed"; } sub warp_text($;$) { my ($text, $width) = @_; $width ||= 80; my @l; foreach (split "\n", $text) { my $t = ''; foreach (split /\s+/, $_) { if (length "$t $_" > $width) { push @l, $t; $t = $_; } else { $t = "$t $_"; } } push @l, $t; } @l; } sub formatAlaTeX($) { my ($t, $tmp); foreach (split "\n", $_[0]) { if (/^$/) { $t .= ($t && "\n") . $tmp; $tmp = ''; } else { $tmp = ($tmp && "$tmp ") . $_; } } $t . ($t && $tmp && "\n") . $tmp; } sub formatLines($) { my ($t, $tmp); foreach (split "\n", $_[0]) { if (/^$/) { $t .= "$tmp\n"; $tmp = ""; } elsif (/^\s/) { $t .= "$tmp\n"; $tmp = $_; } else { $tmp = ($tmp ? "$tmp " : ($t && "\n") . $tmp) . $_; } } "$t$tmp\n"; } sub getVarsFromSh($) { my %l; local *F; open F, $_[0] or return; foreach (<F>) { my ($v, $val, $val2) = /^\s* # leading space (\w+) = # variable ( "([^"]*)" # double-quoted text | '([^']*)' # single-quoted text | [^'"\s]+ # normal text ) \s*$ # end of line /x or next; $l{$v} = $val2 || $val; } %l; } sub setVarsInSh { my ($file, $l, @fields) = @_; @fields = keys %$l unless @fields; #- my $b = 1; $b &&= $l->{$_} foreach @fields; $b or return; local *F; open F, "> $_[0]" or die "cannot create config file $file"; $l->{$_} and print F "$_=$l->{$_}\n" foreach @fields; } sub setVarsInCsh { my ($file, $l, @fields) = @_; @fields = keys %$l unless @fields; #- my $b = 1; $b &&= $l->{$_} foreach @fields; $b or return; local *F; open F, "> $_[0]" or die "cannot create config file $file"; $l->{$_} and print F "setenv $_ $l->{$_}\n" foreach @fields; } sub template2file($$%) { my ($inputfile, $outputfile, %toreplace) = @_; local *OUT; local *IN; open IN, $inputfile or die "Can't open $inputfile $!"; if ($::testing) { *OUT = *STDOUT; } else { open OUT, ">$outputfile" or die "Can't open $outputfile $!"; } map { s/@@@(.*?)@@@/$toreplace{$1}/g; print OUT; } <IN>; } sub substInFile(&@) { my $f = shift; foreach my $file (@_) { if (-e $file) { local @ARGV = $file; local ($^I, $_) = ''; while (<>) { &$f($_); print } } else { local *F; my $old = select F; # that way eof return true local $_ = ''; &$f($_); select $old; eval { output($file, $_) }; } } } sub best_match { my ($str, @lis) = @_; my @words = split /\W+/, $str; my ($max, $res) = 0; foreach (@lis) { my $count = 0; foreach my $i (@words) { $count++ if /$i/i; } $max = $count, $res = $_ if $count >= $max; } $res; } sub bestMatchSentence { my $best = -1; my $bestSentence; my @s = split /\W+/, shift; foreach (@_) { my $count = 0; foreach my $e (@s) { $count++ if /$e/i; } $best = $count, $bestSentence = $_ if $count > $best; } $bestSentence; } # count the number of character that match sub bestMatchSentence2 { my $best = -1; my $bestSentence; my @s = split /\W+/, shift; foreach (@_) { my $count = 0; foreach my $e (@s) { $count+= length ($e) if /$e/i; } $best = $count, $bestSentence = $_ if $count > $best; } $bestSentence; } sub typeFromMagic($@) { my $f = shift; local *F; sysopen F, $f, 0 or return; my $tmp; M: foreach (@_) { my ($name, @l) = @$_; while (@l) { my ($offset, $signature) = splice(@l, 0, 2); sysseek(F, $offset, 0) or next M; sysread(F, $tmp, length $signature); $tmp eq $signature or next M; } return $name; } undef; } sub availableRam() { sum map { /(\d+)/ } grep { /^(MemTotal):/ } cat_("/proc/meminfo"); } sub availableMemory() { sum map { /(\d+)/ } grep { /^(MemTotal|SwapTotal):/ } cat_("/proc/meminfo"); } sub removeXiBSuffix($) { local $_ = shift; /(\d+)k$/i and return $1 * 1024; /(\d+)M$/i and return $1 * 1024 * 1024; /(\d+)G$/i and return $1 * 1024 * 1024 * 1024; $_; } sub formatTime($) { my ($s, $m, $h) = gmtime($_[0]); sprintf "%02d:%02d:%02d", $h, $m, $s; } #-###################################################################################### #- Wonderful perl :( #-###################################################################################### 1; #