sheldonmlee/dotfiles
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove polkit privilege escalation and move old script to attach-usb-polkit

Browse Source
This commit is contained in:
Sheldon Lee 2024-05-17 15:52:58 +08:00
parent 41be8167b6
commit 2ff72c0c5c
2 changed files with 77 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
.config/scripts/libvirt/attach-usb
View File

@ -1,34 +1,12 @@
#!/bin/bash
script_location="$(readlink -f "$0")"
cd "$(dirname $script_location)"
if [ "$(id -u)" -ne "0" ]; then
pkttyagent --process $$ &
pkexec --keep-cwd sh -c "\
WAYLAND_DISPLAY=\"$WAYLAND_DISPLAY\"\
DISPLAY=\"$DISPLAY\"\
XDG_RUNTIME_DIR=\"$XDG_RUNTIME_DIR\"\
\"$0\""
exit 0
fi
run_as_user() {
if [ -z "$PKEXEC_UID" ]; then
USER_ID=1000
else
USER_ID="$PKEXEC_UID"
fi
username="$(id -nu "$USER_ID")"
cmd="$1"
echo "$(su "$username" sh -c "source \"/home/$username/.bash_profile\"; $cmd")"
}
menu() {
if [ -z "$DISPLAY" ]; then
echo -e "$1" | fzf --prompt="$2"
return 0
fi
run_as_user "echo -e '$1' | menucmd -p '$2'"
echo -e "$1" | menucmd -p "$2"
}
csv="$(virsh list | tail --line=+3 | tr ' ' '\n' | uniq | grep -v -e "^[[:space:]]*$" | tr '\n' ',')"

76
.config/scripts/libvirt/attach-usb-polkit Executable file
View File

@ -0,0 +1,76 @@
#!/bin/bash
script_location="$(readlink -f "$0")"
cd "$(dirname $script_location)"
if [ "$(id -u)" -ne "0" ]; then
pkttyagent --process $$ &
pkexec --keep-cwd sh -c "\
WAYLAND_DISPLAY=\"$WAYLAND_DISPLAY\"\
DISPLAY=\"$DISPLAY\"\
XDG_RUNTIME_DIR=\"$XDG_RUNTIME_DIR\"\
\"$0\""
exit 0
fi
run_as_user() {
if [ -z "$PKEXEC_UID" ]; then
USER_ID=1000
else
USER_ID="$PKEXEC_UID"
fi
username="$(id -nu "$USER_ID")"
cmd="$1"
echo "$(su "$username" sh -c "source \"/home/$username/.bash_profile\"; $cmd")"
}
menu() {
if [ -z "$DISPLAY" ]; then
echo -e "$1" | fzf --prompt="$2"
return 0
fi
run_as_user "echo -e '$1' | menucmd -p '$2'"
}
csv="$(virsh list | tail --line=+3 | tr ' ' '\n' | uniq | grep -v -e "^[[:space:]]*$" | tr '\n' ',')"
vms="$(echo "$csv" | awk -F ',' -v n='3' '{
for (i = 1; i+1 <= NF; i++) {
printf "%s ", $i
if (i % n == 0) {
printf "\n"
}
}
}')"
if [ -z "$vms" ]; then
menu "No VMs running" "Error"
exit 1
fi
vm="$(menu "$vms" "VM")"
vm_name=$(echo "$vm" | awk '{print $2}')
directory="./usb-devices"
item_list=""
for file in "$directory"/*; do
item_list="${item_list}$(basename $file)\n"
done
# strip last newline
end=${#item_list}-2
item_list=${item_list:0:end}
device="${directory}/$(menu "$item_list" "Device")"
[ -z "$device" ] && exit 1
output=$(virsh attach-device "$vm_name" "$device" 2>&1) && echo "Attached $device to $vm_name"
if [ $? -ne 0 ]; then
virsh detach-device "$vm_name" "$device" &> /dev/null && echo "Detached $device from $vm_name"
if [ $? -ne 0 ]; then
attached_vm_name="$(echo "$output" | grep domain | rev | awk '{print $1}' | rev)"
virsh detach-device "$attached_vm_name" "$device" &> /dev/null && echo "Detached $device from $attached_vm_name"
virsh attach-device "$vm_name" "$device" &> /dev/null && echo "Attached $device to $vm_name"
fi
fi