"""
FortiMonitor vmstat Countermeasure

Copyright 2023 Fortinet, Inc. All Rights Reserved.
fm-ops@fortinet.com

"""

from CountermeasurePlugin import CountermeasurePlugin


class VmstatCountermeasure(CountermeasurePlugin):
    name = "vmstat"
    textkey = "info.vmstat"
    description = "Gather vmstat output"
    count = 10
    author = "support@panopta.com"

    def run(self):
        """
        Execute the countermeasure action
        """

        return_code, output = self.execute("vmstat 1 %s" % self.count)
        self.save_text_output(output)
        self.save_return_code(return_code)