summary refs log tree commit diff
path: root/Functions/TCP/tcp_output
blob: b22b794121bec00b6ac0621720a08cb43060cdea (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
emulate -L zsh
setopt extendedglob

local opt tprompt sess read_fd tpat quiet

while getopts "F:P:qS:" opt; do
  case $opt in
    (F) read_fd=$OPTARG
	;;
    (P) tprompt=$OPTARG
	;;
    (q) quiet=1
	;;
    (S) sess=$OPTARG
	;;
    (*) [[ $opt != \? ]] && print -r "Can't handle option $opt" >&2
	return 1
	;;
  esac
done
(( OPTIND > 1 )) && shift $(( OPTIND - 1 ))

# Per-session logs don't have the session discriminator in front.
if [[ -n $TCP_LOG_SESS ]]; then
  print -r -- "$*" >>${TCP_LOG_SESS}.$sess
fi
# Always add the TCP prompt.  We used only to do this with
# multiple sessions, but it seems always to be useful to know
# where data is coming from; also, it allows more predictable
# behaviour in tcp_expect.
if [[ -n $tprompt ]]; then
  zgprintf -R -%s=$sess -%f=$read_fd -- $tprompt
  # We will pass this back up.
  REPLY="$REPLY$*"
else
  REPLY="$*"
fi
if [[ -n $TCP_LOG ]]; then
  print -r -- $REPLY >>${TCP_LOG}
fi

if [[ -z $quiet ]]; then
  local skip=
  if [[ ${#tcp_filter} -ne 0 ]]; then
    # Allow tcp_filter to be an associative array, though
    # it doesn't *need* to be.
    for tpat in ${(v)tcp_filter}; do
      [[ $REPLY = ${~tpat} ]] && skip=1 && break
    done
  fi
  if [[ -z $skip ]]; then
    # Check flag passed down probably from tcp_fd_handler:
    # if we have output, we are in zle and need to fix the display first.
    # (The shell is supposed to be smart enough that you can replace
    # all the following with
    #   [[ -o zle ]] && zle -I
    # but I haven't dared try it yet.)
    if [[ -n $TCP_INVALIDATE_ZLE ]]; then
      zle -I
      # Only do this the first time.
      unset TCP_INVALIDATE_ZLE
    fi
    print -r -- $REPLY
  fi
fi