/*
 * Copyright (c) 2011 Vojtech Horky
 * Copyright (c) 2011 Jan Vesely
 * 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 drvusbhub
 * @{
 */
/** @file
 * Hub ports functions.
 */

#include <stdbool.h>
#include <errno.h>
#include <str_error.h>
#include <inttypes.h>
#include <fibril_synch.h>

#include <usb/debug.h>

#include "port.h"
#include "usbhub.h"
#include "status.h"

/** Information for fibril for device discovery. */
struct add_device_phase1 {
	usb_hub_dev_t *hub;
	usb_hub_port_t *port;
	usb_speed_t speed;
};

static int usb_hub_port_device_gone(usb_hub_port_t *port, usb_hub_dev_t *hub);
static void usb_hub_port_reset_completed(usb_hub_port_t *port,
    usb_hub_dev_t *hub, usb_port_status_t status);
static int get_port_status(usb_hub_port_t *port, usb_port_status_t *status);
static int add_device_phase1_worker_fibril(void *arg);
static int create_add_device_fibril(usb_hub_port_t *port, usb_hub_dev_t *hub,
    usb_speed_t speed);

int usb_hub_port_fini(usb_hub_port_t *port, usb_hub_dev_t *hub)
{
	assert(port);
	if (port->device_attached)
		return usb_hub_port_device_gone(port, hub);
	return EOK;
}

/**
 * Clear feature on hub port.
 *
 * @param port Port structure.
 * @param feature Feature selector.
 * @return Operation result
 */
int usb_hub_port_clear_feature(
    usb_hub_port_t *port, usb_hub_class_feature_t feature)
{
	assert(port);
	const usb_device_request_setup_packet_t clear_request = {
		.request_type = USB_HUB_REQ_TYPE_CLEAR_PORT_FEATURE,
		.request = USB_DEVREQ_CLEAR_FEATURE,
		.value = feature,
		.index = port->port_number,
		.length = 0,
	};
	return usb_pipe_control_write(port->control_pipe, &clear_request,
	    sizeof(clear_request), NULL, 0);
}

/**
 * Set feature on hub port.
 *
 * @param port Port structure.
 * @param feature Feature selector.
 * @return Operation result
 */
int usb_hub_port_set_feature(
    usb_hub_port_t *port, usb_hub_class_feature_t feature)
{
	assert(port);
	const usb_device_request_setup_packet_t clear_request = {
		.request_type = USB_HUB_REQ_TYPE_SET_PORT_FEATURE,
		.request = USB_DEVREQ_SET_FEATURE,
		.index = port->port_number,
		.value = feature,
		.length = 0,
	};
	return usb_pipe_control_write(port->control_pipe, &clear_request,
	    sizeof(clear_request), NULL, 0);
}

/**
 * Mark reset process as failed due to external reasons
 *
 * @param port Port structure
 */
void usb_hub_port_reset_fail(usb_hub_port_t *port)
{
	assert(port);
	fibril_mutex_lock(&port->mutex);
	if (port->reset_status == IN_RESET)
		port->reset_status = RESET_FAIL;
	fibril_condvar_broadcast(&port->reset_cv);
	fibril_mutex_unlock(&port->mutex);
}

/**
 * Process interrupts on given port
 *
 * Accepts connection, over current and port reset change.
 * @param port port structure
 * @param hub hub representation
 */
void usb_hub_port_process_interrupt(usb_hub_port_t *port, usb_hub_dev_t *hub)
{
	assert(port);
	assert(hub);
	usb_log_debug2("(%p-%u): Interrupt.\n", hub, port->port_number);

	usb_port_status_t status = 0;
	const int opResult = get_port_status(port, &status);
	if (opResult != EOK) {
		usb_log_error("(%p-%u): Failed to get port status: %s.\n", hub,
		    port->port_number, str_error(opResult));
		return;
	}

	/* Connection change */
	if (status & USB_HUB_PORT_C_STATUS_CONNECTION) {
		const bool connected =
		    (status & USB_HUB_PORT_STATUS_CONNECTION) != 0;
		usb_log_debug("(%p-%u): Connection change: device %s.\n", hub,
		    port->port_number, connected ? "attached" : "removed");

		/* ACK the change */
		const int opResult = usb_hub_port_clear_feature(port,
		    USB_HUB_FEATURE_C_PORT_CONNECTION);
		if (opResult != EOK) {
			usb_log_warning("(%p-%u): Failed to clear "
			    "port-change-connection flag: %s.\n", hub,
			    port->port_number, str_error(opResult));
		}

		if (connected) {
			const int opResult = create_add_device_fibril(port, hub,
			    usb_port_speed(status));
			if (opResult != EOK) {
				usb_log_error("(%p-%u): Cannot handle change on"
				   " port: %s.\n", hub, port->port_number,
				   str_error(opResult));
			}
		} else {
			/* Handle the case we were in reset */
			// FIXME: usb_hub_port_reset_fail(port);
			/* If enabled change was reported leave the removal
			 * to that handler, it shall ACK the change too. */
			if (!(status & USB_HUB_PORT_C_STATUS_ENABLED)) {
				usb_hub_port_device_gone(port, hub);
			}
		}
	}

	/* Enable change, ports are automatically disabled on errors. */
	if (status & USB_HUB_PORT_C_STATUS_ENABLED) {
		// TODO: maybe HS reset failed?
		usb_log_info("(%p-%u): Port disabled because of errors.\n", hub,
		   port->port_number);
		usb_hub_port_device_gone(port, hub);
		const int rc = usb_hub_port_clear_feature(port,
		        USB_HUB_FEATURE_C_PORT_ENABLE);
		if (rc != EOK) {
			usb_log_error("(%p-%u): Failed to clear port enable "
			    "change feature: %s.", hub, port->port_number,
			    str_error(rc));
		}

	}

	/* Suspend change */
	if (status & USB_HUB_PORT_C_STATUS_SUSPEND) {
		usb_log_error("(%p-%u): Port went to suspend state, this should"
		    " NOT happen as we do not support suspend state!", hub,
		    port->port_number);
		const int rc = usb_hub_port_clear_feature(port,
		        USB_HUB_FEATURE_C_PORT_SUSPEND);
		if (rc != EOK) {
			usb_log_error("(%p-%u): Failed to clear port suspend "
			    "change feature: %s.", hub, port->port_number,
			    str_error(rc));
		}
	}

	/* Over current */
	if (status & USB_HUB_PORT_C_STATUS_OC) {
		usb_log_debug("(%p-%u): Port OC reported!.", hub,
		    port->port_number);
		/* According to the USB specs:
		 * 11.13.5 Over-current Reporting and Recovery
		 * Hub device is responsible for putting port in power off
		 * mode. USB system software is responsible for powering port
		 * back on when the over-current condition is gone */
		const int rc = usb_hub_port_clear_feature(port,
		    USB_HUB_FEATURE_C_PORT_OVER_CURRENT);
		if (rc != EOK) {
			usb_log_error("(%p-%u): Failed to clear port OC change "
			    "feature: %s.\n", hub, port->port_number,
			    str_error(rc));
		}
		if (!(status & ~USB_HUB_PORT_STATUS_OC)) {
			const int rc = usb_hub_port_set_feature(
			    port, USB_HUB_FEATURE_PORT_POWER);
			if (rc != EOK) {
				usb_log_error("(%p-%u): Failed to set port "
				    "power after OC: %s.", hub,
				    port->port_number, str_error(rc));
			}
		}
	}

	/* Port reset, set on port reset complete. */
	if (status & USB_HUB_PORT_C_STATUS_RESET) {
		usb_hub_port_reset_completed(port, hub, status);
	}

	usb_log_debug2("(%p-%u): Port status %#08" PRIx32, hub,
	    port->port_number, status);
}

/**
 * routine called when a device on port has been removed
 *
 * If the device on port had default address, it releases default address.
 * Otherwise does not do anything, because DDF does not allow to remove device
 * from it`s device tree.
 * @param port port structure
 * @param hub hub representation
 */
int usb_hub_port_device_gone(usb_hub_port_t *port, usb_hub_dev_t *hub)
{
	assert(port);
	assert(hub);
	async_exch_t *exch = usb_device_bus_exchange_begin(hub->usb_device);
	if (!exch)
		return ENOMEM;
	const int rc = usb_device_remove(exch, port->port_number);
	usb_device_bus_exchange_end(exch);
	if (rc == EOK)
		port->device_attached = false;
	return rc;

}

/**
 * Process port reset change
 *
 * After this change port should be enabled, unless some problem occurred.
 * This functions triggers second phase of enabling new device.
 * @param port Port structure
 * @param status Port status mask
 */
void usb_hub_port_reset_completed(usb_hub_port_t *port, usb_hub_dev_t *hub,
    usb_port_status_t status)
{
	assert(port);
	fibril_mutex_lock(&port->mutex);
	const bool enabled = (status & USB_HUB_PORT_STATUS_ENABLED) != 0;
	/* Finalize device adding. */

	if (enabled) {
		port->reset_status = RESET_OK;
		usb_log_debug("(%p-%u): Port reset complete.\n", hub,
		    port->port_number);
	} else {
		port->reset_status = RESET_FAIL;
		usb_log_warning("(%p-%u): Port reset complete but port not "
		    "enabled.", hub, port->port_number);
	}
	fibril_condvar_broadcast(&port->reset_cv);
	fibril_mutex_unlock(&port->mutex);

	/* Clear the port reset change. */
	int rc = usb_hub_port_clear_feature(port, USB_HUB_FEATURE_C_PORT_RESET);
	if (rc != EOK) {
		usb_log_error("(%p-%u): Failed to clear port reset change: %s.",
		    hub, port->port_number, str_error(rc));
	}
}

/** Retrieve port status.
 *
 * @param[in] port Port structure
 * @param[out] status Where to store the port status.
 * @return Error code.
 */
static int get_port_status(usb_hub_port_t *port, usb_port_status_t *status)
{
	assert(port);
	/* USB hub specific GET_PORT_STATUS request. See USB Spec 11.16.2.6
	 * Generic GET_STATUS request cannot be used because of the difference
	 * in status data size (2B vs. 4B)*/
	const usb_device_request_setup_packet_t request = {
		.request_type = USB_HUB_REQ_TYPE_GET_PORT_STATUS,
		.request = USB_HUB_REQUEST_GET_STATUS,
		.value = 0,
		.index = uint16_host2usb(port->port_number),
		.length = sizeof(usb_port_status_t),
	};
	size_t recv_size;
	usb_port_status_t status_tmp;

	const int rc = usb_pipe_control_read(port->control_pipe,
	    &request, sizeof(usb_device_request_setup_packet_t),
	    &status_tmp, sizeof(status_tmp), &recv_size);
	if (rc != EOK) {
		return rc;
	}

	if (recv_size != sizeof (status_tmp)) {
		return ELIMIT;
	}

	if (status != NULL) {
		*status = status_tmp;
	}

	return EOK;
}

static int port_enable(usb_hub_port_t *port, usb_hub_dev_t *hub, bool enable)
{
	if (enable) {
		int rc =
		    usb_hub_port_set_feature(port, USB_HUB_FEATURE_PORT_RESET);
		if (rc != EOK) {
			usb_log_error("(%p-%u): Port reset request failed: %s.",
			    hub, port->port_number, str_error(rc));
			return rc;
		}
		/* Wait until reset completes. */
		fibril_mutex_lock(&port->mutex);
		port->reset_status = IN_RESET;
		while (port->reset_status == IN_RESET)
			fibril_condvar_wait(&port->reset_cv, &port->mutex);
		rc = port->reset_status == RESET_OK ? EOK : ESTALL;
		fibril_mutex_unlock(&port->mutex);
		return rc;
	} else {
		return usb_hub_port_clear_feature(port,
				USB_HUB_FEATURE_PORT_ENABLE);
	}
}

/** Fibril for adding a new device.
 *
 * Separate fibril is needed because the port reset completion is announced
 * via interrupt pipe and thus we cannot block here.
 *
 * @param arg Pointer to struct add_device_phase1.
 * @return 0 Always.
 */
int add_device_phase1_worker_fibril(void *arg)
{
	struct add_device_phase1 *params = arg;
	assert(params);

	int ret = EOK;
	usb_hub_dev_t *hub = params->hub;
	usb_hub_port_t *port = params->port;
	const usb_speed_t speed = params->speed;
	free(arg);

	usb_log_debug("(%p-%u): New device sequence.", hub, port->port_number);

	async_exch_t *exch = usb_device_bus_exchange_begin(hub->usb_device);
	if (!exch) {
		usb_log_error("(%p-%u): Failed to begin bus exchange.", hub,
		    port->port_number);
		ret = ENOMEM;
		goto out;
	}

	/* Reserve default address */
	while ((ret = usb_reserve_default_address(exch, speed)) == ENOENT) {
		async_usleep(1000000);
	}
	if (ret != EOK) {
		usb_log_error("(%p-%u): Failed to reserve default address: %s",
		    hub, port->port_number, str_error(ret));
		goto out;
	}

	usb_log_debug("(%p-%u): Got default address reseting port.", hub,
	    port->port_number);
	/* Reset port */
	ret = port_enable(port, hub, true);
	if (ret != EOK) {
		usb_log_error("(%p-%u): Failed to reset port.", hub,
		    port->port_number);
		if (usb_release_default_address(exch) != EOK)
			usb_log_warning("(%p-%u): Failed to release default "
			    "address.", hub, port->port_number);
		ret = EIO;
		goto out;
	}
	usb_log_debug("(%p-%u): Port reset, enumerating device", hub,
	    port->port_number);

	ret = usb_device_enumerate(exch, port->port_number);
	if (ret != EOK) {
		usb_log_error("(%p-%u): Failed to enumerate device: %s", hub,
		    port->port_number, str_error(ret));
		const int ret = port_enable(port, hub, false);
		if (ret != EOK) {
			usb_log_warning("(%p-%u)Failed to disable port (%s), "
			    "NOT releasing default address.", hub,
			    port->port_number, str_error(ret));
		} else {
			const int ret = usb_release_default_address(exch);
			if (ret != EOK)
				usb_log_warning("(%p-%u): Failed to release "
				    "default address: %s", hub,
				    port->port_number, str_error(ret));
		}
	} else {
		usb_log_debug("(%p-%u): Device enumerated", hub,
		    port->port_number);
		port->device_attached = true;
		if (usb_release_default_address(exch) != EOK)
			usb_log_warning("(%p-%u): Failed to release default "
			    "address", hub, port->port_number);
	}
out:
	usb_device_bus_exchange_end(exch);

	fibril_mutex_lock(&hub->pending_ops_mutex);
	assert(hub->pending_ops_count > 0);
	--hub->pending_ops_count;
	fibril_condvar_signal(&hub->pending_ops_cv);
	fibril_mutex_unlock(&hub->pending_ops_mutex);

	return ret;
}

/** Start device adding when connection change is detected.
 *
 * This fires a new fibril to complete the device addition.
 *
 * @param hub Hub where the change occured.
 * @param port Port index (starting at 1).
 * @param speed Speed of the device.
 * @return Error code.
 */
static int create_add_device_fibril(usb_hub_port_t *port, usb_hub_dev_t *hub,
    usb_speed_t speed)
{
	assert(hub);
	assert(port);
	struct add_device_phase1 *data
	    = malloc(sizeof(struct add_device_phase1));
	if (data == NULL) {
		return ENOMEM;
	}
	data->hub = hub;
	data->port = port;
	data->speed = speed;

	fid_t fibril = fibril_create(add_device_phase1_worker_fibril, data);
	if (fibril == 0) {
		free(data);
		return ENOMEM;
	}
	fibril_mutex_lock(&hub->pending_ops_mutex);
	++hub->pending_ops_count;
	fibril_mutex_unlock(&hub->pending_ops_mutex);
	fibril_add_ready(fibril);

	return EOK;
}

/**
 * @}
 */