/*
 * Copyright (c) 2009 Lukas Mejdrech
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libnet 
 * @{
 */

/** @file
 * Network interface module skeleton implementation.
 * @see netif.h
 */

#include <async.h>
#include <mem.h>
#include <fibril_synch.h>
#include <stdio.h>
#include <ipc/ipc.h>
#include <ipc/services.h>
#include <ipc/netif.h>
#include <errno.h>

#include <generic.h>
#include <net/modules.h>
#include <net/packet.h>
#include <packet_client.h>
#include <packet_remote.h>
#include <adt/measured_strings.h>
#include <net/device.h>
#include <nil_interface.h>
#include <netif_local.h>
#include <netif_interface.h>

DEVICE_MAP_IMPLEMENT(netif_device_map, netif_device_t);

/** Network interface global data. */
netif_globals_t netif_globals;

/** Probe the existence of the device.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @param[in] irq	The device interrupt number.
 * @param[in] io	The device input/output address.
 * @return		EOK on success.
 * @return		Other error codes as defined for the
 *			netif_probe_message().
 */
int
netif_probe_req_local(int netif_phone, device_id_t device_id, int irq, int io)
{
	fibril_rwlock_write_lock(&netif_globals.lock);
	int result = netif_probe_message(device_id, irq, io);
	fibril_rwlock_write_unlock(&netif_globals.lock);
	
	return result;
}

/** Send the packet queue.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @param[in] packet	The packet queue.
 * @param[in] sender	The sending module service.
 * @return		EOK on success.
 * @return		Other error codes as defined for the generic_send_msg()
 *			function.
 */
int netif_send_msg_local(int netif_phone, device_id_t device_id,
    packet_t *packet, services_t sender)
{
	fibril_rwlock_write_lock(&netif_globals.lock);
	int result = netif_send_message(device_id, packet, sender);
	fibril_rwlock_write_unlock(&netif_globals.lock);
	
	return result;
}

/** Start the device.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @return		EOK on success.
 * @return		Other error codes as defined for the find_device()
 *			function.
 * @return		Other error codes as defined for the
 *			netif_start_message() function.
 */
int netif_start_req_local(int netif_phone, device_id_t device_id)
{
	int rc;
	
	fibril_rwlock_write_lock(&netif_globals.lock);
	
	netif_device_t *device;
	rc = find_device(device_id, &device);
	if (rc != EOK) {
		fibril_rwlock_write_unlock(&netif_globals.lock);
		return rc;
	}
	
	int result = netif_start_message(device);
	if (result > NETIF_NULL) {
		int phone = device->nil_phone;
		fibril_rwlock_write_unlock(&netif_globals.lock);
		nil_device_state_msg(phone, device_id, result);
		return EOK;
	}
	
	fibril_rwlock_write_unlock(&netif_globals.lock);
	
	return result;
}

/** Stop the device.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @return		EOK on success.
 * @return		Other error codes as defined for the find_device()
 *			function.
 * @return		Other error codes as defined for the
 *			netif_stop_message() function.
 */
int netif_stop_req_local(int netif_phone, device_id_t device_id)
{
	int rc;
	
	fibril_rwlock_write_lock(&netif_globals.lock);
	
	netif_device_t *device;
	rc = find_device(device_id, &device);
	if (rc != EOK) {
		fibril_rwlock_write_unlock(&netif_globals.lock);
		return rc;
	}
	
	int result = netif_stop_message(device);
	if (result > NETIF_NULL) {
		int phone = device->nil_phone;
		fibril_rwlock_write_unlock(&netif_globals.lock);
		nil_device_state_msg(phone, device_id, result);
		return EOK;
	}
	
	fibril_rwlock_write_unlock(&netif_globals.lock);
	
	return result;
}

/** Return the device usage statistics.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @param[out] stats	The device usage statistics.
 * @return EOK on success.
 */
int netif_stats_req_local(int netif_phone, device_id_t device_id,
    device_stats_t *stats)
{
	fibril_rwlock_read_lock(&netif_globals.lock);
	int res = netif_get_device_stats(device_id, stats);
	fibril_rwlock_read_unlock(&netif_globals.lock);
	
	return res;
}

/** Return the device local hardware address.
 *
 * @param[in] netif_phone The network interface phone.
 * @param[in] device_id	The device identifier.
 * @param[out] address	The device local hardware address.
 * @param[out] data	The address data.
 * @return		EOK on success.
 * @return		EBADMEM if the address parameter is NULL.
 * @return		ENOENT if there no such device.
 * @return		Other error codes as defined for the
 *			netif_get_addr_message() function.
 */
int netif_get_addr_req_local(int netif_phone, device_id_t device_id,
    measured_string_t **address, char **data)
{
	int rc;
	
	if (!address || !data)
		return EBADMEM;
	
	fibril_rwlock_read_lock(&netif_globals.lock);
	
	measured_string_t translation;
	rc = netif_get_addr_message(device_id, &translation);
	if (rc == EOK) {
		*address = measured_string_copy(&translation);
		rc = (*address) ? EOK : ENOMEM;
	}
	
	fibril_rwlock_read_unlock(&netif_globals.lock);
	
	*data = (**address).value;
	
	return rc;
}

/** Find the device specific data.
 *
 * @param[in] device_id	The device identifier.
 * @param[out] device	The device specific data.
 * @return		EOK on success.
 * @return		ENOENT if device is not found.
 * @return		EPERM if the device is not initialized.
 */
int find_device(device_id_t device_id, netif_device_t **device)
{
	if (!device)
		return EBADMEM;
	
	*device = netif_device_map_find(&netif_globals.device_map, device_id);
	if (*device == NULL)
		return ENOENT;
	
	if ((*device)->state == NETIF_NULL)
		return EPERM;
	
	return EOK;
}

/** Clear the usage statistics.
 *
 * @param[in] stats	The usage statistics.
 */
void null_device_stats(device_stats_t *stats)
{
	bzero(stats, sizeof(device_stats_t));
}

/** Initialize the netif module.
 *
 * @param[in] client_connection The client connection functio to be registered.
 * @return		EOK on success.
 * @return		Other error codes as defined for each specific module
 *			message function.
 */
int netif_init_module(async_client_conn_t client_connection)
{
	int rc;
	
	async_set_client_connection(client_connection);
	
	netif_globals.net_phone = connect_to_service(SERVICE_NETWORKING);
	netif_device_map_initialize(&netif_globals.device_map);
	
	rc = pm_init();
	if (rc != EOK)
		return rc;
	
	fibril_rwlock_initialize(&netif_globals.lock);
	
	rc = netif_initialize();
	if (rc != EOK) {
		pm_destroy();
		return rc;
	}
	
	return EOK;
}

/** Release the given packet.
 *
 * Prepared for future optimization.
 *
 * @param[in] packet_id	The packet identifier.
 */
void netif_pq_release(packet_id_t packet_id)
{
	pq_release_remote(netif_globals.net_phone, packet_id);
}

/** Allocate new packet to handle the given content size.
 *
 * @param[in] content	The minimum content size.
 * @return		The allocated packet.
 * @return		NULL if there is an error.
 *
 */
packet_t *netif_packet_get_1(size_t content)
{
	return packet_get_1_remote(netif_globals.net_phone, content);
}

/** Register the device notification receiver, the network interface layer
 * module.
 *
 * @param[in] name	Module name.
 * @param[in] device_id	The device identifier.
 * @param[in] phone	The network interface layer module phone.
 * @return		EOK on success.
 * @return		ENOENT if there is no such device.
 * @return		ELIMIT if there is another module registered.
 */
static int register_message(const char *name, device_id_t device_id, int phone)
{
	netif_device_t *device;
	int rc;
	
	rc = find_device(device_id, &device);
	if (rc != EOK)
		return rc;
	
	if (device->nil_phone > 0)
		return ELIMIT;
	
	device->nil_phone = phone;
	printf("%s: Receiver of device %d registered (phone: %d)\n",
	    name, device->device_id, device->nil_phone);
	return EOK;
}

/** Process the netif module messages.
 *
 * @param[in] name 	Module name.
 * @param[in] callid	The message identifier.
 * @param[in] call	The message parameters.
 * @param[out] answer	The message answer parameters.
 * @param[out] answer_count The last parameter for the actual answer in the
 *			answer parameter.
 * @return		EOK on success.
 * @return		ENOTSUP if the message is not known.
 * @return		Other error codes as defined for each specific module
 *			message function.
 *
 * @see IS_NET_NETIF_MESSAGE()
 *
 */
int netif_module_message_standalone(const char *name, ipc_callid_t callid,
    ipc_call_t *call, ipc_call_t *answer, int *answer_count)
{
	size_t length;
	device_stats_t stats;
	packet_t *packet;
	measured_string_t address;
	int rc;
	
	*answer_count = 0;
	switch (IPC_GET_METHOD(*call)) {
	case IPC_M_PHONE_HUNGUP:
		return EOK;
	
	case NET_NETIF_PROBE:
		return netif_probe_req_local(0, IPC_GET_DEVICE(call),
		    NETIF_GET_IRQ(call), NETIF_GET_IO(call));
		    
	case IPC_M_CONNECT_TO_ME:
		fibril_rwlock_write_lock(&netif_globals.lock);
		rc = register_message(name, IPC_GET_DEVICE(call),
		    IPC_GET_PHONE(call));
		fibril_rwlock_write_unlock(&netif_globals.lock);
		return rc;
		
	case NET_NETIF_SEND:
		rc = packet_translate_remote(netif_globals.net_phone, &packet,
		    IPC_GET_PACKET(call));
		if (rc != EOK)
			return rc;
		return netif_send_msg_local(0, IPC_GET_DEVICE(call), packet,
		    IPC_GET_SENDER(call));
		
	case NET_NETIF_START:
		return netif_start_req_local(0, IPC_GET_DEVICE(call));
		
	case NET_NETIF_STATS:
		fibril_rwlock_read_lock(&netif_globals.lock);

		rc = async_data_read_receive(&callid, &length);
		if (rc != EOK) {
			fibril_rwlock_read_unlock(&netif_globals.lock);
			return rc;
		}
		if (length < sizeof(device_stats_t)) {
			fibril_rwlock_read_unlock(&netif_globals.lock);
			return EOVERFLOW;
		}

		rc = netif_get_device_stats(IPC_GET_DEVICE(call), &stats);
		if (rc == EOK) {
			rc = async_data_read_finalize(callid, &stats,
			    sizeof(device_stats_t));
		}

		fibril_rwlock_read_unlock(&netif_globals.lock);
		return rc;

	case NET_NETIF_STOP:
		return netif_stop_req_local(0, IPC_GET_DEVICE(call));
		
	case NET_NETIF_GET_ADDR:
		fibril_rwlock_read_lock(&netif_globals.lock);
		rc = netif_get_addr_message(IPC_GET_DEVICE(call), &address);
		if (rc == EOK)
			rc = measured_strings_reply(&address, 1);
		fibril_rwlock_read_unlock(&netif_globals.lock);
		return rc;
	}
	
	return netif_specific_message(callid, call, answer, answer_count);
}

/** Start the network interface module.
 *
 * Initialize the client connection serving function, initialize the module,
 * registers the module service and start the async manager, processing IPC
 * messages in an infinite loop.
 *
 * @param[in] client_connection The client connection processing function.
 *			The module skeleton propagates its own one.
 * @return		EOK on success.
 * @return		Other error codes as defined for each specific module
 *			message function.
 */
int netif_module_start_standalone(async_client_conn_t client_connection)
{
	int rc;
	
	rc = netif_init_module(client_connection);
	if (rc != EOK)
		return rc;
	
	async_manager();
	
	pm_destroy();
	return EOK;
}

/** @}
 */