summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/HTTPParser.py6
-rw-r--r--src/ogClient.py32
-rw-r--r--src/ogProcess.py2
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):