Claude
fd5e2cc761
The hub accepted a WebSocket from anyone on /node/v1, a path in the open code, and replied with a hello nonce: one connection was enough to tell a scanner what it had found. The agent now signs the upgrade request itself (X-Node-Id, X-Node-Ts, X-Node-Sig; on first binding also the join secret and its key), and the hub checks it before Accept. Without a valid signature it answers Go's plain "404 page not found". Inside the socket the protocol is unchanged, and the auth frame must come from the node that signed the request. Older agents send no signature. The hub lets them in while some node has never connected the new way (nodes.preauth), logs them as outdated, and closes that door by itself once every node is updated; HUB_LEGACY=on/off overrides. The timestamp window is a day: node clocks drift, and the headers travel inside pinned TLS. The agent treats a 404 at upgrade as a rejection and backs off for a minute. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_013bNGYiDst62wH3ZGTRPFLJ
90 lines
4.1 KiB
Go
90 lines
4.1 KiB
Go
package ui
|
||
|
||
// Подпись агента ДО апгрейда на WebSocket (stealth хаба нод). Раньше хаб принимал WebSocket у
|
||
// любого и сразу слал hello с nonce: путь /node/v1 есть в открытом коде, и сканер, знающий
|
||
// протокол, узнавал хаб одним подключением. Теперь агент кладёт подпись своим ключом в заголовки
|
||
// запроса, хаб проверяет её до Accept и без неё отвечает 404 пустого Go-сервера.
|
||
//
|
||
// Переходный режим: агенты до этой версии заголовков не шлют. Их пускают по-старому, пока в
|
||
// панели есть нода, ни разу не подключавшаяся с подписью (nodes.preauth = 0): обновили все —
|
||
// путь без подписи закрывается сам. HUB_LEGACY=on / off — принудительно.
|
||
//
|
||
// Внутри WebSocket протокол прежний (hello → auth → ready): кадр auth обязан быть от той же ноды.
|
||
|
||
import (
|
||
"crypto/ed25519"
|
||
"crypto/sha256"
|
||
"crypto/subtle"
|
||
"net/http"
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
// Часы ноды могут расходиться с панелью: окно широкое. Повтор подписи не страшен — заголовки
|
||
// идут внутри TLS с закреплённым сертификатом, подсмотреть их нельзя.
|
||
const preauthWindow = 24 * 3600
|
||
|
||
// preauthMsg — что агент подписывает в заголовке X-Node-Sig. Дублируется в internal/agent.
|
||
func preauthMsg(nodeID, ts int64) []byte {
|
||
return []byte("awg-node-pre-v1\n" + strconv.FormatInt(nodeID, 10) + "\n" + strconv.FormatInt(ts, 10))
|
||
}
|
||
|
||
// joinOK — одноразовый секрет первой привязки ноды верен и не просрочен.
|
||
func joinOK(row *nodeRow, secret string) bool {
|
||
if row.JoinHash == nil || !row.JoinExpires.Valid || row.JoinExpires.Int64 == 0 || row.JoinExpires.Int64 < now() || secret == "" {
|
||
return false
|
||
}
|
||
h := sha256.Sum256(fromB64url(secret))
|
||
return len(row.JoinHash) == len(h) && subtle.ConstantTimeCompare(h[:], row.JoinHash) == 1
|
||
}
|
||
|
||
// preauth — подпись из заголовков. present — агент её прислал (новая версия); row — нода, если
|
||
// подпись верна.
|
||
func (n *Nodes) preauth(r *http.Request) (row *nodeRow, present bool) {
|
||
idS, tsS, sig := r.Header.Get("X-Node-Id"), r.Header.Get("X-Node-Ts"), r.Header.Get("X-Node-Sig")
|
||
if idS == "" && tsS == "" && sig == "" {
|
||
return nil, false
|
||
}
|
||
id, err1 := strconv.ParseInt(idS, 10, 64)
|
||
ts, err2 := strconv.ParseInt(tsS, 10, 64)
|
||
if err1 != nil || err2 != nil || sig == "" {
|
||
return nil, true
|
||
}
|
||
if d := ts - now(); d > preauthWindow || d < -preauthWindow {
|
||
warnf("nodes: подпись ноды #%d вне окна: часы расходятся на %d с", id, d)
|
||
return nil, true
|
||
}
|
||
row = n.row(id)
|
||
if row == nil {
|
||
return nil, true
|
||
}
|
||
pub := row.Pub.String
|
||
if !row.Pub.Valid {
|
||
// Первое подключение: ключа ноды ещё нет, его приносит агент вместе с секретом привязки.
|
||
if !joinOK(row, r.Header.Get("X-Node-Join")) {
|
||
return nil, true
|
||
}
|
||
pub = r.Header.Get("X-Node-Pub")
|
||
}
|
||
key := fromB64url(pub)
|
||
if len(key) != ed25519.PublicKeySize || !ed25519.Verify(key, preauthMsg(id, ts), fromB64url(sig)) {
|
||
return nil, true
|
||
}
|
||
return row, true
|
||
}
|
||
|
||
// legacyOpen — пускать ли подключение без подписи (агент старой версии).
|
||
func (n *Nodes) legacyOpen() bool {
|
||
switch strings.ToLower(os.Getenv("HUB_LEGACY")) {
|
||
case "on":
|
||
return true
|
||
case "off":
|
||
return false
|
||
}
|
||
var waiting int
|
||
n.a.db.QueryRow(`SELECT COUNT(*) FROM nodes WHERE preauth = 0
|
||
AND (pub IS NOT NULL OR (join_hash IS NOT NULL AND join_expires >= ?))`, now()).Scan(&waiting)
|
||
return waiting > 0
|
||
}
|