From ab9f3da989c663a1752184cd5148a65a27f4c78b Mon Sep 17 00:00:00 2001 From: mischa Date: Tue, 26 May 2020 20:11:43 +0200 Subject: [PATCH] move remove.sh to remove.pl --- remove.pl | 163 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ remove.sh | 104 ---------------------------------- 2 files changed, 163 insertions(+), 104 deletions(-) create mode 100755 remove.pl delete mode 100755 remove.sh diff --git a/remove.pl b/remove.pl new file mode 100755 index 0000000..ae9f482 --- /dev/null +++ b/remove.pl @@ -0,0 +1,163 @@ +#!/usr/bin/env perl +# +# Copyright (c) 2020 Mischa Peters +# +# 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 remove script for OpenBSD Amsterdam +# 2020/05/25 version 2 - Perl again! :) +# +use 5.024; +use strict; +use warnings; +use autodie; +use Cwd qw(cwd); +use User::pwent; + +# Get vmXX.txt from arguments +if (! $ARGV[0]) { + print STDERR "Usage: $0 vmXX.txt\n"; + exit 1; +} +my $vmid = $ARGV[0]; + +# 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; +} + +# function to create accounts on the host for vmctl(8) access +sub remove_accounts { + my %conf = %{$_[0]}; + my %vms = %{$_[1]}; + + printf "userdel(8) removal:\n"; + for my $vm_name (sort keys %vms) { + my $_instance = $vms{$vm_name}{'instance'} || $vm_name; + my $_owner = $vms{$vm_name}{'owner'} || $vms{$vm_name}{'username'}; + my $_group = $conf{'conf'}{'VMDUSERS'}; + my $id = getpwnam("$_owner"); + + if ($id) { + qx(/usr/sbin/userdel -r $_owner); + printf "%16s %s account removed\n", $_instance, $_owner; + qx(/usr/sbin/groupdel $_owner); + printf "%16s %s group removed\n", $_instance, $_owner; + } + } +} + +# function to create the disk image files for vmm(4)/vmd(8) +sub backup_img_files { + my %conf = %{$_[0]}; + my %vms = %{$_[1]}; + + printf "vmm(4)/vmd(8) files:\n"; + for my $vm_name (sort keys %vms) { + my $_instance = $vms{$vm_name}{'instance'} || $vm_name; + my $_disk_format = $vms{$vm_name}{'format'} || $conf{'conf'}{'FORMAT'}; + my $_disk = $conf{'conf'}{'IMAGES'} . "/" . $_instance . "." . $_disk_format; + my $_disk2 = $conf{'conf'}{'IMAGES'} . "/" . $_instance . "_extra." . $_disk_format if $vms{$vm_name}{'disk2'}; + + if (-e $_disk) { + qx(mv $_disk $_disk-backup); + printf "%16s %s mv(1) %s-backup\n", $_instance, $_disk, $_disk; + } + if ($_disk2) { + if (-e $_disk2) { + qx(mv $_disk2 $_disk2-backup); + printf "%16s %s mv(1) $_disk2-backup\n", $_instance, $_disk2; + } + } + } +} + +sub backup_vm_files { + my %conf = %{$_[0]}; + my %vms = %{$_[1]}; + + printf "vmXX.txt files:\n"; + for my $vm_name (sort keys %vms) { + my $_instance = $vms{$vm_name}{'instance'} || $vm_name; + my $filename = $conf{'conf'}{'VMS'} . "/" . $vmid; + qx(cp $filename $filename-backup); + printf "%16s %s cp(1) %s-backup\n", $_instance, $filename, $filename; + qx(mv $filename $filename.free); + printf "%16s %s mv(1) %s.free\n", $_instance, $filename, $filename; + qx(chown -R mischa:mischa $filename-backup $filename.free); + } +} + +# function to print all keys & values for debug purposes +sub debug_parse { + my %conf = %{$_[0]}; + my %vms = %{$_[1]}; + + for my $vm_name (sort keys %vms) { + for my $key (keys %{$vms{$vm_name}}) { + printf "VMS: %s %s = %s\n", $vm_name, $key, $vms{$vm_name}{$key}; + } + } +} + +# check if _deploy.conf exists in current working directory +my %conf; +my $dir = cwd; +if (-e "$dir/_deploy.conf") { + %conf = get_variables('conf', "$dir/_deploy.conf"); +} else { + printf "Unable to find config file in current directory (%s).\n", $dir; + printf "Create the config file _deploy.conf in %s.\n", $dir; + exit 1; +} + +# parse all vm*.txt files in the VMS directory +my %vms; +my @files = glob "$conf{'conf'}{'VMS'}/$vmid"; +%vms = get_variables('vms', @files); + +# run all functions +#debug_parse(\%conf, \%vms); +remove_accounts(\%conf, \%vms); +backup_img_files(\%conf, \%vms); +backup_vm_files(\%conf, \%vms); diff --git a/remove.sh b/remove.sh deleted file mode 100755 index 8493419..0000000 --- a/remove.sh +++ /dev/null @@ -1,104 +0,0 @@ -#!/bin/sh -# -# Copyright (c) 2019 Mischa Peters -# -# 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. -# -main () { - CONF_FILE="$PWD/_deploy.conf" - [ -f "$CONF_FILE" ] && . "$CONF_FILE" - - file=${VMS}/$1 - - if [ -f "$file" ]; then - echo "reading $file" - . "$file" - backup_image "$file" - remove_user "$file" - move_file "$file" - else - echo "ERROR file doesn't exist: ${file}" - fi -} - -check_instance() { - # Check if the instance name exists, otherwise return filename as VM. - # Takes vm*.txt and instance - # prints either filename or instance variable - if test -z "$2" - then echo "$1" | sed "s@^$VMS@@;s@^/@@;s/\\.txt$//" - else echo "$2" - fi -} - -check_owner() { - # Check if the owner name exists, otherwise returns username. - # Takes username and owner - # prints either owner or username - if test -z "$2" - then echo "$1" - else echo "$2" - fi -} - -check_format() { - # Check if the image format exists, otherwise returns img. - # Takes format - # prints either format or img - if test -z "$1" - then echo "${FORMAT}" - else echo "$1" - fi -} - -backup_image() { - filename=$1 - _instance=$(check_instance "$filename" "$instance") - _format=$(check_format "$format") - if [ -f "${IMAGES}/${_instance}.${_format}" ]; then - mv ${IMAGES}/${_instance}.${_format} ${IMAGES}/${_instance}.${_format}-backup - echo "vmm(4)/vmd(8) files moved: ${IMAGES}/${_instance}.${_format} ${IMAGES}/${_instance}.${_format}-backup" - else - echo "ERROR vmm(4)/vmd(8) files moved: ${IMAGES}/${_instance}.${_format} doesn't exist" - fi -} - -remove_user() { - _owner=$(check_owner "$username" "$owner") - if [ -n "$_owner" ]; then - if grep -e "^${_owner}:" /etc/passwd > /dev/null; then - userdel -r "$_owner" - echo "userdel(8) removal: $_owner" - else - echo "ERROR userdel(8) removal: $_owner doesn't exist" - fi - - if grep -e "^${_owner}:" /etc/group > /dev/null; then - groupdel "$_owner" - echo "groupdel(8) removal: $_owner" - else - echo "ERROR groupdel(8) removal: $_owner doesn't exist" - fi - fi -} - -move_file() { - filename=$1 - cp ${filename} ${filename}-backup - echo "cp(1): ${filename} ${filename}-backup" - mv ${filename} ${filename}.free - echo "mv(1): ${filename} ${filename}.free" - chown -R mischa:mischa ${filename}-backup ${filename}.free -} - -main "$@"