/*
 * Copyright 2009, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Copyright 2014-2017, Rene Gollent, rene@gollent.com.
 * Distributed under the terms of the MIT License.
 */

#include "DwarfManager.h"

#include <new>

#include <AutoDeleter.h>
#include <AutoLocker.h>

#include "DwarfFile.h"
#include "DwarfFileLoadingState.h"


DwarfManager::DwarfManager(uint8 addressSize, bool isBigEndian)
        :
        fAddressSize(addressSize), fIsBigEndian(isBigEndian),
        fLock("dwarf manager")
{
}


DwarfManager::~DwarfManager()
{
        while (DwarfFile* file = fFiles.RemoveHead())
                file->ReleaseReference();
}


status_t
DwarfManager::Init()
{
        return fLock.InitCheck();
}


status_t
DwarfManager::LoadFile(const char* fileName, DwarfFileLoadingState& _state)
{
        AutoLocker<DwarfManager> locker(this);

        DwarfFile* file = _state.dwarfFile;
        BReference<DwarfFile> fileReference;
        if (file == NULL) {
                file = new(std::nothrow) DwarfFile;
                if (file == NULL)
                        return B_NO_MEMORY;
                fileReference.SetTo(file, true);
                _state.dwarfFile = file;
        } else
                fileReference.SetTo(file);

        status_t error;
        if (_state.externalInfoFileName.IsEmpty()) {
                error = file->StartLoading(fileName, _state.externalInfoFileName);
                if (error != B_OK) {
                        // only preserve state in the failure case if an external
                        // debug information reference was found, but the corresponding
                        // file could not be located on disk.
                        _state.state = _state.externalInfoFileName.IsEmpty()
                                ? DWARF_FILE_LOADING_STATE_FAILED
                                : DWARF_FILE_LOADING_STATE_USER_INPUT_NEEDED;

                        return error;
                }
        }

        error = file->Load(fAddressSize, fIsBigEndian, _state.locatedExternalInfoPath);
        if (error != B_OK) {
                _state.state = DWARF_FILE_LOADING_STATE_FAILED;
                return error;
        }

        fFiles.Add(file);

        fileReference.Detach();
                // keep a reference for ourselves in the list.

        _state.state = DWARF_FILE_LOADING_STATE_SUCCEEDED;

        return B_OK;
}


status_t
DwarfManager::FinishLoading()
{
        AutoLocker<DwarfManager> locker(this);

        for (FileList::Iterator it = fFiles.GetIterator();
                        DwarfFile* file = it.Next();) {
                status_t error = file->FinishLoading(fAddressSize, fIsBigEndian);
                if (error != B_OK)
                        return error;
        }

        return B_OK;
}