Compare commits
1 Commits
fix/67-rea
...
27d2a69026
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
27d2a69026 |
20
README.md
20
README.md
@@ -110,8 +110,8 @@ includes:
|
|||||||
- **NS recoveries**: Which nameserver recovered, which hostname/domain.
|
- **NS recoveries**: Which nameserver recovered, which hostname/domain.
|
||||||
- **NS inconsistencies**: Which nameservers disagree, what each one
|
- **NS inconsistencies**: Which nameservers disagree, what each one
|
||||||
returned, which hostname affected.
|
returned, which hostname affected.
|
||||||
- **Port changes**: Which IP:port, old state, new state, all associated
|
- **Port changes**: Which IP:port, old state, new state, associated
|
||||||
hostnames.
|
hostname.
|
||||||
- **TLS expiry warnings**: Which certificate, days remaining, CN,
|
- **TLS expiry warnings**: Which certificate, days remaining, CN,
|
||||||
issuer, associated hostname and IP.
|
issuer, associated hostname and IP.
|
||||||
- **TLS certificate changes**: Old and new CN/issuer/SANs, associated
|
- **TLS certificate changes**: Old and new CN/issuer/SANs, associated
|
||||||
@@ -290,12 +290,12 @@ not as a merged view, to enable inconsistency detection.
|
|||||||
"ports": {
|
"ports": {
|
||||||
"93.184.216.34:80": {
|
"93.184.216.34:80": {
|
||||||
"open": true,
|
"open": true,
|
||||||
"hostnames": ["www.example.com"],
|
"hostname": "www.example.com",
|
||||||
"lastChecked": "2026-02-19T12:00:00Z"
|
"lastChecked": "2026-02-19T12:00:00Z"
|
||||||
},
|
},
|
||||||
"93.184.216.34:443": {
|
"93.184.216.34:443": {
|
||||||
"open": true,
|
"open": true,
|
||||||
"hostnames": ["www.example.com"],
|
"hostname": "www.example.com",
|
||||||
"lastChecked": "2026-02-19T12:00:00Z"
|
"lastChecked": "2026-02-19T12:00:00Z"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
@@ -367,15 +367,9 @@ docker run -d \
|
|||||||
triggering change notifications).
|
triggering change notifications).
|
||||||
2. **Initial check**: Immediately perform all DNS, port, and TLS checks
|
2. **Initial check**: Immediately perform all DNS, port, and TLS checks
|
||||||
on startup.
|
on startup.
|
||||||
3. **Periodic checks** (DNS always runs first):
|
3. **Periodic checks**:
|
||||||
- DNS checks: every `DNSWATCHER_DNS_INTERVAL` (default 1h). Also
|
- DNS and port checks: every `DNSWATCHER_DNS_INTERVAL` (default 1h).
|
||||||
re-run before every TLS check cycle to ensure fresh IPs.
|
- TLS checks: every `DNSWATCHER_TLS_INTERVAL` (default 12h).
|
||||||
- Port checks: every `DNSWATCHER_DNS_INTERVAL`, after DNS completes.
|
|
||||||
- TLS checks: every `DNSWATCHER_TLS_INTERVAL` (default 12h), after
|
|
||||||
DNS completes.
|
|
||||||
- Port and TLS checks always use freshly resolved IP addresses from
|
|
||||||
the DNS phase that immediately precedes them — never stale IPs
|
|
||||||
from a previous cycle.
|
|
||||||
4. **On change detection**: Send notifications to all configured
|
4. **On change detection**: Send notifications to all configured
|
||||||
endpoints, update in-memory state, persist to disk.
|
endpoints, update in-memory state, persist to disk.
|
||||||
5. **Shutdown**: Persist final state to disk, complete in-flight
|
5. **Shutdown**: Persist final state to disk, complete in-flight
|
||||||
|
|||||||
@@ -1,60 +0,0 @@
|
|||||||
package handlers
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// domainResponse represents a single domain in the API response.
|
|
||||||
type domainResponse struct {
|
|
||||||
Domain string `json:"domain"`
|
|
||||||
Nameservers []string `json:"nameservers,omitempty"`
|
|
||||||
LastChecked string `json:"lastChecked,omitempty"`
|
|
||||||
Status string `json:"status"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// domainsResponse is the top-level response for GET /api/v1/domains.
|
|
||||||
type domainsResponse struct {
|
|
||||||
Domains []domainResponse `json:"domains"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// HandleDomains returns the configured domains and their status.
|
|
||||||
func (h *Handlers) HandleDomains() http.HandlerFunc {
|
|
||||||
return func(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
configured := h.config.Domains
|
|
||||||
snapshot := h.state.GetSnapshot()
|
|
||||||
|
|
||||||
domains := make(
|
|
||||||
[]domainResponse, 0, len(configured),
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, domain := range configured {
|
|
||||||
dr := domainResponse{
|
|
||||||
Domain: domain,
|
|
||||||
Status: "pending",
|
|
||||||
}
|
|
||||||
|
|
||||||
ds, ok := snapshot.Domains[domain]
|
|
||||||
if ok {
|
|
||||||
dr.Nameservers = ds.Nameservers
|
|
||||||
dr.Status = "ok"
|
|
||||||
|
|
||||||
if !ds.LastChecked.IsZero() {
|
|
||||||
dr.LastChecked = ds.LastChecked.
|
|
||||||
Format(time.RFC3339)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
domains = append(domains, dr)
|
|
||||||
}
|
|
||||||
|
|
||||||
h.respondJSON(
|
|
||||||
writer, request,
|
|
||||||
&domainsResponse{Domains: domains},
|
|
||||||
http.StatusOK,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -8,11 +8,9 @@ import (
|
|||||||
|
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
|
|
||||||
"sneak.berlin/go/dnswatcher/internal/config"
|
|
||||||
"sneak.berlin/go/dnswatcher/internal/globals"
|
"sneak.berlin/go/dnswatcher/internal/globals"
|
||||||
"sneak.berlin/go/dnswatcher/internal/healthcheck"
|
"sneak.berlin/go/dnswatcher/internal/healthcheck"
|
||||||
"sneak.berlin/go/dnswatcher/internal/logger"
|
"sneak.berlin/go/dnswatcher/internal/logger"
|
||||||
"sneak.berlin/go/dnswatcher/internal/state"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Params contains dependencies for Handlers.
|
// Params contains dependencies for Handlers.
|
||||||
@@ -22,8 +20,6 @@ type Params struct {
|
|||||||
Logger *logger.Logger
|
Logger *logger.Logger
|
||||||
Globals *globals.Globals
|
Globals *globals.Globals
|
||||||
Healthcheck *healthcheck.Healthcheck
|
Healthcheck *healthcheck.Healthcheck
|
||||||
State *state.State
|
|
||||||
Config *config.Config
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handlers provides HTTP request handlers.
|
// Handlers provides HTTP request handlers.
|
||||||
@@ -32,8 +28,6 @@ type Handlers struct {
|
|||||||
params *Params
|
params *Params
|
||||||
globals *globals.Globals
|
globals *globals.Globals
|
||||||
hc *healthcheck.Healthcheck
|
hc *healthcheck.Healthcheck
|
||||||
state *state.State
|
|
||||||
config *config.Config
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new Handlers instance.
|
// New creates a new Handlers instance.
|
||||||
@@ -43,8 +37,6 @@ func New(_ fx.Lifecycle, params Params) (*Handlers, error) {
|
|||||||
params: ¶ms,
|
params: ¶ms,
|
||||||
globals: params.Globals,
|
globals: params.Globals,
|
||||||
hc: params.Healthcheck,
|
hc: params.Healthcheck,
|
||||||
state: params.State,
|
|
||||||
config: params.Config,
|
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -52,7 +44,7 @@ func (h *Handlers) respondJSON(
|
|||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
_ *http.Request,
|
_ *http.Request,
|
||||||
data any,
|
data any,
|
||||||
status int, //nolint:unparam // general-purpose utility; status varies in future use
|
status int,
|
||||||
) {
|
) {
|
||||||
writer.Header().Set("Content-Type", "application/json")
|
writer.Header().Set("Content-Type", "application/json")
|
||||||
writer.WriteHeader(status)
|
writer.WriteHeader(status)
|
||||||
|
|||||||
@@ -1,120 +0,0 @@
|
|||||||
package handlers
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"sort"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"sneak.berlin/go/dnswatcher/internal/state"
|
|
||||||
)
|
|
||||||
|
|
||||||
// nameserverRecordResponse represents one nameserver's records
|
|
||||||
// in the API response.
|
|
||||||
type nameserverRecordResponse struct {
|
|
||||||
Nameserver string `json:"nameserver"`
|
|
||||||
Records map[string][]string `json:"records"`
|
|
||||||
Status string `json:"status"`
|
|
||||||
Error string `json:"error,omitempty"`
|
|
||||||
LastChecked string `json:"lastChecked,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// hostnameResponse represents a single hostname in the API response.
|
|
||||||
type hostnameResponse struct {
|
|
||||||
Hostname string `json:"hostname"`
|
|
||||||
Nameservers []nameserverRecordResponse `json:"nameservers,omitempty"`
|
|
||||||
LastChecked string `json:"lastChecked,omitempty"`
|
|
||||||
Status string `json:"status"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// hostnamesResponse is the top-level response for
|
|
||||||
// GET /api/v1/hostnames.
|
|
||||||
type hostnamesResponse struct {
|
|
||||||
Hostnames []hostnameResponse `json:"hostnames"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// HandleHostnames returns the configured hostnames and their status.
|
|
||||||
func (h *Handlers) HandleHostnames() http.HandlerFunc {
|
|
||||||
return func(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
) {
|
|
||||||
configured := h.config.Hostnames
|
|
||||||
snapshot := h.state.GetSnapshot()
|
|
||||||
|
|
||||||
hostnames := make(
|
|
||||||
[]hostnameResponse, 0, len(configured),
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, hostname := range configured {
|
|
||||||
hr := hostnameResponse{
|
|
||||||
Hostname: hostname,
|
|
||||||
Status: "pending",
|
|
||||||
}
|
|
||||||
|
|
||||||
hs, ok := snapshot.Hostnames[hostname]
|
|
||||||
if ok {
|
|
||||||
hr.Status = "ok"
|
|
||||||
|
|
||||||
if !hs.LastChecked.IsZero() {
|
|
||||||
hr.LastChecked = hs.LastChecked.
|
|
||||||
Format(time.RFC3339)
|
|
||||||
}
|
|
||||||
|
|
||||||
hr.Nameservers = buildNameserverRecords(
|
|
||||||
hs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
hostnames = append(hostnames, hr)
|
|
||||||
}
|
|
||||||
|
|
||||||
h.respondJSON(
|
|
||||||
writer, request,
|
|
||||||
&hostnamesResponse{Hostnames: hostnames},
|
|
||||||
http.StatusOK,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// buildNameserverRecords converts the per-nameserver state map
|
|
||||||
// into a sorted slice for deterministic JSON output.
|
|
||||||
func buildNameserverRecords(
|
|
||||||
hs *state.HostnameState,
|
|
||||||
) []nameserverRecordResponse {
|
|
||||||
if hs.RecordsByNameserver == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
nsNames := make(
|
|
||||||
[]string, 0, len(hs.RecordsByNameserver),
|
|
||||||
)
|
|
||||||
for ns := range hs.RecordsByNameserver {
|
|
||||||
nsNames = append(nsNames, ns)
|
|
||||||
}
|
|
||||||
|
|
||||||
sort.Strings(nsNames)
|
|
||||||
|
|
||||||
records := make(
|
|
||||||
[]nameserverRecordResponse, 0, len(nsNames),
|
|
||||||
)
|
|
||||||
|
|
||||||
for _, ns := range nsNames {
|
|
||||||
nsr := hs.RecordsByNameserver[ns]
|
|
||||||
|
|
||||||
entry := nameserverRecordResponse{
|
|
||||||
Nameserver: ns,
|
|
||||||
Records: nsr.Records,
|
|
||||||
Status: nsr.Status,
|
|
||||||
Error: nsr.Error,
|
|
||||||
}
|
|
||||||
|
|
||||||
if !nsr.LastChecked.IsZero() {
|
|
||||||
entry.LastChecked = nsr.LastChecked.
|
|
||||||
Format(time.RFC3339)
|
|
||||||
}
|
|
||||||
|
|
||||||
records = append(records, entry)
|
|
||||||
}
|
|
||||||
|
|
||||||
return records
|
|
||||||
}
|
|
||||||
@@ -28,8 +28,6 @@ func (s *Server) SetupRoutes() {
|
|||||||
// API v1 routes
|
// API v1 routes
|
||||||
s.router.Route("/api/v1", func(r chi.Router) {
|
s.router.Route("/api/v1", func(r chi.Router) {
|
||||||
r.Get("/status", s.handlers.HandleStatus())
|
r.Get("/status", s.handlers.HandleStatus())
|
||||||
r.Get("/domains", s.handlers.HandleDomains())
|
|
||||||
r.Get("/hostnames", s.handlers.HandleHostnames())
|
|
||||||
})
|
})
|
||||||
|
|
||||||
// Metrics endpoint (optional, with basic auth)
|
// Metrics endpoint (optional, with basic auth)
|
||||||
|
|||||||
@@ -57,47 +57,8 @@ type HostnameState struct {
|
|||||||
// PortState holds the monitoring state for a port.
|
// PortState holds the monitoring state for a port.
|
||||||
type PortState struct {
|
type PortState struct {
|
||||||
Open bool `json:"open"`
|
Open bool `json:"open"`
|
||||||
Hostnames []string `json:"hostnames"`
|
|
||||||
LastChecked time.Time `json:"lastChecked"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// UnmarshalJSON implements custom unmarshaling to handle both
|
|
||||||
// the old single-hostname format and the new multi-hostname
|
|
||||||
// format for backward compatibility with existing state files.
|
|
||||||
func (ps *PortState) UnmarshalJSON(data []byte) error {
|
|
||||||
// Use an alias to prevent infinite recursion.
|
|
||||||
type portStateAlias struct {
|
|
||||||
Open bool `json:"open"`
|
|
||||||
Hostnames []string `json:"hostnames"`
|
|
||||||
LastChecked time.Time `json:"lastChecked"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var alias portStateAlias
|
|
||||||
|
|
||||||
err := json.Unmarshal(data, &alias)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("unmarshaling port state: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
ps.Open = alias.Open
|
|
||||||
ps.Hostnames = alias.Hostnames
|
|
||||||
ps.LastChecked = alias.LastChecked
|
|
||||||
|
|
||||||
// If Hostnames is empty, try reading the old single-hostname
|
|
||||||
// format for backward compatibility.
|
|
||||||
if len(ps.Hostnames) == 0 {
|
|
||||||
var old struct {
|
|
||||||
Hostname string `json:"hostname"`
|
Hostname string `json:"hostname"`
|
||||||
}
|
LastChecked time.Time `json:"lastChecked"`
|
||||||
|
|
||||||
// Best-effort: ignore errors since the main unmarshal
|
|
||||||
// already succeeded.
|
|
||||||
if json.Unmarshal(data, &old) == nil && old.Hostname != "" {
|
|
||||||
ps.Hostnames = []string{old.Hostname}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// CertificateState holds TLS certificate monitoring state.
|
// CertificateState holds TLS certificate monitoring state.
|
||||||
@@ -302,27 +263,6 @@ func (s *State) GetPortState(key string) (*PortState, bool) {
|
|||||||
return ps, ok
|
return ps, ok
|
||||||
}
|
}
|
||||||
|
|
||||||
// DeletePortState removes a port state entry.
|
|
||||||
func (s *State) DeletePortState(key string) {
|
|
||||||
s.mu.Lock()
|
|
||||||
defer s.mu.Unlock()
|
|
||||||
|
|
||||||
delete(s.snapshot.Ports, key)
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetAllPortKeys returns all port state keys.
|
|
||||||
func (s *State) GetAllPortKeys() []string {
|
|
||||||
s.mu.RLock()
|
|
||||||
defer s.mu.RUnlock()
|
|
||||||
|
|
||||||
keys := make([]string, 0, len(s.snapshot.Ports))
|
|
||||||
for k := range s.snapshot.Ports {
|
|
||||||
keys = append(keys, k)
|
|
||||||
}
|
|
||||||
|
|
||||||
return keys
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetCertificateState updates the state for a certificate.
|
// SetCertificateState updates the state for a certificate.
|
||||||
func (s *State) SetCertificateState(
|
func (s *State) SetCertificateState(
|
||||||
key string,
|
key string,
|
||||||
|
|||||||
@@ -143,16 +143,9 @@ func (w *Watcher) Run(ctx context.Context) {
|
|||||||
|
|
||||||
return
|
return
|
||||||
case <-dnsTicker.C:
|
case <-dnsTicker.C:
|
||||||
w.runDNSChecks(ctx)
|
w.runDNSAndPortChecks(ctx)
|
||||||
|
|
||||||
w.checkAllPorts(ctx)
|
|
||||||
w.saveState()
|
w.saveState()
|
||||||
case <-tlsTicker.C:
|
case <-tlsTicker.C:
|
||||||
// Run DNS first so TLS checks use freshly
|
|
||||||
// resolved IP addresses, not stale ones from
|
|
||||||
// a previous cycle.
|
|
||||||
w.runDNSChecks(ctx)
|
|
||||||
|
|
||||||
w.runTLSChecks(ctx)
|
w.runTLSChecks(ctx)
|
||||||
w.saveState()
|
w.saveState()
|
||||||
}
|
}
|
||||||
@@ -160,26 +153,10 @@ func (w *Watcher) Run(ctx context.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// RunOnce performs a single complete monitoring cycle.
|
// RunOnce performs a single complete monitoring cycle.
|
||||||
// DNS checks run first so that port and TLS checks use
|
|
||||||
// freshly resolved IP addresses. Port checks run before
|
|
||||||
// TLS because TLS checks only target IPs with an open
|
|
||||||
// port 443.
|
|
||||||
func (w *Watcher) RunOnce(ctx context.Context) {
|
func (w *Watcher) RunOnce(ctx context.Context) {
|
||||||
w.detectFirstRun()
|
w.detectFirstRun()
|
||||||
|
w.runDNSAndPortChecks(ctx)
|
||||||
// Phase 1: DNS resolution must complete first so that
|
|
||||||
// subsequent checks use fresh IP addresses.
|
|
||||||
w.runDNSChecks(ctx)
|
|
||||||
|
|
||||||
// Phase 2: Port checks populate port state that TLS
|
|
||||||
// checks depend on (TLS only targets IPs where port
|
|
||||||
// 443 is open).
|
|
||||||
w.checkAllPorts(ctx)
|
|
||||||
|
|
||||||
// Phase 3: TLS checks use fresh DNS IPs and current
|
|
||||||
// port state.
|
|
||||||
w.runTLSChecks(ctx)
|
w.runTLSChecks(ctx)
|
||||||
|
|
||||||
w.saveState()
|
w.saveState()
|
||||||
w.firstRun = false
|
w.firstRun = false
|
||||||
}
|
}
|
||||||
@@ -196,11 +173,7 @@ func (w *Watcher) detectFirstRun() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// runDNSChecks performs DNS resolution for all configured domains
|
func (w *Watcher) runDNSAndPortChecks(ctx context.Context) {
|
||||||
// and hostnames, updating state with freshly resolved records.
|
|
||||||
// This must complete before port or TLS checks run so those
|
|
||||||
// checks operate on current IP addresses.
|
|
||||||
func (w *Watcher) runDNSChecks(ctx context.Context) {
|
|
||||||
for _, domain := range w.config.Domains {
|
for _, domain := range w.config.Domains {
|
||||||
w.checkDomain(ctx, domain)
|
w.checkDomain(ctx, domain)
|
||||||
}
|
}
|
||||||
@@ -208,6 +181,8 @@ func (w *Watcher) runDNSChecks(ctx context.Context) {
|
|||||||
for _, hostname := range w.config.Hostnames {
|
for _, hostname := range w.config.Hostnames {
|
||||||
w.checkHostname(ctx, hostname)
|
w.checkHostname(ctx, hostname)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
w.checkAllPorts(ctx)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (w *Watcher) checkDomain(
|
func (w *Watcher) checkDomain(
|
||||||
@@ -475,94 +450,24 @@ func (w *Watcher) detectInconsistencies(
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (w *Watcher) checkAllPorts(ctx context.Context) {
|
func (w *Watcher) checkAllPorts(ctx context.Context) {
|
||||||
// Phase 1: Build current IP:port → hostname associations
|
for _, hostname := range w.config.Hostnames {
|
||||||
// from fresh DNS data.
|
w.checkPortsForHostname(ctx, hostname)
|
||||||
associations := w.buildPortAssociations()
|
|
||||||
|
|
||||||
// Phase 2: Check each unique IP:port and update state
|
|
||||||
// with the full set of associated hostnames.
|
|
||||||
for key, hostnames := range associations {
|
|
||||||
ip, port := parsePortKey(key)
|
|
||||||
if port == 0 {
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
|
|
||||||
w.checkSinglePort(ctx, ip, port, hostnames)
|
for _, domain := range w.config.Domains {
|
||||||
|
w.checkPortsForHostname(ctx, domain)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Phase 3: Remove port state entries that no longer have
|
func (w *Watcher) checkPortsForHostname(
|
||||||
// any hostname referencing them.
|
ctx context.Context,
|
||||||
w.cleanupStalePorts(associations)
|
hostname string,
|
||||||
}
|
) {
|
||||||
|
ips := w.collectIPs(hostname)
|
||||||
|
|
||||||
// buildPortAssociations constructs a map from IP:port keys to
|
|
||||||
// the sorted set of hostnames currently resolving to that IP.
|
|
||||||
func (w *Watcher) buildPortAssociations() map[string][]string {
|
|
||||||
assoc := make(map[string]map[string]bool)
|
|
||||||
|
|
||||||
allNames := make(
|
|
||||||
[]string, 0,
|
|
||||||
len(w.config.Hostnames)+len(w.config.Domains),
|
|
||||||
)
|
|
||||||
allNames = append(allNames, w.config.Hostnames...)
|
|
||||||
allNames = append(allNames, w.config.Domains...)
|
|
||||||
|
|
||||||
for _, name := range allNames {
|
|
||||||
ips := w.collectIPs(name)
|
|
||||||
for _, ip := range ips {
|
for _, ip := range ips {
|
||||||
for _, port := range monitoredPorts {
|
for _, port := range monitoredPorts {
|
||||||
key := fmt.Sprintf("%s:%d", ip, port)
|
w.checkSinglePort(ctx, ip, port, hostname)
|
||||||
if assoc[key] == nil {
|
|
||||||
assoc[key] = make(map[string]bool)
|
|
||||||
}
|
|
||||||
|
|
||||||
assoc[key][name] = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
result := make(map[string][]string, len(assoc))
|
|
||||||
for key, set := range assoc {
|
|
||||||
hostnames := make([]string, 0, len(set))
|
|
||||||
for h := range set {
|
|
||||||
hostnames = append(hostnames, h)
|
|
||||||
}
|
|
||||||
|
|
||||||
sort.Strings(hostnames)
|
|
||||||
|
|
||||||
result[key] = hostnames
|
|
||||||
}
|
|
||||||
|
|
||||||
return result
|
|
||||||
}
|
|
||||||
|
|
||||||
// parsePortKey splits an "ip:port" key into its components.
|
|
||||||
func parsePortKey(key string) (string, int) {
|
|
||||||
lastColon := strings.LastIndex(key, ":")
|
|
||||||
if lastColon < 0 {
|
|
||||||
return key, 0
|
|
||||||
}
|
|
||||||
|
|
||||||
ip := key[:lastColon]
|
|
||||||
|
|
||||||
var p int
|
|
||||||
|
|
||||||
_, err := fmt.Sscanf(key[lastColon+1:], "%d", &p)
|
|
||||||
if err != nil {
|
|
||||||
return ip, 0
|
|
||||||
}
|
|
||||||
|
|
||||||
return ip, p
|
|
||||||
}
|
|
||||||
|
|
||||||
// cleanupStalePorts removes port state entries that are no
|
|
||||||
// longer referenced by any hostname in the current DNS data.
|
|
||||||
func (w *Watcher) cleanupStalePorts(
|
|
||||||
currentAssociations map[string][]string,
|
|
||||||
) {
|
|
||||||
for _, key := range w.state.GetAllPortKeys() {
|
|
||||||
if _, exists := currentAssociations[key]; !exists {
|
|
||||||
w.state.DeletePortState(key)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -599,7 +504,7 @@ func (w *Watcher) checkSinglePort(
|
|||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
ip string,
|
ip string,
|
||||||
port int,
|
port int,
|
||||||
hostnames []string,
|
hostname string,
|
||||||
) {
|
) {
|
||||||
result, err := w.portCheck.CheckPort(ctx, ip, port)
|
result, err := w.portCheck.CheckPort(ctx, ip, port)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -624,8 +529,8 @@ func (w *Watcher) checkSinglePort(
|
|||||||
}
|
}
|
||||||
|
|
||||||
msg := fmt.Sprintf(
|
msg := fmt.Sprintf(
|
||||||
"Hosts: %s\nAddress: %s\nPort now %s",
|
"Host: %s\nAddress: %s\nPort now %s",
|
||||||
strings.Join(hostnames, ", "), key, stateStr,
|
hostname, key, stateStr,
|
||||||
)
|
)
|
||||||
|
|
||||||
w.notify.SendNotification(
|
w.notify.SendNotification(
|
||||||
@@ -638,7 +543,7 @@ func (w *Watcher) checkSinglePort(
|
|||||||
|
|
||||||
w.state.SetPortState(key, &state.PortState{
|
w.state.SetPortState(key, &state.PortState{
|
||||||
Open: result.Open,
|
Open: result.Open,
|
||||||
Hostnames: hostnames,
|
Hostname: hostname,
|
||||||
LastChecked: now,
|
LastChecked: now,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -682,80 +682,6 @@ func TestGracefulShutdown(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func setupHostnameIP(
|
|
||||||
deps *testDeps,
|
|
||||||
hostname, ip string,
|
|
||||||
) {
|
|
||||||
deps.resolver.allRecords[hostname] = map[string]map[string][]string{
|
|
||||||
"ns1.example.com.": {"A": {ip}},
|
|
||||||
}
|
|
||||||
deps.portChecker.results[ip+":80"] = true
|
|
||||||
deps.portChecker.results[ip+":443"] = true
|
|
||||||
deps.tlsChecker.certs[ip+":"+hostname] = &tlscheck.CertificateInfo{
|
|
||||||
CommonName: hostname,
|
|
||||||
Issuer: "DigiCert",
|
|
||||||
NotAfter: time.Now().Add(90 * 24 * time.Hour),
|
|
||||||
SubjectAlternativeNames: []string{hostname},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func updateHostnameIP(deps *testDeps, hostname, ip string) {
|
|
||||||
deps.resolver.mu.Lock()
|
|
||||||
deps.resolver.allRecords[hostname] = map[string]map[string][]string{
|
|
||||||
"ns1.example.com.": {"A": {ip}},
|
|
||||||
}
|
|
||||||
deps.resolver.mu.Unlock()
|
|
||||||
|
|
||||||
deps.portChecker.mu.Lock()
|
|
||||||
deps.portChecker.results[ip+":80"] = true
|
|
||||||
deps.portChecker.results[ip+":443"] = true
|
|
||||||
deps.portChecker.mu.Unlock()
|
|
||||||
|
|
||||||
deps.tlsChecker.mu.Lock()
|
|
||||||
deps.tlsChecker.certs[ip+":"+hostname] = &tlscheck.CertificateInfo{
|
|
||||||
CommonName: hostname,
|
|
||||||
Issuer: "DigiCert",
|
|
||||||
NotAfter: time.Now().Add(90 * 24 * time.Hour),
|
|
||||||
SubjectAlternativeNames: []string{hostname},
|
|
||||||
}
|
|
||||||
deps.tlsChecker.mu.Unlock()
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestDNSRunsBeforePortAndTLSChecks(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
cfg := defaultTestConfig(t)
|
|
||||||
cfg.Hostnames = []string{"www.example.com"}
|
|
||||||
|
|
||||||
w, deps := newTestWatcher(t, cfg)
|
|
||||||
|
|
||||||
setupHostnameIP(deps, "www.example.com", "10.0.0.1")
|
|
||||||
|
|
||||||
ctx := t.Context()
|
|
||||||
w.RunOnce(ctx)
|
|
||||||
|
|
||||||
snap := deps.state.GetSnapshot()
|
|
||||||
if _, ok := snap.Ports["10.0.0.1:80"]; !ok {
|
|
||||||
t.Fatal("expected port state for 10.0.0.1:80")
|
|
||||||
}
|
|
||||||
|
|
||||||
// DNS changes to a new IP; port and TLS must pick it up.
|
|
||||||
updateHostnameIP(deps, "www.example.com", "10.0.0.2")
|
|
||||||
|
|
||||||
w.RunOnce(ctx)
|
|
||||||
|
|
||||||
snap = deps.state.GetSnapshot()
|
|
||||||
|
|
||||||
if _, ok := snap.Ports["10.0.0.2:80"]; !ok {
|
|
||||||
t.Error("port check used stale DNS: missing 10.0.0.2:80")
|
|
||||||
}
|
|
||||||
|
|
||||||
certKey := "10.0.0.2:443:www.example.com"
|
|
||||||
if _, ok := snap.Certificates[certKey]; !ok {
|
|
||||||
t.Error("TLS check used stale DNS: missing " + certKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestNSFailureAndRecovery(t *testing.T) {
|
func TestNSFailureAndRecovery(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user