306 lines
9.8 KiB
Perl
306 lines
9.8 KiB
Perl
|
#!/usr/bin/env perl
|
||
|
#
|
||
|
# Copyright (c) 2020 Mischa Peters <mischa @ openbsd.amsterdam>
|
||
|
#
|
||
|
# Permission to use, copy, modify, and distribute this software for any
|
||
|
# purpose with or without fee is hereby granted, provided that the above
|
||
|
# copyright notice and this permission notice appear in all copies.
|
||
|
#
|
||
|
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
||
|
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
||
|
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
||
|
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
||
|
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
||
|
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
||
|
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
||
|
#
|
||
|
# vmm(4)/vmd(8) VM notify script for OpenBSD Amsterdam
|
||
|
# 2020/05/17 initial release
|
||
|
#
|
||
|
# This is WIP, currently there is a lot of overlap in the functions.
|
||
|
#
|
||
|
use 5.024;
|
||
|
use strict;
|
||
|
use warnings;
|
||
|
use autodie;
|
||
|
use POSIX qw(strftime);
|
||
|
|
||
|
# Get function and function_variable (vmid) from arguments
|
||
|
my $function = $ARGV[0] || "empty";
|
||
|
my $function_variable = $ARGV[1] || "empty";
|
||
|
|
||
|
# fuction to parse _deploy.conf and vm*.txt files
|
||
|
# all variables are stripped and added to either %vms or %conf
|
||
|
sub get_variables {
|
||
|
my ($hash_name, @files) = @_;
|
||
|
my %hash;
|
||
|
my $filename;
|
||
|
my $vm_name;
|
||
|
my $vm_number;
|
||
|
|
||
|
for my $file (@files) {
|
||
|
# When hash is 'vms' use the vm_name as key
|
||
|
# Otherwise use 'conf' as key
|
||
|
if ($hash_name eq "vms") {
|
||
|
($filename = $file) =~ s/.*\///;
|
||
|
($vm_name = $filename) =~ s/\.txt//;
|
||
|
($vm_number = $vm_name) =~ s/^vm//;
|
||
|
$hash{$vm_name}{'vm_number'} = $vm_number;
|
||
|
}
|
||
|
|
||
|
open my $fh, "<", "$file";
|
||
|
while (my $row = <$fh>) {
|
||
|
next if ($row =~ /^\s*($|#)/);
|
||
|
chomp ($row);
|
||
|
(my $key, my $val) = split(/=/, $row, 2);
|
||
|
if ($hash_name eq "vms") {
|
||
|
($hash{$vm_name}{$key} .= $val) =~ s/^"+|"+$//g;
|
||
|
} else {
|
||
|
($hash{$hash_name}{$key} .= $val) =~ s/^"+|"+$//g;
|
||
|
}
|
||
|
}
|
||
|
close $fh;
|
||
|
}
|
||
|
return %hash;
|
||
|
}
|
||
|
|
||
|
sub deployed {
|
||
|
my %conf = %{$_[0]};
|
||
|
my %vms = %{$_[1]};
|
||
|
my $_etc = $conf{'conf'}{'ETC'};
|
||
|
my $_vms = $conf{'conf'}{'VMS'};
|
||
|
my $_tmpl = $conf{'conf'}{'TEMPLATES'};
|
||
|
my $_server = $conf{'conf'}{'SERVER'};
|
||
|
my $template = "$_tmpl/email-deployed.txt";
|
||
|
|
||
|
for my $vm_name (sort keys %vms) {
|
||
|
my $_date = $vms{$vm_name}{'date'};
|
||
|
my $_payment = $vms{$vm_name}{'payment'};
|
||
|
my $_donated = $vms{$vm_name}{'donated'};
|
||
|
my $_name = $vms{$vm_name}{'name'};
|
||
|
my $_email = $vms{$vm_name}{'email'};
|
||
|
my $_hostname = $vms{$vm_name}{'hostname'};
|
||
|
my $_boot = $vms{$vm_name}{'boot'} || "yes";
|
||
|
|
||
|
if ($function_variable =~ /empty/) { print "No VMID provided\n"; last; }
|
||
|
if ($function_variable =~ /$vm_name/) {
|
||
|
(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
|
||
|
my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');
|
||
|
|
||
|
open(my $fh, '<', $template);
|
||
|
open my $fh_email, "|-", "/usr/sbin/sendmail -t";
|
||
|
printf $fh_email "To: %s\n", $_email;
|
||
|
while (my $row = <$fh>) {
|
||
|
chomp $row;
|
||
|
$row =~ s/NAME/$_firstname/g;
|
||
|
$row =~ s/IP$/$ipaddress/g;
|
||
|
$row =~ s/VMID/$vm_name/g;
|
||
|
$row =~ s/SERVER/$_server/g;
|
||
|
print $fh_email "$row\n";
|
||
|
}
|
||
|
close $fh_email;
|
||
|
print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
# function to collect stopped vms
|
||
|
sub stopped {
|
||
|
my %conf = %{$_[0]};
|
||
|
my %vms = %{$_[1]};
|
||
|
my $_etc = $conf{'conf'}{'ETC'};
|
||
|
my $_vms = $conf{'conf'}{'VMS'};
|
||
|
my $_tmpl = $conf{'conf'}{'TEMPLATES'};
|
||
|
my $_server = $conf{'conf'}{'SERVER'};
|
||
|
my $template = "$_tmpl/email-stopped.txt";
|
||
|
|
||
|
my @stopped_vms = qx(vmctl show | grep stopped | awk '{print \$9}');
|
||
|
|
||
|
for my $vm_name (sort keys %vms) {
|
||
|
my $_date = $vms{$vm_name}{'date'};
|
||
|
my $_payment = $vms{$vm_name}{'payment'};
|
||
|
my $_donated = $vms{$vm_name}{'donated'};
|
||
|
my $_name = $vms{$vm_name}{'name'};
|
||
|
my $_email = $vms{$vm_name}{'email'};
|
||
|
my $_hostname = $vms{$vm_name}{'hostname'};
|
||
|
my $_boot = $vms{$vm_name}{'boot'} || "yes";
|
||
|
|
||
|
if ($_boot =~ /no/) {
|
||
|
print "NOT NOTIFIED: $_name, $_email, $_hostname, $_server ($vm_name)\n";
|
||
|
next;
|
||
|
}
|
||
|
if (grep(/$vm_name/, @stopped_vms)) {
|
||
|
(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
|
||
|
my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');
|
||
|
|
||
|
open(my $fh, '<', $template);
|
||
|
open my $fh_email, "|-", "/usr/sbin/sendmail -t";
|
||
|
printf $fh_email "To: %s\n", $_email;
|
||
|
while (my $row = <$fh>) {
|
||
|
chomp $row;
|
||
|
$row =~ s/NAME/$_firstname/g;
|
||
|
$row =~ s/IP$/$ipaddress/g;
|
||
|
$row =~ s/VMID/$vm_name/g;
|
||
|
$row =~ s/SERVER/$_server/g;
|
||
|
print $fh_email "$row\n";
|
||
|
}
|
||
|
close $fh_email;
|
||
|
print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
sub renewal {
|
||
|
my %conf = %{$_[0]};
|
||
|
my %vms = %{$_[1]};
|
||
|
my $_etc = $conf{'conf'}{'ETC'};
|
||
|
my $_vms = $conf{'conf'}{'VMS'};
|
||
|
my $_tmpl = $conf{'conf'}{'TEMPLATES'};
|
||
|
my $_server = $conf{'conf'}{'SERVER'};
|
||
|
my $template = "$_tmpl/email-renewal.txt";
|
||
|
|
||
|
my $year = strftime("%Y", localtime);
|
||
|
my $month = strftime("%m", localtime);
|
||
|
|
||
|
my $total_donated = qx(ftp -Vo- https://obsda.ms/index.html | grep "donated to OpenBSD" | awk -F';' '{print \$4}' | awk '{printf \$1}');
|
||
|
my $total_vms = qx(ftp -Vo- https://obsda.ms/index.html | grep "deployed" | awk -F '>' '{print \$3}' | awk '{printf \$1}');
|
||
|
my $total_hosts = qx(ftp -Vo- https://obsda.ms/servers.html | grep -c ">Server " | tr -d '\n');
|
||
|
|
||
|
for my $vm_name (sort keys %vms) {
|
||
|
my $_date = $vms{$vm_name}{'date'};
|
||
|
my $_payment = $vms{$vm_name}{'payment'};
|
||
|
my $_donated = $vms{$vm_name}{'donated'};
|
||
|
my $_name = $vms{$vm_name}{'name'};
|
||
|
my $_email = $vms{$vm_name}{'email'};
|
||
|
my $_hostname = $vms{$vm_name}{'hostname'};
|
||
|
my $_boot = $vms{$vm_name}{'boot'} || "yes";
|
||
|
|
||
|
if ($_donated =~ /done/) { next; }
|
||
|
if ($_donated =~ /expire/) { next; }
|
||
|
if ($_donated =~ /sponsor/) { next; }
|
||
|
if ($_donated =~ /renewal/) { next; }
|
||
|
if ($_date =~ /$year\//) { next; }
|
||
|
if ($_date =~ /\/$month\//) {
|
||
|
(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
|
||
|
(my $_year, my $_month, my $_day) = split(/\//, $_date, 3);
|
||
|
my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');
|
||
|
|
||
|
open(my $fh, '<', $template);
|
||
|
open my $fh_email, "|-", "/usr/sbin/sendmail -t";
|
||
|
printf $fh_email "To: %s\n", $_email;
|
||
|
while (my $row = <$fh>) {
|
||
|
chomp $row;
|
||
|
$row =~ s/NAME/$_firstname/g;
|
||
|
$row =~ s/IP$/$ipaddress/g;
|
||
|
$row =~ s/VMID/$vm_name/g;
|
||
|
$row =~ s/SERVER/$_server/g;
|
||
|
|
||
|
$row =~ s/YEAR/$year/g;
|
||
|
$row =~ s/TOTAL_DONATED/$total_donated/g;
|
||
|
$row =~ s/TOTAL_VMS/$total_vms/g;
|
||
|
$row =~ s/TOTAL_HOSTS/$total_hosts/g;
|
||
|
$row =~ s/PAYMENT/$_payment/g;
|
||
|
|
||
|
print $fh_email "$row\n";
|
||
|
}
|
||
|
close $fh_email;
|
||
|
print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
# function to notify people
|
||
|
sub notify {
|
||
|
my %conf = %{$_[0]};
|
||
|
my %vms = %{$_[1]};
|
||
|
my $_etc = $conf{'conf'}{'ETC'};
|
||
|
my $_vms = $conf{'conf'}{'VMS'};
|
||
|
my $_tmpl = $conf{'conf'}{'TEMPLATES'};
|
||
|
my $_server = $conf{'conf'}{'SERVER'};
|
||
|
my $template = "$_tmpl/email-notify.txt";
|
||
|
|
||
|
for my $vm_name (sort keys %vms) {
|
||
|
my $_date = $vms{$vm_name}{'date'};
|
||
|
my $_payment = $vms{$vm_name}{'payment'};
|
||
|
my $_donated = $vms{$vm_name}{'donated'};
|
||
|
my $_name = $vms{$vm_name}{'name'};
|
||
|
my $_email = $vms{$vm_name}{'email'};
|
||
|
my $_hostname = $vms{$vm_name}{'hostname'};
|
||
|
my $_boot = $vms{$vm_name}{'boot'} || "yes";
|
||
|
|
||
|
if (-e "$template") {
|
||
|
#if ($_boot =~ /yes/) {
|
||
|
(my $_firstname, my $_lastname) = split(/ /, $_name, 2);
|
||
|
my $ipaddress = qx(grep -A2 $vm_name $_etc/dhcpd.conf | awk '/fixed-address/{print \$2}' | tr -d ';\n');
|
||
|
|
||
|
open(my $fh, '<', $template);
|
||
|
open my $fh_email, "|-", "/usr/sbin/sendmail -t";
|
||
|
printf $fh_email "To: %s\n", $_email;
|
||
|
while (my $row = <$fh>) {
|
||
|
chomp $row;
|
||
|
$row =~ s/NAME/$_firstname/g;
|
||
|
$row =~ s/IP$/$ipaddress/g;
|
||
|
$row =~ s/VMID/$vm_name/g;
|
||
|
$row =~ s/SERVER/$_server/g;
|
||
|
print $fh_email "$row\n";
|
||
|
}
|
||
|
close $fh_email;
|
||
|
print "NOTIFIED: $_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name), $ipaddress\n";
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
# function to print all keys & values for debug purposes
|
||
|
sub debug_parse {
|
||
|
my %conf = %{$_[0]};
|
||
|
my %vms = %{$_[1]};
|
||
|
my $_etc = $conf{'conf'}{'ETC'};
|
||
|
my $_vms = $conf{'conf'}{'VMS'};
|
||
|
my $_tmpl = $conf{'conf'}{'TEMPLATES'};
|
||
|
my $_server = $conf{'conf'}{'SERVER'};
|
||
|
my $template = "$_tmpl/email-debug.txt";
|
||
|
print "All VMs\n##\n";
|
||
|
for my $vm_name (sort keys %vms) {
|
||
|
my $_date = $vms{$vm_name}{'date'};
|
||
|
my $_payment = $vms{$vm_name}{'payment'};
|
||
|
my $_donated = $vms{$vm_name}{'donated'};
|
||
|
my $_name = $vms{$vm_name}{'name'};
|
||
|
my $_email = $vms{$vm_name}{'email'};
|
||
|
my $_hostname = $vms{$vm_name}{'hostname'};
|
||
|
my $_boot = $vms{$vm_name}{'boot'} || "yes";
|
||
|
print "$_date, $_payment, $_name, $_email, $_hostname, $_server ($vm_name)\n";
|
||
|
}
|
||
|
print "##\n";
|
||
|
}
|
||
|
|
||
|
# check if _deploy.conf exists
|
||
|
my $dev = $ENV{'HOME'} . "/openbsd.amsterdam/deploy.pl";
|
||
|
my $prod = $ENV{'HOME'};
|
||
|
my $dir;
|
||
|
my $debug;
|
||
|
my %conf;
|
||
|
my %vms;
|
||
|
if (-d "$dev") {
|
||
|
$dir = $dev;
|
||
|
$debug = 1;
|
||
|
%conf = get_variables('conf', "$dir/_deploy.conf");
|
||
|
} elsif (-d "$prod") {
|
||
|
$dir = $prod;
|
||
|
%conf = get_variables('conf', "$dir/_deploy.conf");
|
||
|
} else {
|
||
|
printf "Unable to find config file\n";
|
||
|
exit 1;
|
||
|
}
|
||
|
|
||
|
# parse all vm*.txt files in the VMS directory
|
||
|
my @files = glob "$conf{'conf'}{'VMS'}/*.txt";
|
||
|
%vms = get_variables('vms', @files);
|
||
|
|
||
|
# run functions
|
||
|
if ($debug) { debug_parse(\%conf, \%vms); }
|
||
|
if ($function =~ /deployed/) { deployed(\%conf, \%vms); }
|
||
|
elsif ($function =~ /stopped/) { stopped(\%conf, \%vms); }
|
||
|
elsif ($function =~ /renewal/) { renewal(\%conf, \%vms); }
|
||
|
elsif ($function =~ /notify/) { notify(\%conf, \%vms); }
|
||
|
else { print "Specify function: deployed, stopped, renewal, notify\n"; }
|