Rossen Georgiev
9 years ago
4 changed files with 539 additions and 527 deletions
@ -0,0 +1,182 @@ |
|||
"""Classes to (de)serialize message headers""" |
|||
import struct |
|||
from steam.enums.emsg import EMsg |
|||
from steam.protobufs import steammessages_base_pb2 |
|||
from steam.protobufs import gc_pb2 |
|||
from steam.util import set_proto_bit, clear_proto_bit |
|||
|
|||
|
|||
class MsgHdr: |
|||
_size = struct.calcsize("<Iqq") |
|||
msg = EMsg.Invalid |
|||
targetJobID = -1 |
|||
sourceJobID = -1 |
|||
|
|||
def __init__(self, data=None): |
|||
if data: self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<Iqq", self.msg, self.targetJobID, self.sourceJobID) |
|||
|
|||
def load(self, data): |
|||
(msg, self.targetJobID, self.sourceJobID) = struct.unpack_from("<Iqq", data) |
|||
self.msg = EMsg(msg) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["msg: %s" % repr(self.msg), |
|||
"targetJobID: %s" % self.targetJobID, |
|||
"sourceJobID: %s" % self.sourceJobID, |
|||
]) |
|||
|
|||
class ExtendedMsgHdr: |
|||
_size = struct.calcsize("<IBHqqBqi") |
|||
msg = EMsg.Invalid |
|||
headerSize = 36 |
|||
headerVersion = 2 |
|||
targetJobID = -1 |
|||
sourceJobID = -1 |
|||
headerCanary = 239 |
|||
steamID = -1 |
|||
sessionID = -1 |
|||
|
|||
def __init__(self, data=None): |
|||
if data: self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<IBHqqBqi", |
|||
self.msg, |
|||
self.headerSize, |
|||
self.headerVersion, |
|||
self.targetJobID, |
|||
self.sourceJobID, |
|||
self.headerCanary, |
|||
self.steamID, |
|||
self.sessionID, |
|||
) |
|||
|
|||
def load(self, data): |
|||
(msg, |
|||
self.headerSize, |
|||
self.headerVersion, |
|||
self.targetJobID, |
|||
self.sourceJobID, |
|||
self.headerCanary, |
|||
self.steamID, |
|||
self.sessionID, |
|||
) = struct.unpack_from("<IBHqqBqi", data) |
|||
|
|||
self.msg = EMsg(msg) |
|||
|
|||
if self.headerSize != 36 or self.headerVersion != 2: |
|||
raise RuntimeError("Failed to parse header") |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["msg: %s" % self.msg, |
|||
"headerSize: %s" % self.headerSize, |
|||
"headerVersion: %s" % self.headerVersion, |
|||
"targetJobID: %s" % self.targetJobID, |
|||
"sourceJobID: %s" % self.sourceJobID, |
|||
"headerCanary: %s" % self.headerCanary, |
|||
"steamID: %s" % self.steamID, |
|||
"sessionID: %s" % self.sessionID, |
|||
]) |
|||
|
|||
|
|||
class MsgHdrProtoBuf: |
|||
_size = struct.calcsize("<II") |
|||
msg = EMsg.Invalid |
|||
|
|||
def __init__(self, data=None): |
|||
self.proto = steammessages_base_pb2.CMsgProtoBufHeader() |
|||
|
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
proto_data = self.proto.SerializeToString() |
|||
return struct.pack("<II", set_proto_bit(self.msg), len(proto_data)) + proto_data |
|||
|
|||
def load(self, data): |
|||
msg, proto_length = struct.unpack_from("<II", data) |
|||
|
|||
self.msg = EMsg(clear_proto_bit(msg)) |
|||
size = MsgHdrProtoBuf._size |
|||
self._fullsize = size + proto_length |
|||
self.proto.ParseFromString(data[size:self._fullsize]) |
|||
|
|||
|
|||
class GCMsgHdr: |
|||
_size = struct.calcsize("<Hqq") |
|||
proto = None |
|||
headerVersion = 1 |
|||
targetJobID = -1 |
|||
sourceJobID = -1 |
|||
|
|||
def __init__(self, msg, data=None): |
|||
self.msg = clear_proto_bit(msg) |
|||
|
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<Hqq", |
|||
self.headerVersion, |
|||
self.targetJobID, |
|||
self.sourceJobID, |
|||
) |
|||
|
|||
def load(self, data): |
|||
(self.headerVersion, |
|||
self.targetJobID, |
|||
self.sourceJobID, |
|||
) = struct.unpack_from("<Hqq", data) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["headerVersion: %s" % self.headerVersion, |
|||
"targetJobID: %s" % self.targetJobID, |
|||
"sourceJobID: %s" % self.sourceJobID, |
|||
]) |
|||
|
|||
class GCMsgHdrProto: |
|||
_size = struct.calcsize("<Ii") |
|||
headerLength = 0 |
|||
|
|||
def __init__(self, msg, data=None): |
|||
self.proto = gc_pb2.CMsgProtoBufHeader() |
|||
self.msg = clear_proto_bit(msg) |
|||
|
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
proto_data = self.proto.SerializeToString() |
|||
self.headerLength = len(proto_data) |
|||
|
|||
return struct.pack("<Ii", |
|||
set_proto_bit(self.msg), |
|||
self.headerLength, |
|||
) + proto_data |
|||
|
|||
def load(self, data): |
|||
(msg, |
|||
self.headerLength, |
|||
) = struct.unpack_from("<Ii", data) |
|||
|
|||
self.msg = clear_proto_bit(msg) |
|||
|
|||
if self.headerLength: |
|||
x = GCMsgHdrProto._size |
|||
self.proto.ParseFromString(data[x:x+self.headerLength]) |
|||
|
|||
def __str__(self): |
|||
resp = ["msg: %s" % self.msg, |
|||
"headerLength: %s" % self.headerLength, |
|||
] |
|||
|
|||
proto = str(self.proto) |
|||
|
|||
if proto: |
|||
resp.append('-- proto --') |
|||
resp.append(proto) |
|||
|
|||
return '\n'.join(resp) |
|||
@ -0,0 +1,242 @@ |
|||
"""Classes to (de)serialize various struct messages""" |
|||
import struct |
|||
from steam.enums import EResult, EUniverse |
|||
from steam.enums.emsg import EMsg |
|||
|
|||
_emsg_map = {} |
|||
|
|||
def get_struct(emsg): |
|||
return _emsg_map.get(emsg, None) |
|||
|
|||
class MapEMsgMeta(type): |
|||
"""Automatically maps subclasses of :class:`StructMessage` to ``EMsg``""" |
|||
|
|||
def __new__(metacls, name, bases, classdict): |
|||
cls = type.__new__(metacls, name, bases, classdict) |
|||
|
|||
if name != 'StructMessage': |
|||
try: |
|||
_emsg_map[EMsg[name]] = cls |
|||
except KeyError: |
|||
pass |
|||
|
|||
return cls |
|||
|
|||
class StructMessage: |
|||
__metaclass__ = MapEMsgMeta |
|||
|
|||
def __init__(self, data=None): |
|||
if data: self.load(data) |
|||
|
|||
def serialize(self): |
|||
return b'' |
|||
|
|||
def load(self, data): |
|||
pass |
|||
|
|||
class ChannelEncryptRequest(StructMessage): |
|||
protocolVersion = 1 |
|||
universe = EUniverse.Invalid |
|||
challenge = b'' |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<II", self.protocolVersion, self.universe) + self.challenge |
|||
|
|||
def load(self, data): |
|||
(self.protocolVersion, |
|||
universe, |
|||
) = struct.unpack_from("<II", data) |
|||
|
|||
self.universe = EUniverse(universe) |
|||
|
|||
if len(data) > 8: |
|||
self.challenge = data[8:] |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["protocolVersion: %s" % self.protocolVersion, |
|||
"universe: %s" % repr(self.universe), |
|||
"challenge: %s" % repr(self.challenge), |
|||
]) |
|||
|
|||
class ChannelEncryptResponse(StructMessage): |
|||
protocolVersion = 1 |
|||
keySize = 128 |
|||
key = '' |
|||
crc = 0 |
|||
|
|||
def __init__(self, data=None): |
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<II128sII", |
|||
self.protocolVersion, |
|||
self.keySize, |
|||
self.key, |
|||
self.crc, |
|||
0 |
|||
) |
|||
|
|||
def load(self, data): |
|||
(self.protocolVersion, |
|||
self.keySize, |
|||
self.key, |
|||
self.crc, |
|||
_, |
|||
) = struct.unpack_from("<II128sII", data) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["protocolVersion: %s" % self.protocolVersion, |
|||
"keySize: %s" % self.keySize, |
|||
"key: %s" % repr(self.key), |
|||
"crc: %s" % self.crc, |
|||
]) |
|||
|
|||
class ChannelEncryptResult(StructMessage): |
|||
eresult = EResult.Invalid |
|||
|
|||
def __init__(self, data=None): |
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<I", self.eresult) |
|||
|
|||
def load(self, data): |
|||
(result,) = struct.unpack_from("<I", data) |
|||
self.eresult = EResult(result) |
|||
|
|||
def __str__(self): |
|||
return "result: %s" % repr(self.eresult) |
|||
|
|||
class ClientLogOnResponse(StructMessage): |
|||
eresult = EResult.Invalid |
|||
|
|||
def __init__(self, data=None): |
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<I", self.eresult) |
|||
|
|||
def load(self, data): |
|||
(result,) = struct.unpack_from("<I", data) |
|||
self.eresult = EResult(result) |
|||
|
|||
def __str__(self): |
|||
return "eresult: %s" % repr(self.eresult) |
|||
|
|||
class ClientVACBanStatus(StructMessage): |
|||
numBans = 0 |
|||
|
|||
def __init__(self, data=None): |
|||
if data: self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<L", self.steamIdChat) |
|||
|
|||
def load(self, data): |
|||
self.steamIdChat, = struct.unpack_from("<L", data) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["numBans: %d" % self.numBans, |
|||
]) |
|||
|
|||
class ClientChatMsg(StructMessage): |
|||
steamIdChatter = 0 |
|||
steamIdChatRoom = 0 |
|||
ChatMsgType = 0 |
|||
text = "" |
|||
|
|||
def __init__(self, data=None): |
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
rbytes = struct.pack("<QQI", |
|||
self.steamIdChatter, |
|||
self.steamIdChatRoom, |
|||
self.ChatMsgType, |
|||
) |
|||
# utf-8 encode only when unicode in py2 and str in py3 |
|||
rbytes += (self.text.encode('utf-8') |
|||
if (not isinstance(self.text, str) and bytes is str) |
|||
or isinstance(self.text, str) |
|||
else self.text |
|||
) + b'\x00' |
|||
|
|||
return rbytes |
|||
|
|||
def load(self, data): |
|||
(self.steamIdChatter, |
|||
self.steamIdChatRoom, |
|||
self.ChatMsgType, |
|||
) = struct.unpack_from("<QQI", data) |
|||
|
|||
self.text = data[struct.calcsize("<QQI"):-1].decode('utf-8') |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["steamIdChatter: %d" % self.steamIdChatter, |
|||
"steamIdChatRoom: %d" % self.steamIdChatRoom, |
|||
"ChatMsgType: %d" % self.ChatMsgType, |
|||
"text: %s" % repr(self.text), |
|||
]) |
|||
|
|||
class ClientJoinChat(StructMessage): |
|||
steamIdChat = 0 |
|||
isVoiceSpeaker = False |
|||
|
|||
def __init__(self, data=None): |
|||
if data: self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<Q?", |
|||
self.steamIdChat, |
|||
self.isVoiceSpeaker |
|||
) |
|||
|
|||
def load(self, data): |
|||
(self.steamIdChat, |
|||
self.isVoiceSpeaker |
|||
) = struct.unpack_from("<Q?", data) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["steamIdChat: %d" % self.steamIdChat, |
|||
"isVoiceSpeaker: %r" % self.isVoiceSpeaker, |
|||
]) |
|||
|
|||
class ClientChatMemberInfo(StructMessage): |
|||
steamIdChat = 0 |
|||
type = 0 |
|||
steamIdUserActedOn = 0 |
|||
chatAction = 0 |
|||
steamIdUserActedBy = 0 |
|||
|
|||
def __init__(self, data=None): |
|||
if data: |
|||
self.load(data) |
|||
|
|||
def serialize(self): |
|||
return struct.pack("<QIQIQ", |
|||
self.steamIdChat, |
|||
self.type, |
|||
self.steamIdUserActedOn, |
|||
self.chatAction, |
|||
self.steamIdUserActedBy |
|||
) |
|||
|
|||
def load(self, data): |
|||
(self.steamIdChat, |
|||
self.type, |
|||
self.steamIdUserActedOn, |
|||
self.chatAction, |
|||
self.steamIdUserActedBy |
|||
) = struct.unpack_from("<QIQIQ", data) |
|||
|
|||
def __str__(self): |
|||
return '\n'.join(["steamIdChat: %d" % self.steamIdChat, |
|||
"type: %r" % self.type, |
|||
"steamIdUserActedOn: %d" % self.steamIdUserActedOn, |
|||
"chatAction: %d" % self.chatAction, |
|||
"steamIdUserActedBy: %d" % self.steamIdUserActedBy |
|||
]) |
|||
@ -0,0 +1,65 @@ |
|||
import re |
|||
from importlib import import_module |
|||
|
|||
service_lookup = { |
|||
'Broadcast': 'steam.protobufs.steammessages_broadcast_pb2', |
|||
'Cloud': 'steam.protobufs.steammessages_cloud_pb2', |
|||
'DNReport': 'steam.protobufs.steammessages_cloud_pb2', |
|||
'Credentials': 'steam.protobufs.steammessages_credentials_pb2', |
|||
'ContentBuilder': 'steam.protobufs.steammessages_depotbuilder_pb2', |
|||
'DeviceAuth': 'steam.protobufs.steammessages_deviceauth_pb2', |
|||
'Econ': 'steam.protobufs.steammessages_econ_pb2', |
|||
'GameNotifications': 'steam.protobufs.steammessages_gamenotifications_pb2', |
|||
'GameServers': 'steam.protobufs.steammessages_gameservers_pb2', |
|||
'Inventory': 'steam.protobufs.steammessages_inventory_pb2', |
|||
'Community': 'steam.protobufs.steammessages_linkfilter_pb2', |
|||
'Offline': 'steam.protobufs.steammessages_offline_pb2', |
|||
'Parental': 'steam.protobufs.steammessages_parental_pb2', |
|||
'PartnerApps': 'steam.protobufs.steammessages_partnerapps_pb2', |
|||
'PhysicalGoods': 'steam.protobufs.steammessages_physicalgoods_pb2', |
|||
'PlayerClient': 'steam.protobufs.steammessages_player_pb2', |
|||
'Player': 'steam.protobufs.steammessages_player_pb2', |
|||
'PublishedFile': 'steam.protobufs.steammessages_publishedfile_pb2', |
|||
'KeyEscrow': 'steam.protobufs.steammessages_secrets_pb2', |
|||
'TwoFactor': 'steam.protobufs.steammessages_twofactor_pb2', |
|||
'MsgTest': 'steam.protobufs.steammessages_unified_test_pb2', |
|||
'Video': 'steam.protobufs.steammessages_video_pb2', |
|||
} |
|||
|
|||
method_lookup = { |
|||
} |
|||
|
|||
def get_um(method_name, response=False): |
|||
"""Get protobuf for given method name |
|||
|
|||
:param method_name: full method name (e.g. ``Player.GetGameBadgeLevels#1``) |
|||
:type method_name: :class:`str` |
|||
:param response: whether to return proto for response or request |
|||
:type response: :class:`bool` |
|||
:return: protobuf message |
|||
""" |
|||
key = (method_name, response) |
|||
|
|||
if key not in method_lookup: |
|||
match = re.findall(r'^([a-z]+)\.([a-z]+)#(\d)?$', method_name, re.I) |
|||
if not match: |
|||
return None |
|||
|
|||
interface, method, version = match[0] |
|||
|
|||
if interface not in service_lookup: |
|||
raise None |
|||
|
|||
package = import_module(service_lookup[interface]) |
|||
|
|||
service = getattr(package, interface, None) |
|||
if service is None: |
|||
return None |
|||
|
|||
for method_desc in service.GetDescriptor().methods: |
|||
name = "%s.%s#%d" % (interface, method_desc.name, 1) |
|||
|
|||
method_lookup[(name, False)] = getattr(package, method_desc.input_type.full_name, None) |
|||
method_lookup[(name, True)] = getattr(package, method_desc.output_type.full_name, None) |
|||
|
|||
return method_lookup[key] |
|||
Loading…
Reference in new issue