Template for interacting with a netcat server

netcat.py

# A basic template for solving netcat challenges
# Author: aaron@lintile.com

import socket
import string

class Netcat:
    def __init__(self, ip, port):
        self.data_buffer = ''
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.connect((ip, port))

    def read(self, length = 1024):
        if len(self.data_buffer) < length:
            increment = self.socket.recv(length - len(self.data_buffer))
            self.data_buffer += increment

        rval = self.data_buffer[:length]
        self.data_buffer = self.data_buffer[length:]
        return rval
 
    def read_until_found(self, search_string):
        """ Read into the buffer until we encounter the search string """
        while not search_string in self.data_buffer:
            increment = self.socket.recv(1024)
            self.data_buffer += increment
 
        pos = self.data_buffer.find(search_string)
        rval = self.data_buffer[:pos + len(search_string)]
        self.data_buffer = self.data_buffer[pos + len(search_string):]
 
        return rval
 
    def write(self, data):
        self.socket.send(data)
    
    def close(self):
        self.socket.close()

import re
import itertools
import hashlib

nc = Netcat('insert.domain.here', 11235)

data = nc.read_until_found('?')

#TODO: Solve the challenge

nc.close()