include arch and os in local server headers

ee/localserver/krypto-ec-middleware.go

@@ -25,13 +25,15 @@ import (
 )
 
 const (
-	timestampValidityRange                   = 150
-	kolideKryptoEccHeader20230130Value       = "2023-01-30"
-	kolideKryptoHeaderKey                    = "X-Kolide-Krypto"
-	kolideSessionIdHeaderKey                 = "X-Kolide-Session"
-	kolidePresenceDetectionInterval          = "X-Kolide-Presence-Detection-Interval"
-	kolidePresenceDetectionReason            = "X-Kolide-Presence-Detection-Reason"
-	kolideDurationSinceLastPresenceDetection = "X-Kolide-Duration-Since-Last-Presence-Detection"
+	timestampValidityRange                            = 150
+	kolideKryptoEccHeader20230130Value                = "2023-01-30"
+	kolideKryptoHeaderKey                             = "X-Kolide-Krypto"
+	kolideSessionIdHeaderKey                          = "X-Kolide-Session"
+	kolidePresenceDetectionIntervalHeaderKey          = "X-Kolide-Presence-Detection-Interval"
+	kolidePresenceDetectionReasonHeaderKey            = "X-Kolide-Presence-Detection-Reason"
+	kolideDurationSinceLastPresenceDetectionHeaderKey = "X-Kolide-Duration-Since-Last-Presence-Detection"
+	kolideOsHeaderKey                                 = "X-Kolide-Os"
+	kolideArchHeaderKey                               = "X-Kolide-Arch"
 )
 
 type v2CmdRequestType struct {
@@ -316,6 +318,9 @@ func (e *kryptoEcMiddleware) Wrap(next http.Handler) http.Handler {
 		bhr := &bufferedHttpResponse{}
 		next.ServeHTTP(bhr, newReq)
 
+		bhr.Header().Add(kolideOsHeaderKey, runtime.GOOS)
+		bhr.Header().Add(kolideArchHeaderKey, runtime.GOARCH)
+
 		// add headers to the response map
 		// this assumes that the response to `bhr` was a json encoded blob.
 		var responseMap map[string]interface{}

ee/localserver/krypto-ec-middleware_test.go

@@ -24,6 +24,7 @@ import (
 	"github.com/kolide/krypto/pkg/echelper"
 	"github.com/kolide/launcher/ee/agent/keys"
 	"github.com/kolide/launcher/ee/localserver/mocks"
+	"github.com/kolide/launcher/ee/presencedetection"
 
 	"github.com/kolide/launcher/pkg/log/multislogger"
 	"github.com/stretchr/testify/assert"
@@ -42,7 +43,7 @@ func TestKryptoEcMiddleware(t *testing.T) {
 
 	koldieSessionId := ulid.New()
 	cmdRequestHeaders := map[string][]string{
-		kolidePresenceDetectionInterval: {"0s"},
+		kolidePresenceDetectionIntervalHeaderKey: {"0s"},
 	}
 
 	cmdReqCallBackHeaders := map[string][]string{
@@ -240,10 +241,16 @@ func TestKryptoEcMiddleware(t *testing.T) {
 					responseHeaders, err := extractJsonProperty[map[string][]string](opened.ResponseData, "headers")
 					require.NoError(t, err)
 
+					require.Equal(t, runtime.GOOS, responseHeaders[kolideOsHeaderKey][0])
+
 					// check that the presence detection interval is present
 					if runtime.GOOS == "darwin" {
-						require.Equal(t, (0 * time.Second).String(), responseHeaders[kolideDurationSinceLastPresenceDetection][0])
+						require.Equal(t, (0 * time.Second).String(), responseHeaders[kolideDurationSinceLastPresenceDetectionHeaderKey][0])
+						return
 					}
+
+					// not darwin
+					require.Equal(t, presencedetection.DetectionFailedDurationValue.String(), responseHeaders[kolideDurationSinceLastPresenceDetectionHeaderKey][0])
 				})
 			}
 		})

ee/localserver/presence-detection-middleware_test.go

@@ -109,7 +109,7 @@ func TestPresenceDetectionHandler(t *testing.T) {
 			handlerToTest.ServeHTTP(rr, req)
 
 			if tt.shouldHavePresenceDetectionDurationResponseHeader {
-				require.NotEmpty(t, rr.Header().Get(kolideDurationSinceLastPresenceDetection))
+				require.NotEmpty(t, rr.Header().Get(kolideDurationSinceLastPresenceDetectionHeaderKey))
 			}
 			require.Equal(t, tt.expectedStatusCode, rr.Code)
 		})

ee/localserver/server.go

@@ -20,6 +20,7 @@ import (
 	"github.com/kolide/krypto/pkg/echelper"
 	"github.com/kolide/launcher/ee/agent"
 	"github.com/kolide/launcher/ee/agent/types"
+	"github.com/kolide/launcher/ee/presencedetection"
 	"github.com/kolide/launcher/pkg/osquery"
 	"github.com/kolide/launcher/pkg/traces"
 	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
@@ -415,13 +416,14 @@ func (ls *localServer) presenceDetectionHandler(next http.Handler) http.Handler
 
 		// presence detection is only supported on macos currently
 		if runtime.GOOS != "darwin" {
+			w.Header().Add(kolideDurationSinceLastPresenceDetectionHeaderKey, presencedetection.DetectionFailedDurationValue.String())
 			next.ServeHTTP(w, r)
 			return
 		}
 
 		// can test this by adding an unauthed endpoint to the mux and running, for example:
 		// curl -i -H "X-Kolide-Presence-Detection-Interval: 10s" -H "X-Kolide-Presence-Detection-Reason: my reason" localhost:12519/id
-		detectionIntervalStr := r.Header.Get(kolidePresenceDetectionInterval)
+		detectionIntervalStr := r.Header.Get(kolidePresenceDetectionIntervalHeaderKey)
 
 		// no presence detection requested
 		if detectionIntervalStr == "" {
@@ -439,7 +441,7 @@ func (ls *localServer) presenceDetectionHandler(next http.Handler) http.Handler
 
 		// set a default reason, on macos the popup will look like "Kolide is trying to authenticate."
 		reason := "authenticate"
-		reasonHeader := r.Header.Get(kolidePresenceDetectionReason)
+		reasonHeader := r.Header.Get(kolidePresenceDetectionReasonHeaderKey)
 		if reasonHeader != "" {
 			reason = reasonHeader
 		}
@@ -460,7 +462,7 @@ func (ls *localServer) presenceDetectionHandler(next http.Handler) http.Handler
 		// and send the request through
 		// allow the server to decide what to do based on last detection duration
 
-		w.Header().Add(kolideDurationSinceLastPresenceDetection, durationSinceLastDetection.String())
+		w.Header().Add(kolideDurationSinceLastPresenceDetectionHeaderKey, durationSinceLastDetection.String())
 		next.ServeHTTP(w, r)
 	})
 }