ext4_write.c 27 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010
  1. /*
  2. * (C) Copyright 2011 - 2012 Samsung Electronics
  3. * EXT4 filesystem implementation in Uboot by
  4. * Uma Shankar <uma.shankar@samsung.com>
  5. * Manjunatha C Achar <a.manjunatha@samsung.com>
  6. *
  7. * ext4ls and ext4load : Based on ext2 ls and load support in Uboot.
  8. * Ext4 read optimization taken from Open-Moko
  9. * Qi bootloader
  10. *
  11. * (C) Copyright 2004
  12. * esd gmbh <www.esd-electronics.com>
  13. * Reinhard Arlt <reinhard.arlt@esd-electronics.com>
  14. *
  15. * based on code from grub2 fs/ext2.c and fs/fshelp.c by
  16. * GRUB -- GRand Unified Bootloader
  17. * Copyright (C) 2003, 2004 Free Software Foundation, Inc.
  18. *
  19. * ext4write : Based on generic ext4 protocol.
  20. *
  21. * SPDX-License-Identifier: GPL-2.0+
  22. */
  23. #include <common.h>
  24. #include <memalign.h>
  25. #include <linux/stat.h>
  26. #include <div64.h>
  27. #include "ext4_common.h"
  28. static inline void ext4fs_sb_free_inodes_inc(struct ext2_sblock *sb)
  29. {
  30. sb->free_inodes = cpu_to_le32(le32_to_cpu(sb->free_inodes) + 1);
  31. }
  32. static inline void ext4fs_sb_free_blocks_inc(struct ext2_sblock *sb)
  33. {
  34. sb->free_blocks = cpu_to_le32(le32_to_cpu(sb->free_blocks) + 1);
  35. }
  36. static inline void ext4fs_bg_free_inodes_inc(struct ext2_block_group *bg)
  37. {
  38. bg->free_inodes = cpu_to_le16(le16_to_cpu(bg->free_inodes) + 1);
  39. }
  40. static inline void ext4fs_bg_free_blocks_inc(struct ext2_block_group *bg)
  41. {
  42. bg->free_blocks = cpu_to_le16(le16_to_cpu(bg->free_blocks) + 1);
  43. }
  44. static void ext4fs_update(void)
  45. {
  46. short i;
  47. ext4fs_update_journal();
  48. struct ext_filesystem *fs = get_fs();
  49. struct ext2_block_group *bgd = NULL;
  50. /* update super block */
  51. put_ext4((uint64_t)(SUPERBLOCK_SIZE),
  52. (struct ext2_sblock *)fs->sb, (uint32_t)SUPERBLOCK_SIZE);
  53. /* update block bitmaps */
  54. for (i = 0; i < fs->no_blkgrp; i++) {
  55. bgd = ext4fs_get_group_descriptor(fs, i);
  56. bgd->bg_checksum = cpu_to_le16(ext4fs_checksum_update(i));
  57. uint64_t b_bitmap_blk = ext4fs_bg_get_block_id(bgd, fs);
  58. put_ext4(b_bitmap_blk * fs->blksz,
  59. fs->blk_bmaps[i], fs->blksz);
  60. }
  61. /* update inode bitmaps */
  62. for (i = 0; i < fs->no_blkgrp; i++) {
  63. bgd = ext4fs_get_group_descriptor(fs, i);
  64. uint64_t i_bitmap_blk = ext4fs_bg_get_inode_id(bgd, fs);
  65. put_ext4(i_bitmap_blk * fs->blksz,
  66. fs->inode_bmaps[i], fs->blksz);
  67. }
  68. /* update the block group descriptor table */
  69. put_ext4((uint64_t)((uint64_t)fs->gdtable_blkno * (uint64_t)fs->blksz),
  70. (struct ext2_block_group *)fs->gdtable,
  71. (fs->blksz * fs->no_blk_pergdt));
  72. ext4fs_dump_metadata();
  73. gindex = 0;
  74. gd_index = 0;
  75. }
  76. int ext4fs_get_bgdtable(void)
  77. {
  78. int status;
  79. struct ext_filesystem *fs = get_fs();
  80. int gdsize_total = ROUND(fs->no_blkgrp * fs->gdsize, fs->blksz);
  81. fs->no_blk_pergdt = gdsize_total / fs->blksz;
  82. /* allocate memory for gdtable */
  83. fs->gdtable = zalloc(gdsize_total);
  84. if (!fs->gdtable)
  85. return -ENOMEM;
  86. /* read the group descriptor table */
  87. status = ext4fs_devread((lbaint_t)fs->gdtable_blkno * fs->sect_perblk,
  88. 0, fs->blksz * fs->no_blk_pergdt, fs->gdtable);
  89. if (status == 0)
  90. goto fail;
  91. if (ext4fs_log_gdt(fs->gdtable)) {
  92. printf("Error in ext4fs_log_gdt\n");
  93. return -1;
  94. }
  95. return 0;
  96. fail:
  97. free(fs->gdtable);
  98. fs->gdtable = NULL;
  99. return -1;
  100. }
  101. static void delete_single_indirect_block(struct ext2_inode *inode)
  102. {
  103. struct ext2_block_group *bgd = NULL;
  104. static int prev_bg_bmap_idx = -1;
  105. uint32_t blknr;
  106. int remainder;
  107. int bg_idx;
  108. int status;
  109. uint32_t blk_per_grp = le32_to_cpu(ext4fs_root->sblock.blocks_per_group);
  110. struct ext_filesystem *fs = get_fs();
  111. char *journal_buffer = zalloc(fs->blksz);
  112. if (!journal_buffer) {
  113. printf("No memory\n");
  114. return;
  115. }
  116. /* deleting the single indirect block associated with inode */
  117. if (inode->b.blocks.indir_block != 0) {
  118. blknr = le32_to_cpu(inode->b.blocks.indir_block);
  119. debug("SIPB releasing %u\n", blknr);
  120. bg_idx = blknr / blk_per_grp;
  121. if (fs->blksz == 1024) {
  122. remainder = blknr % blk_per_grp;
  123. if (!remainder)
  124. bg_idx--;
  125. }
  126. ext4fs_reset_block_bmap(blknr, fs->blk_bmaps[bg_idx], bg_idx);
  127. /* get block group descriptor table */
  128. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  129. ext4fs_bg_free_blocks_inc(bgd);
  130. ext4fs_sb_free_blocks_inc(fs->sb);
  131. /* journal backup */
  132. if (prev_bg_bmap_idx != bg_idx) {
  133. uint64_t b_bitmap_blk = ext4fs_bg_get_block_id(bgd, fs);
  134. status = ext4fs_devread(
  135. b_bitmap_blk * fs->sect_perblk,
  136. 0, fs->blksz, journal_buffer);
  137. if (status == 0)
  138. goto fail;
  139. if (ext4fs_log_journal(journal_buffer, b_bitmap_blk))
  140. goto fail;
  141. prev_bg_bmap_idx = bg_idx;
  142. }
  143. }
  144. fail:
  145. free(journal_buffer);
  146. }
  147. static void delete_double_indirect_block(struct ext2_inode *inode)
  148. {
  149. int i;
  150. short status;
  151. static int prev_bg_bmap_idx = -1;
  152. uint32_t blknr;
  153. int remainder;
  154. int bg_idx;
  155. uint32_t blk_per_grp = le32_to_cpu(ext4fs_root->sblock.blocks_per_group);
  156. __le32 *di_buffer = NULL;
  157. void *dib_start_addr = NULL;
  158. struct ext2_block_group *bgd = NULL;
  159. struct ext_filesystem *fs = get_fs();
  160. char *journal_buffer = zalloc(fs->blksz);
  161. if (!journal_buffer) {
  162. printf("No memory\n");
  163. return;
  164. }
  165. if (inode->b.blocks.double_indir_block != 0) {
  166. di_buffer = zalloc(fs->blksz);
  167. if (!di_buffer) {
  168. printf("No memory\n");
  169. return;
  170. }
  171. dib_start_addr = di_buffer;
  172. blknr = le32_to_cpu(inode->b.blocks.double_indir_block);
  173. status = ext4fs_devread((lbaint_t)blknr * fs->sect_perblk, 0,
  174. fs->blksz, (char *)di_buffer);
  175. for (i = 0; i < fs->blksz / sizeof(int); i++) {
  176. if (*di_buffer == 0)
  177. break;
  178. debug("DICB releasing %u\n", *di_buffer);
  179. bg_idx = le32_to_cpu(*di_buffer) / blk_per_grp;
  180. if (fs->blksz == 1024) {
  181. remainder = le32_to_cpu(*di_buffer) % blk_per_grp;
  182. if (!remainder)
  183. bg_idx--;
  184. }
  185. /* get block group descriptor table */
  186. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  187. ext4fs_reset_block_bmap(le32_to_cpu(*di_buffer),
  188. fs->blk_bmaps[bg_idx], bg_idx);
  189. di_buffer++;
  190. ext4fs_bg_free_blocks_inc(bgd);
  191. ext4fs_sb_free_blocks_inc(fs->sb);
  192. /* journal backup */
  193. if (prev_bg_bmap_idx != bg_idx) {
  194. uint64_t b_bitmap_blk =
  195. ext4fs_bg_get_block_id(bgd, fs);
  196. status = ext4fs_devread(b_bitmap_blk
  197. * fs->sect_perblk, 0,
  198. fs->blksz,
  199. journal_buffer);
  200. if (status == 0)
  201. goto fail;
  202. if (ext4fs_log_journal(journal_buffer,
  203. b_bitmap_blk))
  204. goto fail;
  205. prev_bg_bmap_idx = bg_idx;
  206. }
  207. }
  208. /* removing the parent double indirect block */
  209. blknr = le32_to_cpu(inode->b.blocks.double_indir_block);
  210. bg_idx = blknr / blk_per_grp;
  211. if (fs->blksz == 1024) {
  212. remainder = blknr % blk_per_grp;
  213. if (!remainder)
  214. bg_idx--;
  215. }
  216. /* get block group descriptor table */
  217. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  218. ext4fs_reset_block_bmap(blknr, fs->blk_bmaps[bg_idx], bg_idx);
  219. ext4fs_bg_free_blocks_inc(bgd);
  220. ext4fs_sb_free_blocks_inc(fs->sb);
  221. /* journal backup */
  222. if (prev_bg_bmap_idx != bg_idx) {
  223. uint64_t b_bitmap_blk = ext4fs_bg_get_block_id(bgd, fs);
  224. status = ext4fs_devread(b_bitmap_blk * fs->sect_perblk,
  225. 0, fs->blksz, journal_buffer);
  226. if (status == 0)
  227. goto fail;
  228. if (ext4fs_log_journal(journal_buffer, b_bitmap_blk))
  229. goto fail;
  230. prev_bg_bmap_idx = bg_idx;
  231. }
  232. debug("DIPB releasing %d\n", blknr);
  233. }
  234. fail:
  235. free(dib_start_addr);
  236. free(journal_buffer);
  237. }
  238. static void delete_triple_indirect_block(struct ext2_inode *inode)
  239. {
  240. int i, j;
  241. short status;
  242. static int prev_bg_bmap_idx = -1;
  243. uint32_t blknr;
  244. int remainder;
  245. int bg_idx;
  246. uint32_t blk_per_grp = le32_to_cpu(ext4fs_root->sblock.blocks_per_group);
  247. __le32 *tigp_buffer = NULL;
  248. void *tib_start_addr = NULL;
  249. __le32 *tip_buffer = NULL;
  250. void *tipb_start_addr = NULL;
  251. struct ext2_block_group *bgd = NULL;
  252. struct ext_filesystem *fs = get_fs();
  253. char *journal_buffer = zalloc(fs->blksz);
  254. if (!journal_buffer) {
  255. printf("No memory\n");
  256. return;
  257. }
  258. if (inode->b.blocks.triple_indir_block != 0) {
  259. tigp_buffer = zalloc(fs->blksz);
  260. if (!tigp_buffer) {
  261. printf("No memory\n");
  262. return;
  263. }
  264. tib_start_addr = tigp_buffer;
  265. blknr = le32_to_cpu(inode->b.blocks.triple_indir_block);
  266. status = ext4fs_devread((lbaint_t)blknr * fs->sect_perblk, 0,
  267. fs->blksz, (char *)tigp_buffer);
  268. for (i = 0; i < fs->blksz / sizeof(int); i++) {
  269. if (*tigp_buffer == 0)
  270. break;
  271. debug("tigp buffer releasing %u\n", *tigp_buffer);
  272. tip_buffer = zalloc(fs->blksz);
  273. if (!tip_buffer)
  274. goto fail;
  275. tipb_start_addr = tip_buffer;
  276. status = ext4fs_devread((lbaint_t)le32_to_cpu(*tigp_buffer) *
  277. fs->sect_perblk, 0, fs->blksz,
  278. (char *)tip_buffer);
  279. for (j = 0; j < fs->blksz / sizeof(int); j++) {
  280. if (le32_to_cpu(*tip_buffer) == 0)
  281. break;
  282. bg_idx = le32_to_cpu(*tip_buffer) / blk_per_grp;
  283. if (fs->blksz == 1024) {
  284. remainder = le32_to_cpu(*tip_buffer) % blk_per_grp;
  285. if (!remainder)
  286. bg_idx--;
  287. }
  288. ext4fs_reset_block_bmap(le32_to_cpu(*tip_buffer),
  289. fs->blk_bmaps[bg_idx],
  290. bg_idx);
  291. tip_buffer++;
  292. /* get block group descriptor table */
  293. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  294. ext4fs_bg_free_blocks_inc(bgd);
  295. ext4fs_sb_free_blocks_inc(fs->sb);
  296. /* journal backup */
  297. if (prev_bg_bmap_idx != bg_idx) {
  298. uint64_t b_bitmap_blk =
  299. ext4fs_bg_get_block_id(bgd, fs);
  300. status =
  301. ext4fs_devread(
  302. b_bitmap_blk *
  303. fs->sect_perblk, 0,
  304. fs->blksz,
  305. journal_buffer);
  306. if (status == 0)
  307. goto fail;
  308. if (ext4fs_log_journal(journal_buffer,
  309. b_bitmap_blk))
  310. goto fail;
  311. prev_bg_bmap_idx = bg_idx;
  312. }
  313. }
  314. free(tipb_start_addr);
  315. tipb_start_addr = NULL;
  316. /*
  317. * removing the grand parent blocks
  318. * which is connected to inode
  319. */
  320. bg_idx = le32_to_cpu(*tigp_buffer) / blk_per_grp;
  321. if (fs->blksz == 1024) {
  322. remainder = le32_to_cpu(*tigp_buffer) % blk_per_grp;
  323. if (!remainder)
  324. bg_idx--;
  325. }
  326. ext4fs_reset_block_bmap(le32_to_cpu(*tigp_buffer),
  327. fs->blk_bmaps[bg_idx], bg_idx);
  328. tigp_buffer++;
  329. /* get block group descriptor table */
  330. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  331. ext4fs_bg_free_blocks_inc(bgd);
  332. ext4fs_sb_free_blocks_inc(fs->sb);
  333. /* journal backup */
  334. if (prev_bg_bmap_idx != bg_idx) {
  335. uint64_t b_bitmap_blk =
  336. ext4fs_bg_get_block_id(bgd, fs);
  337. memset(journal_buffer, '\0', fs->blksz);
  338. status = ext4fs_devread(b_bitmap_blk *
  339. fs->sect_perblk, 0,
  340. fs->blksz,
  341. journal_buffer);
  342. if (status == 0)
  343. goto fail;
  344. if (ext4fs_log_journal(journal_buffer,
  345. b_bitmap_blk))
  346. goto fail;
  347. prev_bg_bmap_idx = bg_idx;
  348. }
  349. }
  350. /* removing the grand parent triple indirect block */
  351. blknr = le32_to_cpu(inode->b.blocks.triple_indir_block);
  352. bg_idx = blknr / blk_per_grp;
  353. if (fs->blksz == 1024) {
  354. remainder = blknr % blk_per_grp;
  355. if (!remainder)
  356. bg_idx--;
  357. }
  358. ext4fs_reset_block_bmap(blknr, fs->blk_bmaps[bg_idx], bg_idx);
  359. /* get block group descriptor table */
  360. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  361. ext4fs_bg_free_blocks_inc(bgd);
  362. ext4fs_sb_free_blocks_inc(fs->sb);
  363. /* journal backup */
  364. if (prev_bg_bmap_idx != bg_idx) {
  365. uint64_t b_bitmap_blk = ext4fs_bg_get_block_id(bgd, fs);
  366. status = ext4fs_devread(b_bitmap_blk * fs->sect_perblk,
  367. 0, fs->blksz, journal_buffer);
  368. if (status == 0)
  369. goto fail;
  370. if (ext4fs_log_journal(journal_buffer, b_bitmap_blk))
  371. goto fail;
  372. prev_bg_bmap_idx = bg_idx;
  373. }
  374. debug("tigp buffer itself releasing %d\n", blknr);
  375. }
  376. fail:
  377. free(tib_start_addr);
  378. free(tipb_start_addr);
  379. free(journal_buffer);
  380. }
  381. static int ext4fs_delete_file(int inodeno)
  382. {
  383. struct ext2_inode inode;
  384. short status;
  385. int i;
  386. int remainder;
  387. long int blknr;
  388. int bg_idx;
  389. int ibmap_idx;
  390. char *read_buffer = NULL;
  391. char *start_block_address = NULL;
  392. uint32_t no_blocks;
  393. static int prev_bg_bmap_idx = -1;
  394. unsigned int inodes_per_block;
  395. uint32_t blkno;
  396. unsigned int blkoff;
  397. uint32_t blk_per_grp = le32_to_cpu(ext4fs_root->sblock.blocks_per_group);
  398. uint32_t inode_per_grp = le32_to_cpu(ext4fs_root->sblock.inodes_per_group);
  399. struct ext2_inode *inode_buffer = NULL;
  400. struct ext2_block_group *bgd = NULL;
  401. struct ext_filesystem *fs = get_fs();
  402. char *journal_buffer = zalloc(fs->blksz);
  403. if (!journal_buffer)
  404. return -ENOMEM;
  405. status = ext4fs_read_inode(ext4fs_root, inodeno, &inode);
  406. if (status == 0)
  407. goto fail;
  408. /* read the block no allocated to a file */
  409. no_blocks = le32_to_cpu(inode.size) / fs->blksz;
  410. if (le32_to_cpu(inode.size) % fs->blksz)
  411. no_blocks++;
  412. if (le32_to_cpu(inode.flags) & EXT4_EXTENTS_FL) {
  413. /* FIXME delete extent index blocks, i.e. eh_depth >= 1 */
  414. struct ext4_extent_header *eh =
  415. (struct ext4_extent_header *)
  416. inode.b.blocks.dir_blocks;
  417. debug("del: dep=%d entries=%d\n", eh->eh_depth, eh->eh_entries);
  418. } else {
  419. delete_single_indirect_block(&inode);
  420. delete_double_indirect_block(&inode);
  421. delete_triple_indirect_block(&inode);
  422. }
  423. /* release data blocks */
  424. for (i = 0; i < no_blocks; i++) {
  425. blknr = read_allocated_block(&inode, i);
  426. if (blknr == 0)
  427. continue;
  428. if (blknr < 0)
  429. goto fail;
  430. bg_idx = blknr / blk_per_grp;
  431. if (fs->blksz == 1024) {
  432. remainder = blknr % blk_per_grp;
  433. if (!remainder)
  434. bg_idx--;
  435. }
  436. ext4fs_reset_block_bmap(blknr, fs->blk_bmaps[bg_idx],
  437. bg_idx);
  438. debug("EXT4 Block releasing %ld: %d\n", blknr, bg_idx);
  439. /* get block group descriptor table */
  440. bgd = ext4fs_get_group_descriptor(fs, bg_idx);
  441. ext4fs_bg_free_blocks_inc(bgd);
  442. ext4fs_sb_free_blocks_inc(fs->sb);
  443. /* journal backup */
  444. if (prev_bg_bmap_idx != bg_idx) {
  445. uint64_t b_bitmap_blk = ext4fs_bg_get_block_id(bgd, fs);
  446. status = ext4fs_devread(b_bitmap_blk * fs->sect_perblk,
  447. 0, fs->blksz,
  448. journal_buffer);
  449. if (status == 0)
  450. goto fail;
  451. if (ext4fs_log_journal(journal_buffer, b_bitmap_blk))
  452. goto fail;
  453. prev_bg_bmap_idx = bg_idx;
  454. }
  455. }
  456. /* release inode */
  457. /* from the inode no to blockno */
  458. inodes_per_block = fs->blksz / fs->inodesz;
  459. ibmap_idx = inodeno / inode_per_grp;
  460. /* get the block no */
  461. inodeno--;
  462. /* get block group descriptor table */
  463. bgd = ext4fs_get_group_descriptor(fs, ibmap_idx);
  464. blkno = ext4fs_bg_get_inode_table_id(bgd, fs) +
  465. (inodeno % inode_per_grp) / inodes_per_block;
  466. /* get the offset of the inode */
  467. blkoff = ((inodeno) % inodes_per_block) * fs->inodesz;
  468. /* read the block no containing the inode */
  469. read_buffer = zalloc(fs->blksz);
  470. if (!read_buffer)
  471. goto fail;
  472. start_block_address = read_buffer;
  473. status = ext4fs_devread((lbaint_t)blkno * fs->sect_perblk,
  474. 0, fs->blksz, read_buffer);
  475. if (status == 0)
  476. goto fail;
  477. if (ext4fs_log_journal(read_buffer, blkno))
  478. goto fail;
  479. read_buffer = read_buffer + blkoff;
  480. inode_buffer = (struct ext2_inode *)read_buffer;
  481. memset(inode_buffer, '\0', fs->inodesz);
  482. /* write the inode to original position in inode table */
  483. if (ext4fs_put_metadata(start_block_address, blkno))
  484. goto fail;
  485. /* update the respective inode bitmaps */
  486. inodeno++;
  487. ext4fs_reset_inode_bmap(inodeno, fs->inode_bmaps[ibmap_idx], ibmap_idx);
  488. ext4fs_bg_free_inodes_inc(bgd);
  489. ext4fs_sb_free_inodes_inc(fs->sb);
  490. /* journal backup */
  491. memset(journal_buffer, '\0', fs->blksz);
  492. status = ext4fs_devread(ext4fs_bg_get_inode_id(bgd, fs) *
  493. fs->sect_perblk, 0, fs->blksz, journal_buffer);
  494. if (status == 0)
  495. goto fail;
  496. if (ext4fs_log_journal(journal_buffer, ext4fs_bg_get_inode_id(bgd, fs)))
  497. goto fail;
  498. ext4fs_update();
  499. ext4fs_deinit();
  500. ext4fs_reinit_global();
  501. if (ext4fs_init() != 0) {
  502. printf("error in File System init\n");
  503. goto fail;
  504. }
  505. free(start_block_address);
  506. free(journal_buffer);
  507. return 0;
  508. fail:
  509. free(start_block_address);
  510. free(journal_buffer);
  511. return -1;
  512. }
  513. int ext4fs_init(void)
  514. {
  515. short status;
  516. int i;
  517. uint32_t real_free_blocks = 0;
  518. struct ext_filesystem *fs = get_fs();
  519. /* populate fs */
  520. fs->blksz = EXT2_BLOCK_SIZE(ext4fs_root);
  521. fs->sect_perblk = fs->blksz >> fs->dev_desc->log2blksz;
  522. /* get the superblock */
  523. fs->sb = zalloc(SUPERBLOCK_SIZE);
  524. if (!fs->sb)
  525. return -ENOMEM;
  526. if (!ext4_read_superblock((char *)fs->sb))
  527. goto fail;
  528. /* init journal */
  529. if (ext4fs_init_journal())
  530. goto fail;
  531. /* get total no of blockgroups */
  532. fs->no_blkgrp = (uint32_t)ext4fs_div_roundup(
  533. le32_to_cpu(ext4fs_root->sblock.total_blocks)
  534. - le32_to_cpu(ext4fs_root->sblock.first_data_block),
  535. le32_to_cpu(ext4fs_root->sblock.blocks_per_group));
  536. /* get the block group descriptor table */
  537. fs->gdtable_blkno = ((EXT2_MIN_BLOCK_SIZE == fs->blksz) + 1);
  538. if (ext4fs_get_bgdtable() == -1) {
  539. printf("Error in getting the block group descriptor table\n");
  540. goto fail;
  541. }
  542. /* load all the available bitmap block of the partition */
  543. fs->blk_bmaps = zalloc(fs->no_blkgrp * sizeof(char *));
  544. if (!fs->blk_bmaps)
  545. goto fail;
  546. for (i = 0; i < fs->no_blkgrp; i++) {
  547. fs->blk_bmaps[i] = zalloc(fs->blksz);
  548. if (!fs->blk_bmaps[i])
  549. goto fail;
  550. }
  551. for (i = 0; i < fs->no_blkgrp; i++) {
  552. struct ext2_block_group *bgd =
  553. ext4fs_get_group_descriptor(fs, i);
  554. status = ext4fs_devread(ext4fs_bg_get_block_id(bgd, fs) *
  555. fs->sect_perblk, 0,
  556. fs->blksz, (char *)fs->blk_bmaps[i]);
  557. if (status == 0)
  558. goto fail;
  559. }
  560. /* load all the available inode bitmap of the partition */
  561. fs->inode_bmaps = zalloc(fs->no_blkgrp * sizeof(unsigned char *));
  562. if (!fs->inode_bmaps)
  563. goto fail;
  564. for (i = 0; i < fs->no_blkgrp; i++) {
  565. fs->inode_bmaps[i] = zalloc(fs->blksz);
  566. if (!fs->inode_bmaps[i])
  567. goto fail;
  568. }
  569. for (i = 0; i < fs->no_blkgrp; i++) {
  570. struct ext2_block_group *bgd =
  571. ext4fs_get_group_descriptor(fs, i);
  572. status = ext4fs_devread(ext4fs_bg_get_inode_id(bgd, fs) *
  573. fs->sect_perblk,
  574. 0, fs->blksz,
  575. (char *)fs->inode_bmaps[i]);
  576. if (status == 0)
  577. goto fail;
  578. }
  579. /*
  580. * check filesystem consistency with free blocks of file system
  581. * some time we observed that superblock freeblocks does not match
  582. * with the blockgroups freeblocks when improper
  583. * reboot of a linux kernel
  584. */
  585. for (i = 0; i < fs->no_blkgrp; i++) {
  586. struct ext2_block_group *bgd =
  587. ext4fs_get_group_descriptor(fs, i);
  588. real_free_blocks = real_free_blocks +
  589. ext4fs_bg_get_free_blocks(bgd, fs);
  590. }
  591. if (real_free_blocks != ext4fs_sb_get_free_blocks(fs->sb))
  592. ext4fs_sb_set_free_blocks(fs->sb, real_free_blocks);
  593. return 0;
  594. fail:
  595. ext4fs_deinit();
  596. return -1;
  597. }
  598. void ext4fs_deinit(void)
  599. {
  600. int i;
  601. struct ext2_inode inode_journal;
  602. struct journal_superblock_t *jsb;
  603. uint32_t blknr;
  604. struct ext_filesystem *fs = get_fs();
  605. uint32_t new_feature_incompat;
  606. /* free journal */
  607. char *temp_buff = zalloc(fs->blksz);
  608. if (temp_buff) {
  609. ext4fs_read_inode(ext4fs_root, EXT2_JOURNAL_INO,
  610. &inode_journal);
  611. blknr = read_allocated_block(&inode_journal,
  612. EXT2_JOURNAL_SUPERBLOCK);
  613. ext4fs_devread((lbaint_t)blknr * fs->sect_perblk, 0, fs->blksz,
  614. temp_buff);
  615. jsb = (struct journal_superblock_t *)temp_buff;
  616. jsb->s_start = 0;
  617. put_ext4((uint64_t) ((uint64_t)blknr * (uint64_t)fs->blksz),
  618. (struct journal_superblock_t *)temp_buff, fs->blksz);
  619. free(temp_buff);
  620. }
  621. ext4fs_free_journal();
  622. /* get the superblock */
  623. ext4_read_superblock((char *)fs->sb);
  624. new_feature_incompat = le32_to_cpu(fs->sb->feature_incompat);
  625. new_feature_incompat &= ~EXT3_FEATURE_INCOMPAT_RECOVER;
  626. fs->sb->feature_incompat = cpu_to_le32(new_feature_incompat);
  627. put_ext4((uint64_t)(SUPERBLOCK_SIZE),
  628. (struct ext2_sblock *)fs->sb, (uint32_t)SUPERBLOCK_SIZE);
  629. free(fs->sb);
  630. fs->sb = NULL;
  631. if (fs->blk_bmaps) {
  632. for (i = 0; i < fs->no_blkgrp; i++) {
  633. free(fs->blk_bmaps[i]);
  634. fs->blk_bmaps[i] = NULL;
  635. }
  636. free(fs->blk_bmaps);
  637. fs->blk_bmaps = NULL;
  638. }
  639. if (fs->inode_bmaps) {
  640. for (i = 0; i < fs->no_blkgrp; i++) {
  641. free(fs->inode_bmaps[i]);
  642. fs->inode_bmaps[i] = NULL;
  643. }
  644. free(fs->inode_bmaps);
  645. fs->inode_bmaps = NULL;
  646. }
  647. free(fs->gdtable);
  648. fs->gdtable = NULL;
  649. /*
  650. * reinitiliazed the global inode and
  651. * block bitmap first execution check variables
  652. */
  653. fs->first_pass_ibmap = 0;
  654. fs->first_pass_bbmap = 0;
  655. fs->curr_inode_no = 0;
  656. fs->curr_blkno = 0;
  657. }
  658. /*
  659. * Write data to filesystem blocks. Uses same optimization for
  660. * contigous sectors as ext4fs_read_file
  661. */
  662. static int ext4fs_write_file(struct ext2_inode *file_inode,
  663. int pos, unsigned int len, char *buf)
  664. {
  665. int i;
  666. int blockcnt;
  667. uint32_t filesize = le32_to_cpu(file_inode->size);
  668. struct ext_filesystem *fs = get_fs();
  669. int log2blksz = fs->dev_desc->log2blksz;
  670. int log2_fs_blocksize = LOG2_BLOCK_SIZE(ext4fs_root) - log2blksz;
  671. int previous_block_number = -1;
  672. int delayed_start = 0;
  673. int delayed_extent = 0;
  674. int delayed_next = 0;
  675. char *delayed_buf = NULL;
  676. /* Adjust len so it we can't read past the end of the file. */
  677. if (len > filesize)
  678. len = filesize;
  679. blockcnt = ((len + pos) + fs->blksz - 1) / fs->blksz;
  680. for (i = pos / fs->blksz; i < blockcnt; i++) {
  681. long int blknr;
  682. int blockend = fs->blksz;
  683. int skipfirst = 0;
  684. blknr = read_allocated_block(file_inode, i);
  685. if (blknr <= 0)
  686. return -1;
  687. blknr = blknr << log2_fs_blocksize;
  688. if (blknr) {
  689. if (previous_block_number != -1) {
  690. if (delayed_next == blknr) {
  691. delayed_extent += blockend;
  692. delayed_next += blockend >> log2blksz;
  693. } else { /* spill */
  694. put_ext4((uint64_t)
  695. ((uint64_t)delayed_start << log2blksz),
  696. delayed_buf,
  697. (uint32_t) delayed_extent);
  698. previous_block_number = blknr;
  699. delayed_start = blknr;
  700. delayed_extent = blockend;
  701. delayed_buf = buf;
  702. delayed_next = blknr +
  703. (blockend >> log2blksz);
  704. }
  705. } else {
  706. previous_block_number = blknr;
  707. delayed_start = blknr;
  708. delayed_extent = blockend;
  709. delayed_buf = buf;
  710. delayed_next = blknr +
  711. (blockend >> log2blksz);
  712. }
  713. } else {
  714. if (previous_block_number != -1) {
  715. /* spill */
  716. put_ext4((uint64_t) ((uint64_t)delayed_start <<
  717. log2blksz),
  718. delayed_buf,
  719. (uint32_t) delayed_extent);
  720. previous_block_number = -1;
  721. }
  722. memset(buf, 0, fs->blksz - skipfirst);
  723. }
  724. buf += fs->blksz - skipfirst;
  725. }
  726. if (previous_block_number != -1) {
  727. /* spill */
  728. put_ext4((uint64_t) ((uint64_t)delayed_start << log2blksz),
  729. delayed_buf, (uint32_t) delayed_extent);
  730. previous_block_number = -1;
  731. }
  732. return len;
  733. }
  734. int ext4fs_write(const char *fname, unsigned char *buffer,
  735. unsigned long sizebytes)
  736. {
  737. int ret = 0;
  738. struct ext2_inode *file_inode = NULL;
  739. unsigned char *inode_buffer = NULL;
  740. int parent_inodeno;
  741. int inodeno;
  742. time_t timestamp = 0;
  743. uint64_t bytes_reqd_for_file;
  744. unsigned int blks_reqd_for_file;
  745. unsigned int blocks_remaining;
  746. int existing_file_inodeno;
  747. char *temp_ptr = NULL;
  748. long int itable_blkno;
  749. long int parent_itable_blkno;
  750. long int blkoff;
  751. struct ext2_sblock *sblock = &(ext4fs_root->sblock);
  752. unsigned int inodes_per_block;
  753. unsigned int ibmap_idx;
  754. struct ext2_block_group *bgd = NULL;
  755. struct ext_filesystem *fs = get_fs();
  756. ALLOC_CACHE_ALIGN_BUFFER(char, filename, 256);
  757. memset(filename, 0x00, 256);
  758. g_parent_inode = zalloc(fs->inodesz);
  759. if (!g_parent_inode)
  760. goto fail;
  761. if (ext4fs_init() != 0) {
  762. printf("error in File System init\n");
  763. return -1;
  764. }
  765. inodes_per_block = fs->blksz / fs->inodesz;
  766. parent_inodeno = ext4fs_get_parent_inode_num(fname, filename, F_FILE);
  767. if (parent_inodeno == -1)
  768. goto fail;
  769. if (ext4fs_iget(parent_inodeno, g_parent_inode))
  770. goto fail;
  771. /* do not mess up a directory using hash trees */
  772. if (le32_to_cpu(g_parent_inode->flags) & EXT4_INDEX_FL) {
  773. printf("hash tree directory\n");
  774. goto fail;
  775. }
  776. /* check if the filename is already present in root */
  777. existing_file_inodeno = ext4fs_filename_unlink(filename);
  778. if (existing_file_inodeno != -1) {
  779. ret = ext4fs_delete_file(existing_file_inodeno);
  780. fs->first_pass_bbmap = 0;
  781. fs->curr_blkno = 0;
  782. fs->first_pass_ibmap = 0;
  783. fs->curr_inode_no = 0;
  784. if (ret)
  785. goto fail;
  786. }
  787. /* calucalate how many blocks required */
  788. bytes_reqd_for_file = sizebytes;
  789. blks_reqd_for_file = lldiv(bytes_reqd_for_file, fs->blksz);
  790. if (do_div(bytes_reqd_for_file, fs->blksz) != 0) {
  791. blks_reqd_for_file++;
  792. debug("total bytes for a file %u\n", blks_reqd_for_file);
  793. }
  794. blocks_remaining = blks_reqd_for_file;
  795. /* test for available space in partition */
  796. if (le32_to_cpu(fs->sb->free_blocks) < blks_reqd_for_file) {
  797. printf("Not enough space on partition !!!\n");
  798. goto fail;
  799. }
  800. inodeno = ext4fs_update_parent_dentry(filename, FILETYPE_REG);
  801. if (inodeno == -1)
  802. goto fail;
  803. /* prepare file inode */
  804. inode_buffer = zalloc(fs->inodesz);
  805. if (!inode_buffer)
  806. goto fail;
  807. file_inode = (struct ext2_inode *)inode_buffer;
  808. file_inode->mode = cpu_to_le16(S_IFREG | S_IRWXU |
  809. S_IRGRP | S_IROTH | S_IXGRP | S_IXOTH);
  810. /* ToDo: Update correct time */
  811. file_inode->mtime = cpu_to_le32(timestamp);
  812. file_inode->atime = cpu_to_le32(timestamp);
  813. file_inode->ctime = cpu_to_le32(timestamp);
  814. file_inode->nlinks = cpu_to_le16(1);
  815. file_inode->size = cpu_to_le32(sizebytes);
  816. /* Allocate data blocks */
  817. ext4fs_allocate_blocks(file_inode, blocks_remaining,
  818. &blks_reqd_for_file);
  819. file_inode->blockcnt = cpu_to_le32((blks_reqd_for_file * fs->blksz) >>
  820. fs->dev_desc->log2blksz);
  821. temp_ptr = zalloc(fs->blksz);
  822. if (!temp_ptr)
  823. goto fail;
  824. ibmap_idx = inodeno / le32_to_cpu(ext4fs_root->sblock.inodes_per_group);
  825. inodeno--;
  826. bgd = ext4fs_get_group_descriptor(fs, ibmap_idx);
  827. itable_blkno = ext4fs_bg_get_inode_table_id(bgd, fs) +
  828. (inodeno % le32_to_cpu(sblock->inodes_per_group)) /
  829. inodes_per_block;
  830. blkoff = (inodeno % inodes_per_block) * fs->inodesz;
  831. ext4fs_devread((lbaint_t)itable_blkno * fs->sect_perblk, 0, fs->blksz,
  832. temp_ptr);
  833. if (ext4fs_log_journal(temp_ptr, itable_blkno))
  834. goto fail;
  835. memcpy(temp_ptr + blkoff, inode_buffer, fs->inodesz);
  836. if (ext4fs_put_metadata(temp_ptr, itable_blkno))
  837. goto fail;
  838. /* copy the file content into data blocks */
  839. if (ext4fs_write_file(file_inode, 0, sizebytes, (char *)buffer) == -1) {
  840. printf("Error in copying content\n");
  841. /* FIXME: Deallocate data blocks */
  842. goto fail;
  843. }
  844. ibmap_idx = parent_inodeno / le32_to_cpu(ext4fs_root->sblock.inodes_per_group);
  845. parent_inodeno--;
  846. bgd = ext4fs_get_group_descriptor(fs, ibmap_idx);
  847. parent_itable_blkno = ext4fs_bg_get_inode_table_id(bgd, fs) +
  848. (parent_inodeno %
  849. le32_to_cpu(sblock->inodes_per_group)) / inodes_per_block;
  850. blkoff = (parent_inodeno % inodes_per_block) * fs->inodesz;
  851. if (parent_itable_blkno != itable_blkno) {
  852. memset(temp_ptr, '\0', fs->blksz);
  853. ext4fs_devread((lbaint_t)parent_itable_blkno * fs->sect_perblk,
  854. 0, fs->blksz, temp_ptr);
  855. if (ext4fs_log_journal(temp_ptr, parent_itable_blkno))
  856. goto fail;
  857. memcpy(temp_ptr + blkoff, g_parent_inode, fs->inodesz);
  858. if (ext4fs_put_metadata(temp_ptr, parent_itable_blkno))
  859. goto fail;
  860. } else {
  861. /*
  862. * If parent and child fall in same inode table block
  863. * both should be kept in 1 buffer
  864. */
  865. memcpy(temp_ptr + blkoff, g_parent_inode, fs->inodesz);
  866. gd_index--;
  867. if (ext4fs_put_metadata(temp_ptr, itable_blkno))
  868. goto fail;
  869. }
  870. ext4fs_update();
  871. ext4fs_deinit();
  872. fs->first_pass_bbmap = 0;
  873. fs->curr_blkno = 0;
  874. fs->first_pass_ibmap = 0;
  875. fs->curr_inode_no = 0;
  876. free(inode_buffer);
  877. free(g_parent_inode);
  878. free(temp_ptr);
  879. g_parent_inode = NULL;
  880. return 0;
  881. fail:
  882. ext4fs_deinit();
  883. free(inode_buffer);
  884. free(g_parent_inode);
  885. free(temp_ptr);
  886. g_parent_inode = NULL;
  887. return -1;
  888. }
  889. int ext4_write_file(const char *filename, void *buf, loff_t offset,
  890. loff_t len, loff_t *actwrite)
  891. {
  892. int ret;
  893. if (offset != 0) {
  894. printf("** Cannot support non-zero offset **\n");
  895. return -1;
  896. }
  897. ret = ext4fs_write(filename, buf, len);
  898. if (ret) {
  899. printf("** Error ext4fs_write() **\n");
  900. goto fail;
  901. }
  902. *actwrite = len;
  903. return 0;
  904. fail:
  905. *actwrite = 0;
  906. return -1;
  907. }