def executeSplitMsg(self, msg):
if msg.getIndex() == 0:
self.splitHandler = SplitHandler(msg.getTotal())
if self.splitHandler.insert(msg):
self.connectionHandler.sendMsg(Messages.ReciveSplitMsg(), (self.ip, self.port))
elif self.splitHandler is not None:
if self.splitHandler.insert(msg):
if self.splitHandler.isReady():
s = self.splitHandler.combineMsg()
self.splitHandler = None
newMsg = self.encoDeco.decodeString(s)
print('new msg', newMsg)
if newMsg is not None:
self.executeMsg(newMsg)
else:
self.connectionHandler.sendMsg(Messages.ReciveSplitMsg(), (self.ip, self.port))
def broadCast(self):
ips = self.connectionHandler.getMyBroadcastIps()
for address in ips:
ip = (address, 1250)
publicKey = 'send BroadCast to phones'
publicKey = hashlib.sha1(bytes(publicKey, 'utf-8')).hexdigest()
msg = Messages.SearchMessage(publicKey)
self.connectionHandler.sendMsg(msg, ip)
def decodeZoomMsg(self, buffer):
offset = 0
opcode, state = struct.unpack_from('1s10s', buffer, offset)
state = str(state, "utf-8")
state = cleanString(state)
if state == 'true':
state = True
else:
state = False
return Messages.ZoomMsg(state)
def createSplitMsgs(self):
split = math.ceil(len(self.encodedMsg) / MAX_UDP_SIZE)
while len(self.encodedMsg) % split != 0:
self.encodedMsg += b'0'
size = len(self.encodedMsg)
part = int(size / split)
for i in range(split):
data = self.encodedMsg[i * part:(i + 1) * part - 1]
splitMsg = Messages.SplitMsg(i, split, data)
self.msgs.append(splitMsg)
def decodeRollerMsg(self, buffer):
offset = 1
size, data = struct.unpack_from('10sc', buffer, offset)
size = str(size, "utf-8")
size = int(cleanString(size))
size, data = struct.unpack_from('10s' + str(size) + 's', buffer,
offset)
data = str(data, "utf-8")
data = json.loads(data)
return Messages.RollerMoveMsg(data)
def decodeGenerationMsgFromString(self, s):
print('here !! :D')
offset = 1
id = s[offset:offset + 40]
id = cleanString(id)
offset += 40
size = s[offset:offset + 10]
size = int(cleanString(size))
offset += 10
data = s[offset:offset + size]
data = json.loads(data)
return Messages.GenerationMessage(id, data)
def decodeMouseClickMsg(self, buffer):
offset = 1
button, state = struct.unpack_from('10s10s', buffer, offset)
button = str(button, "utf-8")
button = cleanString(button)
state = str(state, "utf-8")
state = cleanString(state)
if state == 'true':
state = True
else:
state = False
return Messages.MouseClick(button, state)
def decodeGenerationMsg(self, buffer):
offset = 1
id, rest = struct.unpack_from('40sc', buffer, offset)
id = str(id, "utf-8")
offset += 40
size, rest = struct.unpack_from('10sc', buffer, offset)
size = str(size, "utf-8")
size = int(cleanString(size))
size, data = struct.unpack_from('10s' + str(size) + 's', buffer,
offset)
data = str(data, "utf-8")
data = json.loads(data)
return Messages.GenerationMessage(id, data)
def run(self):
self.running = True
while self.running:
img = self.getCenteredScreen()
arrBytes = io.BytesIO()
img.save(arrBytes, format='JPEG')
value1 = arrBytes.getvalue()
size1 = len(value1)
print('send jpeg')
msg = Messages.ViewerMsg(value1, size1, 'JPEG')
try:
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
except Exception as e:
print(e)
continue
def decodeFileMsgFromString(self, s):
offset = 1
name = s[offset:offset + 40]
name = cleanString(name)
offset += 40
date = s[offset:offset + 10]
date = cleanString(date)
offset += 10
fileSize = s[offset:offset + 15]
fileSize = cleanString(fileSize)
offset += 15
size = s[offset:offset + 20]
size = int(cleanString(size))
offset += 20
data = s[offset:offset + size]
return Messages.FileMsg(name, date, fileSize, data)
def decodeSplitMsg(self, buffer):
offset = 1
index, rest = struct.unpack_from('10sc', buffer, offset)
index = str(index, "utf-8")
index = int(cleanString(index))
offset += 10
total, rest = struct.unpack_from('10sc', buffer, offset)
total = str(total, "utf-8")
total = int(cleanString(total))
offset += 10
size, data = struct.unpack_from('10sc', buffer, offset)
size = str(size, "utf-8")
size = int(cleanString(size))
size, data = struct.unpack_from('10s' + str(size) + 's', buffer,
offset)
data = str(data, "utf-8")
return Messages.SplitMsg(index, total, data)
def decodeFoundMsg(self, buffer):
# print('start decode Found Msg')
offset = 1
sysName, rest = struct.unpack_from('40sc', buffer, offset)
sysName = str(sysName, "utf-8")
sysName = cleanString(sysName)
offset += 40
deviceId, rest = struct.unpack_from('40sc', buffer, offset)
deviceId = str(deviceId, "utf-8")
deviceId = cleanString(deviceId)
offset += 40
deviceName, batrery = struct.unpack_from('40s3s', buffer, offset)
deviceName = str(deviceName, "utf-8")
deviceName = cleanString(deviceName)
batrery = str(batrery, "utf-8")
batrery = cleanString(batrery)
# print('sysName: ', sysName, ' id: ', deviceId, ' deviceName: ', deviceName, ' battery: ', batrery)
return Messages.FoundMessage(sysName, deviceId, deviceName, batrery)
def decodeFileMsg(self, buffer):
offset = 1
name, rest = struct.unpack_from('40sc', buffer, offset)
name = cleanString(str(name, "utf-8"))
offset += 40
date, rest = struct.unpack_from('10sc', buffer, offset)
date = cleanString(str(date, "utf-8"))
offset += 10
fileSize, rest = struct.unpack_from('15sc', buffer, offset)
fileSize = cleanString(str(fileSize, "utf-8"))
offset += 15
size, data = struct.unpack_from('20sc', buffer, offset)
size = str(size, "utf-8")
size = int(cleanString(size))
size, data = struct.unpack_from('20s' + str(size) + 's', buffer,
offset)
data = str(data, "utf-8")
return Messages.FileMsg(name, date, fileSize, data)
def sendAckLogout(self):
self.logged = False
msg = Messages.AckLogoutMsg()
print('send ack logout msg')
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
fin = False
while fin is False:
try:
msg, ip = self.connectionHandler.accept(msgSize=128000) # 2^17
if msg.opcode == Messages.FIN_OPCODE:
print('recived fin msg')
fin = True
else:
print('send ack logout msg')
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
except Exception as e:
continue
self.logoutFunc()
def decodeToouchMoveMsg(self, buffer):
offset = 1
last, rest = struct.unpack_from('10sc', buffer, offset)
last = str(last, "utf-8")
last = cleanString(last)
if last == 'true':
last = True
else:
last = False
offset += 10
size, rest = struct.unpack_from('10sc', buffer, offset)
size = str(size, "utf-8")
size = int(cleanString(size))
size, data = struct.unpack_from('10s' + str(size) + 's', buffer,
offset)
data = str(data, "utf-8")
data = json.loads(data)
return Messages.TouchMoveMsg(last, data)
def decode(self, buffer):
opCode, rest = struct.unpack_from('1cc', buffer, 0)
opCode = str(opCode, "utf-8")
if opCode == chr(Messages.FOUND_OPCODE):
return self.decodeFoundMsg(buffer)
# Not Supported Yet By Client
elif opCode == chr(Messages.GENERATION_OPCODE):
return self.decodeGenerationMsg(buffer)
elif opCode == chr(Messages.MOUSE_CLICK_OPCODE):
return self.decodeMouseClickMsg(buffer)
elif opCode == chr(Messages.MOUSE_MOVE_OPCODE):
return self.decodeMouseMoveMsg(buffer)
elif opCode == chr(Messages.SPLIT_OPCODE):
return self.decodeSplitMsg(buffer)
elif opCode == chr(Messages.RECIVE_SPLIT_OPCODE):
return Messages.ReciveSplitMsg()
elif opCode == chr(Messages.TOUCH_MOVE_OPCODE):
return self.decodeToouchMoveMsg(buffer)
elif opCode == chr(Messages.ROLLER_MOVE_OPCODE):
return self.decodeRollerMsg(buffer)
elif opCode == chr(Messages.FILE_OPCODE):
return self.decodeFileMsg(buffer)
elif opCode == chr(Messages.LOGOUT_OPCODE):
print('recived logout msg')
return Messages.LogoutMsg()
elif opCode == chr(Messages.ACKLOGOUT_OPCODE):
return Messages.AckLogoutMsg()
elif opCode == chr(Messages.FIN_OPCODE):
print('recived fin msg')
return Messages.FinMsg()
elif opCode == chr(Messages.MOUSEY_BATTERY_OPCODE):
return self.decodeBattryMsg(buffer)
elif opCode == chr(Messages.START_VIEWER_OPCODE):
return Messages.StartViewerMsg()
elif opCode == chr(Messages.END_VIEWER_OPCODE):
return Messages.EndViewerMsg()
elif opCode == chr(Messages.ZOOM_OPCODE):
return self.decodeZoomMsg(buffer)
return None
def connect(self):
privateKey = 'noni avraham ofir'
privateKey = hashlib.sha1(bytes(privateKey, 'utf-8')).hexdigest()
msg = Messages.ConnectMessage(privateKey)
print('send Connect Mousey')
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
def sendFin(self):
msg = Messages.FinMsg()
print('send fin msg')
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
self.logged = False
self.logoutFunc()
def sendLogoutMsg(self):
msg = Messages.LogoutMsg()
print('send logout msg')
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
def stopViewer(self):
self.running = False
msg = Messages.AckEndViewerMsg()
self.connectionHandler.sendMsg(msg, (self.ip, self.port))
def decodeBattryMsg(self, buffer):
offset = 0
opcode, batrery = struct.unpack_from('1s3s', buffer, offset)
batrery = str(batrery, "utf-8")
batrery = cleanString(batrery)
return Messages.MouseyBatteryMsg(batrery)