Merge branch 'master' of code.ungleich.ch:ungleich-public/ungleich-tools
This commit is contained in:
commit
619056a29e
5 changed files with 265 additions and 78 deletions
|
@ -2,26 +2,47 @@
|
||||||
|
|
||||||
# This script generates FreeBSD images for OpenNebula, being heavily inspired
|
# This script generates FreeBSD images for OpenNebula, being heavily inspired
|
||||||
# from srht's FreeBSD build image definition. It assumes running on a FreeBSD host.
|
# from srht's FreeBSD build image definition. It assumes running on a FreeBSD host.
|
||||||
|
# ZFS installation as documented by the FreeBSD project
|
||||||
|
# https://wiki.freebsd.org/RootOnZFS/GPTZFSBoot
|
||||||
|
|
||||||
set -e
|
set -e
|
||||||
set -x
|
set -x
|
||||||
|
|
||||||
# XXX: Handle command-line arguments?
|
# XXX: Handle command-line arguments?
|
||||||
RELEASE=13.0-RELEASE
|
RELEASE=13.1-RELEASE
|
||||||
ARCH=amd64
|
ARCH=amd64
|
||||||
IMAGE_PATH=freebsd-$RELEASE-$(date -I).img.qcow2
|
IMAGE_PATH_ZFS="freebsd-zfs-$RELEASE-$(date -I).img.qcow2"
|
||||||
|
IMAGE_PATH_UFS="freebsd-ufs-$RELEASE-$(date -I).img.qcow2"
|
||||||
IMAGE_SIZE=10G
|
IMAGE_SIZE=10G
|
||||||
|
|
||||||
DIST_BASE="https://download.freebsd.org/ftp/releases/$ARCH/$RELEASE"
|
# Comment out to simply use latest version
|
||||||
PORTS_BASE="https://download.freebsd.org/ftp/snapshots/$ARCH/12.1-STABLE"
|
# Hash checking is disabled when specifying this
|
||||||
|
#CLOUDSETUP_VERSION=1.2
|
||||||
|
|
||||||
ONE_CONTEXT_PKG_URL="https://github.com/OpenNebula/addon-context-linux/releases/download/v5.12.0/one-context-5.12.0_1.txz"
|
DIST_BASE="https://download.freebsd.org/ftp/releases/$ARCH/$RELEASE"
|
||||||
|
ZPOOL=zroot
|
||||||
|
ZPOOL_TMP="zinstalling"
|
||||||
|
|
||||||
|
ZFSTARGET="$(mktemp -d /var/tmp/zfsbuild.XXXXX)"
|
||||||
|
UFSTARGET="$(mktemp -d /var/tmp/ufsbuild.XXXXX)"
|
||||||
|
|
||||||
|
if zpool list -Ho name "$ZPOOL_TMP" 2>/dev/null; then
|
||||||
|
echo "The pool $ZPOOL_TMP is already imported." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
cleanup() {
|
cleanup() {
|
||||||
sync || true
|
sync ||:
|
||||||
umount /mnt/dev || true
|
umount "$UFSTARGET/dev" ||:
|
||||||
umount /mnt || true
|
umount "$UFSTARGET/tmp" ||:
|
||||||
mdconfig -du md0 || true
|
umount "$UFSTARGET/var/tmp" ||:
|
||||||
|
umount "$UFSTARGET" ||:
|
||||||
|
zpool export "$ZPOOL_TMP" ||:
|
||||||
|
mdconfig -du md0 ||:
|
||||||
|
mdconfig -du md1 ||:
|
||||||
|
rm -rf "$CLOUDSETUP_WORK" ||:
|
||||||
|
rmdir "$ZFSTARGET" ||:
|
||||||
|
rmdir "$UFSTARGET" ||:
|
||||||
}
|
}
|
||||||
trap cleanup EXIT
|
trap cleanup EXIT
|
||||||
|
|
||||||
|
@ -30,20 +51,89 @@ if [ "$(whoami)" != 'root' ]; then
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# Allocate and partition/format disk image.
|
if ! command -v rsync >/dev/null
|
||||||
disk=$(mktemp)
|
then
|
||||||
truncate -s 6G $disk
|
env ASSUME_ALWAYS_YES=YES pkg install -y rsync
|
||||||
mdconfig -a -t vnode -f $disk -u md0
|
fi
|
||||||
gpart create -s gpt /dev/md0
|
if ! command -v qemu-img >/dev/null
|
||||||
gpart add -t freebsd-boot -l bootfs -b 40 -s 512K md0
|
then
|
||||||
gpart bootcode -b /boot/pmbr -p /boot/gptboot -i 1 md0
|
env ASSUME_ALWAYS_YES=YES pkg install -y qemu-tools
|
||||||
gpart add -t freebsd-ufs -l rootfs -b 1M -s 5G md0
|
fi
|
||||||
newfs -U /dev/md0p2
|
|
||||||
|
portsnap fetch
|
||||||
|
if [ -f /usr/ports/README ]
|
||||||
|
then
|
||||||
|
portsnap update || portsnap extract
|
||||||
|
else
|
||||||
|
portsnap extract
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -n "$CLOUDSETUP_VERSION" ]
|
||||||
|
then
|
||||||
|
sed -i .bak -e '/^PORTVERSION=/ s/[0-9]*\.[0-9]*/'"$CLOUDSETUP_VERSION/" /usr/ports/sysutils/firstboot-cloudsetup/Makefile
|
||||||
|
make -C /usr/ports/sysutils/firstboot-cloudsetup makesum
|
||||||
|
fi
|
||||||
|
make -C /usr/ports/sysutils/firstboot-cloudsetup clean package
|
||||||
|
CLOUDSETUP_VERSION="$(fgrep VERSION /usr/ports/sysutils/firstboot-cloudsetup/Makefile | cut -f2- | tr -d \\t)"
|
||||||
|
CLOUDSETUP_PKG="/usr/ports/sysutils/firstboot-cloudsetup/work/pkg/firstboot-cloudsetup-${CLOUDSETUP_VERSION}.pkg"
|
||||||
|
tar -tzf "$CLOUDSETUP_PKG" >/dev/null # check that it's a valid tar, or we crash due to set -e
|
||||||
|
# tar -t lists the contents of a tar file, but does not extract
|
||||||
|
|
||||||
|
make -C /usr/ports/sysutils/firstboot-freebsd-update clean package
|
||||||
|
FBUPDATE_VERSION="$(fgrep VERSION /usr/ports/sysutils/firstboot-freebsd-update/Makefile | cut -f2- | tr -d \\t)"
|
||||||
|
FBUPDATE_PKG="/usr/ports/sysutils/firstboot-freebsd-update/work/pkg/firstboot-freebsd-update-${FBUPDATE_VERSION}.pkg"
|
||||||
|
tar -tzf "$FBUPDATE_PKG" >/dev/null # check that it's a valid tar, or we crash due to set -e
|
||||||
|
|
||||||
|
ufsdisk="$(mktemp /var/tmp/ufsdisk.XXXXX)"
|
||||||
|
truncate -s 6G "$ufsdisk"
|
||||||
|
mdconfig -a -t vnode -f "$ufsdisk" -u md1
|
||||||
|
gpart create -s gpt /dev/md1
|
||||||
|
#gpart add -t efi -l efiboot0 -s 260M md1
|
||||||
|
gpart add -t freebsd-boot -l gptboot -b 40 -s 512K md1
|
||||||
|
gpart bootcode -b /boot/pmbr -p /boot/gptboot -i 1 md1
|
||||||
|
gpart add -t freebsd-ufs -l rootfs -b 1M -s 5G md1
|
||||||
|
newfs -U /dev/md1p2
|
||||||
|
|
||||||
# Mount allocated image.
|
# Mount allocated image.
|
||||||
mount /dev/md0p2 /mnt
|
mount /dev/md1p2 "$UFSTARGET"
|
||||||
mkdir -p /mnt/dev
|
|
||||||
mount -t devfs devfs /mnt/dev
|
# Allocate and partition/format disk image.
|
||||||
|
# We use "legacy boot", aka BIOS boot
|
||||||
|
# Preferably, we'd use EFI boot here, check the FreeBSD wiki link in the header
|
||||||
|
# to see how to make that change, but make the EFI partition larger
|
||||||
|
zfsdisk="$(mktemp /var/tmp/zfsdisk.XXXXX)"
|
||||||
|
truncate -s 6G "$zfsdisk"
|
||||||
|
mdconfig -a -t vnode -f "$zfsdisk" -u md0
|
||||||
|
gpart create -s gpt /dev/md0
|
||||||
|
#gpart add -t efi -l efiboot0 -s 260M md1
|
||||||
|
gpart add -t freebsd-boot -l gptboot0 -b 40 -s 512K md0
|
||||||
|
gpart bootcode -b /boot/pmbr -p /boot/gptzfsboot -i 1 md0
|
||||||
|
gpart add -t freebsd-zfs -l zfs0 -b 1M -s 5G md0
|
||||||
|
zpool create -O compression=on -o ashift=12 -o "altroot=$ZFSTARGET" -m none -t "$ZPOOL_TMP" "$ZPOOL" md0p2
|
||||||
|
|
||||||
|
zfs create -o mountpoint=none "$ZPOOL_TMP/ROOT"
|
||||||
|
# We set zstd-19 so our image will become smaller, at the cost of a longer build time.
|
||||||
|
# At the end of the process, we disable zstd-19 again using zfs inherit compression,
|
||||||
|
# but all files already written will remain zstd-19 compressed
|
||||||
|
zfs create -o mountpoint=/ -o canmount=noauto "$ZPOOL_TMP/ROOT/default"
|
||||||
|
mount -t zfs "$ZPOOL_TMP/ROOT/default" "$ZFSTARGET"
|
||||||
|
zpool set "bootfs=$ZPOOL_TMP/ROOT/default" "$ZPOOL_TMP"
|
||||||
|
|
||||||
|
zfs create -o mountpoint=/tmp -o exec=on -o setuid=off "$ZPOOL_TMP/tmp"
|
||||||
|
zfs create -o canmount=off -o mountpoint=/usr "$ZPOOL_TMP/usr"
|
||||||
|
zfs create "$ZPOOL_TMP/usr/home"
|
||||||
|
zfs create -o exec=off -o setuid=off "$ZPOOL_TMP/usr/src"
|
||||||
|
zfs create -o mountpoint=/usr/ports -o setuid=off "$ZPOOL_TMP/usr/ports"
|
||||||
|
zfs create -o canmount=off -o mountpoint=/var "$ZPOOL_TMP/var"
|
||||||
|
zfs create -o exec=off -o setuid=off "$ZPOOL_TMP/var/audit"
|
||||||
|
zfs create -o exec=off -o setuid=off "$ZPOOL_TMP/var/crash"
|
||||||
|
zfs create -o exec=off -o setuid=off "$ZPOOL_TMP/var/log"
|
||||||
|
zfs create -o atime=on -o exec=off -o setuid=off "$ZPOOL_TMP/var/mail"
|
||||||
|
zfs create -o exec=on -o setuid=off "$ZPOOL_TMP/var/tmp"
|
||||||
|
|
||||||
|
ln -s /usr/home "$ZFSTARGET/home"
|
||||||
|
chmod 1777 "$ZFSTARGET/var/tmp"
|
||||||
|
chmod 1777 "$ZFSTARGET/tmp"
|
||||||
|
|
||||||
# Download and extract base system.
|
# Download and extract base system.
|
||||||
dist_files="kernel.txz base.txz"
|
dist_files="kernel.txz base.txz"
|
||||||
|
@ -53,64 +143,81 @@ mkdir -p "$dist_dir"
|
||||||
for f in $dist_files
|
for f in $dist_files
|
||||||
do
|
do
|
||||||
fetch -m -o "$dist_dir/$f" "$DIST_BASE/$f"
|
fetch -m -o "$dist_dir/$f" "$DIST_BASE/$f"
|
||||||
tar -C /mnt -xJf "$dist_dir/$f"
|
tar -C "$UFSTARGET" -xJf "$dist_dir/$f"
|
||||||
done
|
done
|
||||||
|
|
||||||
|
# Mount dev and tmp in chroot
|
||||||
|
mount -t devfs devfs "$UFSTARGET/dev"
|
||||||
|
mount_nullfs /tmp "$UFSTARGET/tmp"
|
||||||
|
mount_nullfs /var/tmp "$UFSTARGET/var/tmp"
|
||||||
|
|
||||||
|
# Install the first-boot script that configures the network and ssh key
|
||||||
|
# We must use --rootdir and not --chroot, because the file is read from within the chroot
|
||||||
|
# --automatic means that the package is considered to be installed "automatically",
|
||||||
|
# aka as a dependency of something, so pkg autoremove will remove it.
|
||||||
|
# We do not run pkg autoremove ourselves, that's up to the administrator.
|
||||||
|
pkg --rootdir "$UFSTARGET" add --automatic "$CLOUDSETUP_PKG" "$FBUPDATE_PKG"
|
||||||
|
|
||||||
# Configure new system.
|
# Configure new system.
|
||||||
echo "/dev/gpt/rootfs / ufs rw,noatime 1 1" >/mnt/etc/fstab
|
touch "$UFSTARGET/firstboot"
|
||||||
touch /mnt/firstboot
|
sysrc -f "$UFSTARGET/boot/loader.conf" \
|
||||||
echo 'autoboot_delay="-1"' >>/mnt/boot/loader.conf
|
zfs_load="YES" \
|
||||||
|
autoboot_delay="-1" \
|
||||||
|
|
||||||
cat >>/mnt/etc/rc.conf <<EOF
|
sysrc -f "$UFSTARGET/etc/rc.conf" \
|
||||||
ntpd_enable=YES
|
zfs_enable="YES" \
|
||||||
sshd_enable=YES
|
ntpd_enable="YES" \
|
||||||
growfs_enable=YES
|
sshd_enable="YES" \
|
||||||
hostname="freebsd"
|
growfs_enable="YES" \
|
||||||
EOF
|
hostname="freebsd" \
|
||||||
|
firstboot_cloudsetup_enable="YES" \
|
||||||
|
firstboot_freebsd_update_enable="YES" \
|
||||||
|
|
||||||
cp /etc/resolv.conf /mnt/etc/resolv.conf
|
# The resolv.conf file is written by firstboot_cloudsetup
|
||||||
tzsetup -s -C /mnt UTC
|
#cp /etc/resolv.conf "$UFSTARGET/etc/resolv.conf"
|
||||||
|
|
||||||
cat >>/mnt/etc/ssh/sshd_config <<EOF
|
tzsetup -s -C "$UFSTARGET" UTC
|
||||||
PermitRootLogin yes
|
|
||||||
PasswordAuthentication no
|
|
||||||
PermitEmptyPasswords no
|
|
||||||
EOF
|
|
||||||
|
|
||||||
mkdir -p /mnt/usr/local/etc/pkg/repos/
|
# Add PermitRootLogin without-password, unless PermitRootLogin yes was already set
|
||||||
cat >/mnt/usr/local/etc/pkg/repos/FreeBSD.conf <<EOF
|
sed -i .orig -e '/^#PermitRootLogin[[:blank:]]/a\
|
||||||
FreeBSD: {
|
PermitRootLogin without-password
|
||||||
url: pkg+http://pkg.FreeBSD.org/\$\{ABI\}/latest
|
' -e '/^PermitRootLogin[[:blank:]]*no/ s/\([[:blank:]]\).*$/\1without-password/' \
|
||||||
enabled: yes
|
"$UFSTARGET/etc/ssh/sshd_config"
|
||||||
}
|
if ! grep -Eq '^PermitRootLogin (without-password|yes)' "$UFSTARGET/etc/ssh/sshd_config"
|
||||||
EOF
|
|
||||||
|
|
||||||
# freebsd-update is only supported for RELEASE
|
|
||||||
if [ "${release%-RELEASE}" != "$RELEASE" ]
|
|
||||||
then
|
then
|
||||||
env PAGER=true /usr/sbin/freebsd-update \
|
cat >>"$UFSTARGET/etc/ssh/sshd_config" <<EOF
|
||||||
-b /mnt \
|
|
||||||
--currently-running "$RELEASE" \
|
# Added by Ungleich
|
||||||
--not-running-from-cron -F \
|
PermitRootLogin without-password
|
||||||
fetch install
|
EOF
|
||||||
fi
|
fi
|
||||||
|
|
||||||
env ASSUME_ALWAYS_YES=YES pkg -c /mnt bootstrap -f
|
# Set zstd-19 compression, copy all data to the pool, and then set compression to default again
|
||||||
env ASSUME_ALWAYS_YES=YES pkg -c /mnt install bash curl
|
# This will make the base image smaller, at the cost of taking longer to generate, as zstd-19 is slow to write
|
||||||
|
# Therefore, afterwards we restore compression to default, so written files stay zstd-19, which is fast to read,
|
||||||
|
# but files written by the user afterwards will be written with the default compression algorihtm.
|
||||||
|
zfs set compression=zstd-19 "$ZPOOL_TMP/ROOT/default"
|
||||||
|
umount "$UFSTARGET/dev" "$UFSTARGET/tmp" "$UFSTARGET/var/tmp"
|
||||||
|
rsync -aH --fileflags --inplace "$UFSTARGET/." "$ZFSTARGET"
|
||||||
|
|
||||||
curl -L "$ONE_CONTEXT_PKG_URL" -o /mnt/one-context.txz
|
sysrc -f "$UFSTARGET/boot/loader.conf" -x zfs_load
|
||||||
env ASSUME_ALWAYS_YES=YES pkg -c /mnt add one-context.txz
|
sysrc -f "$UFSTARGET/etc/rc.conf" -x zfs_enable
|
||||||
rm /mnt/one-context.txz
|
printf '# Device\tMountpoint\tFStype\tOptions\t\tDump\tPass#\n' \
|
||||||
|
>"$ZFSTARGET/etc/fstab"
|
||||||
|
printf '# Device\tMountpoint\tFStype\tOptions\t\tDump\tPass#\n%s\t%s\t\t%s\t%s\t%s\t%s\n' \
|
||||||
|
/dev/gpt/rootfs / ufs rw,noatime 1 1 \
|
||||||
|
>"$UFSTARGET/etc/fstab"
|
||||||
|
sync ||:
|
||||||
|
zfs inherit compression "$ZPOOL_TMP/ROOT/default"
|
||||||
|
|
||||||
fetch -m -o "$dist_dir/ports.txz" "$PORTS_BASE/ports.txz"
|
|
||||||
tar -C /mnt -xJf "$dist_dir/ports.txz"
|
|
||||||
|
|
||||||
cleanup
|
|
||||||
trap : EXIT
|
trap : EXIT
|
||||||
|
cleanup
|
||||||
|
|
||||||
mkdir -p "$ARCH"
|
mkdir -p "$ARCH"
|
||||||
qemu-img convert -f raw -O qcow2 "$disk" "$IMAGE_PATH"
|
qemu-img convert -f raw -O qcow2 "$zfsdisk" "$ARCH/$IMAGE_PATH_ZFS"
|
||||||
rm "$disk"
|
qemu-img convert -f raw -O qcow2 "$ufsdisk" "$ARCH/$IMAGE_PATH_UFS"
|
||||||
|
rm "$zfsdisk" "$ufsdisk"
|
||||||
|
|
||||||
# Filesystem will be enlarged by growfs(7) on next startup
|
# Filesystem will be enlarged by growfs(7) on next startup
|
||||||
qemu-img resize "$IMAGE_PATH" "$IMAGE_SIZE"
|
qemu-img resize "$ARCH/$IMAGE_PATH_ZFS" "$IMAGE_SIZE"
|
||||||
|
qemu-img resize "$ARCH/$IMAGE_PATH_UFS" "$IMAGE_SIZE"
|
||||||
|
|
|
@ -9,14 +9,14 @@ set -e
|
||||||
set -x
|
set -x
|
||||||
|
|
||||||
# XXX: Handle command-line arguments?
|
# XXX: Handle command-line arguments?
|
||||||
RELEASE=groovy # 20.10
|
RELEASE=jammy # 22.04 LTS
|
||||||
ARCH=amd64
|
ARCH=amd64
|
||||||
IMAGE_PATH=ubuntu-$RELEASE-$(date --iso-8601).img.qcow2
|
IMAGE_PATH=ubuntu-$RELEASE-$(date --iso-8601).img.qcow2
|
||||||
IMAGE_SIZE=10G
|
IMAGE_SIZE=10G
|
||||||
NBD_DEVICE=/dev/nbd0
|
NBD_DEVICE=/dev/nbd0
|
||||||
|
|
||||||
# TODO: find the package definition and built ourself, publish in some RPM repository.
|
# TODO: find the package definition and built ourself, publish in some RPM repository.
|
||||||
ONE_CONTEXT_DEB_URL="https://github.com/OpenNebula/addon-context-linux/releases/download/v5.10.0/one-context_5.10.0-1.deb"
|
ONE_CONTEXT_DEB_URL="https://github.com/OpenNebula/addon-context-linux/releases/download/v6.4.0/one-context_6.4.0-1.deb"
|
||||||
ONE_CONTEXT_DEB_PATH=/root/one-context.deb
|
ONE_CONTEXT_DEB_PATH=/root/one-context.deb
|
||||||
|
|
||||||
cleanup() {
|
cleanup() {
|
||||||
|
|
|
@ -24,10 +24,10 @@ public_key=$(echo $private_key | wg pubkey)
|
||||||
|
|
||||||
case $my_wireguard_ip in
|
case $my_wireguard_ip in
|
||||||
185.155.29.*)
|
185.155.29.*)
|
||||||
vpn_endpoint_pubkey=6BRnQ+dmeFzVCH9RbM1pbJ7u3y3qrl+zUzzYCmC88kE=
|
vpn_endpoint_pubkey="6BRnQ+dmeFzVCH9RbM1pbJ7u3y3qrl+zUzzYCmC88kE="
|
||||||
;;
|
;;
|
||||||
185.155.30.*)
|
185.155.30.*)
|
||||||
vpn_endpoint_pubkey=5ach7pUQ57aa402LHz1MYh7lyBZS0GvBEw2PC6dMHW4=
|
vpn_endpoint_pubkey="5ach7pUQ57aa402LHz1MYh7lyBZS0GvBEw2PC6dMHW4="
|
||||||
;;
|
;;
|
||||||
*)
|
*)
|
||||||
echo "Unknown VPN host for IP $my_wireguard_ip" >&2
|
echo "Unknown VPN host for IP $my_wireguard_ip" >&2
|
||||||
|
@ -77,4 +77,12 @@ uci commit
|
||||||
|
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
echo "Host ${my_ip} uses ip ${my_wireguard_ip} with public key ${public_key}"
|
echo "Host ${my_ip} uses ip ${my_wireguard_ip} with public key ${public_key}:"
|
||||||
|
|
||||||
|
cat <<EOF
|
||||||
|
|
||||||
|
#
|
||||||
|
[Peer]
|
||||||
|
PublicKey = ${public_key}
|
||||||
|
AllowedIPs = ${my_wireguard_ip}/32
|
||||||
|
EOF
|
||||||
|
|
|
@ -1,5 +1,14 @@
|
||||||
import psycopg2 as pg2
|
import psycopg2 as pg2
|
||||||
from config import config
|
from config import config
|
||||||
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
logger.setLevel(logging.INFO)
|
||||||
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||||||
|
file_handler = logging.FileHandler('/var/log/desktop.log')
|
||||||
|
file_handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(file_handler)
|
||||||
|
|
||||||
|
|
||||||
db_name = config['db']['db_name']
|
db_name = config['db']['db_name']
|
||||||
db_user = config['db']['db_user']
|
db_user = config['db']['db_user']
|
||||||
|
@ -43,6 +52,8 @@ def setconn(u_id, vm_num, vm_port,vm_host):
|
||||||
#clipboard-encoding
|
#clipboard-encoding
|
||||||
cur.execute("INSERT INTO guacamole_connection_parameter VALUES ('{}','clipboard-encoding','UTF-8')".format(cn_id))
|
cur.execute("INSERT INTO guacamole_connection_parameter VALUES ('{}','clipboard-encoding','UTF-8')".format(cn_id))
|
||||||
print("create connection")
|
print("create connection")
|
||||||
|
log = "create connection : " + cn
|
||||||
|
logging.info(log)
|
||||||
else:
|
else:
|
||||||
cur.execute("SELECT MAX(connection_id) FROM guacamole_connection WHERE connection_name = '{}' AND parent_id IS NULL".format(cn))
|
cur.execute("SELECT MAX(connection_id) FROM guacamole_connection WHERE connection_name = '{}' AND parent_id IS NULL".format(cn))
|
||||||
temp_cn_id = cur.fetchone()
|
temp_cn_id = cur.fetchone()
|
||||||
|
@ -53,3 +64,29 @@ def setconn(u_id, vm_num, vm_port,vm_host):
|
||||||
print("no connection")
|
print("no connection")
|
||||||
conn.close()
|
conn.close()
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def delconn(u_id, vm_num):
|
||||||
|
conn2 = pg2.connect("host = localhost dbname={} user={} password={} port={}".format(db_name,db_user,db_password,db_port))
|
||||||
|
conn2.autocommit = True
|
||||||
|
cur2 = conn2.cursor()
|
||||||
|
cur2.execute("SELECT entity_id FROM guacamole_entity WHERE name = '{}'".format(u_id))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 == None:
|
||||||
|
print("no user : " , u_id)
|
||||||
|
else:
|
||||||
|
cn2 = "{}{}".format(u_id,vm_num)
|
||||||
|
cur2.execute("SELECT connection_id FROM guacamole_connection WHERE connection_name = '{}'".format(cn2))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 != None:
|
||||||
|
print("cn_id : ", row2[0])
|
||||||
|
#delete connection
|
||||||
|
cur2.execute("SELECT connection_id from guacamole_connection_permission where connection_id = '{}'".format(row2[0]))
|
||||||
|
row2 = cur2.fetchone()
|
||||||
|
if row2 != None:
|
||||||
|
print("delete connection : ",row2[0])
|
||||||
|
cur2.execute("delete from guacamole_connection_permission where connection_id = '{}'".format(row2[0]))
|
||||||
|
log = "delete connection : " + cn2
|
||||||
|
logging.info(log)
|
||||||
|
conn2.close()
|
||||||
|
|
|
@ -6,6 +6,7 @@ from xmltodict import parse
|
||||||
from config import config
|
from config import config
|
||||||
from ldap_list import vm_list
|
from ldap_list import vm_list
|
||||||
from db_export import setconn
|
from db_export import setconn
|
||||||
|
from db_export import delconn
|
||||||
|
|
||||||
# Constants
|
# Constants
|
||||||
ALL_VM_STATES = -1
|
ALL_VM_STATES = -1
|
||||||
|
@ -61,6 +62,22 @@ class VM:
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
class tVM:
|
||||||
|
def __init__(self, tvm: dict):
|
||||||
|
self.id = vm.get('ID', None)
|
||||||
|
self.owner = {
|
||||||
|
'id': vm.get('UID', None),
|
||||||
|
'name': vm.get('UNAME', None),
|
||||||
|
'gname': vm.get('GNAME', None)
|
||||||
|
}
|
||||||
|
self.name = vm.get('NAME', None)
|
||||||
|
self.status = vm.get('STATE', None)
|
||||||
|
if self.status:
|
||||||
|
self.status = VMState(int(self.status)).name.lower()
|
||||||
|
|
||||||
|
template = vm['TEMPLATE']
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
with RPCClient(opnserver) as rpc_client:
|
with RPCClient(opnserver) as rpc_client:
|
||||||
success, response, *_ = rpc_client.one.vmpool.infoextended(
|
success, response, *_ = rpc_client.one.vmpool.infoextended(
|
||||||
|
@ -76,13 +93,31 @@ def main():
|
||||||
vm_port = vm['TEMPLATE']['GRAPHICS'].get('PORT')
|
vm_port = vm['TEMPLATE']['GRAPHICS'].get('PORT')
|
||||||
vm_host = vm['HISTORY_RECORDS']['HISTORY']['HOSTNAME']
|
vm_host = vm['HISTORY_RECORDS']['HISTORY']['HOSTNAME']
|
||||||
if vm['UNAME'] == temp_uname:
|
if vm['UNAME'] == temp_uname:
|
||||||
#print(entry.uid, vm_id, vm_port, vm_host)
|
print(entry.uid, vm_id, vm_port, vm_host)
|
||||||
setconn(entry.uid, vm_id, vm_port, vm_host)
|
setconn(entry.uid, vm_id, vm_port, vm_host)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
print(response)
|
print(response)
|
||||||
|
|
||||||
|
with RPCClient(opnserver) as rpc_client2:
|
||||||
|
success, response, *_ = rpc_client2.one.vmpool.infoextended(
|
||||||
|
session_string , VmFilterFlag.AllResources.value, START_ID, END_ID, VMState.DONE.value
|
||||||
|
)
|
||||||
|
if success:
|
||||||
|
vms2 = json.loads(json.dumps(parse(response)))['VM_POOL']['VM']
|
||||||
|
for entry in vm_list.entries:
|
||||||
|
temp_uname = entry.uid
|
||||||
|
for i, tvm in enumerate(vms2):
|
||||||
|
vm_user = tvm['UNAME']
|
||||||
|
vm_id = tvm['ID']
|
||||||
|
if tvm['UNAME'] == temp_uname:
|
||||||
|
print("terminated VM : ", entry.uid, vm_id)
|
||||||
|
delconn(entry.uid, vm_id)
|
||||||
|
|
||||||
|
else:
|
||||||
|
print(response)
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue