darroyo
5d5b6c99a7
Fix Wake-on-LAN: set SO_BROADCAST on UDP socket, send 3 magic packets, expose broadcast_addr and wake timeout fields in UI, add Test Wake endpoint, surface send errors in job status, bump default wake timeout to 180s.
48 lines
956 B
Go
48 lines
956 B
Go
package wol
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
var ErrTimeout = fmt.Errorf("timeout waiting for machine to respond")
|
|
|
|
func WaitUntilReady(ctx context.Context, host string, sshPort int, maxWait, checkInterval time.Duration) error {
|
|
deadline := time.Now().Add(maxWait)
|
|
interval := checkInterval
|
|
|
|
ticker := time.NewTicker(checkInterval)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case <-ticker.C:
|
|
}
|
|
|
|
addr := fmt.Sprintf("%s:%d", host, sshPort)
|
|
dialer := net.Dialer{Timeout: 3 * time.Second}
|
|
conn, err := dialer.DialContext(ctx, "tcp", addr)
|
|
if err == nil {
|
|
conn.Close()
|
|
return nil
|
|
}
|
|
|
|
if time.Now().After(deadline) {
|
|
return ErrTimeout
|
|
}
|
|
|
|
elapsed := time.Since(time.Now().Add(-maxWait))
|
|
if elapsed > maxWait/2 && interval < 10*time.Second {
|
|
interval = interval * 3 / 2
|
|
if interval > 10*time.Second {
|
|
interval = 10 * time.Second
|
|
}
|
|
ticker.Reset(interval)
|
|
}
|
|
}
|
|
}
|