update-chroots/update-chroots.pl

79 lines
1.7 KiB
Perl
Raw Permalink Normal View History

2020-11-12 12:32:35 -08:00
#!/usr/local/bin/perl -w
#
# TODO:
# - Be smart about adding ld-elf.so.hints and ld-elf.so.1
#
use strict;
use vars qw( $CONF_FILE %CONF $DEBUG );
use Config::Std;
use File::Basename;
use Data::Dumper;
$CONF_FILE = '/usr/local/etc/chroots.conf';
$DEBUG = 1;
if (-e $CONF_FILE ) {
if ( -r $CONF_FILE ) {
print("$CONF_FILE\n") if $DEBUG;
read_config $CONF_FILE => %CONF;
my @users = keys(%CONF);
foreach my $user (@users) {
print("$user:\n") if $DEBUG;
my $chroot = "/home/$user"; # FIXME: don't guess, know
my @files;
if ( exists $CONF{$user}{'file'} ) {
if ( ref($CONF{$user}->{'file'}) eq 'ARRAY' ) {
@files = @{$CONF{$user}{'file'}};
}
else {
push(@files, $CONF{$user}->${'file'});
}
}
my %uniq_files;
foreach my $file (@files) {
$uniq_files{$file} = 1;
my @libs = `ldd $file | egrep -v ^/ | cut -w -f4`;
foreach my $lib (@libs) {
chomp($lib);
$uniq_files{$lib} = 1;
}
}
foreach my $file ( keys(%uniq_files) ) {
my $file_dir = dirname($file);
if (! -e "$chroot$file_dir" ) {
system("mkdir -p $chroot$file_dir");
}
my $cmd = "cp -f $file $chroot$file";
print("$cmd\n") if $DEBUG;
system($cmd);
}
my @dirs;
if ( exists $CONF{$user}{'dir'} ) {
if ( ref($CONF{$user}->{'dir'}) eq 'ARRAY' ) {
@dirs = @{$CONF{$user}{'dir'}};
}
else {
push(@dirs, $CONF{$user}->{'dir'});
}
}
foreach my $dir ( @dirs ) {
if (! -e "$chroot$dir" ) {
my $cmd = ("mkdir -p $chroot$dir");
print("$cmd\n") if $DEBUG;
system("mkdir -p $chroot$dir");
}
my $cmd = "cp -R $dir/ $chroot$dir/";
print("$cmd\n") if $DEBUG;
system($cmd);
}
}
}
else {
die("Error: Could not read $CONF_FILE\n");
}
}