/*
 * Wireless USB - Cable Based Association
 *
 *
 * Copyright (C) 2006 Intel Corporation
 * Inaky Perez-Gonzalez <inaky.perez-gonzalez@intel.com>
 * Copyright (C) 2008 Cambridge Silicon Radio Ltd.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 *
 *
 * WUSB devices have to be paired (associated in WUSB lingo) so
 * that they can connect to the system.
 *
 * One way of pairing is using CBA-Cable Based Association. First
 * time you plug the device with a cable, association is done between
 * host and device and subsequent times, you can connect wirelessly
 * without having to associate again. That's the idea.
 *
 * This driver does nothing Earth shattering. It just provides an
 * interface to chat with the wire-connected device so we can get a
 * CDID (device ID) that might have been previously associated to a
 * CHID (host ID) and to set up a new <CHID,CDID,CK> triplet
 * (connection context), with the CK being the secret, or connection
 * key. This is the pairing data.
 *
 * When a device with the CBA capability connects, the probe routine
 * just creates a bunch of sysfs files that a user space enumeration
 * manager uses to allow it to connect wirelessly to the system or not.
 *
 * The process goes like this:
 *
 * 1. Device plugs, cbaf is loaded, notifications happen.
 *
 * 2. The connection manager (CM) sees a device with CBAF capability
 *    (the wusb_chid etc. files in /sys/devices/blah/OURDEVICE).
 *
 * 3. The CM writes the host name, supported band groups, and the CHID
 *    (host ID) into the wusb_host_name, wusb_host_band_groups and
 *    wusb_chid files. These get sent to the device and the CDID (if
 *    any) for this host is requested.
 *
 * 4. The CM can verify that the device's supported band groups
 *    (wusb_device_band_groups) are compatible with the host.
 *
 * 5. The CM reads the wusb_cdid file.
 *
 * 6. The CM looks up its database
 *
 * 6.1 If it has a matching CHID,CDID entry, the device has been
 *     authorized before (paired) and nothing further needs to be
 *     done.
 *
 * 6.2 If the CDID is zero (or the CM doesn't find a matching CDID in
 *     its database), the device is assumed to be not known.  The CM
 *     may associate the host with device by: writing a randomly
 *     generated CDID to wusb_cdid and then a random CK to wusb_ck
 *     (this uploads the new CC to the device).
 *
 *     CMD may choose to prompt the user before associating with a new
 *     device.
 *
 * 7. Device is unplugged.
 *
 * When the device tries to connect wirelessly, it will present its
 * CDID to the WUSB host controller.  The CM will query the
 * database. If the CHID/CDID pair found, it will (with a 4-way
 * handshake) challenge the device to demonstrate it has the CK secret
 * key (from our database) without actually exchanging it. Once
 * satisfied, crypto keys are derived from the CK, the device is
 * connected and all communication is encrypted.
 *
 * References:
 *   [WUSB-AM] Association Models Supplement to the Certified Wireless
 *             Universal Serial Bus Specification, version 1.0.
 */
#include <linux/module.h>
#include <linux/ctype.h>
#include <linux/usb.h>
#include <linux/interrupt.h>
#include <linux/delay.h>
#include <linux/random.h>
#include <linux/mutex.h>
#include <linux/uwb.h>
#include <linux/usb/wusb.h>
#include <linux/usb/association.h>

#define CBA_NAME_LEN 0x40 /* [WUSB-AM] table 4-7 */

/* An instance of a Cable-Based-Association-Framework device */
struct cbaf {
        struct usb_device *usb_dev;
        struct usb_interface *usb_iface;
        void *buffer;
        size_t buffer_size;

        struct wusb_ckhdid chid;
        char host_name[CBA_NAME_LEN];
        u16 host_band_groups;

        struct wusb_ckhdid cdid;
        char device_name[CBA_NAME_LEN];
        u16 device_band_groups;

        struct wusb_ckhdid ck;
};

/*
 * Verify that a CBAF USB-interface has what we need
 *
 * According to [WUSB-AM], CBA devices should provide at least two
 * interfaces:
 *  - RETRIEVE_HOST_INFO
 *  - ASSOCIATE
 *
 * If the device doesn't provide these interfaces, we do not know how
 * to deal with it.
 */
static int cbaf_check(struct cbaf *cbaf)
{
        int result;
        struct device *dev = &cbaf->usb_iface->dev;
        struct wusb_cbaf_assoc_info *assoc_info;
        struct wusb_cbaf_assoc_request *assoc_request;
        size_t assoc_size;
        void *itr, *top;
        int ar_rhi = 0, ar_assoc = 0;

        result = usb_control_msg(
                cbaf->usb_dev, usb_rcvctrlpipe(cbaf->usb_dev, 0),
                CBAF_REQ_GET_ASSOCIATION_INFORMATION,
                USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
                0, cbaf->usb_iface->cur_altsetting->desc.bInterfaceNumber,
                cbaf->buffer, cbaf->buffer_size, 1000 /* FIXME: arbitrary */);
        if (result < 0) {
                dev_err(dev, "Cannot get available association types: %d\n",
                        result);
                return result;
        }

        assoc_info = cbaf->buffer;
        if (result < sizeof(*assoc_info)) {
                dev_err(dev, "Not enough data to decode association info "
                        "header (%zu vs %zu bytes required)\n",
                        (size_t)result, sizeof(*assoc_info));
                return result;
        }

        assoc_size = le16_to_cpu(assoc_info->Length);
        if (result < assoc_size) {
                dev_err(dev, "Not enough data to decode association info "
                        "(%zu vs %zu bytes required)\n",
                        (size_t)assoc_size, sizeof(*assoc_info));
                return result;
        }
        /*
         * From now on, we just verify, but won't error out unless we
         * don't find the AR_TYPE_WUSB_{RETRIEVE_HOST_INFO,ASSOCIATE}
         * types.
         */
        itr = cbaf->buffer + sizeof(*assoc_info);
        top = cbaf->buffer + assoc_size;
        dev_dbg(dev, "Found %u association requests (%zu bytes)\n",
                 assoc_info->NumAssociationRequests, assoc_size);

        while (itr < top) {
                u16 ar_type, ar_subtype;
                u32 ar_size;
                const char *ar_name;

                assoc_request = itr;

                if (top - itr < sizeof(*assoc_request)) {
                        dev_err(dev, "Not enough data to decode associaton "
                                "request (%zu vs %zu bytes needed)\n",
                                top - itr, sizeof(*assoc_request));
                        break;
                }

                ar_type = le16_to_cpu(assoc_request->AssociationTypeId);
                ar_subtype = le16_to_cpu(assoc_request->AssociationSubTypeId);
                ar_size = le32_to_cpu(assoc_request->AssociationTypeInfoSize);
                ar_name = "unknown";

                switch (ar_type) {
                case AR_TYPE_WUSB:
                        /* Verify we have what is mandated by [WUSB-AM]. */
                        switch (ar_subtype) {
                        case AR_TYPE_WUSB_RETRIEVE_HOST_INFO:
                                ar_name = "RETRIEVE_HOST_INFO";
                                ar_rhi = 1;
                                break;
                        case AR_TYPE_WUSB_ASSOCIATE:
                                /* send assoc data */
                                ar_name = "ASSOCIATE";
                                ar_assoc = 1;
                                break;
                        };
                        break;
                };

                dev_dbg(dev, "Association request #%02u: 0x%04x/%04x "
                         "(%zu bytes): %s\n",
                         assoc_request->AssociationDataIndex, ar_type,
                         ar_subtype, (size_t)ar_size, ar_name);

                itr += sizeof(*assoc_request);
        }

        if (!ar_rhi) {
                dev_err(dev, "Missing RETRIEVE_HOST_INFO association "
                        "request\n");
                return -EINVAL;
        }
        if (!ar_assoc) {
                dev_err(dev, "Missing ASSOCIATE association request\n");
                return -EINVAL;
        }

        return 0;
}

static const struct wusb_cbaf_host_info cbaf_host_info_defaults = {
        .AssociationTypeId_hdr    = WUSB_AR_AssociationTypeId,
        .AssociationTypeId        = cpu_to_le16(AR_TYPE_WUSB),
        .AssociationSubTypeId_hdr = WUSB_AR_AssociationSubTypeId,
        .AssociationSubTypeId = cpu_to_le16(AR_TYPE_WUSB_RETRIEVE_HOST_INFO),
        .CHID_hdr                 = WUSB_AR_CHID,
        .LangID_hdr               = WUSB_AR_LangID,
        .HostFriendlyName_hdr     = WUSB_AR_HostFriendlyName,
};

/* Send WUSB host information (CHID and name) to a CBAF device */
static int cbaf_send_host_info(struct cbaf *cbaf)
{
        struct wusb_cbaf_host_info *hi;
        size_t name_len;
        size_t hi_size;

        hi = cbaf->buffer;
        memset(hi, 0, sizeof(*hi));
        *hi = cbaf_host_info_defaults;
        hi->CHID = cbaf->chid;
        hi->LangID = 0; /* FIXME: I guess... */
        strlcpy(hi->HostFriendlyName, cbaf->host_name, CBA_NAME_LEN);
        name_len = strlen(cbaf->host_name);
        hi->HostFriendlyName_hdr.len = cpu_to_le16(name_len);
        hi_size = sizeof(*hi) + name_len;

        return usb_control_msg(cbaf->usb_dev, usb_sndctrlpipe(cbaf->usb_dev, 0),
                        CBAF_REQ_SET_ASSOCIATION_RESPONSE,
                        USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
                        0x0101,
                        cbaf->usb_iface->cur_altsetting->desc.bInterfaceNumber,
                        hi, hi_size, 1000 /* FIXME: arbitrary */);
}

/*
 * Get device's information (CDID) associated to CHID
 *
 * The device will return it's information (CDID, name, bandgroups)
 * associated to the CHID we have set before, or 0 CDID and default
 * name and bandgroup if no CHID set or unknown.
 */
static int cbaf_cdid_get(struct cbaf *cbaf)
{
        int result;
        struct device *dev = &cbaf->usb_iface->dev;
        struct wusb_cbaf_device_info *di;
        size_t needed;

        di = cbaf->buffer;
        result = usb_control_msg(
                cbaf->usb_dev, usb_rcvctrlpipe(cbaf->usb_dev, 0),
                CBAF_REQ_GET_ASSOCIATION_REQUEST,
                USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
                0x0200, cbaf->usb_iface->cur_altsetting->desc.bInterfaceNumber,
                di, cbaf->buffer_size, 1000 /* FIXME: arbitrary */);
        if (result < 0) {
                dev_err(dev, "Cannot request device information: %d\n", result);
                return result;
        }

        needed = result < sizeof(*di) ? sizeof(*di) : le32_to_cpu(di->Length);
        if (result < needed) {
                dev_err(dev, "Not enough data in DEVICE_INFO reply (%zu vs "
                        "%zu bytes needed)\n", (size_t)result, needed);
                return result;
        }

        strlcpy(cbaf->device_name, di->DeviceFriendlyName, CBA_NAME_LEN);
        cbaf->cdid = di->CDID;
        cbaf->device_band_groups = le16_to_cpu(di->BandGroups);

        return 0;
}

static ssize_t cbaf_wusb_chid_show(struct device *dev,
                                        struct device_attribute *attr,
                                        char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);
        char pr_chid[WUSB_CKHDID_STRSIZE];

        ckhdid_printf(pr_chid, sizeof(pr_chid), &cbaf->chid);
        return scnprintf(buf, PAGE_SIZE, "%s\n", pr_chid);
}

static ssize_t cbaf_wusb_chid_store(struct device *dev,
                                         struct device_attribute *attr,
                                         const char *buf, size_t size)
{
        ssize_t result;
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        result = sscanf(buf,
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx",
                        &cbaf->chid.data[0] , &cbaf->chid.data[1],
                        &cbaf->chid.data[2] , &cbaf->chid.data[3],
                        &cbaf->chid.data[4] , &cbaf->chid.data[5],
                        &cbaf->chid.data[6] , &cbaf->chid.data[7],
                        &cbaf->chid.data[8] , &cbaf->chid.data[9],
                        &cbaf->chid.data[10], &cbaf->chid.data[11],
                        &cbaf->chid.data[12], &cbaf->chid.data[13],
                        &cbaf->chid.data[14], &cbaf->chid.data[15]);

        if (result != 16)
                return -EINVAL;

        result = cbaf_send_host_info(cbaf);
        if (result < 0)
                return result;
        result = cbaf_cdid_get(cbaf);
        if (result < 0)
                return -result;
        return size;
}
static DEVICE_ATTR(wusb_chid, 0600, cbaf_wusb_chid_show, cbaf_wusb_chid_store);

static ssize_t cbaf_wusb_host_name_show(struct device *dev,
                                        struct device_attribute *attr,
                                        char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        return scnprintf(buf, PAGE_SIZE, "%s\n", cbaf->host_name);
}

static ssize_t cbaf_wusb_host_name_store(struct device *dev,
                                         struct device_attribute *attr,
                                         const char *buf, size_t size)
{
        ssize_t result;
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        result = sscanf(buf, "%63s", cbaf->host_name);
        if (result != 1)
                return -EINVAL;

        return size;
}
static DEVICE_ATTR(wusb_host_name, 0600, cbaf_wusb_host_name_show,
                                         cbaf_wusb_host_name_store);

static ssize_t cbaf_wusb_host_band_groups_show(struct device *dev,
                                               struct device_attribute *attr,
                                               char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        return scnprintf(buf, PAGE_SIZE, "0x%04x\n", cbaf->host_band_groups);
}

static ssize_t cbaf_wusb_host_band_groups_store(struct device *dev,
                                                struct device_attribute *attr,
                                                const char *buf, size_t size)
{
        ssize_t result;
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);
        u16 band_groups = 0;

        result = sscanf(buf, "%04hx", &band_groups);
        if (result != 1)
                return -EINVAL;

        cbaf->host_band_groups = band_groups;

        return size;
}

static DEVICE_ATTR(wusb_host_band_groups, 0600,
                   cbaf_wusb_host_band_groups_show,
                   cbaf_wusb_host_band_groups_store);

static const struct wusb_cbaf_device_info cbaf_device_info_defaults = {
        .Length_hdr               = WUSB_AR_Length,
        .CDID_hdr                 = WUSB_AR_CDID,
        .BandGroups_hdr           = WUSB_AR_BandGroups,
        .LangID_hdr               = WUSB_AR_LangID,
        .DeviceFriendlyName_hdr   = WUSB_AR_DeviceFriendlyName,
};

static ssize_t cbaf_wusb_cdid_show(struct device *dev,
                                   struct device_attribute *attr, char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);
        char pr_cdid[WUSB_CKHDID_STRSIZE];

        ckhdid_printf(pr_cdid, sizeof(pr_cdid), &cbaf->cdid);
        return scnprintf(buf, PAGE_SIZE, "%s\n", pr_cdid);
}

static ssize_t cbaf_wusb_cdid_store(struct device *dev,
                                struct device_attribute *attr,
                                const char *buf, size_t size)
{
        ssize_t result;
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);
        struct wusb_ckhdid cdid;

        result = sscanf(buf,
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx",
                        &cdid.data[0] , &cdid.data[1],
                        &cdid.data[2] , &cdid.data[3],
                        &cdid.data[4] , &cdid.data[5],
                        &cdid.data[6] , &cdid.data[7],
                        &cdid.data[8] , &cdid.data[9],
                        &cdid.data[10], &cdid.data[11],
                        &cdid.data[12], &cdid.data[13],
                        &cdid.data[14], &cdid.data[15]);
        if (result != 16)
                return -EINVAL;

        cbaf->cdid = cdid;

        return size;
}
static DEVICE_ATTR(wusb_cdid, 0600, cbaf_wusb_cdid_show, cbaf_wusb_cdid_store);

static ssize_t cbaf_wusb_device_band_groups_show(struct device *dev,
                                                 struct device_attribute *attr,
                                                 char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        return scnprintf(buf, PAGE_SIZE, "0x%04x\n", cbaf->device_band_groups);
}

static DEVICE_ATTR(wusb_device_band_groups, 0600,
                   cbaf_wusb_device_band_groups_show,
                   NULL);

static ssize_t cbaf_wusb_device_name_show(struct device *dev,
                                        struct device_attribute *attr,
                                        char *buf)
{
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        return scnprintf(buf, PAGE_SIZE, "%s\n", cbaf->device_name);
}
static DEVICE_ATTR(wusb_device_name, 0600, cbaf_wusb_device_name_show, NULL);

static const struct wusb_cbaf_cc_data cbaf_cc_data_defaults = {
        .AssociationTypeId_hdr    = WUSB_AR_AssociationTypeId,
        .AssociationTypeId        = cpu_to_le16(AR_TYPE_WUSB),
        .AssociationSubTypeId_hdr = WUSB_AR_AssociationSubTypeId,
        .AssociationSubTypeId     = cpu_to_le16(AR_TYPE_WUSB_ASSOCIATE),
        .Length_hdr               = WUSB_AR_Length,
        .Length                   = cpu_to_le32(sizeof(struct wusb_cbaf_cc_data)),
        .ConnectionContext_hdr    = WUSB_AR_ConnectionContext,
        .BandGroups_hdr           = WUSB_AR_BandGroups,
};

static const struct wusb_cbaf_cc_data_fail cbaf_cc_data_fail_defaults = {
        .AssociationTypeId_hdr    = WUSB_AR_AssociationTypeId,
        .AssociationSubTypeId_hdr = WUSB_AR_AssociationSubTypeId,
        .Length_hdr               = WUSB_AR_Length,
        .AssociationStatus_hdr    = WUSB_AR_AssociationStatus,
};

/*
 * Send a new CC to the device.
 */
static int cbaf_cc_upload(struct cbaf *cbaf)
{
        int result;
        struct device *dev = &cbaf->usb_iface->dev;
        struct wusb_cbaf_cc_data *ccd;
        char pr_cdid[WUSB_CKHDID_STRSIZE];

        ccd =  cbaf->buffer;
        *ccd = cbaf_cc_data_defaults;
        ccd->CHID = cbaf->chid;
        ccd->CDID = cbaf->cdid;
        ccd->CK = cbaf->ck;
        ccd->BandGroups = cpu_to_le16(cbaf->host_band_groups);

        dev_dbg(dev, "Trying to upload CC:\n");
        ckhdid_printf(pr_cdid, sizeof(pr_cdid), &ccd->CHID);
        dev_dbg(dev, "  CHID       %s\n", pr_cdid);
        ckhdid_printf(pr_cdid, sizeof(pr_cdid), &ccd->CDID);
        dev_dbg(dev, "  CDID       %s\n", pr_cdid);
        dev_dbg(dev, "  Bandgroups 0x%04x\n", cbaf->host_band_groups);

        result = usb_control_msg(
                cbaf->usb_dev, usb_sndctrlpipe(cbaf->usb_dev, 0),
                CBAF_REQ_SET_ASSOCIATION_RESPONSE,
                USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE,
                0x0201, cbaf->usb_iface->cur_altsetting->desc.bInterfaceNumber,
                ccd, sizeof(*ccd), 1000 /* FIXME: arbitrary */);

        return result;
}

static ssize_t cbaf_wusb_ck_store(struct device *dev,
                                  struct device_attribute *attr,
                                  const char *buf, size_t size)
{
        ssize_t result;
        struct usb_interface *iface = to_usb_interface(dev);
        struct cbaf *cbaf = usb_get_intfdata(iface);

        result = sscanf(buf,
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx "
                        "%02hhx %02hhx %02hhx %02hhx",
                        &cbaf->ck.data[0] , &cbaf->ck.data[1],
                        &cbaf->ck.data[2] , &cbaf->ck.data[3],
                        &cbaf->ck.data[4] , &cbaf->ck.data[5],
                        &cbaf->ck.data[6] , &cbaf->ck.data[7],
                        &cbaf->ck.data[8] , &cbaf->ck.data[9],
                        &cbaf->ck.data[10], &cbaf->ck.data[11],
                        &cbaf->ck.data[12], &cbaf->ck.data[13],
                        &cbaf->ck.data[14], &cbaf->ck.data[15]);
        if (result != 16)
                return -EINVAL;

        result = cbaf_cc_upload(cbaf);
        if (result < 0)
                return result;

        return size;
}
static DEVICE_ATTR(wusb_ck, 0600, NULL, cbaf_wusb_ck_store);

static struct attribute *cbaf_dev_attrs[] = {
        &dev_attr_wusb_host_name.attr,
        &dev_attr_wusb_host_band_groups.attr,
        &dev_attr_wusb_chid.attr,
        &dev_attr_wusb_cdid.attr,
        &dev_attr_wusb_device_name.attr,
        &dev_attr_wusb_device_band_groups.attr,
        &dev_attr_wusb_ck.attr,
        NULL,
};

static struct attribute_group cbaf_dev_attr_group = {
        .name = NULL,   /* we want them in the same directory */
        .attrs = cbaf_dev_attrs,
};

static int cbaf_probe(struct usb_interface *iface,
                      const struct usb_device_id *id)
{
        struct cbaf *cbaf;
        struct device *dev = &iface->dev;
        int result = -ENOMEM;

        cbaf = kzalloc(sizeof(*cbaf), GFP_KERNEL);
        if (cbaf == NULL)
                goto error_kzalloc;
        cbaf->buffer = kmalloc(512, GFP_KERNEL);
        if (cbaf->buffer == NULL)
                goto error_kmalloc_buffer;

        cbaf->buffer_size = 512;
        cbaf->usb_dev = usb_get_dev(interface_to_usbdev(iface));
        cbaf->usb_iface = usb_get_intf(iface);
        result = cbaf_check(cbaf);
        if (result < 0) {
                dev_err(dev, "This device is not WUSB-CBAF compliant"
                        "and is not supported yet.\n");
                goto error_check;
        }

        result = sysfs_create_group(&dev->kobj, &cbaf_dev_attr_group);
        if (result < 0) {
                dev_err(dev, "Can't register sysfs attr group: %d\n", result);
                goto error_create_group;
        }
        usb_set_intfdata(iface, cbaf);
        return 0;

error_create_group:
error_check:
        kfree(cbaf->buffer);
error_kmalloc_buffer:
        kfree(cbaf);
error_kzalloc:
        return result;
}

static void cbaf_disconnect(struct usb_interface *iface)
{
        struct cbaf *cbaf = usb_get_intfdata(iface);
        struct device *dev = &iface->dev;
        sysfs_remove_group(&dev->kobj, &cbaf_dev_attr_group);
        usb_set_intfdata(iface, NULL);
        usb_put_intf(iface);
        kfree(cbaf->buffer);
        /* paranoia: clean up crypto keys */
        kzfree(cbaf);
}

static struct usb_device_id cbaf_id_table[] = {
        { USB_INTERFACE_INFO(0xef, 0x03, 0x01), },
        { },
};
MODULE_DEVICE_TABLE(usb, cbaf_id_table);

static struct usb_driver cbaf_driver = {
        .name =         "wusb-cbaf",
        .id_table =     cbaf_id_table,
        .probe =        cbaf_probe,
        .disconnect =   cbaf_disconnect,
};

static int __init cbaf_driver_init(void)
{
        return usb_register(&cbaf_driver);
}
module_init(cbaf_driver_init);

static void __exit cbaf_driver_exit(void)
{
        usb_deregister(&cbaf_driver);
}
module_exit(cbaf_driver_exit);

MODULE_AUTHOR("Inaky Perez-Gonzalez <inaky.perez-gonzalez@intel.com>");
MODULE_DESCRIPTION("Wireless USB Cable Based Association");
MODULE_LICENSE("GPL");