aboutsummaryrefslogtreecommitdiff
path: root/files/usr/local/libexec/idm-ssh-known-hosts.common
blob: 5b784d6ea452c100b16b08d7ffbbf8bfacc7b2c9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
#!/usr/local/bin/perl

use strict;
use warnings;

use Net::LDAP;
use Net::LDAP::Util qw(escape_filter_value);
use Authen::SASL;

sub quit {
  # Prints an error message and exits with code 0.
  # NB: If KnownHostsCommand returns nonzero, the entire SSH connection aborts,
  #     which isn't what we want.
  print STDERR "$0: $_[0]" if @_ > 0;
  exit 0
}

@ARGV == 1 or die "usage: $0 HOSTNAME\n";
my $hostname = $ARGV[0];

open my $fh, '<', '/usr/local/etc/openldap/ldap.conf' or quit($!);
my %config;
while (<$fh>) {
  chomp;
  next if /^#/;
  my @pair = split(' ', $_, 2);
  next unless (@pair == 2);
  $config{$pair[0]} = $pair[1];
}
close($fh);

my $mech   = $config{SASL_MECH} // 'GSSAPI';
my $uri    = $config{URI}  // quit('URI not specified');
my $basedn = $config{HOSTS_BASE} // quit('HOSTS_BASE not specified');

my $conn = Net::LDAP->new($uri, version => '3') or quit($@);
my $sasl = Authen::SASL->new($mech);
my $status = $conn->bind(sasl => $sasl);
$status->code and quit($status->error);

my $search = $conn->search(
  scope  => 'sub',
  base   => $basedn,
  filter => '(&(sshPublicKey=*)(associatedDomain=' . escape_filter_value($hostname) . '))',
  attrs  => ['sshPublicKey']);
$search->code and quit($search->error);

quit if $search->entries == 0;
quit("multiple LDAP entries returned for host: $hostname\n") if $search->entries > 1;

print "$hostname $_\n" foreach (($search->entries)[0]->get_value('sshPublicKey'));