Version 1.3.1: [FIX] Allow disabling header

API.go

@@ -1,13 +1,23 @@
 package main
 
 import (
+	"encoding/json"
+	"fmt"
 	"io/ioutil"
 	"net"
 	"net/http"
+	"os"
+	"sort"
+	"strings"
 )
 
 func getLocalIP() string {
-	resp, _ := http.Get("https://api.ipify.org")
+	resp, err := http.Get("https://api.ipify.org")
+	if err != nil {
+		fmt.Println("FATAL: Cannot get local IP.")
+		os.Exit(2)
+		return ""
+	}
 	body, _ := ioutil.ReadAll(resp.Body)
 	return string(body[:])
 }
@@ -24,3 +34,37 @@ func resolveDNSHostname(hostname string) string {
 	address, _ := net.LookupHost(hostname)
 	return address[0]
 }
+
+func getIPInfo(ipaddress string) IPAddressInfo {
+	apiEndpoint := "http://ip-api.com/json/" + ipaddress
+	resp, err := http.Get(apiEndpoint)
+	if err != nil {
+		fmt.Println("FATAL: Cannot contact IP address information API.")
+		os.Exit(3)
+	}
+	body, _ := ioutil.ReadAll(resp.Body)
+	infoString := string(body)
+	var info IPAddressInfo
+	err = json.Unmarshal([]byte(infoString), &info)
+	if err != nil {
+		fmt.Println("FATAL: Cannot serialize recieved IP address data.")
+		os.Exit(4)
+	}
+	return info
+}
+
+func getBGPPrefixes(as string) {
+	apiEndpoint := "https://api.hackertarget.com/aslookup/?q=" + as
+	resp, err := http.Get(apiEndpoint)
+	if err != nil {
+		fmt.Println("FATAL: Cannot contact BGP Prefixes API.")
+		os.Exit(5)
+	}
+	body, _ := ioutil.ReadAll(resp.Body)
+	prefixesString := string(body)
+	var prefixes = strings.Split(prefixesString, "\n")[1:]
+	sort.Strings(prefixes)
+	for index := range prefixes {
+		fmt.Println(prefixes[index])
+	}
+}

Header.go

@@ -11,4 +11,5 @@ func printHeader() {
 	fmt.Println("|      By Luke Tainton       |")
 	fmt.Println("|        @luketainton        |")
 	fmt.Println("------------------------------")
+	fmt.Println("")
 }

IPInfo.go

@@ -1,11 +1,7 @@
 package main
 
 import (
-	"encoding/json"
 	"fmt"
-	"io/ioutil"
-	"net/http"
-	"sort"
 	"strings"
 )
 
@@ -27,31 +23,6 @@ type IPAddressInfo struct {
 	IPAddress    string  `json:"query"`
 }
 
-func getIPInfo(ipaddress string) IPAddressInfo {
-	apiEndpoint := "http://ip-api.com/json/" + ipaddress
-	resp, _ := http.Get(apiEndpoint)
-	body, _ := ioutil.ReadAll(resp.Body)
-	infoString := string(body)
-	var info IPAddressInfo
-	err := json.Unmarshal([]byte(infoString), &info)
-	if err != nil {
-		fmt.Println(err)
-	}
-	return info
-}
-
-func getBGPPrefixes(as string) {
-	apiEndpoint := "https://api.hackertarget.com/aslookup/?q=" + as
-	resp, _ := http.Get(apiEndpoint)
-	body, _ := ioutil.ReadAll(resp.Body)
-	prefixesString := string(body)
-	var prefixes = strings.Split(prefixesString, "\n")[1:]
-	sort.Strings(prefixes)
-	for i := 0; i < len(prefixes); i++ {
-		fmt.Println(prefixes[i])
-	}
-}
-
 func printIPInfo(input string, wantPrefixes bool) {
 	var IPInfo IPAddressInfo = getIPInfo(input)
 	var location string = IPInfo.Country + "/" + IPInfo.RegionName + "/" + IPInfo.City

LICENSE.md

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 Luke Tainton
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

Main.go

@@ -3,27 +3,51 @@ package main
 import (
 	"flag"
 	"fmt"
+	"os"
 )
 
 func main() {
-	printHeader()
-
 	var input string
 	var wantPrefixes bool
-	localIPAddress := getLocalIP()
+	var wantHeader bool
-	flag.StringVar(&input, "i", localIPAddress, "Specify IP address or domain name.")
+
-	flag.BoolVar(&wantPrefixes, "p", false, "Enable printing of advertised BGP prefixes.")
+	flag.StringVar(&input, "i", "", "IP address or domain")
+	flag.BoolVar(&wantPrefixes, "p", false, "print BGP prefixes")
+	flag.BoolVar(&wantHeader, "b", true, "enable/disable header")
+
+	flag.Usage = func() {
+		fmt.Printf("Usage of iPilot: \n")
+		fmt.Printf("Example: iPilot -b=false -i=me -p \n")
+		fmt.Printf("  -b	bool	enable/disable header (default true)\n")
+		fmt.Printf("  -h	bool	view help\n")
+		fmt.Printf("  -i	string	IP address or domain\n")
+		fmt.Printf("  -p	bool	print BGP prefixes (default false)\n")
+	}
+
 	flag.Parse()
-	var isIPCorrect bool = checkIPSyntax(input)
+
-	if isIPCorrect == true {
+	if wantHeader {
-		printIPInfo(input, wantPrefixes)
+		printHeader()
+	}
+
+	if input == "" {
+		fmt.Println("FATAL: No IP address or domain name was specified.")
+		os.Exit(1)
 	} else {
-		ipaddress := resolveDNSHostname(input)
+		if input == "me" {
-		if checkIPSyntax(ipaddress) == true {
+			input = getLocalIP()
-			fmt.Println("Domain Name:	", input)
+		}
-			printIPInfo(ipaddress, wantPrefixes)
+		var isIPCorrect bool = checkIPSyntax(input)
+		if isIPCorrect == true {
+			printIPInfo(input, wantPrefixes)
 		} else {
-			fmt.Println("Invalid query.")
+			ipaddress := resolveDNSHostname(input)
+			if checkIPSyntax(ipaddress) == true {
+				fmt.Println("Domain Name:	", input)
+				printIPInfo(ipaddress, wantPrefixes)
+			} else {
+				fmt.Println("Invalid query.")
+			}
 		}
 	}
 

README.md

@@ -0,0 +1,20 @@
+# IP Information Lookup Tool (iPilot)
+
+This Go application takes an IP address or domain name and gathers the following information:
+- Location
+- Timezone
+- Internet Service Provider
+- Autonomous System
+- Advertised Prefixes
+
+## Running the script
+Here are some ways that you can run the script:
+| Command                | Description                              |
+| ---------------------- | ---------------------------------------- |
+| `./iPilot -i me`         | Run against your own connection          |
+| `./iPilot -i 1.1.1.1`    | Run against the IP address `1.1.1.1`     |
+| `./iPilot -i google.com` | Run against the domain name `google.com` |
+| `./iPilot -i google.com -p` | Run against the domain name `google.com` and lists BGP prefixes |
+
+## Credits
+This script runs thanks to the APIs provided by [IP-API](http://ip-api.com) and [HackerTarget](https://hackertarget.com/as-ip-lookup).