提交

TUIKit/TUICallKit/TUICallKit-Swift/Debug/GenerateTestUserSig.swift

@@ -0,0 +1,123 @@
+//
+//  GenerateTestUserSig.swift
+//  TUICallKitApp
+//
+//  Created by abyyxwang on 2021/5/7.
+//  Copyright © 2021 Tencent. All rights reserved.
+//
+
+import Foundation
+import CommonCrypto
+import zlib
+import TUICore
+
+@objc
+public class GenerateTestUserSig: NSObject {
+   
+    @objc
+    public class func genTestUserSig(userID: String, sdkAppID: Int, secretKey: String) -> String {
+        // Signature expiration time, it is recommended not to set it too short.
+        // Default time: 7 x 24 x 60 x 60 = 604800 = 7 days
+        let EXPIRETIME = 604_800
+        let current = CFAbsoluteTimeGetCurrent() + kCFAbsoluteTimeIntervalSince1970
+        let TLSTime: CLong = CLong(floor(current))
+        var obj: [String: Any] = [
+            "TLS.ver": "2.0",
+            "TLS.identifier": userID,
+            "TLS.sdkappid": sdkAppID,
+            "TLS.expire": EXPIRETIME,
+            "TLS.time": TLSTime,
+        ]
+        let keyOrder = [
+            "TLS.identifier",
+            "TLS.sdkappid",
+            "TLS.time",
+            "TLS.expire",
+        ]
+        var stringToSign = ""
+        keyOrder.forEach { (key) in
+            if let value = obj[key] {
+                stringToSign += "\(key):\(value)\n"
+            }
+        }
+        print("string to sign: \(stringToSign)")
+        guard var sig = hmac(plainText: stringToSign, secretKey: secretKey) else {
+            print("hmac error: \(stringToSign)")
+            return ""
+        }
+        obj["TLS.sig"] = sig
+        print("sig: \(String(describing: sig))")
+        guard let jsonData = try? JSONSerialization.data(withJSONObject: obj, options: .sortedKeys) else {
+            print("jsonData error: \(obj)")
+            return ""
+        }
+        
+        let bytes = jsonData.withUnsafeBytes { (result) -> UnsafePointer<Bytef>? in
+            return result.bindMemory(to: Bytef.self).baseAddress
+        }
+        let srcLen: uLongf = uLongf(jsonData.count)
+        let upperBound: uLong = compressBound(srcLen)
+        let capacity: Int = Int(upperBound)
+        let dest: UnsafeMutablePointer<Bytef> = UnsafeMutablePointer<Bytef>.allocate(capacity: capacity)
+        var destLen = upperBound
+        let ret = compress2(dest, &destLen, bytes, srcLen, Z_BEST_SPEED)
+        if ret != Z_OK {
+            print("[Error] Compress Error \(ret), upper bound: \(upperBound)")
+            dest.deallocate()
+            return ""
+        }
+        let count = Int(destLen)
+        let result = self.base64URL(data: Data(bytesNoCopy: dest, count: count, deallocator: .free))
+        return result
+    }
+    
+    class func hmac(plainText: String, secretKey: String) -> String? {
+        guard let cKey = secretKey.cString(using: String.Encoding.ascii) else {
+            print("hmac secretKey error: \(secretKey)")
+            return nil
+        }
+        print("hmac secretKey: \(secretKey)")
+        print("hmac cKey: \(cKey)")
+        guard let cData = plainText.cString(using: String.Encoding.ascii) else{
+            print("hmac plainText error: \(plainText)")
+            return nil
+        }
+        print("hmac plainText: \(plainText)")
+        print("hmac cData: \(cData)")
+        let cKeyLen = secretKey.lengthOfBytes(using: .ascii)
+        let cDataLen = plainText.lengthOfBytes(using: .ascii)
+        
+        var cHMAC = [CUnsignedChar].init(repeating: 0, count: Int(CC_SHA256_DIGEST_LENGTH))
+        let pointer = cHMAC.withUnsafeMutableBufferPointer { (unsafeBufferPointer) in
+            return unsafeBufferPointer
+        }
+        CCHmac(CCHmacAlgorithm(kCCHmacAlgSHA256), cKey, cKeyLen, cData, cDataLen, pointer.baseAddress)
+        guard let adress = pointer.baseAddress else {
+            print("adress error: \(String(describing: pointer))")
+            return nil
+        }
+        let data = Data(bytes: adress, count: cHMAC.count)
+        print("cHMAC.count: \(String(describing: cHMAC.count))")
+        print("data: \(String(describing: data))")
+        let result = data.base64EncodedString(options: [])
+        return result
+    }
+    
+    class func base64URL(data: Data) -> String {
+        let result = data.base64EncodedString(options: Data.Base64EncodingOptions(rawValue: 0))
+        var final = ""
+        result.forEach { (char) in
+            switch char {
+            case "+":
+                final += "*"
+            case "/":
+                final += "-"
+            case "=":
+                final += "_"
+            default:
+                final += "\(char)"
+            }
+        }
+        return final
+    }
+}