#include <linux/module.h>
#include <linux/version.h>
#include <linux/pci.h>
#include <linux/netdevice.h>
#include "r8101.h"
#include "rtltool.h"

void mdio_write(void __iomem *ioaddr, u32 RegAddr, u32 value);
u32 mdio_read(void __iomem *ioaddr, u32 RegAddr);
void rtl8101_ephy_write(void __iomem *ioaddr, u32 RegAddr, u32 value);
u16 rtl8101_ephy_read(void __iomem *ioaddr, u32 RegAddr);

int rtltool_ioctl(struct rtl8101_private *tp, struct ifreq *ifr)
{
	struct rtltool_cmd	my_cmd;
	unsigned long flags;
	int	ret;

	if (!capable(CAP_NET_ADMIN))
	{
		return -EPERM;
	}

	if (copy_from_user(&my_cmd, ifr->ifr_data, sizeof(struct rtltool_cmd)))
	{
		return -EFAULT;
	}

	ret = 0;
	switch(my_cmd.cmd)
	{
		case RTLTOOL_READ_MAC:
			if(my_cmd.len==1)
			{
				my_cmd.data = readb(tp->mmio_addr+my_cmd.offset);
			}
			else if(my_cmd.len==2)
			{
				my_cmd.data = readw(tp->mmio_addr+(my_cmd.offset&~1));
			}
			else if(my_cmd.len==4)
			{
				my_cmd.data = readl(tp->mmio_addr+(my_cmd.offset&~3));
			}
			else
			{
				ret = -EOPNOTSUPP;
				break;
			}

			if (copy_to_user(ifr->ifr_data, &my_cmd, sizeof(struct rtltool_cmd)))
			{
				ret = -EFAULT;
				break;
			}
			break;

		case RTLTOOL_WRITE_MAC:
			if(my_cmd.len==1)
			{
				writeb(my_cmd.data, tp->mmio_addr+my_cmd.offset);
			}
			else if(my_cmd.len==2)
			{
				writew(my_cmd.data, tp->mmio_addr+(my_cmd.offset&~1));
			}
			else if(my_cmd.len==4)
			{
				writel(my_cmd.data, tp->mmio_addr+(my_cmd.offset&~3));
			}
			else
			{
				ret = -EOPNOTSUPP;
				break;
			}

			break;

		case RTLTOOL_READ_PHY:
			spin_lock_irqsave(&tp->phy_lock, flags);
			my_cmd.data = mdio_read(tp->mmio_addr, my_cmd.offset);
			spin_unlock_irqrestore(&tp->phy_lock, flags);

			if (copy_to_user(ifr->ifr_data, &my_cmd, sizeof(struct rtltool_cmd)))
			{
				ret = -EFAULT;
				break;
			}

			break;

		case RTLTOOL_WRITE_PHY:
			spin_lock_irqsave(&tp->phy_lock, flags);
			mdio_write(tp->mmio_addr, my_cmd.offset, my_cmd.data);
			spin_unlock_irqrestore(&tp->phy_lock, flags);
			break;

		case RTLTOOL_READ_EPHY:
			my_cmd.data = rtl8101_ephy_read(tp->mmio_addr, my_cmd.offset);

			if (copy_to_user(ifr->ifr_data, &my_cmd, sizeof(struct rtltool_cmd)))
			{
				ret = -EFAULT;
				break;
			}

			break;

		case RTLTOOL_WRITE_EPHY:
			rtl8101_ephy_write(tp->mmio_addr, my_cmd.offset, my_cmd.data);
			break;

		case RTLTOOL_READ_PCI:
			my_cmd.data = 0;
			if(my_cmd.len==1)
			{
				pci_read_config_byte(tp->pci_dev, my_cmd.offset, (u8 *)&my_cmd.data);
			}
			else if(my_cmd.len==2)
			{
				pci_read_config_word(tp->pci_dev, my_cmd.offset, (u16 *)&my_cmd.data);
			}
			else if(my_cmd.len==4)
			{
				pci_read_config_dword(tp->pci_dev, my_cmd.offset, &my_cmd.data);
			}
			else
			{
				ret = -EOPNOTSUPP;
				break;
			}

			if (copy_to_user(ifr->ifr_data, &my_cmd, sizeof(struct rtltool_cmd)))
			{
				ret = -EFAULT;
				break;
			}
			break;

		case RTLTOOL_WRITE_PCI:
			if(my_cmd.len==1)
			{
				pci_write_config_byte(tp->pci_dev, my_cmd.offset, my_cmd.data);
			}
			else if(my_cmd.len==2)
			{
				pci_write_config_word(tp->pci_dev, my_cmd.offset, my_cmd.data);
			}
			else if(my_cmd.len==4)
			{
				pci_write_config_dword(tp->pci_dev, my_cmd.offset, my_cmd.data);
			}
			else
			{
				ret = -EOPNOTSUPP;
				break;
			}

			break;

		default:
			ret = -EOPNOTSUPP;
			break;
	}

	return ret;
}