towk/magellan
1
0
Fork 0
mirror of https://github.com/davidallendj/magellan.git synced 2025-12-20 03:27:03 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add support for scanning PDU components by probing JAWS as well as redfish

Browse source 
Signed-off-by: Ben McDonald <ben.mcdonald@hpe.com>
This commit is contained in:
Ben McDonald 2025-07-03 12:45:02 -07:00
parent 0aa72ce70d
commit 4cfd48da63
No known key found for this signature in database
2 changed files with 46 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

63
pkg/scan.go
View file

@ -16,11 +16,12 @@ import (
)
type RemoteAsset struct {
Host string `json:"host"`
Port int `json:"port"`
Protocol string `json:"protocol"`
State bool `json:"state"`
Timestamp time.Time `json:"timestamp"`
Host string `json:"host"`
Port int `json:"port"`
Protocol string `json:"protocol"`
State bool `json:"state"`
Timestamp time.Time `json:"timestamp"`
ServiceType string `json:"service_type,omitempty"`
}
// ScanParams is a collection of commom parameters passed to the CLI
@ -33,6 +34,8 @@ type ScanParams struct {
DisableProbing bool
Verbose bool
Debug bool
Username string
Password string
}
// ScanForAssets() performs a net scan on a network to find available services
@ -45,7 +48,7 @@ type ScanParams struct {
// to be made concurrently.
//
// If the "disableProbing" flag is set, then the function will skip the extra
// HTTP request made to check if the response was from a Redfish service.
// HTTP request made to check if the response was from a Redfish or JAWS service.
// Otherwise, not receiving a 200 OK response code from the HTTP request will
// remove the service from being stored in the list of scanned results.
//
@ -61,6 +64,17 @@ func ScanForAssets(params *ScanParams) []RemoteAsset {
log.Info().Any("args", params).Msg("starting scan...")
}
probesToRun := []struct {
Type, Path string
}{
{Type: "Redfish", Path: "/redfish/v1/"},
{Type: "JAWS", Path: "/jaws/monitor/outlets"},
}
probeClient := &http.Client{
Timeout: time.Duration(params.Timeout) * time.Second,
}
var wg sync.WaitGroup
wg.Add(params.Concurrency)
for i := 0; i < params.Concurrency; i++ {
@ -78,26 +92,35 @@ func ScanForAssets(params *ScanParams) []RemoteAsset {
if params.Verbose {
log.Debug().Err(err).Msgf("failed to connect to host")
}
wg.Done()
return
// NOTE: This was wg.Done() and return in the original, but that stops the whole worker.
// Continuing allows the worker to process other hosts in its queue.
continue
}
if !params.DisableProbing {
assetsToAdd := []RemoteAsset{}
for _, foundAsset := range foundAssets {
url := fmt.Sprintf("%s:%d/redfish/v1/", foundAsset.Host, foundAsset.Port)
res, _, err := client.MakeRequest(nil, url, http.MethodGet, nil, nil)
if err != nil || res == nil {
if params.Verbose {
log.Printf("failed to make request: %v\n", err)
for _, probe := range probesToRun {
probeURL := fmt.Sprintf("%s:%d%s", foundAsset.Host, foundAsset.Port, probe.Path)
req, err := http.NewRequest("GET", probeURL, nil)
if err != nil {
continue
}
continue
} else if res.StatusCode != http.StatusOK {
if params.Verbose {
log.Printf("request returned code: %v\n", res.StatusCode)
// Add authentication for JAWS endpoints if credentials are provided
if probe.Type == "JAWS" && params.Username != "" && params.Password != "" {
req.SetBasicAuth(params.Username, params.Password)
}
res, err := probeClient.Do(req)
if err == nil && res != nil && res.StatusCode == http.StatusOK {
res.Body.Close()
foundAsset.ServiceType = probe.Type
assetsToAdd = append(assetsToAdd, foundAsset)
break // Found a valid service, no need to probe other types
}
if res != nil {
res.Body.Close()
}
continue
} else {
assetsToAdd = append(assetsToAdd, foundAsset)
}
}
results = append(results, assetsToAdd...)