mirror of https://git.48k.eu/ogclient
124 lines
2.9 KiB
Python
124 lines
2.9 KiB
Python
import errno
|
|
import select
|
|
import socket
|
|
import time
|
|
import email
|
|
from io import StringIO
|
|
|
|
from src.HTTPParser import *
|
|
from src.ogRest import *
|
|
from enum import Enum
|
|
|
|
class State(Enum):
|
|
CONNECTING = 0
|
|
RECEIVING = 1
|
|
FORCE_DISCONNECTED = 2
|
|
|
|
class ogClient:
|
|
def __init__(self, ip, port):
|
|
self.ip = ip
|
|
self.port = port
|
|
|
|
def get_socket(self):
|
|
return self.sock
|
|
|
|
def get_state(self):
|
|
return self.state
|
|
|
|
def connect(self):
|
|
print ('connecting')
|
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.sock.setblocking(0)
|
|
self.state = State.CONNECTING
|
|
self.data = ""
|
|
self.trailer = False
|
|
self.content_len = 0
|
|
|
|
try:
|
|
self.sock.connect((self.ip, self.port))
|
|
except socket.error as err:
|
|
if err.errno == errno.EINPROGRESS:
|
|
return
|
|
elif err.errno == errno.ECONNREFUSED:
|
|
return
|
|
|
|
print ('Error connect ' + str(err))
|
|
|
|
def send(self, msg):
|
|
self.sock.send(bytes(msg, 'utf-8'))
|
|
return len(msg)
|
|
|
|
def connect2(self):
|
|
try:
|
|
self.sock.connect((self.ip, self.port))
|
|
except socket.error as err:
|
|
if err.errno == errno.EISCONN:
|
|
print ('connected')
|
|
self.state = State.RECEIVING
|
|
else:
|
|
print ('connection refused, retrying...')
|
|
self.state = State.CONNECTING
|
|
self.sock.close()
|
|
self.connect()
|
|
|
|
def receive(self):
|
|
try:
|
|
data = self.sock.recv(1024).decode('utf-8')
|
|
except socket.err as err:
|
|
print ('Error3 ' + str(err))
|
|
|
|
if len(data) == 0:
|
|
self.state = State.CONNECTING
|
|
self.sock.close()
|
|
self.connect()
|
|
|
|
self.data = self.data + data
|
|
httpparser = HTTPParser()
|
|
ogrest = ogRest()
|
|
|
|
if not self.trailer:
|
|
if self.data.find("\r\n") > 0:
|
|
# https://stackoverflow.com/questions/4685217/parse-raw-http-headers
|
|
request_line, headers_alone = self.data.split('\n', 1)
|
|
headers = email.message_from_file(StringIO(headers_alone))
|
|
|
|
if 'content-length' in headers.keys():
|
|
self.content_len = int(headers['content-length'])
|
|
|
|
self.trailer = True
|
|
|
|
if self.trailer and len(self.data) >= self.content_len:
|
|
httpparser.parser(self.data)
|
|
ogrest.processOperation(httpparser.getRequestOP(), httpparser.getURI(), self)
|
|
|
|
# Cleanup state information from request
|
|
self.data = ""
|
|
self.content_len = 0
|
|
self.trailer = False
|
|
|
|
def disconnect(self):
|
|
self.state = State.FORCE_DISCONNECTED
|
|
self.sock.close()
|
|
|
|
def run(self):
|
|
while 1:
|
|
sock = self.get_socket()
|
|
state = self.get_state()
|
|
|
|
if state == State.CONNECTING:
|
|
readset = [ sock ]
|
|
writeset = [ sock ]
|
|
elif state == State.FORCE_DISCONNECTED:
|
|
return 0
|
|
else:
|
|
readset = [ sock ]
|
|
writeset = [ ]
|
|
|
|
readable, writable, exception = select.select(readset, writeset, [ ])
|
|
if state == State.CONNECTING and sock in writable:
|
|
self.connect2()
|
|
elif state == State.RECEIVING and sock in readable:
|
|
self.receive()
|
|
else:
|
|
print ('bad state' + str(state))
|