diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/HTTPParser.py | 6 | ||||
| -rw-r--r-- | src/ogClient.py | 32 | ||||
| -rw-r--r-- | src/ogProcess.py | 2 |
3 files changed, 21 insertions, 19 deletions
diff --git a/src/HTTPParser.py b/src/HTTPParser.py index bbced43..7c64805 100644 --- a/src/HTTPParser.py +++ b/src/HTTPParser.py @@ -1,5 +1,5 @@ -from mimetools import Message -from StringIO import StringIO +import email +from io import StringIO class HTTPParser: def __init__(self): @@ -14,7 +14,7 @@ class HTTPParser: def parser(self,data): self.requestLine, self.headersAlone = data.split('\n', 1) - self.headers = Message(StringIO(self.headersAlone)) + self.headers = email.message_from_file(StringIO(self.headersAlone)) if 'host' in self.headers.keys(): self.host = self.headers['host'] diff --git a/src/ogClient.py b/src/ogClient.py index e28193b..4fbdfab 100644 --- a/src/ogClient.py +++ b/src/ogClient.py @@ -2,9 +2,11 @@ import errno import select
import socket
import time
+import email
+from io import StringIO
-from HTTPParser import *
-from ogProcess import *
+from src.HTTPParser import *
+from src.ogProcess import *
from enum import Enum
class State(Enum):
@@ -23,7 +25,7 @@ class ogClient: return self.state
def connect(self):
- print "connecting"
+ print ('connecting')
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.setblocking(0)
self.state = State.CONNECTING
@@ -33,32 +35,32 @@ class ogClient: try:
self.sock.connect((self.ip, self.port))
- except socket.error, err:
+ except socket.error as err:
if err.errno == errno.EINPROGRESS:
return
elif err.errno == errno.ECONNREFUSED:
return
- print "Error connect " + str(err)
+ print ('Error connect ' + str(err))
def connect2(self):
try:
self.sock.connect((self.ip, self.port))
- except socket.error, err:
+ except socket.error as err:
if err.errno == errno.EISCONN:
- print "connected"
+ print ('connected')
self.state = State.RECEIVING
else:
- print "connection refused, retrying..."
+ print ('connection refused, retrying...')
self.state = State.CONNECTING
self.sock.close()
self.connect()
def receive(self):
try:
- data = self.sock.recv(1024)
- except socket.err, err:
- print "Error3 " + str(err)
+ 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
@@ -73,7 +75,7 @@ class ogClient: 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 = Message(StringIO(headers_alone))
+ headers = email.message_from_file(StringIO(headers_alone))
if 'content-length' in headers.keys():
self.content_len = int(headers['content-length'])
@@ -83,9 +85,9 @@ class ogClient: if self.trailer and len(self.data) >= self.content_len:
httpparser.parser(self.data)
if not ogprocess.processOperation(httpparser.getRequestOP(), httpparser.getURI()):
- self.sock.send("HTTP/1.0 400 Bad request\r\n\r\n")
+ self.sock.send(bytes('HTTP/1.0 400 Bad request\r\n\r\n', 'utf-8'))
else:
- self.sock.send("HTTP/1.0 200 OK\r\n\r\n")
+ self.sock.send(bytes('HTTP/1.0 200 OK\r\n\r\n', 'utf-8'))
# Cleanup state information from request
self.data = ""
@@ -110,4 +112,4 @@ class ogClient: elif state == State.RECEIVING and sock in readable:
self.receive()
else:
- print "bad state" + str(state)
+ print ('bad state' + str(state))
diff --git a/src/ogProcess.py b/src/ogProcess.py index 143f49a..6601b8c 100644 --- a/src/ogProcess.py +++ b/src/ogProcess.py @@ -3,7 +3,7 @@ import platform import time if platform.system() == 'Linux': - from linux import ogOperations + from src.linux import ogOperations class ogProcess(): def processOperation(self, op, URI): |