devtime.c

来自「linux下基于c++的处理器仿真平台。具有处理器流水线」· C语言 代码 · 共 199 行

/*
 * Copyright (c) 2004
 * The Regents of The University of Michigan
 * All Rights Reserved
 *
 * This code is part of the M5 simulator, developed by Nathan Binkert,
 * Erik Hallnor, Steve Raasch, and Steve Reinhardt, with contributions
 * from Ron Dreslinski, Dave Greene, Lisa Hsu, Kevin Lim, Ali Saidi,
 * and Andrew Schultz.
 *
 * Permission is granted to use, copy, create derivative works and
 * redistribute this software and such derivative works for any
 * purpose, so long as the copyright notice above, this grant of
 * permission, and the disclaimer below appear in all copies made; and
 * so long as the name of The University of Michigan is not used in
 * any advertising or publicity pertaining to the use or distribution
 * of this software without specific, written prior authorization.
 *
 * THIS SOFTWARE IS PROVIDED AS IS, WITHOUT REPRESENTATION FROM THE
 * UNIVERSITY OF MICHIGAN AS TO ITS FITNESS FOR ANY PURPOSE, AND
 * WITHOUT WARRANTY BY THE UNIVERSITY OF MICHIGAN OF ANY KIND, EITHER
 * EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE. THE REGENTS OF THE UNIVERSITY OF MICHIGAN SHALL NOT BE
 * LIABLE FOR ANY DAMAGES, INCLUDING DIRECT, SPECIAL, INDIRECT,
 * INCIDENTAL, OR CONSEQUENTIAL DAMAGES, WITH RESPECT TO ANY CLAIM
 * ARISING OUT OF OR IN CONNECTION WITH THE USE OF THE SOFTWARE, EVEN
 * IF IT HAS BEEN OR IS HEREAFTER ADVISED OF THE POSSIBILITY OF SUCH
 * DAMAGES.
 */

#include <linux/module.h>
#include <linux/config.h>
#include <linux/moduleparam.h>
#include <linux/init.h>
#include <linux/fs.h>
#include <asm/uaccess.h>
#include <linux/kernel.h>
#include <asm/io.h>
#include <asm/page.h>
#include <linux/netdevice.h>

#ifdef __i386__
#include <asm/processor.h>
#include <asm/msr.h>
#endif

#define DRIVER_AUTHOR "Ali Saidi"
#define DRIVER_DESC   "Interface to time uncacachable read and writes to device registers"
#define DRIVER_VER    "0.1"

static char *dataAddr = NULL;
static int count = 0;
#ifdef __alpha__
static int memTest = 0;
#endif

static inline uint32_t cycleCounter(uint32_t dep);

static int __init devtime_start(void) 
{
    uint64_t addr;
    uint32_t t1, t2;
    uint32_t trash;
    int x;
    uint32_t *times;
    uint32_t num = 0;
    struct net_device *dev; 

    printk("Devtime Driver Version %s Loaded...\n", DRIVER_VER);

#ifdef __alpha__
    if (memTest) {
           addr = 0xfffffc0000000000;
//         addr += 16*1024*1024;

            printk("Preparing memory test.\n");

            t1 = cycleCounter(trash);
            for (x = 0; x < count; x++) {
                trash = readl(addr);
                t2 = cycleCounter(trash);
                times[num++] = t2 - t1;
                t1 = t2;
               addr += 4096;
            }

            printk("Measurements:\n");
            for (x = 0; x < count; x++) {
                printk("%d ", times[x]);
                if (((x + 1) % 10) == 0)
                    printk("\n");
            }
            printk("\nDone.\n");
    } else
#endif
    if (dataAddr != 0 && count != 0) {
        addr = simple_strtoull(dataAddr, NULL, 0);

        addr = ioremap(addr, PAGE_SIZE); 
        /**
         * Make sure that the remapping actually worked. On alpha we have
         * linear addressing, so its not a problem. But it can fail in x86
         * if physical memory is mapped to this address.
         */
        times = kmalloc(sizeof(uint32_t) * count, GFP_USER);
        if (!times) {
            printk("Could not allocate memory... Try again later.\n");
            return -1;
        }

        if (addr) {
            printk("Preparing to read %#llx %d times.\n", addr, count);

            t1 = cycleCounter(trash);
            for (x = 0; x < count; x++) {
                trash = readl(addr);
                t2 = cycleCounter(trash);
                times[num++] = t2 - t1;
                t1 = t2;
            }
   
            /**
             * Unmap the address.
             */
            iounmap(addr);

            printk("Measurements:\n");
            for (x = 0; x < count; x++) {
                printk("%d ", times[x]);
                if (((x + 1) % 10) == 0)
                    printk("\n");
            }
            printk("\nDone.\n");
        } else {
            printk("Unable to remap address. Please try again later.\n");
        }
    } else {
        dev = dev_get_by_name("eth0");
        if (dev) {
            printk("Eth0: MemStart: %#lx MemEnd: %#lx I/O Addr: %#lx\n",
                   dev->mem_start, dev->mem_end, dev->base_addr);
            dev_put(dev);
        }
        dev = 0;
        dev = dev_get_by_name("eth1");
        if (dev) {
            printk("Eth1: MemStart: %#lx MemEnd: %#lx I/O Addr: %#lx\n",
                   dev->mem_start, dev->mem_end, dev->base_addr);
            dev_put(dev);
        }

        printk("Required information not supplied.\n");
    }
        
    return 0; 
}

#ifdef __i386__

static inline uint32_t cycleCounter(uint32_t dep)
{
    uint32_t time;
    cpuid_eax(0);
    rdtscl(time);
    cpuid_eax(0);
    return time;
}

#elif __alpha__

inline uint32_t cycleCounter(uint32_t dep)
{
    uint32_t res;
    asm volatile ("rpcc %0, %1" : "=r"(res) : "r" (dep) : "memory");
    return res;
}
#else
#error Architecture NOT SUPPORTED
#endif

static void __exit devtime_end(void)
{
    printk("Devtime Driver Version %s Unloaded...\n", DRIVER_VER);
}


module_init(devtime_start);
module_exit(devtime_end);

MODULE_LICENSE("Dual BSD/GPL");
MODULE_AUTHOR(DRIVER_AUTHOR);
MODULE_DESCRIPTION(DRIVER_DESC);
module_param(dataAddr, charp, 0);
module_param(count, int, 0);
#ifdef __alpha__
module_param(memTest, int, 0);
#endif