lyadmin/perl-script/create_user.pl

121 linhas
2.6 KiB
Perl

2020-11-24 14:51:43 -05:00
#!/usr/bin/perl
use warnings;
use strict;
use JSON;
2020-11-24 14:51:43 -05:00
2020-11-30 00:05:17 -05:00
# create-user.pl
#
# processes new user request files
2020-11-24 14:51:43 -05:00
my $working_dir = "./";
my $account_dir = $working_dir."req/";
my $conf_path = $working_dir."lyadmin.conf.json";
my $ul_path = $working_dir."user_list.txt";
my $SHELL_ENUM;
2020-11-24 14:51:43 -05:00
my @g;
2020-11-30 00:05:17 -05:00
# Given a username... prompts and creates that user
2020-11-24 23:05:19 -05:00
sub create($){
2020-11-24 22:33:03 -05:00
my $id = $_[0];
my $fn1 = $account_dir.$id.".ident";
2020-11-24 22:33:03 -05:00
my $username;
my $shell_pref;
my $user_email;
2020-11-26 17:31:34 -05:00
my $pub_key;
2020-11-24 22:33:03 -05:00
2020-11-30 00:05:17 -05:00
# read in username and validate
2020-11-26 17:31:34 -05:00
open FILE, $fn1 or die "could not open file $fn1";
2020-11-24 22:33:03 -05:00
$username = <FILE>;
chomp $username;
2020-11-30 00:05:17 -05:00
if(length($username) > 31 || !($username =~ /^[A-Za-z][A-Za-z0-9]+$/)){
printf("%s has an INVALID username\n", $id);
die ("oh no");
}
2020-11-30 00:05:17 -05:00
# read in email
2020-11-24 22:33:03 -05:00
$user_email = <FILE>;
chomp $user_email;
2020-11-30 00:05:17 -05:00
# read in shell and validate
2020-11-24 22:33:03 -05:00
{
2020-11-26 17:31:34 -05:00
my $s0 = <FILE>;
chomp $s0;
unless($SHELL_ENUM->{$s0}){
die "invalid shell setting $s0 in file $id.ident";
}
2020-11-26 17:31:34 -05:00
$shell_pref = $SHELL_ENUM->{$s0};
2020-11-24 22:33:03 -05:00
}
2020-11-30 00:05:17 -05:00
# read in pub key
2020-11-26 17:31:34 -05:00
$pub_key = <FILE>;
chomp $pub_key;
2020-11-24 22:33:03 -05:00
{
2020-11-30 00:05:17 -05:00
# Prompt to make sure the username looks OK
2020-11-24 22:33:03 -05:00
my $cmd;
$cmd = "useradd -m -s " . $shell_pref . " " . $username;
printf("Y/N is this command OK?: %s\n", $cmd);
2020-11-26 17:31:34 -05:00
if(!(<STDIN> =~ /^y/i)){
die "invalid characters?!!";
}
2020-11-30 00:05:17 -05:00
# create the user
system($cmd);
2020-11-27 00:49:15 -05:00
system("echo '".$pub_key."' > /home/$username/.ssh/authorized_keys");
2020-11-26 01:14:59 -05:00
system("chmod 711 /home/$username");
2020-11-26 17:36:27 -05:00
system("mv $fn1 $fn1.done");
system("echo $username >> $ul_path");
2020-11-24 22:33:03 -05:00
}
close FILE;
}
2020-11-30 00:05:17 -05:00
# MAIN starts here
# Checks if user is root
if(!(`id` =~ /uid=0/)){
die "please run this script as root";
}
2020-11-30 00:05:17 -05:00
# Adjusts the relative file paths based on where
# the script runs from
if( `pwd` =~ /perl-script\/?\s*$/){
$working_dir = "../";
$account_dir = $working_dir."req/";
$conf_path = $working_dir."lyadmin.conf.json";
$ul_path = $working_dir."user_list.txt";
printf("%s\n", $conf_path);
}elsif(!(join(" ", glob("./*")) =~ /perl-script/)){
die "please run this script with ./perl-script/ as the present working directory";
}
2020-11-30 00:05:17 -05:00
# Opens the conf file to read
# shell enums
open FILE, $conf_path or die "could not open file $conf_path";
{
my $conf_str;
my $conf_obj;
local $/=undef;
$conf_str = <FILE>;
chomp $conf_str;
$conf_obj = decode_json($conf_str);
$SHELL_ENUM = $conf_obj->{"shell"};
};
close FILE;
2020-11-30 00:05:17 -05:00
# Saves all new user requests to an array
@g = glob("$account_dir*");
2020-11-26 17:31:34 -05:00
@g = map { s/.*\/([^\/]*).ident$/$1/; $_ } grep {$_ =~ /ident$/} @g;
2020-11-24 14:51:43 -05:00
2020-11-30 00:05:17 -05:00
# Iterate and create the users
2020-11-24 14:51:43 -05:00
for my $fn (@g){
2020-11-24 23:05:19 -05:00
create($fn);
2020-11-24 14:51:43 -05:00
}