# Copyright (C) 2021-2025 Free Software Foundation, Inc.

# 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 sys

# Avoid generating
# src/gdb/testsuite/gdb.python/__pycache__/gdb_leak_detector.cpython-<n>.pyc.
sys.dont_write_bytecode = True

import re

import gdb_leak_detector


class inferior_leak_detector(gdb_leak_detector.gdb_leak_detector):
    def __init__(self):
        super().__init__(__file__)
        self.inferior = None
        self.__handler = lambda event: setattr(self, "inferior", event.inferior)
        gdb.events.new_inferior.connect(self.__handler)

    def __del__(self):
        gdb.events.new_inferior.disconnect(self.__handler)

    def allocate(self):
        output = gdb.execute("add-inferior", False, True)
        m = re.search(r"Added inferior (\d+)", output)
        if m:
            num = int(m.group(1))
        else:
            raise RuntimeError("no match")

        gdb.execute("remove-inferiors %s" % num)

    def deallocate(self):
        self.inferior = None


inferior_leak_detector().run()