/*
 *	Copyright 2009 Jasmin Fazlic All rights reserved.
 */
/*
 *	Cleaned and merged by iNDi
 */

#include "libsaio.h"
#include "boot.h"
#include "bootstruct.h"
#include "pci.h"
#include "pci_root.h"
#include "device_inject.h"
#include "convert.h"

#ifndef DEBUG_INJECT
#define DEBUG_INJECT 0
#endif

#if DEBUG_INJECT
#define DBG(x...)	printf(x)
#else
#define DBG(x...)
#endif

uint32_t devices_number		= 1;
uint32_t onboard_number		= 1; // TODO: increment the onboard-x(+1) value if the duallink option is true for nVidia and ATi (gfx and HDAU)
uint32_t builtin_set		= 0;
DevPropString *string		= NULL;
uint8_t *stringdata		= NULL;
uint32_t stringlength		= 0;

char *efi_inject_get_devprop_string(uint32_t *len)
{
	if(string)
	{
		*len = string->length;
		return devprop_generate_string(string);
	}
	verbose("efi_inject_get_devprop_string NULL trying stringdata\n");
	return NULL;
}

void setupDeviceProperties(Node *node)
{
  const char *val;
  uint8_t *binStr;
  int cnt, cnt2;

  static char DEVICE_PROPERTIES_PROP[] = "device-properties";

  /* Generate devprop string.
   */
  uint32_t strlength;
  char *string = efi_inject_get_devprop_string(&strlength);

  /* Use the static "device-properties" boot config key contents if available,
   * otheriwse use the generated one.
   */  
	if (!getValueForKey(kDeviceProperties, &val, &cnt, &bootInfo->chameleonConfig) && string)
	{
		val = (const char*)string;
		cnt = strlength * 2;
	}

	if (cnt > 1)
	{
		binStr = convertHexStr2Binary(val, &cnt2);
		if (cnt2 > 0)
		{
			DT__AddProperty(node, DEVICE_PROPERTIES_PROP, cnt2, binStr);
		}
	}
}

DevPropString *devprop_create_string(void)
{
	string = (struct DevPropString*)malloc(sizeof(struct DevPropString));
	
	if(string == NULL)
	{
		return NULL;
	}
	
	memset(string, 0, sizeof(struct DevPropString));
	string->length = 12;
	string->WHAT2 = 0x01000000;
	return string;
}

DevPropDevice *devprop_add_device(DevPropString *string, char *path)
{
	DevPropDevice	*device;
	const char		pciroot_string[] = "PciRoot(0x";
	const char		pci_device_string[] = "Pci(0x";

	if (string == NULL || path == NULL) {
		return NULL;
	}
	device = malloc(sizeof(DevPropDevice));
	if (!device)
	{
		return NULL;
	}
	if (strncmp(path, pciroot_string, strlen(pciroot_string)))
	{
		free(device);
		printf("ERROR parsing device path\n");
		return NULL;
	}

	memset(device, 0, sizeof(DevPropDevice));
	device->acpi_dev_path._UID = getPciRootUID();

	int numpaths = 0;
	int		x, curr = 0;
	char	buff[] = "00";

	for (x = 0; x < strlen(path); x++)
	{
		if (!strncmp(&path[x], pci_device_string, strlen(pci_device_string)))
		{
			x+=strlen(pci_device_string);
			curr=x;
			while(path[++x] != ',');
			if(x-curr == 2)
			{
				sprintf(buff, "%c%c", path[curr], path[curr+1]);
			}
			else if(x-curr == 1)
			{
				sprintf(buff, "%c", path[curr]);
			}
			else
			{
				printf("ERROR parsing device path\n");
				numpaths = 0;
				break;
			}
			device->pci_dev_path[numpaths].device =	ascii_hex_to_int(buff);

			x += 3; // 0x
			curr = x;
			while(path[++x] != ')');
			if(x-curr == 2)
			{
				sprintf(buff, "%c%c", path[curr], path[curr+1]);
			}
			else if(x-curr == 1)
			{
				sprintf(buff, "%c", path[curr]);
			}
			else
			{
				printf("ERROR parsing device path\n");
				numpaths = 0;
				break;
			}
			device->pci_dev_path[numpaths].function = ascii_hex_to_int(buff); // TODO: find dev from char *path

			numpaths++;
		}
	}
	
	if(!numpaths)
	{
		free(device);
		return NULL;
	}
	
	device->numentries = 0x00;
	
	device->acpi_dev_path.length = 0x0c;
	device->acpi_dev_path.type = 0x02;
	device->acpi_dev_path.subtype = 0x01;
	device->acpi_dev_path._HID = 0xd041030a;
	
	device->num_pci_devpaths = numpaths;
	device->length = 24 + (6*numpaths);
	
	int		i;
	
	for(i = 0; i < numpaths; i++)
	{
		device->pci_dev_path[i].length = 0x06;
		device->pci_dev_path[i].type = 0x01;
		device->pci_dev_path[i].subtype = 0x01;
	}
	
	device->path_end.length = 0x04;
	device->path_end.type = 0x7f;
	device->path_end.subtype = 0xff;
	
	device->string = string;
	device->data = NULL;
	string->length += device->length;
	
	if(!string->entries)
	{
		if((string->entries = (DevPropDevice**)malloc(sizeof(device)*DEV_PROP_DEVICE_MAX_ENTRIES))== NULL)
		{
			printf("ERROR parsing device path 2\n");

			free(device);
			return 0;
		}
	}
	
	string->entries[string->numentries++] = (DevPropDevice*)malloc(sizeof(device));
	string->entries[string->numentries-1] = device;

	return device;
}

int devprop_add_value(DevPropDevice *device, char *nm, uint8_t *vl, uint32_t len)
{

	if(!nm || !vl || !len)
		return 0;

	uint32_t length = ((strlen(nm) * 2) + len + (2 * sizeof(uint32_t)) + 2);
	uint8_t *data = (uint8_t*)malloc(length);
	{
		if(!data)
		{
			return 0;
		}

		memset(data, 0, length);
		uint32_t off= 0;
		data[off+1] = ((strlen(nm) * 2) + 6) >> 8;
		data[off] =   ((strlen(nm) * 2) + 6) & 0x00FF;
		
		off += 4;
		uint32_t i=0, l = strlen(nm);
		for(i = 0 ; i < l ; i++, off += 2)
		{
			data[off] = *nm++;
		}

		off += 2;
		l = len;
		uint32_t *datalength = (uint32_t*)&data[off];
		*datalength = (uint32_t)(l + 4);
		off += 4;
		for(i = 0 ; i < l ; i++, off++)
		{
			data[off] = *vl++;
		}
	}
	
	uint32_t offset = device->length - (24 + (6 * device->num_pci_devpaths));
	
	uint8_t *newdata = (uint8_t*)malloc((length + offset));
	if(!newdata)
	{
		return 0;
	}
	if(device->data)
	{
		if(offset > 1)
		{
			memcpy(newdata, device->data, offset);
		}
	}

	memcpy(newdata + offset, data, length);
	
	device->length += length;
	device->string->length += length;
	device->numentries++;

	if(!device->data)
	{
		device->data = (uint8_t*)malloc(sizeof(uint8_t));
	}
	else
	{
		free(device->data);
	}

	free(data);
	device->data = newdata;

	return 1;
}

char *devprop_generate_string(DevPropString *string)
{
	uint32_t len = string->length * 2;
	int i = 0;
	uint32_t x = 0;

	char *buffer = (char*)malloc(len +1);
	char *ptr = buffer;

	if(!buffer)
	{
		return NULL;
	}

	sprintf(buffer, "%08x%08x%04x%04x", dp_swap32(string->length), string->WHAT2,
			dp_swap16(string->numentries), string->WHAT3);
	buffer += 24;

	while(i < string->numentries)
	{
		sprintf(buffer, "%08x%04x%04x", dp_swap32(string->entries[i]->length),
				dp_swap16(string->entries[i]->numentries), string->entries[i]->WHAT2);
		
		buffer += 16;
		sprintf(buffer, "%02x%02x%04x%08x%08x", string->entries[i]->acpi_dev_path.type,
				string->entries[i]->acpi_dev_path.subtype,
				dp_swap16(string->entries[i]->acpi_dev_path.length),
				string->entries[i]->acpi_dev_path._HID,
				dp_swap32(string->entries[i]->acpi_dev_path._UID));

		buffer += 24;
		for(x = 0;x < string->entries[i]->num_pci_devpaths; x++)
		{
			sprintf(buffer, "%02x%02x%04x%02x%02x", string->entries[i]->pci_dev_path[x].type,
					string->entries[i]->pci_dev_path[x].subtype,
					dp_swap16(string->entries[i]->pci_dev_path[x].length),
					string->entries[i]->pci_dev_path[x].function,
					string->entries[i]->pci_dev_path[x].device);
			buffer += 12;
		}
		
		sprintf(buffer, "%02x%02x%04x", string->entries[i]->path_end.type,
				string->entries[i]->path_end.subtype,
				dp_swap16(string->entries[i]->path_end.length));
		
		buffer += 8;
		uint8_t *dataptr = string->entries[i]->data;
		for(x = 0; (uint32_t)x < (string->entries[i]->length) - (24 + (6 * string->entries[i]->num_pci_devpaths)) ; x++)
		{
			sprintf(buffer, "%02x", *dataptr++);
			buffer += 2;
		}
		i++;
	}
	return ptr;
}

void devprop_free_string(DevPropString *string)
{

	if(!string)
	{
		return;
	}

	int i;
	for(i = 0; i < string->numentries; i++)
	{
		if(string->entries[i])
		{
			if(string->entries[i]->data)
			{
				free(string->entries[i]->data);
				string->entries[i]->data = NULL;
			}
			free(string->entries[i]);
			string->entries[i] = NULL;
		}
	}

	free(string);
	string = NULL;
}

/* ======================================================= */

/* Converters */

/*static int hexdigit(int c)
{
	unsigned  d;
    
	d = (unsigned)(c - '0');
	if (d < 10) return d;

	d = (unsigned)(c - 'a');
	if (d < 6) return d+10;

	d = (unsigned)(c - 'A');
	if (d < 6) return d+10;
	printf("[ERROR] hexdigit input error\n");
	return -1;
}*/

/*********************************************************************
 * decodes a sequence of 'len' hexadecimal chars from 'hex' into     *
 * an integer. returns -1 in case of error (i.e. badly formed chars) *
 *********************************************************************/
static const long hextable[] = {
 	[0 ... 255] = -1, // bit aligned access into this table is considerably
 	['0'] = 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, // faster for most modern processors,
 	['A'] = 10, 11, 12, 13, 14, 15,       // for the space conscious, reduce to
 	['a'] = 10, 11, 12, 13, 14, 15        // signed char.
};

int hex2dec(unsigned const char *hex) { //long... unsigned const char
	int ret = 0;
	while (*hex && ret >= 0) {
	ret = (ret << 4) | hextable[*hex++];
	}
	return ret;
}

/************************************************************************
 * Encodes an integer 'val' into 'len' hexadecimal charaters into 'hex' *
 ************************************************************************/
//char int2hex(int val)


/***********************************************************
 * bin2hex creates hexadecimal presentation of binary data *
 ***********************************************************/
char *bin2hex(const uint8_t* bin, char* hex, int len)
{
	int i;
	const char *hexc = "0123456789abcdef";
	for (i = 0; i < len; i++) {
		hex[i * 2] = hexc[(bin[i] >> 4) & 0xf];
		hex[i * 2 + 1] = hexc[bin[i] &0xf];
	}
	hex[i * 2] = '\0';
	return hex;
}
/* */

/*******************************************************************
 * Decodes a sequence of 'len' hexadecimal chars from 'hex' into   *
 * a binary. returns -1 in case of error (i.e. badly formed chars) *
 *******************************************************************/
int hex2bin(const char *hex, uint8_t *bin, int len)
{
	char	*p;
	int	i;
	char	buf[3];

	if (hex == NULL || bin == NULL || len <= 0 || strlen(hex) != len * 2)
	{
		printf("[ERROR] bin2hex input error\n");
		return -1;
	}

	buf[2] = '\0';
	p = (char *) hex;

	for (i = 0; i < len; i++)
	{
		if (p[0] == '\0' || p[1] == '\0' || !isxdigit(p[0]) || !isxdigit(p[1]))
		{
			printf("[ERROR] bin2hex '%s' syntax error\n", hex);
			return -2;
		}
		buf[0] = *p++;
		buf[1] = *p++;
		bin[i] = (unsigned char) strtoul(buf, NULL, 16);
	}
	return 0;
}

/* ======================================================= */

/* a fine place for this code */
int devprop_add_network_template(DevPropDevice *device, uint16_t vendor_id)
{
	uint8_t builtin = 0x00;
	if(device)
	{

		if((vendor_id != 0x168c) && (builtin_set == 0))
		{
			builtin_set = 1;
			builtin = 0x01;
		}
		if(!devprop_add_value(device, "device_type", (uint8_t*)"ethernet", 8))
		{
			return 0;
		}
		if(!devprop_add_value(device, "built-in", (uint8_t*)&builtin, 1))
		{
			return 0;
		}
		devices_number++;
		return 1;
	}
	else
	{
		return 0;
	}
}

void set_eth_builtin(pci_dt_t *eth_dev)
{
	char *devicepath = get_pci_dev_path(eth_dev);
	DevPropDevice *device = (DevPropDevice*)malloc(sizeof(DevPropDevice));

	verbose("LAN Controller [%04x:%04x] :: %s\n", eth_dev->vendor_id, eth_dev->device_id, devicepath);

	if (!string)
	{
		string = devprop_create_string();
		if (!string)
		{
			return;
		}
	}

	device = devprop_add_device(string, devicepath);
	if(device)
	{
		verbose("Setting up lan keys\n");
		devprop_add_network_template(device, eth_dev->vendor_id);
		stringdata = (uint8_t*)malloc(sizeof(uint8_t) * string->length);
		if(stringdata)
		{
			memcpy(stringdata, (uint8_t*)devprop_generate_string(string), string->length);
			stringlength = string->length;
		}
	}
}