/*
 * This file is part of the libopencm3 project.
 *
 * Copyright (C) 2022 Rachel Mant <git@dragonmux.network>
 *
 * This library is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <string.h>
#include <libopencm3/usb/usbd.h>
#include <libopencm3/usb/microsoft.h>
#include "usb_private.h"

#define MICROSOFT_OS_FEATURE_REGISTRY_PROPERTY_DESCRIPTOR_SIZE_CHUNK1 8U
#define MICROSOFT_OS_FEATURE_REGISTRY_PROPERTY_DESCRIPTOR_SIZE_CHUNK2 2U

static uint16_t build_function_subset(const microsoft_os_descriptor_function_subset_header *const subset,
	uint8_t *const buf, uint16_t len)
{
	uint16_t count = MIN(len, subset->wLength);
	memcpy(buf, subset, count);
	uint16_t total = count;
	uint16_t total_length = subset->wLength;
	size_t offset = 0;

	for (uint8_t i = 0; i < subset->num_feature_descriptors; ++i) {
		const microsoft_os_feature_descriptor *const feature =
			(const microsoft_os_feature_descriptor *)(((const uint8_t *)subset->feature_descriptors) + offset);

		switch (feature->wDescriptorType) {
		case MICROSOFT_OS_FEATURE_COMPATIBLE_ID:
		case MICROSOFT_OS_FEATURE_MIN_RESUME_TIME:
		case MICROSOFT_OS_FEATURE_MODEL_ID:
		case MICROSOFT_OS_FEATURE_CCGP_DEVICE:
		case MICROSOFT_OS_FEATURE_VENDOR_REVISION:
			count = MIN(len, feature->wLength);
			memcpy(buf + total, feature, count);
			total_length += feature->wLength;
			offset += feature->wLength;
			break;
		case MICROSOFT_OS_FEATURE_REG_PROPERTY: {
			const microsoft_os_feature_registry_property_descriptor *const registry_property =
				(const microsoft_os_feature_registry_property_descriptor *)feature;
			uint16_t *descriptor_length = (uint16_t *)(buf + total);
			/* Copy in the first chunk of the descriptor */
			count = MIN(len, MICROSOFT_OS_FEATURE_REGISTRY_PROPERTY_DESCRIPTOR_SIZE_CHUNK1);
			memcpy(buf + total, registry_property, count);
			len -= count;
			total += count;
			/* Copy in the property name */
			count = MIN(len, registry_property->wPropertyNameLength);
			memcpy(buf + total, registry_property->PropertyName, count);
			len -= count;
			total += count;
			/* Copy in the second chunk of the descriptor */
			count = MIN(len, MICROSOFT_OS_FEATURE_REGISTRY_PROPERTY_DESCRIPTOR_SIZE_CHUNK2);
			memcpy(buf + total, &registry_property->wPropertyDataLength, count);
			len -= count;
			total += count;
			/* Copy in the property data */
			count = MIN(len, registry_property->wPropertyDataLength);
			memcpy(buf + total, registry_property->PropertyData, count);
			/* Compute the total length of this descriptor and update total_length + offset */
			*descriptor_length = MICROSOFT_OS_FEATURE_REGISTRY_PROPERTY_DESCRIPTOR_SIZE_BASE +
				registry_property->wPropertyNameLength + registry_property->wPropertyDataLength;
			total_length += *descriptor_length;
			offset += sizeof(microsoft_os_feature_registry_property_descriptor);
			break;
		}
		default:
			return 0;
		}
		len -= count;
		total += count;
	}

	((microsoft_os_descriptor_function_subset_header *)buf)->wTotalLength = total_length;
	return total_length;
}

static uint16_t build_config_subset(const microsoft_os_descriptor_config_subset_header *const subset,
	uint8_t *const buf, uint16_t len)
{
	uint16_t total = 0;
	uint16_t count = MIN(len, subset->wLength);
	memcpy(buf, subset, count);
	len -= count;
	total += count;
	uint16_t total_length = subset->wLength;

	for (size_t i = 0; i < subset->num_function_subset_headers; ++i) {
		const uint16_t subset_header_len = build_function_subset(&subset->function_subset_headers[i],
			buf + total, len);
		count = MIN(len, subset_header_len);
		len -= count;
		total += count;
		total_length += subset_header_len;
	}

	((microsoft_os_descriptor_config_subset_header *)buf)->wTotalLength = total_length;
	return total_length;
}

static uint16_t build_descriptor_set(const microsoft_os_descriptor_set_header *const set,
	uint8_t *const buf, uint16_t len)
{
	uint16_t count = MIN(len, set->wLength);
	memcpy(buf, set, count);
	len -= count;
	uint16_t total = count;
	uint16_t total_length = set->wLength;

	for (size_t i = 0; i < set->num_config_subset_headers; ++i) {
		const uint16_t subset_header_len = build_config_subset(&set->config_subset_headers[i], buf + total, len);
		count = MIN(len, subset_header_len);
		len -= count;
		total += count;
		total_length += subset_header_len;
	}

	((microsoft_os_descriptor_set_header *)buf)->wTotalLength = total_length;
	return total;
}

static enum usbd_request_return_codes microsoft_os_get_descriptor_set(usbd_device *const usbd_dev,
	struct usb_setup_data *const req, uint8_t **const buf, uint16_t *const len)
{
	if (req->wValue != 0)
		return USBD_REQ_NOTSUPP;

	const microsoft_os_descriptor_set_header *const sets = usbd_dev->microsoft_os_descriptor_sets;
	for (size_t i = 0; i < usbd_dev->num_microsoft_os_descriptor_sets; ++i) {
		if (sets[i].vendor_code == req->bRequest) {
			*buf = usbd_dev->ctrl_buf;
			*len = build_descriptor_set(&sets[i], *buf, *len);
			return USBD_REQ_HANDLED;
		}
	}
	return USBD_REQ_NOTSUPP;
}

static enum usbd_request_return_codes microsoft_os_control_request(usbd_device *const usbd_dev,
	struct usb_setup_data *const req, uint8_t **const buf, uint16_t *const len)
{
	switch (req->wIndex) {
	case MICROSOFT_GET_DESCRIPTOR_SET:
		return microsoft_os_get_descriptor_set(usbd_dev, req, buf, len);
	case MICROSOFT_SET_ALTERNATE_ENUM:
		return USBD_REQ_NOTSUPP;
	}
	return USBD_REQ_NEXT_CALLBACK;
}

void microsoft_os_register_descriptor_sets(usbd_device *const dev,
	const microsoft_os_descriptor_set_header *const sets, const uint8_t num_sets)
{
	dev->microsoft_os_req_callback = microsoft_os_control_request;
	dev->microsoft_os_descriptor_sets = sets;
	dev->num_microsoft_os_descriptor_sets = num_sets;
}