/*
 *  Bus Glue for Realtek rtl819x built-in EHCI controller.
 *
 *  Copyright (C) 2008-2010 Gabor Juhos <juhosg@openwrt.org>
 *  Copyright (C) 2008 Imre Kaloz <kaloz@openwrt.org>
 *
 *  Parts of this file are based on Realtek' 2.6.30 BSP
 *	Copyright (C) 2013 Realtek Semiconductor, Corp.
 *
 *  This program is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License version 2 as published
 *  by the Free Software Foundation.
 */

#include <linux/platform_device.h>
#include <linux/delay.h>

#if defined(CONFIG_RTL8685) || defined(CONFIG_RTL8685S)
#define USB_PATCH_00 1
#endif 

#ifdef USB_PATCH_00 
#include <rtl_usb_phy.h>
enum {
	EVT_UNKNOWN,
	EVT_HS_CONNECT,
	EVT_FS_CONNECT,
	EVT_DISCONNECT,
	EVT_DISCONNECT_ERR,
};

static struct timer_list T0, T1;
static int patch00_state = 0;
static u8 orig_phy[3];
#define PATCH_DEBUG_FLAGS 0
#define PATCH_DEBUG(fmt, args...) if (PATCH_DEBUG_FLAGS) printk(fmt,##args)
	
/* save original phy and apply new */
static void usb_phy_save(void) {
	#ifdef CONFIG_RTL8685
	orig_phy[0]	= ehci_phy_read(0, 0xE0);
	orig_phy[1]	= ehci_phy_read(0, 0xE2);
	orig_phy[2]	= ehci_phy_read(0, 0xE6);
	ehci_phy_write(0, 0xE0, 0x19);
	ehci_phy_write(0, 0xE2, 0x9a);
	ehci_phy_write(0, 0xE6, 0x91);
	#elif defined(CONFIG_RTL8685S)
	orig_phy[0]	= ehci_phy_read(0, 0xE0);	
	orig_phy[2]	= ehci_phy_read(0, 0xE6);
	ehci_phy_write(0, 0xE0, 0x1c);
	ehci_phy_write(0, 0xE6, 0x91);
	#endif 
	
}
static void usb_phy_restore(void) {
	#ifdef CONFIG_RTL8685
	ehci_phy_write(0, 0xE0, orig_phy[0]);
	ehci_phy_write(0, 0xE2, orig_phy[1]);
	ehci_phy_write(0, 0xE6, orig_phy[2]);
	#elif defined(CONFIG_RTL8685S)
	ehci_phy_write(0, 0xE0, orig_phy[0]);	
	ehci_phy_write(0, 0xE6, orig_phy[2]);
	#endif
}

/* Reset State */
static void T0_func(unsigned long data) {
	PATCH_DEBUG("T0 time-out");
	patch00_state = 0;
}

/* Suspend and re-start */ 
static void T1_func(unsigned long data) {
	struct usb_hcd *hcd = (struct usb_hcd *)data;
	struct ehci_hcd *ehci = hcd_to_ehci(hcd);				
	//printk("%s: %p\n",__func__,&ehci->regs->port_status[0]);	
	ehci_writel(ehci, 0x1085, &ehci->regs->port_status[0]);
	if (3==patch00_state)
		patch00_state = 4;
}

void rtk_usb_patch00(struct usb_device *hdev, int port, u16 portstatus, u16 portchange) {
	int event, old_state;
	
	if (hdev->parent!=NULL)
		return;
	PATCH_DEBUG("rtk_usb_patch00: %p port%d %x %x\n", hdev->parent,port,portstatus,portchange);
	if (portchange == USB_PORT_STAT_CONNECTION) {		
		switch (portstatus) {
		case (USB_PORT_STAT_HIGH_SPEED|USB_PORT_STAT_CONNECTION|USB_PORT_STAT_POWER):
			event = EVT_HS_CONNECT;
			break;
		case USB_PORT_STAT_CONNECTION|USB_PORT_STAT_POWER:
			event = EVT_FS_CONNECT;
			break;
		case USB_PORT_STAT_POWER:
			event = EVT_DISCONNECT;
			break;
		default:
			event = EVT_UNKNOWN;
			break;
		}
	} else if (portchange == (USB_PORT_STAT_CONNECTION|USB_PORT_STAT_ENABLE)) {
		if (portstatus==USB_PORT_STAT_POWER)
			event = EVT_DISCONNECT_ERR;
	}	
	old_state = patch00_state;
	switch (patch00_state) {

	case 1:
		if (EVT_DISCONNECT == event)
			patch00_state = 2;
		break;
	case 2:
		if (EVT_FS_CONNECT == event) {
			if (timer_pending(&T0))
				del_timer(&T0);
			
			usb_phy_save();
						
			patch00_state = 3;
			mod_timer(&T1, jiffies+msecs_to_jiffies(1500));
		}
		break;	
	case 3:
	case 4:
		if (EVT_DISCONNECT == event) {
			if (timer_pending(&T1))
				del_timer(&T1);
			
			usb_phy_restore();
			
			patch00_state = 0;
		} 
	}	
	PATCH_DEBUG("State change %d->%d\n",old_state,patch00_state);
}


static void rtk_ehci_relinquish_port(struct usb_hcd *hcd, int portnum) {
	if (patch00_state==0) {
		patch00_state = 1;
		mod_timer(&T0, jiffies+msecs_to_jiffies(1500));
	}
		
	ehci_relinquish_port(hcd, portnum);
}

#endif

extern int usb_disabled(void);

/**
 * usb_hcd_rtl819x_probe - initialize FSL-based HCDs
 * @drvier: Driver to be used for this HCD
 * @pdev: USB Host Controller being probed
 * Context: !in_interrupt()                                                                                         
 *
 * Allocates basic resources for this USB host controller.
 *
 */
static int ehci_rtl86xx_probe(const struct hc_driver *driver,
			     struct usb_hcd **hcd_out,
			     struct platform_device *pdev)
{
	struct usb_hcd *hcd;
	struct resource *res;
	int irq;
	int ret;

    pr_debug("initializing RTL-SOC USB Controller\n"); 

	res = platform_get_resource(pdev, IORESOURCE_IRQ, 0);
	if (!res) {
		dev_dbg(&pdev->dev, "no IRQ specified for %s\n",
			dev_name(&pdev->dev));
		return -ENODEV;
	}
	irq = res->start;

	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	if (!res) {
		dev_dbg(&pdev->dev, "no base address specified for %s\n",
			dev_name(&pdev->dev));
		return -ENODEV;
	}

	hcd = usb_create_hcd(driver, &pdev->dev, dev_name(&pdev->dev));
	if (!hcd) 
		return -ENOMEM;

	hcd->rsrc_start	= res->start;
	hcd->rsrc_len	= res->end - res->start + 1;

	if (!request_mem_region(hcd->rsrc_start, hcd->rsrc_len, hcd_name)) {
		dev_dbg(&pdev->dev, "controller already in use\n");
		ret = -EBUSY;
		goto err_put_hcd;
	}

	hcd->regs = ioremap(hcd->rsrc_start, hcd->rsrc_len);
	if (!hcd->regs) {
		dev_dbg(&pdev->dev, "error mapping memory\n");
		ret = -EFAULT;
		goto err_release_region;
	}

	ret = usb_add_hcd(hcd, irq, IRQF_DISABLED | IRQF_SHARED);
	if (ret) 
		goto err_iounmap;

	#ifdef USB_PATCH_00
	setup_timer(&T0, T0_func, (unsigned long)hcd);	
	setup_timer(&T1, T1_func, (unsigned long)hcd);	
	#endif

	return 0;

 err_iounmap:
	iounmap(hcd->regs);

 err_release_region:
	release_mem_region(hcd->rsrc_start, hcd->rsrc_len);
 err_put_hcd:
	usb_put_hcd(hcd);
	return ret;
}

/**
 * usb_hcd_rtl86xx_remove - shutdown processing for FSL-based HCDs
 * @dev: USB Host Controller being removed
 * Context: !in_interrupt()
 *      
 * Reverses the effect of usb_hcd_rtl8652_probe().
 *
 */ 
static void usb_hcd_rtl86xx_remove(struct usb_hcd *hcd,
			       struct platform_device *pdev)
{
	usb_remove_hcd(hcd);
	iounmap(hcd->regs);
	release_mem_region(hcd->rsrc_start, hcd->rsrc_len);
	usb_put_hcd(hcd);
}

#if 0 
static int synopsys_usb_setup(struct ehci_hcd *ehci)
{
#if 1  //tony: patch for synopsys 
    printk("read synopsys=%x\n",readl (&ehci->regs->command+0x84));
    writel(0x01fd0020,&ehci->regs->command+0x84);
    printk("read synopsys2=%x\n",readl (&ehci->regs->command+0x84));
#endif        
    return 0;
}  
#endif

static int ehci_rtl86xx_setup(struct usb_hcd *hcd)
{
    struct ehci_hcd *ehci = hcd_to_ehci(hcd);
    int ret;

	ehci->big_endian_desc = 1;
	ehci->big_endian_mmio = 1;
    ehci->caps = hcd->regs;
    ehci->regs = hcd->regs +
            HC_LENGTH(ehci,ehci_readl(ehci, &ehci->caps->hc_capbase));
    ehci->hcs_params = ehci_readl(ehci, &ehci->caps->hcs_params);

    ehci->sbrn = 0x20;                                                                                 
    //ehci->has_synopsys_hc_bug = 1;

    ehci_reset(ehci);

    ret = ehci_init(hcd);
    if (ret)
        return ret;

    //ehci_port_power(ehci, 0); 

    return 0;
}

static const struct hc_driver ehci_rtl819x_hc_driver = {
	.description		= hcd_name,
	.product_desc		= "Realtek rtl819x On-Chip EHCI Host Controller",
	.hcd_priv_size		= sizeof(struct ehci_hcd),

	.irq				= ehci_irq,
	.flags				= HCD_MEMORY | HCD_USB2 | HCD_BH,
	//.flags				= HCD_USB2,

	.reset				= ehci_rtl86xx_setup,
	.start				= ehci_run,
	.stop				= ehci_stop,
	.shutdown			= ehci_shutdown,

	.urb_enqueue		= ehci_urb_enqueue,
	.urb_dequeue		= ehci_urb_dequeue,
	.endpoint_disable	= ehci_endpoint_disable,
	.endpoint_reset		= ehci_endpoint_reset,

	.get_frame_number	= ehci_get_frame,

	.hub_status_data	= ehci_hub_status_data,
	.hub_control		= ehci_hub_control,
#ifdef CONFIG_PM
	.bus_suspend		= ehci_bus_suspend,
	.bus_resume			= ehci_bus_resume,
#endif
#ifdef USB_PATCH_00
	.relinquish_port 	= rtk_ehci_relinquish_port,
#else
	.relinquish_port	= ehci_relinquish_port,
#endif	
	.port_handed_over	= ehci_port_handed_over,

	.clear_tt_buffer_complete = ehci_clear_tt_buffer_complete,
};

static int ehci_rtl86xx_driver_probe(struct platform_device *pdev)
{
	struct usb_hcd *hcd = NULL;
	int ret;

	printk("ehci_rtl86xx_driver_probe\n");
	if (usb_disabled())
		return -ENODEV;
	printk("2 ehci_rtl86xx_driver_probe\n");

	ret = ehci_rtl86xx_probe(&ehci_rtl819x_hc_driver, &hcd, pdev);
	//ret = usb_hcd_rtl8652_probe(&ehci_rtl819x_hc_driver, pdev); 

	return ret;
}

static int ehci_rtl86xx_driver_remove(struct platform_device *pdev)
{
	struct usb_hcd *hcd = platform_get_drvdata(pdev);

    usb_hcd_rtl86xx_remove(hcd, pdev);
	return 0;
}

MODULE_ALIAS("platform:rtl86xx-ehci");

static struct platform_driver ehci_rtl86xx_driver = {
	.probe		= ehci_rtl86xx_driver_probe,
	.remove		= ehci_rtl86xx_driver_remove,
	.driver = {
		.name	= "rtl86xx-ehci",
	}
};