/*
 * Copyright 2001-2009, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Distributed under the terms of the MIT License.
 */

#include "FileSystemInitializer.h"

#include <new>

#include <image.h>
#include <OS.h>

#include <port.h>
#include <team.h>

#include "AutoDeleter.h"

#include "FileSystem.h"
#include "RequestAllocator.h"
#include "RequestPort.h"
#include "Requests.h"
#include "SingleReplyRequestHandler.h"
#include "UserlandFSDefs.h"


// constructor
FileSystemInitializer::FileSystemInitializer(const char* name)
        :
        BReferenceable(),
        fName(name),
        fFileSystem(NULL)
{
        // Note: We don't copy the name. It's only needed in FirstTimeInit() and
        // the UserlandFS makes sure it is valid until then.
}


// destructor
FileSystemInitializer::~FileSystemInitializer()
{
        delete fFileSystem;
}


// FirstTimeInit
status_t
FileSystemInitializer::FirstTimeInit()
{
        // First check whether a server for this FS is already loaded. Look for a
        // port with the respective name.
        char portName[B_OS_NAME_LENGTH];
        snprintf(portName, sizeof(portName), "_userlandfs_%s", fName);

        port_id port = find_port(portName);
        if (port >= 0)
                return _Init(port);

        // We have to start the server ourselves. First create the port we're going
        // to use.
        port = create_port(1, portName);
        if (port < 0)
                RETURN_ERROR(port);

        // prepare the command line arguments
        char portID[16];
        snprintf(portID, sizeof(portID), "%" B_PRId32, port);

        const char* args[4] = {
                "/system/servers/userlandfs_server",
                fName,
                portID,
                NULL
        };

        // start the server
        thread_id thread = load_image_etc(3, args, NULL, B_NORMAL_PRIORITY,
                B_SYSTEM_TEAM, 0);
        if (thread < 0) {
                delete_port(port);
                RETURN_ERROR(thread);
        }

        // let it own the port and start the team
        status_t error = set_port_owner(port, thread);
        if (error == B_OK)
                resume_thread(thread);

        // and do the initialization
        if (error == B_OK)
                error = _Init(port);

        if (error != B_OK) {
                kill_team(thread);
                delete_port(port);
                RETURN_ERROR(error);
        }

        return B_OK;
}


void
FileSystemInitializer::LastReferenceReleased()
{
        // don't delete
}


status_t
FileSystemInitializer::_Init(port_id port)
{
        // allocate a buffer for the FS info
        const size_t bufferSize = 1024;
        fs_init_info* info = (fs_init_info*)malloc(bufferSize);
        MemoryDeleter _(info);

        // Read the info -- we peek only, so it won't go away and we can do the
        // initialization again, in case we get completely unloaded are reloaded
        // later.
        ssize_t bytesRead = read_port_etc(port, NULL, info, bufferSize,
            B_PEEK_PORT_MESSAGE | B_CAN_INTERRUPT, 0);
        if (bytesRead < 0)
                RETURN_ERROR(bytesRead);

        // sanity check
        if ((size_t)bytesRead < sizeof(fs_init_info)
                || info->portInfoCount < 2
                || (size_t)bytesRead < sizeof(fs_init_info)
                        + info->portInfoCount * sizeof(Port::Info)) {
                RETURN_ERROR(B_BAD_DATA);
        }

        // get the port's team
        port_info portInfo;
        status_t error = get_port_info(port, &portInfo);
        if (error != B_OK)
                RETURN_ERROR(error);

        // create and init the FileSystem
        fFileSystem = new(std::nothrow) FileSystem;
        if (!fFileSystem)
                return B_NO_MEMORY;

        error = fFileSystem->Init(fName, portInfo.team, info->portInfos,
                info->portInfoCount, info->capabilities);
        if (error != B_OK)
                RETURN_ERROR(error);

        return B_OK;
}