def check_ports(ports):
    """
    Check ports to determine running gobuster and/or smbmap.
    0 = none
    1 = gobuster
    2 = smbmap
    3 = both
    """
    web = 0
    smb = 0

    # Determine if we ports open
    if (('80' in ports) or ('443' in ports)):
        web = 1

    # Determine if SMB ports open
    if (('139' in ports) or ('445' in ports)):
        smb = 2
    return web + smb