增加使用字符串房间号生成privateMapKey

yutingzeng · yutingzeng · commit f91496e3b192 · 2020-09-08T15:27:48.000+08:00
diff --git a/TLSSigAPIv2.py b/TLSSigAPIv2.py
@@ -47,14 +47,19 @@ def __init__(self, sdkappid, key):
     # @param dwExpTime 过期时间：该权限加密串的过期时间，实际过期时间：当前时间+dwExpTime
     # @param dwPrivilegeMap 用户权限，255表示所有权限
     # @param dwAccountType 用户类型,默认为0
+    # @param roomStr 字符串房间号,数字房间号非0时存在
     # @return userbuf  {string}  返回的userbuf
     #/
-    def _gen_userbuf(self,account, dwAuthID, dwExpTime,
-               dwPrivilegeMap, dwAccountType):
+    def _gen_userbuf(self, account, dwAuthID, dwExpTime,
+               dwPrivilegeMap, dwAccountType, roomStr):
         userBuf = b''
 
+        if len(roomStr) > 0 :
+            userBuf += bytearray([1])
+        else :
+            userBuf += bytearray([0])
+
         userBuf += bytearray([
-            0,
             ((len(account) & 0xFF00) >> 8),
             (len(account) & 0x00FF),
         ])
@@ -100,6 +105,12 @@ def _gen_userbuf(self,account, dwAuthID, dwExpTime,
             ((dwAccountType & 0x0000FF00) >> 8),
             (dwAccountType & 0x000000FF),
         ])
+        if len(roomStr) > 0 :
+            userBuf += bytearray([
+                ((len(roomStr) & 0xFF00) >> 8),
+                (len(roomStr) & 0x00FF),
+            ])
+            userBuf += bytearray(map(ord, roomStr))
         return userBuf
     def __hmacsha256(self, identifier, curr_time, expire, base64_userbuf=None):
         """ 通过固定串进行 hmac 然后 base64 得的 sig 字段的值"""
@@ -171,17 +182,48 @@ def genUserSig(self, userid, expire=180*86400):
     #/
     def genPrivateMapKey(self, userid, expire, roomid, privilegeMap):
         """ 带 userbuf 生成签名 """
-        userbuf = self._gen_userbuf(userid,roomid,expire,privilegeMap,0)
+        userbuf = self._gen_userbuf(userid,roomid,expire,privilegeMap,0,"")
         print(userbuf)
         return self.__gen_sig(userid, expire, userbuf)
 
+    ##
+    #【功能说明】
+    # 用于签发 TRTC 进房参数中可选的 PrivateMapKey 权限票据。
+    # PrivateMapKey  需要跟 UserSig 一起使用，但比 UserSig 有更强的权限控制能力：
+    # - UserSig 只能控制某个 UserID 有无使用 TRTC 服务的权限，只要 UserSig 正确，其对应的 UserID 可以进出任意房间。
+    # - PrivateMapKey 则是将 UserID 的权限控制的更加严格，包括能不能进入某个房间，能不能在该房间里上行音视频等等。
+    # 如果要开启 PrivateMapKey 严格权限位校验，需要在【实时音视频控制台】=>【应用管理】=>【应用信息】中“启动权限密钥”。
+    # 
+    #【参数说明】
+    # userid - 用户id，限制长度为32字节，只允许包含大小写英文字母（a-zA-Z）、数字（0-9）及下划线和连词符。
+    # expire - PrivateMapKey 票据的过期时间，单位是秒，比如 86400 生成的 PrivateMapKey 票据在一天后就无法再使用了。
+    # roomstr - 字符串房间号，用于指定该 userid 可以进入的房间号
+    # privilegeMap - 权限位，使用了一个字节中的 8 个比特位，分别代表八个具体的功能权限：
+    #  - 第 1 位：0000 0001 = 1，创建房间的权限
+    #  - 第 2 位：0000 0010 = 2，加入房间的权限
+    #  - 第 3 位：0000 0100 = 4，发送语音的权限
+    #  - 第 4 位：0000 1000 = 8，接收语音的权限
+    #  - 第 5 位：0001 0000 = 16，发送视频的权限  
+    #  - 第 6 位：0010 0000 = 32，接收视频的权限  
+    #  - 第 7 位：0100 0000 = 64，发送辅路（也就是屏幕分享）视频的权限
+    #  - 第 8 位：1000 0000 = 200，接收辅路（也就是屏幕分享）视频的权限  
+    #  - privilegeMap == 1111 1111 == 255 代表该 userid 在该 roomid 房间内的所有功能权限。
+    #  - privilegeMap == 0010 1010 == 42  代表该 userid 拥有加入房间和接收音视频数据的权限，但不具备其他权限。
+    #/
+    def genPrivateMapKeyWithStringRoomID(self, userid, expire, roomstr, privilegeMap):
+        """ 带 userbuf 生成签名 """
+        userbuf = self._gen_userbuf(userid,0,expire,privilegeMap,0,roomstr)
+        print(userbuf)
+        return self.__gen_sig(userid, expire, userbuf)
 
 def main():
     api = TLSSigAPIv2(1400000000, '5bd2850fff3ecb11d7c805251c51ee463a25727bddc2385f3fa8bfee1bb93b5e')
     sig = api.genUserSig("xiaojun")
     print(sig)
     sig = api.genPrivateMapKey("xiaojun", 86400*180,10000,255)
     print(sig)
+    sig = api.genPrivateMapKeyWithStringRoomID("xiaojun", 86400*180,"adbjkl",255)
+    print(sig)
 
 
 if __name__ == "__main__":
