48 lines
1.4 KiB
Ruby
48 lines
1.4 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'pcaprub'
|
||
|
|
require 'socket'
|
||
|
|
require_relative 'databasemanager'
|
||
|
|
require_relative 'logg_man'
|
||
|
|
require_relative 'redis_queue'
|
||
|
|
|
||
|
|
# Class used to capture packets and not much else
|
||
|
|
class PacketCapture
|
||
|
|
INTERFACE_NAME = 'netrave0'
|
||
|
|
|
||
|
|
def initialize(queue, logger)
|
||
|
|
@loggman = logger
|
||
|
|
@loggman.log_info("Initializing packet capture for #{INTERFACE_NAME}...")
|
||
|
|
@capture = Pcap.open_live(INTERFACE_NAME, 65_535, true, 1)
|
||
|
|
@capture.setfilter('')
|
||
|
|
@loggman.log_info('Packet capture initialized successfully!')
|
||
|
|
@queue = queue
|
||
|
|
end
|
||
|
|
|
||
|
|
def start_capture_loop # rubocop:disable Metrics/MethodLength
|
||
|
|
@loggman.log_info("Starting packet capture loop for #{@interface}...")
|
||
|
|
packet_count = 0
|
||
|
|
begin
|
||
|
|
@loggman.log_info("Packet capture loop started for #{@interface}...")
|
||
|
|
@capture.each_packet do |packet|
|
||
|
|
# Add packet to queue
|
||
|
|
@queue.push(packet)
|
||
|
|
@loggman.log_info("Packet #{packet_count += 1} added to queue.")
|
||
|
|
end
|
||
|
|
rescue StopIteration
|
||
|
|
@loggman.log_warn("Packet capture loop stopped for #{@interface}.")
|
||
|
|
rescue StandardError => e
|
||
|
|
@loggman.log_fatal("Packet capture loop stopped for #{@interface}: #{e.message}\n#{e.backtrace}", false)
|
||
|
|
sleep 1
|
||
|
|
retry
|
||
|
|
ensure
|
||
|
|
@capture.close
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def stop_capture
|
||
|
|
@loggman.log_warn("Stopping packet capture loop for #{@interface}...")
|
||
|
|
@stop_flag = true
|
||
|
|
end
|
||
|
|
end
|