simple_lambda_portscan.py

import json, socket

print('Loading function')


def portScan(ip,ports) :
    r = []
    for p in ports :
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        socket.setdefaulttimeout(1)
          
        # returns an error indicator
        result = s.connect_ex((ip,p))
        if result ==0:
            r.append(p)    
            print("Port {} is open".format(p))
        s.close()

    return r

def lambda_handler(event, context):
    print("Received event: " + json.dumps(event, indent=2))
    body = json.loads(event["body"])
    result = portScan(body["ip"],body["ports"])
    return json.dumps(result)