mirror of
https://github.com/notwa/rc
synced 2024-11-05 02:59:03 -08:00
92 lines
2.5 KiB
Bash
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 "$@"
|