mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-08 21:25:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify data transmission

Browse Source
This commit is contained in:
Kovid Goyal
2022-02-24 08:32:30 +05:30
parent 3c28a4f723
commit a71e7d7eb1
3 changed files with 25 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
shell-integration/ssh/bootstrap.sh
View File

@@ -3,18 +3,17 @@
# Distributed under terms of the GPLv3 license.
# read the transmitted data from STDIN
saved_tty_settings=$(command stty -g)
command stty raw -echo
encoded_data_file=$(mktemp)
cleanup_on_bootstrap_exit() {
[ ! -z "$encoded_data_file" ] && command rm -f "$encoded_data_file"
[ ! -z "$saved_tty_settings" ] && command stty "$saved_tty_settings"
}
trap 'cleanup_on_bootstrap_exit' EXIT
die() { echo "$*" >/dev/stderr; cleanup_on_bootstrap_exit; exit 1; }
debug() { printf "\033P@kitty-print|%s\033\\" "$(printf "%s" "debug: $1" | base64)"; }
saved_tty_settings=$(command stty -g)
command stty raw min 1 time 0 || die "stty not available"
trap 'cleanup_on_bootstrap_exit' EXIT
data_started="n"
data_complete="n"
pending_data=""
@@ -34,38 +33,25 @@ data_password="DATA_PASSWORD"
password_filename="PASSWORD_FILENAME"
pending_data=""
data_complete="n"
printf "\033P@kitty-ssh|%s:%s:%s\033\\" "$hostname" "$password_filename" "$data_password"
while [ "$data_complete" = "n" ]; do
IFS= read -r line || die "Incomplete ssh data";
case "$line" in
*"KITTY_SSH_DATA_START")
prefix=$(command expr "$line" : "\(.*\)KITTY_SSH_DATA_START")
pending_data="$pending_data$prefix"
data_started="y";
;;
"KITTY_SSH_DATA_END")
data_complete="y";
;;
*)
if [ "$data_started" = "y" ]; then
printf "%s" "$line" >> "$encoded_data_file"
else
pending_data="$pending_data$line\n"
fi
;;
esac
done
printf "\033P@kitty-ssh|%s:%s:%s\033\\" "$hostname" "$password_filename" "$data_password"
IFS= read -r size || die "Failed to read data size";
case "$size" in
" "*)
die "$size";
;;
"")
die "Got empty size";
;;
esac
command head -c "$size" < /dev/stdin | command base64 -d | command tar xjf - --no-same-owner -C "$HOME"
rc="$?"
command stty "$saved_tty_settings"
saved_tty_settings=""
if [ "$rc" != "0" ]; then die "Failed to extract data transmitted by ssh kitten over the TTY device"; fi
if [ -n "$pending_data" ]; then
printf "\033P@kitty-echo|%s\033\\" "$(printf "%s" "$pending_data" | base64)"
fi
command base64 -d < "$encoded_data_file" | command tar xjf - --no-same-owner -C "$HOME"
rc=$?
command rm -f "$encoded_data_file"
encoded_data_file=""
if [ "$rc" != "0" ]; then die "Failed to extract data transmitted by ssh kitten over the TTY device"; fi
if [ ! -f "$HOME/.terminfo/kitty.terminfo" ]; then die "Extracted data transmitted by ssh kitten is incomplete"; fi
# export TERMINFO