genhd.c 44 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816
  1. /*
  2. * gendisk handling
  3. */
  4. #include <linux/module.h>
  5. #include <linux/fs.h>
  6. #include <linux/genhd.h>
  7. #include <linux/kdev_t.h>
  8. #include <linux/kernel.h>
  9. #include <linux/blkdev.h>
  10. #include <linux/init.h>
  11. #include <linux/spinlock.h>
  12. #include <linux/proc_fs.h>
  13. #include <linux/seq_file.h>
  14. #include <linux/slab.h>
  15. #include <linux/kmod.h>
  16. #include <linux/kobj_map.h>
  17. #include <linux/mutex.h>
  18. #include <linux/idr.h>
  19. #include <linux/log2.h>
  20. #include "blk.h"
  21. static DEFINE_MUTEX(block_class_lock);
  22. struct kobject *block_depr;
  23. /* for extended dynamic devt allocation, currently only one major is used */
  24. #define MAX_EXT_DEVT (1 << MINORBITS)
  25. /* For extended devt allocation. ext_devt_mutex prevents look up
  26. * results from going away underneath its user.
  27. */
  28. static DEFINE_MUTEX(ext_devt_mutex);
  29. static DEFINE_IDR(ext_devt_idr);
  30. static struct device_type disk_type;
  31. static void disk_alloc_events(struct gendisk *disk);
  32. static void disk_add_events(struct gendisk *disk);
  33. static void disk_del_events(struct gendisk *disk);
  34. static void disk_release_events(struct gendisk *disk);
  35. /**
  36. * disk_get_part - get partition
  37. * @disk: disk to look partition from
  38. * @partno: partition number
  39. *
  40. * Look for partition @partno from @disk. If found, increment
  41. * reference count and return it.
  42. *
  43. * CONTEXT:
  44. * Don't care.
  45. *
  46. * RETURNS:
  47. * Pointer to the found partition on success, NULL if not found.
  48. */
  49. struct hd_struct *disk_get_part(struct gendisk *disk, int partno)
  50. {
  51. struct hd_struct *part = NULL;
  52. struct disk_part_tbl *ptbl;
  53. if (unlikely(partno < 0))
  54. return NULL;
  55. rcu_read_lock();
  56. ptbl = rcu_dereference(disk->part_tbl);
  57. if (likely(partno < ptbl->len)) {
  58. part = rcu_dereference(ptbl->part[partno]);
  59. if (part)
  60. get_device(part_to_dev(part));
  61. }
  62. rcu_read_unlock();
  63. return part;
  64. }
  65. EXPORT_SYMBOL_GPL(disk_get_part);
  66. /**
  67. * disk_part_iter_init - initialize partition iterator
  68. * @piter: iterator to initialize
  69. * @disk: disk to iterate over
  70. * @flags: DISK_PITER_* flags
  71. *
  72. * Initialize @piter so that it iterates over partitions of @disk.
  73. *
  74. * CONTEXT:
  75. * Don't care.
  76. */
  77. void disk_part_iter_init(struct disk_part_iter *piter, struct gendisk *disk,
  78. unsigned int flags)
  79. {
  80. struct disk_part_tbl *ptbl;
  81. rcu_read_lock();
  82. ptbl = rcu_dereference(disk->part_tbl);
  83. piter->disk = disk;
  84. piter->part = NULL;
  85. if (flags & DISK_PITER_REVERSE)
  86. piter->idx = ptbl->len - 1;
  87. else if (flags & (DISK_PITER_INCL_PART0 | DISK_PITER_INCL_EMPTY_PART0))
  88. piter->idx = 0;
  89. else
  90. piter->idx = 1;
  91. piter->flags = flags;
  92. rcu_read_unlock();
  93. }
  94. EXPORT_SYMBOL_GPL(disk_part_iter_init);
  95. /**
  96. * disk_part_iter_next - proceed iterator to the next partition and return it
  97. * @piter: iterator of interest
  98. *
  99. * Proceed @piter to the next partition and return it.
  100. *
  101. * CONTEXT:
  102. * Don't care.
  103. */
  104. struct hd_struct *disk_part_iter_next(struct disk_part_iter *piter)
  105. {
  106. struct disk_part_tbl *ptbl;
  107. int inc, end;
  108. /* put the last partition */
  109. disk_put_part(piter->part);
  110. piter->part = NULL;
  111. /* get part_tbl */
  112. rcu_read_lock();
  113. ptbl = rcu_dereference(piter->disk->part_tbl);
  114. /* determine iteration parameters */
  115. if (piter->flags & DISK_PITER_REVERSE) {
  116. inc = -1;
  117. if (piter->flags & (DISK_PITER_INCL_PART0 |
  118. DISK_PITER_INCL_EMPTY_PART0))
  119. end = -1;
  120. else
  121. end = 0;
  122. } else {
  123. inc = 1;
  124. end = ptbl->len;
  125. }
  126. /* iterate to the next partition */
  127. for (; piter->idx != end; piter->idx += inc) {
  128. struct hd_struct *part;
  129. part = rcu_dereference(ptbl->part[piter->idx]);
  130. if (!part)
  131. continue;
  132. if (!part_nr_sects_read(part) &&
  133. !(piter->flags & DISK_PITER_INCL_EMPTY) &&
  134. !(piter->flags & DISK_PITER_INCL_EMPTY_PART0 &&
  135. piter->idx == 0))
  136. continue;
  137. get_device(part_to_dev(part));
  138. piter->part = part;
  139. piter->idx += inc;
  140. break;
  141. }
  142. rcu_read_unlock();
  143. return piter->part;
  144. }
  145. EXPORT_SYMBOL_GPL(disk_part_iter_next);
  146. /**
  147. * disk_part_iter_exit - finish up partition iteration
  148. * @piter: iter of interest
  149. *
  150. * Called when iteration is over. Cleans up @piter.
  151. *
  152. * CONTEXT:
  153. * Don't care.
  154. */
  155. void disk_part_iter_exit(struct disk_part_iter *piter)
  156. {
  157. disk_put_part(piter->part);
  158. piter->part = NULL;
  159. }
  160. EXPORT_SYMBOL_GPL(disk_part_iter_exit);
  161. static inline int sector_in_part(struct hd_struct *part, sector_t sector)
  162. {
  163. return part->start_sect <= sector &&
  164. sector < part->start_sect + part_nr_sects_read(part);
  165. }
  166. /**
  167. * disk_map_sector_rcu - map sector to partition
  168. * @disk: gendisk of interest
  169. * @sector: sector to map
  170. *
  171. * Find out which partition @sector maps to on @disk. This is
  172. * primarily used for stats accounting.
  173. *
  174. * CONTEXT:
  175. * RCU read locked. The returned partition pointer is valid only
  176. * while preemption is disabled.
  177. *
  178. * RETURNS:
  179. * Found partition on success, part0 is returned if no partition matches
  180. */
  181. struct hd_struct *disk_map_sector_rcu(struct gendisk *disk, sector_t sector)
  182. {
  183. struct disk_part_tbl *ptbl;
  184. struct hd_struct *part;
  185. int i;
  186. ptbl = rcu_dereference(disk->part_tbl);
  187. part = rcu_dereference(ptbl->last_lookup);
  188. if (part && sector_in_part(part, sector))
  189. return part;
  190. for (i = 1; i < ptbl->len; i++) {
  191. part = rcu_dereference(ptbl->part[i]);
  192. if (part && sector_in_part(part, sector)) {
  193. rcu_assign_pointer(ptbl->last_lookup, part);
  194. return part;
  195. }
  196. }
  197. return &disk->part0;
  198. }
  199. EXPORT_SYMBOL_GPL(disk_map_sector_rcu);
  200. /*
  201. * Can be deleted altogether. Later.
  202. *
  203. */
  204. static struct blk_major_name {
  205. struct blk_major_name *next;
  206. int major;
  207. char name[16];
  208. } *major_names[BLKDEV_MAJOR_HASH_SIZE];
  209. /* index in the above - for now: assume no multimajor ranges */
  210. static inline int major_to_index(unsigned major)
  211. {
  212. return major % BLKDEV_MAJOR_HASH_SIZE;
  213. }
  214. #ifdef CONFIG_PROC_FS
  215. void blkdev_show(struct seq_file *seqf, off_t offset)
  216. {
  217. struct blk_major_name *dp;
  218. if (offset < BLKDEV_MAJOR_HASH_SIZE) {
  219. mutex_lock(&block_class_lock);
  220. for (dp = major_names[offset]; dp; dp = dp->next)
  221. seq_printf(seqf, "%3d %s\n", dp->major, dp->name);
  222. mutex_unlock(&block_class_lock);
  223. }
  224. }
  225. #endif /* CONFIG_PROC_FS */
  226. /**
  227. * register_blkdev - register a new block device
  228. *
  229. * @major: the requested major device number [1..255]. If @major=0, try to
  230. * allocate any unused major number.
  231. * @name: the name of the new block device as a zero terminated string
  232. *
  233. * The @name must be unique within the system.
  234. *
  235. * The return value depends on the @major input parameter.
  236. * - if a major device number was requested in range [1..255] then the
  237. * function returns zero on success, or a negative error code
  238. * - if any unused major number was requested with @major=0 parameter
  239. * then the return value is the allocated major number in range
  240. * [1..255] or a negative error code otherwise
  241. */
  242. int register_blkdev(unsigned int major, const char *name)
  243. {
  244. struct blk_major_name **n, *p;
  245. int index, ret = 0;
  246. mutex_lock(&block_class_lock);
  247. /* temporary */
  248. if (major == 0) {
  249. for (index = ARRAY_SIZE(major_names)-1; index > 0; index--) {
  250. if (major_names[index] == NULL)
  251. break;
  252. }
  253. if (index == 0) {
  254. printk("register_blkdev: failed to get major for %s\n",
  255. name);
  256. ret = -EBUSY;
  257. goto out;
  258. }
  259. major = index;
  260. ret = major;
  261. }
  262. p = kmalloc(sizeof(struct blk_major_name), GFP_KERNEL);
  263. if (p == NULL) {
  264. ret = -ENOMEM;
  265. goto out;
  266. }
  267. p->major = major;
  268. strlcpy(p->name, name, sizeof(p->name));
  269. p->next = NULL;
  270. index = major_to_index(major);
  271. for (n = &major_names[index]; *n; n = &(*n)->next) {
  272. if ((*n)->major == major)
  273. break;
  274. }
  275. if (!*n)
  276. *n = p;
  277. else
  278. ret = -EBUSY;
  279. if (ret < 0) {
  280. printk("register_blkdev: cannot get major %d for %s\n",
  281. major, name);
  282. kfree(p);
  283. }
  284. out:
  285. mutex_unlock(&block_class_lock);
  286. return ret;
  287. }
  288. EXPORT_SYMBOL(register_blkdev);
  289. void unregister_blkdev(unsigned int major, const char *name)
  290. {
  291. struct blk_major_name **n;
  292. struct blk_major_name *p = NULL;
  293. int index = major_to_index(major);
  294. mutex_lock(&block_class_lock);
  295. for (n = &major_names[index]; *n; n = &(*n)->next)
  296. if ((*n)->major == major)
  297. break;
  298. if (!*n || strcmp((*n)->name, name)) {
  299. WARN_ON(1);
  300. } else {
  301. p = *n;
  302. *n = p->next;
  303. }
  304. mutex_unlock(&block_class_lock);
  305. kfree(p);
  306. }
  307. EXPORT_SYMBOL(unregister_blkdev);
  308. static struct kobj_map *bdev_map;
  309. /**
  310. * blk_mangle_minor - scatter minor numbers apart
  311. * @minor: minor number to mangle
  312. *
  313. * Scatter consecutively allocated @minor number apart if MANGLE_DEVT
  314. * is enabled. Mangling twice gives the original value.
  315. *
  316. * RETURNS:
  317. * Mangled value.
  318. *
  319. * CONTEXT:
  320. * Don't care.
  321. */
  322. static int blk_mangle_minor(int minor)
  323. {
  324. #ifdef CONFIG_DEBUG_BLOCK_EXT_DEVT
  325. int i;
  326. for (i = 0; i < MINORBITS / 2; i++) {
  327. int low = minor & (1 << i);
  328. int high = minor & (1 << (MINORBITS - 1 - i));
  329. int distance = MINORBITS - 1 - 2 * i;
  330. minor ^= low | high; /* clear both bits */
  331. low <<= distance; /* swap the positions */
  332. high >>= distance;
  333. minor |= low | high; /* and set */
  334. }
  335. #endif
  336. return minor;
  337. }
  338. /**
  339. * blk_alloc_devt - allocate a dev_t for a partition
  340. * @part: partition to allocate dev_t for
  341. * @devt: out parameter for resulting dev_t
  342. *
  343. * Allocate a dev_t for block device.
  344. *
  345. * RETURNS:
  346. * 0 on success, allocated dev_t is returned in *@devt. -errno on
  347. * failure.
  348. *
  349. * CONTEXT:
  350. * Might sleep.
  351. */
  352. int blk_alloc_devt(struct hd_struct *part, dev_t *devt)
  353. {
  354. struct gendisk *disk = part_to_disk(part);
  355. int idx, rc;
  356. /* in consecutive minor range? */
  357. if (part->partno < disk->minors) {
  358. *devt = MKDEV(disk->major, disk->first_minor + part->partno);
  359. return 0;
  360. }
  361. /* allocate ext devt */
  362. do {
  363. if (!idr_pre_get(&ext_devt_idr, GFP_KERNEL))
  364. return -ENOMEM;
  365. rc = idr_get_new(&ext_devt_idr, part, &idx);
  366. } while (rc == -EAGAIN);
  367. if (rc)
  368. return rc;
  369. if (idx > MAX_EXT_DEVT) {
  370. idr_remove(&ext_devt_idr, idx);
  371. return -EBUSY;
  372. }
  373. *devt = MKDEV(BLOCK_EXT_MAJOR, blk_mangle_minor(idx));
  374. return 0;
  375. }
  376. /**
  377. * blk_free_devt - free a dev_t
  378. * @devt: dev_t to free
  379. *
  380. * Free @devt which was allocated using blk_alloc_devt().
  381. *
  382. * CONTEXT:
  383. * Might sleep.
  384. */
  385. void blk_free_devt(dev_t devt)
  386. {
  387. might_sleep();
  388. if (devt == MKDEV(0, 0))
  389. return;
  390. if (MAJOR(devt) == BLOCK_EXT_MAJOR) {
  391. mutex_lock(&ext_devt_mutex);
  392. idr_remove(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
  393. mutex_unlock(&ext_devt_mutex);
  394. }
  395. }
  396. static char *bdevt_str(dev_t devt, char *buf)
  397. {
  398. if (MAJOR(devt) <= 0xff && MINOR(devt) <= 0xff) {
  399. char tbuf[BDEVT_SIZE];
  400. snprintf(tbuf, BDEVT_SIZE, "%02x%02x", MAJOR(devt), MINOR(devt));
  401. snprintf(buf, BDEVT_SIZE, "%-9s", tbuf);
  402. } else
  403. snprintf(buf, BDEVT_SIZE, "%03x:%05x", MAJOR(devt), MINOR(devt));
  404. return buf;
  405. }
  406. /*
  407. * Register device numbers dev..(dev+range-1)
  408. * range must be nonzero
  409. * The hash chain is sorted on range, so that subranges can override.
  410. */
  411. void blk_register_region(dev_t devt, unsigned long range, struct module *module,
  412. struct kobject *(*probe)(dev_t, int *, void *),
  413. int (*lock)(dev_t, void *), void *data)
  414. {
  415. kobj_map(bdev_map, devt, range, module, probe, lock, data);
  416. }
  417. EXPORT_SYMBOL(blk_register_region);
  418. void blk_unregister_region(dev_t devt, unsigned long range)
  419. {
  420. kobj_unmap(bdev_map, devt, range);
  421. }
  422. EXPORT_SYMBOL(blk_unregister_region);
  423. static struct kobject *exact_match(dev_t devt, int *partno, void *data)
  424. {
  425. struct gendisk *p = data;
  426. return &disk_to_dev(p)->kobj;
  427. }
  428. static int exact_lock(dev_t devt, void *data)
  429. {
  430. struct gendisk *p = data;
  431. if (!get_disk(p))
  432. return -1;
  433. return 0;
  434. }
  435. static void register_disk(struct gendisk *disk)
  436. {
  437. struct device *ddev = disk_to_dev(disk);
  438. struct block_device *bdev;
  439. struct disk_part_iter piter;
  440. struct hd_struct *part;
  441. int err;
  442. ddev->parent = disk->driverfs_dev;
  443. dev_set_name(ddev, disk->disk_name);
  444. /* delay uevents, until we scanned partition table */
  445. dev_set_uevent_suppress(ddev, 1);
  446. if (device_add(ddev))
  447. return;
  448. if (!sysfs_deprecated) {
  449. err = sysfs_create_link(block_depr, &ddev->kobj,
  450. kobject_name(&ddev->kobj));
  451. if (err) {
  452. device_del(ddev);
  453. return;
  454. }
  455. }
  456. disk->part0.holder_dir = kobject_create_and_add("holders", &ddev->kobj);
  457. disk->slave_dir = kobject_create_and_add("slaves", &ddev->kobj);
  458. /* No minors to use for partitions */
  459. if (!disk_part_scan_enabled(disk))
  460. goto exit;
  461. /* No such device (e.g., media were just removed) */
  462. if (!get_capacity(disk))
  463. goto exit;
  464. bdev = bdget_disk(disk, 0);
  465. if (!bdev)
  466. goto exit;
  467. bdev->bd_invalidated = 1;
  468. err = blkdev_get(bdev, FMODE_READ, NULL);
  469. if (err < 0)
  470. goto exit;
  471. blkdev_put(bdev, FMODE_READ);
  472. exit:
  473. /* announce disk after possible partitions are created */
  474. dev_set_uevent_suppress(ddev, 0);
  475. kobject_uevent(&ddev->kobj, KOBJ_ADD);
  476. /* announce possible partitions */
  477. disk_part_iter_init(&piter, disk, 0);
  478. while ((part = disk_part_iter_next(&piter)))
  479. kobject_uevent(&part_to_dev(part)->kobj, KOBJ_ADD);
  480. disk_part_iter_exit(&piter);
  481. }
  482. /**
  483. * add_disk - add partitioning information to kernel list
  484. * @disk: per-device partitioning information
  485. *
  486. * This function registers the partitioning information in @disk
  487. * with the kernel.
  488. *
  489. * FIXME: error handling
  490. */
  491. void add_disk(struct gendisk *disk)
  492. {
  493. struct backing_dev_info *bdi;
  494. dev_t devt;
  495. int retval;
  496. /* minors == 0 indicates to use ext devt from part0 and should
  497. * be accompanied with EXT_DEVT flag. Make sure all
  498. * parameters make sense.
  499. */
  500. WARN_ON(disk->minors && !(disk->major || disk->first_minor));
  501. WARN_ON(!disk->minors && !(disk->flags & GENHD_FL_EXT_DEVT));
  502. disk->flags |= GENHD_FL_UP;
  503. retval = blk_alloc_devt(&disk->part0, &devt);
  504. if (retval) {
  505. WARN_ON(1);
  506. return;
  507. }
  508. disk_to_dev(disk)->devt = devt;
  509. /* ->major and ->first_minor aren't supposed to be
  510. * dereferenced from here on, but set them just in case.
  511. */
  512. disk->major = MAJOR(devt);
  513. disk->first_minor = MINOR(devt);
  514. disk_alloc_events(disk);
  515. /* Register BDI before referencing it from bdev */
  516. bdi = &disk->queue->backing_dev_info;
  517. bdi_register_dev(bdi, disk_devt(disk));
  518. blk_register_region(disk_devt(disk), disk->minors, NULL,
  519. exact_match, exact_lock, disk);
  520. register_disk(disk);
  521. blk_register_queue(disk);
  522. /*
  523. * Take an extra ref on queue which will be put on disk_release()
  524. * so that it sticks around as long as @disk is there.
  525. */
  526. WARN_ON_ONCE(!blk_get_queue(disk->queue));
  527. retval = sysfs_create_link(&disk_to_dev(disk)->kobj, &bdi->dev->kobj,
  528. "bdi");
  529. WARN_ON(retval);
  530. disk_add_events(disk);
  531. }
  532. EXPORT_SYMBOL(add_disk);
  533. void del_gendisk(struct gendisk *disk)
  534. {
  535. struct disk_part_iter piter;
  536. struct hd_struct *part;
  537. disk_del_events(disk);
  538. /* invalidate stuff */
  539. disk_part_iter_init(&piter, disk,
  540. DISK_PITER_INCL_EMPTY | DISK_PITER_REVERSE);
  541. while ((part = disk_part_iter_next(&piter))) {
  542. invalidate_partition(disk, part->partno);
  543. delete_partition(disk, part->partno);
  544. }
  545. disk_part_iter_exit(&piter);
  546. invalidate_partition(disk, 0);
  547. blk_free_devt(disk_to_dev(disk)->devt);
  548. set_capacity(disk, 0);
  549. disk->flags &= ~GENHD_FL_UP;
  550. sysfs_remove_link(&disk_to_dev(disk)->kobj, "bdi");
  551. bdi_unregister(&disk->queue->backing_dev_info);
  552. blk_unregister_queue(disk);
  553. blk_unregister_region(disk_devt(disk), disk->minors);
  554. part_stat_set_all(&disk->part0, 0);
  555. disk->part0.stamp = 0;
  556. kobject_put(disk->part0.holder_dir);
  557. kobject_put(disk->slave_dir);
  558. disk->driverfs_dev = NULL;
  559. if (!sysfs_deprecated)
  560. sysfs_remove_link(block_depr, dev_name(disk_to_dev(disk)));
  561. device_del(disk_to_dev(disk));
  562. }
  563. EXPORT_SYMBOL(del_gendisk);
  564. /**
  565. * get_gendisk - get partitioning information for a given device
  566. * @devt: device to get partitioning information for
  567. * @partno: returned partition index
  568. *
  569. * This function gets the structure containing partitioning
  570. * information for the given device @devt.
  571. */
  572. struct gendisk *get_gendisk(dev_t devt, int *partno)
  573. {
  574. struct gendisk *disk = NULL;
  575. if (MAJOR(devt) != BLOCK_EXT_MAJOR) {
  576. struct kobject *kobj;
  577. kobj = kobj_lookup(bdev_map, devt, partno);
  578. if (kobj)
  579. disk = dev_to_disk(kobj_to_dev(kobj));
  580. } else {
  581. struct hd_struct *part;
  582. mutex_lock(&ext_devt_mutex);
  583. part = idr_find(&ext_devt_idr, blk_mangle_minor(MINOR(devt)));
  584. if (part && get_disk(part_to_disk(part))) {
  585. *partno = part->partno;
  586. disk = part_to_disk(part);
  587. }
  588. mutex_unlock(&ext_devt_mutex);
  589. }
  590. return disk;
  591. }
  592. EXPORT_SYMBOL(get_gendisk);
  593. /**
  594. * bdget_disk - do bdget() by gendisk and partition number
  595. * @disk: gendisk of interest
  596. * @partno: partition number
  597. *
  598. * Find partition @partno from @disk, do bdget() on it.
  599. *
  600. * CONTEXT:
  601. * Don't care.
  602. *
  603. * RETURNS:
  604. * Resulting block_device on success, NULL on failure.
  605. */
  606. struct block_device *bdget_disk(struct gendisk *disk, int partno)
  607. {
  608. struct hd_struct *part;
  609. struct block_device *bdev = NULL;
  610. part = disk_get_part(disk, partno);
  611. if (part)
  612. bdev = bdget(part_devt(part));
  613. disk_put_part(part);
  614. return bdev;
  615. }
  616. EXPORT_SYMBOL(bdget_disk);
  617. /*
  618. * print a full list of all partitions - intended for places where the root
  619. * filesystem can't be mounted and thus to give the victim some idea of what
  620. * went wrong
  621. */
  622. void __init printk_all_partitions(void)
  623. {
  624. struct class_dev_iter iter;
  625. struct device *dev;
  626. class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
  627. while ((dev = class_dev_iter_next(&iter))) {
  628. struct gendisk *disk = dev_to_disk(dev);
  629. struct disk_part_iter piter;
  630. struct hd_struct *part;
  631. char name_buf[BDEVNAME_SIZE];
  632. char devt_buf[BDEVT_SIZE];
  633. /*
  634. * Don't show empty devices or things that have been
  635. * suppressed
  636. */
  637. if (get_capacity(disk) == 0 ||
  638. (disk->flags & GENHD_FL_SUPPRESS_PARTITION_INFO))
  639. continue;
  640. /*
  641. * Note, unlike /proc/partitions, I am showing the
  642. * numbers in hex - the same format as the root=
  643. * option takes.
  644. */
  645. disk_part_iter_init(&piter, disk, DISK_PITER_INCL_PART0);
  646. while ((part = disk_part_iter_next(&piter))) {
  647. bool is_part0 = part == &disk->part0;
  648. printk("%s%s %10llu %s %s", is_part0 ? "" : " ",
  649. bdevt_str(part_devt(part), devt_buf),
  650. (unsigned long long)part_nr_sects_read(part) >> 1
  651. , disk_name(disk, part->partno, name_buf),
  652. part->info ? part->info->uuid : "");
  653. if (is_part0) {
  654. if (disk->driverfs_dev != NULL &&
  655. disk->driverfs_dev->driver != NULL)
  656. printk(" driver: %s\n",
  657. disk->driverfs_dev->driver->name);
  658. else
  659. printk(" (driver?)\n");
  660. } else
  661. printk("\n");
  662. }
  663. disk_part_iter_exit(&piter);
  664. }
  665. class_dev_iter_exit(&iter);
  666. }
  667. #ifdef CONFIG_PROC_FS
  668. /* iterator */
  669. static void *disk_seqf_start(struct seq_file *seqf, loff_t *pos)
  670. {
  671. loff_t skip = *pos;
  672. struct class_dev_iter *iter;
  673. struct device *dev;
  674. iter = kmalloc(sizeof(*iter), GFP_KERNEL);
  675. if (!iter)
  676. return ERR_PTR(-ENOMEM);
  677. seqf->private = iter;
  678. class_dev_iter_init(iter, &block_class, NULL, &disk_type);
  679. do {
  680. dev = class_dev_iter_next(iter);
  681. if (!dev)
  682. return NULL;
  683. } while (skip--);
  684. return dev_to_disk(dev);
  685. }
  686. static void *disk_seqf_next(struct seq_file *seqf, void *v, loff_t *pos)
  687. {
  688. struct device *dev;
  689. (*pos)++;
  690. dev = class_dev_iter_next(seqf->private);
  691. if (dev)
  692. return dev_to_disk(dev);
  693. return NULL;
  694. }
  695. static void disk_seqf_stop(struct seq_file *seqf, void *v)
  696. {
  697. struct class_dev_iter *iter = seqf->private;
  698. /* stop is called even after start failed :-( */
  699. if (iter) {
  700. class_dev_iter_exit(iter);
  701. kfree(iter);
  702. }
  703. }
  704. static void *show_partition_start(struct seq_file *seqf, loff_t *pos)
  705. {
  706. void *p;
  707. p = disk_seqf_start(seqf, pos);
  708. if (!IS_ERR_OR_NULL(p) && !*pos)
  709. seq_puts(seqf, "major minor #blocks name\n\n");
  710. return p;
  711. }
  712. static int show_partition(struct seq_file *seqf, void *v)
  713. {
  714. struct gendisk *sgp = v;
  715. struct disk_part_iter piter;
  716. struct hd_struct *part;
  717. char buf[BDEVNAME_SIZE];
  718. /* Don't show non-partitionable removeable devices or empty devices */
  719. if (!get_capacity(sgp) || (!disk_max_parts(sgp) &&
  720. (sgp->flags & GENHD_FL_REMOVABLE)))
  721. return 0;
  722. if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
  723. return 0;
  724. /* show the full disk and all non-0 size partitions of it */
  725. disk_part_iter_init(&piter, sgp, DISK_PITER_INCL_PART0);
  726. while ((part = disk_part_iter_next(&piter)))
  727. seq_printf(seqf, "%4d %7d %10llu %s\n",
  728. MAJOR(part_devt(part)), MINOR(part_devt(part)),
  729. (unsigned long long)part_nr_sects_read(part) >> 1,
  730. disk_name(sgp, part->partno, buf));
  731. disk_part_iter_exit(&piter);
  732. return 0;
  733. }
  734. static const struct seq_operations partitions_op = {
  735. .start = show_partition_start,
  736. .next = disk_seqf_next,
  737. .stop = disk_seqf_stop,
  738. .show = show_partition
  739. };
  740. static int partitions_open(struct inode *inode, struct file *file)
  741. {
  742. return seq_open(file, &partitions_op);
  743. }
  744. static const struct file_operations proc_partitions_operations = {
  745. .open = partitions_open,
  746. .read = seq_read,
  747. .llseek = seq_lseek,
  748. .release = seq_release,
  749. };
  750. #endif
  751. static struct kobject *base_probe(dev_t devt, int *partno, void *data)
  752. {
  753. if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
  754. /* Make old-style 2.4 aliases work */
  755. request_module("block-major-%d", MAJOR(devt));
  756. return NULL;
  757. }
  758. static int __init genhd_device_init(void)
  759. {
  760. int error;
  761. block_class.dev_kobj = sysfs_dev_block_kobj;
  762. error = class_register(&block_class);
  763. if (unlikely(error))
  764. return error;
  765. bdev_map = kobj_map_init(base_probe, &block_class_lock);
  766. blk_dev_init();
  767. register_blkdev(BLOCK_EXT_MAJOR, "blkext");
  768. /* create top-level block dir */
  769. if (!sysfs_deprecated)
  770. block_depr = kobject_create_and_add("block", NULL);
  771. return 0;
  772. }
  773. subsys_initcall(genhd_device_init);
  774. static ssize_t disk_range_show(struct device *dev,
  775. struct device_attribute *attr, char *buf)
  776. {
  777. struct gendisk *disk = dev_to_disk(dev);
  778. return sprintf(buf, "%d\n", disk->minors);
  779. }
  780. static ssize_t disk_ext_range_show(struct device *dev,
  781. struct device_attribute *attr, char *buf)
  782. {
  783. struct gendisk *disk = dev_to_disk(dev);
  784. return sprintf(buf, "%d\n", disk_max_parts(disk));
  785. }
  786. static ssize_t disk_removable_show(struct device *dev,
  787. struct device_attribute *attr, char *buf)
  788. {
  789. struct gendisk *disk = dev_to_disk(dev);
  790. return sprintf(buf, "%d\n",
  791. (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
  792. }
  793. static ssize_t disk_ro_show(struct device *dev,
  794. struct device_attribute *attr, char *buf)
  795. {
  796. struct gendisk *disk = dev_to_disk(dev);
  797. return sprintf(buf, "%d\n", get_disk_ro(disk) ? 1 : 0);
  798. }
  799. static ssize_t disk_capability_show(struct device *dev,
  800. struct device_attribute *attr, char *buf)
  801. {
  802. struct gendisk *disk = dev_to_disk(dev);
  803. return sprintf(buf, "%x\n", disk->flags);
  804. }
  805. static ssize_t disk_alignment_offset_show(struct device *dev,
  806. struct device_attribute *attr,
  807. char *buf)
  808. {
  809. struct gendisk *disk = dev_to_disk(dev);
  810. return sprintf(buf, "%d\n", queue_alignment_offset(disk->queue));
  811. }
  812. static ssize_t disk_discard_alignment_show(struct device *dev,
  813. struct device_attribute *attr,
  814. char *buf)
  815. {
  816. struct gendisk *disk = dev_to_disk(dev);
  817. return sprintf(buf, "%d\n", queue_discard_alignment(disk->queue));
  818. }
  819. static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
  820. static DEVICE_ATTR(ext_range, S_IRUGO, disk_ext_range_show, NULL);
  821. static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
  822. static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
  823. static DEVICE_ATTR(size, S_IRUGO, part_size_show, NULL);
  824. static DEVICE_ATTR(alignment_offset, S_IRUGO, disk_alignment_offset_show, NULL);
  825. static DEVICE_ATTR(discard_alignment, S_IRUGO, disk_discard_alignment_show,
  826. NULL);
  827. static DEVICE_ATTR(capability, S_IRUGO, disk_capability_show, NULL);
  828. static DEVICE_ATTR(stat, S_IRUGO, part_stat_show, NULL);
  829. static DEVICE_ATTR(inflight, S_IRUGO, part_inflight_show, NULL);
  830. #ifdef CONFIG_FAIL_MAKE_REQUEST
  831. static struct device_attribute dev_attr_fail =
  832. __ATTR(make-it-fail, S_IRUGO|S_IWUSR, part_fail_show, part_fail_store);
  833. #endif
  834. #ifdef CONFIG_FAIL_IO_TIMEOUT
  835. static struct device_attribute dev_attr_fail_timeout =
  836. __ATTR(io-timeout-fail, S_IRUGO|S_IWUSR, part_timeout_show,
  837. part_timeout_store);
  838. #endif
  839. static struct attribute *disk_attrs[] = {
  840. &dev_attr_range.attr,
  841. &dev_attr_ext_range.attr,
  842. &dev_attr_removable.attr,
  843. &dev_attr_ro.attr,
  844. &dev_attr_size.attr,
  845. &dev_attr_alignment_offset.attr,
  846. &dev_attr_discard_alignment.attr,
  847. &dev_attr_capability.attr,
  848. &dev_attr_stat.attr,
  849. &dev_attr_inflight.attr,
  850. #ifdef CONFIG_FAIL_MAKE_REQUEST
  851. &dev_attr_fail.attr,
  852. #endif
  853. #ifdef CONFIG_FAIL_IO_TIMEOUT
  854. &dev_attr_fail_timeout.attr,
  855. #endif
  856. NULL
  857. };
  858. static struct attribute_group disk_attr_group = {
  859. .attrs = disk_attrs,
  860. };
  861. static const struct attribute_group *disk_attr_groups[] = {
  862. &disk_attr_group,
  863. NULL
  864. };
  865. /**
  866. * disk_replace_part_tbl - replace disk->part_tbl in RCU-safe way
  867. * @disk: disk to replace part_tbl for
  868. * @new_ptbl: new part_tbl to install
  869. *
  870. * Replace disk->part_tbl with @new_ptbl in RCU-safe way. The
  871. * original ptbl is freed using RCU callback.
  872. *
  873. * LOCKING:
  874. * Matching bd_mutx locked.
  875. */
  876. static void disk_replace_part_tbl(struct gendisk *disk,
  877. struct disk_part_tbl *new_ptbl)
  878. {
  879. struct disk_part_tbl *old_ptbl = disk->part_tbl;
  880. rcu_assign_pointer(disk->part_tbl, new_ptbl);
  881. if (old_ptbl) {
  882. rcu_assign_pointer(old_ptbl->last_lookup, NULL);
  883. kfree_rcu(old_ptbl, rcu_head);
  884. }
  885. }
  886. /**
  887. * disk_expand_part_tbl - expand disk->part_tbl
  888. * @disk: disk to expand part_tbl for
  889. * @partno: expand such that this partno can fit in
  890. *
  891. * Expand disk->part_tbl such that @partno can fit in. disk->part_tbl
  892. * uses RCU to allow unlocked dereferencing for stats and other stuff.
  893. *
  894. * LOCKING:
  895. * Matching bd_mutex locked, might sleep.
  896. *
  897. * RETURNS:
  898. * 0 on success, -errno on failure.
  899. */
  900. int disk_expand_part_tbl(struct gendisk *disk, int partno)
  901. {
  902. struct disk_part_tbl *old_ptbl = disk->part_tbl;
  903. struct disk_part_tbl *new_ptbl;
  904. int len = old_ptbl ? old_ptbl->len : 0;
  905. int target = partno + 1;
  906. size_t size;
  907. int i;
  908. /* disk_max_parts() is zero during initialization, ignore if so */
  909. if (disk_max_parts(disk) && target > disk_max_parts(disk))
  910. return -EINVAL;
  911. if (target <= len)
  912. return 0;
  913. size = sizeof(*new_ptbl) + target * sizeof(new_ptbl->part[0]);
  914. new_ptbl = kzalloc_node(size, GFP_KERNEL, disk->node_id);
  915. if (!new_ptbl)
  916. return -ENOMEM;
  917. new_ptbl->len = target;
  918. for (i = 0; i < len; i++)
  919. rcu_assign_pointer(new_ptbl->part[i], old_ptbl->part[i]);
  920. disk_replace_part_tbl(disk, new_ptbl);
  921. return 0;
  922. }
  923. static void disk_release(struct device *dev)
  924. {
  925. struct gendisk *disk = dev_to_disk(dev);
  926. disk_release_events(disk);
  927. kfree(disk->random);
  928. disk_replace_part_tbl(disk, NULL);
  929. free_part_stats(&disk->part0);
  930. free_part_info(&disk->part0);
  931. if (disk->queue)
  932. blk_put_queue(disk->queue);
  933. kfree(disk);
  934. }
  935. struct class block_class = {
  936. .name = "block",
  937. };
  938. static char *block_devnode(struct device *dev, umode_t *mode)
  939. {
  940. struct gendisk *disk = dev_to_disk(dev);
  941. if (disk->devnode)
  942. return disk->devnode(disk, mode);
  943. return NULL;
  944. }
  945. static struct device_type disk_type = {
  946. .name = "disk",
  947. .groups = disk_attr_groups,
  948. .release = disk_release,
  949. .devnode = block_devnode,
  950. };
  951. #ifdef CONFIG_PROC_FS
  952. /*
  953. * aggregate disk stat collector. Uses the same stats that the sysfs
  954. * entries do, above, but makes them available through one seq_file.
  955. *
  956. * The output looks suspiciously like /proc/partitions with a bunch of
  957. * extra fields.
  958. */
  959. static int diskstats_show(struct seq_file *seqf, void *v)
  960. {
  961. struct gendisk *gp = v;
  962. struct disk_part_iter piter;
  963. struct hd_struct *hd;
  964. char buf[BDEVNAME_SIZE];
  965. int cpu;
  966. /*
  967. if (&disk_to_dev(gp)->kobj.entry == block_class.devices.next)
  968. seq_puts(seqf, "major minor name"
  969. " rio rmerge rsect ruse wio wmerge "
  970. "wsect wuse running use aveq"
  971. "\n\n");
  972. */
  973. disk_part_iter_init(&piter, gp, DISK_PITER_INCL_EMPTY_PART0);
  974. while ((hd = disk_part_iter_next(&piter))) {
  975. cpu = part_stat_lock();
  976. part_round_stats(cpu, hd);
  977. part_stat_unlock();
  978. seq_printf(seqf, "%4d %7d %s %lu %lu %lu "
  979. "%u %lu %lu %lu %u %u %u %u\n",
  980. MAJOR(part_devt(hd)), MINOR(part_devt(hd)),
  981. disk_name(gp, hd->partno, buf),
  982. part_stat_read(hd, ios[READ]),
  983. part_stat_read(hd, merges[READ]),
  984. part_stat_read(hd, sectors[READ]),
  985. jiffies_to_msecs(part_stat_read(hd, ticks[READ])),
  986. part_stat_read(hd, ios[WRITE]),
  987. part_stat_read(hd, merges[WRITE]),
  988. part_stat_read(hd, sectors[WRITE]),
  989. jiffies_to_msecs(part_stat_read(hd, ticks[WRITE])),
  990. part_in_flight(hd),
  991. jiffies_to_msecs(part_stat_read(hd, io_ticks)),
  992. jiffies_to_msecs(part_stat_read(hd, time_in_queue))
  993. );
  994. }
  995. disk_part_iter_exit(&piter);
  996. return 0;
  997. }
  998. static const struct seq_operations diskstats_op = {
  999. .start = disk_seqf_start,
  1000. .next = disk_seqf_next,
  1001. .stop = disk_seqf_stop,
  1002. .show = diskstats_show
  1003. };
  1004. static int diskstats_open(struct inode *inode, struct file *file)
  1005. {
  1006. return seq_open(file, &diskstats_op);
  1007. }
  1008. static const struct file_operations proc_diskstats_operations = {
  1009. .open = diskstats_open,
  1010. .read = seq_read,
  1011. .llseek = seq_lseek,
  1012. .release = seq_release,
  1013. };
  1014. static int __init proc_genhd_init(void)
  1015. {
  1016. proc_create("diskstats", 0, NULL, &proc_diskstats_operations);
  1017. proc_create("partitions", 0, NULL, &proc_partitions_operations);
  1018. return 0;
  1019. }
  1020. module_init(proc_genhd_init);
  1021. #endif /* CONFIG_PROC_FS */
  1022. dev_t blk_lookup_devt(const char *name, int partno)
  1023. {
  1024. dev_t devt = MKDEV(0, 0);
  1025. struct class_dev_iter iter;
  1026. struct device *dev;
  1027. class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
  1028. while ((dev = class_dev_iter_next(&iter))) {
  1029. struct gendisk *disk = dev_to_disk(dev);
  1030. struct hd_struct *part;
  1031. if (strcmp(dev_name(dev), name))
  1032. continue;
  1033. if (partno < disk->minors) {
  1034. /* We need to return the right devno, even
  1035. * if the partition doesn't exist yet.
  1036. */
  1037. devt = MKDEV(MAJOR(dev->devt),
  1038. MINOR(dev->devt) + partno);
  1039. break;
  1040. }
  1041. part = disk_get_part(disk, partno);
  1042. if (part) {
  1043. devt = part_devt(part);
  1044. disk_put_part(part);
  1045. break;
  1046. }
  1047. disk_put_part(part);
  1048. }
  1049. class_dev_iter_exit(&iter);
  1050. return devt;
  1051. }
  1052. EXPORT_SYMBOL(blk_lookup_devt);
  1053. struct gendisk *alloc_disk(int minors)
  1054. {
  1055. return alloc_disk_node(minors, NUMA_NO_NODE);
  1056. }
  1057. EXPORT_SYMBOL(alloc_disk);
  1058. struct gendisk *alloc_disk_node(int minors, int node_id)
  1059. {
  1060. struct gendisk *disk;
  1061. disk = kmalloc_node(sizeof(struct gendisk),
  1062. GFP_KERNEL | __GFP_ZERO, node_id);
  1063. if (disk) {
  1064. if (!init_part_stats(&disk->part0)) {
  1065. kfree(disk);
  1066. return NULL;
  1067. }
  1068. disk->node_id = node_id;
  1069. if (disk_expand_part_tbl(disk, 0)) {
  1070. free_part_stats(&disk->part0);
  1071. kfree(disk);
  1072. return NULL;
  1073. }
  1074. disk->part_tbl->part[0] = &disk->part0;
  1075. /*
  1076. * set_capacity() and get_capacity() currently don't use
  1077. * seqcounter to read/update the part0->nr_sects. Still init
  1078. * the counter as we can read the sectors in IO submission
  1079. * patch using seqence counters.
  1080. *
  1081. * TODO: Ideally set_capacity() and get_capacity() should be
  1082. * converted to make use of bd_mutex and sequence counters.
  1083. */
  1084. seqcount_init(&disk->part0.nr_sects_seq);
  1085. hd_ref_init(&disk->part0);
  1086. disk->minors = minors;
  1087. rand_initialize_disk(disk);
  1088. disk_to_dev(disk)->class = &block_class;
  1089. disk_to_dev(disk)->type = &disk_type;
  1090. device_initialize(disk_to_dev(disk));
  1091. }
  1092. return disk;
  1093. }
  1094. EXPORT_SYMBOL(alloc_disk_node);
  1095. struct kobject *get_disk(struct gendisk *disk)
  1096. {
  1097. struct module *owner;
  1098. struct kobject *kobj;
  1099. if (!disk->fops)
  1100. return NULL;
  1101. owner = disk->fops->owner;
  1102. if (owner && !try_module_get(owner))
  1103. return NULL;
  1104. kobj = kobject_get(&disk_to_dev(disk)->kobj);
  1105. if (kobj == NULL) {
  1106. module_put(owner);
  1107. return NULL;
  1108. }
  1109. return kobj;
  1110. }
  1111. EXPORT_SYMBOL(get_disk);
  1112. void put_disk(struct gendisk *disk)
  1113. {
  1114. if (disk)
  1115. kobject_put(&disk_to_dev(disk)->kobj);
  1116. }
  1117. EXPORT_SYMBOL(put_disk);
  1118. static void set_disk_ro_uevent(struct gendisk *gd, int ro)
  1119. {
  1120. char event[] = "DISK_RO=1";
  1121. char *envp[] = { event, NULL };
  1122. if (!ro)
  1123. event[8] = '0';
  1124. kobject_uevent_env(&disk_to_dev(gd)->kobj, KOBJ_CHANGE, envp);
  1125. }
  1126. void set_device_ro(struct block_device *bdev, int flag)
  1127. {
  1128. bdev->bd_part->policy = flag;
  1129. }
  1130. EXPORT_SYMBOL(set_device_ro);
  1131. void set_disk_ro(struct gendisk *disk, int flag)
  1132. {
  1133. struct disk_part_iter piter;
  1134. struct hd_struct *part;
  1135. if (disk->part0.policy != flag) {
  1136. set_disk_ro_uevent(disk, flag);
  1137. disk->part0.policy = flag;
  1138. }
  1139. disk_part_iter_init(&piter, disk, DISK_PITER_INCL_EMPTY);
  1140. while ((part = disk_part_iter_next(&piter)))
  1141. part->policy = flag;
  1142. disk_part_iter_exit(&piter);
  1143. }
  1144. EXPORT_SYMBOL(set_disk_ro);
  1145. int bdev_read_only(struct block_device *bdev)
  1146. {
  1147. if (!bdev)
  1148. return 0;
  1149. return bdev->bd_part->policy;
  1150. }
  1151. EXPORT_SYMBOL(bdev_read_only);
  1152. int invalidate_partition(struct gendisk *disk, int partno)
  1153. {
  1154. int res = 0;
  1155. struct block_device *bdev = bdget_disk(disk, partno);
  1156. if (bdev) {
  1157. fsync_bdev(bdev);
  1158. res = __invalidate_device(bdev, true);
  1159. bdput(bdev);
  1160. }
  1161. return res;
  1162. }
  1163. EXPORT_SYMBOL(invalidate_partition);
  1164. /*
  1165. * Disk events - monitor disk events like media change and eject request.
  1166. */
  1167. struct disk_events {
  1168. struct list_head node; /* all disk_event's */
  1169. struct gendisk *disk; /* the associated disk */
  1170. spinlock_t lock;
  1171. struct mutex block_mutex; /* protects blocking */
  1172. int block; /* event blocking depth */
  1173. unsigned int pending; /* events already sent out */
  1174. unsigned int clearing; /* events being cleared */
  1175. long poll_msecs; /* interval, -1 for default */
  1176. struct delayed_work dwork;
  1177. };
  1178. static const char *disk_events_strs[] = {
  1179. [ilog2(DISK_EVENT_MEDIA_CHANGE)] = "media_change",
  1180. [ilog2(DISK_EVENT_EJECT_REQUEST)] = "eject_request",
  1181. };
  1182. static char *disk_uevents[] = {
  1183. [ilog2(DISK_EVENT_MEDIA_CHANGE)] = "DISK_MEDIA_CHANGE=1",
  1184. [ilog2(DISK_EVENT_EJECT_REQUEST)] = "DISK_EJECT_REQUEST=1",
  1185. };
  1186. /* list of all disk_events */
  1187. static DEFINE_MUTEX(disk_events_mutex);
  1188. static LIST_HEAD(disk_events);
  1189. /* disable in-kernel polling by default */
  1190. static unsigned long disk_events_dfl_poll_msecs = 0;
  1191. static unsigned long disk_events_poll_jiffies(struct gendisk *disk)
  1192. {
  1193. struct disk_events *ev = disk->ev;
  1194. long intv_msecs = 0;
  1195. /*
  1196. * If device-specific poll interval is set, always use it. If
  1197. * the default is being used, poll iff there are events which
  1198. * can't be monitored asynchronously.
  1199. */
  1200. if (ev->poll_msecs >= 0)
  1201. intv_msecs = ev->poll_msecs;
  1202. else if (disk->events & ~disk->async_events)
  1203. intv_msecs = disk_events_dfl_poll_msecs;
  1204. return msecs_to_jiffies(intv_msecs);
  1205. }
  1206. /**
  1207. * disk_block_events - block and flush disk event checking
  1208. * @disk: disk to block events for
  1209. *
  1210. * On return from this function, it is guaranteed that event checking
  1211. * isn't in progress and won't happen until unblocked by
  1212. * disk_unblock_events(). Events blocking is counted and the actual
  1213. * unblocking happens after the matching number of unblocks are done.
  1214. *
  1215. * Note that this intentionally does not block event checking from
  1216. * disk_clear_events().
  1217. *
  1218. * CONTEXT:
  1219. * Might sleep.
  1220. */
  1221. void disk_block_events(struct gendisk *disk)
  1222. {
  1223. struct disk_events *ev = disk->ev;
  1224. unsigned long flags;
  1225. bool cancel;
  1226. if (!ev)
  1227. return;
  1228. /*
  1229. * Outer mutex ensures that the first blocker completes canceling
  1230. * the event work before further blockers are allowed to finish.
  1231. */
  1232. mutex_lock(&ev->block_mutex);
  1233. spin_lock_irqsave(&ev->lock, flags);
  1234. cancel = !ev->block++;
  1235. spin_unlock_irqrestore(&ev->lock, flags);
  1236. if (cancel)
  1237. cancel_delayed_work_sync(&disk->ev->dwork);
  1238. mutex_unlock(&ev->block_mutex);
  1239. }
  1240. static void __disk_unblock_events(struct gendisk *disk, bool check_now)
  1241. {
  1242. struct disk_events *ev = disk->ev;
  1243. unsigned long intv;
  1244. unsigned long flags;
  1245. spin_lock_irqsave(&ev->lock, flags);
  1246. if (WARN_ON_ONCE(ev->block <= 0))
  1247. goto out_unlock;
  1248. if (--ev->block)
  1249. goto out_unlock;
  1250. /*
  1251. * Not exactly a latency critical operation, set poll timer
  1252. * slack to 25% and kick event check.
  1253. */
  1254. intv = disk_events_poll_jiffies(disk);
  1255. set_timer_slack(&ev->dwork.timer, intv / 4);
  1256. if (check_now)
  1257. queue_delayed_work(system_freezable_wq, &ev->dwork, 0);
  1258. else if (intv)
  1259. queue_delayed_work(system_freezable_wq, &ev->dwork, intv);
  1260. out_unlock:
  1261. spin_unlock_irqrestore(&ev->lock, flags);
  1262. }
  1263. /**
  1264. * disk_unblock_events - unblock disk event checking
  1265. * @disk: disk to unblock events for
  1266. *
  1267. * Undo disk_block_events(). When the block count reaches zero, it
  1268. * starts events polling if configured.
  1269. *
  1270. * CONTEXT:
  1271. * Don't care. Safe to call from irq context.
  1272. */
  1273. void disk_unblock_events(struct gendisk *disk)
  1274. {
  1275. if (disk->ev)
  1276. __disk_unblock_events(disk, false);
  1277. }
  1278. /**
  1279. * disk_flush_events - schedule immediate event checking and flushing
  1280. * @disk: disk to check and flush events for
  1281. * @mask: events to flush
  1282. *
  1283. * Schedule immediate event checking on @disk if not blocked. Events in
  1284. * @mask are scheduled to be cleared from the driver. Note that this
  1285. * doesn't clear the events from @disk->ev.
  1286. *
  1287. * CONTEXT:
  1288. * If @mask is non-zero must be called with bdev->bd_mutex held.
  1289. */
  1290. void disk_flush_events(struct gendisk *disk, unsigned int mask)
  1291. {
  1292. struct disk_events *ev = disk->ev;
  1293. if (!ev)
  1294. return;
  1295. spin_lock_irq(&ev->lock);
  1296. ev->clearing |= mask;
  1297. if (!ev->block)
  1298. mod_delayed_work(system_freezable_wq, &ev->dwork, 0);
  1299. spin_unlock_irq(&ev->lock);
  1300. }
  1301. /**
  1302. * disk_clear_events - synchronously check, clear and return pending events
  1303. * @disk: disk to fetch and clear events from
  1304. * @mask: mask of events to be fetched and clearted
  1305. *
  1306. * Disk events are synchronously checked and pending events in @mask
  1307. * are cleared and returned. This ignores the block count.
  1308. *
  1309. * CONTEXT:
  1310. * Might sleep.
  1311. */
  1312. unsigned int disk_clear_events(struct gendisk *disk, unsigned int mask)
  1313. {
  1314. const struct block_device_operations *bdops = disk->fops;
  1315. struct disk_events *ev = disk->ev;
  1316. unsigned int pending;
  1317. if (!ev) {
  1318. /* for drivers still using the old ->media_changed method */
  1319. if ((mask & DISK_EVENT_MEDIA_CHANGE) &&
  1320. bdops->media_changed && bdops->media_changed(disk))
  1321. return DISK_EVENT_MEDIA_CHANGE;
  1322. return 0;
  1323. }
  1324. /* tell the workfn about the events being cleared */
  1325. spin_lock_irq(&ev->lock);
  1326. ev->clearing |= mask;
  1327. spin_unlock_irq(&ev->lock);
  1328. /* uncondtionally schedule event check and wait for it to finish */
  1329. disk_block_events(disk);
  1330. /*
  1331. * We need to put the work on system_nrt_wq here since there is a
  1332. * deadlock that happens while probing a usb device while suspending. If
  1333. * we put work on a freezable workqueue here, a usb probe will wait here
  1334. * until the workqueue is unfrozen during suspend. Since suspend waits
  1335. * on all probes to complete, we have a deadlock
  1336. */
  1337. queue_delayed_work(system_nrt_wq, &ev->dwork, 0);
  1338. flush_delayed_work(&ev->dwork);
  1339. __disk_unblock_events(disk, false);
  1340. /* then, fetch and clear pending events */
  1341. spin_lock_irq(&ev->lock);
  1342. WARN_ON_ONCE(ev->clearing & mask); /* cleared by workfn */
  1343. pending = ev->pending & mask;
  1344. ev->pending &= ~mask;
  1345. spin_unlock_irq(&ev->lock);
  1346. return pending;
  1347. }
  1348. static void disk_events_workfn(struct work_struct *work)
  1349. {
  1350. struct delayed_work *dwork = to_delayed_work(work);
  1351. struct disk_events *ev = container_of(dwork, struct disk_events, dwork);
  1352. struct gendisk *disk = ev->disk;
  1353. char *envp[ARRAY_SIZE(disk_uevents) + 1] = { };
  1354. unsigned int clearing = ev->clearing;
  1355. unsigned int events;
  1356. unsigned long intv;
  1357. int nr_events = 0, i;
  1358. /* check events */
  1359. events = disk->fops->check_events(disk, clearing);
  1360. /* accumulate pending events and schedule next poll if necessary */
  1361. spin_lock_irq(&ev->lock);
  1362. events &= ~ev->pending;
  1363. ev->pending |= events;
  1364. ev->clearing &= ~clearing;
  1365. intv = disk_events_poll_jiffies(disk);
  1366. if (!ev->block && intv)
  1367. queue_delayed_work(system_freezable_wq, &ev->dwork, intv);
  1368. spin_unlock_irq(&ev->lock);
  1369. /*
  1370. * Tell userland about new events. Only the events listed in
  1371. * @disk->events are reported. Unlisted events are processed the
  1372. * same internally but never get reported to userland.
  1373. */
  1374. for (i = 0; i < ARRAY_SIZE(disk_uevents); i++)
  1375. if (events & disk->events & (1 << i))
  1376. envp[nr_events++] = disk_uevents[i];
  1377. if (nr_events)
  1378. kobject_uevent_env(&disk_to_dev(disk)->kobj, KOBJ_CHANGE, envp);
  1379. }
  1380. /*
  1381. * A disk events enabled device has the following sysfs nodes under
  1382. * its /sys/block/X/ directory.
  1383. *
  1384. * events : list of all supported events
  1385. * events_async : list of events which can be detected w/o polling
  1386. * events_poll_msecs : polling interval, 0: disable, -1: system default
  1387. */
  1388. static ssize_t __disk_events_show(unsigned int events, char *buf)
  1389. {
  1390. const char *delim = "";
  1391. ssize_t pos = 0;
  1392. int i;
  1393. for (i = 0; i < ARRAY_SIZE(disk_events_strs); i++)
  1394. if (events & (1 << i)) {
  1395. pos += sprintf(buf + pos, "%s%s",
  1396. delim, disk_events_strs[i]);
  1397. delim = " ";
  1398. }
  1399. if (pos)
  1400. pos += sprintf(buf + pos, "\n");
  1401. return pos;
  1402. }
  1403. static ssize_t disk_events_show(struct device *dev,
  1404. struct device_attribute *attr, char *buf)
  1405. {
  1406. struct gendisk *disk = dev_to_disk(dev);
  1407. return __disk_events_show(disk->events, buf);
  1408. }
  1409. static ssize_t disk_events_async_show(struct device *dev,
  1410. struct device_attribute *attr, char *buf)
  1411. {
  1412. struct gendisk *disk = dev_to_disk(dev);
  1413. return __disk_events_show(disk->async_events, buf);
  1414. }
  1415. static ssize_t disk_events_poll_msecs_show(struct device *dev,
  1416. struct device_attribute *attr,
  1417. char *buf)
  1418. {
  1419. struct gendisk *disk = dev_to_disk(dev);
  1420. return sprintf(buf, "%ld\n", disk->ev->poll_msecs);
  1421. }
  1422. static ssize_t disk_events_poll_msecs_store(struct device *dev,
  1423. struct device_attribute *attr,
  1424. const char *buf, size_t count)
  1425. {
  1426. struct gendisk *disk = dev_to_disk(dev);
  1427. long intv;
  1428. if (!count || !sscanf(buf, "%ld", &intv))
  1429. return -EINVAL;
  1430. if (intv < 0 && intv != -1)
  1431. return -EINVAL;
  1432. disk_block_events(disk);
  1433. disk->ev->poll_msecs = intv;
  1434. __disk_unblock_events(disk, true);
  1435. return count;
  1436. }
  1437. static const DEVICE_ATTR(events, S_IRUGO, disk_events_show, NULL);
  1438. static const DEVICE_ATTR(events_async, S_IRUGO, disk_events_async_show, NULL);
  1439. static const DEVICE_ATTR(events_poll_msecs, S_IRUGO|S_IWUSR,
  1440. disk_events_poll_msecs_show,
  1441. disk_events_poll_msecs_store);
  1442. static const struct attribute *disk_events_attrs[] = {
  1443. &dev_attr_events.attr,
  1444. &dev_attr_events_async.attr,
  1445. &dev_attr_events_poll_msecs.attr,
  1446. NULL,
  1447. };
  1448. /*
  1449. * The default polling interval can be specified by the kernel
  1450. * parameter block.events_dfl_poll_msecs which defaults to 0
  1451. * (disable). This can also be modified runtime by writing to
  1452. * /sys/module/block/events_dfl_poll_msecs.
  1453. */
  1454. static int disk_events_set_dfl_poll_msecs(const char *val,
  1455. const struct kernel_param *kp)
  1456. {
  1457. struct disk_events *ev;
  1458. int ret;
  1459. ret = param_set_ulong(val, kp);
  1460. if (ret < 0)
  1461. return ret;
  1462. mutex_lock(&disk_events_mutex);
  1463. list_for_each_entry(ev, &disk_events, node)
  1464. disk_flush_events(ev->disk, 0);
  1465. mutex_unlock(&disk_events_mutex);
  1466. return 0;
  1467. }
  1468. static const struct kernel_param_ops disk_events_dfl_poll_msecs_param_ops = {
  1469. .set = disk_events_set_dfl_poll_msecs,
  1470. .get = param_get_ulong,
  1471. };
  1472. #undef MODULE_PARAM_PREFIX
  1473. #define MODULE_PARAM_PREFIX "block."
  1474. module_param_cb(events_dfl_poll_msecs, &disk_events_dfl_poll_msecs_param_ops,
  1475. &disk_events_dfl_poll_msecs, 0644);
  1476. /*
  1477. * disk_{alloc|add|del|release}_events - initialize and destroy disk_events.
  1478. */
  1479. static void disk_alloc_events(struct gendisk *disk)
  1480. {
  1481. struct disk_events *ev;
  1482. if (!disk->fops->check_events)
  1483. return;
  1484. ev = kzalloc(sizeof(*ev), GFP_KERNEL);
  1485. if (!ev) {
  1486. pr_warn("%s: failed to initialize events\n", disk->disk_name);
  1487. return;
  1488. }
  1489. INIT_LIST_HEAD(&ev->node);
  1490. ev->disk = disk;
  1491. spin_lock_init(&ev->lock);
  1492. mutex_init(&ev->block_mutex);
  1493. ev->block = 1;
  1494. ev->poll_msecs = -1;
  1495. INIT_DELAYED_WORK(&ev->dwork, disk_events_workfn);
  1496. disk->ev = ev;
  1497. }
  1498. static void disk_add_events(struct gendisk *disk)
  1499. {
  1500. if (!disk->ev)
  1501. return;
  1502. /* FIXME: error handling */
  1503. if (sysfs_create_files(&disk_to_dev(disk)->kobj, disk_events_attrs) < 0)
  1504. pr_warn("%s: failed to create sysfs files for events\n",
  1505. disk->disk_name);
  1506. mutex_lock(&disk_events_mutex);
  1507. list_add_tail(&disk->ev->node, &disk_events);
  1508. mutex_unlock(&disk_events_mutex);
  1509. /*
  1510. * Block count is initialized to 1 and the following initial
  1511. * unblock kicks it into action.
  1512. */
  1513. __disk_unblock_events(disk, true);
  1514. }
  1515. static void disk_del_events(struct gendisk *disk)
  1516. {
  1517. if (!disk->ev)
  1518. return;
  1519. disk_block_events(disk);
  1520. mutex_lock(&disk_events_mutex);
  1521. list_del_init(&disk->ev->node);
  1522. mutex_unlock(&disk_events_mutex);
  1523. sysfs_remove_files(&disk_to_dev(disk)->kobj, disk_events_attrs);
  1524. }
  1525. static void disk_release_events(struct gendisk *disk)
  1526. {
  1527. /* the block count should be 1 from disk_del_events() */
  1528. WARN_ON_ONCE(disk->ev && disk->ev->block != 1);
  1529. kfree(disk->ev);
  1530. }