-
Notifications
You must be signed in to change notification settings - Fork 3
/
give_me_hostname
executable file
·111 lines (101 loc) · 2.72 KB
/
give_me_hostname
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/bin/bash
#
# Copyright (C) 2015 barbUk
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation, either version 3 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.
#
set -o errexit -o pipefail -o nounset
readonly mode="${1:-}"
awk_command=awk
if [ "$(uname)" = "Darwin" ]; then
awk_command=gawk
fi
dsh_hosts() {
local file
file="$HOME/.dsh/group/all"
if [ -f "$file" ]; then
grep -vE '^#' "$file"
fi
}
known_hosts() {
local file
file="$HOME/.ssh/known_hosts"
if [ -f "$file" ]; then
"$awk_command" '{print $1}' "$file" \
| grep -v ^\| \
| cut -d, -f 1 \
| sed -e 's/\[//g' \
| sed -e 's/\]//g' \
| cut -d: -f1 \
| grep -v ssh-rsa
fi
}
bash_history() {
local file
file="$HOME/.bash_history"
if [ -f "$file" ]; then
grep -E "^host|^dig|^drill|^ping|^whois" "$file" \
| "$awk_command" '{print $NF}'
fi
}
ssh_config() {
local file
file="$HOME/.ssh/config"
if [ -d "$(dirname "$file")" ]; then
"$awk_command" '/^Host/ && ! /\*/ {print $2}' "$file"
fi
}
ssh_config_d() {
local file
file="$HOME/.ssh/config.d/*"
if [ -d "$(dirname "$file")" ]; then
# shellcheck disable=2086
"$awk_command" '/^Host/ && ! /*/ {print $2}' $file
fi
}
main() {
case "$mode" in
"dsh") dsh_hosts | sort -u;;
"hosts") known_hosts | sort -u;;
"ssh")
(
ssh_config_d
dsh_hosts
known_hosts
) | grep -vE "\\b([0-9]{1,3}\\.){3}[0-9]{1,3}\\b" | sort -u
;;
"ips")
(
bash_history
known_hosts
) | grep -oE "\\b([0-9]{1,3}\\.){3}[0-9]{1,3}\\b" | sort -u
;;
"all")
(
dsh_hosts
ssh_config_d
known_hosts
bash_history
) | sort -u
;;
*)
(
dsh_hosts
known_hosts
bash_history
) | grep -vE "\\b([0-9]{1,3}\\.){3}[0-9]{1,3}\\b|^@|^\\+|^\\.'" | sort -u
;;
esac
return 0
}
main