Skip to content

P
Production Test Suite - base
Commit 8c817685 authored by Matthieu Cattin's avatar Matthieu Cattin
Browse files
Options

Add rrlib and python wrapper sources.

parent 015ed5a1
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 795 additions and 0 deletions
gnurabbit/python/Makefile 0 → 100644
View file @ 8c817685
CFLAGS = -Wall -ggdb -I../kernel
AS = $(CROSS_COMPILE)as
LD = $(CROSS_COMPILE)ld
CC = $(CROSS_COMPILE)gcc
CPP = $(CC) -E
AR = $(CROSS_COMPILE)ar
NM = $(CROSS_COMPILE)nm
STRIP = $(CROSS_COMPILE)strip
OBJCOPY = $(CROSS_COMPILE)objcopy
OBJDUMP = $(CROSS_COMPILE)objdump
ALL = rrlib.so
all: $(ALL)
rrlib.so: rrlib.o
$(CC) $(CFLAGS) -o $@ -shared $^
clean:
rm -f $(ALL) *.o *~ *.so *.pyc
gnurabbit/python/rr.html 0 → 100644
View file @ 8c817685
This diff is collapsed.
gnurabbit/python/rr.py 0 → 100644
View file @ 8c817685
#! /usr/bin/env python
# :vi:ts=4 sw=4 et
from ctypes import *
import os, errno, re, sys, struct
# unsigned formats to unpack words
fmt = { 1: 'B', 2: 'H', 4: 'I', 8: 'L' }
# some defaults from rawrabbit.h
RR_DEVSEL_UNUSED = 0xffff
RR_DEFAULT_VENDOR = 0x1a39
RR_DEFAULT_DEVICE = 0x0004
RR_BAR_0 = 0x00000000
RR_BAR_2 = 0x20000000
RR_BAR_4 = 0x40000000
RR_BAR_BUF = 0xc0000000
bar_map = {
0 : RR_BAR_0,
2: RR_BAR_2,
4: RR_BAR_4,
0xc: RR_BAR_BUF }
# classes to interface with the driver via ctypes
Plist = c_int * 256
class RR_Devsel(Structure):
_fields_ = [
("vendor", c_ushort),
("device", c_ushort),
("subvendor", c_ushort),
("subdevice", c_ushort),
("bus", c_ushort),
("devfn", c_ushort),
]
class RR_U(Union):
_fields_ = [
("data8", c_ubyte),
("data16", c_ushort),
("data32", c_uint),
("data64", c_ulonglong),
]
class RR_Iocmd(Structure):
_anonymous_ = [ "data", ]
_fields_ = [
("address", c_uint),
("datasize", c_uint),
("data", RR_U),
]
class Gennum(object):
device = '/dev/rawrabbit'
rrlib = './rrlib.so'
def __init__(self):
"""get a file descriptor for the Gennum device"""
self.lib = CDLL(Gennum.rrlib)
self.fd = os.open(Gennum.device, os.O_RDWR)
def iread(self, bar, offset, width):
"""do a read by means of the ioctl interface
bar = 0, 2, 4 (or c for DMA buffer access
offset = address within bar
width = data size (1, 2, 4 or 8 bytes)
"""
address = bar_map[bar] + offset
ds = RR_Iocmd(address=address, datasize=width)
print self.lib.rr_read(self.fd, byref(ds))
return ds.data32
def read(self, bar, offset, width):
"""do a read by means of lseek+read
bar = 0, 2, 4 (or c for DMA buffer access
offset = address within bar
width = data size (1, 2, 4 or 8 bytes)
"""
address = bar_map[bar] + offset
os.lseek(self.fd, address, os.SEEK_SET)
buf = os.read(self.fd, width)
return struct.unpack(fmt[width], buf)[0]
def iwrite(self, bar, offset, width, datum):
"""do a write by means of the ioctl interface
bar = 0, 2, 4 (or c for DMA buffer access
offset = address within bar
width = data size (1, 2, 4 or 8 bytes)
datum = value to be written
"""
address = bar_map[bar] + offset
ds = RR_Iocmd(address=address, datasize=width, data32=datum)
print self.lib.rr_iwrite(self.fd, byref(ds))
return ds.data32
def write(self, bar, offset, width, datum):
"""do a write by means of lseek+write
bar = 0, 2, 4 (or c for DMA buffer access
offset = address within bar
width = data size (1, 2, 4 or 8 bytes)
datum = value to be written
"""
address = bar_map[bar] + offset
os.lseek(self.fd, address, os.SEEK_SET)
return os.write(self.fd, struct.pack(fmt[width], datum))
def irqwait(self):
"""wait for an interrupt"""
return self.lib.rr_irqwait(self.fd);
def irqena(self):
"""enable the interrupt line"""
return self.lib.rr_irqena(self.fd);
def getdmasize(self):
"""return the size of the allocated DMA buffer (in bytes)"""
return self.lib.rr_getdmasize(self.fd);
def getplist(self):
"""get a list of pages for DMA access
The addresses returned, shifted by 12 bits, give the physical
addresses of the allocated pages
"""
plist = Plist()
self.lib.rr_getplist(self.fd, plist);
return plist
def info(self):
"""get a string describing the interface the driver is bound to
The syntax of the string is
vendor:device/dubvendor:subdevice@bus:devfn
"""
ds = RR_Devsel()
self.errno = self.lib.rr_devget(self.fd, byref(ds))
for key in RR_Devsel._fields_:
setattr(self, key[0], getattr(ds, key[0], RR_DEVSEL_UNUSED))
return '%04x:%04x/%04x:%04x@%04x:%04x' % (
ds.vendor, ds.device,
ds.subvendor, ds.subdevice,
ds.bus, ds.devfn)
def parse_addr(self, addr):
"""take a string of the form
vendor:device[/subvendor:subdevice][@bus:devfn]
and return a dictionary object with the corresponding values,
initialized to RR_DEVSEL_UNUSED when absent
"""
# address format
reg = ( r'(?i)^'
r'(?P<vendor>[a-f0-9]{1,4}):(?P<device>[a-f0-9]{1,4})'
r'(/(?P<subvendor>[a-f0-9]{1,4}):(?P<subdevice>[a-f0-9]{1,4}))?'
r'(@(?P<bus>[a-f0-9]{1,4}):(?P<devfn>[a-f0-9]{1,4}))?$' )
match = re.match(reg, addr).groupdict()
if not 'sub' in match:
match['subvendor'] = match['subdevice'] = RR_DEVSEL_UNUSED
if not 'geo' in match:
match['bus'] = match['devfn'] = RR_DEVSEL_UNUSED
for k, v in match.items():
if type(v) is str:
match[k] = int(v, 16)
return match
def bind(self, device):
"""bind the rawrabbit driver to a device
The device is specified with a syntax described in parse_addr
"""
d = self.parse_addr(device)
ds = RR_Devsel(**d)
self.errno = self.lib.rr_devsel(self.fd, byref(ds))
return self.errno
if __name__ == '__main__':
g = Gennum()
print g.parse_addr('1a39:0004/1a39:0004@0020:0000')
print g.bind('1a39:0004/1a39:0004@0020:0000')
print '%x' % g.write(bar=RR_BAR_4, offset=0xa08, width=4, datum=0xdeadface)
print '%x' % g.read(bar=RR_BAR_4, offset=0xa08, width=4)
print g.getdmasize()
for page in g.getplist():
print '%08x ' % (page<<12),
gnurabbit/python/rrcmd.py 0 → 100644
View file @ 8c817685
#! /usr/bin/env python
# coding: utf8
import rr
import cmd
class RrCmd(cmd.Cmd):
def preloop(self):
print 'Initializing...'
self.gennum = rr.Gennum()
self.do_info('')
self.prompt = 'rr> '
def do_info(self, args):
print 'bound to ' , self.gennum.info()
def do_bind(self, args):
self.gennum.bind(args)
print 'bound to ' , self.gennum.info()
def do_irqwait(self, args):
self.gennum.irqwait()
def do_read(self, args):
try:
size, bar, offset = args.split()
size = int(size)
bar = int(bar)
offset = int(offset, 16)
except ValueError:
print "syntax: read size bar address"
return
print hex(self.gennum.read(bar, offset, size))
def do_write(self, args):
try:
size, bar, offset, datum = args.split()
size = int(size)
bar = int(bar)
offset = int(offset, 16)
datum = int(datum, 16)
except ValueError:
print "syntax: write size bar address datum"
return
self.gennum.write(bar, offset, size, datum)
def do_irqena(self, args):
self.gennum.irqena()
def do_getdmasize(self, args):
print hex(self.gennum.getdmasize())
def do_getplist(self, args):
plist = self.gennum.getplist()
width = 8
for i, addr in enumerate(plist):
if i % width == 0:
print '%08x ' % i,
addr <<= 12
print '%08x ' % addr,
if i % width == width-1:
print
else:
if i % width != width-1:
print
def emptyline(self):
pass
def do_EOF(self, args):
return True
do_quit = do_exit = do_EOF
if __name__ == '__main__':
gennum = rr.Gennum()
cmd = RrCmd()
cmd.cmdloop()
gnurabbit/python/rrlib.c 0 → 100644
View file @ 8c817685
#include <sys/ioctl.h>
#include <errno.h>
#include <stdint.h>
#include <fcntl.h>
#include <rawrabbit.h>
#include <stdio.h>
int rr_devsel(int fd, struct rr_devsel *ds)
{
if (ioctl(fd, RR_DEVSEL, ds) < 0)
return -errno;
return 0;
}
int rr_devget(int fd, struct rr_devsel *ds)
{
if (ioctl(fd, RR_DEVGET, ds) < 0)
return -errno;
return 0;
}
int rr_iread(int fd, struct rr_iocmd *iocmd)
{
if (ioctl(fd, RR_READ, iocmd) < 0)
return -errno;
return 0;
}
int rr_iwrite(int fd, struct rr_iocmd *iocmd)
{
if (ioctl(fd, RR_WRITE, iocmd) < 0)
return -errno;
return 0;
}
int rr_irqwait(int fd)
{
if (ioctl(fd, RR_IRQWAIT) < 0)
return -errno;
return 0;
}
int rr_irqena(int fd)
{
if (ioctl(fd, RR_IRQENA) < 0)
return -errno;
return 0;
}
int rr_getdmasize(int fd)
{
return ioctl(fd, RR_GETDMASIZE);
}
int rr_getplist(int fd, uintptr_t *plist)
{
int i, size;
size = ioctl(fd, RR_GETDMASIZE);
if (size < 0)
return -errno;
i = ioctl(fd, RR_GETPLIST, plist);
if (i < 0)
return -errno;
return 0;
}
int main(int argc, char *argv[])
{
struct rr_iocmd cmd, *cmdp = &cmd;
int ret;
int fd = open("/dev/rawrabbit", O_RDWR);
printf("fd = %d\n", fd);
cmdp->address = 0x40000a08;
cmdp->datasize = 4;
cmdp->data32 = 0xdeadface;
cmdp->data32 = 0x00000000;
ret = rr_iread(fd, cmdp);
printf("ret = %d, errno = %d\n", ret, errno);
printf("0x%08x 0x%08x 0x%08x\n", cmdp->address, cmdp->datasize, cmdp->data32);
return 0;
}
gnurabbit/python/rrlib.h 0 → 100644
View file @ 8c817685
#include <rawrabbit.h>
int rr_devsel(int fd, struct rr_devsel *ds);
int rr_devget(int fd, struct rr_devsel *ds);
int rr_read(int fd, struct rr_iocmd *iocmd);
int rr_write(int fd, struct rr_iocmd *iocmd);
int rr_irqwait(int fd);
int rr_irqena(int fd);
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment