ext4_write.c 26 KB

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