At the Pentester Academy I took a course called Pentesting with Python.  It’s a great course… if you have an interest in Python and Security, I highly recommend it.


The instructor (Vivek Ramachandran) provides student exercises at the end of most of the lectures.  In lesson 16 (on signaling) he asks the students to write a Python script that does the following:

  1. Create a TCP server that binds to a socket
  2. Create signals to ensure it automatically shuts down after a pre-configured duration which is given from the command line as an argument (i.e. -s 10
  3. Shutdown the server after listening on the port for X seconds

My solution required a bit of research… I wasn’t sure how to create a server in Python (rusty on this) and also needed to research python signals on alarms.  I also wasn’t sure how to pass in command line arguments to the application.


For reference I used the following resources:

Raw Attempt

My raw attempt here isn’t very OO… it simply gets the job done.  I’d like to revisit it in the future and make it more OO.

import socket
import signal
import sys, getopt

# Defining host and port to server on
HOST, PORT = '', 777

def handler(signum, frame):
  print 'Starting...'
  raise IOError("Timmer tripped, shutting down...")

time_seconds = 0

# Catch command line argument -s
  opts, args = getopt.getopt(sys.argv[1:],"hs:")
except getopt.GetoptError:
for opt, arg in opts:
if opt in ("-s"):
time_seconds = arg


listen_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
listen_socket.bind((HOST, PORT))
# Listen for connections
print 'Serving on port %s ' % PORT
while True:
  client_connection, client_address = listen_socket.accept()
  request = client_connection.recv(1024)
  print request


Screen Shot 2016-07-27 at 6.02.21 PM


No responses yet

Leave a Reply

Your email address will not be published.