137 lines
3.9 KiB
Python
137 lines
3.9 KiB
Python
import socket
|
|
import platform
|
|
|
|
import websocket
|
|
import ssl
|
|
from PyQt4.QtCore import QString
|
|
|
|
from constants import *
|
|
|
|
printPackets = False # Enable for debugging packets sent and received
|
|
|
|
class AOtcpSocket(object):
|
|
def __init__(self):
|
|
self.sock = socket.socket()
|
|
self.isWS = False
|
|
|
|
def connect(self, ip, port):
|
|
try:
|
|
self.sock.connect((ip, port))
|
|
except:
|
|
return False
|
|
|
|
self.sock.settimeout(0.1)
|
|
return True
|
|
|
|
def recv(self):
|
|
tempdata = ""
|
|
gotIt = False
|
|
while not gotIt:
|
|
try:
|
|
contents = self.sock.recv(16384)
|
|
except (socket.timeout, socket.error) as err:
|
|
error = err.args[0]
|
|
if error == "timed out" or error == 10035 or err.errno == 11:
|
|
return -2, []
|
|
else: # connection lost
|
|
return -1, []
|
|
|
|
if not contents.endswith("%"):
|
|
tempdata += contents
|
|
continue
|
|
else:
|
|
gotIt = True
|
|
if tempdata:
|
|
contents = tempdata + contents
|
|
tempdata = ""
|
|
|
|
totals = contents.split('%')
|
|
del totals[-1]
|
|
for i in range(len(totals)):
|
|
totals[i] = totals[i].split("#")
|
|
del totals[i][-1]
|
|
if printPackets:
|
|
try:
|
|
print "[packet] <-", totals[0]
|
|
except:
|
|
print "(unable to print)"
|
|
return 0, totals
|
|
|
|
def send(self, data):
|
|
if printPackets:
|
|
try:
|
|
print "[packet] ->", data
|
|
except:
|
|
print "(unable to print)"
|
|
# Qt pls
|
|
if isinstance(data, QString):
|
|
#return self.sock.send(unicode(data).encode('utf-8'))
|
|
return self.sock.send(str(data.toUtf8()))
|
|
else:
|
|
return self.sock.send(data.encode('utf-8'))
|
|
|
|
def close(self):
|
|
self.sock.close()
|
|
|
|
class AOwebSocket(object):
|
|
def __init__(self):
|
|
self.sock = websocket.WebSocket(sslopt={"cert_reqs": ssl.CERT_NONE})
|
|
self.isWS = True
|
|
self.isSecure = False
|
|
self.header = {
|
|
"User-Agent": "AO2XP %s" % GAME_VERSION
|
|
}
|
|
|
|
def connect(self, ip, port, secure_port):
|
|
try:
|
|
if secure_port:
|
|
try:
|
|
print "[debug]", "Trying secure websocket..."
|
|
self.sock.connect("wss://%s:%s" % (ip, secure_port), header=self.header)
|
|
self.isSecure = True
|
|
except:
|
|
print "[debug]", "Connecting to secure websocket failed. Trying websocket..."
|
|
self.sock.connect("ws://%s:%s" % (ip, port), header=self.header)
|
|
else:
|
|
print "[debug]", "Trying websocket..."
|
|
self.sock.connect("ws://%s:%s" % (ip, port), header=self.header)
|
|
except:
|
|
return False
|
|
|
|
self.sock.settimeout(0.1)
|
|
return True
|
|
|
|
def recv(self):
|
|
try:
|
|
contents = self.sock.recv()
|
|
except websocket.WebSocketTimeoutException:
|
|
return -2, []
|
|
except websocket.WebSocketConnectionClosedException:
|
|
return -1, []
|
|
except Exception as e:
|
|
return -3, e
|
|
|
|
totals = contents.split('%')
|
|
del totals[-1]
|
|
for i in range(len(totals)):
|
|
totals[i] = totals[i].split("#")
|
|
del totals[i][-1]
|
|
|
|
if printPackets:
|
|
try:
|
|
print "[packet] <-", totals[0]
|
|
except:
|
|
print "(unable to print)"
|
|
return 0, totals
|
|
|
|
def send(self, data):
|
|
if printPackets:
|
|
try:
|
|
print "[packet] ->", data
|
|
except:
|
|
print "(unable to print)"
|
|
return self.sock.send(unicode(data))
|
|
|
|
def close(self):
|
|
self.sock.close()
|