vmt.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828
  1. /*
  2. * Copyright (c) International Business Machines Corp., 2006
  3. *
  4. * This program is free software; you can redistribute it and/or modify
  5. * it under the terms of the GNU General Public License as published by
  6. * the Free Software Foundation; either version 2 of the License, or
  7. * (at your option) any later version.
  8. *
  9. * This program is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
  12. * the GNU General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU General Public License
  15. * along with this program; if not, write to the Free Software
  16. * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  17. *
  18. * Author: Artem Bityutskiy (Битюцкий Артём)
  19. */
  20. /*
  21. * This file contains implementation of volume creation, deletion, updating and
  22. * resizing.
  23. */
  24. #include <linux/err.h>
  25. #include <asm/div64.h>
  26. #include "ubi.h"
  27. #ifdef CONFIG_MTD_UBI_DEBUG_PARANOID
  28. static void paranoid_check_volumes(struct ubi_device *ubi);
  29. #else
  30. #define paranoid_check_volumes(ubi)
  31. #endif
  32. static ssize_t vol_attribute_show(struct device *dev,
  33. struct device_attribute *attr, char *buf);
  34. /* Device attributes corresponding to files in '/<sysfs>/class/ubi/ubiX_Y' */
  35. static struct device_attribute attr_vol_reserved_ebs =
  36. __ATTR(reserved_ebs, S_IRUGO, vol_attribute_show, NULL);
  37. static struct device_attribute attr_vol_type =
  38. __ATTR(type, S_IRUGO, vol_attribute_show, NULL);
  39. static struct device_attribute attr_vol_name =
  40. __ATTR(name, S_IRUGO, vol_attribute_show, NULL);
  41. static struct device_attribute attr_vol_corrupted =
  42. __ATTR(corrupted, S_IRUGO, vol_attribute_show, NULL);
  43. static struct device_attribute attr_vol_alignment =
  44. __ATTR(alignment, S_IRUGO, vol_attribute_show, NULL);
  45. static struct device_attribute attr_vol_usable_eb_size =
  46. __ATTR(usable_eb_size, S_IRUGO, vol_attribute_show, NULL);
  47. static struct device_attribute attr_vol_data_bytes =
  48. __ATTR(data_bytes, S_IRUGO, vol_attribute_show, NULL);
  49. static struct device_attribute attr_vol_upd_marker =
  50. __ATTR(upd_marker, S_IRUGO, vol_attribute_show, NULL);
  51. /*
  52. * "Show" method for files in '/<sysfs>/class/ubi/ubiX_Y/'.
  53. *
  54. * Consider a situation:
  55. * A. process 1 opens a sysfs file related to volume Y, say
  56. * /<sysfs>/class/ubi/ubiX_Y/reserved_ebs;
  57. * B. process 2 removes volume Y;
  58. * C. process 1 starts reading the /<sysfs>/class/ubi/ubiX_Y/reserved_ebs file;
  59. *
  60. * What we want to do in a situation like that is to return error when the file
  61. * is read. This is done by means of the 'removed' flag and the 'vol_lock' of
  62. * the UBI volume description object.
  63. */
  64. static ssize_t vol_attribute_show(struct device *dev,
  65. struct device_attribute *attr, char *buf)
  66. {
  67. int ret;
  68. struct ubi_volume *vol = container_of(dev, struct ubi_volume, dev);
  69. spin_lock(&vol->ubi->volumes_lock);
  70. if (vol->removed) {
  71. spin_unlock(&vol->ubi->volumes_lock);
  72. return -ENODEV;
  73. }
  74. if (attr == &attr_vol_reserved_ebs)
  75. ret = sprintf(buf, "%d\n", vol->reserved_pebs);
  76. else if (attr == &attr_vol_type) {
  77. const char *tp;
  78. if (vol->vol_type == UBI_DYNAMIC_VOLUME)
  79. tp = "dynamic";
  80. else
  81. tp = "static";
  82. ret = sprintf(buf, "%s\n", tp);
  83. } else if (attr == &attr_vol_name)
  84. ret = sprintf(buf, "%s\n", vol->name);
  85. else if (attr == &attr_vol_corrupted)
  86. ret = sprintf(buf, "%d\n", vol->corrupted);
  87. else if (attr == &attr_vol_alignment)
  88. ret = sprintf(buf, "%d\n", vol->alignment);
  89. else if (attr == &attr_vol_usable_eb_size) {
  90. ret = sprintf(buf, "%d\n", vol->usable_leb_size);
  91. } else if (attr == &attr_vol_data_bytes)
  92. ret = sprintf(buf, "%lld\n", vol->used_bytes);
  93. else if (attr == &attr_vol_upd_marker)
  94. ret = sprintf(buf, "%d\n", vol->upd_marker);
  95. else
  96. BUG();
  97. spin_unlock(&vol->ubi->volumes_lock);
  98. return ret;
  99. }
  100. /* Release method for volume devices */
  101. static void vol_release(struct device *dev)
  102. {
  103. struct ubi_volume *vol = container_of(dev, struct ubi_volume, dev);
  104. ubi_assert(vol->removed);
  105. kfree(vol);
  106. }
  107. /**
  108. * volume_sysfs_init - initialize sysfs for new volume.
  109. * @ubi: UBI device description object
  110. * @vol: volume description object
  111. *
  112. * This function returns zero in case of success and a negative error code in
  113. * case of failure.
  114. *
  115. * Note, this function does not free allocated resources in case of failure -
  116. * the caller does it. This is because this would cause release() here and the
  117. * caller would oops.
  118. */
  119. static int volume_sysfs_init(struct ubi_device *ubi, struct ubi_volume *vol)
  120. {
  121. int err;
  122. err = device_create_file(&vol->dev, &attr_vol_reserved_ebs);
  123. if (err)
  124. return err;
  125. err = device_create_file(&vol->dev, &attr_vol_type);
  126. if (err)
  127. return err;
  128. err = device_create_file(&vol->dev, &attr_vol_name);
  129. if (err)
  130. return err;
  131. err = device_create_file(&vol->dev, &attr_vol_corrupted);
  132. if (err)
  133. return err;
  134. err = device_create_file(&vol->dev, &attr_vol_alignment);
  135. if (err)
  136. return err;
  137. err = device_create_file(&vol->dev, &attr_vol_usable_eb_size);
  138. if (err)
  139. return err;
  140. err = device_create_file(&vol->dev, &attr_vol_data_bytes);
  141. if (err)
  142. return err;
  143. err = device_create_file(&vol->dev, &attr_vol_upd_marker);
  144. if (err)
  145. return err;
  146. return 0;
  147. }
  148. /**
  149. * volume_sysfs_close - close sysfs for a volume.
  150. * @vol: volume description object
  151. */
  152. static void volume_sysfs_close(struct ubi_volume *vol)
  153. {
  154. device_remove_file(&vol->dev, &attr_vol_upd_marker);
  155. device_remove_file(&vol->dev, &attr_vol_data_bytes);
  156. device_remove_file(&vol->dev, &attr_vol_usable_eb_size);
  157. device_remove_file(&vol->dev, &attr_vol_alignment);
  158. device_remove_file(&vol->dev, &attr_vol_corrupted);
  159. device_remove_file(&vol->dev, &attr_vol_name);
  160. device_remove_file(&vol->dev, &attr_vol_type);
  161. device_remove_file(&vol->dev, &attr_vol_reserved_ebs);
  162. device_unregister(&vol->dev);
  163. }
  164. /**
  165. * ubi_create_volume - create volume.
  166. * @ubi: UBI device description object
  167. * @req: volume creation request
  168. *
  169. * This function creates volume described by @req. If @req->vol_id id
  170. * %UBI_VOL_NUM_AUTO, this function automatically assigne ID to the new volume
  171. * and saves it in @req->vol_id. Returns zero in case of success and a negative
  172. * error code in case of failure.
  173. */
  174. int ubi_create_volume(struct ubi_device *ubi, struct ubi_mkvol_req *req)
  175. {
  176. int i, err, vol_id = req->vol_id;
  177. struct ubi_volume *vol;
  178. struct ubi_vtbl_record vtbl_rec;
  179. uint64_t bytes;
  180. if (ubi->ro_mode)
  181. return -EROFS;
  182. vol = kzalloc(sizeof(struct ubi_volume), GFP_KERNEL);
  183. if (!vol)
  184. return -ENOMEM;
  185. spin_lock(&ubi->volumes_lock);
  186. if (vol_id == UBI_VOL_NUM_AUTO) {
  187. /* Find unused volume ID */
  188. dbg_msg("search for vacant volume ID");
  189. for (i = 0; i < ubi->vtbl_slots; i++)
  190. if (!ubi->volumes[i]) {
  191. vol_id = i;
  192. break;
  193. }
  194. if (vol_id == UBI_VOL_NUM_AUTO) {
  195. dbg_err("out of volume IDs");
  196. err = -ENFILE;
  197. goto out_unlock;
  198. }
  199. req->vol_id = vol_id;
  200. }
  201. dbg_msg("volume ID %d, %llu bytes, type %d, name %s",
  202. vol_id, (unsigned long long)req->bytes,
  203. (int)req->vol_type, req->name);
  204. /* Ensure that this volume does not exist */
  205. err = -EEXIST;
  206. if (ubi->volumes[vol_id]) {
  207. dbg_err("volume %d already exists", vol_id);
  208. goto out_unlock;
  209. }
  210. /* Ensure that the name is unique */
  211. for (i = 0; i < ubi->vtbl_slots; i++)
  212. if (ubi->volumes[i] &&
  213. ubi->volumes[i]->name_len == req->name_len &&
  214. !strcmp(ubi->volumes[i]->name, req->name)) {
  215. dbg_err("volume \"%s\" exists (ID %d)", req->name, i);
  216. goto out_unlock;
  217. }
  218. /* Calculate how many eraseblocks are requested */
  219. vol->usable_leb_size = ubi->leb_size - ubi->leb_size % req->alignment;
  220. bytes = req->bytes;
  221. if (do_div(bytes, vol->usable_leb_size))
  222. vol->reserved_pebs = 1;
  223. vol->reserved_pebs += bytes;
  224. /* Reserve physical eraseblocks */
  225. if (vol->reserved_pebs > ubi->avail_pebs) {
  226. dbg_err("not enough PEBs, only %d available", ubi->avail_pebs);
  227. err = -ENOSPC;
  228. goto out_unlock;
  229. }
  230. ubi->avail_pebs -= vol->reserved_pebs;
  231. ubi->rsvd_pebs += vol->reserved_pebs;
  232. vol->vol_id = vol_id;
  233. vol->alignment = req->alignment;
  234. vol->data_pad = ubi->leb_size % vol->alignment;
  235. vol->vol_type = req->vol_type;
  236. vol->name_len = req->name_len;
  237. memcpy(vol->name, req->name, vol->name_len + 1);
  238. vol->exclusive = 1;
  239. vol->ubi = ubi;
  240. ubi->volumes[vol_id] = vol;
  241. spin_unlock(&ubi->volumes_lock);
  242. /*
  243. * Finish all pending erases because there may be some LEBs belonging
  244. * to the same volume ID.
  245. */
  246. err = ubi_wl_flush(ubi);
  247. if (err)
  248. goto out_acc;
  249. vol->eba_tbl = kmalloc(vol->reserved_pebs * sizeof(int), GFP_KERNEL);
  250. if (!vol->eba_tbl) {
  251. err = -ENOMEM;
  252. goto out_acc;
  253. }
  254. for (i = 0; i < vol->reserved_pebs; i++)
  255. vol->eba_tbl[i] = UBI_LEB_UNMAPPED;
  256. if (vol->vol_type == UBI_DYNAMIC_VOLUME) {
  257. vol->used_ebs = vol->reserved_pebs;
  258. vol->last_eb_bytes = vol->usable_leb_size;
  259. vol->used_bytes =
  260. (long long)vol->used_ebs * vol->usable_leb_size;
  261. } else {
  262. bytes = vol->used_bytes;
  263. vol->last_eb_bytes = do_div(bytes, vol->usable_leb_size);
  264. vol->used_ebs = bytes;
  265. if (vol->last_eb_bytes)
  266. vol->used_ebs += 1;
  267. else
  268. vol->last_eb_bytes = vol->usable_leb_size;
  269. }
  270. /* Register character device for the volume */
  271. cdev_init(&vol->cdev, &ubi_vol_cdev_operations);
  272. vol->cdev.owner = THIS_MODULE;
  273. err = cdev_add(&vol->cdev, MKDEV(ubi->major, vol_id + 1), 1);
  274. if (err) {
  275. ubi_err("cannot add character device for volume %d", vol_id);
  276. goto out_mapping;
  277. }
  278. err = ubi_create_gluebi(ubi, vol);
  279. if (err)
  280. goto out_cdev;
  281. vol->dev.release = vol_release;
  282. vol->dev.parent = &ubi->dev;
  283. vol->dev.devt = MKDEV(ubi->major, vol->vol_id + 1);
  284. vol->dev.class = ubi_class;
  285. sprintf(&vol->dev.bus_id[0], "%s_%d", ubi->ubi_name, vol->vol_id);
  286. err = device_register(&vol->dev);
  287. if (err)
  288. goto out_gluebi;
  289. err = volume_sysfs_init(ubi, vol);
  290. if (err)
  291. goto out_sysfs;
  292. /* Fill volume table record */
  293. memset(&vtbl_rec, 0, sizeof(struct ubi_vtbl_record));
  294. vtbl_rec.reserved_pebs = cpu_to_be32(vol->reserved_pebs);
  295. vtbl_rec.alignment = cpu_to_be32(vol->alignment);
  296. vtbl_rec.data_pad = cpu_to_be32(vol->data_pad);
  297. vtbl_rec.name_len = cpu_to_be16(vol->name_len);
  298. if (vol->vol_type == UBI_DYNAMIC_VOLUME)
  299. vtbl_rec.vol_type = UBI_VID_DYNAMIC;
  300. else
  301. vtbl_rec.vol_type = UBI_VID_STATIC;
  302. memcpy(vtbl_rec.name, vol->name, vol->name_len + 1);
  303. err = ubi_change_vtbl_record(ubi, vol_id, &vtbl_rec);
  304. if (err)
  305. goto out_sysfs;
  306. spin_lock(&ubi->volumes_lock);
  307. ubi->vol_count += 1;
  308. vol->exclusive = 0;
  309. spin_unlock(&ubi->volumes_lock);
  310. paranoid_check_volumes(ubi);
  311. return 0;
  312. out_gluebi:
  313. err = ubi_destroy_gluebi(vol);
  314. out_cdev:
  315. cdev_del(&vol->cdev);
  316. out_mapping:
  317. kfree(vol->eba_tbl);
  318. out_acc:
  319. spin_lock(&ubi->volumes_lock);
  320. ubi->rsvd_pebs -= vol->reserved_pebs;
  321. ubi->avail_pebs += vol->reserved_pebs;
  322. ubi->volumes[vol_id] = NULL;
  323. out_unlock:
  324. spin_unlock(&ubi->volumes_lock);
  325. kfree(vol);
  326. return err;
  327. /*
  328. * We are registered, so @vol is destroyed in the release function and
  329. * we have to de-initialize differently.
  330. */
  331. out_sysfs:
  332. err = ubi_destroy_gluebi(vol);
  333. cdev_del(&vol->cdev);
  334. kfree(vol->eba_tbl);
  335. spin_lock(&ubi->volumes_lock);
  336. ubi->rsvd_pebs -= vol->reserved_pebs;
  337. ubi->avail_pebs += vol->reserved_pebs;
  338. ubi->volumes[vol_id] = NULL;
  339. spin_unlock(&ubi->volumes_lock);
  340. volume_sysfs_close(vol);
  341. return err;
  342. }
  343. /**
  344. * ubi_remove_volume - remove volume.
  345. * @desc: volume descriptor
  346. *
  347. * This function removes volume described by @desc. The volume has to be opened
  348. * in "exclusive" mode. Returns zero in case of success and a negative error
  349. * code in case of failure.
  350. */
  351. int ubi_remove_volume(struct ubi_volume_desc *desc)
  352. {
  353. struct ubi_volume *vol = desc->vol;
  354. struct ubi_device *ubi = vol->ubi;
  355. int i, err, vol_id = vol->vol_id, reserved_pebs = vol->reserved_pebs;
  356. dbg_msg("remove UBI volume %d", vol_id);
  357. ubi_assert(desc->mode == UBI_EXCLUSIVE);
  358. ubi_assert(vol == ubi->volumes[vol_id]);
  359. if (ubi->ro_mode)
  360. return -EROFS;
  361. err = ubi_destroy_gluebi(vol);
  362. if (err)
  363. return err;
  364. err = ubi_change_vtbl_record(ubi, vol_id, NULL);
  365. if (err)
  366. return err;
  367. for (i = 0; i < vol->reserved_pebs; i++) {
  368. err = ubi_eba_unmap_leb(ubi, vol_id, i);
  369. if (err)
  370. return err;
  371. }
  372. spin_lock(&ubi->volumes_lock);
  373. vol->removed = 1;
  374. ubi->volumes[vol_id] = NULL;
  375. spin_unlock(&ubi->volumes_lock);
  376. kfree(vol->eba_tbl);
  377. vol->eba_tbl = NULL;
  378. cdev_del(&vol->cdev);
  379. volume_sysfs_close(vol);
  380. kfree(desc);
  381. spin_lock(&ubi->volumes_lock);
  382. ubi->rsvd_pebs -= reserved_pebs;
  383. ubi->avail_pebs += reserved_pebs;
  384. i = ubi->beb_rsvd_level - ubi->beb_rsvd_pebs;
  385. if (i > 0) {
  386. i = ubi->avail_pebs >= i ? i : ubi->avail_pebs;
  387. ubi->avail_pebs -= i;
  388. ubi->rsvd_pebs += i;
  389. ubi->beb_rsvd_pebs += i;
  390. if (i > 0)
  391. ubi_msg("reserve more %d PEBs", i);
  392. }
  393. ubi->vol_count -= 1;
  394. spin_unlock(&ubi->volumes_lock);
  395. paranoid_check_volumes(ubi);
  396. module_put(THIS_MODULE);
  397. return 0;
  398. }
  399. /**
  400. * ubi_resize_volume - re-size volume.
  401. * @desc: volume descriptor
  402. * @reserved_pebs: new size in physical eraseblocks
  403. *
  404. * This function returns zero in case of success, and a negative error code in
  405. * case of failure.
  406. */
  407. int ubi_resize_volume(struct ubi_volume_desc *desc, int reserved_pebs)
  408. {
  409. int i, err, pebs, *new_mapping;
  410. struct ubi_volume *vol = desc->vol;
  411. struct ubi_device *ubi = vol->ubi;
  412. struct ubi_vtbl_record vtbl_rec;
  413. int vol_id = vol->vol_id;
  414. if (ubi->ro_mode)
  415. return -EROFS;
  416. dbg_msg("re-size volume %d to from %d to %d PEBs",
  417. vol_id, vol->reserved_pebs, reserved_pebs);
  418. ubi_assert(desc->mode == UBI_EXCLUSIVE);
  419. ubi_assert(vol == ubi->volumes[vol_id]);
  420. if (vol->vol_type == UBI_STATIC_VOLUME &&
  421. reserved_pebs < vol->used_ebs) {
  422. dbg_err("too small size %d, %d LEBs contain data",
  423. reserved_pebs, vol->used_ebs);
  424. return -EINVAL;
  425. }
  426. /* If the size is the same, we have nothing to do */
  427. if (reserved_pebs == vol->reserved_pebs)
  428. return 0;
  429. new_mapping = kmalloc(reserved_pebs * sizeof(int), GFP_KERNEL);
  430. if (!new_mapping)
  431. return -ENOMEM;
  432. for (i = 0; i < reserved_pebs; i++)
  433. new_mapping[i] = UBI_LEB_UNMAPPED;
  434. /* Reserve physical eraseblocks */
  435. pebs = reserved_pebs - vol->reserved_pebs;
  436. if (pebs > 0) {
  437. spin_lock(&ubi->volumes_lock);
  438. if (pebs > ubi->avail_pebs) {
  439. dbg_err("not enough PEBs: requested %d, available %d",
  440. pebs, ubi->avail_pebs);
  441. spin_unlock(&ubi->volumes_lock);
  442. err = -ENOSPC;
  443. goto out_free;
  444. }
  445. ubi->avail_pebs -= pebs;
  446. ubi->rsvd_pebs += pebs;
  447. for (i = 0; i < vol->reserved_pebs; i++)
  448. new_mapping[i] = vol->eba_tbl[i];
  449. kfree(vol->eba_tbl);
  450. vol->eba_tbl = new_mapping;
  451. spin_unlock(&ubi->volumes_lock);
  452. }
  453. /* Change volume table record */
  454. memcpy(&vtbl_rec, &ubi->vtbl[vol_id], sizeof(struct ubi_vtbl_record));
  455. vtbl_rec.reserved_pebs = cpu_to_be32(reserved_pebs);
  456. err = ubi_change_vtbl_record(ubi, vol_id, &vtbl_rec);
  457. if (err)
  458. goto out_acc;
  459. if (pebs < 0) {
  460. for (i = 0; i < -pebs; i++) {
  461. err = ubi_eba_unmap_leb(ubi, vol_id, reserved_pebs + i);
  462. if (err)
  463. goto out_acc;
  464. }
  465. spin_lock(&ubi->volumes_lock);
  466. ubi->rsvd_pebs += pebs;
  467. ubi->avail_pebs -= pebs;
  468. pebs = ubi->beb_rsvd_level - ubi->beb_rsvd_pebs;
  469. if (pebs > 0) {
  470. pebs = ubi->avail_pebs >= pebs ? pebs : ubi->avail_pebs;
  471. ubi->avail_pebs -= pebs;
  472. ubi->rsvd_pebs += pebs;
  473. ubi->beb_rsvd_pebs += pebs;
  474. if (pebs > 0)
  475. ubi_msg("reserve more %d PEBs", pebs);
  476. }
  477. for (i = 0; i < reserved_pebs; i++)
  478. new_mapping[i] = vol->eba_tbl[i];
  479. kfree(vol->eba_tbl);
  480. vol->eba_tbl = new_mapping;
  481. spin_unlock(&ubi->volumes_lock);
  482. }
  483. vol->reserved_pebs = reserved_pebs;
  484. if (vol->vol_type == UBI_DYNAMIC_VOLUME) {
  485. vol->used_ebs = reserved_pebs;
  486. vol->last_eb_bytes = vol->usable_leb_size;
  487. vol->used_bytes =
  488. (long long)vol->used_ebs * vol->usable_leb_size;
  489. }
  490. paranoid_check_volumes(ubi);
  491. return 0;
  492. out_acc:
  493. if (pebs > 0) {
  494. spin_lock(&ubi->volumes_lock);
  495. ubi->rsvd_pebs -= pebs;
  496. ubi->avail_pebs += pebs;
  497. spin_unlock(&ubi->volumes_lock);
  498. }
  499. out_free:
  500. kfree(new_mapping);
  501. return err;
  502. }
  503. /**
  504. * ubi_add_volume - add volume.
  505. * @ubi: UBI device description object
  506. * @vol_id: volume ID
  507. *
  508. * This function adds an existin volume and initializes all its data
  509. * structures. Returnes zero in case of success and a negative error code in
  510. * case of failure.
  511. */
  512. int ubi_add_volume(struct ubi_device *ubi, int vol_id)
  513. {
  514. int err;
  515. struct ubi_volume *vol = ubi->volumes[vol_id];
  516. dbg_msg("add volume %d", vol_id);
  517. ubi_dbg_dump_vol_info(vol);
  518. ubi_assert(vol);
  519. /* Register character device for the volume */
  520. cdev_init(&vol->cdev, &ubi_vol_cdev_operations);
  521. vol->cdev.owner = THIS_MODULE;
  522. err = cdev_add(&vol->cdev, MKDEV(ubi->major, vol->vol_id + 1), 1);
  523. if (err) {
  524. ubi_err("cannot add character device for volume %d", vol_id);
  525. return err;
  526. }
  527. err = ubi_create_gluebi(ubi, vol);
  528. if (err)
  529. goto out_cdev;
  530. vol->dev.release = vol_release;
  531. vol->dev.parent = &ubi->dev;
  532. vol->dev.devt = MKDEV(ubi->major, vol->vol_id + 1);
  533. vol->dev.class = ubi_class;
  534. sprintf(&vol->dev.bus_id[0], "%s_%d", ubi->ubi_name, vol->vol_id);
  535. err = device_register(&vol->dev);
  536. if (err)
  537. goto out_gluebi;
  538. err = volume_sysfs_init(ubi, vol);
  539. if (err) {
  540. cdev_del(&vol->cdev);
  541. err = ubi_destroy_gluebi(vol);
  542. volume_sysfs_close(vol);
  543. return err;
  544. }
  545. paranoid_check_volumes(ubi);
  546. return 0;
  547. out_gluebi:
  548. err = ubi_destroy_gluebi(vol);
  549. out_cdev:
  550. cdev_del(&vol->cdev);
  551. return err;
  552. }
  553. /**
  554. * ubi_free_volume - free volume.
  555. * @ubi: UBI device description object
  556. * @vol_id: volume ID
  557. *
  558. * This function frees all resources for volume @vol_id but does not remove it.
  559. * Used only when the UBI device is detached.
  560. */
  561. void ubi_free_volume(struct ubi_device *ubi, int vol_id)
  562. {
  563. int err;
  564. struct ubi_volume *vol = ubi->volumes[vol_id];
  565. dbg_msg("free volume %d", vol_id);
  566. ubi_assert(vol);
  567. vol->removed = 1;
  568. err = ubi_destroy_gluebi(vol);
  569. ubi->volumes[vol_id] = NULL;
  570. cdev_del(&vol->cdev);
  571. volume_sysfs_close(vol);
  572. }
  573. #ifdef CONFIG_MTD_UBI_DEBUG_PARANOID
  574. /**
  575. * paranoid_check_volume - check volume information.
  576. * @ubi: UBI device description object
  577. * @vol_id: volume ID
  578. */
  579. static void paranoid_check_volume(struct ubi_device *ubi, int vol_id)
  580. {
  581. int idx = vol_id2idx(ubi, vol_id);
  582. int reserved_pebs, alignment, data_pad, vol_type, name_len, upd_marker;
  583. const struct ubi_volume *vol;
  584. long long n;
  585. const char *name;
  586. spin_lock(&ubi->volumes_lock);
  587. reserved_pebs = be32_to_cpu(ubi->vtbl[vol_id].reserved_pebs);
  588. vol = ubi->volumes[idx];
  589. if (!vol) {
  590. if (reserved_pebs) {
  591. ubi_err("no volume info, but volume exists");
  592. goto fail;
  593. }
  594. spin_unlock(&ubi->volumes_lock);
  595. return;
  596. }
  597. if (vol->exclusive) {
  598. /*
  599. * The volume may be being created at the moment, do not check
  600. * it (e.g., it may be in the middle of ubi_create_volume().
  601. */
  602. spin_unlock(&ubi->volumes_lock);
  603. return;
  604. }
  605. if (vol->reserved_pebs < 0 || vol->alignment < 0 || vol->data_pad < 0 ||
  606. vol->name_len < 0) {
  607. ubi_err("negative values");
  608. goto fail;
  609. }
  610. if (vol->alignment > ubi->leb_size || vol->alignment == 0) {
  611. ubi_err("bad alignment");
  612. goto fail;
  613. }
  614. n = vol->alignment % ubi->min_io_size;
  615. if (vol->alignment != 1 && n) {
  616. ubi_err("alignment is not multiple of min I/O unit");
  617. goto fail;
  618. }
  619. n = ubi->leb_size % vol->alignment;
  620. if (vol->data_pad != n) {
  621. ubi_err("bad data_pad, has to be %lld", n);
  622. goto fail;
  623. }
  624. if (vol->vol_type != UBI_DYNAMIC_VOLUME &&
  625. vol->vol_type != UBI_STATIC_VOLUME) {
  626. ubi_err("bad vol_type");
  627. goto fail;
  628. }
  629. if (vol->upd_marker != 0 && vol->upd_marker != 1) {
  630. ubi_err("bad upd_marker");
  631. goto fail;
  632. }
  633. if (vol->upd_marker && vol->corrupted) {
  634. dbg_err("update marker and corrupted simultaneously");
  635. goto fail;
  636. }
  637. if (vol->reserved_pebs > ubi->good_peb_count) {
  638. ubi_err("too large reserved_pebs");
  639. goto fail;
  640. }
  641. n = ubi->leb_size - vol->data_pad;
  642. if (vol->usable_leb_size != ubi->leb_size - vol->data_pad) {
  643. ubi_err("bad usable_leb_size, has to be %lld", n);
  644. goto fail;
  645. }
  646. if (vol->name_len > UBI_VOL_NAME_MAX) {
  647. ubi_err("too long volume name, max is %d", UBI_VOL_NAME_MAX);
  648. goto fail;
  649. }
  650. if (!vol->name) {
  651. ubi_err("NULL volume name");
  652. goto fail;
  653. }
  654. n = strnlen(vol->name, vol->name_len + 1);
  655. if (n != vol->name_len) {
  656. ubi_err("bad name_len %lld", n);
  657. goto fail;
  658. }
  659. n = (long long)vol->used_ebs * vol->usable_leb_size;
  660. if (vol->vol_type == UBI_DYNAMIC_VOLUME) {
  661. if (vol->corrupted != 0) {
  662. ubi_err("corrupted dynamic volume");
  663. goto fail;
  664. }
  665. if (vol->used_ebs != vol->reserved_pebs) {
  666. ubi_err("bad used_ebs");
  667. goto fail;
  668. }
  669. if (vol->last_eb_bytes != vol->usable_leb_size) {
  670. ubi_err("bad last_eb_bytes");
  671. goto fail;
  672. }
  673. if (vol->used_bytes != n) {
  674. ubi_err("bad used_bytes");
  675. goto fail;
  676. }
  677. } else {
  678. if (vol->corrupted != 0 && vol->corrupted != 1) {
  679. ubi_err("bad corrupted");
  680. goto fail;
  681. }
  682. if (vol->used_ebs < 0 || vol->used_ebs > vol->reserved_pebs) {
  683. ubi_err("bad used_ebs");
  684. goto fail;
  685. }
  686. if (vol->last_eb_bytes < 0 ||
  687. vol->last_eb_bytes > vol->usable_leb_size) {
  688. ubi_err("bad last_eb_bytes");
  689. goto fail;
  690. }
  691. if (vol->used_bytes < 0 || vol->used_bytes > n ||
  692. vol->used_bytes < n - vol->usable_leb_size) {
  693. ubi_err("bad used_bytes");
  694. goto fail;
  695. }
  696. }
  697. alignment = be32_to_cpu(ubi->vtbl[vol_id].alignment);
  698. data_pad = be32_to_cpu(ubi->vtbl[vol_id].data_pad);
  699. name_len = be16_to_cpu(ubi->vtbl[vol_id].name_len);
  700. upd_marker = ubi->vtbl[vol_id].upd_marker;
  701. name = &ubi->vtbl[vol_id].name[0];
  702. if (ubi->vtbl[vol_id].vol_type == UBI_VID_DYNAMIC)
  703. vol_type = UBI_DYNAMIC_VOLUME;
  704. else
  705. vol_type = UBI_STATIC_VOLUME;
  706. if (alignment != vol->alignment || data_pad != vol->data_pad ||
  707. upd_marker != vol->upd_marker || vol_type != vol->vol_type ||
  708. name_len!= vol->name_len || strncmp(name, vol->name, name_len)) {
  709. ubi_err("volume info is different");
  710. goto fail;
  711. }
  712. spin_unlock(&ubi->volumes_lock);
  713. return;
  714. fail:
  715. ubi_err("paranoid check failed for volume %d", vol_id);
  716. ubi_dbg_dump_vol_info(vol);
  717. ubi_dbg_dump_vtbl_record(&ubi->vtbl[vol_id], vol_id);
  718. spin_unlock(&ubi->volumes_lock);
  719. BUG();
  720. }
  721. /**
  722. * paranoid_check_volumes - check information about all volumes.
  723. * @ubi: UBI device description object
  724. */
  725. static void paranoid_check_volumes(struct ubi_device *ubi)
  726. {
  727. int i;
  728. mutex_lock(&ubi->vtbl_mutex);
  729. for (i = 0; i < ubi->vtbl_slots; i++)
  730. paranoid_check_volume(ubi, i);
  731. mutex_unlock(&ubi->vtbl_mutex);
  732. }
  733. #endif