From 137415cb5062ae3b5ce4ba8056754c1fc4febcaf Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Sun, 23 Mar 2025 12:04:27 +0800 Subject: Separate proxy stuff --- token.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 token.go (limited to 'token.go') diff --git a/token.go b/token.go new file mode 100644 index 0000000..74bf903 --- /dev/null +++ b/token.go @@ -0,0 +1,39 @@ +package main + +import ( + "crypto/hmac" + "crypto/sha256" + "encoding/binary" + "net/http" + "time" +) + +func makeSignedToken(request *http.Request) []byte { + buf := make([]byte, 0, 2*sha256.Size) + + timeBuf := make([]byte, binary.MaxVarintLen64) + binary.PutVarint(timeBuf, time.Now().Unix()/604800) + + remoteIP := getRemoteIP(request) + + h := sha256.New() + h.Write(timeBuf) + h.Write(stringToBytes(remoteIP)) + h.Write(stringToBytes(request.Header.Get("User-Agent"))) + h.Write(stringToBytes(request.Header.Get("Accept-Encoding"))) + h.Write(stringToBytes(request.Header.Get("Accept-Language"))) + h.Write(privkeyHash) + buf = h.Sum(buf) + if len(buf) != sha256.Size { + panic("unexpected buffer length after hashing contents") + } + + mac := hmac.New(sha256.New, privkey) + mac.Write(buf) + buf = mac.Sum(buf) + if len(buf) != 2*sha256.Size { + panic("unexpected buffer length after hmac") + } + + return buf +} -- cgit v1.2.3