pangolin_mirror/olm
1
0
Fork 0
mirror of https://github.com/fosrl/olm.git synced 2026-03-06 18:56:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ping): send fingerprint and posture checks as part of ping/register

Browse Source
This commit is contained in:
Varun Narravula
2026-01-08 20:40:41 -08:00
committed by Owen Schwartz
parent 566084683a
commit 70a7e83291
2 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
olm/olm.go
View File

@@ -362,6 +362,8 @@ func (o *Olm) StartTunnel(config TunnelConfig) {
"olmAgent": o.olmConfig.Agent, "olmAgent": o.olmConfig.Agent,
"orgId": config.OrgID, "orgId": config.OrgID,
"userToken": userToken, "userToken": userToken,
"fingerprint": o.fingerprint,
"postures": o.postures,
}, 1*time.Second) }, 1*time.Second)
// Invoke onRegistered callback if configured // Invoke onRegistered callback if configured

8
olm/ping.go
View File

@@ -8,10 +8,12 @@ import (
"github.com/fosrl/olm/websocket" "github.com/fosrl/olm/websocket"
) )
func sendPing(olm *websocket.Client) error { func (o *Olm) sendPing(olm *websocket.Client) error {
err := olm.SendMessage("olm/ping", map[string]any{ err := olm.SendMessage("olm/ping", map[string]any{
"timestamp": time.Now().Unix(), "timestamp": time.Now().Unix(),
"userToken": olm.GetConfig().UserToken, "userToken": olm.GetConfig().UserToken,
"fingerprint": o.fingerprint,
"postures": o.postures,
}) })
if err != nil { if err != nil {
logger.Error("Failed to send ping message: %v", err) logger.Error("Failed to send ping message: %v", err)
@@ -23,7 +25,7 @@ func sendPing(olm *websocket.Client) error {
func (o *Olm) keepSendingPing(olm *websocket.Client) { func (o *Olm) keepSendingPing(olm *websocket.Client) {
// Send ping immediately on startup // Send ping immediately on startup
if err := sendPing(olm); err != nil { if err := o.sendPing(olm); err != nil {
logger.Error("Failed to send initial ping: %v", err) logger.Error("Failed to send initial ping: %v", err)
} else { } else {
logger.Info("Sent initial ping message") logger.Info("Sent initial ping message")
@@ -39,7 +41,7 @@ func (o *Olm) keepSendingPing(olm *websocket.Client) {
logger.Info("Stopping ping messages") logger.Info("Stopping ping messages")
return return
case <-ticker.C: case <-ticker.C:
if err := sendPing(olm); err != nil { if err := o.sendPing(olm); err != nil {
logger.Error("Failed to send periodic ping: %v", err) logger.Error("Failed to send periodic ping: %v", err)
} }
} }