Brijesh's Git Server — network-scan @ main

main.go (view raw)

 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
 
package main

import (
	"fmt"
	"log"
	"os"

	"network-scan/arp"
	"network-scan/database"
	"network-scan/definitions"
	"network-scan/icmp"
	"network-scan/oui"
	"network-scan/utils"
)

const dbPath string = "./oui.db"

func init() {
	err := database.CreateDbIfNotExists(dbPath)
	if err != nil {
		log.Fatalf("Error creating database file: %v", err)
	}

	db, err := database.New(dbPath)
	if err != nil {
		log.Fatalf("Error opening database: %v", err)
	}
	defer db.Close()

	err = db.CreateOuiTable()
	if err != nil {
		log.Fatalf("Error creating table: %v", err)
	}
}

func main() {
	var devices []definitions.Device

	db, err := database.New(dbPath)
	if err != nil {
		log.Fatalf("Error opening database: %v", err)
	} else {
		utils.PrintInColor("Database opened successfully", 28)
	}
	defer db.Close()

	utils.ConfirmBeforeRunning("Do you want to seed sqlite from IEEE data?", func() {
		file, err := os.Open("standards-oui.ieee.org.txt")
		if err != nil {
			log.Fatalf("Error opening file: %v", err)
		}
		defer file.Close()

		oui.ScanDataFromTextFile("/Users/brijesh/projects/ongoing/network-scan/standards-oui.ieee.org.txt", db)
		utils.PrintInColor("OUI data imported successfully", 28)
	})

	fmt.Println("Scanning network for devices...")
	devices, err = arp.GetDevices()
	if err != nil {
		fmt.Println("Error:", err)
		return
	}

	fmt.Println("Devices found:")
	for _, device := range devices {
		fmt.Printf("IP: %s, MAC: %s\n", device.IP, device.MAC)
	}

	utils.ConfirmBeforeRunning("Do you want to check reachability of all devices?", func() {
		isRoot, err := utils.CheckSudoPrivileges()
		if err != nil {
			fmt.Printf("Error checking root status: %v\n", err)
			return
		}
		if !isRoot {
			fmt.Println("ERROR: sudo previleges are needed for creating raw sockets which is needed to send ICMP packets")
			return
		}

		for _, device := range devices {
			ip := device.IP

			online, err := icmp.CheckICMPReachability(ip)
			if err != nil {
				log.Fatalf("Error checking device status: %v", err)
			}

			if online {
				fmt.Printf("Device at %s is online\n", ip)
			} else {
				fmt.Printf("Device at %s is offline\n", ip)
			}
		}
	})
}