]> code.delx.au - pymsnt/blobdiff - src/legacy/msn/msn.py
Applied patch from the old svn memorytest tree
[pymsnt] / src / legacy / msn / msn.py
index d3da1e474145fecd93271d47a7ab4647acdf45b8..c27142368ac028b36d07cc7d2a81b610e3132096 100644 (file)
@@ -84,6 +84,9 @@ Use of this module requires that PyOpenSSL is installed.
 
 from __future__ import nested_scopes
 
+import twistfix
+twistfix.main()
+
 # Sibling imports
 from twisted.protocols.basic import LineReceiver
 from twisted.web.http import HTTPClient
@@ -113,6 +116,7 @@ MSN_MAX_MESSAGE      = 1664               # max message length
 MSN_CVR_STR          = "0x040c winnt 5.1 i386 MSNMSGR 7.0.0777 msmsgs"
 MSN_AVATAR_GUID      = "{A4268EEC-FEC5-49E5-95C3-F126696BDBF6}"
 MSN_MSNFTP_GUID      = "{5D3E02AB-6190-11D3-BBBB-00C04F795683}"
+MSN_MAXINT           = 2**31 - 1
 
 # auth constants
 LOGIN_SUCCESS  = 1
@@ -464,7 +468,7 @@ class MSNObject:
     Used to represent a MSNObject. This can be currently only be an avatar.
 
     @ivar creator: The userHandle of the creator of this picture.
-    @ivar imageData: The PNG image data (only for our own avatar)
+    @ivar imageDataFunc: A function to return the PNG image data (only for our own avatar)
     @ivar type: Always set to 3, for avatar.
     @ivar size: The size of the image.
     @ivar location: The filename of the image.
@@ -477,10 +481,11 @@ class MSNObject:
         if s:
             self.parse(s)
     
-    def setData(self, creator, imageData):
+    def setData(self, creator, imageDataFunc):
         """ Set the creator and imageData for this object """
+        imageData = imageDataFunc()
         self.creator = creator
-        self.imageData = imageData
+        self.imageDataFunc = imageDataFunc
         self.size = len(imageData)
         self.type = 3
         self.location = "TMP" + str(random.randint(1000,9999))
@@ -490,7 +495,7 @@ class MSNObject:
     
     def setNull(self):
         self.creator = ""
-        self.imageData = ""
+        self.imageDataFunc = lambda: None
         self.size = 0
         self.type = 0
         self.location = ""
@@ -520,13 +525,18 @@ class MSNObject:
         e = parseText(s, True)
         if not e:
             return # Parse failed
-        self.creator = e.getAttribute("Creator")
-        self.size = int(e.getAttribute("Size"))
-        self.type = int(e.getAttribute("Type"))
-        self.location = e.getAttribute("Location")
-        self.friendly = e.getAttribute("Friendly")
-        self.sha1d = e.getAttribute("SHA1D")
-        self.text = s
+        try:
+            self.creator = e.getAttribute("Creator")
+            self.size = int(e.getAttribute("Size"))
+            self.type = int(e.getAttribute("Type"))
+            self.location = e.getAttribute("Location")
+            self.friendly = e.getAttribute("Friendly")
+            self.sha1d = e.getAttribute("SHA1D")
+            self.text = s
+        except TypeError:
+            self.setNull()
+        except ValueError:
+            self.setNull()
 
 
 class MSNContact:
@@ -1130,7 +1140,7 @@ class NotificationClient(MSNEventBase):
         msnContact.status = params[1]
         msnContact.screenName = unquote(params[3])
         if len(params) > 4: msnContact.caps = int(params[4])
-        if len(params) > 5:
+        if len(params) > 5 and params[5] != "0":
             self.handleAvatarHelper(msnContact, params[5])
         else:
             self.handleAvatarGoneHelper(msnContact)
@@ -1166,7 +1176,7 @@ class NotificationClient(MSNEventBase):
         msnContact.status = params[0]
         msnContact.screenName = unquote(params[2])
         if len(params) > 3: msnContact.caps = int(params[3])
-        if len(params) > 4:
+        if len(params) > 4 and params[4] != "0":
             self.handleAvatarHelper(msnContact, params[4])
         else:
             self.handleAvatarGoneHelper(msnContact)
@@ -1937,11 +1947,11 @@ class NotificationClient(MSNEventBase):
             return (personal,)
         return d.addCallback(_cb)
 
-    def changeAvatar(self, imageData, push):
+    def changeAvatar(self, imageDataFunc, push):
         """
         Used to change the avatar that other users see.
 
-        @param imageData: the PNG image data to set as the avatar
+        @param imageDataFunc: a function to return the PNG image data to set as the avatar
         @param push     : whether to push the update to the server
                           (it will otherwise be sent with the next
                           changeStatus())
@@ -1951,12 +1961,18 @@ class NotificationClient(MSNEventBase):
                  The callback argument will be the same as for changeStatus.
         """
 
-        if self.msnobj and imageData == self.msnobj.imageData: return
-        if imageData:
-            self.msnobj.setData(self.factory.userHandle, imageData)
+        checkMsnobj = MSNObject()
+        checkMsnobj.setData(self.factory.userHandle, imageDataFunc)
+        if self.msnobj and self.msnobj.sha1d == checkMsnobj.sha1d:
+            return # Avatar hasn't changed
+        if imageDataFunc:
+            # We need to keep the same MSNObject instance, as it is
+            # passed on to SwitchboardClient objects
+            self.msnobj.setData(self.factory.userHandle, imageDataFunc)
         else:
             self.msnobj.setNull()
-        if push: return self.changeStatus(self.factory.status) # Push to server
+        if push:
+            return self.changeStatus(self.factory.status) # Push to server
 
 
     def requestSwitchboardServer(self):
@@ -2147,7 +2163,7 @@ class SwitchboardClient(MSNEventBase):
                     # Check that we have an avatar to send
                     if self.msnobj:
                         slpLink = SLPLink_AvatarSend(remoteUser=slpMessage.fro, switchboard=self, filesize=self.msnobj.size, sessionID=slpMessage.sessionID, sessionGuid=slpMessage.sessionGuid)
-                        slpLink.write(self.msnobj.imageData)
+                        slpLink.write(self.msnobj.imageDataFunc())
                         slpLink.close()
                     else:
                         # They shouldn't have sent a request if we have
@@ -2681,7 +2697,7 @@ class SeqID:
         if baseID:
             self.baseID = baseID
         else:
-            self.baseID = random.randint(1000, sys.maxint)
+            self.baseID = random.randint(1000, MSN_MAXINT)
         self.pos = -1
 
     def get(self):
@@ -2708,7 +2724,7 @@ class SLPLink:
     def __init__(self, remoteUser, switchboard, sessionID, sessionGuid):
         self.dataFlag = 0
         if not sessionID:
-            sessionID = random.randint(1000, sys.maxint)
+            sessionID = random.randint(1000, MSN_MAXINT)
         if not sessionGuid:
             sessionGuid = random_guid()
         self.remoteUser = remoteUser
@@ -2754,7 +2770,7 @@ class SLPLink:
         binaryFields[1] = self.seqID.next()
         binaryFields[3] = len(msgStr)
         binaryFields[4] = binaryFields[3]
-        binaryFields[6] = random.randint(1000, sys.maxint)
+        binaryFields[6] = random.randint(1000, MSN_MAXINT)
         self.sendP2PMessage(binaryFields, msgStr)
 
     def sendP2PMessage(self, binaryFields, msgStr):
@@ -2788,7 +2804,7 @@ class SLPLink_Send(SLPLink):
         binaryFields[1] = self.seqID.next()
         binaryFields[3] = 4
         binaryFields[4] = 4
-        binaryFields[6] = random.randint(1000, sys.maxint)
+        binaryFields[6] = random.randint(1000, MSN_MAXINT)
         binaryFields[9] = 1
         self.sendP2PMessage(binaryFields, chr(0) * 4)
 
@@ -2818,7 +2834,7 @@ class SLPLink_Send(SLPLink):
         binaryFields[3] = self.filesize
         binaryFields[4] = len(chunk)
         binaryFields[5] = self.dataFlag
-        binaryFields[6] = random.randint(1000, sys.maxint)
+        binaryFields[6] = random.randint(1000, MSN_MAXINT)
         binaryFields[9] = 1
         self.offset += len(chunk)
         self.sendP2PMessage(binaryFields, chunk)