main.go

package main

import (
	"bufio"
	"fmt"
	"net"
	"os"
	"strconv"
	"strings"

	//l "./latex"
	p "./pcap"
)

func listInterfaces() []string {
	interfaces, err := net.Interfaces()
	var ifaceArr []string

	if err != nil {
		fmt.Print(fmt.Errorf("listInterfaces: %+v\n", err.Error()))
		os.Exit(1)
	}

	for _, iface := range interfaces {
		arr, _ := iface.Addrs()
		if len(arr) == 0 {
			continue
		}
		ifaceArr = append(ifaceArr, iface.Name)
	}

	for index, iface := range ifaceArr {
		fmt.Println(index, ": ", iface)
	}
	return ifaceArr
}

func readUserInput(size int) int {
	reader := bufio.NewReader(os.Stdin)
	count := 0

	for {
		fmt.Print("Enter index of the device: ")
		indexStr, _ := reader.ReadString('\n')
		indexStr = strings.Replace(indexStr, "\n", "", -1)
		index, convErr := strconv.Atoi(indexStr)

		if convErr != nil {
			fmt.Println("Cannot convert input to int")
			os.Exit(1)
		}

		if index >= 0 && index < size {
			return index
		}

		if count == 2 {
			fmt.Println("Invalid choices made. Shutting down...")
			os.Exit(1)
		}

		fmt.Println("Invalid choice. ( 0 -", size-1, ")")
	}
}

func main() {
	// http://www.ascii-art.de/ascii/s/sharks.txt
	shark := "                     ^`.                     o\n     ^_              \\  \\                  o  o\n     \\ \\             {   \\                 o\n     {  \\           /     `~~~--__\n     {   \\___----~~'              `~~-_     ______          _____\n      \\                         /// a  `~._(_||___)________/___\n      / /~~~~-, ,__.    ,      ///  __,,,,)      o  ______/    \\ \n      \\/      \\/    `~~~;   ,---~~-_`~= \\ \\------o-'            \\ \n                       /   /            / /\n                      '._.'           _/_/\n                                      ';|\\ \n            -David 'TAZ' Baltazar-	 \n"
	fmt.Println(shark, "\n|:|:|:|:|:| GoShark |:|:|:|:|:|")
	fmt.Println("Available connected interfaces")

	iface := listInterfaces()
	input := readUserInput(len(iface))
	p.Capture(iface[input])
	//l.MakeReport("toRead.txt")

	os.Exit(0)
}