-
Notifications
You must be signed in to change notification settings - Fork 1
/
mtalkconn.py
140 lines (110 loc) · 4.26 KB
/
mtalkconn.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
import socket
import ssl
import threading
import time
import queue
import google.protobuf.internal.decoder as GoogleProtobufDecoder
from gcm import mtalk, googleplayapi
from gcm.protos import GoogleServicesFramework_pb2
import config
import receive_textsecure
import dialogs
app = "org.thoughtcrime.securesms.gcm"
senderId = "312334754206"
#app = "com.iapplize.gcm.test"
#senderId = "879830610296"
q = queue.Queue()
quit_thread = False
def configGCM(email, password, app=app, senderId=senderId):
gapi = googleplayapi.GooglePlayAPI(email, password)
androidId = gapi.getAndroidId()
securityToken = gapi.getSecurityToken()
regid = gapi.register(app, senderId)
config.setConfigOption('gcmandroidId', str(androidId))
config.setConfigOption('gcmsecurityToken', str(securityToken))
config.setConfigOption('gcmregid', regid)
def connectGCM(androidId, securityToken):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
context = ssl.create_default_context()
conn = context.wrap_socket(s, server_hostname=mtalk.server)
conn.connect((mtalk.server, 5228))
conn.sendall(b'\x07')
data = conn.recv(1024)
mtalkpacket = mtalk.Packets()
loginRequestPacket = mtalkpacket.LoginRequestPacket(androidId, securityToken, hex(androidId)[2:])
conn.sendall(loginRequestPacket)
threading.Thread(target=send_heartbeats,
args=(conn, ),
daemon=True, ).start()
lastStreamId = 1
while True:
data = b''
while 1:
data += conn.recv(1024)
if len(data) != 0:
packetlength, varintlength = GoogleProtobufDecoder._DecodeVarint(data, 1)
if (len(data) - varintlength) == packetlength:
break
tag = data[0]
data = data[varintlength:]
print('message parsed with tag: ' + str(tag))
print(data)
if tag == 0x00:
print('Disconnected from GCM')
if tag == 0x03:
# Recieved LoginResponse
loginResponse = GoogleServicesFramework_pb2.LoginResponse()
loginResponse.ParseFromString(data)
if loginResponse.error.code == 0:
print('Login Successful')
elif tag == 0x07:
# Recieved DataMessageStanza
iqStanza = GoogleServicesFramework_pb2.IQStanza()
iqStanza.ParseFromString(data)
elif tag == 0x08:
# Recieved DataMessageStanza
dataMessageStanza = GoogleServicesFramework_pb2.DataMessageStanza()
dataMessageStanza.ParseFromString(data)
if dataMessageStanza.category == app:
for appdata in dataMessageStanza.appdata:
#print(appdata.key + ' - ' + appdata.value)
if appdata.key == 'message':
print('Received GCM message for textsecure')
receive_textsecure.receive_textsecure_message(appdata.value, q)
else:
pass
#if debug:
# for appdata in dataMessageStanza.appdata:
# print(appdata.key + ' - ' + appdata.value)
notificationRequestPacket = mtalkpacket.NotificationPacket(lastStreamId)
conn.sendall(notificationRequestPacket)
lastStreamId += 1
def send_heartbeats(conn):
while True:
time.sleep(120)
send_heartbeat(conn)
def send_heartbeat(conn):
mtalkpacket = mtalk.Packets()
heartBeatPacket = mtalkpacket.HeartBeatPacket()
conn.sendall(heartBeatPacket)
def start_gcm(GUIObject):
#get config here to prevent reading from database in a thread
androidId = int(config.getConfigOption('gcmandroidId'))
securityToken = int(config.getConfigOption('gcmsecurityToken'))
threading.Thread(target=connectGCM,
args=(androidId,
securityToken, ),
daemon=True,
).start()
threading.Thread(target=handle_queue,
args=(GUIObject, ),
daemon=True,
).start()
def handle_queue(GUIObject):
from gi.repository import Gdk
while True:
data = q.get()
Gdk.threads_enter()
GUIObject.received_message( data )
Gdk.threads_leave()
#GUIObject.set_chat_text(text)