From cd39231b9adb1fc700ff8a690cbca68f0ea61cdc Mon Sep 17 00:00:00 2001
From: lopezloo <lopez@multitheftauto.pl>
Date: Tue, 20 Dec 2016 05:25:49 +0100
Subject: [PATCH] Implement Logon ID (#63)

---
 steam/client/__init__.py | 11 +++++++++--
 steam/core/connection.py |  4 ++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/steam/client/__init__.py b/steam/client/__init__.py
index 679b10e..71fbb55 100644
--- a/steam/client/__init__.py
+++ b/steam/client/__init__.py
@@ -28,7 +28,7 @@ from steam.core.msg import MsgProto
 from steam.core.cm import CMClient
 from steam import SteamID
 from steam.client.builtins import BuiltinBase
-from steam.util import ip_from_int, proto_fill_from_dict
+from steam.util import ip_from_int, ip_to_int, proto_fill_from_dict
 
 if six.PY2:
     _cli_input = raw_input
@@ -436,7 +436,7 @@ class SteamClient(CMClient, BuiltinBase):
         if self.relogin_available:
             self.login(self.username, '', self.login_key)
 
-    def login(self, username, password='', login_key=None, auth_code=None, two_factor_code=None):
+    def login(self, username, password='', login_key=None, auth_code=None, two_factor_code=None, login_id=None):
         """Login as a specific user
 
         :param username: username
@@ -449,6 +449,8 @@ class SteamClient(CMClient, BuiltinBase):
         :type auth_code: :class:`str`
         :param two_factor_code: 2FA authentication code
         :type two_factor_code: :class:`str`
+        :param login_id: number used for identifying logon session
+        :type login_id: :class:`int`
         :return: logon result, see `CMsgClientLogonResponse.eresult <https://github.com/ValvePython/steam/blob/513c68ca081dc9409df932ad86c66100164380a6/protobufs/steammessages_clientserver.proto#L95-L118>`_
         :rtype: :class:`.EResult`
 
@@ -487,6 +489,11 @@ class SteamClient(CMClient, BuiltinBase):
         message.body.should_remember_password = True
         message.body.supports_rate_limit_response = True
 
+        if login_id is None:
+            message.body.obfustucated_private_ip = ip_to_int(self.connection.local_address) ^ 0xBAADF00D
+        else:
+            message.body.obfustucated_private_ip = login_id
+
         message.body.account_name = username
 
         if login_key:
diff --git a/steam/core/connection.py b/steam/core/connection.py
index ec53ab7..b382d21 100644
--- a/steam/core/connection.py
+++ b/steam/core/connection.py
@@ -28,6 +28,10 @@ class Connection(object):
 
         self.event_connected = event.Event()
 
+    @property
+    def local_address(self):
+        return self.socket.getsockname()[0]
+
     def connect(self, server_addr):
         self._new_socket()