git.oblomov.eu Git - linux-2.6/blob - drivers/pci/hotplug/fakephp.c

   1 /*
   2  * Fake PCI Hot Plug Controller Driver
   3  *
   4  * Copyright (C) 2003 Greg Kroah-Hartman <greg@kroah.com>
   5  * Copyright (C) 2003 IBM Corp.
   6  * Copyright (C) 2003 Rolf Eike Beer <eike-kernel@sf-tec.de>
   7  *
   8  * Based on ideas and code from:
   9  *      Vladimir Kondratiev <vladimir.kondratiev@intel.com>
  10  *      Rolf Eike Beer <eike-kernel@sf-tec.de>
  11  *
  12  * All rights reserved.
  13  *
  14  * This program is free software; you can redistribute it and/or modify
  15  * it under the terms of the GNU General Public License as published by
  16  * the Free Software Foundation, version 2 of the License.
  17  *
  18  * Send feedback to <greg@kroah.com>
  19  */
  20
  21 /*
  22  *
  23  * This driver will "emulate" removing PCI devices from the system.  If
  24  * the "power" file is written to with "0" then the specified PCI device
  25  * will be completely removed from the kernel.
  26  *
  27  * WARNING, this does NOT turn off the power to the PCI device.  This is
  28  * a "logical" removal, not a physical or electrical removal.
  29  *
  30  * Use this module at your own risk, you have been warned!
  31  *
  32  * Enabling PCI devices is left as an exercise for the reader...
  33  *
  34  */
  35 #include <linux/kernel.h>
  36 #include <linux/module.h>
  37 #include <linux/pci.h>
  38 #include <linux/pci_hotplug.h>
  39 #include <linux/init.h>
  40 #include <linux/string.h>
  41 #include <linux/slab.h>
  42 #include <linux/workqueue.h>
  43 #include "../pci.h"
  44
  45 #if !defined(MODULE)
  46         #define MY_NAME "fakephp"
  47 #else
  48         #define MY_NAME THIS_MODULE->name
  49 #endif
  50
  51 #define dbg(format, arg...)                                     \
  52         do {                                                    \
  53                 if (debug)                                      \
  54                         printk(KERN_DEBUG "%s: " format,        \
  55                                 MY_NAME , ## arg);              \
  56         } while (0)
  57 #define err(format, arg...) printk(KERN_ERR "%s: " format, MY_NAME , ## arg)
  58 #define info(format, arg...) printk(KERN_INFO "%s: " format, MY_NAME , ## arg)
  59
  60 #define DRIVER_AUTHOR   "Greg Kroah-Hartman <greg@kroah.com>"
  61 #define DRIVER_DESC     "Fake PCI Hot Plug Controller Driver"
  62
  63 struct dummy_slot {
  64         struct list_head node;
  65         struct hotplug_slot *slot;
  66         struct pci_dev *dev;
  67         struct work_struct remove_work;
  68         unsigned long removed;
  69 };
  70
  71 static int debug;
  72 static int dup_slots;
  73 static LIST_HEAD(slot_list);
  74 static struct workqueue_struct *dummyphp_wq;
  75
  76 static void pci_rescan_worker(struct work_struct *work);
  77 static DECLARE_WORK(pci_rescan_work, pci_rescan_worker);
  78
  79 static int enable_slot (struct hotplug_slot *slot);
  80 static int disable_slot (struct hotplug_slot *slot);
  81
  82 static struct hotplug_slot_ops dummy_hotplug_slot_ops = {
  83         .owner                  = THIS_MODULE,
  84         .enable_slot            = enable_slot,
  85         .disable_slot           = disable_slot,
  86 };
  87
  88 static void dummy_release(struct hotplug_slot *slot)
  89 {
  90         struct dummy_slot *dslot = slot->private;
  91
  92         list_del(&dslot->node);
  93         kfree(dslot->slot->info);
  94         kfree(dslot->slot);
  95         pci_dev_put(dslot->dev);
  96         kfree(dslot);
  97 }
  98
  99 #define SLOT_NAME_SIZE  8
 100
 101 static int add_slot(struct pci_dev *dev)
 102 {
 103         struct dummy_slot *dslot;
 104         struct hotplug_slot *slot;
 105         char name[SLOT_NAME_SIZE];
 106         int retval = -ENOMEM;
 107         static int count = 1;
 108
 109         slot = kzalloc(sizeof(struct hotplug_slot), GFP_KERNEL);
 110         if (!slot)
 111                 goto error;
 112
 113         slot->info = kzalloc(sizeof(struct hotplug_slot_info), GFP_KERNEL);
 114         if (!slot->info)
 115                 goto error_slot;
 116
 117         slot->info->power_status = 1;
 118         slot->info->max_bus_speed = PCI_SPEED_UNKNOWN;
 119         slot->info->cur_bus_speed = PCI_SPEED_UNKNOWN;
 120
 121         dslot = kzalloc(sizeof(struct dummy_slot), GFP_KERNEL);
 122         if (!dslot)
 123                 goto error_info;
 124
 125         if (dup_slots)
 126                 snprintf(name, SLOT_NAME_SIZE, "fake");
 127         else
 128                 snprintf(name, SLOT_NAME_SIZE, "fake%d", count++);
 129         dbg("slot->name = %s\n", name);
 130         slot->ops = &dummy_hotplug_slot_ops;
 131         slot->release = &dummy_release;
 132         slot->private = dslot;
 133
 134         retval = pci_hp_register(slot, dev->bus, PCI_SLOT(dev->devfn), name);
 135         if (retval) {
 136                 err("pci_hp_register failed with error %d\n", retval);
 137                 goto error_dslot;
 138         }
 139
 140         dbg("slot->name = %s\n", hotplug_slot_name(slot));
 141         dslot->slot = slot;
 142         dslot->dev = pci_dev_get(dev);
 143         list_add (&dslot->node, &slot_list);
 144         return retval;
 145
 146 error_dslot:
 147         kfree(dslot);
 148 error_info:
 149         kfree(slot->info);
 150 error_slot:
 151         kfree(slot);
 152 error:
 153         return retval;
 154 }
 155
 156 static int __init pci_scan_buses(void)
 157 {
 158         struct pci_dev *dev = NULL;
 159         int lastslot = 0;
 160
 161         while ((dev = pci_get_device(PCI_ANY_ID, PCI_ANY_ID, dev)) != NULL) {
 162                 if (PCI_FUNC(dev->devfn) > 0 &&
 163                                 lastslot == PCI_SLOT(dev->devfn))
 164                         continue;
 165                 lastslot = PCI_SLOT(dev->devfn);
 166                 add_slot(dev);
 167         }
 168
 169         return 0;
 170 }
 171
 172 static void remove_slot(struct dummy_slot *dslot)
 173 {
 174         int retval;
 175
 176         dbg("removing slot %s\n", hotplug_slot_name(dslot->slot));
 177         retval = pci_hp_deregister(dslot->slot);
 178         if (retval)
 179                 err("Problem unregistering a slot %s\n",
 180                         hotplug_slot_name(dslot->slot));
 181 }
 182
 183 /* called from the single-threaded workqueue handler to remove a slot */
 184 static void remove_slot_worker(struct work_struct *work)
 185 {
 186         struct dummy_slot *dslot =
 187                 container_of(work, struct dummy_slot, remove_work);
 188         remove_slot(dslot);
 189 }
 190
 191 /**
 192  * pci_rescan_slot - Rescan slot
 193  * @temp: Device template. Should be set: bus and devfn.
 194  *
 195  * Tries hard not to re-enable already existing devices;
 196  * also handles scanning of subfunctions.
 197  */
 198 static int pci_rescan_slot(struct pci_dev *temp)
 199 {
 200         struct pci_bus *bus = temp->bus;
 201         struct pci_dev *dev;
 202         int func;
 203         u8 hdr_type;
 204         int count = 0;
 205
 206         if (!pci_read_config_byte(temp, PCI_HEADER_TYPE, &hdr_type)) {
 207                 temp->hdr_type = hdr_type & 0x7f;
 208                 if ((dev = pci_get_slot(bus, temp->devfn)) != NULL)
 209                         pci_dev_put(dev);
 210                 else {
 211                         dev = pci_scan_single_device(bus, temp->devfn);
 212                         if (dev) {
 213                                 dbg("New device on %s function %x:%x\n",
 214                                         bus->name, temp->devfn >> 3,
 215                                         temp->devfn & 7);
 216                                 count++;
 217                         }
 218                 }
 219                 /* multifunction device? */
 220                 if (!(hdr_type & 0x80))
 221                         return count;
 222
 223                 /* continue scanning for other functions */
 224                 for (func = 1, temp->devfn++; func < 8; func++, temp->devfn++) {
 225                         if (pci_read_config_byte(temp, PCI_HEADER_TYPE, &hdr_type))
 226                                 continue;
 227                         temp->hdr_type = hdr_type & 0x7f;
 228
 229                         if ((dev = pci_get_slot(bus, temp->devfn)) != NULL)
 230                                 pci_dev_put(dev);
 231                         else {
 232                                 dev = pci_scan_single_device(bus, temp->devfn);
 233                                 if (dev) {
 234                                         dbg("New device on %s function %x:%x\n",
 235                                                 bus->name, temp->devfn >> 3,
 236                                                 temp->devfn & 7);
 237                                         count++;
 238                                 }
 239                         }
 240                 }
 241         }
 242
 243         return count;
 244 }
 245
 246
 247 /**
 248  * pci_rescan_bus - Rescan PCI bus
 249  * @bus: the PCI bus to rescan
 250  *
 251  * Call pci_rescan_slot for each possible function of the bus.
 252  */
 253 static void pci_rescan_bus(const struct pci_bus *bus)
 254 {
 255         unsigned int devfn;
 256         struct pci_dev *dev;
 257         int retval;
 258         int found = 0;
 259         dev = alloc_pci_dev();
 260         if (!dev)
 261                 return;
 262
 263         dev->bus = (struct pci_bus*)bus;
 264         dev->sysdata = bus->sysdata;
 265         for (devfn = 0; devfn < 0x100; devfn += 8) {
 266                 dev->devfn = devfn;
 267                 found += pci_rescan_slot(dev);
 268         }
 269
 270         if (found) {
 271                 pci_bus_assign_resources(bus);
 272                 list_for_each_entry(dev, &bus->devices, bus_list) {
 273                         /* Skip already-added devices */
 274                         if (dev->is_added)
 275                                         continue;
 276                         retval = pci_bus_add_device(dev);
 277                         if (retval)
 278                                 dev_err(&dev->dev,
 279                                         "Error adding device, continuing\n");
 280                         else
 281                                 add_slot(dev);
 282                 }
 283                 pci_bus_add_devices(bus);
 284         }
 285         kfree(dev);
 286 }
 287
 288 /* recursively scan all buses */
 289 static void pci_rescan_buses(const struct list_head *list)
 290 {
 291         const struct list_head *l;
 292         list_for_each(l,list) {
 293                 const struct pci_bus *b = pci_bus_b(l);
 294                 pci_rescan_bus(b);
 295                 pci_rescan_buses(&b->children);
 296         }
 297 }
 298
 299 /* initiate rescan of all pci buses */
 300 static inline void pci_rescan(void) {
 301         pci_rescan_buses(&pci_root_buses);
 302 }
 303
 304 /* called from the single-threaded workqueue handler to rescan all pci buses */
 305 static void pci_rescan_worker(struct work_struct *work)
 306 {
 307         pci_rescan();
 308 }
 309
 310 static int enable_slot(struct hotplug_slot *hotplug_slot)
 311 {
 312         /* mis-use enable_slot for rescanning of the pci bus */
 313         cancel_work_sync(&pci_rescan_work);
 314         queue_work(dummyphp_wq, &pci_rescan_work);
 315         return 0;
 316 }
 317
 318 static int disable_slot(struct hotplug_slot *slot)
 319 {
 320         struct dummy_slot *dslot;
 321         struct pci_dev *dev;
 322         int func;
 323
 324         if (!slot)
 325                 return -ENODEV;
 326         dslot = slot->private;
 327
 328         dbg("%s - physical_slot = %s\n", __func__, hotplug_slot_name(slot));
 329
 330         for (func = 7; func >= 0; func--) {
 331                 dev = pci_get_slot(dslot->dev->bus, dslot->dev->devfn + func);
 332                 if (!dev)
 333                         continue;
 334
 335                 if (test_and_set_bit(0, &dslot->removed)) {
 336                         dbg("Slot already scheduled for removal\n");
 337                         pci_dev_put(dev);
 338                         return -ENODEV;
 339                 }
 340
 341                 /* remove the device from the pci core */
 342                 pci_remove_bus_device(dev);
 343
 344                 /* queue work item to blow away this sysfs entry and other
 345                  * parts.
 346                  */
 347                 INIT_WORK(&dslot->remove_work, remove_slot_worker);
 348                 queue_work(dummyphp_wq, &dslot->remove_work);
 349
 350                 pci_dev_put(dev);
 351         }
 352         return 0;
 353 }
 354
 355 static void cleanup_slots (void)
 356 {
 357         struct list_head *tmp;
 358         struct list_head *next;
 359         struct dummy_slot *dslot;
 360
 361         destroy_workqueue(dummyphp_wq);
 362         list_for_each_safe (tmp, next, &slot_list) {
 363                 dslot = list_entry (tmp, struct dummy_slot, node);
 364                 remove_slot(dslot);
 365         }
 366
 367 }
 368
 369 static int __init dummyphp_init(void)
 370 {
 371         info(DRIVER_DESC "\n");
 372
 373         dummyphp_wq = create_singlethread_workqueue(MY_NAME);
 374         if (!dummyphp_wq)
 375                 return -ENOMEM;
 376
 377         return pci_scan_buses();
 378 }
 379
 380
 381 static void __exit dummyphp_exit(void)
 382 {
 383         cleanup_slots();
 384 }
 385
 386 module_init(dummyphp_init);
 387 module_exit(dummyphp_exit);
 388
 389 MODULE_AUTHOR(DRIVER_AUTHOR);
 390 MODULE_DESCRIPTION(DRIVER_DESC);
 391 MODULE_LICENSE("GPL");
 392 module_param(debug, bool, S_IRUGO | S_IWUSR);
 393 MODULE_PARM_DESC(debug, "Debugging mode enabled or not");
 394 module_param(dup_slots, bool, S_IRUGO | S_IWUSR);
 395 MODULE_PARM_DESC(dup_slots, "Force duplicate slot names for debugging");