/*
 * (C) 2004-2006  Sebastian Witt <se.witt@gmx.net>
 *
 *  Licensed under the terms of the GNU GPL License version 2.
 *  Based upon reverse engineered information
 *
 *  BIG FAT DISCLAIMER: Work in progress code. Possibly *dangerous*
 */

#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/cpufreq.h>
#include <linux/pci.h>
#include <linux/delay.h>

#define NFORCE2_XTAL 25
#define NFORCE2_BOOTFSB 0x48
#define NFORCE2_PLLENABLE 0xa8
#define NFORCE2_PLLREG 0xa4
#define NFORCE2_PLLADR 0xa0
#define NFORCE2_PLL(mul, div) (0x100000 | (mul << 8) | div)

#define NFORCE2_MIN_FSB 50
#define NFORCE2_SAFE_DISTANCE 50

/* Delay in ms between FSB changes */
/* #define NFORCE2_DELAY 10 */

/*
 * nforce2_chipset:
 * FSB is changed using the chipset
 */
static struct pci_dev *nforce2_dev;

/* fid:
 * multiplier * 10
 */
static int fid;

/* min_fsb, max_fsb:
 * minimum and maximum FSB (= FSB at boot time)
 */
static int min_fsb;
static int max_fsb;

MODULE_AUTHOR("Sebastian Witt <se.witt@gmx.net>");
MODULE_DESCRIPTION("nForce2 FSB changing cpufreq driver");
MODULE_LICENSE("GPL");

module_param(fid, int, 0444);
module_param(min_fsb, int, 0444);

MODULE_PARM_DESC(fid, "CPU multiplier to use (11.5 = 115)");
MODULE_PARM_DESC(min_fsb,
                "Minimum FSB to use, if not defined: current FSB - 50");

#define PFX "cpufreq-nforce2: "

/**
 * nforce2_calc_fsb - calculate FSB
 * @pll: PLL value
 *
 *   Calculates FSB from PLL value
 */
static int nforce2_calc_fsb(int pll)
{
        unsigned char mul, div;

        mul = (pll >> 8) & 0xff;
        div = pll & 0xff;

        if (div > 0)
                return NFORCE2_XTAL * mul / div;

        return 0;
}

/**
 * nforce2_calc_pll - calculate PLL value
 * @fsb: FSB
 *
 *   Calculate PLL value for given FSB
 */
static int nforce2_calc_pll(unsigned int fsb)
{
        unsigned char xmul, xdiv;
        unsigned char mul = 0, div = 0;
        int tried = 0;

        /* Try to calculate multiplier and divider up to 4 times */
        while (((mul == 0) || (div == 0)) && (tried <= 3)) {
                for (xdiv = 2; xdiv <= 0x80; xdiv++)
                        for (xmul = 1; xmul <= 0xfe; xmul++)
                                if (nforce2_calc_fsb(NFORCE2_PLL(xmul, xdiv)) ==
                                    fsb + tried) {
                                        mul = xmul;
                                        div = xdiv;
                                }
                tried++;
        }

        if ((mul == 0) || (div == 0))
                return -1;

        return NFORCE2_PLL(mul, div);
}

/**
 * nforce2_write_pll - write PLL value to chipset
 * @pll: PLL value
 *
 *   Writes new FSB PLL value to chipset
 */
static void nforce2_write_pll(int pll)
{
        int temp;

        /* Set the pll addr. to 0x00 */
        pci_write_config_dword(nforce2_dev, NFORCE2_PLLADR, 0);

        /* Now write the value in all 64 registers */
        for (temp = 0; temp <= 0x3f; temp++)
                pci_write_config_dword(nforce2_dev, NFORCE2_PLLREG, pll);

        return;
}

/**
 * nforce2_fsb_read - Read FSB
 *
 *   Read FSB from chipset
 *   If bootfsb != 0, return FSB at boot-time
 */
static unsigned int nforce2_fsb_read(int bootfsb)
{
        struct pci_dev *nforce2_sub5;
        u32 fsb, temp = 0;

        /* Get chipset boot FSB from subdevice 5 (FSB at boot-time) */
        nforce2_sub5 = pci_get_subsys(PCI_VENDOR_ID_NVIDIA, 0x01EF,
                                PCI_ANY_ID, PCI_ANY_ID, NULL);
        if (!nforce2_sub5)
                return 0;

        pci_read_config_dword(nforce2_sub5, NFORCE2_BOOTFSB, &fsb);
        fsb /= 1000000;

        /* Check if PLL register is already set */
        pci_read_config_byte(nforce2_dev, NFORCE2_PLLENABLE, (u8 *)&temp);

        if (bootfsb || !temp)
                return fsb;

        /* Use PLL register FSB value */
        pci_read_config_dword(nforce2_dev, NFORCE2_PLLREG, &temp);
        fsb = nforce2_calc_fsb(temp);

        return fsb;
}

/**
 * nforce2_set_fsb - set new FSB
 * @fsb: New FSB
 *
 *   Sets new FSB
 */
static int nforce2_set_fsb(unsigned int fsb)
{
        u32 temp = 0;
        unsigned int tfsb;
        int diff;
        int pll = 0;

        if ((fsb > max_fsb) || (fsb < NFORCE2_MIN_FSB)) {
                printk(KERN_ERR PFX "FSB %d is out of range!\n", fsb);
                return -EINVAL;
        }

        tfsb = nforce2_fsb_read(0);
        if (!tfsb) {
                printk(KERN_ERR PFX "Error while reading the FSB\n");
                return -EINVAL;
        }

        /* First write? Then set actual value */
        pci_read_config_byte(nforce2_dev, NFORCE2_PLLENABLE, (u8 *)&temp);
        if (!temp) {
                pll = nforce2_calc_pll(tfsb);

                if (pll < 0)
                        return -EINVAL;

                nforce2_write_pll(pll);
        }

        /* Enable write access */
        temp = 0x01;
        pci_write_config_byte(nforce2_dev, NFORCE2_PLLENABLE, (u8)temp);

        diff = tfsb - fsb;

        if (!diff)
                return 0;

        while ((tfsb != fsb) && (tfsb <= max_fsb) && (tfsb >= min_fsb)) {
                if (diff < 0)
                        tfsb++;
                else
                        tfsb--;

                /* Calculate the PLL reg. value */
                pll = nforce2_calc_pll(tfsb);
                if (pll == -1)
                        return -EINVAL;

                nforce2_write_pll(pll);
#ifdef NFORCE2_DELAY
                mdelay(NFORCE2_DELAY);
#endif
        }

        temp = 0x40;
        pci_write_config_byte(nforce2_dev, NFORCE2_PLLADR, (u8)temp);

        return 0;
}

/**
 * nforce2_get - get the CPU frequency
 * @cpu: CPU number
 *
 * Returns the CPU frequency
 */
static unsigned int nforce2_get(unsigned int cpu)
{
        if (cpu)
                return 0;
        return nforce2_fsb_read(0) * fid * 100;
}

/**
 * nforce2_target - set a new CPUFreq policy
 * @policy: new policy
 * @target_freq: the target frequency
 * @relation: how that frequency relates to achieved frequency
 *  (CPUFREQ_RELATION_L or CPUFREQ_RELATION_H)
 *
 * Sets a new CPUFreq policy.
 */
static int nforce2_target(struct cpufreq_policy *policy,
                          unsigned int target_freq, unsigned int relation)
{
/*        unsigned long         flags; */
        struct cpufreq_freqs freqs;
        unsigned int target_fsb;

        if ((target_freq > policy->max) || (target_freq < policy->min))
                return -EINVAL;

        target_fsb = target_freq / (fid * 100);

        freqs.old = nforce2_get(policy->cpu);
        freqs.new = target_fsb * fid * 100;

        if (freqs.old == freqs.new)
                return 0;

        pr_debug("Old CPU frequency %d kHz, new %d kHz\n",
               freqs.old, freqs.new);

        cpufreq_freq_transition_begin(policy, &freqs);

        /* Disable IRQs */
        /* local_irq_save(flags); */

        if (nforce2_set_fsb(target_fsb) < 0)
                printk(KERN_ERR PFX "Changing FSB to %d failed\n",
                        target_fsb);
        else
                pr_debug("Changed FSB successfully to %d\n",
                        target_fsb);

        /* Enable IRQs */
        /* local_irq_restore(flags); */

        cpufreq_freq_transition_end(policy, &freqs, 0);

        return 0;
}

/**
 * nforce2_verify - verifies a new CPUFreq policy
 * @policy: new policy
 */
static int nforce2_verify(struct cpufreq_policy *policy)
{
        unsigned int fsb_pol_max;

        fsb_pol_max = policy->max / (fid * 100);

        if (policy->min < (fsb_pol_max * fid * 100))
                policy->max = (fsb_pol_max + 1) * fid * 100;

        cpufreq_verify_within_cpu_limits(policy);
        return 0;
}

static int nforce2_cpu_init(struct cpufreq_policy *policy)
{
        unsigned int fsb;
        unsigned int rfid;

        /* capability check */
        if (policy->cpu != 0)
                return -ENODEV;

        /* Get current FSB */
        fsb = nforce2_fsb_read(0);

        if (!fsb)
                return -EIO;

        /* FIX: Get FID from CPU */
        if (!fid) {
                if (!cpu_khz) {
                        printk(KERN_WARNING PFX
                        "cpu_khz not set, can't calculate multiplier!\n");
                        return -ENODEV;
                }

                fid = cpu_khz / (fsb * 100);
                rfid = fid % 5;

                if (rfid) {
                        if (rfid > 2)
                                fid += 5 - rfid;
                        else
                                fid -= rfid;
                }
        }

        printk(KERN_INFO PFX "FSB currently at %i MHz, FID %d.%d\n", fsb,
               fid / 10, fid % 10);

        /* Set maximum FSB to FSB at boot time */
        max_fsb = nforce2_fsb_read(1);

        if (!max_fsb)
                return -EIO;

        if (!min_fsb)
                min_fsb = max_fsb - NFORCE2_SAFE_DISTANCE;

        if (min_fsb < NFORCE2_MIN_FSB)
                min_fsb = NFORCE2_MIN_FSB;

        /* cpuinfo and default policy values */
        policy->min = policy->cpuinfo.min_freq = min_fsb * fid * 100;
        policy->max = policy->cpuinfo.max_freq = max_fsb * fid * 100;
        policy->cpuinfo.transition_latency = CPUFREQ_ETERNAL;

        return 0;
}

static int nforce2_cpu_exit(struct cpufreq_policy *policy)
{
        return 0;
}

static struct cpufreq_driver nforce2_driver = {
        .name = "nforce2",
        .verify = nforce2_verify,
        .target = nforce2_target,
        .get = nforce2_get,
        .init = nforce2_cpu_init,
        .exit = nforce2_cpu_exit,
};

#ifdef MODULE
static const struct pci_device_id nforce2_ids[] = {
        { PCI_VENDOR_ID_NVIDIA, PCI_DEVICE_ID_NVIDIA_NFORCE2 },
        {}
};
MODULE_DEVICE_TABLE(pci, nforce2_ids);
#endif

/**
 * nforce2_detect_chipset - detect the Southbridge which contains FSB PLL logic
 *
 * Detects nForce2 A2 and C1 stepping
 *
 */
static int nforce2_detect_chipset(void)
{
        nforce2_dev = pci_get_subsys(PCI_VENDOR_ID_NVIDIA,
                                        PCI_DEVICE_ID_NVIDIA_NFORCE2,
                                        PCI_ANY_ID, PCI_ANY_ID, NULL);

        if (nforce2_dev == NULL)
                return -ENODEV;

        printk(KERN_INFO PFX "Detected nForce2 chipset revision %X\n",
               nforce2_dev->revision);
        printk(KERN_INFO PFX
               "FSB changing is maybe unstable and can lead to "
               "crashes and data loss.\n");

        return 0;
}

/**
 * nforce2_init - initializes the nForce2 CPUFreq driver
 *
 * Initializes the nForce2 FSB support. Returns -ENODEV on unsupported
 * devices, -EINVAL on problems during initialization, and zero on
 * success.
 */
static int __init nforce2_init(void)
{
        /* TODO: do we need to detect the processor? */

        /* detect chipset */
        if (nforce2_detect_chipset()) {
                printk(KERN_INFO PFX "No nForce2 chipset.\n");
                return -ENODEV;
        }

        return cpufreq_register_driver(&nforce2_driver);
}

/**
 * nforce2_exit - unregisters cpufreq module
 *
 *   Unregisters nForce2 FSB change support.
 */
static void __exit nforce2_exit(void)
{
        cpufreq_unregister_driver(&nforce2_driver);
}

module_init(nforce2_init);
module_exit(nforce2_exit);