Anonymous

#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/module.h>
#include <linux/config.h>
#include <linux/init.h>
#include <linux/fs.h>
#include "vhd.h"
#include "vhdheader.h"
#include <asm/semaphore.h>

struct file_operations fileOp;
struct vhd_block *rBlock;
struct vhd_block *wBlock;
struct semaphore sem0;
struct semaphore sem1;

int ioctl_function(struct inode *ip, struct file *fp, unsigned int cmd, unsigned long param) {
	struct vhd_header *userSpace  = (struct vhd_header *)kmalloc(sizeof(struct vhd_header), GFP_KERNEL);

	printk("ioctl function called\n");

	copy_from_user(userSpace, (struct vhd_header *)param, sizeof(struct vhd_header));
	down(&sem1);
	printk("copied ok\n");
	if (cmd == VHD_WRITE) {
		printk("WriteIOCTL\n");
		wBlock -> block_number = userSpace->block_number;
		wBlock -> operation = VHD_WRITE;
		memcpy(wBlock->data, userSpace->data, 64);

	printk(">>>%s and %s\n", wBlock->data, userSpace->data);

		vhd_request_operation(wBlock);

	}

	if (cmd == VHD_READ) {
		printk("ReadIOCTL\n");
		rBlock -> block_number = userSpace->block_number;
		rBlock -> operation = VHD_READ;
		vhd_request_operation(rBlock);
		down(&sem0);
		printk("IOCTL read data: %s\n", rBlock->data);
		copy_to_user((struct vhd_header *)param, rBlock, sizeof(struct vhd_header));

	}


	kfree(userSpace);
	printk("end of ioctl\n");

}

void callBack(struct vhd_block *block) {

	if (block->operation == VHD_READ) {
		printk("Data read: %s\n", block->data);
		up(&sem0);
	}
	else if (block->operation == VHD_WRITE) {
		printk("Data: %s - Written to: %d\n", block->data, block->block_number);
	}

	kfree(block);
	up(&sem1);

}

int init(void)
{

	memset(&fileOp, 0, sizeof (struct file_operations));

	sema_init(&sem0, 0);
	sema_init(&sem1, 1);

	fileOp.ioctl = ioctl_function;
	register_chrdev (241, "VHD", &fileOp);

	rBlock = (struct vhd_block *)kmalloc(sizeof(struct vhd_block), GFP_KERNEL);
	wBlock = (struct vhd_block *)kmalloc(sizeof(struct vhd_block), GFP_KERNEL);
	vhd_init();

	vhd_register_interrupt(callBack);

	return 0;
}

void cleanup(void)
{
	unregister_chrdev(241, "VHD");
	kfree(rBlock);
	kfree(wBlock);
	vhd_cleanup();
}


module_init(init);
module_exit(cleanup);
MODULE_LICENSE("GPL");