chore(ci): add ci

Author: jakobmoellerdev

cmd/signing-server/main_test.go

@@ -0,0 +1,142 @@
+package main
+
+import (
+	"bytes"
+	"encoding/pem"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"os"
+	"os/exec"
+	"strings"
+	"sync"
+	"testing"
+	"time"
+
+	"go.uber.org/zap"
+)
+
+func TestSoftHSMConcurrentSignRequests(t *testing.T) {
+	hsmModule, err := exec.Command("softhsm2-util", "--show-config", "default-pkcs11-lib").CombinedOutput()
+	if err != nil {
+		hsmModule = []byte("/usr/lib/softhsm/libsofthsm2.so")
+	}
+	tokenLabel := os.Getenv("TOKEN_LABEL")
+	if tokenLabel == "" {
+		t.Skip("TOKEN_LABEL environment variable is not set")
+	}
+	keyLabel := os.Getenv("KEY_LABEL")
+	if keyLabel == "" {
+		t.Skip("KEY_LABEL environment variable is not set")
+	}
+	hsmPin := os.Getenv("HSM_PIN")
+	if hsmPin == "" {
+		t.Skip("HSM_PIN environment variable is not set")
+	}
+
+	const (
+		base      = "http://localhost:8080"
+		healthURL = base + "/healthz"
+		url       = base + "/sign/rsassa-pss?hashAlgorithm=sha256"
+		bodyHex   = "aec070645fe53ee3b3763059376134f058cc337247c978add178b6ccdfb0019f"
+	)
+
+	var (
+		headers = map[string]string{
+			"Content-Type":     "text/plain",
+			"Content-Encoding": "hex",
+			"Accept":           "application/x-pem-file",
+		}
+	)
+
+	l, err := zap.NewDevelopment()
+	if err != nil {
+		t.Fatalf("Failed to create logger: %v", err)
+	}
+
+	go func() {
+
+		if err := run(&Config{
+			HSMModule:        strings.TrimSpace(string(hsmModule)),
+			HSMTokenLabel:    tokenLabel,
+			HSMSlot:          -1,
+			HSMKeyLabel:      keyLabel,
+			HSMPass:          hsmPin,
+			Port:             "8080",
+			DisableHTTPS:     true,
+			DisableAuth:      true,
+			MaxBodySizeBytes: 2048,
+			Logger:           l,
+			RunServer:        true,
+		}); err != nil {
+			t.Fatalf("Failed to start signing server: %v", err)
+		}
+	}()
+
+	var wg sync.WaitGroup
+	client := &http.Client{Timeout: 10 * time.Second}
+
+outer:
+	for {
+		select {
+		case <-t.Context().Done():
+			t.Fatal("Test context was cancelled before health check passed")
+		default:
+			healthReq, err := http.NewRequest(http.MethodGet, healthURL, nil)
+			if err != nil {
+				t.Fatalf("Health request creation failed: %v", err)
+			}
+			healthResp, _ := client.Do(healthReq)
+			if healthResp != nil && healthResp.StatusCode == http.StatusOK {
+				log.Println("Health check passed")
+				break outer
+			}
+		}
+	}
+
+	for i := 0; i < 10; i++ {
+		wg.Add(1)
+
+		go func(index int) {
+			defer wg.Done()
+
+			req, err := http.NewRequest(http.MethodPost, url, bytes.NewBufferString(bodyHex))
+			if err != nil {
+				t.Errorf("Request %d creation failed: %v", index, err)
+				return
+			}
+
+			for k, v := range headers {
+				req.Header.Set(k, v)
+			}
+
+			resp, err := client.Do(req)
+			if err != nil {
+				t.Errorf("Request %d failed: %v", index, err)
+				return
+			}
+			defer resp.Body.Close()
+
+			body, err := io.ReadAll(resp.Body)
+			if err != nil {
+				t.Errorf("Request %d read body failed: %v", index, err)
+				return
+			}
+			if resp.StatusCode != http.StatusOK {
+				t.Errorf("Request %d failed with status %d: %s", index, resp.StatusCode, body)
+				return
+			}
+
+			block, _ := pem.Decode(body)
+			if block == nil || block.Type != "SIGNATURE" {
+				log.Fatal("Failed to parse PEM block")
+			}
+			fmt.Printf("Decoded signature: %x\n", block.Bytes)
+
+			log.Printf("Response %d: %d - %.100q\n", index, resp.StatusCode, body)
+		}(i)
+	}
+
+	wg.Wait()
+}