home/lib/login_functions.sh

   1 #
   2
   3 ## open : string -> unit
   4 ##
   5 ## Fork xdg-open so we don't block current terminal session when opening
   6 ## things like pdf files. For example:
   7 ##
   8 ##    open book.pdf
   9 ##
  10 open() {
  11     (xdg-open "$1" &) &
  12 }
  13
  14 ## notify_done : unit -> unit
  15 notify_done() {
  16     local -r _status_code="$?"
  17     local -r _program="$1"
  18     local _timestamp
  19     _timestamp="$(timestamp)"
  20     local -r _msg="$_timestamp [$_program] done "
  21     if [[ "$_status_code" -eq 0 ]]
  22     then
  23         notify-send -u normal "$_msg OK: $_status_code"
  24     else
  25         notify-send -u critical "$_msg ERROR: $_status_code"
  26     fi
  27 }
  28
  29 ## p : string -> unit
  30 p() {
  31     awk \
  32         -v _s="$1" \
  33         '
  34             BEGIN {_s = tolower(_s)}
  35
  36             # TODO fzf/dmenu select instead of searching:
  37             /^[a-zA-Z]/ && tolower($1) ~ _s && NF >= 2 {
  38                 n++
  39                 s = $1
  40                 p = $NF
  41                 if (NF == 2) {
  42                     e = ""
  43                     u = ""
  44                 } else if (NF == 3) {
  45                     e = $2
  46                     u = ""
  47                 } else {
  48                     e = $2
  49                     u = $3
  50                 } # TODO What would NF > 4 mean?
  51
  52                 printf("%d [O] s:\"%s\", e:\"%s\", u:\"%s\"\n", n, s, e, u) > "/dev/stderr"
  53                 if (match(u, "@")) {
  54                     tmp = e
  55                     e = u
  56                     u = tmp
  57                     printf("%d [C] s:\"%s\", e:\"%s\", u:\"%s\"\n", n, s, e, u) > "/dev/stderr"
  58                 }
  59
  60                 printf "%s", p # XXX Intentionally avoiding newline in the result.
  61             }
  62
  63             END {
  64                 if (n == 1) {
  65                     exit 0
  66                 } else if (n == 0) {
  67                     printf "[ERROR] Found nothing.\n" > "/dev/stderr"
  68                     exit 1
  69                 } else if (n > 1) {
  70                     # TODO fzf-select which of the records the user (ahem, me) wants.
  71                     printf "[WARNING] Found more than one record.\n" > "/dev/stderr"
  72                     exit 0
  73                 }
  74             }
  75         ' \
  76         ~/._p/p \
  77         | xsel -i -b -t 30000
  78 }
  79
  80 ## web search
  81 ## ws : string -> unit
  82 ws() {
  83     local line search_string0 search_string
  84
  85     search_string0="$*"
  86     case "$search_string0" in
  87         '')
  88             while read -r line; do
  89                 search_string="${search_string} ${line}"
  90             done;;
  91          *)
  92             search_string="$search_string0";;
  93     esac
  94
  95     firefox --search "$search_string"
  96 }
  97
  98
  99 ## dictionary
 100 ## d : string -> string list
 101 d() {
 102     local -r word=$(fzf < /usr/share/dict/words)
 103     dict "$word"
 104 }
 105
 106 ## shell_activity_report : (mon | dow) -> string list
 107 shell_activity_report() {
 108     # TODO: optional concrete number output
 109     # TODO: optional combinations of granularities: hour, weekday, month, year
 110     local group_by="$1"
 111     case "$group_by" in
 112         'mon') ;;
 113         'dow') ;;
 114         '') group_by='dow';;
 115         *)
 116             echo "Usage: $0 [mon|dow]" >&2
 117             kill -INT $$
 118     esac
 119     history \
 120     | awk -v group_by="$group_by" '
 121         function date2dow(y, m, d,    _t, _i) {
 122             # Contract:
 123             #   y > 1752,  1 <= m <= 12.
 124             # Source:
 125             #   Sakamoto`s methods
 126             #   https://en.wikipedia.org/wiki/Determination_of_the_day_of_the_week#Sakamoto%27s_methods
 127             _t[ 0] = 0
 128             _t[ 1] = 3
 129             _t[ 2] = 2
 130             _t[ 3] = 5
 131             _t[ 4] = 0
 132             _t[ 5] = 3
 133             _t[ 6] = 5
 134             _t[ 7] = 1
 135             _t[ 8] = 4
 136             _t[ 9] = 6
 137             _t[10] = 2
 138             _t[11] = 4
 139             y -= m < 3
 140             _i = int(y + y/4 - y/100 + y/400 + _t[m - 1] + d) % 7
 141             _i = _i == 0 ? 7 : _i  # Make Sunday last
 142             return _i
 143
 144         }
 145
 146         {
 147             # NOTE: $2 & $3 are specific to oh-my-zsh history output
 148             date = $2
 149             time = $3
 150             d_fields = split(date, d, "-")
 151             t_fields = split(time, t, ":")
 152             if (t_fields && d_fields) {
 153                 # +0 to coerce number from string
 154                 year  = d[1] + 0
 155                 month = d[2] + 0
 156                 day   = d[3] + 0
 157                 hour = t[1] + 0
 158                 dow = date2dow(year, month, day)
 159                 g = group_by == "mon" ? month : dow  # dow is default
 160                 c = count[g, hour]++
 161             }
 162             if (c > max)
 163                 max = c
 164         }
 165
 166         END {
 167             w[1] = "Monday"
 168             w[2] = "Tuesday"
 169             w[3] = "Wednesday"
 170             w[4] = "Thursday"
 171             w[5] = "Friday"
 172             w[6] = "Saturday"
 173             w[7] = "Sunday"
 174
 175             m[ 1] = "January"
 176             m[ 2] = "February"
 177             m[ 3] = "March"
 178             m[ 4] = "April"
 179             m[ 5] = "May"
 180             m[ 6] = "June"
 181             m[ 7] = "July"
 182             m[ 8] = "August"
 183             m[ 9] = "September"
 184             m[10] = "October"
 185             m[11] = "November"
 186             m[12] = "December"
 187
 188             n = group_by == "mon" ? 12 : 7  # dow is default
 189
 190             for (gid = 1; gid <= n; gid++) {
 191                 group = group_by == "mon" ? m[gid] : w[gid]
 192                 printf "%s\n", group;
 193                 for (hour=0; hour<24; hour++) {
 194                     c = count[gid, hour]
 195                     printf "  %2d ", hour
 196                     for (i = 1; i <= (c * 100) / max; i++)
 197                         printf "|"
 198                     printf "\n"
 199                 }
 200             }
 201         }'
 202 }
 203
 204 ## top_commands : unit -> (command:string * count:number * bar:string) list
 205 top_commands() {
 206     history \
 207     | awk '
 208         {
 209             count[$4]++
 210         }
 211
 212         END {
 213             for (cmd in count)
 214                 print count[cmd], cmd
 215         }' \
 216     | sort -n -r -k 1 \
 217     | head -50 \
 218     | awk '
 219         {
 220             cmd[NR] = $2
 221             c = count[NR] = $1 + 0  # + 0 to coerce number from string
 222             if (c > max)
 223                 max = c
 224         }
 225
 226         END {
 227             for (i = 1; i <= NR; i++) {
 228                 c = count[i]
 229                 printf "%s %d ", cmd[i], c
 230                 scaled = (c * 100) / max
 231                 for (j = 1; j <= scaled; j++)
 232                     printf "|"
 233                 printf "\n"
 234             }
 235         }' \
 236     | column -t
 237 }
 238
 239 ## Top Disk-Using directories
 240 ## TODO: Consider using numfmt instead of awk
 241 ## tdu : path-string -> (size:number * directory:path-string) list
 242 tdu() {
 243     local -r root_path="$1"
 244
 245     du "$root_path" \
 246     | awk '
 247         {
 248             size = $1
 249             path = $0
 250             sub("^" $1 "\t+", "", path)
 251             paths[path] = size
 252             if (size > max)
 253                 max = size
 254         }
 255
 256         END {
 257             for (path in paths) {
 258                 size = paths[path]
 259                 pct = 100 * (size / max)
 260                 gb = size / 1024 / 1024
 261                 printf("%6.2f %3d%% %s\n", gb, pct, path)
 262             }
 263         }
 264     ' \
 265     | sort -r -n -k 1 \
 266     | head -50 \
 267     | tac
 268     # A slight optimization: head can exit before traversing the full input.
 269 }
 270
 271 ## Top Disk-Using Files
 272 ## tduf : path-string list -> (size:number * file:path-string) list
 273 tduf() {
 274     find "$@" -type f -printf '%s\t%p\0' \
 275     | sort -z -n -k 1 \
 276     | tail -z -n 50 \
 277     | gawk -v RS='\0' '
 278         {
 279             size = $1
 280             path = $0
 281             sub("^" $1 "\t+", "", path)
 282             gb = size / 1024 / 1024 / 1024
 283             printf("%f\t%s\n", gb, path)
 284         }'
 285 }
 286
 287 # Most-recently modified file system objects
 288 ## recent : ?(path-string list) -> path-string list
 289 recent() {
 290     # NOTES:
 291     # - %T+ is a GNU extension;
 292     # - gawk is able to split records on \0, while awk cannot.
 293     find "$@" -printf '%T@ %T+ %p\0' \
 294     | tee >(gawk -v RS='\0' 'END { printf("[INFO] Total found: %d\n", NR); }') \
 295     | sort -z -k 1 -n -r \
 296     | head -n "$(stty size | awk 'NR == 1 {print $1 - 5}')" -z \
 297     | gawk -v RS='\0' '
 298         {
 299             sub("^" $1 " +", "")  # Remove epoch time
 300             sub("+", " ")         # Blank-out the default separator
 301             sub("\\.[0-9]+", "")  # Remove fractional seconds
 302             print
 303         }'
 304 }
 305
 306 ## recent_dirs : ?(path-string list) -> path-string list
 307 recent_dirs() {
 308     recent "$@" -type d
 309 }
 310
 311 ## recent_files : ?(path-string list) -> path-string list
 312 recent_files() {
 313     recent "$@" -type f
 314 }
 315
 316 ## pa_def_sink : unit -> string
 317 pa_def_sink() {
 318     pactl info | awk '/^Default Sink:/ {print $3}'
 319 }
 320
 321 ## void_pkgs : ?(string) -> json
 322 void_pkgs() {
 323     curl "https://xq-api.voidlinux.org/v1/query/x86_64?q=$1" | jq '.data'
 324 }
 325
 326 ## Colorful man
 327 ## man : string -> string
 328 man() {
 329     # mb: begin blink
 330     # md: begin bold
 331     # me: end   bold, blink and underline
 332     #
 333     # so: begin standout (reverse video)
 334     # se: end   standout
 335     #
 336     # us: begin underline
 337     # ue: end   underline
 338
 339     LESS_TERMCAP_md=$'\e[01;30m' \
 340     LESS_TERMCAP_me=$'\e[0m' \
 341     LESS_TERMCAP_so=$'\e[01;44;33m' \
 342     LESS_TERMCAP_se=$'\e[0m' \
 343     LESS_TERMCAP_us=$'\e[01;33m' \
 344     LESS_TERMCAP_ue=$'\e[0m' \
 345     command man "$@"
 346 }
 347
 348 ## new experiment
 349 ## x : string list -> unit
 350 x() {
 351     cd "$(~/bin/x $@)" || kill -INT $$
 352 }
 353
 354 ## ocaml repl
 355 ## hump : unit -> unit
 356 hump() {
 357     ledit -l "$(stty size | awk '{print $2}')" ocaml $@
 358 }
 359
 360 ## search howtos
 361 ## howto : unit -> string
 362 howto() {
 363     cat "$(find  ~/arc/doc/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
 364 }
 365
 366 _yt() {
 367     local -r base_dir="$1"
 368     local -r uri="$2"
 369     local -r opts="$3"
 370
 371     local -r yt=youtube-dl
 372     local -r id=$("$yt" --get-id "$uri")
 373     local -r title=$("$yt" --get-title "$uri" | sed 's/[^А-Яа-яA-Za-z0-9._-]/_/g')
 374     local -r dir="${base_dir}/${title}--${id}"
 375
 376     mkdir -p "$dir"
 377     cd "$dir" || kill -INT $$
 378     echo "$uri" > 'uri'
 379     "$yt" $opts -c --write-all-thumbnails --write-description --write-info-json "$uri"
 380 }
 381
 382 yt_audio() {
 383     local -r uri="$1"
 384     _yt "${DIR_YOUTUBE_AUDIO}/individual" "$uri" '-f 140'
 385 }
 386
 387 yt_video() {
 388     local -r uri="$1"
 389     _yt "${DIR_YOUTUBE_VIDEO}/individual" "$uri"
 390 }
 391
 392 gh_fetch_repos() {
 393     local -r user_type="$1"
 394     local -r user_name="$2"
 395
 396     curl "https://api.github.com/$user_type/$user_name/repos?page=1&per_page=10000"
 397 }
 398
 399 gh_clone() {
 400     local -r gh_user_type="$1"
 401     local -r gh_user_name="$2"
 402
 403     local -r gh_dir="${DIR_GITHUB}/${gh_user_name}"
 404     mkdir -p "$gh_dir"
 405     cd "$gh_dir" || kill -INT $$
 406     gh_fetch_repos "$gh_user_type" "$gh_user_name" \
 407     | jq --raw-output '.[] | select(.fork | not) | .clone_url' \
 408     | parallel -j 25 \
 409     git clone {}
 410 }
 411
 412 gh_clone_user() {
 413     gh_clone 'users' "$1"
 414 }
 415
 416 gh_clone_org() {
 417     gh_clone 'orgs' "$1"
 418 }
 419
 420 gh_clone_repo() {
 421     gh_username=$(echo "$1" | awk -F / '"$1 == "https" && $3 == github.com" {print $4}')
 422     gh_dir="${DIR_GITHUB}/${gh_username}"
 423     mkdir -p "$gh_dir"
 424     cd "$gh_dir" || kill -INT $$
 425     git clone "$1"
 426 }
 427
 428 daily_todo_file_template() {
 429 cat << EOF
 430 ===============================================================================
 431 $(date '+%F %A')
 432 ===============================================================================
 433
 434
 435 backlog
 436 -------------------------------------------------------------------------------
 437 EOF
 438 }
 439
 440 todo() {
 441     local date
 442     date="$(date +%F)"
 443     local -r dir="$DIR_TODO/daily"
 444     local -r file="$dir/$date.txt"
 445
 446     mkdir -p "$dir"
 447     if [ ! -f "$file" ]
 448     then
 449         daily_todo_file_template > "$file"
 450     fi
 451     cd "$DIR_TODO" && "$EDITOR" "$file"
 452 }
 453
 454 work_log_template() {
 455 cat << EOF
 456 $(date '+%F %A')
 457 ==========
 458
 459 Morning report
 460 --------------
 461
 462 ### Prev
 463
 464 ### Curr
 465
 466 ### Next
 467
 468 ### Blockers
 469
 470 Day's notes
 471 -----------
 472 EOF
 473 }
 474
 475 work_log() {
 476     mkdir -p "$DIR_WORK_LOG"
 477     local -r file_work_log_today="${DIR_WORK_LOG}/daily-$(date +%F).md"
 478     if [ ! -f "$file_work_log_today" ]
 479     then
 480         work_log_template > "$file_work_log_today"
 481     fi
 482     vim -c 'set spell' "$file_work_log_today"
 483
 484 }
 485
 486 note() {
 487     mkdir -p "$DIR_NOTES"
 488     vim -c 'set spell' "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
 489 }
 490
 491 _bt_devs_infos() {
 492     # grep's defintion of a line does not include \r, wile awk's does and
 493     # which bluetoothctl outputs
 494     awk '/^Device +/ {print $2}' \
 495     | xargs -I% sh -c 'echo info % | bluetoothctl' \
 496     | awk '/^Device |^\t[A-Z][A-Za-z0-9]+: /'
 497 }
 498
 499 bt_devs_paired() {
 500     echo 'paired-devices' | bluetoothctl | _bt_devs_infos
 501 }
 502
 503 bt_devs() {
 504     echo 'devices' | bluetoothctl | _bt_devs_infos
 505 }
 506
 507 run() {
 508     local -r stderr="$(mktemp)"
 509
 510     local code urgency
 511
 512     $@ 2> >(tee "$stderr")
 513     code="$?"
 514     case "$code" in
 515         0) urgency='normal';;
 516         *) urgency='critical'
 517     esac
 518     notify-send -u "$urgency" "Job done: $code" "$(cat $stderr)"
 519     rm "$stderr"
 520 }
 521
 522 bar_gauge() {
 523     awk "$@" '
 524         BEGIN {
 525             # CLI options
 526             width    = width    ? width    : 80
 527             ch_left  = ch_left  ? ch_left  : "["
 528             ch_right = ch_right ? ch_right : "]"
 529             ch_blank = ch_blank ? ch_blank : "-"
 530             ch_used  = ch_used  ? ch_used  : "|"
 531             num      = num      ? 1        : 0
 532             pct      = pct      ? 1        : 0
 533         }
 534
 535         {
 536             cur = $1
 537             max = $2
 538             lab = $3
 539
 540             cur_scaled = num_scale(cur, max, 1, width)
 541
 542             printf \
 543                 "%s%s%s%s", \
 544                 lab ? lab         " " : "", \
 545                 num ? cur "/" max " " : "", \
 546                 pct ? sprintf("%3.0f%% ", cur / max * 100) : "", \
 547                 ch_left
 548             for (i=1; i<=width; i++) {
 549                 c = i <= cur_scaled ? ch_used : ch_blank
 550                 printf "%s", c
 551             }
 552             printf "%s\n", ch_right
 553         }
 554
 555         function num_scale(src_cur, src_max, dst_min, dst_max) {
 556             return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
 557         }
 558     '
 559 }
 560
 561 flat_top_5() {
 562     sort -n -k 1 -r \
 563     | head -5 \
 564     | awk '
 565         {
 566             cur  = $1
 567             max  = $2
 568             name = $3
 569             pct  = cur / max * 100
 570             printf "%s%s %.2f%%", sep, name, pct
 571             sep = ",  "
 572         }
 573
 574         END {printf "\n"}
 575         '
 576 }
 577
 578 internet_addr() {
 579     curl --silent --show-error --max-time "${1:=1}" 'https://api.ipify.org' 2>&1
 580 }
 581
 582 status_batt() {
 583     case "$(uname)" in
 584         'Linux')
 585             if which upower > /dev/null
 586             then
 587                 upower --dump \
 588                 | awk '
 589                     /^Device:[ \t]+/ {
 590                         device["path"] = $2
 591                         next
 592                     }
 593
 594                     /  battery/ && device["path"] {
 595                         device["is_battery"] = 1
 596                         next
 597                     }
 598
 599                     /    percentage:/ && device["is_battery"] {
 600                         device["battery_percentage"] = $2
 601                         sub("%$", "", device["battery_percentage"])
 602                         next
 603                     }
 604
 605                     /^$/ {
 606                         if (device["is_battery"] && device["path"] == "/org/freedesktop/UPower/devices/DisplayDevice")
 607                             print device["battery_percentage"], 100, "batt"
 608                         delete device
 609                     }
 610                 '
 611             fi
 612         ;;
 613     esac
 614 }
 615
 616 indent() {
 617     awk -v unit="$1" '{printf "%s%s\n", unit, $0}'
 618 }
 619
 620 status() {
 621     local -r indent_unit='    '
 622
 623     uname -srvmo
 624     hostname | figlet
 625     uptime
 626
 627     echo
 628
 629     echo 'accounting'
 630
 631     printf '%stmux\n%ssessions %d, clients %d\n' \
 632         "$indent_unit" \
 633         "${indent_unit}${indent_unit}" \
 634         "$(tmux list-sessions 2> /dev/null | wc -l)" \
 635         "$(tmux list-clients  2> /dev/null | wc -l)"
 636
 637     echo
 638
 639     printf '%sprocs by user\n' "${indent_unit}"
 640     ps -eo user \
 641     | awk '
 642         NR > 1 {
 643             count_by_user[$1]++
 644             total++
 645         }
 646
 647         END {
 648             for (user in count_by_user)
 649                 print count_by_user[user], total, user
 650         }
 651         ' \
 652     | flat_top_5 \
 653     | indent "${indent_unit}${indent_unit}"
 654
 655     echo
 656
 657     echo 'resources'
 658     (
 659         free | awk '$1 == "Mem:" {print $3, $2, "mem"}'
 660         df ~ | awk 'NR == 2 {print $3, $3 + $4, "disk"}'
 661         status_batt
 662     ) \
 663     | bar_gauge -v width=60 -v pct=1 \
 664     | column -t \
 665     | indent "$indent_unit"
 666
 667     echo
 668
 669     printf '%smem by proc\n' "$indent_unit"
 670     ps -eo rss,comm \
 671     | awk -v total="$(free | awk '$1 == "Mem:" {print $2; exit}')" '
 672         NR > 1 {
 673             rss = $1
 674             proc = $2
 675             by_proc[proc] += rss
 676         }
 677
 678         END {
 679             for (proc in by_proc)
 680                 print by_proc[proc], total, proc
 681             }
 682         ' \
 683         | flat_top_5 \
 684         | indent "${indent_unit}${indent_unit}"
 685
 686     echo
 687
 688     local _dir temp_input label_file label
 689
 690     printf '%sthermal\n' "$indent_unit"
 691     for _dir in /sys/class/hwmon/hwmon*; do
 692         cat "$_dir"/name
 693         find "$_dir"/ -name 'temp*_input' \
 694             | while read -r temp_input; do
 695                 label_file=${temp_input//_input/_label}
 696                 if [ -f "$label_file" ]; then
 697                     label=$(< "$label_file")
 698                 else
 699                     label=''
 700                 fi
 701                 awk -v label="$label" '{
 702                         if (label)
 703                             label = sprintf(" (%s)", label)
 704                             printf("%.2f°C%s\n", $1 / 1000, label)
 705                     }' \
 706                     "$temp_input"
 707             done \
 708             | sort \
 709             | indent "$indent_unit"
 710     done \
 711     | indent "${indent_unit}${indent_unit}"
 712
 713     echo 'net'
 714     #local -r internet_addr=$(internet_addr 0.5)
 715     #local -r internet_ptr=$(host -W 1 "$internet_addr" | awk 'NR == 1 {print $NF}' )
 716
 717     #echo "${indent_unit}internet"
 718     #echo "${indent_unit}${indent_unit}$internet_addr  $internet_ptr"
 719     echo "${indent_unit}if"
 720     (ifconfig; iwconfig) 2> /dev/null \
 721     | awk '
 722         /^[^ ]/ {
 723             device = $1
 724             sub(":$", "", device)
 725             if ($4 ~ "ESSID:") {
 726                 _essid = $4
 727                 sub("^ESSID:\"", "", _essid)
 728                 sub("\"$", "", _essid)
 729                 essid[device] = _essid
 730             }
 731             next
 732         }
 733
 734         /^ / && $1 == "inet" {
 735             address[device] = $2
 736             next
 737         }
 738
 739         /^ +Link Quality=[0-9]+\/[0-9]+ +Signal level=/ {
 740             split($2, lq_parts_eq, "=")
 741             split(lq_parts_eq[2], lq_parts_slash, "/")
 742             cur = lq_parts_slash[1]
 743             max = lq_parts_slash[2]
 744             link[device] = cur / max * 100
 745             next
 746         }
 747
 748         END {
 749             for (device in address)
 750                 if (device != "lo") {
 751                     l = link[device]
 752                     e = essid[device]
 753                     l = l ? sprintf("%.0f%%", l) : "--"
 754                     e = e ? e : "--"
 755                     print device, address[device], e, l
 756                 }
 757         }
 758         ' \
 759     | column -t \
 760     | indent "${indent_unit}${indent_unit}"
 761
 762     # WARN: ensure: $USER ALL=(ALL) NOPASSWD:/bin/netstat
 763
 764     echo "${indent_unit}-->"
 765
 766     sudo -n netstat -tulnp \
 767     | awk -v indent="${indent_unit}${indent_unit}" '
 768         NR > 2 && ((/^tcp/ && proc = $7) || (/^udp/ && proc = $6)) {
 769             protocol = $1
 770             addr = $4
 771             port = a[split(addr, a, ":")]
 772             name = p[split(proc, p, "/")]
 773             names[name] = 1
 774             protocols[protocol] = 1
 775             if (!seen[protocol, name, port]++)
 776                 ports[protocol, name, ++seen[protocol, name]] = port
 777         }
 778
 779         END {
 780             for (protocol in protocols) {
 781                 printf "%s%s\t", indent, toupper(protocol)
 782                 for (name in names) {
 783                     if (n = seen[protocol, name]) {
 784                         sep = ""
 785                         printf "%s:", name
 786                         for (i = 1; i <= n; i++) {
 787                             printf "%s%d", sep, ports[protocol, name, i]
 788                             sep = ","
 789                         }
 790                         printf "  "
 791                     }
 792                 }
 793                 printf "\n"
 794             }
 795         }'
 796
 797     echo "${indent_unit}<->"
 798
 799     printf '%sTCP\t' "${indent_unit}${indent_unit}"
 800     sudo -n netstat -tnp \
 801     | awk 'NR > 2 && $6 == "ESTABLISHED" {print $7}' \
 802     | awk '{sub("^[0-9]+/", ""); print}' \
 803     | sort -u \
 804     | xargs \
 805     | column -t
 806
 807     # TODO: iptables summary
 808 }
 809
 810 ssh_invalid_by_addr() {
 811     awk '
 812         /: Invalid user/ && $5 ~ /^sshd/ {
 813             addr=$10 == "port" ? $9 : $10
 814             max++
 815             by_addr[addr]++
 816         }
 817
 818         END {
 819             for (addr in by_addr)
 820                 if ((c = by_addr[addr]) > 1)
 821                     printf "%d %d %s\n", c, max, addr
 822         }
 823         ' \
 824         /var/log/auth.log \
 825         /var/log/auth.log.1 \
 826     | sort -n -k 1 \
 827     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 828     | column -t
 829 }
 830
 831 ssh_invalid_by_day() {
 832     awk '
 833         BEGIN {
 834             m["Jan"] = "01"
 835             m["Feb"] = "02"
 836             m["Mar"] = "03"
 837             m["Apr"] = "04"
 838             m["May"] = "05"
 839             m["Jun"] = "06"
 840             m["Jul"] = "07"
 841             m["Aug"] = "08"
 842             m["Sep"] = "09"
 843             m["Oct"] = "10"
 844             m["Nov"] = "11"
 845             m["Dec"] = "12"
 846         }
 847
 848         /: Invalid user/ && $5 ~ /^sshd/ {
 849             day = m[$1] "-" $2
 850             max++
 851             by_day[day]++
 852         }
 853
 854         END {
 855             for (day in by_day)
 856                 if ((c = by_day[day]) > 1)
 857                     printf "%d %d %s\n", c, max, day
 858         }
 859         ' \
 860         /var/log/auth.log \
 861         /var/log/auth.log.1 \
 862     | sort -k 3 \
 863     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 864     | column -t
 865 }
 866
 867 ssh_invalid_by_user() {
 868     awk '
 869         /: Invalid user/ && $5 ~ /^sshd/ {
 870             user=$8
 871             max++
 872             by_user[user]++
 873         }
 874
 875         END {
 876             for (user in by_user)
 877                 if ((c = by_user[user]) > 1)
 878                     printf "%d %d %s\n", c, max, user
 879         }
 880         ' \
 881         /var/log/auth.log \
 882         /var/log/auth.log.1 \
 883     | sort -n -k 1 \
 884     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 885     | column -t
 886 }
 887
 888 loggers() {
 889     awk '
 890         {
 891             split($5, prog, "[")
 892             sub(":$", "", prog[1]) # if there were no [], than : will is left behind
 893             print prog[1]
 894         }' /var/log/syslog /var/log/syslog.1 \
 895     | awk '
 896         {
 897             n = split($1, path, "/")  # prog may be in path form
 898             prog = path[n]
 899             total++
 900             count[prog]++
 901         }
 902
 903         END {
 904             for (prog in count)
 905                 print count[prog], total, prog
 906         }' \
 907     | sort -n -k 1 \
 908     | bar_gauge -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 909     | column -t
 910 }