Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2025-01-02 18:39:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

make auth key header a constant

Browse Source
This commit is contained in:
Evan Simkowitz 2024-08-26 13:55:47 -07:00
parent 2118c24c0d
commit 4e68211f22
No known key found for this signature in database
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pkg/authkey/authkey.go
View File

@ -12,6 +12,18 @@ import (
var authkey string var authkey string
const AuthKeyEnv = "AUTH_KEY" const AuthKeyEnv = "AUTH_KEY"
const AuthKeyHeader = "X-AuthKey"
func ValidateIncomingRequest(r *http.Request) error {
reqAuthKey := r.Header.Get(AuthKeyHeader)
if reqAuthKey == "" {
return fmt.Errorf("no x-authkey header")
}
if reqAuthKey != GetAuthKey() {
return fmt.Errorf("x-authkey header is invalid")
}
return nil
}
func SetAuthKeyFromEnv() error { func SetAuthKeyFromEnv() error {
authkey = os.Getenv(AuthKeyEnv) authkey = os.Getenv(AuthKeyEnv)
@ -25,14 +37,3 @@ func SetAuthKeyFromEnv() error {
func GetAuthKey() string { func GetAuthKey() string {
return authkey return authkey
} }
func ValidateIncomingRequest(r *http.Request) error {
reqAuthKey := r.Header.Get("X-AuthKey")
if reqAuthKey == "" {
return fmt.Errorf("no x-authkey header")
}
if reqAuthKey != GetAuthKey() {
return fmt.Errorf("x-authkey header is invalid")
}
return nil
}