#! /usr/bin/python -Es
# Copyright (C) 2012 Red Hat 
# see file 'COPYING' for use and warranty information
#
# setrans is a tool for analyzing process transistions in SELinux policy
#
#    This program is free software; you can redistribute it and/or
#    modify it under the terms of the GNU General Public License as
#    published by the Free Software Foundation; either version 2 of
#    the License, or (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program; if not, write to the Free Software
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA     
#                                        02111-1307  USA
#
#  
import sys
import seobject
import setools
search=setools.sesearch
seinfo=setools.seinfo

portRec = seobject.portRecords()
portrecs = portRec.get_all_by_type()
portrecsbynum = portRec.get_all()
port_types =  setools.seinfo(setools.ATTRIBUTE,"port_type")[0]["types"]

def get_types(src, tclass, perm):
    allows=search([setools.ALLOW],{setools.SCONTEXT:src,setools.CLASS:tclass, setools.PERMS:perm})
    nlist=[]
    if allows:
        for i in map(lambda y: y[setools.TCONTEXT], filter(lambda x: set(perm).issubset(x[setools.PERMS]), allows)):
            if i not in nlist:
                nlist.append(i)
    return nlist
   

def get_network_connect(src, protocol, perm):
    d={}
    tlist = get_types(src, "%s_socket" % protocol, [perm])
    if len(tlist) > 0:
        if "port_type" in tlist:
            d[(src,protocol,perm)] = ["all ports"]
            return d

        d[(src,protocol,perm)] = []

        for i in tlist:
            if i == "ephemeral_port_type":
                if "unreserved_port_type" in tlist:
                    continue
                i = "ephemeral_port_t"
            if i == "unreserved_port_t":
                if "unreserved_port_type" in tlist:
                    continue
                if "port_t" in tlist:
                    continue
            if i == "port_t":
                d[(src,protocol,perm)].append("all ports with out defined types")
            elif i == "unreserved_port_type":
                d[(src,protocol,perm)].append("%s: all ports > 1024" % i)
            elif i == "reserved_port_type":
                d[(src,protocol,perm)].append("%s: all ports < 1024" % i)
            elif i == "rpc_port_type":
                d[(src,protocol,perm)].append("%s: all ports > 500 and  < 1024" % i)
            else:
                try:
                    d[(src,protocol,perm)].append("%s: %s" % (i, ",".join(portrecs[(i, protocol)])))
                except KeyError:
                    pass
    return d

def print_net(src, protocol, perm):
    portdict = get_network_connect(src, protocol, perm)
    if len(portdict) > 0:
        print "%s: %s %s" % (src, protocol, perm)
        for p in portdict:
            for recs in portdict[p]:
                print "\t" + recs


if __name__ == '__main__':
    setype = sys.argv[1]
    if setype.isdigit():
        port = int(setype)

        found = False
        for i in portrecsbynum:
            if i[0] <= port and port <= i[1]:
                if i[0] == i[1]:
                    range = i[0]
                else:
                    range = "%s-%s" % (i[0], i[1])
                found = True
                print "%s: %s %s %s" % (setype, i[2], portrecsbynum[i][0], range)
        if not found:
            if port < 500:
                print "Undefined reserved port type" 
            else:
                print "Undefined port type" 
    elif setype in port_types:
        if (setype,'tcp') in portrecs.keys():
            print "%s: tcp: %s" % (setype, ",".join(portrecs[setype,'tcp']))
        if (setype,'udp') in portrecs.keys():
            print "%s: udp: %s" % (setype, ",".join(portrecs[setype,'udp']))
    else:
        print_net(setype, "tcp", "name_connect")
        for net in ("tcp", "udp"):
            print_net(setype, net, "name_bind")