添加代码

weijunyi · weijunyi · commit 541eda4c3bea · 2019-06-17T20:06:43.000+08:00
diff --git a/.gitignore b/.gitignore
@@ -10,3 +10,5 @@
 
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
+
+.idea
diff --git a/TLSSigAPITest.go b/TLSSigAPITest.go
@@ -0,0 +1,20 @@
+package main
+
+import (
+	"./tencentyun"
+	"fmt"
+)
+
+const (
+	sdkappid = 1400000000
+	key = "5bd2850fff3ecb11d7c805251c51ee463a25727bddc2385f3fa8bfee1bb93b5e"
+)
+
+func main()  {
+	sig, err := tencentyun.GenSig(sdkappid, key, "xiaojun", 86400*180)
+	if err != nil {
+		fmt.Println(err.Error())
+	} else {
+		fmt.Println(sig)
+	}
+}
diff --git a/tencentyun/TLSSigAPI.go b/tencentyun/TLSSigAPI.go
@@ -0,0 +1,50 @@
+package tencentyun
+
+import (
+	"bytes"
+	"compress/zlib"
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"strconv"
+	"time"
+)
+
+func hmacsha256(sdkappid int, key string, identifier string, currTime int64, expire int) string {
+	var contentToBeSigned string
+	contentToBeSigned = "TLS.identifier:" + identifier + "\n"
+	contentToBeSigned += "TLS.sdkappid:" + strconv.Itoa(sdkappid) + "\n"
+	contentToBeSigned += "TLS.time:" + strconv.FormatInt(currTime, 10) + "\n"
+	contentToBeSigned += "TLS.expire:" + strconv.Itoa(expire) + "\n"
+
+	fmt.Println(contentToBeSigned)
+
+	h := hmac.New(sha256.New, []byte(key))
+	h.Write([]byte(contentToBeSigned))
+	return base64.StdEncoding.EncodeToString(h.Sum(nil))
+}
+
+func GenSig(sdkappid int, key string, identifier string, expire int) (string, error) {
+	currTime := time.Now().Unix()
+	var sigDoc map[string]interface{}
+	sigDoc = make(map[string]interface{})
+	sigDoc["TLS.ver"] = "2.0"
+	sigDoc["TLS.identifier"] = identifier
+	sigDoc["TLS.sdkappid"] = sdkappid
+	sigDoc["TLS.expire"] = expire
+	sigDoc["TLS.time"] = currTime
+	sigDoc["TLS.sig"] = hmacsha256(sdkappid, key, identifier, currTime, expire)
+
+	data, err := json.Marshal(sigDoc)
+	if err != nil {
+		return "", err
+	}
+
+	var b bytes.Buffer
+	w := zlib.NewWriter(&b)
+	w.Write(data)
+	w.Close()
+	return base64urlEncode(b.Bytes()), nil
+}
diff --git a/tencentyun/base64url.go b/tencentyun/base64url.go
@@ -0,0 +1,21 @@
+package tencentyun
+
+import (
+	"encoding/base64"
+	"strings"
+)
+
+func base64urlEncode(data []byte) string {
+	str := base64.StdEncoding.EncodeToString(data)
+	str = strings.Replace(str, "+", "*", -1)
+	str = strings.Replace(str, "/", "-", -1)
+	str = strings.Replace(str, "=", "_", -1)
+	return str
+}
+
+func base64urlDecode(str string) ([]byte, error) {
+	str = strings.Replace(str, "_", "=", -1)
+	str = strings.Replace(str, "-", "/", -1)
+	str = strings.Replace(str, "*", "+", -1)
+	return base64.StdEncoding.DecodeString(str)
+}

-Original file line number
+Diff line change
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
++
 +.idea