On Fri, May 25, 2001 at 09:47:34AM +0200, Alessandro Ronchi wrote:
> 
> Can i have the corrected file? You can send it to my address, if you don't 
> want to post in the mailing list.

Here it is.  I need to still fix the things in the TODO section, add
more documentation, and actually test the thing :)

thanks,

greg k-h
/*
 * USB Skeleton driver
 *
 * Copyright (c) 2001 Greg Kroah-Hartman ([EMAIL PROTECTED])
 *
 *      This program is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU General Public License as
 *      published by the Free Software Foundation; either version 2 of
 *      the License, or (at your option) any later version.
 *
 *
 * This driver is to be used as a skeleton driver to be able to create a
 * USB driver quickly.  The design of it is based on the usb-serial and
 * dc2xx drivers.
 *
 * TODO:
 *      - fix urb->status race condition in write sequence
 *      - move minor_table to a dynamic list.
 *
 * History:
 *
 * 2001_05_24 - 0.2 - bug fixes based on review from linux-usb-devel people
 * 2001_05_01 - 0.1 - first version
 * 
 */

#include <linux/config.h>
#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/signal.h>
#include <linux/errno.h>
#include <linux/poll.h>
#include <linux/init.h>
#include <linux/slab.h>
#include <linux/fcntl.h>
#include <linux/module.h>
#include <linux/spinlock.h>
#include <linux/list.h>
#include <linux/smp_lock.h>
#include <linux/devfs_fs_kernel.h>
#include <linux/usb.h>

#ifdef CONFIG_USB_DEBUG
        static int debug = 1;
#else
        static int debug;
#endif

/* Use our own dbg macro */
#undef dbg
#define dbg(format, arg...) do { if (debug) printk(KERN_DEBUG __FILE__ ": " format 
"\n" , ## arg); } while (0)


/* Version Information */
#define DRIVER_VERSION "v0.1"
#define DRIVER_AUTHOR "Greg Kroah-Hartman, [EMAIL PROTECTED]"
#define DRIVER_DESC "USB Skeleton Driver"

/* Module paramaters */
MODULE_PARM(debug, "i");
MODULE_PARM_DESC(debug, "Debug enabled or not");


/* Define these values to match your device */
#define USB_SKEL_VENDOR_ID      0x082d
#define USB_SKEL_PRODUCT_ID     0x0100

/* table of devices that work with this driver */
static struct usb_device_id skel_table [] = {
        { USB_DEVICE(USB_SKEL_VENDOR_ID, USB_SKEL_PRODUCT_ID) },
        { }                                     /* Terminating entry */
};

MODULE_DEVICE_TABLE (usb, skel_table);



/* Get a minor range for your devices from the usb maintainer */
#define USB_SKEL_MINOR_BASE     200     

/* we can have up to this number of device plugged in at once */
#define MAX_DEVICES             16

/* Structure to hold all of our device specific stuff */
struct usb_skel {
        struct usb_device *             dev;                    /* save off the usb 
device pointer */
        struct usb_interface *          interface;              /* the interface for 
this device */
        devfs_handle_t                  devfs;                  /* devfs device node */
        unsigned char                   minor;                  /* the starting minor 
number for this device */
        unsigned char                   num_ports;              /* the number of ports 
this device has */
        char                            num_interrupt_in;       /* number of interrupt 
in endpoints we have */
        char                            num_bulk_in;            /* number of bulk in 
endpoints we have */
        char                            num_bulk_out;           /* number of bulk out 
endpoints we have */

        unsigned char *                 bulk_in_buffer;         /* the buffer to 
receive data */
        int                             bulk_in_size;           /* the size of the 
receive buffer */
        __u8                            bulk_in_endpointAddr;   /* the address of the 
bulk in endpoint */

        unsigned char *                 bulk_out_buffer;        /* the buffer to send 
data */
        int                             bulk_out_size;          /* the size of the 
send buffer */
        struct urb *                    write_urb;              /* the urb used to 
send data */
        __u8                            bulk_out_endpointAddr;  /* the address of the 
bulk out endpoint */

        struct tq_struct                tqueue;                 /* task queue for line 
discipline waking up */
        int                             open_count;             /* number of times 
this port has been opened */
        struct semaphore                sem;                    /* locks this 
structure */
};


/* the global usb devfs handle */
extern devfs_handle_t usb_devfs_handle;


/* local function prototypes */
static ssize_t skel_read        (struct file *file, char *buffer, size_t count, loff_t 
*ppos);
static ssize_t skel_write       (struct file *file, const char *buffer, size_t count, 
loff_t *ppos);
static int skel_ioctl           (struct inode *inode, struct file *file, unsigned int 
cmd, unsigned long arg);
static int skel_open            (struct inode *inode, struct file *file);
static int skel_release         (struct inode *inode, struct file *file);
        
static void * skel_probe        (struct usb_device *dev, unsigned int ifnum, const 
struct usb_device_id *id);
static void skel_disconnect     (struct usb_device *dev, void *ptr);

static void skel_write_bulk_callback    (struct urb *urb);


/* array of pointers to our devices that are currently connected */
static struct usb_skel          *minor_table[MAX_DEVICES];

/* lock to protect the minor_table structure */
static DECLARE_MUTEX (minor_table_mutex);

/* file operations needed when we register this driver */
static struct file_operations skel_fops = {
        owner:          THIS_MODULE,
        read:           skel_read,
        write:          skel_write,
        ioctl:          skel_ioctl,
        open:           skel_open,
        release:        skel_release,
};      


/* usb specific object needed to register this driver with the usb subsystem */
static struct usb_driver skel_driver = {
        name:           "skeleton",
        probe:          skel_probe,
        disconnect:     skel_disconnect,
        fops:           &skel_fops,
        minor:          USB_SKEL_MINOR_BASE,
        id_table:       skel_table,
};




static inline struct usb_skel *get_serial_by_minor (int minor)
{
        return minor_table[minor-USB_SKEL_MINOR_BASE];
}



/**
 *      usb_skel_debug_data
 */
static inline void usb_skel_debug_data (const char *function, int size, const unsigned 
char *data)
{
        int i;

        if (!debug)
                return;
        
        printk (KERN_DEBUG __FILE__": %s - length = %d, data = ", function, size);
        for (i = 0; i < size; ++i) {
                printk ("%.2x ", data[i]);
        }
        printk ("\n");
}


/**
 *      skel_open
 */
static int skel_open (struct inode *inode, struct file *file)
{
        struct usb_skel *skel = NULL;
        int subminor;
        int retval = 0;
        
        dbg(__FUNCTION__);

        subminor = MINOR (inode->i_rdev) - USB_SKEL_MINOR_BASE;
        if ((subminor < 0) ||
            (subminor >= MAX_DEVICES)) {
                return -ENODEV;
        }

        /* lock our minor table and get our local data for this minor */
        down (&minor_table_mutex);
        skel = minor_table[subminor];
        if (skel == NULL) {
                up (&minor_table_mutex);
                return -ENODEV;
        }

        /* increment our usage count for the module */
        MOD_INC_USE_COUNT;

        /* lock this device */
        down (&skel->sem);

        /* unlock the minor table */
        up (&minor_table_mutex);

        /* increment our usage count for the driver */
        ++skel->open_count;

        /* save our object in the file's private structure */
        file->private_data = skel;

        /* unlock this device */
        up (&skel->sem);

        return retval;
}


/**
 *      skel_release
 */
static int skel_release (struct inode *inode, struct file *file)
{
        struct usb_skel *skel;
        int retval = 0;

        skel = (struct usb_skel *)file->private_data;
        if (skel == NULL) {
                dbg (__FUNCTION__ " - object is NULL");
                return -ENODEV;
        }

        dbg(__FUNCTION__ " - minor %d", skel->minor);

        /* lock our minor table */
        down (&minor_table_mutex);

        /* lock our device */
        down (&skel->sem);

        if (skel->open_count <= 0) {
                dbg (__FUNCTION__ " - device not opened");
                retval = -ENODEV;
                goto exit_not_opened;
        }

        if (skel->dev == NULL) {
                /* the device was unplugged before the file was released */
                minor_table[skel->minor] = NULL;
                if (skel->bulk_in_buffer != NULL)
                        kfree (skel->bulk_in_buffer);
                if (skel->bulk_out_buffer != NULL)
                        kfree (skel->bulk_out_buffer);
                if (skel->write_urb != NULL)
                        usb_free_urb (skel->write_urb);
                kfree (skel);
                goto exit;
        }

        /* decrement our usage count for the device */
        --skel->open_count;
        if (skel->open_count <= 0) {
                /* shutdown any bulk writes that might be going on */
                usb_unlink_urb (skel->write_urb);
                skel->open_count = 0;
        }

        up (&skel->sem);

exit:
        /* decrement our usage count for the module */
        MOD_DEC_USE_COUNT;

exit_not_opened:
        up (&minor_table_mutex);

        return retval;
}


/**
 *      skel_read
 */
static ssize_t skel_read (struct file *file, char *buffer, size_t count, loff_t *ppos)
{
        struct usb_skel *skel;
        int retval = 0;

        skel = (struct usb_skel *)file->private_data;
        
        dbg(__FUNCTION__ " - minor %d, count = %d", skel->minor, count);

        /* lock this object */
        down (&skel->sem);

        /* verify that the device wasn't unplugged */
        if (skel->dev == NULL) {
                up (&skel->sem);
                return -ENODEV;
        }
        
        /* do an immediate bulk read to get data from the device */
        retval = usb_bulk_msg (skel->dev,
                               usb_rcvbulkpipe (skel->dev, skel->bulk_in_endpointAddr),
                               skel->bulk_in_buffer, skel->bulk_in_size,
                               &count, HZ*10);

        /* if the read was successful, copy the data to userspace */
        if (!retval) {
                if (copy_to_user (buffer, skel->bulk_in_buffer, count))
                        retval = -EFAULT;
                else
                        retval = count;
        }
        
        /* unlock the device */
        up (&skel->sem);
        return retval;
}


/**
 *      skel_write
 */
static ssize_t skel_write (struct file *file, const char *buffer, size_t count, loff_t 
*ppos)
{
        struct usb_skel *skel;
        ssize_t bytes_written = 0;
        int retval = 0;

        skel = (struct usb_skel *)file->private_data;

        dbg(__FUNCTION__ " - minor %d, count = %d", skel->minor, count);

        /* lock this object */
        down (&skel->sem);

        /* verify that the device wasn't unplugged */
        if (skel->dev == NULL) {
                retval = -ENODEV;
                goto exit;
        }

        /* verify that we actually have some data to write */
        if (count == 0) {
                dbg(__FUNCTION__ " - write request of 0 bytes");
                goto exit;
        }

        /* see if we are already in the middle of a write */
        if (skel->write_urb->status == -EINPROGRESS) {
                dbg (__FUNCTION__ " - already writing");
                goto exit;
        }

        /* we can only write as much as 1 urb will hold */
        bytes_written = (count > skel->bulk_out_size) ? skel->bulk_out_size : count;

        /* copy the data from userspace into our urb */
        if (copy_from_user(skel->write_urb->transfer_buffer, buffer, bytes_written)) {
                retval = -EFAULT;
                goto exit;
        }

        usb_skel_debug_data (__FUNCTION__, bytes_written, 
skel->write_urb->transfer_buffer);

        /* set up our urb */
        FILL_BULK_URB(skel->write_urb, skel->dev, 
                      usb_sndbulkpipe(skel->dev, skel->bulk_out_endpointAddr),
                      skel->write_urb->transfer_buffer, bytes_written,
                      skel_write_bulk_callback, skel);

        /* send the data out the bulk port */
        retval = usb_submit_urb(skel->write_urb);
        if (retval) {
                err(__FUNCTION__ " - failed submitting write urb, error %d", retval);
        } else {
                retval = bytes_written;
        }

exit:
        /* unlock the device */
        up (&skel->sem);

        return retval;
}


/**
 *      skel_ioctl
 */
static int skel_ioctl (struct inode *inode, struct file *file, unsigned int cmd, 
unsigned long arg)
{
        struct usb_skel *skel;

        skel = (struct usb_skel *)file->private_data;

        /* lock this object */
        down (&skel->sem);

        /* verify that the device wasn't unplugged */
        if (skel->dev == NULL) {
                up (&skel->sem);
                return -ENODEV;
        }

        dbg(__FUNCTION__ " - minor %d, cmd 0x%.4x, arg %ld", skel->minor, cmd, arg);


        /* fill in your device specific stuff here */
        
        /* unlock the device */
        up (&skel->sem);
        
        /* return that we did not understand this ioctl call */
        return -ENOTTY;
}


/**
 *      skel_write_bulk_callback
 */
static void skel_write_bulk_callback (struct urb *urb)
{
        struct usb_skel *skel = (struct usb_skel *)urb->context;

        dbg(__FUNCTION__ " - minor %d", skel->minor);

        if (urb->status) {
                dbg(__FUNCTION__ " - nonzero write bulk status received: %d", 
urb->status);
                return;
        }

        return;
}





static void * skel_probe(struct usb_device *dev, unsigned int ifnum, const struct 
usb_device_id *id)
{
        struct usb_skel *skel = NULL;
        struct usb_interface *interface;
        struct usb_interface_descriptor *iface_desc;
        struct usb_endpoint_descriptor *endpoint;
        int minor;
        int buffer_size;
        int i;
        char name[10];

        
        /* See if the device offered us matches what we can accept */
        if ((dev->descriptor.idVendor != USB_SKEL_VENDOR_ID) ||
            (dev->descriptor.idProduct != USB_SKEL_PRODUCT_ID)) {
                return NULL;
        }

        /* select a "subminor" number (part of a minor number) */
        down (&minor_table_mutex);
        for (minor = 0; minor < MAX_DEVICES; i++) {
                if (minor_table[minor] != NULL)
                        break;
        }
        if (minor >= MAX_DEVICES) {
                info ("Too many devices plugged in, can not handle this device.");
                goto exit;
        }

        /* allocate memory for our device state and intialize it */
        skel = kmalloc (sizeof(struct usb_skel), GFP_KERNEL);
        if (skel == NULL) {
                err ("Out of memory");
                goto exit;
        }
        minor_table[minor] = skel;

        interface = &dev->actconfig->interface[ifnum];

        init_MUTEX (&skel->sem);
        skel->dev = dev;
        skel->interface = interface;
        skel->minor = minor;

        /* set up the endpoint information */
        /* check out the endpoints */
        iface_desc = &interface->altsetting[0];
        for (i = 0; i < iface_desc->bNumEndpoints; ++i) {
                endpoint = &iface_desc->endpoint[i];

                if ((endpoint->bEndpointAddress & 0x80) &&
                    ((endpoint->bmAttributes & 3) == 0x02)) {
                        /* we found a bulk in endpoint */
                        buffer_size = endpoint->wMaxPacketSize;
                        skel->bulk_in_size = buffer_size;
                        skel->bulk_in_endpointAddr = endpoint->bEndpointAddress;
                        skel->bulk_in_buffer = kmalloc (buffer_size, GFP_KERNEL);
                        if (!skel->bulk_in_buffer) {
                                err("Couldn't allocate bulk_in_buffer");
                                goto error;
                        }
                }
                
                if (((endpoint->bEndpointAddress & 0x80) == 0x00) &&
                    ((endpoint->bmAttributes & 3) == 0x02)) {
                        /* we found a bulk out endpoint */
                        skel->write_urb = usb_alloc_urb(0);
                        if (!skel->write_urb) {
                                err("No free urbs available");
                                goto error;
                        }
                        buffer_size = endpoint->wMaxPacketSize;
                        skel->bulk_out_size = buffer_size;
                        skel->bulk_out_endpointAddr = endpoint->bEndpointAddress;
                        skel->bulk_out_buffer = kmalloc (buffer_size, GFP_KERNEL);
                        if (!skel->bulk_out_buffer) {
                                err("Couldn't allocate bulk_out_buffer");
                                goto error;
                        }
                        FILL_BULK_URB(skel->write_urb, dev, 
                                      usb_sndbulkpipe(dev, endpoint->bEndpointAddress),
                                      skel->bulk_out_buffer, buffer_size,
                                      skel_write_bulk_callback, skel);
                }
        }

        /* initialize the devfs node for this device and register it */
        sprintf(name, "skel%d", skel->minor);
        
        skel->devfs = devfs_register (usb_devfs_handle, name,
                                      DEVFS_FL_DEFAULT, USB_MAJOR,
                                      USB_SKEL_MINOR_BASE + skel->minor,
                                      S_IFCHR | S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP 
| S_IROTH, 
                                      &skel_fops, NULL);

        /* let the user know what node this device is now attached to */
        info ("USB Skeleton device now attached to USBSkel%d", skel->minor);
        goto exit;
        
error:
        minor_table [skel->minor] = NULL;
        if (skel->bulk_in_buffer != NULL)
                kfree (skel->bulk_in_buffer);
        if (skel->bulk_out_buffer != NULL)
                kfree (skel->bulk_out_buffer);
        if (skel->write_urb != NULL)
                usb_free_urb (skel->write_urb);
        kfree (skel);
        skel = NULL;

exit:
        up (&minor_table_mutex);
        return skel;
}


/**
 *      skel_disconnect
 */
static void skel_disconnect(struct usb_device *dev, void *ptr)
{
        struct usb_skel *skel;
        int minor;

        skel = (struct usb_skel *)ptr;
        
        down (&minor_table_mutex);
        down (&skel->sem);
                
        minor = skel->minor;

        /* if the device is not opened, then we clean up right now */
        if (skel->open_count) {
                minor_table[skel->minor] = NULL;
                if (skel->bulk_in_buffer != NULL)
                        kfree (skel->bulk_in_buffer);
                if (skel->bulk_out_buffer != NULL)
                        kfree (skel->bulk_out_buffer);
                if (skel->write_urb != NULL)
                        usb_free_urb (skel->write_urb);
                kfree (skel);
        } else {
                skel->dev = NULL;
                up (&skel->sem);
        }

        /* remove our devfs node */
        devfs_unregister(skel->devfs);

        info("USB Skeleton #%d now disconnected", minor);
        up (&minor_table_mutex);
}



/**
 *      usb_skel_init
 */
static int __init usb_skel_init(void)
{
        int result;

        /* register this driver with the USB subsystem */
        result = usb_register(&skel_driver);
        if (result < 0) {
                err("usb_register failed for the "__FILE__" driver. Error number %d", 
result);
                return -1;
        }

        info(DRIVER_VERSION " " DRIVER_AUTHOR);
        info(DRIVER_DESC);
        return 0;
}


/**
 *      usb_skel_exit
 */
static void __exit usb_skel_exit(void)
{
        /* deregister this driver with the USB subsystem */
        usb_deregister(&skel_driver);
}


module_init (usb_skel_init);
module_exit (usb_skel_exit);

MODULE_AUTHOR(DRIVER_AUTHOR);
MODULE_DESCRIPTION(DRIVER_DESC);

Reply via email to