1 /* 2 * This file is part of UBIFS. 3 * 4 * Copyright (C) 2006-2008 Nokia Corporation. 5 * 6 * (C) Copyright 2008-2010 7 * Stefan Roese, DENX Software Engineering, sr@denx.de. 8 * 9 * This program is free software; you can redistribute it and/or modify it 10 * under the terms of the GNU General Public License version 2 as published by 11 * the Free Software Foundation. 12 * 13 * This program is distributed in the hope that it will be useful, but WITHOUT 14 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 15 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for 16 * more details. 17 * 18 * You should have received a copy of the GNU General Public License along with 19 * this program; if not, write to the Free Software Foundation, Inc., 51 20 * Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 21 * 22 * Authors: Artem Bityutskiy (Битюцкий Артём) 23 * Adrian Hunter 24 */ 25 26 #include <common.h> 27 #include <memalign.h> 28 #include "ubifs.h" 29 #include <u-boot/zlib.h> 30 31 #include <linux/err.h> 32 #include <linux/lzo.h> 33 34 DECLARE_GLOBAL_DATA_PTR; 35 36 /* compress.c */ 37 38 /* 39 * We need a wrapper for zunzip() because the parameters are 40 * incompatible with the lzo decompressor. 41 */ 42 static int gzip_decompress(const unsigned char *in, size_t in_len, 43 unsigned char *out, size_t *out_len) 44 { 45 return zunzip(out, *out_len, (unsigned char *)in, 46 (unsigned long *)out_len, 0, 0); 47 } 48 49 /* Fake description object for the "none" compressor */ 50 static struct ubifs_compressor none_compr = { 51 .compr_type = UBIFS_COMPR_NONE, 52 .name = "none", 53 .capi_name = "", 54 .decompress = NULL, 55 }; 56 57 static struct ubifs_compressor lzo_compr = { 58 .compr_type = UBIFS_COMPR_LZO, 59 #ifndef __UBOOT__ 60 .comp_mutex = &lzo_mutex, 61 #endif 62 .name = "lzo", 63 .capi_name = "lzo", 64 .decompress = lzo1x_decompress_safe, 65 }; 66 67 static struct ubifs_compressor zlib_compr = { 68 .compr_type = UBIFS_COMPR_ZLIB, 69 #ifndef __UBOOT__ 70 .comp_mutex = &deflate_mutex, 71 .decomp_mutex = &inflate_mutex, 72 #endif 73 .name = "zlib", 74 .capi_name = "deflate", 75 .decompress = gzip_decompress, 76 }; 77 78 /* All UBIFS compressors */ 79 struct ubifs_compressor *ubifs_compressors[UBIFS_COMPR_TYPES_CNT]; 80 81 82 #ifdef __UBOOT__ 83 /* from mm/util.c */ 84 85 /** 86 * kmemdup - duplicate region of memory 87 * 88 * @src: memory region to duplicate 89 * @len: memory region length 90 * @gfp: GFP mask to use 91 */ 92 void *kmemdup(const void *src, size_t len, gfp_t gfp) 93 { 94 void *p; 95 96 p = kmalloc(len, gfp); 97 if (p) 98 memcpy(p, src, len); 99 return p; 100 } 101 102 struct crypto_comp { 103 int compressor; 104 }; 105 106 static inline struct crypto_comp 107 *crypto_alloc_comp(const char *alg_name, u32 type, u32 mask) 108 { 109 struct ubifs_compressor *comp; 110 struct crypto_comp *ptr; 111 int i = 0; 112 113 ptr = malloc_cache_aligned(sizeof(struct crypto_comp)); 114 while (i < UBIFS_COMPR_TYPES_CNT) { 115 comp = ubifs_compressors[i]; 116 if (!comp) { 117 i++; 118 continue; 119 } 120 if (strncmp(alg_name, comp->capi_name, strlen(alg_name)) == 0) { 121 ptr->compressor = i; 122 return ptr; 123 } 124 i++; 125 } 126 if (i >= UBIFS_COMPR_TYPES_CNT) { 127 dbg_gen("invalid compression type %s", alg_name); 128 free (ptr); 129 return NULL; 130 } 131 return ptr; 132 } 133 static inline int 134 crypto_comp_decompress(const struct ubifs_info *c, struct crypto_comp *tfm, 135 const u8 *src, unsigned int slen, u8 *dst, 136 unsigned int *dlen) 137 { 138 struct ubifs_compressor *compr = ubifs_compressors[tfm->compressor]; 139 int err; 140 141 if (compr->compr_type == UBIFS_COMPR_NONE) { 142 memcpy(dst, src, slen); 143 *dlen = slen; 144 return 0; 145 } 146 147 err = compr->decompress(src, slen, dst, (size_t *)dlen); 148 if (err) 149 ubifs_err(c, "cannot decompress %d bytes, compressor %s, " 150 "error %d", slen, compr->name, err); 151 152 return err; 153 154 return 0; 155 } 156 157 /* from shrinker.c */ 158 159 /* Global clean znode counter (for all mounted UBIFS instances) */ 160 atomic_long_t ubifs_clean_zn_cnt; 161 162 #endif 163 164 /** 165 * ubifs_decompress - decompress data. 166 * @in_buf: data to decompress 167 * @in_len: length of the data to decompress 168 * @out_buf: output buffer where decompressed data should 169 * @out_len: output length is returned here 170 * @compr_type: type of compression 171 * 172 * This function decompresses data from buffer @in_buf into buffer @out_buf. 173 * The length of the uncompressed data is returned in @out_len. This functions 174 * returns %0 on success or a negative error code on failure. 175 */ 176 int ubifs_decompress(const struct ubifs_info *c, const void *in_buf, 177 int in_len, void *out_buf, int *out_len, int compr_type) 178 { 179 int err; 180 struct ubifs_compressor *compr; 181 182 if (unlikely(compr_type < 0 || compr_type >= UBIFS_COMPR_TYPES_CNT)) { 183 ubifs_err(c, "invalid compression type %d", compr_type); 184 return -EINVAL; 185 } 186 187 compr = ubifs_compressors[compr_type]; 188 189 if (unlikely(!compr->capi_name)) { 190 ubifs_err(c, "%s compression is not compiled in", compr->name); 191 return -EINVAL; 192 } 193 194 if (compr_type == UBIFS_COMPR_NONE) { 195 memcpy(out_buf, in_buf, in_len); 196 *out_len = in_len; 197 return 0; 198 } 199 200 if (compr->decomp_mutex) 201 mutex_lock(compr->decomp_mutex); 202 err = crypto_comp_decompress(c, compr->cc, in_buf, in_len, out_buf, 203 (unsigned int *)out_len); 204 if (compr->decomp_mutex) 205 mutex_unlock(compr->decomp_mutex); 206 if (err) 207 ubifs_err(c, "cannot decompress %d bytes, compressor %s," 208 " error %d", in_len, compr->name, err); 209 210 return err; 211 } 212 213 /** 214 * compr_init - initialize a compressor. 215 * @compr: compressor description object 216 * 217 * This function initializes the requested compressor and returns zero in case 218 * of success or a negative error code in case of failure. 219 */ 220 static int __init compr_init(struct ubifs_compressor *compr) 221 { 222 ubifs_compressors[compr->compr_type] = compr; 223 224 #ifdef CONFIG_NEEDS_MANUAL_RELOC 225 ubifs_compressors[compr->compr_type]->name += gd->reloc_off; 226 ubifs_compressors[compr->compr_type]->capi_name += gd->reloc_off; 227 ubifs_compressors[compr->compr_type]->decompress += gd->reloc_off; 228 #endif 229 230 if (compr->capi_name) { 231 compr->cc = crypto_alloc_comp(compr->capi_name, 0, 0); 232 if (IS_ERR(compr->cc)) { 233 dbg_gen("cannot initialize compressor %s," 234 " error %ld", compr->name, 235 PTR_ERR(compr->cc)); 236 return PTR_ERR(compr->cc); 237 } 238 } 239 240 return 0; 241 } 242 243 /** 244 * ubifs_compressors_init - initialize UBIFS compressors. 245 * 246 * This function initializes the compressor which were compiled in. Returns 247 * zero in case of success and a negative error code in case of failure. 248 */ 249 int __init ubifs_compressors_init(void) 250 { 251 int err; 252 253 err = compr_init(&lzo_compr); 254 if (err) 255 return err; 256 257 err = compr_init(&zlib_compr); 258 if (err) 259 return err; 260 261 err = compr_init(&none_compr); 262 if (err) 263 return err; 264 265 return 0; 266 } 267 268 /* 269 * ubifsls... 270 */ 271 272 static int filldir(struct ubifs_info *c, const char *name, int namlen, 273 u64 ino, unsigned int d_type) 274 { 275 struct inode *inode; 276 char filetime[32]; 277 278 switch (d_type) { 279 case UBIFS_ITYPE_REG: 280 printf("\t"); 281 break; 282 case UBIFS_ITYPE_DIR: 283 printf("<DIR>\t"); 284 break; 285 case UBIFS_ITYPE_LNK: 286 printf("<LNK>\t"); 287 break; 288 default: 289 printf("other\t"); 290 break; 291 } 292 293 inode = ubifs_iget(c->vfs_sb, ino); 294 if (IS_ERR(inode)) { 295 printf("%s: Error in ubifs_iget(), ino=%lld ret=%p!\n", 296 __func__, ino, inode); 297 return -1; 298 } 299 ctime_r((time_t *)&inode->i_mtime, filetime); 300 printf("%9lld %24.24s ", inode->i_size, filetime); 301 #ifndef __UBOOT__ 302 ubifs_iput(inode); 303 #endif 304 305 printf("%s\n", name); 306 307 return 0; 308 } 309 310 static int ubifs_printdir(struct file *file, void *dirent) 311 { 312 int err, over = 0; 313 struct qstr nm; 314 union ubifs_key key; 315 struct ubifs_dent_node *dent; 316 struct inode *dir = file->f_path.dentry->d_inode; 317 struct ubifs_info *c = dir->i_sb->s_fs_info; 318 319 dbg_gen("dir ino %lu, f_pos %#llx", dir->i_ino, file->f_pos); 320 321 if (file->f_pos > UBIFS_S_KEY_HASH_MASK || file->f_pos == 2) 322 /* 323 * The directory was seek'ed to a senseless position or there 324 * are no more entries. 325 */ 326 return 0; 327 328 if (file->f_pos == 1) { 329 /* Find the first entry in TNC and save it */ 330 lowest_dent_key(c, &key, dir->i_ino); 331 nm.name = NULL; 332 dent = ubifs_tnc_next_ent(c, &key, &nm); 333 if (IS_ERR(dent)) { 334 err = PTR_ERR(dent); 335 goto out; 336 } 337 338 file->f_pos = key_hash_flash(c, &dent->key); 339 file->private_data = dent; 340 } 341 342 dent = file->private_data; 343 if (!dent) { 344 /* 345 * The directory was seek'ed to and is now readdir'ed. 346 * Find the entry corresponding to @file->f_pos or the 347 * closest one. 348 */ 349 dent_key_init_hash(c, &key, dir->i_ino, file->f_pos); 350 nm.name = NULL; 351 dent = ubifs_tnc_next_ent(c, &key, &nm); 352 if (IS_ERR(dent)) { 353 err = PTR_ERR(dent); 354 goto out; 355 } 356 file->f_pos = key_hash_flash(c, &dent->key); 357 file->private_data = dent; 358 } 359 360 while (1) { 361 dbg_gen("feed '%s', ino %llu, new f_pos %#x", 362 dent->name, (unsigned long long)le64_to_cpu(dent->inum), 363 key_hash_flash(c, &dent->key)); 364 ubifs_assert(le64_to_cpu(dent->ch.sqnum) > ubifs_inode(dir)->creat_sqnum); 365 366 nm.len = le16_to_cpu(dent->nlen); 367 over = filldir(c, (char *)dent->name, nm.len, 368 le64_to_cpu(dent->inum), dent->type); 369 if (over) 370 return 0; 371 372 /* Switch to the next entry */ 373 key_read(c, &dent->key, &key); 374 nm.name = (char *)dent->name; 375 dent = ubifs_tnc_next_ent(c, &key, &nm); 376 if (IS_ERR(dent)) { 377 err = PTR_ERR(dent); 378 goto out; 379 } 380 381 kfree(file->private_data); 382 file->f_pos = key_hash_flash(c, &dent->key); 383 file->private_data = dent; 384 cond_resched(); 385 } 386 387 out: 388 if (err != -ENOENT) { 389 ubifs_err(c, "cannot find next direntry, error %d", err); 390 return err; 391 } 392 393 kfree(file->private_data); 394 file->private_data = NULL; 395 file->f_pos = 2; 396 return 0; 397 } 398 399 static int ubifs_finddir(struct super_block *sb, char *dirname, 400 unsigned long root_inum, unsigned long *inum) 401 { 402 int err; 403 struct qstr nm; 404 union ubifs_key key; 405 struct ubifs_dent_node *dent; 406 struct ubifs_info *c; 407 struct file *file; 408 struct dentry *dentry; 409 struct inode *dir; 410 int ret = 0; 411 412 file = kzalloc(sizeof(struct file), 0); 413 dentry = kzalloc(sizeof(struct dentry), 0); 414 dir = kzalloc(sizeof(struct inode), 0); 415 if (!file || !dentry || !dir) { 416 printf("%s: Error, no memory for malloc!\n", __func__); 417 err = -ENOMEM; 418 goto out; 419 } 420 421 dir->i_sb = sb; 422 file->f_path.dentry = dentry; 423 file->f_path.dentry->d_parent = dentry; 424 file->f_path.dentry->d_inode = dir; 425 file->f_path.dentry->d_inode->i_ino = root_inum; 426 c = sb->s_fs_info; 427 428 dbg_gen("dir ino %lu, f_pos %#llx", dir->i_ino, file->f_pos); 429 430 /* Find the first entry in TNC and save it */ 431 lowest_dent_key(c, &key, dir->i_ino); 432 nm.name = NULL; 433 dent = ubifs_tnc_next_ent(c, &key, &nm); 434 if (IS_ERR(dent)) { 435 err = PTR_ERR(dent); 436 goto out; 437 } 438 439 file->f_pos = key_hash_flash(c, &dent->key); 440 file->private_data = dent; 441 442 while (1) { 443 dbg_gen("feed '%s', ino %llu, new f_pos %#x", 444 dent->name, (unsigned long long)le64_to_cpu(dent->inum), 445 key_hash_flash(c, &dent->key)); 446 ubifs_assert(le64_to_cpu(dent->ch.sqnum) > ubifs_inode(dir)->creat_sqnum); 447 448 nm.len = le16_to_cpu(dent->nlen); 449 if ((strncmp(dirname, (char *)dent->name, nm.len) == 0) && 450 (strlen(dirname) == nm.len)) { 451 *inum = le64_to_cpu(dent->inum); 452 ret = 1; 453 goto out_free; 454 } 455 456 /* Switch to the next entry */ 457 key_read(c, &dent->key, &key); 458 nm.name = (char *)dent->name; 459 dent = ubifs_tnc_next_ent(c, &key, &nm); 460 if (IS_ERR(dent)) { 461 err = PTR_ERR(dent); 462 goto out; 463 } 464 465 kfree(file->private_data); 466 file->f_pos = key_hash_flash(c, &dent->key); 467 file->private_data = dent; 468 cond_resched(); 469 } 470 471 out: 472 if (err != -ENOENT) 473 dbg_gen("cannot find next direntry, error %d", err); 474 475 out_free: 476 if (file->private_data) 477 kfree(file->private_data); 478 if (file) 479 free(file); 480 if (dentry) 481 free(dentry); 482 if (dir) 483 free(dir); 484 485 return ret; 486 } 487 488 static unsigned long ubifs_findfile(struct super_block *sb, char *filename) 489 { 490 int ret; 491 char *next; 492 char fpath[128]; 493 char symlinkpath[128]; 494 char *name = fpath; 495 unsigned long root_inum = 1; 496 unsigned long inum; 497 int symlink_count = 0; /* Don't allow symlink recursion */ 498 char link_name[64]; 499 500 strcpy(fpath, filename); 501 502 /* Remove all leading slashes */ 503 while (*name == '/') 504 name++; 505 506 /* 507 * Handle root-direcoty ('/') 508 */ 509 inum = root_inum; 510 if (!name || *name == '\0') 511 return inum; 512 513 for (;;) { 514 struct inode *inode; 515 struct ubifs_inode *ui; 516 517 /* Extract the actual part from the pathname. */ 518 next = strchr(name, '/'); 519 if (next) { 520 /* Remove all leading slashes. */ 521 while (*next == '/') 522 *(next++) = '\0'; 523 } 524 525 ret = ubifs_finddir(sb, name, root_inum, &inum); 526 if (!ret) 527 return 0; 528 inode = ubifs_iget(sb, inum); 529 530 if (!inode) 531 return 0; 532 ui = ubifs_inode(inode); 533 534 if ((inode->i_mode & S_IFMT) == S_IFLNK) { 535 char buf[128]; 536 537 /* We have some sort of symlink recursion, bail out */ 538 if (symlink_count++ > 8) { 539 printf("Symlink recursion, aborting\n"); 540 return 0; 541 } 542 memcpy(link_name, ui->data, ui->data_len); 543 link_name[ui->data_len] = '\0'; 544 545 if (link_name[0] == '/') { 546 /* Absolute path, redo everything without 547 * the leading slash */ 548 next = name = link_name + 1; 549 root_inum = 1; 550 continue; 551 } 552 /* Relative to cur dir */ 553 sprintf(buf, "%s/%s", 554 link_name, next == NULL ? "" : next); 555 memcpy(symlinkpath, buf, sizeof(buf)); 556 next = name = symlinkpath; 557 continue; 558 } 559 560 /* 561 * Check if directory with this name exists 562 */ 563 564 /* Found the node! */ 565 if (!next || *next == '\0') 566 return inum; 567 568 root_inum = inum; 569 name = next; 570 } 571 572 return 0; 573 } 574 575 int ubifs_set_blk_dev(block_dev_desc_t *rbdd, disk_partition_t *info) 576 { 577 if (rbdd) { 578 debug("UBIFS cannot be used with normal block devices\n"); 579 return -1; 580 } 581 582 /* 583 * Should never happen since get_device_and_partition() already checks 584 * this, but better safe then sorry. 585 */ 586 if (!ubifs_is_mounted()) { 587 debug("UBIFS not mounted, use ubifsmount to mount volume first!\n"); 588 return -1; 589 } 590 591 return 0; 592 } 593 594 int ubifs_ls(const char *filename) 595 { 596 struct ubifs_info *c = ubifs_sb->s_fs_info; 597 struct file *file; 598 struct dentry *dentry; 599 struct inode *dir; 600 void *dirent = NULL; 601 unsigned long inum; 602 int ret = 0; 603 604 c->ubi = ubi_open_volume(c->vi.ubi_num, c->vi.vol_id, UBI_READONLY); 605 inum = ubifs_findfile(ubifs_sb, (char *)filename); 606 if (!inum) { 607 ret = -1; 608 goto out; 609 } 610 611 file = kzalloc(sizeof(struct file), 0); 612 dentry = kzalloc(sizeof(struct dentry), 0); 613 dir = kzalloc(sizeof(struct inode), 0); 614 if (!file || !dentry || !dir) { 615 printf("%s: Error, no memory for malloc!\n", __func__); 616 ret = -ENOMEM; 617 goto out_mem; 618 } 619 620 dir->i_sb = ubifs_sb; 621 file->f_path.dentry = dentry; 622 file->f_path.dentry->d_parent = dentry; 623 file->f_path.dentry->d_inode = dir; 624 file->f_path.dentry->d_inode->i_ino = inum; 625 file->f_pos = 1; 626 file->private_data = NULL; 627 ubifs_printdir(file, dirent); 628 629 out_mem: 630 if (file) 631 free(file); 632 if (dentry) 633 free(dentry); 634 if (dir) 635 free(dir); 636 637 out: 638 ubi_close_volume(c->ubi); 639 return ret; 640 } 641 642 int ubifs_exists(const char *filename) 643 { 644 struct ubifs_info *c = ubifs_sb->s_fs_info; 645 unsigned long inum; 646 647 c->ubi = ubi_open_volume(c->vi.ubi_num, c->vi.vol_id, UBI_READONLY); 648 inum = ubifs_findfile(ubifs_sb, (char *)filename); 649 ubi_close_volume(c->ubi); 650 651 return inum != 0; 652 } 653 654 int ubifs_size(const char *filename, loff_t *size) 655 { 656 struct ubifs_info *c = ubifs_sb->s_fs_info; 657 unsigned long inum; 658 struct inode *inode; 659 int err = 0; 660 661 c->ubi = ubi_open_volume(c->vi.ubi_num, c->vi.vol_id, UBI_READONLY); 662 663 inum = ubifs_findfile(ubifs_sb, (char *)filename); 664 if (!inum) { 665 err = -1; 666 goto out; 667 } 668 669 inode = ubifs_iget(ubifs_sb, inum); 670 if (IS_ERR(inode)) { 671 printf("%s: Error reading inode %ld!\n", __func__, inum); 672 err = PTR_ERR(inode); 673 goto out; 674 } 675 676 *size = inode->i_size; 677 678 ubifs_iput(inode); 679 out: 680 ubi_close_volume(c->ubi); 681 return err; 682 } 683 684 /* 685 * ubifsload... 686 */ 687 688 /* file.c */ 689 690 static inline void *kmap(struct page *page) 691 { 692 return page->addr; 693 } 694 695 static int read_block(struct inode *inode, void *addr, unsigned int block, 696 struct ubifs_data_node *dn) 697 { 698 struct ubifs_info *c = inode->i_sb->s_fs_info; 699 int err, len, out_len; 700 union ubifs_key key; 701 unsigned int dlen; 702 703 data_key_init(c, &key, inode->i_ino, block); 704 err = ubifs_tnc_lookup(c, &key, dn); 705 if (err) { 706 if (err == -ENOENT) 707 /* Not found, so it must be a hole */ 708 memset(addr, 0, UBIFS_BLOCK_SIZE); 709 return err; 710 } 711 712 ubifs_assert(le64_to_cpu(dn->ch.sqnum) > ubifs_inode(inode)->creat_sqnum); 713 714 len = le32_to_cpu(dn->size); 715 if (len <= 0 || len > UBIFS_BLOCK_SIZE) 716 goto dump; 717 718 dlen = le32_to_cpu(dn->ch.len) - UBIFS_DATA_NODE_SZ; 719 out_len = UBIFS_BLOCK_SIZE; 720 err = ubifs_decompress(c, &dn->data, dlen, addr, &out_len, 721 le16_to_cpu(dn->compr_type)); 722 if (err || len != out_len) 723 goto dump; 724 725 /* 726 * Data length can be less than a full block, even for blocks that are 727 * not the last in the file (e.g., as a result of making a hole and 728 * appending data). Ensure that the remainder is zeroed out. 729 */ 730 if (len < UBIFS_BLOCK_SIZE) 731 memset(addr + len, 0, UBIFS_BLOCK_SIZE - len); 732 733 return 0; 734 735 dump: 736 ubifs_err(c, "bad data node (block %u, inode %lu)", 737 block, inode->i_ino); 738 ubifs_dump_node(c, dn); 739 return -EINVAL; 740 } 741 742 static int do_readpage(struct ubifs_info *c, struct inode *inode, 743 struct page *page, int last_block_size) 744 { 745 void *addr; 746 int err = 0, i; 747 unsigned int block, beyond; 748 struct ubifs_data_node *dn; 749 loff_t i_size = inode->i_size; 750 751 dbg_gen("ino %lu, pg %lu, i_size %lld", 752 inode->i_ino, page->index, i_size); 753 754 addr = kmap(page); 755 756 block = page->index << UBIFS_BLOCKS_PER_PAGE_SHIFT; 757 beyond = (i_size + UBIFS_BLOCK_SIZE - 1) >> UBIFS_BLOCK_SHIFT; 758 if (block >= beyond) { 759 /* Reading beyond inode */ 760 memset(addr, 0, PAGE_CACHE_SIZE); 761 goto out; 762 } 763 764 dn = kmalloc(UBIFS_MAX_DATA_NODE_SZ, GFP_NOFS); 765 if (!dn) 766 return -ENOMEM; 767 768 i = 0; 769 while (1) { 770 int ret; 771 772 if (block >= beyond) { 773 /* Reading beyond inode */ 774 err = -ENOENT; 775 memset(addr, 0, UBIFS_BLOCK_SIZE); 776 } else { 777 /* 778 * Reading last block? Make sure to not write beyond 779 * the requested size in the destination buffer. 780 */ 781 if (((block + 1) == beyond) || last_block_size) { 782 void *buff; 783 int dlen; 784 785 /* 786 * We need to buffer the data locally for the 787 * last block. This is to not pad the 788 * destination area to a multiple of 789 * UBIFS_BLOCK_SIZE. 790 */ 791 buff = malloc_cache_aligned(UBIFS_BLOCK_SIZE); 792 if (!buff) { 793 printf("%s: Error, malloc fails!\n", 794 __func__); 795 err = -ENOMEM; 796 break; 797 } 798 799 /* Read block-size into temp buffer */ 800 ret = read_block(inode, buff, block, dn); 801 if (ret) { 802 err = ret; 803 if (err != -ENOENT) { 804 free(buff); 805 break; 806 } 807 } 808 809 if (last_block_size) 810 dlen = last_block_size; 811 else 812 dlen = le32_to_cpu(dn->size); 813 814 /* Now copy required size back to dest */ 815 memcpy(addr, buff, dlen); 816 817 free(buff); 818 } else { 819 ret = read_block(inode, addr, block, dn); 820 if (ret) { 821 err = ret; 822 if (err != -ENOENT) 823 break; 824 } 825 } 826 } 827 if (++i >= UBIFS_BLOCKS_PER_PAGE) 828 break; 829 block += 1; 830 addr += UBIFS_BLOCK_SIZE; 831 } 832 if (err) { 833 if (err == -ENOENT) { 834 /* Not found, so it must be a hole */ 835 dbg_gen("hole"); 836 goto out_free; 837 } 838 ubifs_err(c, "cannot read page %lu of inode %lu, error %d", 839 page->index, inode->i_ino, err); 840 goto error; 841 } 842 843 out_free: 844 kfree(dn); 845 out: 846 return 0; 847 848 error: 849 kfree(dn); 850 return err; 851 } 852 853 int ubifs_read(const char *filename, void *buf, loff_t offset, 854 loff_t size, loff_t *actread) 855 { 856 struct ubifs_info *c = ubifs_sb->s_fs_info; 857 unsigned long inum; 858 struct inode *inode; 859 struct page page; 860 int err = 0; 861 int i; 862 int count; 863 int last_block_size = 0; 864 865 *actread = 0; 866 867 if (offset & (PAGE_SIZE - 1)) { 868 printf("ubifs: Error offset must be a multple of %d\n", 869 PAGE_SIZE); 870 return -1; 871 } 872 873 c->ubi = ubi_open_volume(c->vi.ubi_num, c->vi.vol_id, UBI_READONLY); 874 /* ubifs_findfile will resolve symlinks, so we know that we get 875 * the real file here */ 876 inum = ubifs_findfile(ubifs_sb, (char *)filename); 877 if (!inum) { 878 err = -1; 879 goto out; 880 } 881 882 /* 883 * Read file inode 884 */ 885 inode = ubifs_iget(ubifs_sb, inum); 886 if (IS_ERR(inode)) { 887 printf("%s: Error reading inode %ld!\n", __func__, inum); 888 err = PTR_ERR(inode); 889 goto out; 890 } 891 892 if (offset > inode->i_size) { 893 printf("ubifs: Error offset (%lld) > file-size (%lld)\n", 894 offset, size); 895 err = -1; 896 goto put_inode; 897 } 898 899 /* 900 * If no size was specified or if size bigger than filesize 901 * set size to filesize 902 */ 903 if ((size == 0) || (size > (inode->i_size - offset))) 904 size = inode->i_size - offset; 905 906 count = (size + UBIFS_BLOCK_SIZE - 1) >> UBIFS_BLOCK_SHIFT; 907 908 page.addr = buf; 909 page.index = offset / PAGE_SIZE; 910 page.inode = inode; 911 for (i = 0; i < count; i++) { 912 /* 913 * Make sure to not read beyond the requested size 914 */ 915 if (((i + 1) == count) && (size < inode->i_size)) 916 last_block_size = size - (i * PAGE_SIZE); 917 918 err = do_readpage(c, inode, &page, last_block_size); 919 if (err) 920 break; 921 922 page.addr += PAGE_SIZE; 923 page.index++; 924 } 925 926 if (err) { 927 printf("Error reading file '%s'\n", filename); 928 *actread = i * PAGE_SIZE; 929 } else { 930 *actread = size; 931 } 932 933 put_inode: 934 ubifs_iput(inode); 935 936 out: 937 ubi_close_volume(c->ubi); 938 return err; 939 } 940 941 void ubifs_close(void) 942 { 943 } 944 945 /* Compat wrappers for common/cmd_ubifs.c */ 946 int ubifs_load(char *filename, u32 addr, u32 size) 947 { 948 loff_t actread; 949 int err; 950 951 printf("Loading file '%s' to addr 0x%08x...\n", filename, addr); 952 953 err = ubifs_read(filename, (void *)addr, 0, size, &actread); 954 if (err == 0) { 955 setenv_hex("filesize", actread); 956 printf("Done\n"); 957 } 958 959 return err; 960 } 961 962 void uboot_ubifs_umount(void) 963 { 964 if (ubifs_sb) { 965 printf("Unmounting UBIFS volume %s!\n", 966 ((struct ubifs_info *)(ubifs_sb->s_fs_info))->vi.name); 967 ubifs_umount(ubifs_sb->s_fs_info); 968 ubifs_sb = NULL; 969 } 970 } 971