github
/
fail2ban
mirror of https://github.com/fail2ban/fail2ban
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix up for client/server socket for python3

pull/128/merge^2
Steven Hiscocks 2013-02-24 10:10:15 +00:00
parent 6f4da8f3c4
commit 46a2b6e428
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
client/csocket.py
View File

@ -29,10 +29,13 @@ __license__ = "GPL"
#from cPickle import dumps, loads, HIGHEST_PROTOCOL #from cPickle import dumps, loads, HIGHEST_PROTOCOL
from pickle import dumps, loads, HIGHEST_PROTOCOL from pickle import dumps, loads, HIGHEST_PROTOCOL
import socket import socket, sys
class CSocket: class CSocket:
if sys.version_info >= (3,):
END_STRING = b"<F2B_END_COMMAND>"
else:
END_STRING = "<F2B_END_COMMAND>" END_STRING = "<F2B_END_COMMAND>"
def __init__(self, sock = "/var/run/fail2ban/fail2ban.sock"): def __init__(self, sock = "/var/run/fail2ban/fail2ban.sock"):
@ -52,6 +55,9 @@ class CSocket:
#@staticmethod #@staticmethod
def receive(sock): def receive(sock):
if sys.version_info >= (3,):
msg = b''
else:
msg = '' msg = ''
while msg.rfind(CSocket.END_STRING) == -1: while msg.rfind(CSocket.END_STRING) == -1:
chunk = sock.recv(6) chunk = sock.recv(6)

6
server/asyncserver.py
View File

@ -42,6 +42,9 @@ logSys = logging.getLogger("fail2ban.server")
class RequestHandler(asynchat.async_chat): class RequestHandler(asynchat.async_chat):
if sys.version_info >= (3,):
END_STRING = b"<F2B_END_COMMAND>"
else:
END_STRING = "<F2B_END_COMMAND>" END_STRING = "<F2B_END_COMMAND>"
def __init__(self, conn, transmitter): def __init__(self, conn, transmitter):
@ -62,6 +65,9 @@ class RequestHandler(asynchat.async_chat):
def found_terminator(self): def found_terminator(self):
# Joins the buffer items. # Joins the buffer items.
if sys.version_info >= (3,):
message = loads(b"".join(self.__buffer))
else:
message = loads("".join(self.__buffer)) message = loads("".join(self.__buffer))
# Gives the message to the transmitter. # Gives the message to the transmitter.
message = self.__transmitter.proceed(message) message = self.__transmitter.proceed(message)