adding reading agent ssh key from file

This commit is contained in:
Ben Morgenstren
2025-01-14 22:13:36 -05:00
parent 5d88599c9a
commit 1c4533f1f2

View File

@@ -7,6 +7,7 @@ import (
"log"
"os"
"strings"
"io/ioutil"
)
func main() {
@@ -25,7 +26,16 @@ func main() {
if pubKeyEnv, exists := os.LookupEnv("KEY"); exists {
pubKey = []byte(pubKeyEnv)
} else {
log.Fatal("KEY environment variable is not set")
keyFile := os.Getenv("KEY_FILE")
if keyFile != "" {
if keyData, err := ioutil.ReadFile(keyFile); err == nil {
pubKey = keyData
} else {
log.Fatalf("Failed to read key from file '%s': %v", keyFile, err)
}
} else {
log.Fatal("KEY environment variable is not set, and KEY_FILE environment variable is not set")
}
}
addr := ":45876"