From 54bb0393e2dff29a6ae1099cf1d46e7405a5016d Mon Sep 17 00:00:00 2001
From: Massaki Archambault <badjware@massaki.ca>
Date: Mon, 26 Mar 2018 09:06:05 -0400
Subject: [PATCH] further improved nmap helper functions; default params, tun,
 generic function

---
 files/.config/zsh/custom/101-function.zsh | 29 +++++++++++++----------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/files/.config/zsh/custom/101-function.zsh b/files/.config/zsh/custom/101-function.zsh
index d5eb3bf..9cf5e02 100644
--- a/files/.config/zsh/custom/101-function.zsh
+++ b/files/.config/zsh/custom/101-function.zsh
@@ -9,23 +9,28 @@ vifunction() {
 }
 
 nmap-libvirt() {
-    local addr_list=""
-    for link in $(ip link show up | grep -Eo 'virbr[0-9]+'); do
-        for addr in $(ip addr show $link | grep -oP '(?<=inet )([0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]{1,2}'); do
-            addr_list="$addr_list $addr"
-        done
-    done
-    echo "Scanning $addr_list"
-    nmap $@ $addr_list
+    __nmap_iface 'virbr[0-9]+' $@
+}
+
+nmap-tun() {
+    __nmap_iface 'tun[0-9]+' $@
 }
 
 nmap-local() {
-    local addr_list=""
-    for link in $(ip link show up | grep -Eo '(eth|wlan)[0-9]+'); do
+    __nmap_iface '(eth|wlan)[0-9]+' $@
+}
+
+__nmap_iface() {
+    local iface="$1"
+    shift
+    local opt=${@:--T5}
+
+    for link in $(ip link show up | grep -Eo "$iface"); do
         for addr in $(ip addr show $link | grep -oP '(?<=inet )([0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]{1,2}'); do
             addr_list="$addr_list $addr"
         done
     done
-    echo "Scanning $addr_list"
-    nmap $@ $addr_list
+
+    printf "Scanning %s\n" "$addr"
+    nmap $opt $addr_list
 }