From 1ee1229cb75086f569f1793b3472c11d92ec620c Mon Sep 17 00:00:00 2001 From: Jonas Smedegaard Date: Fri, 11 Mar 2005 16:57:22 +0000 Subject: Added scripts for updating Linux Counter. --- machine-update-beta | 1117 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 1117 insertions(+) create mode 100755 machine-update-beta (limited to 'machine-update-beta') diff --git a/machine-update-beta b/machine-update-beta new file mode 100755 index 0000000..6a4db3e --- /dev/null +++ b/machine-update-beta @@ -0,0 +1,1117 @@ +#!/usr/bin/perl -w +# For Emacs: -*- mode:cperl; mode:folding; -*- +# +# Get a machine's critical features, And mail/http them to the Linux Counter +# +# (c) 1999 - Harald Tveit Alvestrand, the Linux Counter Project +# 2003 - PetaMem Group (www.petamem.com) +# License: GNU Copyleft - see bottom of file. +# Changelog: see even more bottom of the file +# +# As a matter of courtesy, if you change this file on your own, +# make sure it does NOT mail to the counter! +# +use strict; +use POSIX; + +our $VERSION = '0.20'; +our $CVS_VERSION = '$Revision: 1.1 $ $Date: 2005-03-11 16:57:22 $ $Author: jonas $'; +our $IsInTestHarness; +use vars qw(%values %oldvalues $errordata $debugdata); # data that is sent +use vars qw($progname %option); +use vars qw(%is_sys_account %is_user %is_account); + +# Some variables are for internal use, and never prompted for in +# the loop of askquestions +our %dontask = ('uniqueid' => 1, # Internal use + 'manual' => 1, + 'method' => 1, + 'owner' => 1, # These 2 are always prompted for + 'key' => 1, + 'uptime' => 1, # We think we know how to get these + ); + +# stuff that controls defaults for passwdscan & accounts subroutines +my ($UID_MIN, $UID_MAX, $got_defs) = (100, 65533, ''); + +# Make sure nothing happens, so that the script's routines +# can be debugged from another file +return 1 if($IsInTestHarness); + +&preparation; +&options; + +&readfile; +&checkconfig; + +if ($option{ask}) { + &askquestions; +} else { + ©manuals; +} + +&writefile; +&sendfile; + +# {{{ preparation + +# +sub preparation { + die "No HOME environment variable\n" if (!$ENV{HOME}); + die "No home diretory\n" if ! -d $ENV{HOME}; + my $infodir = "$ENV{HOME}/.linuxcounter"; + if (! -d $infodir) { + mkdir($infodir, 0766) || die "Unable to make $infodir\n"; + } + # Keep track of where I am; need it to install crontab entry + # progname is a global. + $progname = $0; + if ($progname !~ /^\//) { + my $progdir = `pwd`; + chop $progdir; + $progname = "$progdir/$progname"; + $progname =~ s!/./!/!; + } + chdir($infodir) || die "Unable to change to $infodir\n"; + my ($sysname, $nodename, $release, $version, $machine ) = POSIX::uname(); + if (! -f $nodename) { + print STDERR "Machine-update $VERSION. Use $0 -l to display license.\n"; + print STDERR "Creating the infofile for your computer.\n"; + # Create the infodir + open(INFO, ">$nodename"); + print INFO "uniqueid: ", randomnumber(), "\n"; + close INFO; + } + + srand time % $$; # do some seed "randomization" +} + +# }}} +# {{{ options + +# +sub options { + my $opt; + + while (defined($ARGV[0]) && $ARGV[0] =~ /^-/) { + $opt = shift @ARGV; + $opt =~ /c/ && &installcrontab; + $opt =~ /d/ && $option{DEBUG}++ && print STDERR "Debug is $option{DEBUG}\n"; + $opt =~ /h/ && &help; + $opt =~ /i/ && ($option{ask} = 1); + $opt =~ /l/ && &license; + $opt =~ /m/ && ($option{mail} = 1); + $opt =~ /t/ && ($option{mail} = 0); + $opt =~ /u/ && &uninstallcrontab; + $opt =~ /v/ && die "\n\t Linux Counter machine-update version $VERSION\n" + . "\tCVS version $CVS_VERSION\n"; + $opt =~ /x/ && ($option{info} = 1); + } +} + +# }}} + +# {{{ askquestions + +# +sub askquestions { + return if ! -t STDIN || ! -t STDOUT; + $| = 1; + print "Here you can specify some info that the script can't know for itself\n"; + $values{owner} = askone("Your Linux Counter reg#, if any", $values{owner}); + $values{key} = askone("Your machine's counter reg#, if any", $values{key}); + print "Here is what the program has found:\n"; + for my $key (keys(%values)) { + if (!$dontask{$key}) { + printf "%-10s%1s: %s\n", $key, $dontask{$key}?"*":" ", $values{$key}; + } + } + my $domore = &askone('Do you want to override some of the found values?', 'no'); + if ($domore =~ /^Y/i) { + my $manual; + for my $key (keys(%values)) { + next if $dontask{$key}; + my $value = &askone($key, $oldvalues{$key}, $values{$key}); + if ($values{$key} eq $value) { # go to automatic + &Debug("auto value: $key\n"); + } else { + &Debug("still manual value: $key\n"); + $manual .= " $key"; + } + $values{$key} = $value; + } + $values{manual} = $manual; + } else { + delete $values{manual}; + } +} + +# }}} +# {{{ askone +# +sub askone { + my $prompt = shift; + my $default = shift; + my $probed = shift; + + print $prompt; + if (!defined($default) && defined($probed)) { + $default = $probed; + } + if (defined($default)) { + print " [$default]"; + } + if (defined($probed) && $probed ne $default) { + print "(program found $probed)"; + } + print ':'; + + my $ans = ; + + chop $ans; + &Debug("Answer was $ans\n"); + $ans = $default if (!length($ans)); + + return $ans; +} +# }}} + +# {{{ copymanuals +# +sub copymanuals { + my %keeps; + if ($values{manual}) { + %keeps = map {$_ => 1} split(' ', $values{manual}); + } + + &Debug('Keeping '.join(' ', keys(%keeps))."\n"); + for my $key (keys(%keeps)) { + $values{$key} = $oldvalues{$key}; + } +} +# }}} + +# {{{ readfile + +# +sub readfile { + my ($sysname, $nodename, $release, $version, $machine ) = POSIX::uname(); + open(INFO, $nodename) || die "Did not find infofile $nodename\n"; + while () { + chop; + s/#.*//; + if (/^(\S+): *(.+)/) { + &Debug("Read $1: $2\n"); + $values{$1} = $2; + } else { + print STDERR "Unparsed info line: $_ - discarded\n"; + } + } + close INFO; + %oldvalues = %values; +} + +# }}} +# {{{ writefile + +# +sub writefile { + my ($sysname, $nodename, $release, $version, $machine ) = POSIX::uname(); + + open(INFO, ">$nodename.new"); + for my $val (sort keys(%values)) { + &Debug("Saving $val: $values{$val}\n"); + print INFO "$val: $values{$val}\n"; + } + close INFO; + rename("$nodename.new", $nodename) || die "Rename failed\n"; +} + +# }}} +# {{{ sendfile + +# +sub sendfile { + if ($option{mail}) { + open(MAIL, "|/usr/lib/sendmail machine-registration\@counter.li.org") + || die "Unable to open sendmail\n"; + } else { + warn "--------------------------------------------------------\n"; + warn "This is what will be sent to the Linux Counter if you\n"; + warn "run the program with the -m switch. Now, NOTHING IS SENT\n"; + warn "--------------------------------------------------------\n"; + open(MAIL, ">&STDOUT"); + } + # note that $ENV{USER} isn't (always) set in a cron job... + my $user = (getpwuid($<))[0]; + $user = "unknown-id-$<" if !$user; + print MAIL < 0; + } + print MAIL "//END\n"; + # Attach possible other info + if ($errordata) { + print MAIL "----- Problem info gathered during probing -----\n"; + print MAIL $errordata; + } + $option{info} && do { + print MAIL "----- Debug data for the script maintainer's aid -----\n"; + print MAIL $debugdata; + }; + close MAIL; +} + +# }}} + +# {{{ randomnumber +# +sub randomnumber { + return int(rand(1_000_000_000)); +} +# }}} + +# {{{ checkconfig + +# +sub checkconfig { + my ($sysname, $nodename, $release, $version, $machine ) = POSIX::uname(); + + warn "This is not Linux, but $sysname!\n" if($sysname ne 'Linux'); + $values{method} = "machine-update version $VERSION"; + $values{os} = $sysname; + $values{kernel} = $release; + $values{cpu_uname} = $machine; + $values{name} = $nodename; # First order guess + + # Credit for some of the code below goes to + # Denis Havlik: + # Blame is, of course, all mine - HTA - + # Note - there are numerous problems with df, including: + # - early versions don't support the -l option + # - at least some include SAMBA filesystems in the -l option + open (TMP,"df -l -x shm |"); # exclude shmfs, its 50% of RAM (mostly) + my $HD = 0; + while () { + my @line = split(/\s+/); + ($line[0] =~ /\/dev/) && ($HD += $line[1]); + } + &Debug("$HD kbytes of disk found\n"); + $HD /= 1024; + $values{disk} = sprintf("%d", $HD); + $values{accounts} = &accounts; + $values{users} = &active_users; + + my $uptime = &xbin('uptime'); + if($uptime) { + $uptime = `$uptime`; + # get the uptime string here (could be more elegant, but so it is + # robust against various localizations) + if ($uptime =~ /^\s+[^\s]+\s+[^\s]+\s+([^,]+), /) { + $values{uptime} = $1; + } else { + &ErrorInfo("Can't parse uptime output: $uptime\n"); + } + if ($option{info}) { + &DebugInfo("***** Uptime output *****\n$uptime"); + } + } + # Not sure this is a Right Thing...so not saving it for the moment + # This section based on a patch from Mark-Jason Dominus + # try to guess mailer based on content of /usr/lib/sendmail link + if (-l '/usr/lib/sendmail') { + my $realsendmail = readlink('/usr/lib/sendmail'); + if ($realsendmail eq '../sbin/sendmail') { + $realsendmail = '/usr/sbin/sendmail'; + if (-l $realsendmail) { + $realsendmail = readlink($realsendmail); + } + } + if ($realsendmail =~ m{^/var/qmail}) { + $values{mailer} = "qmail"; + } else { + &DebugInfo("Found sendmail as a link to $realsendmail\n"); + } + } + # Link method did not work. Try to guess based on presence of + # config files. (this is more susceptible to the old-junk problem) + if (!$values{mailer}) { + if ( -d '/var/qmail') { + $values{mailer} = 'qmail'; + } elsif ( -f '/etc/sendmail.cf') { + $values{mailer} = 'sendmail'; + } elsif ( -d '/etc/postfix') { + $values{mailer} = 'postfix'; + } + } + + # forget about /proc/kcore: It's not reliable above 960MB. It also doesn't show + # real memory, because some may be eaten by graphics + # forget also about the "free" command: We don't want to be dependant on that + $values{memory} = int(&getval_from_file('/proc/meminfo',1,1) / (1024*1024)); + # But actually: Shouldn't we return the total of virtual memory here? Seems to + # be relevant to me - RJ - + + &cpuinfo; +} + +# }}} + +# {{{ cpuinfo + +# +sub cpuinfo { + my $cpufile = '/proc/cpuinfo'; # Linux: Place to get info on CPU + my %interesting = (# 2.0 and 2.2 kernels + 'bogomips' => '+bogomips', + 'processor' => '1+processors', + 'vendor_id' => 'cpu_vendor', + # 2.0 kernels + 'cpu' => 'cpu_only', + 'model' => 'cpu_model', + 'model name' => 'cpu_model_name', + # 2.2 kernels + 'cpu MHz' => 'cpu_mhz', + 'cpu family' => 'cpu_family', + # from an Alpha processor + 'cycle frequency [Hz]' => 'cpu_hz', + 'BogoMIPS' => '+bogomips', + 'cpu model' => 'cpu_model', + 'system type' => 'cpu_system_type', + 'cpus detected' => 'processors', + # from a PowerMAC + 'machine' => 'cpu_machine', + 'clock' => 'cpu_clock', + 'motherboard' => 'cpu_motherboard', + # from an UltraSparc + 'BogoMips' => '+bogomips', + 'ncpus active' => 'processors', + ); + + # Zero out the accumulative values + $values{bogomips} = 0; + $values{processors} = 0; + if (open (TMP,"<$cpufile")) { + &DebugInfo("**** Contents of $cpufile ****\n"); + while () { + &DebugInfo($_); # Save /proc/cpuinfo to debugdata if -d + chop; + # A bizarre selection of names are "interesting". + # Make a data-driven pick routine + if (/^(\S+[^:]+\S)\s+: /) { + my $name = $1; + my $value = $'; + if ($interesting{$name}) { + if ($interesting{$name} =~ /^\+/) { + $values{$'} += $value; + } elsif ($interesting{$name} =~ /^1\+/) { + $values{$'}++; + } else { + $values{$interesting{$name}} = $value; + } + } + } + } + } else { + &ErrorInfo("Could not open $cpufile\n"); + } +} + +# }}} + +# {{{ accounts + +# +sub accounts { + my $s; + my $niss; + my $ypcatbin; # will hold path to the ypcat binary (if any) + + open (TMP," /dev/null|" + || ($errordata .= "ypcat failed: $!\n"); + $niss = &passwdscan; + $s += $niss; + close TMP; + &Debug("Status of ypcat: $?\n"); + &DebugErr("Found $niss accounts in ypcat passwd\n"); + } + + &DebugErr('Sysaccounts: ', join(' ', keys(%is_sys_account)), "\n"); + &DebugErr("Found $s accounts total\n"); + + return $s; +} + +# }}} +# {{{ passwdscan + +# +sub passwdscan { + # Code for reading login.defs courtesy of Vassilii Khachaturov + # + local (*DEFS); + # Try importing UID_MIN and UID_MAX from /etc/login.defs, if possible + # else just assume the above defaults for min and max non-system UID + if (!$got_defs && open (DEFS, '/etc/login.defs')) { + while () { + if (/^\s*(UID_(?:MIN|MAX))\s+(\d+)/) { + # elegant, but not compatible with "strict refs": + #${ $1 } = $2; + if ($1 eq "UID_MIN") { + $UID_MIN = $2; + } else { + $UID_MAX = $2; + } + &Debug("DEFS match: $1 = $2\n"); + } + } + close (DEFS); + $got_defs = 1; + } + &Debug("UID_MIN = $UID_MIN, UID_MAX = $UID_MAX\n"); + # I suppose this is as good as it gets - + # Usually user accounts have UID > 100 and + # "system accounts" have UID < 100, but there is no guarantee + # that + # this will hold for pseudo-users like "postgress" etc. + # Also nobody is usually 99 on linux, but -1 on "standard" unices. + # RedHat places the dividing line at 500. Others use 400... + my @line; + my $s = 0; + + while () { + @line = split ':'; + if ($line[2] >= $UID_MIN && $line[2] <= $UID_MAX + && !($line[0] eq 'nobody')) { + $s++; + $is_account{$line[0]} = 1; + } else { + $is_sys_account{$line[0]} = 1; + } + } + + return $s; +} + +# }}} +# {{{ active_users + +# +# This is kind of alpha, but please test it. +# It calculates the number of "active" users based on the "wtmp" entries +# unfortunately at least Mandrake 8 and 9 ship with non-world-read wtmp +# and non-set-uid last, so this does not work any more... +# +# RJ: Actually I think the best thing to do is to bury this code and be silent about it. +# +sub active_users { + my $userslisted; + + for (qw(reboot wtmp runlevel)) { # This sysaccounts shouldn't be counted. Who else? + $is_sys_account{$_} = 1; + } + open( TMP, "/usr/bin/last 2>&1|"); + while () { + chop; + if (m!/var/log/wtmp: Permission denied!) { # RJ: ***Boom*** on every non-EN system + &ErrorInfo("/usr/bin/last failed because /var/log/wtmp isn't readable\n"); + last; + } + last if(!$_); # RJ: quick hack to safe bad code from harm + my @tmp = split; + my $name = $tmp[0]; + if ($is_sys_account{$name}) { + # do nothing + } elsif (defined $is_account{$name}) { + $is_user{$name} = 1; + } elsif (/^\s*$/) { # blank line - do nothing + } elsif ($#tmp == 9) { # OK line, but unknown user + $option{DEBUG} && do { + if (!$userslisted) { + print STDERR 'Know users are: ', + join(' ', keys(%is_account)), "\n"; + $userslisted = 1; + } + print STDERR "Unknown user: $name\n"; + } + } else { + &DebugErr("Strange line: $_\n"); + } + } + close TMP; + my $i = 0; + for (sort keys %is_user) { + $option{DEBUG} && printf "Active user %3d: %s\n", ++$i, $_; + } + &Debug("$i active users found.\n"); + + return $i; +} + +# }}} + +# {{{ installcrontab + +# +sub installcrontab { + my $hour = int(rand(24)); + my $min = int(rand(60)); + my $day = int(rand(7)); # Weekday. This version runs once a week. + my $cron; + + warn "Installing start of script into your crontab\n"; + if (open(CRON, "crontab -l |")) { + &Debug("Checking crontab for machine-update\n"); + &Debug("Want to install as $progname\n"); + while () { + if (/^#/ && $. <= 3) { # initial comment + &Debug("Skipping comment: $_"); + next; + } + if (/machine-update/) { + if (/ $progname -m/) { + die "Crontab entry already installed: $_\n"; + } else { + die "Another entry with machine-update: $_\n"; + } + } + $cron .= $_; + } + close CRON; + &Debug("Result from crontab -l: ", $? / 256, "\n"); + if ($? == 0) { + &Debug("Crontab successfully read\n"); + } elsif ($? == 256) { + warn "You don't seem to have a crontab. I will create one.\n"; + } else { + die "Failed to read your crontab. Please report this as a bug: $?\n"; + } + } else { + &Debug("Result from crontab open(): $?\n"); + die "Unable to execute crontab command. Please check your system\n"; + } + open(CRON, "|crontab -"); + print CRON $cron; + print CRON "$min $hour * * $day $progname -m\n"; + close CRON; + &Debug("Result from crontab: $?\n"); + if ($?) { + die(<) { + if (/^#/ && $. <= 3) { # initial comment + &Debug("Skipping comment: $_"); + next; + } + if (/machine-update/) { + if (/ $progname -m/) { + print STDERR "Crontab entry found and removed\n"; + $found = 1; + next; # skip stuff at end.... + } else { + die "Another entry with machine-update: $_\nUninstall manually?\n"; + } + } + $cron .= $_; + } + close CRON; + &Debug("Result from crontab -l: $?\n"); + if ($?) { + die "Failed to read your crontab. You may not have one?\n"; + } + if ($found) { + open(CRON, "|crontab -"); + print CRON $cron; + close CRON; + &Debug("Result from crontab: $?\n"); + if ($?) { + die(<; # read whole file to array + close FH; + + @cols = split /\s+/, $file[$row]; # get the right row + + return $cols[$col]; # return the right column +} + +# }}} + +# {{{ Debug print debug information if flag is set + +# +sub Debug { + $option{DEBUG} && print @_; +} + +# }}} +# {{{ DebugErr print debug on STDERR if flag is set + +# +sub DebugErr { + $option{DEBUG} && print STDERR @_; +} + +# }}} +# {{{ ErrorInfo +sub ErrorInfo { + $errordata .= join('', @_); +} +# }}} +# {{{ DebugInfo + +sub DebugInfo { + $option{info} && ($debugdata .= join('', @_)); +} + +# }}} + +# {{{ help print help & exit + +# +sub help { + my $host = `uname -n`; + + print <960MB capable +# - slightly better randomness +# +#vim:ts=8:sw=4:sts=4 + -- cgit v1.2.3