X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=dsa-nagios-checks%2Fshare%2Fweak-ssh-keys-check;h=a9bd75239814b91c07c1744cd5066d849d3f2e34;hb=2bcf5c18f38e8288bb766dbf1a9fd70e5eccacd5;hp=e35a8a10c96e5f2b069a3b8ad58d88b7d4b02cdf;hpb=42e99da9d896a33803e763c746e9a103183b6b34;p=mirror%2Fdsa-nagios.git

diff --git a/dsa-nagios-checks/share/weak-ssh-keys-check b/dsa-nagios-checks/share/weak-ssh-keys-check
index e35a8a1..a9bd752 100755
--- a/dsa-nagios-checks/share/weak-ssh-keys-check
+++ b/dsa-nagios-checks/share/weak-ssh-keys-check
@@ -92,11 +92,13 @@ use IPC::Open3;
 my $fprdb_fname = "/var/lib/dsa/ssh-weak-keys.db" ;
 my ($outfile, $help);
 my $dsa_nowarn = 0;
+my $debian_org = 1;
 
 GetOptions(     'help|h' => \$help, #Help function
 		'statusfile|s=s' => \$outfile, 
 		'fprdb|f=s' => \$fprdb_fname,
-		'n|dsa_nowarn' => \$dsa_nowarn,  
+		'n|dsa_nowarn' => \$dsa_nowarn,
+		'd|debian-org!' => \$debian_org,
 );
 
 pod2usage(1) if $help;
@@ -123,9 +125,11 @@ my $text = '';
 my %key_sizes;
 
 
-
-#&from_user_all;
-&from_debianorg_places;
+if ($debian_org) {
+	&from_debianorg_places;
+} else {
+	&from_user_all;
+}
 &from_ssh_host(qw(localhost));
 
 my $status="OK";
@@ -198,6 +202,7 @@ sub from_ssh_key_file ($) {
     my $name = shift;
     if (open (my $FH, '<', $name)) {
 	my $key = <$FH>; 
+	close($FH);
 	if (! defined $key) {
 		$weird_keyfiles++;
 		$text .= "cannot read $name properly - empty?\n";
@@ -332,7 +337,7 @@ sub from_debianorg_places () {
 	for my $file (grep { ! -d $d.'/'.$_ } readdir(D)) {
 	    next if ($file eq 'README-DSA-BUILDD');
 	    my $f = $d.'/'.$file;
-	    from_ssh_key_file $f if -r $f;
+	    from_ssh_auth_file $f if -r $f;
 	};
     };
 }