video1394.c

1394在linux下单独的驱动程序代码

			else {
				/* Wake up dma context if necessary */
				if (!(reg_read(ohci, d->ctrlSet) & 0x400)) {
					PRINT(KERN_INFO, ohci->id, 
						"Waking up iso dma ctx=%d", d->ctx);
					reg_write(ohci, d->ctrlSet, 0x1000);
				}
			}
			return 0;
			
		}
	case VIDEO1394_LISTEN_WAIT_BUFFER:
	case VIDEO1394_LISTEN_POLL_BUFFER:
	case VIDEO1394_IOC_LISTEN_WAIT_BUFFER:
	case VIDEO1394_IOC_LISTEN_POLL_BUFFER:
		{
			struct video1394_wait v;
			struct dma_iso_ctx *d;
			int i;
			
			if (copy_from_user(&v, (void *)arg, sizeof(v)))
				return -EFAULT;
			
			d = find_ctx(&ctx->context_list, OHCI_ISO_RECEIVE, v.channel);
			
			if ((v.buffer<0) || (v.buffer>d->num_desc)) {
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d out of range",v.buffer);
				return -EFAULT;
			}
			
			/*
			* I change the way it works so that it returns 
			* the last received frame.
			*/
			spin_lock_irqsave(&d->lock, flags);
			switch(d->buffer_status[v.buffer]) {
			case VIDEO1394_BUFFER_READY:
				d->buffer_status[v.buffer]=VIDEO1394_BUFFER_FREE;
				break;
			case VIDEO1394_BUFFER_QUEUED:
				if (cmd == VIDEO1394_IOC_LISTEN_POLL_BUFFER ||
					cmd == VIDEO1394_LISTEN_POLL_BUFFER) {
					/* for polling, return error code EINTR */
					spin_unlock_irqrestore(&d->lock, flags);
					return -EINTR;
				}
				
#if 1
				while (d->buffer_status[v.buffer]!=
					VIDEO1394_BUFFER_READY) {
					spin_unlock_irqrestore(&d->lock, flags);
					interruptible_sleep_on(&d->waitq);
					spin_lock_irqsave(&d->lock, flags);
					if (signal_pending(current)) {
						spin_unlock_irqrestore(&d->lock,flags);
						return -EINTR;
					}
				}
#else
				if (wait_event_interruptible(d->waitq, 
					d->buffer_status[v.buffer]
					== VIDEO1394_BUFFER_READY)
					== -ERESTARTSYS)
					return -EINTR;
#endif
				d->buffer_status[v.buffer]=VIDEO1394_BUFFER_FREE;
				break;
			default:
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d is not queued",v.buffer);
				spin_unlock_irqrestore(&d->lock, flags);
				return -EFAULT;
			}
			
			/* set time of buffer */
			v.filltime = d->buffer_time[v.buffer];
			//		printk("Buffer %d time %d\n", v.buffer, (d->buffer_time[v.buffer]).tv_usec);
			
			/*
			* Look ahead to see how many more buffers have been received
			*/
			i=0;
			while (d->buffer_status[(v.buffer+1)%d->num_desc]==
				VIDEO1394_BUFFER_READY) {
				v.buffer=(v.buffer+1)%d->num_desc;
				i++;
			}
			spin_unlock_irqrestore(&d->lock, flags);
			
			v.buffer=i;
			if (copy_to_user((void *)arg, &v, sizeof(v)))
				return -EFAULT;
			
			return 0;
		}
	case VIDEO1394_TALK_QUEUE_BUFFER:
	case VIDEO1394_IOC_TALK_QUEUE_BUFFER:
		{
			struct video1394_wait v;
			struct video1394_queue_variable qv;
			struct dma_iso_ctx *d;
			
			qv.packet_sizes = NULL;
			
			if (copy_from_user(&v, (void *)arg, sizeof(v)))
				return -EFAULT;
			
			d = find_ctx(&ctx->context_list, OHCI_ISO_TRANSMIT, v.channel);
			
			if ((v.buffer<0) || (v.buffer>d->num_desc)) {
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d out of range",v.buffer);
				return -EFAULT;
			}
			
			if (d->flags & VIDEO1394_VARIABLE_PACKET_SIZE) {
				unsigned int *psizes;
				int buf_size = d->nb_cmd * sizeof(unsigned int);
				
				if (copy_from_user(&qv, (void *)arg, sizeof(qv))) 
					return -EFAULT;
				
				psizes = kmalloc(buf_size, GFP_KERNEL);
				if (!psizes)
					return -ENOMEM;
				
				if (copy_from_user(psizes, qv.packet_sizes, buf_size)) {
					kfree(psizes);
					return -EFAULT;
				}
				
				qv.packet_sizes = psizes;
			}
			
			spin_lock_irqsave(&d->lock,flags);
			
			if (d->buffer_status[v.buffer]!=VIDEO1394_BUFFER_FREE) {
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d is already used",v.buffer);
				spin_unlock_irqrestore(&d->lock,flags);
				if (qv.packet_sizes)
					kfree(qv.packet_sizes);
				return -EFAULT;
			}
			
			if (d->flags & VIDEO1394_VARIABLE_PACKET_SIZE) {
				initialize_dma_it_prg_var_packet_queue(
					d, v.buffer, qv.packet_sizes,
					ohci);
			}
			
			d->buffer_status[v.buffer]=VIDEO1394_BUFFER_QUEUED;
			
			if (d->last_buffer >= 0) {
				d->it_prg[d->last_buffer]
					[ d->last_used_cmd[d->last_buffer] ].end.branchAddress = 
					cpu_to_le32((dma_prog_region_offset_to_bus(&d->prg_reg[v.buffer],
					0) & 0xfffffff0) | 0x3);
				
				d->it_prg[d->last_buffer]
					[ d->last_used_cmd[d->last_buffer] ].begin.branchAddress =
					cpu_to_le32((dma_prog_region_offset_to_bus(&d->prg_reg[v.buffer],
					0) & 0xfffffff0) | 0x3);
				d->next_buffer[d->last_buffer] = v.buffer;
			}
			d->last_buffer = v.buffer;
			d->next_buffer[d->last_buffer] = -1;
			
			d->it_prg[d->last_buffer][d->last_used_cmd[d->last_buffer]].end.branchAddress = 0;
			
			spin_unlock_irqrestore(&d->lock,flags);
			
			if (!(reg_read(ohci, d->ctrlSet) & 0x8000)) 
			{
				DBGMSG(ohci->id, "Starting iso transmit DMA ctx=%d",
					d->ctx);
				put_timestamp(ohci, d, d->last_buffer);
				
				/* Tell the controller where the first program is */
				reg_write(ohci, d->cmdPtr,
					dma_prog_region_offset_to_bus(&d->prg_reg[v.buffer], 0) | 0x3);
				
				/* Run IT context */
				reg_write(ohci, d->ctrlSet, 0x8000);
			}
			else {
				/* Wake up dma context if necessary */
				if (!(reg_read(ohci, d->ctrlSet) & 0x400)) {
					PRINT(KERN_INFO, ohci->id, 
						"Waking up iso transmit dma ctx=%d", 
						d->ctx);
					put_timestamp(ohci, d, d->last_buffer);
					reg_write(ohci, d->ctrlSet, 0x1000);
				}
			}
			
			if (qv.packet_sizes)
				kfree(qv.packet_sizes);
			
			return 0;
			
	}
	case VIDEO1394_TALK_WAIT_BUFFER:
	case VIDEO1394_IOC_TALK_WAIT_BUFFER:
		{
			struct video1394_wait v;
			struct dma_iso_ctx *d;
			
			if (copy_from_user(&v, (void *)arg, sizeof(v)))
				return -EFAULT;
			
			d = find_ctx(&ctx->context_list, OHCI_ISO_TRANSMIT, v.channel);
			
			if ((v.buffer<0) || (v.buffer>d->num_desc)) {
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d out of range",v.buffer);
				return -EFAULT;
			}
			
			switch(d->buffer_status[v.buffer]) {
			case VIDEO1394_BUFFER_READY:
				d->buffer_status[v.buffer]=VIDEO1394_BUFFER_FREE;
				return 0;
			case VIDEO1394_BUFFER_QUEUED:
#if 1
				while (d->buffer_status[v.buffer]!=
					VIDEO1394_BUFFER_READY) {
					interruptible_sleep_on(&d->waitq);
					if (signal_pending(current)) return -EINTR;
				}
#else
				if (wait_event_interruptible(d->waitq, 
					d->buffer_status[v.buffer]
					== VIDEO1394_BUFFER_READY)
					== -ERESTARTSYS)
					return -EINTR;
#endif
				d->buffer_status[v.buffer]=VIDEO1394_BUFFER_FREE;
				return 0;
			default:
				PRINT(KERN_ERR, ohci->id, 
					"Buffer %d is not queued",v.buffer);
				return -EFAULT;
			}
		}
	default:
		return -EINVAL;
	}
}

/*
*	This maps the vmalloced and reserved buffer to user space.
*
*  FIXME: 
*  - PAGE_READONLY should suffice!?
*  - remap_page_range is kind of inefficient for page by page remapping.
*    But e.g. pte_alloc() does not work in modules ... :-(
*/

int video1394_mmap(struct file *file, struct vm_area_struct *vma)
{
	struct file_ctx *ctx = (struct file_ctx *)file->private_data;
	struct video_card *video = ctx->video;
	struct ti_ohci *ohci = video->ohci;
	int res = -EINVAL;
	
	lock_kernel();
	ohci = video->ohci;
	
	if (ctx->current_ctx == NULL) {
		PRINT(KERN_ERR, ohci->id, "Current iso context not set");
	} else
		res = dma_region_mmap(&ctx->current_ctx->dma, file, vma);
	unlock_kernel();
	return res;
}

static int video1394_open(struct inode *inode, struct file *file)
{
	int i = ieee1394_file_to_instance(file);
	struct video_card *video;
	struct file_ctx *ctx;
	
	video = hpsb_get_hostinfo_bykey(&video1394_highlevel, i);
	if (video == NULL)
		return -EIO;
	
	ctx = kmalloc(sizeof(struct file_ctx), GFP_KERNEL);
	if (ctx == NULL)  {
		PRINT(KERN_ERR, video->ohci->id, "Cannot malloc file_ctx");
		return -ENOMEM;
	}
	
	memset(ctx, 0, sizeof(struct file_ctx));
	ctx->video = video;
	INIT_LIST_HEAD(&ctx->context_list);
	ctx->current_ctx = NULL;
	file->private_data = ctx;
	
	return 0;
}

static int video1394_release(struct inode *inode, struct file *file)
{
	struct file_ctx *ctx = (struct file_ctx *)file->private_data;
	struct video_card *video = ctx->video;
	struct ti_ohci *ohci = video->ohci;
	struct list_head *lh, *next;
	u64 mask;
	
	lock_kernel();
	list_for_each_safe(lh, next, &ctx->context_list) {
		struct dma_iso_ctx *d;
		d = list_entry(lh, struct dma_iso_ctx, link);
		mask = (u64) 1 << d->channel;
		
		if (!(ohci->ISO_channel_usage & mask))
			PRINT(KERN_ERR, ohci->id, "On release: Channel %d "
			"is not being used", d->channel);
		else
			ohci->ISO_channel_usage &= ~mask;
		PRINT(KERN_INFO, ohci->id, "On release: Iso %s context "
			"%d stop listening on channel %d",
			d->type == OHCI_ISO_RECEIVE ? "receive" : "transmit",
			d->ctx, d->channel);
		free_dma_iso_ctx(d);
	}
	
	kfree(ctx);
	file->private_data = NULL;
	
	unlock_kernel();
	return 0;
}

static struct file_operations video1394_fops=
{
	.owner =	THIS_MODULE,
		.ioctl =	video1394_ioctl,
		.mmap =		video1394_mmap,
		.open =		video1394_open,
		.release =	video1394_release
};

/*** HOTPLUG STUFF **********************************************************/
/*
* Export information about protocols/devices supported by this driver.
*/
//#define IEEE1394_MATCH_SPECIFIER_ID	0x0004
//#define IEEE1394_MATCH_VERSION			0x0008
//#define CAMERA_UNIT_SPEC_ID_ENTRY		0x0000A02D
//#define CAMERA_SW_VERSION_ENTRY			0x00000100
static struct ieee1394_device_id video1394_id_table[] = {
	{
		.match_flags	= IEEE1394_MATCH_SPECIFIER_ID | IEEE1394_MATCH_VERSION,
			.specifier_id	= CAMERA_UNIT_SPEC_ID_ENTRY & 0xffffff,
			.version	= CAMERA_SW_VERSION_ENTRY & 0xffffff
	},
	{ }
};

MODULE_DEVICE_TABLE(ieee1394, video1394_id_table);

static struct hpsb_protocol_driver video1394_driver = {
	.name		= "1394 Digital Camera Driver",
		.id_table	= video1394_id_table,
};


static void video1394_add_host (struct hpsb_host *host)
{
	struct ti_ohci *ohci;
	struct video_card *video;
	char name[16];
	int minor;
	
	/* We only work with the OHCI-1394 driver */
	if (strcmp(host->driver->name, OHCI1394_DRIVER_NAME))
		return;
	
	ohci = (struct ti_ohci *)host->hostdata;
	
	video = hpsb_create_hostinfo(&video1394_highlevel, host, sizeof(*video));
	if (video == NULL) {
		PRINT(KERN_ERR, ohci->id, "Cannot allocate video_card");
		return;
	}
	
	hpsb_set_hostinfo_key(&video1394_highlevel, host, ohci->id);
	video->ohci = ohci;
	
	sprintf(name, "%d", ohci->id);
	minor = IEEE1394_MINOR_BLOCK_VIDEO1394 * 16 + ohci->id;
	video->devfs = devfs_register(devfs_handle, name,
		DEVFS_FL_AUTO_OWNER,
		IEEE1394_MAJOR, minor,
		S_IFCHR | S_IRUSR | S_IWUSR,
		&video1394_fops, NULL);
	
	return;
}


static void video1394_remove_host (struct hpsb_host *host)
{
	struct video_card *video = hpsb_get_hostinfo(&video1394_highlevel, host);
	
	if (video)
		devfs_unregister(video->devfs);
	
	return;
}


static struct hpsb_highlevel video1394_highlevel = {
	.name =		VIDEO1394_DRIVER_NAME,
		.add_host =	video1394_add_host,
		.remove_host =	video1394_remove_host,
};

MODULE_AUTHOR("Sebastien Rougeaux <[e-mail]>");
MODULE_DESCRIPTION("driver for digital video on OHCI board");
MODULE_SUPPORTED_DEVICE(VIDEO1394_DRIVER_NAME);
MODULE_LICENSE("GPL");

static void __exit video1394_exit_module (void)
{
	hpsb_unregister_protocol(&video1394_driver);
	
	hpsb_unregister_highlevel(&video1394_highlevel);
	
	devfs_unregister(devfs_handle);
	ieee1394_unregister_chardev(IEEE1394_MINOR_BLOCK_VIDEO1394);
	
	PRINT_G(KERN_INFO, "Removed " VIDEO1394_DRIVER_NAME " module");
}

static int __init video1394_init_module (void)
{
	if (ieee1394_register_chardev(IEEE1394_MINOR_BLOCK_VIDEO1394,
		THIS_MODULE, &video1394_fops)) {
		PRINT_G(KERN_ERR, "video1394: unable to get minor device block");
		return -EIO;
	}
	
	devfs_handle = devfs_mk_dir(NULL, VIDEO1394_DRIVER_NAME, NULL);
	
	hpsb_register_highlevel(&video1394_highlevel);
	
	hpsb_register_protocol(&video1394_driver);
	
	PRINT_G(KERN_INFO, "Installed " VIDEO1394_DRIVER_NAME " module");
	return 0;
}

module_init(video1394_init_module);
module_exit(video1394_exit_module);