ANTFARM/PCAP/PCAP 脚本

1. ! /usr/bin/ruby

 #
 # Copyright (2008) Sandia Corporation.
 # Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
 # the U.S. Government retains certain rights in this software.
 #
 # Original Author: Michael Berg, Sandia National Laboratories <mjberg@sandia.gov>
 # Modified By: Bryan T. Richardson, Sandia National Laboratories <btricha@sandia.gov>
 # Further Modified By: Melissa Myerly & Cassandra Trevino, Sandia National Laboratories
 # This library is free software; you can redistribute it and/or modify it
 # under the terms of the GNU Lesser General Public License as published by
 # the Free Software Foundation; either version 2.1 of the License, or (at
 # your option) any later version.
 #
 # This library 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 Lesser General Public License for more
 # details.
 #
 # You should have received a copy of the GNU Lesser General Public License
 # along with this library; if not, write to the Free Software Foundation, Inc.,
 # 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 
 #
 require 'pcap'
 #
 def print_help
   puts "Usage: antfarm [options] pcap [options] parse-pcap-file [options] <pcap file>"
   puts
   puts "This script parses a libPcap file containing traffic capture data,"
   puts "creating an IP interface for each endpoint and a traffic object"
   puts "for the traffic between them.  Node device types are set to 'PCAP',"
   puts "as well as traffic descriptions."
   puts 
   puts "Script Options:"
   puts "  --create-new-networks    Create new networks if networks containing the"
   puts "                           source or destination address don't already exist."
 end
 #
 def parse(file, options = [])
   puts file
 #
 cap = Pcap::Capture.open_offline(file)
 cap.each do |pkt|
   if pkt.ip?
    source_addr = pkt.src.to_num_s
    target_addr = pkt.dst.to_num_s
    source_in_networks = false               
    target_in_networks = false
    network_list.each do |network|
     net = Antfarm::IPAddrExt.new(network)
     if net.network_in_network?(source_addr)
      source_in_networks = true
      break
      end
    end
   network_list.each do |network|
    net = Antfarm::IPAddrExt.new(network)
    if net.network_in_network?(target_addr)
     target_in_networks = true
     break
    end
   end

解析 PCAP 文件并确定每个源或目标地址是否已属于数据库中的网络。在使用“创建新网络”选项时会执行此操作。

 if source_in_networks || target_in_networks
  if options.include?('--create-new-networks')     
   source_iface = IpInterface.find_or_initialize_by_address(source_addr)
   if source_iface.new_record?
    source_iface.node_name = source_addr             #Assigns the source_iface.node_name the current source address
    source_iface.node_device_type = 'PCAP'           #Informs the database this is coming from PCAP information/file
    source_iface.save false
   end
  target_iface = IpInterface.find_or_initialize_by_address(target_addr)
  if target_iface.new_record?
   target_iface.node_name = target_addr
   target_iface.node_device_type = 'PCAP'
   target_iface.save false
  end
 if pkt.tcp? || pkt.udp?
  traffic = Traffic.first(:conditions => { :source_layer3_interface_id => source_iface.layer3_interface.id,
            :target_layer3_interface_id => target_iface.layer3_interface.id,
            :port => pkt.dport})
 else
  traffic = Traffic.first(:conditions => { :source_layer3_interface_id => source_iface.layer3_interface.id,
            :target_layer3_interface_id => target_iface.layer3_interface.id })
 end
 unless traffic
 if pkt.tcp? || pkt.udp?
  traffic = Traffic.create :source_layer3_interface => source_iface.layer3_interface,
            :target_layer3_interface => target_iface.layer3_interface,
            :port => pkt.dport,
            :description => "PCAP"
 else
  traffic = Traffic.create :source_layer3_interface => source_iface.layer3_interface,
            :target_layer3_interface => target_iface.layer3_interface,
            :description => "PCAP"
 end
  puts "Added traffic -- #{source_addr} ==> #{target_addr}"  #Prints traffic added to the database to the screen.
end
else
 source_net = Layer3Network.network_containing(source_addr)
 target_net = Layer3Network.network_containing(target_addr)
 if source_net && target_net
  source_iface = IpInterface.find_or_initialize_by_address(source_addr)
  if source_iface.new_record?
   source_iface.node_name = source_addr
   source_iface.node_device_type = 'PCAP'
   source_iface.save false
  end
 target_iface = IpInterface.find_or_initialize_by_address(target_addr)
  if target_iface.new_record?
   target_iface.node_name = target_addr
   target_iface.node_device_type = 'PCAP'
   target_iface.save false
  end
 if pkt.tcp? || pkt.udp?
  traffic = Traffic.first(:conditions => { :source_layer3_interface_id => source_iface.layer3_interface.id,
            :target_layer3_interface_id => target_iface.layer3_interface.id,
            :port => pkt.dport})
 else
  traffic = Traffic.first(:conditions => { :source_layer3_interface_id => source_iface.layer3_interface.id,
            :target_layer3_interface_id =>target_iface.layer3_interface.id })
 end
 unless traffic
  if pkt.tcp? || pkt.udp?
   traffic = Traffic.create :source_layer3_interface => source_iface.layer3_interface,
             :target_layer3_interface => target_iface.layer3_interface,
             :port => pkt.dport,
             :description => "PCAP"
  else
   traffic = Traffic.create :source_layer3_interface => source_iface.layer3_interface,
             :target_layer3_interface => target_iface.layer3_interface,:description => "PCAP"
  end
   puts "Added traffic -- #{source_addr} ==> #{target_addr}"
  end
   end
    end
     end
    end
   end
 end
 #
 if ['-h', '--help'].include?(ARGV[0])
  print_help
 else
  file_dir = ARGV.pop
  if File.directory?(file_dir)
   Find.find(file_dir) do |path|
   if File.file?(path)
    parse(path, ARGV)
   end
  end
 else
  parse(file_dir, ARGV)
 end
end