113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package shodan
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/ns3777k/go-shodan/v4/shodan"
|
|
|
|
"git.tcp.direct/perp/shogo/internal/utils"
|
|
)
|
|
|
|
// Search query
|
|
type Search struct {
|
|
Query string // Search query
|
|
Page int // Current page
|
|
Fields []string // Filter fields
|
|
Separator string // Filter separator
|
|
Results chan string // Results channel
|
|
}
|
|
|
|
// Return shodan search results
|
|
func (s *Search) Search() {
|
|
// Setup options
|
|
options := &shodan.HostQueryOptions{
|
|
Query: s.Query,
|
|
Page: s.Page,
|
|
Minify: false,
|
|
}
|
|
|
|
// Get results
|
|
results, err := utils.Client.GetHostsForQuery(context.Background(), options)
|
|
if err != nil {
|
|
fmt.Printf("%s: %s\n", utils.Red("Error"), err.Error())
|
|
s.Results <- ""
|
|
return
|
|
}
|
|
|
|
// More than 2 matches
|
|
if len(results.Matches) > 2 {
|
|
// Go through matches
|
|
for match := range results.Matches {
|
|
if len(s.Fields) != 0 {
|
|
fields := map[string]string{
|
|
"product": results.Matches[match].Product,
|
|
"ip_str": results.Matches[match].IP.String(),
|
|
"org": results.Matches[match].Organization,
|
|
"isp": results.Matches[match].ISP,
|
|
"transport": results.Matches[match].Transport,
|
|
"data": results.Matches[match].Data,
|
|
"asn": results.Matches[match].ASN,
|
|
"port": strconv.Itoa(results.Matches[match].Port),
|
|
"timestamp": results.Matches[match].Timestamp,
|
|
"os": results.Matches[match].OS,
|
|
}
|
|
|
|
var result string
|
|
var content string
|
|
|
|
// Go through custom fields
|
|
for key, value := range fields {
|
|
// Go through real fields
|
|
for index := range s.Fields {
|
|
// Real field matched custom field
|
|
if s.Fields[index] == key {
|
|
// Port may go on the wrong side
|
|
if key == "port" {
|
|
content += value
|
|
} else {
|
|
content += value + s.Separator
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
lines := strings.Split(content, "\n")
|
|
for line := range lines {
|
|
result += strings.TrimRight(lines[line], ":")
|
|
}
|
|
|
|
// Print results
|
|
fmt.Println(result)
|
|
|
|
} else {
|
|
// Send results
|
|
content := fmt.Sprintf("%s: %s\n%s: %s\n%s: %s\n",
|
|
utils.Green("IP"), utils.Blue(results.Matches[match].IP.String()),
|
|
utils.Blue("Port"), utils.Green(results.Matches[match].Port),
|
|
utils.Green("Hostname"), utils.Blue(strings.Join(results.Matches[0].Hostnames, ",")),
|
|
)
|
|
|
|
lines := strings.Split(results.Matches[match].Data, "\n")
|
|
green := true
|
|
for line := range lines {
|
|
values := strings.Split(lines[line], ":")
|
|
if green {
|
|
content += fmt.Sprintf("%s:%s\n", utils.Green(values[0]), utils.Blue(strings.Join(values[1:], "")))
|
|
green = false
|
|
} else {
|
|
content += fmt.Sprintf("%s:%s\n", utils.Blue(values[0]), utils.Green(strings.Join(values[1:], "")))
|
|
green = true
|
|
}
|
|
}
|
|
|
|
s.Results <- content
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Todo: Change pages to limit, redo field filtering
|