drivers/base/firmware_class.c

   1 /*
   2  * firmware_class.c - Multi purpose firmware loading support
   3  *
   4  * Copyright (c) 2003 Manuel Estrada Sainz <ranty@debian.org>
   5  *
   6  * Please see Documentation/firmware_class/ for more information.
   7  *
   8  */
   9
  10 #include <linux/capability.h>
  11 #include <linux/device.h>
  12 #include <linux/module.h>
  13 #include <linux/init.h>
  14 #include <linux/timer.h>
  15 #include <linux/vmalloc.h>
  16 #include <linux/interrupt.h>
  17 #include <linux/bitops.h>
  18 #include <asm/semaphore.h>
  19
  20 #include <linux/firmware.h>
  21 #include "base.h"
  22
  23 MODULE_AUTHOR("Manuel Estrada Sainz <ranty@debian.org>");
  24 MODULE_DESCRIPTION("Multi purpose firmware loading support");
  25 MODULE_LICENSE("GPL");
  26
  27 enum {
  28         FW_STATUS_LOADING,
  29         FW_STATUS_DONE,
  30         FW_STATUS_ABORT,
  31         FW_STATUS_READY,
  32         FW_STATUS_READY_NOHOTPLUG,
  33 };
  34
  35 static int loading_timeout = 10;        /* In seconds */
  36
  37 /* fw_lock could be moved to 'struct firmware_priv' but since it is just
  38  * guarding for corner cases a global lock should be OK */
  39 static DECLARE_MUTEX(fw_lock);
  40
  41 struct firmware_priv {
  42         char fw_id[FIRMWARE_NAME_MAX];
  43         struct completion completion;
  44         struct bin_attribute attr_data;
  45         struct firmware *fw;
  46         unsigned long status;
  47         int alloc_size;
  48         struct timer_list timeout;
  49 };
  50
  51 static inline void
  52 fw_load_abort(struct firmware_priv *fw_priv)
  53 {
  54         set_bit(FW_STATUS_ABORT, &fw_priv->status);
  55         wmb();
  56         complete(&fw_priv->completion);
  57 }
  58
  59 static ssize_t
  60 firmware_timeout_show(struct class *class, char *buf)
  61 {
  62         return sprintf(buf, "%d\n", loading_timeout);
  63 }
  64
  65 /**
  66  * firmware_timeout_store - set number of seconds to wait for firmware
  67  * @class: device class pointer
  68  * @buf: buffer to scan for timeout value
  69  * @count: number of bytes in @buf
  70  *
  71  *      Sets the number of seconds to wait for the firmware.  Once
  72  *      this expires an error will be returned to the driver and no
  73  *      firmware will be provided.
  74  *
  75  *      Note: zero means 'wait forever'.
  76  **/
  77 static ssize_t
  78 firmware_timeout_store(struct class *class, const char *buf, size_t count)
  79 {
  80         loading_timeout = simple_strtol(buf, NULL, 10);
  81         if (loading_timeout < 0)
  82                 loading_timeout = 0;
  83         return count;
  84 }
  85
  86 static CLASS_ATTR(timeout, 0644, firmware_timeout_show, firmware_timeout_store);
  87
  88 static void  fw_class_dev_release(struct class_device *class_dev);
  89 int firmware_class_uevent(struct class_device *dev, char **envp,
  90                            int num_envp, char *buffer, int buffer_size);
  91
  92 static struct class firmware_class = {
  93         .name           = "firmware",
  94         .uevent = firmware_class_uevent,
  95         .release        = fw_class_dev_release,
  96 };
  97
  98 int
  99 firmware_class_uevent(struct class_device *class_dev, char **envp,
 100                        int num_envp, char *buffer, int buffer_size)
 101 {
 102         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 103         int i = 0, len = 0;
 104
 105         if (!test_bit(FW_STATUS_READY, &fw_priv->status))
 106                 return -ENODEV;
 107
 108         if (add_uevent_var(envp, num_envp, &i, buffer, buffer_size, &len,
 109                            "FIRMWARE=%s", fw_priv->fw_id))
 110                 return -ENOMEM;
 111         if (add_uevent_var(envp, num_envp, &i, buffer, buffer_size, &len,
 112                            "TIMEOUT=%i", loading_timeout))
 113                 return -ENOMEM;
 114         envp[i] = NULL;
 115
 116         return 0;
 117 }
 118
 119 static ssize_t
 120 firmware_loading_show(struct class_device *class_dev, char *buf)
 121 {
 122         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 123         int loading = test_bit(FW_STATUS_LOADING, &fw_priv->status);
 124         return sprintf(buf, "%d\n", loading);
 125 }
 126
 127 /**
 128  * firmware_loading_store - set value in the 'loading' control file
 129  * @class_dev: class_device pointer
 130  * @buf: buffer to scan for loading control value
 131  * @count: number of bytes in @buf
 132  *
 133  *      The relevant values are:
 134  *
 135  *       1: Start a load, discarding any previous partial load.
 136  *       0: Conclude the load and hand the data to the driver code.
 137  *      -1: Conclude the load with an error and discard any written data.
 138  **/
 139 static ssize_t
 140 firmware_loading_store(struct class_device *class_dev,
 141                        const char *buf, size_t count)
 142 {
 143         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 144         int loading = simple_strtol(buf, NULL, 10);
 145
 146         switch (loading) {
 147         case 1:
 148                 down(&fw_lock);
 149                 if (!fw_priv->fw) {
 150                         up(&fw_lock);
 151                         break;
 152                 }
 153                 vfree(fw_priv->fw->data);
 154                 fw_priv->fw->data = NULL;
 155                 fw_priv->fw->size = 0;
 156                 fw_priv->alloc_size = 0;
 157                 set_bit(FW_STATUS_LOADING, &fw_priv->status);
 158                 up(&fw_lock);
 159                 break;
 160         case 0:
 161                 if (test_bit(FW_STATUS_LOADING, &fw_priv->status)) {
 162                         complete(&fw_priv->completion);
 163                         clear_bit(FW_STATUS_LOADING, &fw_priv->status);
 164                         break;
 165                 }
 166                 /* fallthrough */
 167         default:
 168                 printk(KERN_ERR "%s: unexpected value (%d)\n", __FUNCTION__,
 169                        loading);
 170                 /* fallthrough */
 171         case -1:
 172                 fw_load_abort(fw_priv);
 173                 break;
 174         }
 175
 176         return count;
 177 }
 178
 179 static CLASS_DEVICE_ATTR(loading, 0644,
 180                         firmware_loading_show, firmware_loading_store);
 181
 182 static ssize_t
 183 firmware_data_read(struct kobject *kobj,
 184                    char *buffer, loff_t offset, size_t count)
 185 {
 186         struct class_device *class_dev = to_class_dev(kobj);
 187         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 188         struct firmware *fw;
 189         ssize_t ret_count = count;
 190
 191         down(&fw_lock);
 192         fw = fw_priv->fw;
 193         if (!fw || test_bit(FW_STATUS_DONE, &fw_priv->status)) {
 194                 ret_count = -ENODEV;
 195                 goto out;
 196         }
 197         if (offset > fw->size) {
 198                 ret_count = 0;
 199                 goto out;
 200         }
 201         if (offset + ret_count > fw->size)
 202                 ret_count = fw->size - offset;
 203
 204         memcpy(buffer, fw->data + offset, ret_count);
 205 out:
 206         up(&fw_lock);
 207         return ret_count;
 208 }
 209
 210 static int
 211 fw_realloc_buffer(struct firmware_priv *fw_priv, int min_size)
 212 {
 213         u8 *new_data;
 214
 215         if (min_size <= fw_priv->alloc_size)
 216                 return 0;
 217
 218         new_data = vmalloc(fw_priv->alloc_size + PAGE_SIZE);
 219         if (!new_data) {
 220                 printk(KERN_ERR "%s: unable to alloc buffer\n", __FUNCTION__);
 221                 /* Make sure that we don't keep incomplete data */
 222                 fw_load_abort(fw_priv);
 223                 return -ENOMEM;
 224         }
 225         fw_priv->alloc_size += PAGE_SIZE;
 226         if (fw_priv->fw->data) {
 227                 memcpy(new_data, fw_priv->fw->data, fw_priv->fw->size);
 228                 vfree(fw_priv->fw->data);
 229         }
 230         fw_priv->fw->data = new_data;
 231         BUG_ON(min_size > fw_priv->alloc_size);
 232         return 0;
 233 }
 234
 235 /**
 236  * firmware_data_write - write method for firmware
 237  * @kobj: kobject for the class_device
 238  * @buffer: buffer being written
 239  * @offset: buffer offset for write in total data store area
 240  * @count: buffer size
 241  *
 242  *      Data written to the 'data' attribute will be later handed to
 243  *      the driver as a firmware image.
 244  **/
 245 static ssize_t
 246 firmware_data_write(struct kobject *kobj,
 247                     char *buffer, loff_t offset, size_t count)
 248 {
 249         struct class_device *class_dev = to_class_dev(kobj);
 250         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 251         struct firmware *fw;
 252         ssize_t retval;
 253
 254         if (!capable(CAP_SYS_RAWIO))
 255                 return -EPERM;
 256
 257         down(&fw_lock);
 258         fw = fw_priv->fw;
 259         if (!fw || test_bit(FW_STATUS_DONE, &fw_priv->status)) {
 260                 retval = -ENODEV;
 261                 goto out;
 262         }
 263         retval = fw_realloc_buffer(fw_priv, offset + count);
 264         if (retval)
 265                 goto out;
 266
 267         memcpy(fw->data + offset, buffer, count);
 268
 269         fw->size = max_t(size_t, offset + count, fw->size);
 270         retval = count;
 271 out:
 272         up(&fw_lock);
 273         return retval;
 274 }
 275
 276 static struct bin_attribute firmware_attr_data_tmpl = {
 277         .attr = {.name = "data", .mode = 0644, .owner = THIS_MODULE},
 278         .size = 0,
 279         .read = firmware_data_read,
 280         .write = firmware_data_write,
 281 };
 282
 283 static void
 284 fw_class_dev_release(struct class_device *class_dev)
 285 {
 286         struct firmware_priv *fw_priv = class_get_devdata(class_dev);
 287
 288         kfree(fw_priv);
 289         kfree(class_dev);
 290
 291         module_put(THIS_MODULE);
 292 }
 293
 294 static void
 295 firmware_class_timeout(u_long data)
 296 {
 297         struct firmware_priv *fw_priv = (struct firmware_priv *) data;
 298         fw_load_abort(fw_priv);
 299 }
 300
 301 static inline void
 302 fw_setup_class_device_id(struct class_device *class_dev, struct device *dev)
 303 {
 304         /* XXX warning we should watch out for name collisions */
 305         strlcpy(class_dev->class_id, dev->bus_id, BUS_ID_SIZE);
 306 }
 307
 308 static int
 309 fw_register_class_device(struct class_device **class_dev_p,
 310                          const char *fw_name, struct device *device)
 311 {
 312         int retval;
 313         struct firmware_priv *fw_priv = kzalloc(sizeof(*fw_priv),
 314                                                 GFP_KERNEL);
 315         struct class_device *class_dev = kzalloc(sizeof(*class_dev),
 316                                                  GFP_KERNEL);
 317
 318         *class_dev_p = NULL;
 319
 320         if (!fw_priv || !class_dev) {
 321                 printk(KERN_ERR "%s: kmalloc failed\n", __FUNCTION__);
 322                 retval = -ENOMEM;
 323                 goto error_kfree;
 324         }
 325
 326         init_completion(&fw_priv->completion);
 327         fw_priv->attr_data = firmware_attr_data_tmpl;
 328         strlcpy(fw_priv->fw_id, fw_name, FIRMWARE_NAME_MAX);
 329
 330         fw_priv->timeout.function = firmware_class_timeout;
 331         fw_priv->timeout.data = (u_long) fw_priv;
 332         init_timer(&fw_priv->timeout);
 333
 334         fw_setup_class_device_id(class_dev, device);
 335         class_dev->dev = device;
 336         class_dev->class = &firmware_class;
 337         class_set_devdata(class_dev, fw_priv);
 338         retval = class_device_register(class_dev);
 339         if (retval) {
 340                 printk(KERN_ERR "%s: class_device_register failed\n",
 341                        __FUNCTION__);
 342                 goto error_kfree;
 343         }
 344         *class_dev_p = class_dev;
 345         return 0;
 346
 347 error_kfree:
 348         kfree(fw_priv);
 349         kfree(class_dev);
 350         return retval;
 351 }
 352
 353 static int
 354 fw_setup_class_device(struct firmware *fw, struct class_device **class_dev_p,
 355                       const char *fw_name, struct device *device, int uevent)
 356 {
 357         struct class_device *class_dev;
 358         struct firmware_priv *fw_priv;
 359         int retval;
 360
 361         *class_dev_p = NULL;
 362         retval = fw_register_class_device(&class_dev, fw_name, device);
 363         if (retval)
 364                 goto out;
 365
 366         /* Need to pin this module until class device is destroyed */
 367         __module_get(THIS_MODULE);
 368
 369         fw_priv = class_get_devdata(class_dev);
 370
 371         fw_priv->fw = fw;
 372         retval = sysfs_create_bin_file(&class_dev->kobj, &fw_priv->attr_data);
 373         if (retval) {
 374                 printk(KERN_ERR "%s: sysfs_create_bin_file failed\n",
 375                        __FUNCTION__);
 376                 goto error_unreg;
 377         }
 378
 379         retval = class_device_create_file(class_dev,
 380                                           &class_device_attr_loading);
 381         if (retval) {
 382                 printk(KERN_ERR "%s: class_device_create_file failed\n",
 383                        __FUNCTION__);
 384                 goto error_unreg;
 385         }
 386
 387         if (uevent)
 388                 set_bit(FW_STATUS_READY, &fw_priv->status);
 389         else
 390                 set_bit(FW_STATUS_READY_NOHOTPLUG, &fw_priv->status);
 391         *class_dev_p = class_dev;
 392         goto out;
 393
 394 error_unreg:
 395         class_device_unregister(class_dev);
 396 out:
 397         return retval;
 398 }
 399
 400 static int
 401 _request_firmware(const struct firmware **firmware_p, const char *name,
 402                  struct device *device, int uevent)
 403 {
 404         struct class_device *class_dev;
 405         struct firmware_priv *fw_priv;
 406         struct firmware *firmware;
 407         int retval;
 408
 409         if (!firmware_p)
 410                 return -EINVAL;
 411
 412         *firmware_p = firmware = kzalloc(sizeof(*firmware), GFP_KERNEL);
 413         if (!firmware) {
 414                 printk(KERN_ERR "%s: kmalloc(struct firmware) failed\n",
 415                        __FUNCTION__);
 416                 retval = -ENOMEM;
 417                 goto out;
 418         }
 419
 420         retval = fw_setup_class_device(firmware, &class_dev, name, device,
 421                                        uevent);
 422         if (retval)
 423                 goto error_kfree_fw;
 424
 425         fw_priv = class_get_devdata(class_dev);
 426
 427         if (uevent) {
 428                 if (loading_timeout > 0) {
 429                         fw_priv->timeout.expires = jiffies + loading_timeout * HZ;
 430                         add_timer(&fw_priv->timeout);
 431                 }
 432
 433                 kobject_uevent(&class_dev->kobj, KOBJ_ADD);
 434                 wait_for_completion(&fw_priv->completion);
 435                 set_bit(FW_STATUS_DONE, &fw_priv->status);
 436                 del_timer_sync(&fw_priv->timeout);
 437         } else
 438                 wait_for_completion(&fw_priv->completion);
 439
 440         down(&fw_lock);
 441         if (!fw_priv->fw->size || test_bit(FW_STATUS_ABORT, &fw_priv->status)) {
 442                 retval = -ENOENT;
 443                 release_firmware(fw_priv->fw);
 444                 *firmware_p = NULL;
 445         }
 446         fw_priv->fw = NULL;
 447         up(&fw_lock);
 448         class_device_unregister(class_dev);
 449         goto out;
 450
 451 error_kfree_fw:
 452         kfree(firmware);
 453         *firmware_p = NULL;
 454 out:
 455         return retval;
 456 }
 457
 458 /**
 459  * request_firmware: - send firmware request and wait for it
 460  * @firmware_p: pointer to firmware image
 461  * @name: name of firmware file
 462  * @device: device for which firmware is being loaded
 463  *
 464  *      @firmware_p will be used to return a firmware image by the name
 465  *      of @name for device @device.
 466  *
 467  *      Should be called from user context where sleeping is allowed.
 468  *
 469  *      @name will be used as $FIRMWARE in the uevent environment and
 470  *      should be distinctive enough not to be confused with any other
 471  *      firmware image for this or any other device.
 472  **/
 473 int
 474 request_firmware(const struct firmware **firmware_p, const char *name,
 475                  struct device *device)
 476 {
 477         int uevent = 1;
 478         return _request_firmware(firmware_p, name, device, uevent);
 479 }
 480
 481 /**
 482  * release_firmware: - release the resource associated with a firmware image
 483  * @fw: firmware resource to release
 484  **/
 485 void
 486 release_firmware(const struct firmware *fw)
 487 {
 488         if (fw) {
 489                 vfree(fw->data);
 490                 kfree(fw);
 491         }
 492 }
 493
 494 /**
 495  * register_firmware: - provide a firmware image for later usage
 496  * @name: name of firmware image file
 497  * @data: buffer pointer for the firmware image
 498  * @size: size of the data buffer area
 499  *
 500  *      Make sure that @data will be available by requesting firmware @name.
 501  *
 502  *      Note: This will not be possible until some kind of persistence
 503  *      is available.
 504  **/
 505 void
 506 register_firmware(const char *name, const u8 *data, size_t size)
 507 {
 508         /* This is meaningless without firmware caching, so until we
 509          * decide if firmware caching is reasonable just leave it as a
 510          * noop */
 511 }
 512
 513 /* Async support */
 514 struct firmware_work {
 515         struct work_struct work;
 516         struct module *module;
 517         const char *name;
 518         struct device *device;
 519         void *context;
 520         void (*cont)(const struct firmware *fw, void *context);
 521         int uevent;
 522 };
 523
 524 static int
 525 request_firmware_work_func(void *arg)
 526 {
 527         struct firmware_work *fw_work = arg;
 528         const struct firmware *fw;
 529         int ret;
 530         if (!arg) {
 531                 WARN_ON(1);
 532                 return 0;
 533         }
 534         daemonize("%s/%s", "firmware", fw_work->name);
 535         ret = _request_firmware(&fw, fw_work->name, fw_work->device,
 536                 fw_work->uevent);
 537         if (ret < 0)
 538                 fw_work->cont(NULL, fw_work->context);
 539         else {
 540                 fw_work->cont(fw, fw_work->context);
 541                 release_firmware(fw);
 542         }
 543         module_put(fw_work->module);
 544         kfree(fw_work);
 545         return ret;
 546 }
 547
 548 /**
 549  * request_firmware_nowait: asynchronous version of request_firmware
 550  * @module: module requesting the firmware
 551  * @uevent: sends uevent to copy the firmware image if this flag
 552  *      is non-zero else the firmware copy must be done manually.
 553  * @name: name of firmware file
 554  * @device: device for which firmware is being loaded
 555  * @context: will be passed over to @cont, and
 556  *      @fw may be %NULL if firmware request fails.
 557  * @cont: function will be called asynchronously when the firmware
 558  *      request is over.
 559  *
 560  *      Asynchronous variant of request_firmware() for contexts where
 561  *      it is not possible to sleep.
 562  **/
 563 int
 564 request_firmware_nowait(
 565         struct module *module, int uevent,
 566         const char *name, struct device *device, void *context,
 567         void (*cont)(const struct firmware *fw, void *context))
 568 {
 569         struct firmware_work *fw_work = kmalloc(sizeof (struct firmware_work),
 570                                                 GFP_ATOMIC);
 571         int ret;
 572
 573         if (!fw_work)
 574                 return -ENOMEM;
 575         if (!try_module_get(module)) {
 576                 kfree(fw_work);
 577                 return -EFAULT;
 578         }
 579
 580         *fw_work = (struct firmware_work) {
 581                 .module = module,
 582                 .name = name,
 583                 .device = device,
 584                 .context = context,
 585                 .cont = cont,
 586                 .uevent = uevent,
 587         };
 588
 589         ret = kernel_thread(request_firmware_work_func, fw_work,
 590                             CLONE_FS | CLONE_FILES);
 591
 592         if (ret < 0) {
 593                 fw_work->cont(NULL, fw_work->context);
 594                 module_put(fw_work->module);
 595                 kfree(fw_work);
 596                 return ret;
 597         }
 598         return 0;
 599 }
 600
 601 static int __init
 602 firmware_class_init(void)
 603 {
 604         int error;
 605         error = class_register(&firmware_class);
 606         if (error) {
 607                 printk(KERN_ERR "%s: class_register failed\n", __FUNCTION__);
 608                 return error;
 609         }
 610         error = class_create_file(&firmware_class, &class_attr_timeout);
 611         if (error) {
 612                 printk(KERN_ERR "%s: class_create_file failed\n",
 613                        __FUNCTION__);
 614                 class_unregister(&firmware_class);
 615         }
 616         return error;
 617
 618 }
 619 static void __exit
 620 firmware_class_exit(void)
 621 {
 622         class_unregister(&firmware_class);
 623 }
 624
 625 module_init(firmware_class_init);
 626 module_exit(firmware_class_exit);
 627
 628 EXPORT_SYMBOL(release_firmware);
 629 EXPORT_SYMBOL(request_firmware);
 630 EXPORT_SYMBOL(request_firmware_nowait);
 631 EXPORT_SYMBOL(register_firmware);