eduard256
3fec89be7f
- Add HTTPProber: parallel HEAD+GET on ports 80/8080, extracts Server header - Reduce mDNS timeout from 1s to 100ms using context wrapper around mdns.Query (HomeKit devices respond in 2-10ms, no need to wait 1s) - Add Trassir (F0:23:B9) and ZOSI (00:05:FE) to camera OUI database - Probe response time improved from ~1s to ~110ms for reachable devices
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
package discovery
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/eduard256/Strix/internal/models"
|
|
)
|
|
|
|
// HTTPProber identifies the device by checking HTTP server headers.
|
|
// It sends HEAD and GET requests in parallel to port 80 (some devices
|
|
// like XMEye/JAWS don't respond to HEAD), and returns whichever
|
|
// responds first.
|
|
type HTTPProber struct{}
|
|
|
|
func (p *HTTPProber) Name() string { return "http" }
|
|
|
|
// Probe sends parallel HEAD+GET to port 80 and extracts Server header.
|
|
// Returns nil if no HTTP server is found.
|
|
func (p *HTTPProber) Probe(ctx context.Context, ip string) (any, error) {
|
|
ports := []int{80, 8080}
|
|
|
|
client := &http.Client{
|
|
// Don't follow redirects -- we want the original response headers
|
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
},
|
|
Transport: &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
},
|
|
}
|
|
|
|
type result struct {
|
|
resp *http.Response
|
|
port int
|
|
err error
|
|
}
|
|
|
|
for _, port := range ports {
|
|
url := fmt.Sprintf("http://%s:%d/", ip, port)
|
|
ch := make(chan result, 2)
|
|
|
|
// HEAD and GET in parallel -- take whichever responds first
|
|
for _, method := range []string{"HEAD", "GET"} {
|
|
go func(method string) {
|
|
req, err := http.NewRequestWithContext(ctx, method, url, nil)
|
|
if err != nil {
|
|
ch <- result{err: err}
|
|
return
|
|
}
|
|
req.Header.Set("User-Agent", "Strix/1.0")
|
|
resp, err := client.Do(req)
|
|
ch <- result{resp: resp, port: port, err: err}
|
|
}(method)
|
|
}
|
|
|
|
// Wait for first success
|
|
for i := 0; i < 2; i++ {
|
|
select {
|
|
case <-ctx.Done():
|
|
return nil, ctx.Err()
|
|
case r := <-ch:
|
|
if r.err != nil {
|
|
continue
|
|
}
|
|
if r.resp.Body != nil {
|
|
r.resp.Body.Close()
|
|
}
|
|
|
|
server := r.resp.Header.Get("Server")
|
|
if server == "" && r.resp.StatusCode == 0 {
|
|
continue
|
|
}
|
|
|
|
return &models.HTTPProbeResult{
|
|
Port: r.port,
|
|
StatusCode: r.resp.StatusCode,
|
|
Server: server,
|
|
}, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil, nil
|
|
}
|