pangolin_mirror/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2026-03-03 17:26:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add round trip tracking for any message

Browse Source
This commit is contained in:
Owen
2026-02-16 20:29:19 -08:00
committed by Owen Schwartz
parent 5b884042cd
commit d98eaa88b3
1 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
main.go
View File

@@ -1329,7 +1329,7 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
// Define the structure of the incoming message // Define the structure of the incoming message
type SSHCertData struct { type SSHCertData struct {
TraceID string `json:"traceId"` MessageId string `json:"messageId"`
AgentPort int `json:"agentPort"` AgentPort int `json:"agentPort"`
AgentHost string `json:"agentHost"` AgentHost string `json:"agentHost"`
CACert string `json:"caCert"` CACert string `json:"caCert"`
@@ -1357,9 +1357,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
if authDaemonKey == "" { if authDaemonKey == "" {
logger.Error("Auth daemon key not configured, cannot process SSH certificate") logger.Error("Auth daemon key not configured, cannot process SSH certificate")
// Send failure response back to cloud // Send failure response back to cloud
err := client.SendMessage("newt/pam/connection/response", map[string]interface{}{ err := client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": false, "complete": true,
"error": "auth daemon key not configured", "error": "auth daemon key not configured",
}) })
if err != nil { if err != nil {
@@ -1383,9 +1383,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
if err != nil { if err != nil {
logger.Error("Failed to marshal auth daemon request: %v", err) logger.Error("Failed to marshal auth daemon request: %v", err)
// Send failure response // Send failure response
client.SendMessage("newt/pam/ssh-cert-response", map[string]interface{}{ client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": false, "complete": true,
"error": fmt.Sprintf("failed to marshal request: %v", err), "error": fmt.Sprintf("failed to marshal request: %v", err),
}) })
return return
@@ -1407,9 +1407,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
req, err := http.NewRequest("POST", url, bytes.NewBuffer(requestJSON)) req, err := http.NewRequest("POST", url, bytes.NewBuffer(requestJSON))
if err != nil { if err != nil {
logger.Error("Failed to create auth daemon request: %v", err) logger.Error("Failed to create auth daemon request: %v", err)
client.SendMessage("newt/pam/connection/response", map[string]interface{}{ client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": false, "complete": true,
"error": fmt.Sprintf("failed to create request: %v", err), "error": fmt.Sprintf("failed to create request: %v", err),
}) })
return return
@@ -1425,9 +1425,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
resp, err := httpClient.Do(req) resp, err := httpClient.Do(req)
if err != nil { if err != nil {
logger.Error("Failed to connect to auth daemon: %v", err) logger.Error("Failed to connect to auth daemon: %v", err)
client.SendMessage("newt/pam/connection/response", map[string]interface{}{ client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": false, "complete": true,
"error": fmt.Sprintf("failed to connect to auth daemon: %v", err), "error": fmt.Sprintf("failed to connect to auth daemon: %v", err),
}) })
return return
@@ -1437,9 +1437,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
// Check response status // Check response status
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
logger.Error("Auth daemon returned non-OK status: %d", resp.StatusCode) logger.Error("Auth daemon returned non-OK status: %d", resp.StatusCode)
client.SendMessage("newt/pam/connection/response", map[string]interface{}{ client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": false, "complete": true,
"error": fmt.Sprintf("auth daemon returned status %d", resp.StatusCode), "error": fmt.Sprintf("auth daemon returned status %d", resp.StatusCode),
}) })
return return
@@ -1448,9 +1448,9 @@ persistent_keepalive_interval=5`, util.FixKey(privateKey.String()), util.FixKey(
logger.Info("Successfully registered SSH certificate with auth daemon for user %s", certData.Username) logger.Info("Successfully registered SSH certificate with auth daemon for user %s", certData.Username)
// Send success response back to cloud // Send success response back to cloud
err = client.SendMessage("newt/pam/connection/response", map[string]interface{}{ err = client.SendMessage("ws/round-trip/complete", map[string]interface{}{
"traceId": certData.TraceID, "messageId": certData.MessageId,
"success": true, "complete": true,
}) })
if err != nil { if err != nil {
logger.Error("Failed to send SSH cert success response: %v", err) logger.Error("Failed to send SSH cert success response: %v", err)