btree.c 57 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202
  1. /*
  2. * btree.c - NILFS B-tree.
  3. *
  4. * Copyright (C) 2005-2008 Nippon Telegraph and Telephone Corporation.
  5. *
  6. * This program is free software; you can redistribute it and/or modify
  7. * it under the terms of the GNU General Public License as published by
  8. * the Free Software Foundation; either version 2 of the License, or
  9. * (at your option) any later version.
  10. *
  11. * This program is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. * GNU General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU General Public License
  17. * along with this program; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  19. *
  20. * Written by Koji Sato <koji@osrg.net>.
  21. */
  22. #include <linux/slab.h>
  23. #include <linux/string.h>
  24. #include <linux/errno.h>
  25. #include <linux/pagevec.h>
  26. #include "nilfs.h"
  27. #include "page.h"
  28. #include "btnode.h"
  29. #include "btree.h"
  30. #include "alloc.h"
  31. #include "dat.h"
  32. static struct nilfs_btree_path *nilfs_btree_alloc_path(void)
  33. {
  34. struct nilfs_btree_path *path;
  35. int level = NILFS_BTREE_LEVEL_DATA;
  36. path = kmem_cache_alloc(nilfs_btree_path_cache, GFP_NOFS);
  37. if (path == NULL)
  38. goto out;
  39. for (; level < NILFS_BTREE_LEVEL_MAX; level++) {
  40. path[level].bp_bh = NULL;
  41. path[level].bp_sib_bh = NULL;
  42. path[level].bp_index = 0;
  43. path[level].bp_oldreq.bpr_ptr = NILFS_BMAP_INVALID_PTR;
  44. path[level].bp_newreq.bpr_ptr = NILFS_BMAP_INVALID_PTR;
  45. path[level].bp_op = NULL;
  46. }
  47. out:
  48. return path;
  49. }
  50. static void nilfs_btree_free_path(struct nilfs_btree_path *path)
  51. {
  52. int level = NILFS_BTREE_LEVEL_DATA;
  53. for (; level < NILFS_BTREE_LEVEL_MAX; level++)
  54. brelse(path[level].bp_bh);
  55. kmem_cache_free(nilfs_btree_path_cache, path);
  56. }
  57. /*
  58. * B-tree node operations
  59. */
  60. static int nilfs_btree_get_block(const struct nilfs_bmap *btree, __u64 ptr,
  61. struct buffer_head **bhp)
  62. {
  63. struct address_space *btnc = &NILFS_BMAP_I(btree)->i_btnode_cache;
  64. struct buffer_head *bh;
  65. int err;
  66. err = nilfs_btnode_submit_block(btnc, ptr, 0, bhp);
  67. if (err)
  68. return err == -EEXIST ? 0 : err;
  69. bh = *bhp;
  70. wait_on_buffer(bh);
  71. if (!buffer_uptodate(bh)) {
  72. brelse(bh);
  73. return -EIO;
  74. }
  75. if (nilfs_btree_broken_node_block(bh)) {
  76. clear_buffer_uptodate(bh);
  77. brelse(bh);
  78. return -EINVAL;
  79. }
  80. return 0;
  81. }
  82. static int nilfs_btree_get_new_block(const struct nilfs_bmap *btree,
  83. __u64 ptr, struct buffer_head **bhp)
  84. {
  85. struct address_space *btnc = &NILFS_BMAP_I(btree)->i_btnode_cache;
  86. struct buffer_head *bh;
  87. bh = nilfs_btnode_create_block(btnc, ptr);
  88. if (!bh)
  89. return -ENOMEM;
  90. set_buffer_nilfs_volatile(bh);
  91. *bhp = bh;
  92. return 0;
  93. }
  94. static inline int
  95. nilfs_btree_node_get_flags(const struct nilfs_btree_node *node)
  96. {
  97. return node->bn_flags;
  98. }
  99. static inline void
  100. nilfs_btree_node_set_flags(struct nilfs_btree_node *node, int flags)
  101. {
  102. node->bn_flags = flags;
  103. }
  104. static inline int nilfs_btree_node_root(const struct nilfs_btree_node *node)
  105. {
  106. return nilfs_btree_node_get_flags(node) & NILFS_BTREE_NODE_ROOT;
  107. }
  108. static inline int
  109. nilfs_btree_node_get_level(const struct nilfs_btree_node *node)
  110. {
  111. return node->bn_level;
  112. }
  113. static inline void
  114. nilfs_btree_node_set_level(struct nilfs_btree_node *node, int level)
  115. {
  116. node->bn_level = level;
  117. }
  118. static inline int
  119. nilfs_btree_node_get_nchildren(const struct nilfs_btree_node *node)
  120. {
  121. return le16_to_cpu(node->bn_nchildren);
  122. }
  123. static inline void
  124. nilfs_btree_node_set_nchildren(struct nilfs_btree_node *node, int nchildren)
  125. {
  126. node->bn_nchildren = cpu_to_le16(nchildren);
  127. }
  128. static inline int nilfs_btree_node_size(const struct nilfs_bmap *btree)
  129. {
  130. return 1 << btree->b_inode->i_blkbits;
  131. }
  132. static inline int
  133. nilfs_btree_node_nchildren_min(const struct nilfs_btree_node *node,
  134. const struct nilfs_bmap *btree)
  135. {
  136. return nilfs_btree_node_root(node) ?
  137. NILFS_BTREE_ROOT_NCHILDREN_MIN :
  138. NILFS_BTREE_NODE_NCHILDREN_MIN(nilfs_btree_node_size(btree));
  139. }
  140. static inline int
  141. nilfs_btree_node_nchildren_max(const struct nilfs_btree_node *node,
  142. const struct nilfs_bmap *btree)
  143. {
  144. return nilfs_btree_node_root(node) ?
  145. NILFS_BTREE_ROOT_NCHILDREN_MAX :
  146. NILFS_BTREE_NODE_NCHILDREN_MAX(nilfs_btree_node_size(btree));
  147. }
  148. static inline __le64 *
  149. nilfs_btree_node_dkeys(const struct nilfs_btree_node *node)
  150. {
  151. return (__le64 *)((char *)(node + 1) +
  152. (nilfs_btree_node_root(node) ?
  153. 0 : NILFS_BTREE_NODE_EXTRA_PAD_SIZE));
  154. }
  155. static inline __le64 *
  156. nilfs_btree_node_dptrs(const struct nilfs_btree_node *node,
  157. const struct nilfs_bmap *btree)
  158. {
  159. return (__le64 *)(nilfs_btree_node_dkeys(node) +
  160. nilfs_btree_node_nchildren_max(node, btree));
  161. }
  162. static inline __u64
  163. nilfs_btree_node_get_key(const struct nilfs_btree_node *node, int index)
  164. {
  165. return le64_to_cpu(*(nilfs_btree_node_dkeys(node) + index));
  166. }
  167. static inline void
  168. nilfs_btree_node_set_key(struct nilfs_btree_node *node, int index, __u64 key)
  169. {
  170. *(nilfs_btree_node_dkeys(node) + index) = cpu_to_le64(key);
  171. }
  172. static inline __u64
  173. nilfs_btree_node_get_ptr(const struct nilfs_bmap *btree,
  174. const struct nilfs_btree_node *node, int index)
  175. {
  176. return le64_to_cpu(*(nilfs_btree_node_dptrs(node, btree) + index));
  177. }
  178. static inline void
  179. nilfs_btree_node_set_ptr(struct nilfs_bmap *btree,
  180. struct nilfs_btree_node *node, int index, __u64 ptr)
  181. {
  182. *(nilfs_btree_node_dptrs(node, btree) + index) = cpu_to_le64(ptr);
  183. }
  184. static void nilfs_btree_node_init(struct nilfs_bmap *btree,
  185. struct nilfs_btree_node *node,
  186. int flags, int level, int nchildren,
  187. const __u64 *keys, const __u64 *ptrs)
  188. {
  189. __le64 *dkeys;
  190. __le64 *dptrs;
  191. int i;
  192. nilfs_btree_node_set_flags(node, flags);
  193. nilfs_btree_node_set_level(node, level);
  194. nilfs_btree_node_set_nchildren(node, nchildren);
  195. dkeys = nilfs_btree_node_dkeys(node);
  196. dptrs = nilfs_btree_node_dptrs(node, btree);
  197. for (i = 0; i < nchildren; i++) {
  198. dkeys[i] = cpu_to_le64(keys[i]);
  199. dptrs[i] = cpu_to_le64(ptrs[i]);
  200. }
  201. }
  202. /* Assume the buffer heads corresponding to left and right are locked. */
  203. static void nilfs_btree_node_move_left(struct nilfs_bmap *btree,
  204. struct nilfs_btree_node *left,
  205. struct nilfs_btree_node *right,
  206. int n)
  207. {
  208. __le64 *ldkeys, *rdkeys;
  209. __le64 *ldptrs, *rdptrs;
  210. int lnchildren, rnchildren;
  211. ldkeys = nilfs_btree_node_dkeys(left);
  212. ldptrs = nilfs_btree_node_dptrs(left, btree);
  213. lnchildren = nilfs_btree_node_get_nchildren(left);
  214. rdkeys = nilfs_btree_node_dkeys(right);
  215. rdptrs = nilfs_btree_node_dptrs(right, btree);
  216. rnchildren = nilfs_btree_node_get_nchildren(right);
  217. memcpy(ldkeys + lnchildren, rdkeys, n * sizeof(*rdkeys));
  218. memcpy(ldptrs + lnchildren, rdptrs, n * sizeof(*rdptrs));
  219. memmove(rdkeys, rdkeys + n, (rnchildren - n) * sizeof(*rdkeys));
  220. memmove(rdptrs, rdptrs + n, (rnchildren - n) * sizeof(*rdptrs));
  221. lnchildren += n;
  222. rnchildren -= n;
  223. nilfs_btree_node_set_nchildren(left, lnchildren);
  224. nilfs_btree_node_set_nchildren(right, rnchildren);
  225. }
  226. /* Assume that the buffer heads corresponding to left and right are locked. */
  227. static void nilfs_btree_node_move_right(struct nilfs_bmap *btree,
  228. struct nilfs_btree_node *left,
  229. struct nilfs_btree_node *right,
  230. int n)
  231. {
  232. __le64 *ldkeys, *rdkeys;
  233. __le64 *ldptrs, *rdptrs;
  234. int lnchildren, rnchildren;
  235. ldkeys = nilfs_btree_node_dkeys(left);
  236. ldptrs = nilfs_btree_node_dptrs(left, btree);
  237. lnchildren = nilfs_btree_node_get_nchildren(left);
  238. rdkeys = nilfs_btree_node_dkeys(right);
  239. rdptrs = nilfs_btree_node_dptrs(right, btree);
  240. rnchildren = nilfs_btree_node_get_nchildren(right);
  241. memmove(rdkeys + n, rdkeys, rnchildren * sizeof(*rdkeys));
  242. memmove(rdptrs + n, rdptrs, rnchildren * sizeof(*rdptrs));
  243. memcpy(rdkeys, ldkeys + lnchildren - n, n * sizeof(*rdkeys));
  244. memcpy(rdptrs, ldptrs + lnchildren - n, n * sizeof(*rdptrs));
  245. lnchildren -= n;
  246. rnchildren += n;
  247. nilfs_btree_node_set_nchildren(left, lnchildren);
  248. nilfs_btree_node_set_nchildren(right, rnchildren);
  249. }
  250. /* Assume that the buffer head corresponding to node is locked. */
  251. static void nilfs_btree_node_insert(struct nilfs_bmap *btree,
  252. struct nilfs_btree_node *node,
  253. __u64 key, __u64 ptr, int index)
  254. {
  255. __le64 *dkeys;
  256. __le64 *dptrs;
  257. int nchildren;
  258. dkeys = nilfs_btree_node_dkeys(node);
  259. dptrs = nilfs_btree_node_dptrs(node, btree);
  260. nchildren = nilfs_btree_node_get_nchildren(node);
  261. if (index < nchildren) {
  262. memmove(dkeys + index + 1, dkeys + index,
  263. (nchildren - index) * sizeof(*dkeys));
  264. memmove(dptrs + index + 1, dptrs + index,
  265. (nchildren - index) * sizeof(*dptrs));
  266. }
  267. dkeys[index] = cpu_to_le64(key);
  268. dptrs[index] = cpu_to_le64(ptr);
  269. nchildren++;
  270. nilfs_btree_node_set_nchildren(node, nchildren);
  271. }
  272. /* Assume that the buffer head corresponding to node is locked. */
  273. static void nilfs_btree_node_delete(struct nilfs_bmap *btree,
  274. struct nilfs_btree_node *node,
  275. __u64 *keyp, __u64 *ptrp, int index)
  276. {
  277. __u64 key;
  278. __u64 ptr;
  279. __le64 *dkeys;
  280. __le64 *dptrs;
  281. int nchildren;
  282. dkeys = nilfs_btree_node_dkeys(node);
  283. dptrs = nilfs_btree_node_dptrs(node, btree);
  284. key = le64_to_cpu(dkeys[index]);
  285. ptr = le64_to_cpu(dptrs[index]);
  286. nchildren = nilfs_btree_node_get_nchildren(node);
  287. if (keyp != NULL)
  288. *keyp = key;
  289. if (ptrp != NULL)
  290. *ptrp = ptr;
  291. if (index < nchildren - 1) {
  292. memmove(dkeys + index, dkeys + index + 1,
  293. (nchildren - index - 1) * sizeof(*dkeys));
  294. memmove(dptrs + index, dptrs + index + 1,
  295. (nchildren - index - 1) * sizeof(*dptrs));
  296. }
  297. nchildren--;
  298. nilfs_btree_node_set_nchildren(node, nchildren);
  299. }
  300. static int nilfs_btree_node_lookup(const struct nilfs_btree_node *node,
  301. __u64 key, int *indexp)
  302. {
  303. __u64 nkey;
  304. int index, low, high, s;
  305. /* binary search */
  306. low = 0;
  307. high = nilfs_btree_node_get_nchildren(node) - 1;
  308. index = 0;
  309. s = 0;
  310. while (low <= high) {
  311. index = (low + high) / 2;
  312. nkey = nilfs_btree_node_get_key(node, index);
  313. if (nkey == key) {
  314. s = 0;
  315. goto out;
  316. } else if (nkey < key) {
  317. low = index + 1;
  318. s = -1;
  319. } else {
  320. high = index - 1;
  321. s = 1;
  322. }
  323. }
  324. /* adjust index */
  325. if (nilfs_btree_node_get_level(node) > NILFS_BTREE_LEVEL_NODE_MIN) {
  326. if (s > 0 && index > 0)
  327. index--;
  328. } else if (s < 0)
  329. index++;
  330. out:
  331. *indexp = index;
  332. return s == 0;
  333. }
  334. /**
  335. * nilfs_btree_node_broken - verify consistency of btree node
  336. * @node: btree node block to be examined
  337. * @size: node size (in bytes)
  338. * @blocknr: block number
  339. *
  340. * Return Value: If node is broken, 1 is returned. Otherwise, 0 is returned.
  341. */
  342. static int nilfs_btree_node_broken(const struct nilfs_btree_node *node,
  343. size_t size, sector_t blocknr)
  344. {
  345. int level, flags, nchildren;
  346. int ret = 0;
  347. level = nilfs_btree_node_get_level(node);
  348. flags = nilfs_btree_node_get_flags(node);
  349. nchildren = nilfs_btree_node_get_nchildren(node);
  350. if (unlikely(level < NILFS_BTREE_LEVEL_NODE_MIN ||
  351. level >= NILFS_BTREE_LEVEL_MAX ||
  352. (flags & NILFS_BTREE_NODE_ROOT) ||
  353. nchildren < 0 ||
  354. nchildren > NILFS_BTREE_NODE_NCHILDREN_MAX(size))) {
  355. printk(KERN_CRIT "NILFS: bad btree node (blocknr=%llu): "
  356. "level = %d, flags = 0x%x, nchildren = %d\n",
  357. (unsigned long long)blocknr, level, flags, nchildren);
  358. ret = 1;
  359. }
  360. return ret;
  361. }
  362. int nilfs_btree_broken_node_block(struct buffer_head *bh)
  363. {
  364. return nilfs_btree_node_broken((struct nilfs_btree_node *)bh->b_data,
  365. bh->b_size, bh->b_blocknr);
  366. }
  367. static inline struct nilfs_btree_node *
  368. nilfs_btree_get_root(const struct nilfs_bmap *btree)
  369. {
  370. return (struct nilfs_btree_node *)btree->b_u.u_data;
  371. }
  372. static inline struct nilfs_btree_node *
  373. nilfs_btree_get_nonroot_node(const struct nilfs_btree_path *path, int level)
  374. {
  375. return (struct nilfs_btree_node *)path[level].bp_bh->b_data;
  376. }
  377. static inline struct nilfs_btree_node *
  378. nilfs_btree_get_sib_node(const struct nilfs_btree_path *path, int level)
  379. {
  380. return (struct nilfs_btree_node *)path[level].bp_sib_bh->b_data;
  381. }
  382. static inline int nilfs_btree_height(const struct nilfs_bmap *btree)
  383. {
  384. return nilfs_btree_node_get_level(nilfs_btree_get_root(btree)) + 1;
  385. }
  386. static inline struct nilfs_btree_node *
  387. nilfs_btree_get_node(const struct nilfs_bmap *btree,
  388. const struct nilfs_btree_path *path,
  389. int level)
  390. {
  391. return (level == nilfs_btree_height(btree) - 1) ?
  392. nilfs_btree_get_root(btree) :
  393. nilfs_btree_get_nonroot_node(path, level);
  394. }
  395. static inline int
  396. nilfs_btree_bad_node(struct nilfs_btree_node *node, int level)
  397. {
  398. if (unlikely(nilfs_btree_node_get_level(node) != level)) {
  399. dump_stack();
  400. printk(KERN_CRIT "NILFS: btree level mismatch: %d != %d\n",
  401. nilfs_btree_node_get_level(node), level);
  402. return 1;
  403. }
  404. return 0;
  405. }
  406. static int nilfs_btree_do_lookup(const struct nilfs_bmap *btree,
  407. struct nilfs_btree_path *path,
  408. __u64 key, __u64 *ptrp, int minlevel)
  409. {
  410. struct nilfs_btree_node *node;
  411. __u64 ptr;
  412. int level, index, found, ncmax, ret;
  413. node = nilfs_btree_get_root(btree);
  414. level = nilfs_btree_node_get_level(node);
  415. if (level < minlevel || nilfs_btree_node_get_nchildren(node) <= 0)
  416. return -ENOENT;
  417. found = nilfs_btree_node_lookup(node, key, &index);
  418. ptr = nilfs_btree_node_get_ptr(btree, node, index);
  419. path[level].bp_bh = NULL;
  420. path[level].bp_index = index;
  421. ncmax = NILFS_BTREE_NODE_NCHILDREN_MAX(nilfs_btree_node_size(btree));
  422. for (level--; level >= minlevel; level--) {
  423. ret = nilfs_btree_get_block(btree, ptr, &path[level].bp_bh);
  424. if (ret < 0)
  425. return ret;
  426. node = nilfs_btree_get_nonroot_node(path, level);
  427. if (nilfs_btree_bad_node(node, level))
  428. return -EINVAL;
  429. if (!found)
  430. found = nilfs_btree_node_lookup(node, key, &index);
  431. else
  432. index = 0;
  433. if (index < ncmax) {
  434. ptr = nilfs_btree_node_get_ptr(btree, node, index);
  435. } else {
  436. WARN_ON(found || level != NILFS_BTREE_LEVEL_NODE_MIN);
  437. /* insert */
  438. ptr = NILFS_BMAP_INVALID_PTR;
  439. }
  440. path[level].bp_index = index;
  441. }
  442. if (!found)
  443. return -ENOENT;
  444. if (ptrp != NULL)
  445. *ptrp = ptr;
  446. return 0;
  447. }
  448. static int nilfs_btree_do_lookup_last(const struct nilfs_bmap *btree,
  449. struct nilfs_btree_path *path,
  450. __u64 *keyp, __u64 *ptrp)
  451. {
  452. struct nilfs_btree_node *node;
  453. __u64 ptr;
  454. int index, level, ret;
  455. node = nilfs_btree_get_root(btree);
  456. index = nilfs_btree_node_get_nchildren(node) - 1;
  457. if (index < 0)
  458. return -ENOENT;
  459. level = nilfs_btree_node_get_level(node);
  460. ptr = nilfs_btree_node_get_ptr(btree, node, index);
  461. path[level].bp_bh = NULL;
  462. path[level].bp_index = index;
  463. for (level--; level > 0; level--) {
  464. ret = nilfs_btree_get_block(btree, ptr, &path[level].bp_bh);
  465. if (ret < 0)
  466. return ret;
  467. node = nilfs_btree_get_nonroot_node(path, level);
  468. if (nilfs_btree_bad_node(node, level))
  469. return -EINVAL;
  470. index = nilfs_btree_node_get_nchildren(node) - 1;
  471. ptr = nilfs_btree_node_get_ptr(btree, node, index);
  472. path[level].bp_index = index;
  473. }
  474. if (keyp != NULL)
  475. *keyp = nilfs_btree_node_get_key(node, index);
  476. if (ptrp != NULL)
  477. *ptrp = ptr;
  478. return 0;
  479. }
  480. static int nilfs_btree_lookup(const struct nilfs_bmap *btree,
  481. __u64 key, int level, __u64 *ptrp)
  482. {
  483. struct nilfs_btree_path *path;
  484. int ret;
  485. path = nilfs_btree_alloc_path();
  486. if (path == NULL)
  487. return -ENOMEM;
  488. ret = nilfs_btree_do_lookup(btree, path, key, ptrp, level);
  489. nilfs_btree_free_path(path);
  490. return ret;
  491. }
  492. static int nilfs_btree_lookup_contig(const struct nilfs_bmap *btree,
  493. __u64 key, __u64 *ptrp, unsigned maxblocks)
  494. {
  495. struct nilfs_btree_path *path;
  496. struct nilfs_btree_node *node;
  497. struct inode *dat = NULL;
  498. __u64 ptr, ptr2;
  499. sector_t blocknr;
  500. int level = NILFS_BTREE_LEVEL_NODE_MIN;
  501. int ret, cnt, index, maxlevel;
  502. path = nilfs_btree_alloc_path();
  503. if (path == NULL)
  504. return -ENOMEM;
  505. ret = nilfs_btree_do_lookup(btree, path, key, &ptr, level);
  506. if (ret < 0)
  507. goto out;
  508. if (NILFS_BMAP_USE_VBN(btree)) {
  509. dat = nilfs_bmap_get_dat(btree);
  510. ret = nilfs_dat_translate(dat, ptr, &blocknr);
  511. if (ret < 0)
  512. goto out;
  513. ptr = blocknr;
  514. }
  515. cnt = 1;
  516. if (cnt == maxblocks)
  517. goto end;
  518. maxlevel = nilfs_btree_height(btree) - 1;
  519. node = nilfs_btree_get_node(btree, path, level);
  520. index = path[level].bp_index + 1;
  521. for (;;) {
  522. while (index < nilfs_btree_node_get_nchildren(node)) {
  523. if (nilfs_btree_node_get_key(node, index) !=
  524. key + cnt)
  525. goto end;
  526. ptr2 = nilfs_btree_node_get_ptr(btree, node, index);
  527. if (dat) {
  528. ret = nilfs_dat_translate(dat, ptr2, &blocknr);
  529. if (ret < 0)
  530. goto out;
  531. ptr2 = blocknr;
  532. }
  533. if (ptr2 != ptr + cnt || ++cnt == maxblocks)
  534. goto end;
  535. index++;
  536. continue;
  537. }
  538. if (level == maxlevel)
  539. break;
  540. /* look-up right sibling node */
  541. node = nilfs_btree_get_node(btree, path, level + 1);
  542. index = path[level + 1].bp_index + 1;
  543. if (index >= nilfs_btree_node_get_nchildren(node) ||
  544. nilfs_btree_node_get_key(node, index) != key + cnt)
  545. break;
  546. ptr2 = nilfs_btree_node_get_ptr(btree, node, index);
  547. path[level + 1].bp_index = index;
  548. brelse(path[level].bp_bh);
  549. path[level].bp_bh = NULL;
  550. ret = nilfs_btree_get_block(btree, ptr2, &path[level].bp_bh);
  551. if (ret < 0)
  552. goto out;
  553. node = nilfs_btree_get_nonroot_node(path, level);
  554. index = 0;
  555. path[level].bp_index = index;
  556. }
  557. end:
  558. *ptrp = ptr;
  559. ret = cnt;
  560. out:
  561. nilfs_btree_free_path(path);
  562. return ret;
  563. }
  564. static void nilfs_btree_promote_key(struct nilfs_bmap *btree,
  565. struct nilfs_btree_path *path,
  566. int level, __u64 key)
  567. {
  568. if (level < nilfs_btree_height(btree) - 1) {
  569. do {
  570. nilfs_btree_node_set_key(
  571. nilfs_btree_get_nonroot_node(path, level),
  572. path[level].bp_index, key);
  573. if (!buffer_dirty(path[level].bp_bh))
  574. nilfs_btnode_mark_dirty(path[level].bp_bh);
  575. } while ((path[level].bp_index == 0) &&
  576. (++level < nilfs_btree_height(btree) - 1));
  577. }
  578. /* root */
  579. if (level == nilfs_btree_height(btree) - 1) {
  580. nilfs_btree_node_set_key(nilfs_btree_get_root(btree),
  581. path[level].bp_index, key);
  582. }
  583. }
  584. static void nilfs_btree_do_insert(struct nilfs_bmap *btree,
  585. struct nilfs_btree_path *path,
  586. int level, __u64 *keyp, __u64 *ptrp)
  587. {
  588. struct nilfs_btree_node *node;
  589. if (level < nilfs_btree_height(btree) - 1) {
  590. node = nilfs_btree_get_nonroot_node(path, level);
  591. nilfs_btree_node_insert(btree, node, *keyp, *ptrp,
  592. path[level].bp_index);
  593. if (!buffer_dirty(path[level].bp_bh))
  594. nilfs_btnode_mark_dirty(path[level].bp_bh);
  595. if (path[level].bp_index == 0)
  596. nilfs_btree_promote_key(btree, path, level + 1,
  597. nilfs_btree_node_get_key(node,
  598. 0));
  599. } else {
  600. node = nilfs_btree_get_root(btree);
  601. nilfs_btree_node_insert(btree, node, *keyp, *ptrp,
  602. path[level].bp_index);
  603. }
  604. }
  605. static void nilfs_btree_carry_left(struct nilfs_bmap *btree,
  606. struct nilfs_btree_path *path,
  607. int level, __u64 *keyp, __u64 *ptrp)
  608. {
  609. struct nilfs_btree_node *node, *left;
  610. int nchildren, lnchildren, n, move;
  611. node = nilfs_btree_get_nonroot_node(path, level);
  612. left = nilfs_btree_get_sib_node(path, level);
  613. nchildren = nilfs_btree_node_get_nchildren(node);
  614. lnchildren = nilfs_btree_node_get_nchildren(left);
  615. move = 0;
  616. n = (nchildren + lnchildren + 1) / 2 - lnchildren;
  617. if (n > path[level].bp_index) {
  618. /* move insert point */
  619. n--;
  620. move = 1;
  621. }
  622. nilfs_btree_node_move_left(btree, left, node, n);
  623. if (!buffer_dirty(path[level].bp_bh))
  624. nilfs_btnode_mark_dirty(path[level].bp_bh);
  625. if (!buffer_dirty(path[level].bp_sib_bh))
  626. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  627. nilfs_btree_promote_key(btree, path, level + 1,
  628. nilfs_btree_node_get_key(node, 0));
  629. if (move) {
  630. brelse(path[level].bp_bh);
  631. path[level].bp_bh = path[level].bp_sib_bh;
  632. path[level].bp_sib_bh = NULL;
  633. path[level].bp_index += lnchildren;
  634. path[level + 1].bp_index--;
  635. } else {
  636. brelse(path[level].bp_sib_bh);
  637. path[level].bp_sib_bh = NULL;
  638. path[level].bp_index -= n;
  639. }
  640. nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
  641. }
  642. static void nilfs_btree_carry_right(struct nilfs_bmap *btree,
  643. struct nilfs_btree_path *path,
  644. int level, __u64 *keyp, __u64 *ptrp)
  645. {
  646. struct nilfs_btree_node *node, *right;
  647. int nchildren, rnchildren, n, move;
  648. node = nilfs_btree_get_nonroot_node(path, level);
  649. right = nilfs_btree_get_sib_node(path, level);
  650. nchildren = nilfs_btree_node_get_nchildren(node);
  651. rnchildren = nilfs_btree_node_get_nchildren(right);
  652. move = 0;
  653. n = (nchildren + rnchildren + 1) / 2 - rnchildren;
  654. if (n > nchildren - path[level].bp_index) {
  655. /* move insert point */
  656. n--;
  657. move = 1;
  658. }
  659. nilfs_btree_node_move_right(btree, node, right, n);
  660. if (!buffer_dirty(path[level].bp_bh))
  661. nilfs_btnode_mark_dirty(path[level].bp_bh);
  662. if (!buffer_dirty(path[level].bp_sib_bh))
  663. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  664. path[level + 1].bp_index++;
  665. nilfs_btree_promote_key(btree, path, level + 1,
  666. nilfs_btree_node_get_key(right, 0));
  667. path[level + 1].bp_index--;
  668. if (move) {
  669. brelse(path[level].bp_bh);
  670. path[level].bp_bh = path[level].bp_sib_bh;
  671. path[level].bp_sib_bh = NULL;
  672. path[level].bp_index -= nilfs_btree_node_get_nchildren(node);
  673. path[level + 1].bp_index++;
  674. } else {
  675. brelse(path[level].bp_sib_bh);
  676. path[level].bp_sib_bh = NULL;
  677. }
  678. nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
  679. }
  680. static void nilfs_btree_split(struct nilfs_bmap *btree,
  681. struct nilfs_btree_path *path,
  682. int level, __u64 *keyp, __u64 *ptrp)
  683. {
  684. struct nilfs_btree_node *node, *right;
  685. __u64 newkey;
  686. __u64 newptr;
  687. int nchildren, n, move;
  688. node = nilfs_btree_get_nonroot_node(path, level);
  689. right = nilfs_btree_get_sib_node(path, level);
  690. nchildren = nilfs_btree_node_get_nchildren(node);
  691. move = 0;
  692. n = (nchildren + 1) / 2;
  693. if (n > nchildren - path[level].bp_index) {
  694. n--;
  695. move = 1;
  696. }
  697. nilfs_btree_node_move_right(btree, node, right, n);
  698. if (!buffer_dirty(path[level].bp_bh))
  699. nilfs_btnode_mark_dirty(path[level].bp_bh);
  700. if (!buffer_dirty(path[level].bp_sib_bh))
  701. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  702. newkey = nilfs_btree_node_get_key(right, 0);
  703. newptr = path[level].bp_newreq.bpr_ptr;
  704. if (move) {
  705. path[level].bp_index -= nilfs_btree_node_get_nchildren(node);
  706. nilfs_btree_node_insert(btree, right, *keyp, *ptrp,
  707. path[level].bp_index);
  708. *keyp = nilfs_btree_node_get_key(right, 0);
  709. *ptrp = path[level].bp_newreq.bpr_ptr;
  710. brelse(path[level].bp_bh);
  711. path[level].bp_bh = path[level].bp_sib_bh;
  712. path[level].bp_sib_bh = NULL;
  713. } else {
  714. nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
  715. *keyp = nilfs_btree_node_get_key(right, 0);
  716. *ptrp = path[level].bp_newreq.bpr_ptr;
  717. brelse(path[level].bp_sib_bh);
  718. path[level].bp_sib_bh = NULL;
  719. }
  720. path[level + 1].bp_index++;
  721. }
  722. static void nilfs_btree_grow(struct nilfs_bmap *btree,
  723. struct nilfs_btree_path *path,
  724. int level, __u64 *keyp, __u64 *ptrp)
  725. {
  726. struct nilfs_btree_node *root, *child;
  727. int n;
  728. root = nilfs_btree_get_root(btree);
  729. child = nilfs_btree_get_sib_node(path, level);
  730. n = nilfs_btree_node_get_nchildren(root);
  731. nilfs_btree_node_move_right(btree, root, child, n);
  732. nilfs_btree_node_set_level(root, level + 1);
  733. if (!buffer_dirty(path[level].bp_sib_bh))
  734. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  735. path[level].bp_bh = path[level].bp_sib_bh;
  736. path[level].bp_sib_bh = NULL;
  737. nilfs_btree_do_insert(btree, path, level, keyp, ptrp);
  738. *keyp = nilfs_btree_node_get_key(child, 0);
  739. *ptrp = path[level].bp_newreq.bpr_ptr;
  740. }
  741. static __u64 nilfs_btree_find_near(const struct nilfs_bmap *btree,
  742. const struct nilfs_btree_path *path)
  743. {
  744. struct nilfs_btree_node *node;
  745. int level;
  746. if (path == NULL)
  747. return NILFS_BMAP_INVALID_PTR;
  748. /* left sibling */
  749. level = NILFS_BTREE_LEVEL_NODE_MIN;
  750. if (path[level].bp_index > 0) {
  751. node = nilfs_btree_get_node(btree, path, level);
  752. return nilfs_btree_node_get_ptr(btree, node,
  753. path[level].bp_index - 1);
  754. }
  755. /* parent */
  756. level = NILFS_BTREE_LEVEL_NODE_MIN + 1;
  757. if (level <= nilfs_btree_height(btree) - 1) {
  758. node = nilfs_btree_get_node(btree, path, level);
  759. return nilfs_btree_node_get_ptr(btree, node,
  760. path[level].bp_index);
  761. }
  762. return NILFS_BMAP_INVALID_PTR;
  763. }
  764. static __u64 nilfs_btree_find_target_v(const struct nilfs_bmap *btree,
  765. const struct nilfs_btree_path *path,
  766. __u64 key)
  767. {
  768. __u64 ptr;
  769. ptr = nilfs_bmap_find_target_seq(btree, key);
  770. if (ptr != NILFS_BMAP_INVALID_PTR)
  771. /* sequential access */
  772. return ptr;
  773. else {
  774. ptr = nilfs_btree_find_near(btree, path);
  775. if (ptr != NILFS_BMAP_INVALID_PTR)
  776. /* near */
  777. return ptr;
  778. }
  779. /* block group */
  780. return nilfs_bmap_find_target_in_group(btree);
  781. }
  782. static int nilfs_btree_prepare_insert(struct nilfs_bmap *btree,
  783. struct nilfs_btree_path *path,
  784. int *levelp, __u64 key, __u64 ptr,
  785. struct nilfs_bmap_stats *stats)
  786. {
  787. struct buffer_head *bh;
  788. struct nilfs_btree_node *node, *parent, *sib;
  789. __u64 sibptr;
  790. int pindex, level, ncmax, ret;
  791. struct inode *dat = NULL;
  792. stats->bs_nblocks = 0;
  793. level = NILFS_BTREE_LEVEL_DATA;
  794. /* allocate a new ptr for data block */
  795. if (NILFS_BMAP_USE_VBN(btree)) {
  796. path[level].bp_newreq.bpr_ptr =
  797. nilfs_btree_find_target_v(btree, path, key);
  798. dat = nilfs_bmap_get_dat(btree);
  799. }
  800. ret = nilfs_bmap_prepare_alloc_ptr(btree, &path[level].bp_newreq, dat);
  801. if (ret < 0)
  802. goto err_out_data;
  803. ncmax = NILFS_BTREE_NODE_NCHILDREN_MAX(nilfs_btree_node_size(btree));
  804. for (level = NILFS_BTREE_LEVEL_NODE_MIN;
  805. level < nilfs_btree_height(btree) - 1;
  806. level++) {
  807. node = nilfs_btree_get_nonroot_node(path, level);
  808. if (nilfs_btree_node_get_nchildren(node) < ncmax) {
  809. path[level].bp_op = nilfs_btree_do_insert;
  810. stats->bs_nblocks++;
  811. goto out;
  812. }
  813. parent = nilfs_btree_get_node(btree, path, level + 1);
  814. pindex = path[level + 1].bp_index;
  815. /* left sibling */
  816. if (pindex > 0) {
  817. sibptr = nilfs_btree_node_get_ptr(btree, parent,
  818. pindex - 1);
  819. ret = nilfs_btree_get_block(btree, sibptr, &bh);
  820. if (ret < 0)
  821. goto err_out_child_node;
  822. sib = (struct nilfs_btree_node *)bh->b_data;
  823. if (nilfs_btree_node_get_nchildren(sib) < ncmax) {
  824. path[level].bp_sib_bh = bh;
  825. path[level].bp_op = nilfs_btree_carry_left;
  826. stats->bs_nblocks++;
  827. goto out;
  828. } else
  829. brelse(bh);
  830. }
  831. /* right sibling */
  832. if (pindex <
  833. nilfs_btree_node_get_nchildren(parent) - 1) {
  834. sibptr = nilfs_btree_node_get_ptr(btree, parent,
  835. pindex + 1);
  836. ret = nilfs_btree_get_block(btree, sibptr, &bh);
  837. if (ret < 0)
  838. goto err_out_child_node;
  839. sib = (struct nilfs_btree_node *)bh->b_data;
  840. if (nilfs_btree_node_get_nchildren(sib) < ncmax) {
  841. path[level].bp_sib_bh = bh;
  842. path[level].bp_op = nilfs_btree_carry_right;
  843. stats->bs_nblocks++;
  844. goto out;
  845. } else
  846. brelse(bh);
  847. }
  848. /* split */
  849. path[level].bp_newreq.bpr_ptr =
  850. path[level - 1].bp_newreq.bpr_ptr + 1;
  851. ret = nilfs_bmap_prepare_alloc_ptr(btree,
  852. &path[level].bp_newreq, dat);
  853. if (ret < 0)
  854. goto err_out_child_node;
  855. ret = nilfs_btree_get_new_block(btree,
  856. path[level].bp_newreq.bpr_ptr,
  857. &bh);
  858. if (ret < 0)
  859. goto err_out_curr_node;
  860. stats->bs_nblocks++;
  861. nilfs_btree_node_init(btree,
  862. (struct nilfs_btree_node *)bh->b_data,
  863. 0, level, 0, NULL, NULL);
  864. path[level].bp_sib_bh = bh;
  865. path[level].bp_op = nilfs_btree_split;
  866. }
  867. /* root */
  868. node = nilfs_btree_get_root(btree);
  869. if (nilfs_btree_node_get_nchildren(node) <
  870. NILFS_BTREE_ROOT_NCHILDREN_MAX) {
  871. path[level].bp_op = nilfs_btree_do_insert;
  872. stats->bs_nblocks++;
  873. goto out;
  874. }
  875. /* grow */
  876. path[level].bp_newreq.bpr_ptr = path[level - 1].bp_newreq.bpr_ptr + 1;
  877. ret = nilfs_bmap_prepare_alloc_ptr(btree, &path[level].bp_newreq, dat);
  878. if (ret < 0)
  879. goto err_out_child_node;
  880. ret = nilfs_btree_get_new_block(btree, path[level].bp_newreq.bpr_ptr,
  881. &bh);
  882. if (ret < 0)
  883. goto err_out_curr_node;
  884. nilfs_btree_node_init(btree, (struct nilfs_btree_node *)bh->b_data,
  885. 0, level, 0, NULL, NULL);
  886. path[level].bp_sib_bh = bh;
  887. path[level].bp_op = nilfs_btree_grow;
  888. level++;
  889. path[level].bp_op = nilfs_btree_do_insert;
  890. /* a newly-created node block and a data block are added */
  891. stats->bs_nblocks += 2;
  892. /* success */
  893. out:
  894. *levelp = level;
  895. return ret;
  896. /* error */
  897. err_out_curr_node:
  898. nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
  899. err_out_child_node:
  900. for (level--; level > NILFS_BTREE_LEVEL_DATA; level--) {
  901. nilfs_btnode_delete(path[level].bp_sib_bh);
  902. nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
  903. }
  904. nilfs_bmap_abort_alloc_ptr(btree, &path[level].bp_newreq, dat);
  905. err_out_data:
  906. *levelp = level;
  907. stats->bs_nblocks = 0;
  908. return ret;
  909. }
  910. static void nilfs_btree_commit_insert(struct nilfs_bmap *btree,
  911. struct nilfs_btree_path *path,
  912. int maxlevel, __u64 key, __u64 ptr)
  913. {
  914. struct inode *dat = NULL;
  915. int level;
  916. set_buffer_nilfs_volatile((struct buffer_head *)((unsigned long)ptr));
  917. ptr = path[NILFS_BTREE_LEVEL_DATA].bp_newreq.bpr_ptr;
  918. if (NILFS_BMAP_USE_VBN(btree)) {
  919. nilfs_bmap_set_target_v(btree, key, ptr);
  920. dat = nilfs_bmap_get_dat(btree);
  921. }
  922. for (level = NILFS_BTREE_LEVEL_NODE_MIN; level <= maxlevel; level++) {
  923. nilfs_bmap_commit_alloc_ptr(btree,
  924. &path[level - 1].bp_newreq, dat);
  925. path[level].bp_op(btree, path, level, &key, &ptr);
  926. }
  927. if (!nilfs_bmap_dirty(btree))
  928. nilfs_bmap_set_dirty(btree);
  929. }
  930. static int nilfs_btree_insert(struct nilfs_bmap *btree, __u64 key, __u64 ptr)
  931. {
  932. struct nilfs_btree_path *path;
  933. struct nilfs_bmap_stats stats;
  934. int level, ret;
  935. path = nilfs_btree_alloc_path();
  936. if (path == NULL)
  937. return -ENOMEM;
  938. ret = nilfs_btree_do_lookup(btree, path, key, NULL,
  939. NILFS_BTREE_LEVEL_NODE_MIN);
  940. if (ret != -ENOENT) {
  941. if (ret == 0)
  942. ret = -EEXIST;
  943. goto out;
  944. }
  945. ret = nilfs_btree_prepare_insert(btree, path, &level, key, ptr, &stats);
  946. if (ret < 0)
  947. goto out;
  948. nilfs_btree_commit_insert(btree, path, level, key, ptr);
  949. nilfs_bmap_add_blocks(btree, stats.bs_nblocks);
  950. out:
  951. nilfs_btree_free_path(path);
  952. return ret;
  953. }
  954. static void nilfs_btree_do_delete(struct nilfs_bmap *btree,
  955. struct nilfs_btree_path *path,
  956. int level, __u64 *keyp, __u64 *ptrp)
  957. {
  958. struct nilfs_btree_node *node;
  959. if (level < nilfs_btree_height(btree) - 1) {
  960. node = nilfs_btree_get_nonroot_node(path, level);
  961. nilfs_btree_node_delete(btree, node, keyp, ptrp,
  962. path[level].bp_index);
  963. if (!buffer_dirty(path[level].bp_bh))
  964. nilfs_btnode_mark_dirty(path[level].bp_bh);
  965. if (path[level].bp_index == 0)
  966. nilfs_btree_promote_key(btree, path, level + 1,
  967. nilfs_btree_node_get_key(node, 0));
  968. } else {
  969. node = nilfs_btree_get_root(btree);
  970. nilfs_btree_node_delete(btree, node, keyp, ptrp,
  971. path[level].bp_index);
  972. }
  973. }
  974. static void nilfs_btree_borrow_left(struct nilfs_bmap *btree,
  975. struct nilfs_btree_path *path,
  976. int level, __u64 *keyp, __u64 *ptrp)
  977. {
  978. struct nilfs_btree_node *node, *left;
  979. int nchildren, lnchildren, n;
  980. nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
  981. node = nilfs_btree_get_nonroot_node(path, level);
  982. left = nilfs_btree_get_sib_node(path, level);
  983. nchildren = nilfs_btree_node_get_nchildren(node);
  984. lnchildren = nilfs_btree_node_get_nchildren(left);
  985. n = (nchildren + lnchildren) / 2 - nchildren;
  986. nilfs_btree_node_move_right(btree, left, node, n);
  987. if (!buffer_dirty(path[level].bp_bh))
  988. nilfs_btnode_mark_dirty(path[level].bp_bh);
  989. if (!buffer_dirty(path[level].bp_sib_bh))
  990. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  991. nilfs_btree_promote_key(btree, path, level + 1,
  992. nilfs_btree_node_get_key(node, 0));
  993. brelse(path[level].bp_sib_bh);
  994. path[level].bp_sib_bh = NULL;
  995. path[level].bp_index += n;
  996. }
  997. static void nilfs_btree_borrow_right(struct nilfs_bmap *btree,
  998. struct nilfs_btree_path *path,
  999. int level, __u64 *keyp, __u64 *ptrp)
  1000. {
  1001. struct nilfs_btree_node *node, *right;
  1002. int nchildren, rnchildren, n;
  1003. nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
  1004. node = nilfs_btree_get_nonroot_node(path, level);
  1005. right = nilfs_btree_get_sib_node(path, level);
  1006. nchildren = nilfs_btree_node_get_nchildren(node);
  1007. rnchildren = nilfs_btree_node_get_nchildren(right);
  1008. n = (nchildren + rnchildren) / 2 - nchildren;
  1009. nilfs_btree_node_move_left(btree, node, right, n);
  1010. if (!buffer_dirty(path[level].bp_bh))
  1011. nilfs_btnode_mark_dirty(path[level].bp_bh);
  1012. if (!buffer_dirty(path[level].bp_sib_bh))
  1013. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  1014. path[level + 1].bp_index++;
  1015. nilfs_btree_promote_key(btree, path, level + 1,
  1016. nilfs_btree_node_get_key(right, 0));
  1017. path[level + 1].bp_index--;
  1018. brelse(path[level].bp_sib_bh);
  1019. path[level].bp_sib_bh = NULL;
  1020. }
  1021. static void nilfs_btree_concat_left(struct nilfs_bmap *btree,
  1022. struct nilfs_btree_path *path,
  1023. int level, __u64 *keyp, __u64 *ptrp)
  1024. {
  1025. struct nilfs_btree_node *node, *left;
  1026. int n;
  1027. nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
  1028. node = nilfs_btree_get_nonroot_node(path, level);
  1029. left = nilfs_btree_get_sib_node(path, level);
  1030. n = nilfs_btree_node_get_nchildren(node);
  1031. nilfs_btree_node_move_left(btree, left, node, n);
  1032. if (!buffer_dirty(path[level].bp_sib_bh))
  1033. nilfs_btnode_mark_dirty(path[level].bp_sib_bh);
  1034. nilfs_btnode_delete(path[level].bp_bh);
  1035. path[level].bp_bh = path[level].bp_sib_bh;
  1036. path[level].bp_sib_bh = NULL;
  1037. path[level].bp_index += nilfs_btree_node_get_nchildren(left);
  1038. }
  1039. static void nilfs_btree_concat_right(struct nilfs_bmap *btree,
  1040. struct nilfs_btree_path *path,
  1041. int level, __u64 *keyp, __u64 *ptrp)
  1042. {
  1043. struct nilfs_btree_node *node, *right;
  1044. int n;
  1045. nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
  1046. node = nilfs_btree_get_nonroot_node(path, level);
  1047. right = nilfs_btree_get_sib_node(path, level);
  1048. n = nilfs_btree_node_get_nchildren(right);
  1049. nilfs_btree_node_move_left(btree, node, right, n);
  1050. if (!buffer_dirty(path[level].bp_bh))
  1051. nilfs_btnode_mark_dirty(path[level].bp_bh);
  1052. nilfs_btnode_delete(path[level].bp_sib_bh);
  1053. path[level].bp_sib_bh = NULL;
  1054. path[level + 1].bp_index++;
  1055. }
  1056. static void nilfs_btree_shrink(struct nilfs_bmap *btree,
  1057. struct nilfs_btree_path *path,
  1058. int level, __u64 *keyp, __u64 *ptrp)
  1059. {
  1060. struct nilfs_btree_node *root, *child;
  1061. int n;
  1062. nilfs_btree_do_delete(btree, path, level, keyp, ptrp);
  1063. root = nilfs_btree_get_root(btree);
  1064. child = nilfs_btree_get_nonroot_node(path, level);
  1065. nilfs_btree_node_delete(btree, root, NULL, NULL, 0);
  1066. nilfs_btree_node_set_level(root, level);
  1067. n = nilfs_btree_node_get_nchildren(child);
  1068. nilfs_btree_node_move_left(btree, root, child, n);
  1069. nilfs_btnode_delete(path[level].bp_bh);
  1070. path[level].bp_bh = NULL;
  1071. }
  1072. static int nilfs_btree_prepare_delete(struct nilfs_bmap *btree,
  1073. struct nilfs_btree_path *path,
  1074. int *levelp,
  1075. struct nilfs_bmap_stats *stats,
  1076. struct inode *dat)
  1077. {
  1078. struct buffer_head *bh;
  1079. struct nilfs_btree_node *node, *parent, *sib;
  1080. __u64 sibptr;
  1081. int pindex, level, ncmin, ret;
  1082. ret = 0;
  1083. stats->bs_nblocks = 0;
  1084. ncmin = NILFS_BTREE_NODE_NCHILDREN_MIN(nilfs_btree_node_size(btree));
  1085. for (level = NILFS_BTREE_LEVEL_NODE_MIN;
  1086. level < nilfs_btree_height(btree) - 1;
  1087. level++) {
  1088. node = nilfs_btree_get_nonroot_node(path, level);
  1089. path[level].bp_oldreq.bpr_ptr =
  1090. nilfs_btree_node_get_ptr(btree, node,
  1091. path[level].bp_index);
  1092. ret = nilfs_bmap_prepare_end_ptr(btree,
  1093. &path[level].bp_oldreq, dat);
  1094. if (ret < 0)
  1095. goto err_out_child_node;
  1096. if (nilfs_btree_node_get_nchildren(node) > ncmin) {
  1097. path[level].bp_op = nilfs_btree_do_delete;
  1098. stats->bs_nblocks++;
  1099. goto out;
  1100. }
  1101. parent = nilfs_btree_get_node(btree, path, level + 1);
  1102. pindex = path[level + 1].bp_index;
  1103. if (pindex > 0) {
  1104. /* left sibling */
  1105. sibptr = nilfs_btree_node_get_ptr(btree, parent,
  1106. pindex - 1);
  1107. ret = nilfs_btree_get_block(btree, sibptr, &bh);
  1108. if (ret < 0)
  1109. goto err_out_curr_node;
  1110. sib = (struct nilfs_btree_node *)bh->b_data;
  1111. if (nilfs_btree_node_get_nchildren(sib) > ncmin) {
  1112. path[level].bp_sib_bh = bh;
  1113. path[level].bp_op = nilfs_btree_borrow_left;
  1114. stats->bs_nblocks++;
  1115. goto out;
  1116. } else {
  1117. path[level].bp_sib_bh = bh;
  1118. path[level].bp_op = nilfs_btree_concat_left;
  1119. stats->bs_nblocks++;
  1120. /* continue; */
  1121. }
  1122. } else if (pindex <
  1123. nilfs_btree_node_get_nchildren(parent) - 1) {
  1124. /* right sibling */
  1125. sibptr = nilfs_btree_node_get_ptr(btree, parent,
  1126. pindex + 1);
  1127. ret = nilfs_btree_get_block(btree, sibptr, &bh);
  1128. if (ret < 0)
  1129. goto err_out_curr_node;
  1130. sib = (struct nilfs_btree_node *)bh->b_data;
  1131. if (nilfs_btree_node_get_nchildren(sib) > ncmin) {
  1132. path[level].bp_sib_bh = bh;
  1133. path[level].bp_op = nilfs_btree_borrow_right;
  1134. stats->bs_nblocks++;
  1135. goto out;
  1136. } else {
  1137. path[level].bp_sib_bh = bh;
  1138. path[level].bp_op = nilfs_btree_concat_right;
  1139. stats->bs_nblocks++;
  1140. /* continue; */
  1141. }
  1142. } else {
  1143. /* no siblings */
  1144. /* the only child of the root node */
  1145. WARN_ON(level != nilfs_btree_height(btree) - 2);
  1146. if (nilfs_btree_node_get_nchildren(node) - 1 <=
  1147. NILFS_BTREE_ROOT_NCHILDREN_MAX) {
  1148. path[level].bp_op = nilfs_btree_shrink;
  1149. stats->bs_nblocks += 2;
  1150. } else {
  1151. path[level].bp_op = nilfs_btree_do_delete;
  1152. stats->bs_nblocks++;
  1153. }
  1154. goto out;
  1155. }
  1156. }
  1157. node = nilfs_btree_get_root(btree);
  1158. path[level].bp_oldreq.bpr_ptr =
  1159. nilfs_btree_node_get_ptr(btree, node, path[level].bp_index);
  1160. ret = nilfs_bmap_prepare_end_ptr(btree, &path[level].bp_oldreq, dat);
  1161. if (ret < 0)
  1162. goto err_out_child_node;
  1163. /* child of the root node is deleted */
  1164. path[level].bp_op = nilfs_btree_do_delete;
  1165. stats->bs_nblocks++;
  1166. /* success */
  1167. out:
  1168. *levelp = level;
  1169. return ret;
  1170. /* error */
  1171. err_out_curr_node:
  1172. nilfs_bmap_abort_end_ptr(btree, &path[level].bp_oldreq, dat);
  1173. err_out_child_node:
  1174. for (level--; level >= NILFS_BTREE_LEVEL_NODE_MIN; level--) {
  1175. brelse(path[level].bp_sib_bh);
  1176. nilfs_bmap_abort_end_ptr(btree, &path[level].bp_oldreq, dat);
  1177. }
  1178. *levelp = level;
  1179. stats->bs_nblocks = 0;
  1180. return ret;
  1181. }
  1182. static void nilfs_btree_commit_delete(struct nilfs_bmap *btree,
  1183. struct nilfs_btree_path *path,
  1184. int maxlevel, struct inode *dat)
  1185. {
  1186. int level;
  1187. for (level = NILFS_BTREE_LEVEL_NODE_MIN; level <= maxlevel; level++) {
  1188. nilfs_bmap_commit_end_ptr(btree, &path[level].bp_oldreq, dat);
  1189. path[level].bp_op(btree, path, level, NULL, NULL);
  1190. }
  1191. if (!nilfs_bmap_dirty(btree))
  1192. nilfs_bmap_set_dirty(btree);
  1193. }
  1194. static int nilfs_btree_delete(struct nilfs_bmap *btree, __u64 key)
  1195. {
  1196. struct nilfs_btree_path *path;
  1197. struct nilfs_bmap_stats stats;
  1198. struct inode *dat;
  1199. int level, ret;
  1200. path = nilfs_btree_alloc_path();
  1201. if (path == NULL)
  1202. return -ENOMEM;
  1203. ret = nilfs_btree_do_lookup(btree, path, key, NULL,
  1204. NILFS_BTREE_LEVEL_NODE_MIN);
  1205. if (ret < 0)
  1206. goto out;
  1207. dat = NILFS_BMAP_USE_VBN(btree) ? nilfs_bmap_get_dat(btree) : NULL;
  1208. ret = nilfs_btree_prepare_delete(btree, path, &level, &stats, dat);
  1209. if (ret < 0)
  1210. goto out;
  1211. nilfs_btree_commit_delete(btree, path, level, dat);
  1212. nilfs_bmap_sub_blocks(btree, stats.bs_nblocks);
  1213. out:
  1214. nilfs_btree_free_path(path);
  1215. return ret;
  1216. }
  1217. static int nilfs_btree_last_key(const struct nilfs_bmap *btree, __u64 *keyp)
  1218. {
  1219. struct nilfs_btree_path *path;
  1220. int ret;
  1221. path = nilfs_btree_alloc_path();
  1222. if (path == NULL)
  1223. return -ENOMEM;
  1224. ret = nilfs_btree_do_lookup_last(btree, path, keyp, NULL);
  1225. nilfs_btree_free_path(path);
  1226. return ret;
  1227. }
  1228. static int nilfs_btree_check_delete(struct nilfs_bmap *btree, __u64 key)
  1229. {
  1230. struct buffer_head *bh;
  1231. struct nilfs_btree_node *root, *node;
  1232. __u64 maxkey, nextmaxkey;
  1233. __u64 ptr;
  1234. int nchildren, ret;
  1235. root = nilfs_btree_get_root(btree);
  1236. switch (nilfs_btree_height(btree)) {
  1237. case 2:
  1238. bh = NULL;
  1239. node = root;
  1240. break;
  1241. case 3:
  1242. nchildren = nilfs_btree_node_get_nchildren(root);
  1243. if (nchildren > 1)
  1244. return 0;
  1245. ptr = nilfs_btree_node_get_ptr(btree, root, nchildren - 1);
  1246. ret = nilfs_btree_get_block(btree, ptr, &bh);
  1247. if (ret < 0)
  1248. return ret;
  1249. node = (struct nilfs_btree_node *)bh->b_data;
  1250. break;
  1251. default:
  1252. return 0;
  1253. }
  1254. nchildren = nilfs_btree_node_get_nchildren(node);
  1255. maxkey = nilfs_btree_node_get_key(node, nchildren - 1);
  1256. nextmaxkey = (nchildren > 1) ?
  1257. nilfs_btree_node_get_key(node, nchildren - 2) : 0;
  1258. if (bh != NULL)
  1259. brelse(bh);
  1260. return (maxkey == key) && (nextmaxkey < NILFS_BMAP_LARGE_LOW);
  1261. }
  1262. static int nilfs_btree_gather_data(struct nilfs_bmap *btree,
  1263. __u64 *keys, __u64 *ptrs, int nitems)
  1264. {
  1265. struct buffer_head *bh;
  1266. struct nilfs_btree_node *node, *root;
  1267. __le64 *dkeys;
  1268. __le64 *dptrs;
  1269. __u64 ptr;
  1270. int nchildren, i, ret;
  1271. root = nilfs_btree_get_root(btree);
  1272. switch (nilfs_btree_height(btree)) {
  1273. case 2:
  1274. bh = NULL;
  1275. node = root;
  1276. break;
  1277. case 3:
  1278. nchildren = nilfs_btree_node_get_nchildren(root);
  1279. WARN_ON(nchildren > 1);
  1280. ptr = nilfs_btree_node_get_ptr(btree, root, nchildren - 1);
  1281. ret = nilfs_btree_get_block(btree, ptr, &bh);
  1282. if (ret < 0)
  1283. return ret;
  1284. node = (struct nilfs_btree_node *)bh->b_data;
  1285. break;
  1286. default:
  1287. node = NULL;
  1288. return -EINVAL;
  1289. }
  1290. nchildren = nilfs_btree_node_get_nchildren(node);
  1291. if (nchildren < nitems)
  1292. nitems = nchildren;
  1293. dkeys = nilfs_btree_node_dkeys(node);
  1294. dptrs = nilfs_btree_node_dptrs(node, btree);
  1295. for (i = 0; i < nitems; i++) {
  1296. keys[i] = le64_to_cpu(dkeys[i]);
  1297. ptrs[i] = le64_to_cpu(dptrs[i]);
  1298. }
  1299. if (bh != NULL)
  1300. brelse(bh);
  1301. return nitems;
  1302. }
  1303. static int
  1304. nilfs_btree_prepare_convert_and_insert(struct nilfs_bmap *btree, __u64 key,
  1305. union nilfs_bmap_ptr_req *dreq,
  1306. union nilfs_bmap_ptr_req *nreq,
  1307. struct buffer_head **bhp,
  1308. struct nilfs_bmap_stats *stats)
  1309. {
  1310. struct buffer_head *bh;
  1311. struct inode *dat = NULL;
  1312. int ret;
  1313. stats->bs_nblocks = 0;
  1314. /* for data */
  1315. /* cannot find near ptr */
  1316. if (NILFS_BMAP_USE_VBN(btree)) {
  1317. dreq->bpr_ptr = nilfs_btree_find_target_v(btree, NULL, key);
  1318. dat = nilfs_bmap_get_dat(btree);
  1319. }
  1320. ret = nilfs_bmap_prepare_alloc_ptr(btree, dreq, dat);
  1321. if (ret < 0)
  1322. return ret;
  1323. *bhp = NULL;
  1324. stats->bs_nblocks++;
  1325. if (nreq != NULL) {
  1326. nreq->bpr_ptr = dreq->bpr_ptr + 1;
  1327. ret = nilfs_bmap_prepare_alloc_ptr(btree, nreq, dat);
  1328. if (ret < 0)
  1329. goto err_out_dreq;
  1330. ret = nilfs_btree_get_new_block(btree, nreq->bpr_ptr, &bh);
  1331. if (ret < 0)
  1332. goto err_out_nreq;
  1333. *bhp = bh;
  1334. stats->bs_nblocks++;
  1335. }
  1336. /* success */
  1337. return 0;
  1338. /* error */
  1339. err_out_nreq:
  1340. nilfs_bmap_abort_alloc_ptr(btree, nreq, dat);
  1341. err_out_dreq:
  1342. nilfs_bmap_abort_alloc_ptr(btree, dreq, dat);
  1343. stats->bs_nblocks = 0;
  1344. return ret;
  1345. }
  1346. static void
  1347. nilfs_btree_commit_convert_and_insert(struct nilfs_bmap *btree,
  1348. __u64 key, __u64 ptr,
  1349. const __u64 *keys, const __u64 *ptrs,
  1350. int n,
  1351. union nilfs_bmap_ptr_req *dreq,
  1352. union nilfs_bmap_ptr_req *nreq,
  1353. struct buffer_head *bh)
  1354. {
  1355. struct nilfs_btree_node *node;
  1356. struct inode *dat;
  1357. __u64 tmpptr;
  1358. /* free resources */
  1359. if (btree->b_ops->bop_clear != NULL)
  1360. btree->b_ops->bop_clear(btree);
  1361. /* ptr must be a pointer to a buffer head. */
  1362. set_buffer_nilfs_volatile((struct buffer_head *)((unsigned long)ptr));
  1363. /* convert and insert */
  1364. dat = NILFS_BMAP_USE_VBN(btree) ? nilfs_bmap_get_dat(btree) : NULL;
  1365. nilfs_btree_init(btree);
  1366. if (nreq != NULL) {
  1367. nilfs_bmap_commit_alloc_ptr(btree, dreq, dat);
  1368. nilfs_bmap_commit_alloc_ptr(btree, nreq, dat);
  1369. /* create child node at level 1 */
  1370. node = (struct nilfs_btree_node *)bh->b_data;
  1371. nilfs_btree_node_init(btree, node, 0, 1, n, keys, ptrs);
  1372. nilfs_btree_node_insert(btree, node, key, dreq->bpr_ptr, n);
  1373. if (!buffer_dirty(bh))
  1374. nilfs_btnode_mark_dirty(bh);
  1375. if (!nilfs_bmap_dirty(btree))
  1376. nilfs_bmap_set_dirty(btree);
  1377. brelse(bh);
  1378. /* create root node at level 2 */
  1379. node = nilfs_btree_get_root(btree);
  1380. tmpptr = nreq->bpr_ptr;
  1381. nilfs_btree_node_init(btree, node, NILFS_BTREE_NODE_ROOT,
  1382. 2, 1, &keys[0], &tmpptr);
  1383. } else {
  1384. nilfs_bmap_commit_alloc_ptr(btree, dreq, dat);
  1385. /* create root node at level 1 */
  1386. node = nilfs_btree_get_root(btree);
  1387. nilfs_btree_node_init(btree, node, NILFS_BTREE_NODE_ROOT,
  1388. 1, n, keys, ptrs);
  1389. nilfs_btree_node_insert(btree, node, key, dreq->bpr_ptr, n);
  1390. if (!nilfs_bmap_dirty(btree))
  1391. nilfs_bmap_set_dirty(btree);
  1392. }
  1393. if (NILFS_BMAP_USE_VBN(btree))
  1394. nilfs_bmap_set_target_v(btree, key, dreq->bpr_ptr);
  1395. }
  1396. /**
  1397. * nilfs_btree_convert_and_insert -
  1398. * @bmap:
  1399. * @key:
  1400. * @ptr:
  1401. * @keys:
  1402. * @ptrs:
  1403. * @n:
  1404. */
  1405. int nilfs_btree_convert_and_insert(struct nilfs_bmap *btree,
  1406. __u64 key, __u64 ptr,
  1407. const __u64 *keys, const __u64 *ptrs, int n)
  1408. {
  1409. struct buffer_head *bh;
  1410. union nilfs_bmap_ptr_req dreq, nreq, *di, *ni;
  1411. struct nilfs_bmap_stats stats;
  1412. int ret;
  1413. if (n + 1 <= NILFS_BTREE_ROOT_NCHILDREN_MAX) {
  1414. di = &dreq;
  1415. ni = NULL;
  1416. } else if ((n + 1) <= NILFS_BTREE_NODE_NCHILDREN_MAX(
  1417. 1 << btree->b_inode->i_blkbits)) {
  1418. di = &dreq;
  1419. ni = &nreq;
  1420. } else {
  1421. di = NULL;
  1422. ni = NULL;
  1423. BUG();
  1424. }
  1425. ret = nilfs_btree_prepare_convert_and_insert(btree, key, di, ni, &bh,
  1426. &stats);
  1427. if (ret < 0)
  1428. return ret;
  1429. nilfs_btree_commit_convert_and_insert(btree, key, ptr, keys, ptrs, n,
  1430. di, ni, bh);
  1431. nilfs_bmap_add_blocks(btree, stats.bs_nblocks);
  1432. return 0;
  1433. }
  1434. static int nilfs_btree_propagate_p(struct nilfs_bmap *btree,
  1435. struct nilfs_btree_path *path,
  1436. int level,
  1437. struct buffer_head *bh)
  1438. {
  1439. while ((++level < nilfs_btree_height(btree) - 1) &&
  1440. !buffer_dirty(path[level].bp_bh))
  1441. nilfs_btnode_mark_dirty(path[level].bp_bh);
  1442. return 0;
  1443. }
  1444. static int nilfs_btree_prepare_update_v(struct nilfs_bmap *btree,
  1445. struct nilfs_btree_path *path,
  1446. int level, struct inode *dat)
  1447. {
  1448. struct nilfs_btree_node *parent;
  1449. int ret;
  1450. parent = nilfs_btree_get_node(btree, path, level + 1);
  1451. path[level].bp_oldreq.bpr_ptr =
  1452. nilfs_btree_node_get_ptr(btree, parent,
  1453. path[level + 1].bp_index);
  1454. path[level].bp_newreq.bpr_ptr = path[level].bp_oldreq.bpr_ptr + 1;
  1455. ret = nilfs_dat_prepare_update(dat, &path[level].bp_oldreq.bpr_req,
  1456. &path[level].bp_newreq.bpr_req);
  1457. if (ret < 0)
  1458. return ret;
  1459. if (buffer_nilfs_node(path[level].bp_bh)) {
  1460. path[level].bp_ctxt.oldkey = path[level].bp_oldreq.bpr_ptr;
  1461. path[level].bp_ctxt.newkey = path[level].bp_newreq.bpr_ptr;
  1462. path[level].bp_ctxt.bh = path[level].bp_bh;
  1463. ret = nilfs_btnode_prepare_change_key(
  1464. &NILFS_BMAP_I(btree)->i_btnode_cache,
  1465. &path[level].bp_ctxt);
  1466. if (ret < 0) {
  1467. nilfs_dat_abort_update(dat,
  1468. &path[level].bp_oldreq.bpr_req,
  1469. &path[level].bp_newreq.bpr_req);
  1470. return ret;
  1471. }
  1472. }
  1473. return 0;
  1474. }
  1475. static void nilfs_btree_commit_update_v(struct nilfs_bmap *btree,
  1476. struct nilfs_btree_path *path,
  1477. int level, struct inode *dat)
  1478. {
  1479. struct nilfs_btree_node *parent;
  1480. nilfs_dat_commit_update(dat, &path[level].bp_oldreq.bpr_req,
  1481. &path[level].bp_newreq.bpr_req,
  1482. btree->b_ptr_type == NILFS_BMAP_PTR_VS);
  1483. if (buffer_nilfs_node(path[level].bp_bh)) {
  1484. nilfs_btnode_commit_change_key(
  1485. &NILFS_BMAP_I(btree)->i_btnode_cache,
  1486. &path[level].bp_ctxt);
  1487. path[level].bp_bh = path[level].bp_ctxt.bh;
  1488. }
  1489. set_buffer_nilfs_volatile(path[level].bp_bh);
  1490. parent = nilfs_btree_get_node(btree, path, level + 1);
  1491. nilfs_btree_node_set_ptr(btree, parent, path[level + 1].bp_index,
  1492. path[level].bp_newreq.bpr_ptr);
  1493. }
  1494. static void nilfs_btree_abort_update_v(struct nilfs_bmap *btree,
  1495. struct nilfs_btree_path *path,
  1496. int level, struct inode *dat)
  1497. {
  1498. nilfs_dat_abort_update(dat, &path[level].bp_oldreq.bpr_req,
  1499. &path[level].bp_newreq.bpr_req);
  1500. if (buffer_nilfs_node(path[level].bp_bh))
  1501. nilfs_btnode_abort_change_key(
  1502. &NILFS_BMAP_I(btree)->i_btnode_cache,
  1503. &path[level].bp_ctxt);
  1504. }
  1505. static int nilfs_btree_prepare_propagate_v(struct nilfs_bmap *btree,
  1506. struct nilfs_btree_path *path,
  1507. int minlevel, int *maxlevelp,
  1508. struct inode *dat)
  1509. {
  1510. int level, ret;
  1511. level = minlevel;
  1512. if (!buffer_nilfs_volatile(path[level].bp_bh)) {
  1513. ret = nilfs_btree_prepare_update_v(btree, path, level, dat);
  1514. if (ret < 0)
  1515. return ret;
  1516. }
  1517. while ((++level < nilfs_btree_height(btree) - 1) &&
  1518. !buffer_dirty(path[level].bp_bh)) {
  1519. WARN_ON(buffer_nilfs_volatile(path[level].bp_bh));
  1520. ret = nilfs_btree_prepare_update_v(btree, path, level, dat);
  1521. if (ret < 0)
  1522. goto out;
  1523. }
  1524. /* success */
  1525. *maxlevelp = level - 1;
  1526. return 0;
  1527. /* error */
  1528. out:
  1529. while (--level > minlevel)
  1530. nilfs_btree_abort_update_v(btree, path, level, dat);
  1531. if (!buffer_nilfs_volatile(path[level].bp_bh))
  1532. nilfs_btree_abort_update_v(btree, path, level, dat);
  1533. return ret;
  1534. }
  1535. static void nilfs_btree_commit_propagate_v(struct nilfs_bmap *btree,
  1536. struct nilfs_btree_path *path,
  1537. int minlevel, int maxlevel,
  1538. struct buffer_head *bh,
  1539. struct inode *dat)
  1540. {
  1541. int level;
  1542. if (!buffer_nilfs_volatile(path[minlevel].bp_bh))
  1543. nilfs_btree_commit_update_v(btree, path, minlevel, dat);
  1544. for (level = minlevel + 1; level <= maxlevel; level++)
  1545. nilfs_btree_commit_update_v(btree, path, level, dat);
  1546. }
  1547. static int nilfs_btree_propagate_v(struct nilfs_bmap *btree,
  1548. struct nilfs_btree_path *path,
  1549. int level, struct buffer_head *bh)
  1550. {
  1551. int maxlevel = 0, ret;
  1552. struct nilfs_btree_node *parent;
  1553. struct inode *dat = nilfs_bmap_get_dat(btree);
  1554. __u64 ptr;
  1555. get_bh(bh);
  1556. path[level].bp_bh = bh;
  1557. ret = nilfs_btree_prepare_propagate_v(btree, path, level, &maxlevel,
  1558. dat);
  1559. if (ret < 0)
  1560. goto out;
  1561. if (buffer_nilfs_volatile(path[level].bp_bh)) {
  1562. parent = nilfs_btree_get_node(btree, path, level + 1);
  1563. ptr = nilfs_btree_node_get_ptr(btree, parent,
  1564. path[level + 1].bp_index);
  1565. ret = nilfs_dat_mark_dirty(dat, ptr);
  1566. if (ret < 0)
  1567. goto out;
  1568. }
  1569. nilfs_btree_commit_propagate_v(btree, path, level, maxlevel, bh, dat);
  1570. out:
  1571. brelse(path[level].bp_bh);
  1572. path[level].bp_bh = NULL;
  1573. return ret;
  1574. }
  1575. static int nilfs_btree_propagate(struct nilfs_bmap *btree,
  1576. struct buffer_head *bh)
  1577. {
  1578. struct nilfs_btree_path *path;
  1579. struct nilfs_btree_node *node;
  1580. __u64 key;
  1581. int level, ret;
  1582. WARN_ON(!buffer_dirty(bh));
  1583. path = nilfs_btree_alloc_path();
  1584. if (path == NULL)
  1585. return -ENOMEM;
  1586. if (buffer_nilfs_node(bh)) {
  1587. node = (struct nilfs_btree_node *)bh->b_data;
  1588. key = nilfs_btree_node_get_key(node, 0);
  1589. level = nilfs_btree_node_get_level(node);
  1590. } else {
  1591. key = nilfs_bmap_data_get_key(btree, bh);
  1592. level = NILFS_BTREE_LEVEL_DATA;
  1593. }
  1594. ret = nilfs_btree_do_lookup(btree, path, key, NULL, level + 1);
  1595. if (ret < 0) {
  1596. if (unlikely(ret == -ENOENT))
  1597. printk(KERN_CRIT "%s: key = %llu, level == %d\n",
  1598. __func__, (unsigned long long)key, level);
  1599. goto out;
  1600. }
  1601. ret = NILFS_BMAP_USE_VBN(btree) ?
  1602. nilfs_btree_propagate_v(btree, path, level, bh) :
  1603. nilfs_btree_propagate_p(btree, path, level, bh);
  1604. out:
  1605. nilfs_btree_free_path(path);
  1606. return ret;
  1607. }
  1608. static int nilfs_btree_propagate_gc(struct nilfs_bmap *btree,
  1609. struct buffer_head *bh)
  1610. {
  1611. return nilfs_dat_mark_dirty(nilfs_bmap_get_dat(btree), bh->b_blocknr);
  1612. }
  1613. static void nilfs_btree_add_dirty_buffer(struct nilfs_bmap *btree,
  1614. struct list_head *lists,
  1615. struct buffer_head *bh)
  1616. {
  1617. struct list_head *head;
  1618. struct buffer_head *cbh;
  1619. struct nilfs_btree_node *node, *cnode;
  1620. __u64 key, ckey;
  1621. int level;
  1622. get_bh(bh);
  1623. node = (struct nilfs_btree_node *)bh->b_data;
  1624. key = nilfs_btree_node_get_key(node, 0);
  1625. level = nilfs_btree_node_get_level(node);
  1626. if (level < NILFS_BTREE_LEVEL_NODE_MIN ||
  1627. level >= NILFS_BTREE_LEVEL_MAX) {
  1628. dump_stack();
  1629. printk(KERN_WARNING
  1630. "%s: invalid btree level: %d (key=%llu, ino=%lu, "
  1631. "blocknr=%llu)\n",
  1632. __func__, level, (unsigned long long)key,
  1633. NILFS_BMAP_I(btree)->vfs_inode.i_ino,
  1634. (unsigned long long)bh->b_blocknr);
  1635. return;
  1636. }
  1637. list_for_each(head, &lists[level]) {
  1638. cbh = list_entry(head, struct buffer_head, b_assoc_buffers);
  1639. cnode = (struct nilfs_btree_node *)cbh->b_data;
  1640. ckey = nilfs_btree_node_get_key(cnode, 0);
  1641. if (key < ckey)
  1642. break;
  1643. }
  1644. list_add_tail(&bh->b_assoc_buffers, head);
  1645. }
  1646. static void nilfs_btree_lookup_dirty_buffers(struct nilfs_bmap *btree,
  1647. struct list_head *listp)
  1648. {
  1649. struct address_space *btcache = &NILFS_BMAP_I(btree)->i_btnode_cache;
  1650. struct list_head lists[NILFS_BTREE_LEVEL_MAX];
  1651. struct pagevec pvec;
  1652. struct buffer_head *bh, *head;
  1653. pgoff_t index = 0;
  1654. int level, i;
  1655. for (level = NILFS_BTREE_LEVEL_NODE_MIN;
  1656. level < NILFS_BTREE_LEVEL_MAX;
  1657. level++)
  1658. INIT_LIST_HEAD(&lists[level]);
  1659. pagevec_init(&pvec, 0);
  1660. while (pagevec_lookup_tag(&pvec, btcache, &index, PAGECACHE_TAG_DIRTY,
  1661. PAGEVEC_SIZE)) {
  1662. for (i = 0; i < pagevec_count(&pvec); i++) {
  1663. bh = head = page_buffers(pvec.pages[i]);
  1664. do {
  1665. if (buffer_dirty(bh))
  1666. nilfs_btree_add_dirty_buffer(btree,
  1667. lists, bh);
  1668. } while ((bh = bh->b_this_page) != head);
  1669. }
  1670. pagevec_release(&pvec);
  1671. cond_resched();
  1672. }
  1673. for (level = NILFS_BTREE_LEVEL_NODE_MIN;
  1674. level < NILFS_BTREE_LEVEL_MAX;
  1675. level++)
  1676. list_splice_tail(&lists[level], listp);
  1677. }
  1678. static int nilfs_btree_assign_p(struct nilfs_bmap *btree,
  1679. struct nilfs_btree_path *path,
  1680. int level,
  1681. struct buffer_head **bh,
  1682. sector_t blocknr,
  1683. union nilfs_binfo *binfo)
  1684. {
  1685. struct nilfs_btree_node *parent;
  1686. __u64 key;
  1687. __u64 ptr;
  1688. int ret;
  1689. parent = nilfs_btree_get_node(btree, path, level + 1);
  1690. ptr = nilfs_btree_node_get_ptr(btree, parent,
  1691. path[level + 1].bp_index);
  1692. if (buffer_nilfs_node(*bh)) {
  1693. path[level].bp_ctxt.oldkey = ptr;
  1694. path[level].bp_ctxt.newkey = blocknr;
  1695. path[level].bp_ctxt.bh = *bh;
  1696. ret = nilfs_btnode_prepare_change_key(
  1697. &NILFS_BMAP_I(btree)->i_btnode_cache,
  1698. &path[level].bp_ctxt);
  1699. if (ret < 0)
  1700. return ret;
  1701. nilfs_btnode_commit_change_key(
  1702. &NILFS_BMAP_I(btree)->i_btnode_cache,
  1703. &path[level].bp_ctxt);
  1704. *bh = path[level].bp_ctxt.bh;
  1705. }
  1706. nilfs_btree_node_set_ptr(btree, parent,
  1707. path[level + 1].bp_index, blocknr);
  1708. key = nilfs_btree_node_get_key(parent, path[level + 1].bp_index);
  1709. /* on-disk format */
  1710. binfo->bi_dat.bi_blkoff = cpu_to_le64(key);
  1711. binfo->bi_dat.bi_level = level;
  1712. return 0;
  1713. }
  1714. static int nilfs_btree_assign_v(struct nilfs_bmap *btree,
  1715. struct nilfs_btree_path *path,
  1716. int level,
  1717. struct buffer_head **bh,
  1718. sector_t blocknr,
  1719. union nilfs_binfo *binfo)
  1720. {
  1721. struct nilfs_btree_node *parent;
  1722. struct inode *dat = nilfs_bmap_get_dat(btree);
  1723. __u64 key;
  1724. __u64 ptr;
  1725. union nilfs_bmap_ptr_req req;
  1726. int ret;
  1727. parent = nilfs_btree_get_node(btree, path, level + 1);
  1728. ptr = nilfs_btree_node_get_ptr(btree, parent, path[level + 1].bp_index);
  1729. req.bpr_ptr = ptr;
  1730. ret = nilfs_dat_prepare_start(dat, &req.bpr_req);
  1731. if (ret < 0)
  1732. return ret;
  1733. nilfs_dat_commit_start(dat, &req.bpr_req, blocknr);
  1734. key = nilfs_btree_node_get_key(parent, path[level + 1].bp_index);
  1735. /* on-disk format */
  1736. binfo->bi_v.bi_vblocknr = cpu_to_le64(ptr);
  1737. binfo->bi_v.bi_blkoff = cpu_to_le64(key);
  1738. return 0;
  1739. }
  1740. static int nilfs_btree_assign(struct nilfs_bmap *btree,
  1741. struct buffer_head **bh,
  1742. sector_t blocknr,
  1743. union nilfs_binfo *binfo)
  1744. {
  1745. struct nilfs_btree_path *path;
  1746. struct nilfs_btree_node *node;
  1747. __u64 key;
  1748. int level, ret;
  1749. path = nilfs_btree_alloc_path();
  1750. if (path == NULL)
  1751. return -ENOMEM;
  1752. if (buffer_nilfs_node(*bh)) {
  1753. node = (struct nilfs_btree_node *)(*bh)->b_data;
  1754. key = nilfs_btree_node_get_key(node, 0);
  1755. level = nilfs_btree_node_get_level(node);
  1756. } else {
  1757. key = nilfs_bmap_data_get_key(btree, *bh);
  1758. level = NILFS_BTREE_LEVEL_DATA;
  1759. }
  1760. ret = nilfs_btree_do_lookup(btree, path, key, NULL, level + 1);
  1761. if (ret < 0) {
  1762. WARN_ON(ret == -ENOENT);
  1763. goto out;
  1764. }
  1765. ret = NILFS_BMAP_USE_VBN(btree) ?
  1766. nilfs_btree_assign_v(btree, path, level, bh, blocknr, binfo) :
  1767. nilfs_btree_assign_p(btree, path, level, bh, blocknr, binfo);
  1768. out:
  1769. nilfs_btree_free_path(path);
  1770. return ret;
  1771. }
  1772. static int nilfs_btree_assign_gc(struct nilfs_bmap *btree,
  1773. struct buffer_head **bh,
  1774. sector_t blocknr,
  1775. union nilfs_binfo *binfo)
  1776. {
  1777. struct nilfs_btree_node *node;
  1778. __u64 key;
  1779. int ret;
  1780. ret = nilfs_dat_move(nilfs_bmap_get_dat(btree), (*bh)->b_blocknr,
  1781. blocknr);
  1782. if (ret < 0)
  1783. return ret;
  1784. if (buffer_nilfs_node(*bh)) {
  1785. node = (struct nilfs_btree_node *)(*bh)->b_data;
  1786. key = nilfs_btree_node_get_key(node, 0);
  1787. } else
  1788. key = nilfs_bmap_data_get_key(btree, *bh);
  1789. /* on-disk format */
  1790. binfo->bi_v.bi_vblocknr = cpu_to_le64((*bh)->b_blocknr);
  1791. binfo->bi_v.bi_blkoff = cpu_to_le64(key);
  1792. return 0;
  1793. }
  1794. static int nilfs_btree_mark(struct nilfs_bmap *btree, __u64 key, int level)
  1795. {
  1796. struct buffer_head *bh;
  1797. struct nilfs_btree_path *path;
  1798. __u64 ptr;
  1799. int ret;
  1800. path = nilfs_btree_alloc_path();
  1801. if (path == NULL)
  1802. return -ENOMEM;
  1803. ret = nilfs_btree_do_lookup(btree, path, key, &ptr, level + 1);
  1804. if (ret < 0) {
  1805. WARN_ON(ret == -ENOENT);
  1806. goto out;
  1807. }
  1808. ret = nilfs_btree_get_block(btree, ptr, &bh);
  1809. if (ret < 0) {
  1810. WARN_ON(ret == -ENOENT);
  1811. goto out;
  1812. }
  1813. if (!buffer_dirty(bh))
  1814. nilfs_btnode_mark_dirty(bh);
  1815. brelse(bh);
  1816. if (!nilfs_bmap_dirty(btree))
  1817. nilfs_bmap_set_dirty(btree);
  1818. out:
  1819. nilfs_btree_free_path(path);
  1820. return ret;
  1821. }
  1822. static const struct nilfs_bmap_operations nilfs_btree_ops = {
  1823. .bop_lookup = nilfs_btree_lookup,
  1824. .bop_lookup_contig = nilfs_btree_lookup_contig,
  1825. .bop_insert = nilfs_btree_insert,
  1826. .bop_delete = nilfs_btree_delete,
  1827. .bop_clear = NULL,
  1828. .bop_propagate = nilfs_btree_propagate,
  1829. .bop_lookup_dirty_buffers = nilfs_btree_lookup_dirty_buffers,
  1830. .bop_assign = nilfs_btree_assign,
  1831. .bop_mark = nilfs_btree_mark,
  1832. .bop_last_key = nilfs_btree_last_key,
  1833. .bop_check_insert = NULL,
  1834. .bop_check_delete = nilfs_btree_check_delete,
  1835. .bop_gather_data = nilfs_btree_gather_data,
  1836. };
  1837. static const struct nilfs_bmap_operations nilfs_btree_ops_gc = {
  1838. .bop_lookup = NULL,
  1839. .bop_lookup_contig = NULL,
  1840. .bop_insert = NULL,
  1841. .bop_delete = NULL,
  1842. .bop_clear = NULL,
  1843. .bop_propagate = nilfs_btree_propagate_gc,
  1844. .bop_lookup_dirty_buffers = nilfs_btree_lookup_dirty_buffers,
  1845. .bop_assign = nilfs_btree_assign_gc,
  1846. .bop_mark = NULL,
  1847. .bop_last_key = NULL,
  1848. .bop_check_insert = NULL,
  1849. .bop_check_delete = NULL,
  1850. .bop_gather_data = NULL,
  1851. };
  1852. int nilfs_btree_init(struct nilfs_bmap *bmap)
  1853. {
  1854. bmap->b_ops = &nilfs_btree_ops;
  1855. return 0;
  1856. }
  1857. void nilfs_btree_init_gc(struct nilfs_bmap *bmap)
  1858. {
  1859. bmap->b_ops = &nilfs_btree_ops_gc;
  1860. }