# osmo_gsm_tester: specifics for running an osmo-nitb
#
# Copyright (C) 2016-2017 by sysmocom - s.f.m.c. GmbH
#
# Author: Pau Espin Pedrol <pespin@sysmocom.de>
#
# 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 3 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, see <http://www.gnu.org/licenses/>.

import os

from ..core import log, process

class PcapRecorder(log.Origin):

    def __init__(self, testenv, run_dir, iface=None, filters='', netns=None):
        self.iface = iface
        if not self.iface:
            self.iface = "any"
        self.filters = filters
        super().__init__(log.C_RUN, 'pcap-recorder_%s' % self.iface, filters=self.filters)
        self.testenv = testenv
        self.run_dir = run_dir
        self.netns = netns
        self.start()

    def start(self):
        self.dbg('Recording pcap', self.run_dir, self.filters)
        dumpfile = os.path.join(os.path.abspath(self.run_dir), self.name() + ".pcap")
        popen_args = ('tcpdump', '-n',
                      '-i', self.iface,
                      '-w', dumpfile,
                      self.filters)
        if self.netns:
            self.process = process.NetNSProcess(self.name(), self.run_dir, self.netns, popen_args)
        else:
            self.process = process.Process(self.name(), self.run_dir, popen_args)
        self.testenv.remember_to_stop(self.process)
        self.process.launch()

    def running(self):
        return not self.process.terminated()

# vim: expandtab tabstop=4 shiftwidth=4