| /* |
| * fileio.c --- Simple file I/O routines |
| * |
| * Copyright (C) 1997 Theodore Ts'o. |
| * |
| * %Begin-Header% |
| * This file may be redistributed under the terms of the GNU Public |
| * License. |
| * %End-Header% |
| */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| #if HAVE_UNISTD_H |
| #include <unistd.h> |
| #endif |
| |
| #if EXT2_FLAT_INCLUDES |
| #include "ext2_fs.h" |
| #else |
| #include <linux/ext2_fs.h> |
| #endif |
| |
| #include "ext2fs.h" |
| |
| struct ext2_file { |
| errcode_t magic; |
| ext2_filsys fs; |
| ino_t ino; |
| struct ext2_inode inode; |
| int flags; |
| ext2_off_t pos; |
| blk_t blockno; |
| blk_t physblock; |
| char *buf; |
| }; |
| |
| errcode_t ext2fs_file_open(ext2_filsys fs, ino_t ino, |
| int flags, ext2_file_t *ret) |
| { |
| ext2_file_t file; |
| errcode_t retval; |
| |
| /* |
| * Don't let caller create or open a file for writing if the |
| * filesystem is read-only. |
| */ |
| if ((flags & (EXT2_FILE_WRITE | EXT2_FILE_CREATE)) && |
| !(fs->flags & EXT2_FLAG_RW)) |
| return EXT2_ET_RO_FILSYS; |
| |
| retval = ext2fs_get_mem(sizeof(struct ext2_file), (void **) &file); |
| if (retval) |
| return retval; |
| |
| memset(file, 0, sizeof(struct ext2_file)); |
| file->magic = EXT2_ET_MAGIC_EXT2_FILE; |
| file->fs = fs; |
| file->ino = ino; |
| file->flags = flags & EXT2_FILE_MASK; |
| |
| retval = ext2fs_read_inode(fs, ino, &file->inode); |
| if (retval) |
| goto fail; |
| |
| retval = ext2fs_get_mem(fs->blocksize * 2, (void **) &file->buf); |
| if (retval) |
| goto fail; |
| |
| *ret = file; |
| return 0; |
| |
| fail: |
| if (file->buf) |
| ext2fs_free_mem((void **) &file->buf); |
| ext2fs_free_mem((void **) &file); |
| return retval; |
| } |
| |
| /* |
| * This function returns the filesystem handle of a file from the structure |
| */ |
| ext2_filsys ext2fs_file_get_fs(ext2_file_t file) |
| { |
| if (file->magic != EXT2_ET_MAGIC_EXT2_FILE) |
| return 0; |
| return file->fs; |
| } |
| |
| /* |
| * This function flushes the dirty block buffer out to disk if |
| * necessary. |
| */ |
| static errcode_t ext2fs_file_flush(ext2_file_t file) |
| { |
| errcode_t retval; |
| |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| |
| if (!(file->flags & EXT2_FILE_BUF_VALID) || |
| !(file->flags & EXT2_FILE_BUF_DIRTY)) |
| return 0; |
| |
| /* |
| * OK, the physical block hasn't been allocated yet. |
| * Allocate it. |
| */ |
| if (!file->physblock) { |
| retval = ext2fs_bmap(file->fs, file->ino, &file->inode, |
| file->buf, BMAP_ALLOC, |
| file->blockno, &file->physblock); |
| if (retval) |
| return retval; |
| } |
| |
| retval = io_channel_write_blk(file->fs->io, file->physblock, |
| 1, file->buf); |
| if (retval) |
| return retval; |
| |
| file->flags &= ~EXT2_FILE_BUF_DIRTY; |
| |
| return retval; |
| } |
| |
| errcode_t ext2fs_file_close(ext2_file_t file) |
| { |
| errcode_t retval; |
| |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| |
| retval = ext2fs_file_flush(file); |
| |
| if (file->buf) |
| ext2fs_free_mem((void **) &file->buf); |
| ext2fs_free_mem((void **) &file); |
| |
| return retval; |
| } |
| |
| |
| errcode_t ext2fs_file_read(ext2_file_t file, void *buf, |
| unsigned int wanted, unsigned int *got) |
| { |
| ext2_filsys fs; |
| errcode_t retval; |
| blk_t b, pb; |
| unsigned int start, left, c, count = 0; |
| char *ptr = (char *) buf; |
| |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| fs = file->fs; |
| |
| again: |
| if (file->pos >= file->inode.i_size) |
| goto done; |
| |
| b = file->pos / fs->blocksize; |
| if (b != file->blockno) { |
| retval = ext2fs_file_flush(file); |
| if (retval) |
| goto fail; |
| file->flags &= ~EXT2_FILE_BUF_VALID; |
| } |
| file->blockno = b; |
| if (!(file->flags & EXT2_FILE_BUF_VALID)) { |
| retval = ext2fs_bmap(fs, file->ino, &file->inode, |
| file->buf, 0, b, &pb); |
| if (retval) |
| goto fail; |
| if (pb) { |
| file->physblock = pb; |
| retval = io_channel_read_blk(fs->io, pb, 1, file->buf); |
| if (retval) |
| goto fail; |
| } else { |
| file->physblock = 0; |
| memset(file->buf, 0, fs->blocksize); |
| } |
| |
| file->flags |= EXT2_FILE_BUF_VALID; |
| } |
| start = file->pos % fs->blocksize; |
| c = fs->blocksize - start; |
| if (c > wanted) |
| c = wanted; |
| left = file->inode.i_size - file->pos ; |
| if (c > left) |
| c = left; |
| |
| memcpy(ptr, file->buf+start, c); |
| file->pos += c; |
| ptr += c; |
| count += c; |
| wanted -= c; |
| |
| if (wanted > 0) |
| goto again; |
| |
| done: |
| if (got) |
| *got = count; |
| return 0; |
| |
| fail: |
| if (count) |
| goto done; |
| return retval; |
| } |
| |
| |
| errcode_t ext2fs_file_write(ext2_file_t file, void *buf, |
| unsigned int nbytes, unsigned int *written) |
| { |
| ext2_filsys fs; |
| errcode_t retval; |
| blk_t b, pb; |
| unsigned int start, c, count = 0; |
| char *ptr = (char *) buf; |
| |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| fs = file->fs; |
| |
| if (!(file->flags & EXT2_FILE_WRITE)) |
| return EXT2_ET_FILE_RO; |
| |
| again: |
| b = file->pos / fs->blocksize; |
| if (b != file->blockno) { |
| retval = ext2fs_file_flush(file); |
| if (retval) |
| goto fail; |
| file->flags &= ~EXT2_FILE_BUF_VALID; |
| } |
| file->blockno = b; |
| if (!(file->flags & EXT2_FILE_BUF_VALID)) { |
| retval = ext2fs_bmap(fs, file->ino, &file->inode, |
| file->buf, BMAP_ALLOC, b, &pb); |
| if (retval) |
| goto fail; |
| file->physblock = pb; |
| |
| retval = io_channel_read_blk(fs->io, pb, 1, file->buf); |
| if (retval) |
| goto fail; |
| file->flags |= EXT2_FILE_BUF_VALID; |
| } |
| start = file->pos % fs->blocksize; |
| c = fs->blocksize - start; |
| if (c > nbytes) |
| c = nbytes; |
| |
| file->flags |= EXT2_FILE_BUF_DIRTY; |
| memcpy(file->buf+start, ptr, c); |
| file->pos += c; |
| ptr += c; |
| count += c; |
| nbytes -= c; |
| |
| if (nbytes > 0) |
| goto again; |
| |
| done: |
| if (written) |
| *written = count; |
| return 0; |
| |
| fail: |
| if (count) |
| goto done; |
| return retval; |
| } |
| |
| |
| errcode_t ext2fs_file_lseek(ext2_file_t file, ext2_off_t offset, |
| int whence, ext2_off_t *ret_pos) |
| { |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| |
| if (whence == EXT2_SEEK_SET) |
| file->pos = offset; |
| else if (whence == EXT2_SEEK_CUR) |
| file->pos += offset; |
| else if (whence == EXT2_SEEK_END) |
| file->pos = file->inode.i_size + offset; |
| else |
| return EXT2_ET_INVALID_ARGUMENT; |
| |
| if (ret_pos) |
| *ret_pos = file->pos; |
| |
| return 0; |
| } |
| |
| /* |
| * This function returns the size of the file, according to the inode |
| */ |
| ext2_off_t ext2fs_file_get_size(ext2_file_t file) |
| { |
| if (file->magic != EXT2_ET_MAGIC_EXT2_FILE) |
| return 0; |
| return file->inode.i_size; |
| } |
| |
| /* |
| * This function sets the size of the file, truncating it if necessary |
| * |
| * XXX still need to call truncate |
| */ |
| errcode_t ext2fs_file_set_size(ext2_file_t file, ext2_off_t size) |
| { |
| errcode_t retval; |
| EXT2_CHECK_MAGIC(file, EXT2_ET_MAGIC_EXT2_FILE); |
| |
| file->inode.i_size = size; |
| retval = ext2fs_write_inode(file->fs, file->ino, &file->inode); |
| if (retval) |
| return retval; |
| |
| /* |
| * XXX truncate inode if necessary |
| */ |
| |
| return 0; |
| } |