finalized the tcpconnect service. should be ready for takeoff now.... ;)

This commit is contained in:
Johannes Findeisen 2013-08-15 02:03:12 +02:00
commit 06e29776b2

View file

@ -11,7 +11,6 @@ from lib.services.service import Service
class TcpconnectService(Service):
def __init__(self, **kwargs):
#Service.__init__(self, **kwargs)
super(TcpconnectService, self).__init__(**kwargs)
args = self.get_arguments()
@ -28,19 +27,24 @@ class TcpconnectService(Service):
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
except socket.error, msg:
jobInfo.set_errorcode(1)
jobInfo.set_message("Could not create socket to host: " + jobInfo.get_host() + " to port: " + str(self.port) + " (" + str(msg) + ")")
jobInfo.set_message("Could not create socket to host: " + jobInfo.get_host() + " to port: " +
str(self.port) + " (" + str(msg) + ")")
try:
sock.connect((jobInfo.get_host(), self.port))
except socket.error, msg:
jobInfo.set_errorcode(2)
jobInfo.set_message("Could not establish connection to host: " + jobInfo.get_host() + " to port: " + str(self.port) + " (" + str(msg) + ")")
jobInfo.set_message("Could not establish connection to host: " + jobInfo.get_host() + " to port: " +
str(self.port) + " (" + str(msg) + ")")
if jobInfo.get_errorcode() == -1:
jobInfo.set_execution_successful(True)
jobInfo.set_errorcode(0)
jobInfo.set_message("Connection successful established to host: " + jobInfo.get_host() + " to port: " + str(self.port))
jobInfo.set_message("Connection successful established to host: " + jobInfo.get_host() + " to port: " +
str(self.port))
sock.close()
def create(kwargs):
return TcpconnectService(**kwargs)