vm flag and wip vsh rework

Signed-off-by: Ava Hahn <a.hahn@f5.com>
This commit is contained in:
Ava Hahn 2025-01-21 14:25:05 -08:00
parent 8c91778fe1
commit b47476bd69
5 changed files with 266 additions and 195 deletions

138
virt.sh
View file

@ -23,53 +23,44 @@ if [[ ! $PASSP ]]; then
fi
function vms_avail() {
ret=$(sudo virsh list --all --name | sort)
sudo virsh list --all --name | sort
}
function vms_on() {
ret=$(sudo virsh list --name | sort)
sudo virsh list --name | sort
}
function vms_off() {
ret=$(sudo virsh list --inactive --name | sort)
sudo virsh list --inactive --name | sort
}
function get_vm_ip() {
vms_avail
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_avail) =~ $1 ]]; then
log "VM $1 doesnt exist"
ret=""
return 1
fi
vms_on
if ! [[ $ret =~ $1 ]]; then
log "VM $1 already off"
ret=""
if ! [[ $(vms_on) =~ $1 ]]; then
log "VM $1 is off"
return 1
fi
ret=$(sudo virsh net-dhcp-leases default | grep $1 | awk '{print $5}' | rev | cut -c 4- | rev)
sudo virsh net-dhcp-leases default | grep $1 | awk '{print $5}' | rev | cut -c 4- | rev
}
function turn_on_vm_and_wait() {
if [[ ! $1 ]]; then
log "no VM specified"
ret=""
return 1
fi
vms_avail
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_avail) =~ $1 ]]; then
log "VM $1 doesnt exist"
ret=""
return 1
fi
vms_off
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_off) =~ $1 ]]; then
log "VM $1 already on"
ret=""
return 0
fi
@ -78,9 +69,7 @@ function turn_on_vm_and_wait() {
# wait for an IP
ret=""
while [[ "$ret" == "" ]]; do
get_vm_ip $1
while ! [[ $(get_vm_ip $1) ]]; do
sleep 0.5
done
log "Got IP for VM $1"
@ -89,121 +78,82 @@ function turn_on_vm_and_wait() {
# ret set by get_vm_ip above
while ! sshpass -p $PASSP \
ssh -o PreferredAuthentications=password \
-o StrictHostKeyChecking=no $USERN@$ret \
exit; do
sleep 0.1
done
log "Got SSH on VM $1"
}
function vm_shell() {
vms_avail
if ! [[ $ret =~ $1 ]]; then
log "VM $1 doesnt exist"
ret=""
return 1
fi
vms_on
if ! [[ $ret =~ $1 ]]; then
log "VM $1 is off"
ret=""
return 1
fi
if [[ "$2" == "" ]]; then
log "wont execute empty command"
ret=""
return 1
fi
# wait for an IP
ret=""
while [[ "$ret" == "" ]]; do
get_vm_ip $1
sleep 0.5
-o StrictHostKeyChecking=no $USERN@$(get_vm_ip $1) \
exit; do
sleep 0.1
done
# wait for successful ssh
# ret set by get_vm_ip above
sshpass -p $PASSP \
ssh -o PreferredAuthentications=password \
-o StrictHostKeyChecking=no \
$USERN@$ret "$2"
return $?
log "Got SSH on VM $1"
}
function turn_off_vm() {
if [[ ! $1 ]]; then
log "no VM specified"
ret=""
return 1
fi
vms_avail
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_avail) =~ $1 ]]; then
log "VM $1 doesnt exist"
ret=""
return 1
fi
vms_on
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_on) =~ $1 ]]; then
log "VM $1 already off"
ret=""
return 1
return 0
fi
ret=$(sudo virsh shutdown $1)
sudo virsh shutdown $1
}
function sync_dir_to_vm(){
if [ ! -d $2 ]; then
if [[ ! -d $2 ]]; then
log "directory $2 does not exist."
ret=""
return 1
fi
vms_avail
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_avail) =~ $1 ]]; then
log "VM $1 doesnt exist"
ret=""
return 1
fi
vms_on
if ! [[ $ret =~ $1 ]]; then
if ! [[ $(vms_on) =~ $1 ]]; then
log "VM $1 is off"
ret=""
return 1
fi
# wait for an IP
ret=""
while [[ "$ret" == "" ]]; do
get_vm_ip $1
log "waiting for ip"
while [[ ! $(get_vm_ip $1) ]]; do
sleep 0.5
done
sshpass -p $PASSP \
rsync -avze "ssh -o PreferredAuthentications=password -o StrictHostKeyChecking=no" \
$2 $USERN@$ret:
return $?
$2 $USERN@$(get_vm_ip $1):
}
# boots vm, enters SSH, shuts down VM
function vsh() {
# takes care of error cases
if ! turn_on_vm_and_wait $1; then
exit 1
function vm_cmd() {
if ! [[ $(vms_avail) =~ $1 ]]; then
log "VM $1 doesnt exist"
return 1
fi
get_vm_ip $1
if ! [[ $(vms_on) =~ $1 ]]; then
log "VM $1 is off"
return 1
fi
if [[ ! $2 ]]; then
log "wont execute empty command"
return 1
fi
# wait for an IP
while [[ ! $(get_vm_ip $1) ]]; do
sleep 0.5
done
# wait for successful ssh
sshpass -p $PASSP \
ssh -o PreferredAuthentications=password \
-o StrictHostKeyChecking=no \
$USERN@$ret
turn_off_vm $1
$USERN@$(get_vm_ip $1) "$2"
}