mirror of
https://github.com/davidallendj/magellan.git
synced 2025-12-20 11:37:01 -07:00
109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
magellan "github.com/OpenCHAMI/magellan/internal"
|
|
"github.com/OpenCHAMI/magellan/internal/db/sqlite"
|
|
|
|
"github.com/cznic/mathutil"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
begin uint8
|
|
end uint8
|
|
subnets []string
|
|
subnetMasks []net.IP
|
|
disableProbing bool
|
|
)
|
|
|
|
var scanCmd = &cobra.Command{
|
|
Use: "scan",
|
|
Short: "Scan for BMC nodes on a network",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
hostsToScan []string
|
|
portsToScan []int
|
|
)
|
|
|
|
// start by adding `--host` supplied to scan
|
|
if len(hosts) > 0 {
|
|
hostsToScan = hosts
|
|
}
|
|
|
|
// add hosts from `--subnets` and `--subnet-mask`
|
|
for i, subnet := range subnets {
|
|
// subnet string is empty so nothing to do here
|
|
if subnet == "" {
|
|
continue
|
|
}
|
|
|
|
// NOTE: should we check if subnet is valid here or is it done elsewhere (maybe in GenerateHosts)?
|
|
|
|
// no subnet masks supplied so add a default one for class C private networks
|
|
if len(subnetMasks) < i+1 {
|
|
subnetMasks = append(subnetMasks, net.IP{255, 255, 255, 0})
|
|
}
|
|
|
|
// generate a slice of all hosts to scan from subnets
|
|
hostsToScan = append(hostsToScan, magellan.GenerateHosts(subnet, &subnetMasks[i])...)
|
|
}
|
|
|
|
// add ports to use for scanning
|
|
if len(ports) > 0 {
|
|
portsToScan = ports
|
|
} else {
|
|
// no ports supplied so only use defaults
|
|
portsToScan = magellan.GetDefaultPorts()
|
|
}
|
|
|
|
// scan and store scanned data in cache
|
|
if concurrency <= 0 {
|
|
concurrency = mathutil.Clamp(len(hostsToScan), 1, 255)
|
|
}
|
|
probeStates := magellan.ScanForAssets(hostsToScan, portsToScan, concurrency, timeout, disableProbing, verbose)
|
|
if verbose {
|
|
format = strings.ToLower(format)
|
|
if format == "json" {
|
|
b, _ := json.Marshal(probeStates)
|
|
fmt.Printf("%s\n", string(b))
|
|
} else {
|
|
for _, r := range probeStates {
|
|
fmt.Printf("%s:%d (%s)\n", r.Host, r.Port, r.Protocol)
|
|
}
|
|
}
|
|
}
|
|
|
|
// make the dbpath dir if needed
|
|
err := os.MkdirAll(path.Dir(cachePath), 0766)
|
|
if err != nil {
|
|
fmt.Printf("failed tomake database directory: %v", err)
|
|
}
|
|
|
|
sqlite.InsertProbeResults(cachePath, &probeStates)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
scanCmd.Flags().StringSliceVar(&hosts, "host", []string{}, "set additional hosts to scan")
|
|
scanCmd.Flags().IntSliceVar(&ports, "port", []int{}, "set the ports to scan")
|
|
scanCmd.Flags().StringVar(&format, "format", "", "set the output format")
|
|
scanCmd.Flags().StringSliceVar(&subnets, "subnet", []string{}, "set additional subnets")
|
|
scanCmd.Flags().IPSliceVar(&subnetMasks, "subnet-mask", []net.IP{}, "set the subnet masks to use for network (must match number of subnets)")
|
|
scanCmd.Flags().BoolVar(&disableProbing, "disable-probing", false, "disable probing scanned results for BMC nodes")
|
|
|
|
viper.BindPFlag("scan.hosts", scanCmd.Flags().Lookup("host"))
|
|
viper.BindPFlag("scan.ports", scanCmd.Flags().Lookup("port"))
|
|
viper.BindPFlag("scan.subnets", scanCmd.Flags().Lookup("subnet"))
|
|
viper.BindPFlag("scan.subnet-masks", scanCmd.Flags().Lookup("subnet-mask"))
|
|
viper.BindPFlag("scan.disable-probing", scanCmd.Flags().Lookup("disable-probing"))
|
|
|
|
rootCmd.AddCommand(scanCmd)
|
|
}
|