1
0
Fork 0
mirror of https://github.com/notwa/rc synced 2024-11-05 07:29:04 -08:00
rc/sh/asn
2024-08-06 21:07:05 -07:00

92 lines
2.5 KiB
Bash

#!/usr/bin/env bash
# compat: -ash +bash -dash -hush -ksh -loksh -mksh -oksh +osh -posh -yash +zsh
__asn_rev() { # byte-swap
REPLY=
local ip="${1:?missing argument}" byte=
while [ -n "$ip" ]; do
byte="${ip%%.*}" ip="${ip#"$byte"}" ip="${ip#.}"
REPLY="$byte.$REPLY"
done
REPLY="${REPLY%.}"
}
__asn_txt() {
REPLY="$(exec dig +short +timeout=1 +tries=1 "$@" TXT)"
case "$REPLY" in
(";;"*) return 2;;
('"'*'"') true;;
(*) return 1;;
esac
REPLY="${REPLY#\"}" REPLY="${REPLY%\"}"
}
declare -A __asn_replies __asn_descs
__asn_lookup() {
local remote= ip= ish= REPLY= sep= asn= desc=
remote=asn.cymru.com
ip="${1:?missing argument}"
ish="${ip%.*}"
sep=' | '
__asn_rev "$ip"
if [ -n "${__asn_replies["$ish"]}" ]; then
REPLY="${__asn_replies["$ish"]}"
else
__asn_txt "$REPLY.origin.$remote" || return
__asn_replies["$ish"]="$REPLY"
fi
asns="${REPLY%%"$sep"*}" REPLY="${REPLY#"$asns$sep"}"
subnet="${REPLY%%"$sep"*}" REPLY="${REPLY#"$subnet$sep"}"
country="${REPLY%%"$sep"*}" REPLY="${REPLY#"$country$sep"}"
registry="${REPLY%%"$sep"*}" REPLY="${REPLY#"$registry$sep"}"
date="${REPLY}"
description=
while [ -n "$asns" ]; do
asn="${asns%% *}" asns="${asns#"$asn"}" asns="${asns# }"
if [ -n "${__asn_descs["$asn"]}" ]; then
desc="${__asn_descs["$asn"]}"
else
__asn_txt "AS$asn.$remote" || return
desc="${REPLY##* | }"
__asn_descs["$asn"]="$desc"
fi
if [ -z "$description" ]; then
description="$desc"
elif [ "${description%"$desc"}" = "$description" ]; then
description="$description,$desc"
fi
done
printf '%-7s | %-16s | %s\n' "$asn" "$ip" "$description"
}
__asn() {
local attempt="${2:-1}"
__asn_lookup "$1" ||
if [ $? = 1 ]; then
printf '%-7s | %-16s | %s\n' "n/a" "$1" "n/a"
elif [ "$attempt" -lt 2 ]; then
sleep 4 && __asn "$1" $((attempt+1))
else
printf '%-7s | %-16s | %s\n' "(fail)" "$1" "(failed to retrieve data, try again)"
fi
}
asn() {
echo 'AS | IP | AS Name'
if [ $# = 0 ]; then
while read -r ip; do
__asn "$ip"
done < <(grep -oP '(?<=^|[^\d.])((0|1\d\d|2[0-4]\d|25[0-5]|[1-9]\d?)\.((?2))\.((?2))\.((?2)))(?=$|[^\d.])')
else
for ip; do
__asn "$ip"
done
fi
}
[ -n "${preload+-}" ] || asn "$@"