Compare commits
3 Commits
962a06fa70
...
b51b314d74
Author | SHA1 | Date |
---|---|---|
dece | b51b314d74 | 2 years ago |
dece | 8c5bd53ba0 | 2 years ago |
dece | aa1ee1d0ad | 2 years ago |
@ -0,0 +1,51 @@
|
||||
#!/bin/bash
|
||||
# Share data to a running Shrlok instance.
|
||||
|
||||
socket_path="/run/shrlok/shr.sock"
|
||||
|
||||
usage() {
|
||||
echo "Usage: $0 [-s SOCKET_PATH] [-o key1=value1,key2=value2,…] TYPE FILE"
|
||||
echo " -s override shrlok socket path (default: $socket_path)"
|
||||
echo " -o comma separated "k=v" pairs to add to the shrlok header"
|
||||
}
|
||||
|
||||
header_args=
|
||||
while getopts "hs:o:" OPTION; do
|
||||
case $OPTION in
|
||||
h) usage; exit 0 ;;
|
||||
s) socket_path="$OPTARG" ;;
|
||||
o) readarray -d "," -t header_args <<< "$OPTARG" ;;
|
||||
*) usage; exit 1 ;;
|
||||
esac
|
||||
done
|
||||
shift $(( OPTIND - 1 ))
|
||||
[ $# != 2 ] && usage && exit 1
|
||||
|
||||
share_type="$1"
|
||||
input_file="$2"
|
||||
|
||||
if [ ! -S "$socket_path" ]; then
|
||||
echo "$socket_path is not a valid Unix socket. Is shrlok running?"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
header_args_str=""
|
||||
if [[ -n "$header_args" ]]; then
|
||||
for item in "${header_args[@]}"; do
|
||||
readarray -d "=" -t kv <<< "$item"
|
||||
key="${kv[0]}"
|
||||
val="$(echo "${kv[1]}" | xargs)" # strip trailing LFs
|
||||
header_args_str+=",\"$key\":\"$val\""
|
||||
done
|
||||
fi
|
||||
|
||||
# The first thing sent through the socket is the total packet length, as ASCII
|
||||
# digits for the convenience of scripts. Build the header before hand to get its
|
||||
# size, then get the file length and push everything to the socket with socat.
|
||||
header="{\"type\":\"$share_type\"$header_args_str}"
|
||||
header_length="${#header}"
|
||||
file_length="$(wc -c "$input_file" | cut -d' ' -f1)"
|
||||
full_length="$(( $header_length + 1 + $file_length ))"
|
||||
|
||||
cat <(printf "${full_length}\x00${header}\x00") "$input_file" \
|
||||
| socat - UNIX-CONNECT:"$socket_path" ; echo
|
Loading…
Reference in new issue