/* 
 * Copyright 2005, Ingo Weinhold, bonefish@users.sf.net. All rights reserved.
 * Distributed under the terms of the MIT License.
 */

#include <map>
#include <new>

#include <string.h>

#include <Autolock.h>

#include <syscalls.h>

#include "Debug.h"
#include "MessageDeliverer.h"
#include "MessagingService.h"

using std::map;
using std::nothrow;

// sService -- the singleton instance
MessagingService *MessagingService::sService = NULL;

/*!	\class MessagingArea
	\brief Represents an area of the messaging service shared between kernel
		   and registrar.

	The main purpose of the class is to retrieve (and remove) commands from
	the area.
*/

// constructor
MessagingArea::MessagingArea()
{
}

// destructor
MessagingArea::~MessagingArea()
{
	if (fID >= 0)
		delete_area(fID);
}

// Create
status_t
MessagingArea::Create(area_id kernelAreaID, sem_id lockSem, sem_id counterSem,
	MessagingArea *&_area)
{
	// allocate the object on the heap
	MessagingArea *area = new(nothrow) MessagingArea;
	if (!area)
		return B_NO_MEMORY;

	// clone the kernel area
	area_id areaID = clone_area("messaging", (void**)&area->fHeader,
		B_ANY_ADDRESS, B_READ_AREA | B_WRITE_AREA, kernelAreaID);
	if (areaID < 0) {
		delete area;
		return areaID;
	}

	// finish the initialization of the object
	area->fID = areaID;
	area->fSize = area->fHeader->size;
	area->fLockSem = lockSem;
	area->fCounterSem = counterSem;
	area->fNextArea = NULL;

	_area = area;
	return B_OK;
}

// Lock
bool
MessagingArea::Lock()
{
	// benaphore-like locking
	if (atomic_add(&fHeader->lock_counter, 1) == 0)
		return true;

	return (acquire_sem(fLockSem) == B_OK);
}

// Unlock
void
MessagingArea::Unlock()
{
	if (atomic_add(&fHeader->lock_counter, -1) > 1)
		release_sem(fLockSem);
}

// ID
area_id
MessagingArea::ID() const
{
	return fID;
}

// Size
int32
MessagingArea::Size() const
{
	return fSize;
}

// CountCommands
int32
MessagingArea::CountCommands() const
{
	return fHeader->command_count;
}

// PopCommand
const messaging_command *
MessagingArea::PopCommand()
{
	if (fHeader->command_count == 0)
		return NULL;

	// get the command
	messaging_command *command
		= (messaging_command*)((char*)fHeader + fHeader->first_command);

	// remove it from the area
	// (as long as the area is still locked, noone will overwrite the contents)
	if (--fHeader->command_count == 0)
		fHeader->first_command = fHeader->last_command = 0;
	else
		fHeader->first_command = command->next_command;

	return command;
}

// Discard
void
MessagingArea::Discard()
{
	fHeader->size = 0;
}

// NextKernelAreaID
area_id
MessagingArea::NextKernelAreaID() const
{
	return fHeader->next_kernel_area;
}

// SetNextArea
void
MessagingArea::SetNextArea(MessagingArea *area)
{
	fNextArea = area;
}

// NextArea
MessagingArea *
MessagingArea::NextArea() const
{
	return fNextArea;
}


// #pragma mark -

// constructor
MessagingCommandHandler::MessagingCommandHandler()
{
}

// destructor
MessagingCommandHandler::~MessagingCommandHandler()
{
}


// #pragma mark -

// DefaultSendCommandHandler
class MessagingService::DefaultSendCommandHandler
	: public MessagingCommandHandler {

	virtual void HandleMessagingCommand(uint32 _command, const void *data,
		int32 dataSize)
	{
		const messaging_command_send_message *sendData
			= (const messaging_command_send_message*)data;
		const void *messageData = (uint8*)data
			+ sizeof(messaging_command_send_message)
			+ sizeof(messaging_target) * sendData->target_count;

		DefaultMessagingTargetSet set(sendData->targets,
			sendData->target_count);
		MessageDeliverer::Default()->DeliverMessage(messageData,
			sendData->message_size, set);
	}
};

// CommandHandlerMap
struct MessagingService::CommandHandlerMap
	: map<uint32, MessagingCommandHandler*> {
};


/*! \class MessagingService
	\brief Userland implementation of the kernel -> userland messaging service.

	This service provides a way for the kernel to send BMessages (usually
	notification (e.g. node monitoring) messages) to userland applications.

	The kernel could write the messages directly to the respective target ports,
	but this has the disadvantage, that a message needs to be dropped, if the
	port is full at the moment of sending. By transferring the message to the
	registrar, it is possible to use the MessageDeliverer which retries sending
	messages on full ports.

	The message transfer is implemented via areas shared between kernel
	and registrar. By default one area is used as a ring buffer. The kernel
	adds messages to it, the registrar removes them. If the area is full, the
	kernel creates a new one and adds it to the area list.

	While the service is called `messaging service' and we were speaking of
	`messages' being passed through the areas, the service is actually more
	general. In fact `commands' are passed through the areas. Currently the
	only implemented command type is to send a message, but it is very easy
	to add further command types (e.g. one for alerting the user in case of
	errors).

	The MessagingService maintains a mapping of command types to command
	handlers (MessagingCommandHandler, which perform the actual processing
	of the commands), that can be altered via
	MessagingService::SetCommandHandler().
*/

// constructor
MessagingService::MessagingService()
	: fLock("messaging service"),
	  fLockSem(-1),
	  fCounterSem(-1),
	  fFirstArea(NULL),
	  fCommandHandlers(NULL),
	  fCommandProcessor(-1),
	  fTerminating(false)
{
}

// destructor
MessagingService::~MessagingService()
{
	fTerminating = true;

	if (fLockSem >= 0)
		delete_sem(fLockSem);
	if (fCounterSem >= 0)
		delete_sem(fCounterSem);

	if (fCommandProcessor >= 0) {
		int32 result;
		wait_for_thread(fCommandProcessor, &result);
	}

	delete fCommandHandlers;

	delete fFirstArea;
}

// Init
status_t
MessagingService::Init()
{
	// create the semaphores
	fLockSem = create_sem(0, "messaging lock");
	if (fLockSem < 0)
		return fLockSem;

	fCounterSem = create_sem(0, "messaging counter");
	if (fCounterSem < 0)
		return fCounterSem;

	// create the command handler map
	fCommandHandlers = new(nothrow) CommandHandlerMap;
	if (!fCommandHandlers)
		return B_NO_MEMORY;

	// spawn the command processor
	fCommandProcessor = spawn_thread(MessagingService::_CommandProcessorEntry,
		"messaging command processor", B_DISPLAY_PRIORITY, this);
	if (fCommandProcessor < 0)
		return fCommandProcessor;

	// register with the kernel
	area_id areaID = _kern_register_messaging_service(fLockSem, fCounterSem);
	if (areaID < 0)
		return areaID;

	// create the area
	status_t error = MessagingArea::Create(areaID, fLockSem, fCounterSem,
		fFirstArea);
	if (error != B_OK) {
		_kern_unregister_messaging_service();
		return error;
	}

	// resume the command processor
	resume_thread(fCommandProcessor);

	// install the default send message command handler
	MessagingCommandHandler *handler = new(nothrow) DefaultSendCommandHandler;
	if (!handler)
		return B_NO_MEMORY;
	SetCommandHandler(MESSAGING_COMMAND_SEND_MESSAGE, handler);

	return B_OK;
}

// CreateDefault
status_t
MessagingService::CreateDefault()
{
	if (sService)
		return B_OK;

	// create the service
	MessagingService *service = new(nothrow) MessagingService;
	if (!service)
		return B_NO_MEMORY;

	// init it
	status_t error = service->Init();
	if (error != B_OK) {
		delete service;
		return error;
	}

	sService = service;
	return B_OK;
}

// DeleteDefault
void
MessagingService::DeleteDefault()
{
	if (sService) {
		delete sService;
		sService = NULL;
	}
}

// Default
MessagingService *
MessagingService::Default()
{
	return sService;
}

// SetCommandHandler
void
MessagingService::SetCommandHandler(uint32 command,
	MessagingCommandHandler *handler)
{
	BAutolock _(fLock);

	if (handler) {
		(*fCommandHandlers)[command] = handler;
	} else {
		// no handler: remove and existing entry
		CommandHandlerMap::iterator it = fCommandHandlers->find(command);
		if (it != fCommandHandlers->end())
			fCommandHandlers->erase(it);
	}
}

// _GetCommandHandler
MessagingCommandHandler *
MessagingService::_GetCommandHandler(uint32 command) const
{
	BAutolock _(fLock);

	CommandHandlerMap::iterator it = fCommandHandlers->find(command);
	return (it != fCommandHandlers->end() ? it->second : NULL);
}

// _CommandProcessorEntry
int32
MessagingService::_CommandProcessorEntry(void *data)
{
	return ((MessagingService*)data)->_CommandProcessor();
}

// _CommandProcessor
int32
MessagingService::_CommandProcessor()
{
	bool commandWaiting = false;
	while (!fTerminating) {
		// wait for the next command
		if (!commandWaiting) {
			status_t error = acquire_sem(fCounterSem);
			if (error != B_OK)
				continue;
		} else
			commandWaiting = false;

		// get it from the first area
		MessagingArea *area = fFirstArea;
		area->Lock();
		while (area->CountCommands() > 0) {
			const messaging_command *command = area->PopCommand();
			if (!command) {
				// something's seriously wrong
				ERROR("MessagingService::_CommandProcessor(): area %p (%"
					B_PRId32 ") has command count %" B_PRId32 ", but doesn't "
					"return any more commands.", area, area->ID(),
					area->CountCommands());
				break;
			}
PRINT("MessagingService::_CommandProcessor(): got command %" B_PRIu32 "\n",
command->command);

			// dispatch the command
			MessagingCommandHandler *handler
				= _GetCommandHandler(command->command);
			if (handler) {
				handler->HandleMessagingCommand(command->command, command->data,
					command->size - sizeof(messaging_command));
			} else {
				WARNING("MessagingService::_CommandProcessor(): No handler "
					"found for command %" B_PRIu32 "\n", command->command);
			}
		}

		// there is a new area we don't know yet
		if (!area->NextArea() && area->NextKernelAreaID() >= 0) {
			// create it
			MessagingArea *nextArea;
			status_t error = MessagingArea::Create(area->NextKernelAreaID(),
				fLockSem, fCounterSem, nextArea);
			if (error == B_OK) {
				area->SetNextArea(nextArea);
				commandWaiting = true;
			} else {
				// Bad, but what can we do?
				ERROR("MessagingService::_CommandProcessor(): Failed to clone "
					"kernel area %" B_PRId32 ": %s\n", area->NextKernelAreaID(),
					strerror(error));
			}

		}

		// if the current area is empty and there is a next one, we discard the
		// current one
		if (area->NextArea() && area->CountCommands() == 0) {
			fFirstArea = area->NextArea();
			area->Discard();
			area->Unlock();
			delete area;
		} else {
			area->Unlock();
		}
	}

	return 0;
}