diff options
Diffstat (limited to 'lib/libdisk/write_i386_disk.c')
| -rw-r--r-- | lib/libdisk/write_i386_disk.c | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/lib/libdisk/write_i386_disk.c b/lib/libdisk/write_i386_disk.c index e0cac80..15317d3 100644 --- a/lib/libdisk/write_i386_disk.c +++ b/lib/libdisk/write_i386_disk.c @@ -35,7 +35,8 @@ Write_FreeBSD(int fd, const struct disk *new, const struct chunk *c1) u_char buf[BBSIZE]; for (i = 0; i < BBSIZE/512; i++) { - p = read_block(fd, i + c1->offset, 512); + if (!(p = read_block(fd, i + c1->offset, 512))) + return (1); memcpy(buf + 512 * i, p, 512); free(p); } @@ -101,7 +102,10 @@ Write_Disk(const struct disk *d1) return 1; memset(s, 0, sizeof s); - mbrblk = read_block(fd, 0, d1->sector_size); + if (!(mbrblk = read_block(fd, 0, d1->sector_size))) { + close (fd); + return (1); + } dp = (struct dos_partition *)(mbrblk + DOSPARTOFF); memcpy(work, dp, sizeof work); dp = work; @@ -178,7 +182,10 @@ Write_Disk(const struct disk *d1) if (dp[i].dp_typ == 0xa5) dp[i].dp_flag = 0x80; - mbrblk = read_block(fd, 0, d1->sector_size); + if (!(mbrblk = read_block(fd, 0, d1->sector_size))) { + close (fd); + return (1); + } if (d1->bootmgr) { memcpy(mbrblk, d1->bootmgr, DOSPARTOFF); Cfg_Boot_Mgr(mbrblk, need_edd); |
