JZ2440笔记：DMA驱动

vi dma.c

#include <linux/module.h>
#include <linux/interrupt.h>
#include <linux/irq.h>
#include <linux/io.h>
#include <linux/input.h>
#include <linux/dma-mapping.h>


#define MEM_CPY_NO_DMA 0
#define MEM_CPY_DMA 1

#define BUF_SIZE (512*1024)

#define DMA0_BASE_ADDR 0x4b000000
#define DMA1_BASE_ADDR 0x4b000040
#define DMA2_BASE_ADDR 0x4b000080
#define DMA3_BASE_ADDR 0x4b0000c0


struct s3c_dma_regs {
	unsigned long disrc;
	unsigned long disrcc;
	unsigned long didst;
	unsigned long didstc;
	unsigned long dcon; 
	unsigned long dstat; 
	unsigned long dcsrc; 
	unsigned long dcdst; 
	unsigned long dmasktrig;
};

static char *src;
static u32 src_phys;
static char *dst;
static u32 dst_phys;

static int major = 0;
static struct class *cls;
static volatile struct s3c_dma_regs *dma_regs;

static DECLARE_WAIT_QUEUE_HEAD(dma_waitq);
static volatile int ev_dma = 0;

static int s3c_dma_ioctl (struct inode *inode, struct file *file, unsigned int cmd, unsigned long data)
{
	int i;

	memset(src,0xAA,BUF_SIZE);
	memset(dst,0x55,BUF_SIZE);
	
	switch(cmd)
	{
		case MEM_CPY_NO_DMA:
		{
			for(i=0;i<BUF_SIZE;i++)
				dst[i] = src[i];
			if(memcmp(src,dst,BUF_SIZE)==0)
				printk("MEM_CPY_NO_DMA OK\n");
			else
				printk("MEM_CPY_NO_DMA ERROR\n");
			break;
		}
		case MEM_CPY_DMA:
		{
			dma_regs->disrc  = src_phys;
			dma_regs->disrcc = (0<<1)|(0<<0);
			dma_regs->didst  = dst_phys;
			dma_regs->didstc = (0<<2)|(0<<1)|(0<<0);
			//dma_regs->dcon   = (1<<29)|(0<<28)|(0<<23)|(0<<20)|(BUF_SIZE<<0);
			dma_regs->dcon   = (1<<30)|(1<<29)|(0<<28)|(1<<27)|(0<<23)|(0<<20)|(BUF_SIZE<<0);
            dma_regs->dmasktrig = (1<<1)|(1<<0);
			wait_event_interruptible(dma_waitq, ev_dma);
			if(memcmp(src,dst,BUF_SIZE)==0)
				printk("MEM_CPY_DMA OK\n");
			else
				printk("MEM_CPY_DMA ERROR\n");
			break;
		}
	}
	return 0;
}


static struct file_operations dma_fops = {
	.owner = THIS_MODULE,
	.ioctl = s3c_dma_ioctl,
};

static irqreturn_t s3c_dma_irq(int irq, void *dev_id)
{
	ev_dma = 1;
	wake_up_interruptible(&dma_waitq);
	return IRQ_HANDLED;
}


static int s3c_dma_init(void)
{
	if(request_irq(IRQ_DMA3,s3c_dma_irq,0,"s3c_dma",1))
	{
		printk("can't request_irq for DMA3\n");
		return -EBUSY;
	}
	src = dma_alloc_writecombine(NULL, BUF_SIZE, &src_phys, GFP_KERNEL);
	if(NULL==src)
	{
		free_irq(IRQ_DMA3,1);
		printk("can't alloc buffer for src\n");
		return -ENOMEM;
	}
	dst = dma_alloc_writecombine(NULL, BUF_SIZE, &dst_phys, GFP_KERNEL);
	if(NULL==dst)
	{
		dma_free_writecombine(NULL, BUF_SIZE, src, src_phys);
		printk("can't alloc buffer for dst\n");
		return -ENOMEM;
	}

	major = register_chrdev(0,"s3c_dma",&dma_fops);
	cls = class_create(THIS_MODULE, "s3c_dma");
	class_device_create(cls, NULL, MKDEV(major, 0), NULL, "dma");

	dma_regs = ioremap(DMA3_BASE_ADDR,sizeof(struct s3c_dma_regs));

	return 0;
}

static void s3c_dma_exit(void)
{
	iounmap(dma_regs);
	class_device_destroy(cls, MKDEV(major, 0));
	class_destroy(cls);
	unregister_chrdev(major,"s3c_dma");
	dma_free_writecombine(NULL, BUF_SIZE, src, src_phys);
	dma_free_writecombine(NULL, BUF_SIZE, dst, dst_phys);
	free_irq(IRQ_DMA3,1);
}

module_init(s3c_dma_init); 
module_exit(s3c_dma_exit);

MODULE_LICENSE("GPL");

vi dma_test.c

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <string.h>


#define MEM_CPY_NO_DMA 0
#define MEM_CPY_DMA 1


void print_usage(char *name)
{
	printf("Usage:\n");
	printf("%s [nodma | dma]\n",name);
}

int main(int argc, char **argv)
{
	int fd;
	if(argc != 2)
	{
		print_usage(argv[0]);
		return -1;
	}
	fd = open("/dev/dma",O_RDWR);
	if(fd<0)
	{
		printf("can't open /dev/dma\n");
		return -1;
	}
	if(strcmp(argv[1],"nodma")==0)
	{
		while(1)
		{
			ioctl(fd,MEM_CPY_NO_DMA);
		}
	}
	else if(strcmp(argv[1],"dma")==0)
	{
		while(1)
		{
			ioctl(fd,MEM_CPY_DMA);
		}
	}
	else
	{
		print_usage(argv[0]);
		return -1;
	}
	return 0;
}	

测试：

# ls /dev/dma -l
ls: /dev/dma: No such file or directory
# insmod dma.ko
# ls /dev/dma -l
crw-rw----    1 0        0        252,   0 Jun  4 13:59 /dev/dma

# ./dma_test nodma &
[1] - Done(255)                  ./dma_test
# MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
lsMEM_CPY_NO_DMA OK

MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
MEM_CPY_NO_DMA OK
Makefile        dma.ko          dma.o
Module.symvers  dma.mod.c       dma_test
dma.c           dma.mod.o       dma_test.c
 

# ./dma_test dma &
# MEM_CPY_DMA OK
# ls
MEM_CPY_DMA OK
Makefile        dma.ko          dma.o
Module.symvers  dma.mod.c       dma_test
dma.c           dma.mod.o       dma_test.c
# ls
MEM_CPY_DMA OK
Makefile        dma.ko          dma.o
Module.symvers  dma.mod.c       dma_test
dma.c           dma.mod.o       dma_test.c