mirror of
https://github.com/holub/mame
synced 2025-05-29 17:13:05 +03:00
86f50b0d65
* Created a more flexible imgtool::datetime structure for use within Imgtool This is intended to replace most usage of time_t * Changing the granularity of imgtool_clock from 1ms to 100ns, as per Vas' suggestion * Created arbitrary_datetime in timeconv.h to facilitate interpretation of datetime info I concluded that invoking std::mktime on manually assembled std::tm is bad, because it is indeterminate how the std::tm members may be "dominant". This required that I go further in imgtool, and update a number of drivers and eliminate the parameter of imgtool::datetime that takes std::tm.
2417 lines
60 KiB
C++
2417 lines
60 KiB
C++
// license:GPL-2.0+
|
|
// copyright-holders:Dirk Best
|
|
/****************************************************************************
|
|
|
|
amiga.cpp
|
|
|
|
Amiga floppies
|
|
|
|
****************************************************************************/
|
|
|
|
|
|
/*****************************************************************************
|
|
Includes
|
|
*****************************************************************************/
|
|
|
|
|
|
#include <time.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
|
|
#include "imgtool.h"
|
|
#include "iflopimg.h"
|
|
#include "formats/imageutl.h"
|
|
|
|
|
|
|
|
/*****************************************************************************
|
|
Data structures
|
|
*****************************************************************************/
|
|
|
|
|
|
#define BSIZE (512) /* Block size is always 512 bytes for floppies */
|
|
#define TSIZE ((BSIZE/4) - 56) /* Size of data tables */
|
|
#define MSIZE ((BSIZE/4) - 1) /* Size of bitmaps */
|
|
|
|
|
|
enum disk_type
|
|
{
|
|
DT_UNKNOWN = -1,
|
|
DT_OFS = 0,
|
|
DT_FFS = 1,
|
|
DT_OFS_INTL = 2,
|
|
DT_FFS_INTL = 3,
|
|
DT_OFS_INTL_DIRC = 4,
|
|
DT_FFS_INTL_DIRC = 5
|
|
};
|
|
|
|
|
|
enum
|
|
{
|
|
T_INVALID = 0,
|
|
T_HEADER = 2,
|
|
T_DATA = 8,
|
|
T_LIST = 16,
|
|
T_DIRCACHE = 33
|
|
};
|
|
|
|
|
|
enum sec_type
|
|
{
|
|
ST_INVALID = 0,
|
|
ST_ROOT = 1,
|
|
ST_USERDIR = 2,
|
|
ST_FILE = -3,
|
|
ST_LINKFILE = -4,
|
|
ST_LINKDIR = 4,
|
|
ST_SOFTLINK = 3
|
|
};
|
|
|
|
|
|
struct amiga_date
|
|
{
|
|
uint32_t days; /* days since 1 jan 78 */
|
|
uint32_t mins; /* minutes past midnight */
|
|
uint32_t ticks; /* ticks (1/50 sec) past last minute */
|
|
};
|
|
|
|
|
|
struct root_block
|
|
{
|
|
uint32_t ht_size; /* Hash table size in long */
|
|
uint32_t chksum; /* Rootblock checksum */
|
|
uint32_t ht[TSIZE]; /* Hash table (entry block number) */
|
|
uint32_t bm_flag; /* bitmap flag, -1 means VALID */
|
|
uint32_t bm_pages[25]; /* bitmap blocks pointers (first one at bm_pages[0]) */
|
|
amiga_date r; /* last root alteration date */
|
|
uint8_t name_len; /* volume name length */
|
|
uint8_t diskname[30]; /* volume name */
|
|
amiga_date v; /* last disk alteration date */
|
|
amiga_date c; /* filesystem creation date */
|
|
uint32_t extension; /* FFS: first directory cache block, 0 otherwise */
|
|
};
|
|
|
|
|
|
struct bitmap_block
|
|
{
|
|
uint32_t chksum; /* checksum, normal algorithm */
|
|
uint32_t map[MSIZE]; /* bitmap */
|
|
};
|
|
|
|
|
|
struct bitmap_ext_block
|
|
{
|
|
uint32_t map[MSIZE]; /* bitmap */
|
|
uint32_t next; /* next extension block */
|
|
};
|
|
|
|
|
|
struct file_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t high_seq; /* number of data block ptr stored here */
|
|
uint32_t first_data; /* first data block ptr */
|
|
uint32_t chksum; /* same algorithm as rootblock */
|
|
uint32_t data_blocks[TSIZE]; /* data blk ptr */
|
|
uint16_t uid; /* UserID */
|
|
uint16_t gid; /* GroupID */
|
|
uint32_t protect; /* protection flags (0 by default) */
|
|
uint32_t byte_size; /* file size in bytes */
|
|
uint8_t comm_len; /* file comment length */
|
|
uint8_t comment[79]; /* comment (max. 79 chars permitted) */
|
|
amiga_date date; /* last change date */
|
|
uint8_t name_len; /* filename length */
|
|
uint8_t filename[30]; /* filename (max. 30 chars permitted) */
|
|
uint32_t real_entry; /* FFS: unused, set to 0 */
|
|
uint32_t next_link; /* FFS: hardlinks chained list (first == newest */
|
|
uint32_t hash_chain; /* next entry ptr with same hash */
|
|
uint32_t parent; /* parent directory */
|
|
uint32_t extension; /* pointer to 1st file extension block */
|
|
};
|
|
|
|
|
|
struct file_ext_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t high_seq; /* number of data block ptr stored here */
|
|
uint32_t chksum; /* same algorithm as rootblock */
|
|
uint32_t data_blocks[TSIZE]; /* data blk ptr */
|
|
uint32_t parent; /* file header block */
|
|
uint32_t extension; /* pointer to next file extension block */
|
|
};
|
|
|
|
|
|
struct data_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t seq_num; /* file data block number */
|
|
uint32_t data_size; /* data size */
|
|
uint32_t next_data; /* next data block ptr */
|
|
uint32_t chksum; /* checksum, rootblock algorithm */
|
|
uint8_t data[BSIZE-24]; /* file data */
|
|
};
|
|
|
|
|
|
struct dir_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t chksum; /* same algorithm as rootblock */
|
|
uint32_t ht[TSIZE]; /* hash table (entry block number) */
|
|
uint8_t uid; /* UserID */
|
|
uint8_t gid; /* GroupID */
|
|
uint32_t protect; /* protection flags (0 by default) */
|
|
uint8_t comm_len; /* file comment length */
|
|
uint8_t comment[79]; /* comment (max. 79 chars permitted) */
|
|
amiga_date date; /* last access date */
|
|
uint8_t name_len; /* directory name length */
|
|
uint8_t dirname[30]; /* directory name (max. 30 chars permitted) */
|
|
uint32_t next_link; /* FFS: hardlinks chained list (first == newest */
|
|
uint32_t hash_chain; /* next entry ptr with same hash */
|
|
uint32_t parent; /* parent directory */
|
|
uint32_t extension; /* FFS: first directory cache block */
|
|
};
|
|
|
|
|
|
struct hardlink_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t chksum; /* same algorithm as rootblock */
|
|
uint32_t protect; /* protection flags (0 by default) */
|
|
uint8_t comm_len; /* file comment length */
|
|
uint8_t comment[79]; /* comment (max. 79 chars permitted) */
|
|
amiga_date date; /* last access date */
|
|
uint8_t name_len; /* hard link name length */
|
|
uint8_t hlname[30]; /* hard link name (max. 30 chars permitted) */
|
|
uint32_t real_entry; /* FFS: pointer to "real" file or directory */
|
|
uint32_t next_link; /* FFS: hardlinks chained list (first == newest */
|
|
uint32_t hash_chain; /* next entry ptr with same hash */
|
|
uint32_t parent; /* parent directory */
|
|
uint32_t sec_type; /* secondary type, ST_LINKFILE/ST_LINKDIR */
|
|
};
|
|
|
|
|
|
struct softlink_block
|
|
{
|
|
uint32_t header_key; /* self pointer (to this block) */
|
|
uint32_t chksum; /* same algorithm as rootblock */
|
|
uint8_t symbolic_name[BSIZE-224]; /* path name to referenced object */
|
|
uint32_t protect; /* protection flags (0 by default) */
|
|
uint8_t comm_len; /* file comment length */
|
|
uint8_t comment[79]; /* comment (max. 79 chars permitted) */
|
|
amiga_date date; /* last access date */
|
|
uint8_t name_len; /* soft link name length */
|
|
uint8_t slname[30]; /* soft link name (max. 30 chars permitted) */
|
|
uint32_t hash_chain; /* next entry ptr with same hash */
|
|
uint32_t parent; /* parent directory */
|
|
};
|
|
|
|
|
|
/* Basic Amiga floppy disk image info */
|
|
struct amiga_floppy
|
|
{
|
|
imgtool::stream *stream;
|
|
uint8_t sectors;
|
|
};
|
|
|
|
|
|
/* iterator used to walk through directory entries */
|
|
struct amiga_iterator
|
|
{
|
|
unsigned int index; /* current file index */
|
|
int block; /* block number we are iterating */
|
|
uint32_t next_block; /* next block in hash chain */
|
|
int ht_index; /* current index in the hash table */
|
|
unsigned int eof : 1; /* end of file listing reached? */
|
|
};
|
|
|
|
|
|
/*****************************************************************************
|
|
Prototypes
|
|
*****************************************************************************/
|
|
|
|
|
|
static imgtoolerr_t amiga_image_read_sector(imgtool::image &img,
|
|
uint32_t track, uint32_t head, uint32_t sector, void *buf, size_t len);
|
|
static imgtoolerr_t amiga_image_read_sector(imgtool::image &img,
|
|
uint32_t track, uint32_t head, uint32_t sector, std::vector<uint8_t> &buffer);
|
|
static imgtoolerr_t amiga_image_write_sector(imgtool::image &img,
|
|
uint32_t track, uint32_t head, uint32_t sector, const void *buf, size_t len, int ddam);
|
|
|
|
|
|
|
|
/*****************************************************************************
|
|
Utility functions
|
|
*****************************************************************************/
|
|
|
|
|
|
/* Amiga version of the toupper function */
|
|
static int intl_toupper(int c)
|
|
{
|
|
return (c>='a' && c<='z') || (c>=224 && c<=254 && c!=247) ? c - ('a'-'A') : c ;
|
|
}
|
|
|
|
|
|
/* Amiga filename case insensitive string compare */
|
|
static int intl_stricmp(const char *s1, const char *s2)
|
|
{
|
|
for (;;)
|
|
{
|
|
int c1 = intl_toupper(*s1++);
|
|
int c2 = intl_toupper(*s2++);
|
|
|
|
if (c1 == 0 || c1 != c2)
|
|
return c1 - c2;
|
|
}
|
|
}
|
|
|
|
|
|
/* Calculate the hash value for a filename */
|
|
static int hash_name(const char *name, int intl)
|
|
{
|
|
int i, l = strlen(name);
|
|
uint32_t hash = l;
|
|
|
|
for(i = 0; i < l; i++)
|
|
{
|
|
hash *= 13;
|
|
hash += (uint8_t) (intl ? intl_toupper(name[i]) : toupper(name[i]));
|
|
hash &= 0x7ff;
|
|
}
|
|
|
|
return hash % TSIZE; /* 0 < hash < 71 in case of BSIZE=512 */
|
|
}
|
|
|
|
|
|
/* Returns true if year is a leap year */
|
|
static int is_leap(int year)
|
|
{
|
|
return ((year % 4 == 0) && (year % 100 != 0)) || (year % 400 == 0);
|
|
}
|
|
|
|
|
|
/* Convert amiga time to standard time */
|
|
static imgtool::datetime amiga_crack_time(amiga_date *date)
|
|
{
|
|
int month_days[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
|
|
int year = 1978, month = 1, year_days = 365; /* base date */
|
|
int day = date->days;
|
|
|
|
/* first calculate the year */
|
|
while (day >= year_days)
|
|
{
|
|
day -= year_days;
|
|
year_days = is_leap(++year) ? 366 : 365;
|
|
}
|
|
|
|
/* then the month */
|
|
while(day >= month_days[month-1])
|
|
{
|
|
day -= month_days[month-1];
|
|
if (month == 2 && is_leap(year))
|
|
day -= 1;
|
|
month++;
|
|
}
|
|
|
|
// fill the struct with our calculated values
|
|
util::arbitrary_datetime dt;
|
|
dt.year = year;
|
|
dt.month = month;
|
|
dt.day_of_month = day;
|
|
dt.hour = date->mins / 60;
|
|
dt.minute = date->mins % 60;
|
|
dt.second = date->ticks / 50;
|
|
|
|
return imgtool::datetime(imgtool::datetime::datetime_type::LOCAL, dt);
|
|
}
|
|
|
|
|
|
/* convert standard time to amiga time */
|
|
static void amiga_setup_time(time_t time, amiga_date *dest)
|
|
{
|
|
struct tm t = *localtime(&time);
|
|
int year;
|
|
|
|
dest->days = 0;
|
|
|
|
for (year = 1978; year < t.tm_year + 1900; year++)
|
|
{
|
|
dest->days += is_leap(year) ? 366 : 365;
|
|
}
|
|
|
|
dest->days += t.tm_yday;
|
|
dest->mins = t.tm_hour * 60 + t.tm_min;
|
|
dest->ticks = t.tm_sec * 50;
|
|
}
|
|
|
|
|
|
/* convert flags to human readable form */
|
|
static void amiga_decode_flags(uint32_t flags, char *dest)
|
|
{
|
|
/* test for flags */
|
|
dest[0] = (flags & 0x80) ? 'h' : '-';
|
|
dest[1] = (flags & 0x40) ? 's' : '-';
|
|
dest[2] = (flags & 0x20) ? 'p' : '-';
|
|
dest[3] = (flags & 0x10) ? 'a' : '-';
|
|
dest[4] = (flags & 0x08) ? '-' : 'r';
|
|
dest[5] = (flags & 0x04) ? '-' : 'w';
|
|
dest[6] = (flags & 0x02) ? '-' : 'e';
|
|
dest[7] = (flags & 0x01) ? '-' : 'd';
|
|
dest[8] = '\0';
|
|
}
|
|
|
|
|
|
static void copy_integer_array_be(uint32_t *dest, const uint32_t *source, int size)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < size; i++)
|
|
{
|
|
dest[i] = big_endianize_int32(source[i]);
|
|
}
|
|
}
|
|
|
|
|
|
/* This function converts an array of UINT32s to an amiga_date */
|
|
static void copy_date_be(amiga_date *dest, const uint32_t *source)
|
|
{
|
|
dest->days = big_endianize_int32(source[0]);
|
|
dest->mins = big_endianize_int32(source[1]);
|
|
dest->ticks = big_endianize_int32(source[2]);
|
|
}
|
|
|
|
|
|
/* Calculate the block checksum of a byte array */
|
|
static uint32_t block_checksum(uint8_t *buffer, int length)
|
|
{
|
|
uint32_t chksum = 0;
|
|
int i;
|
|
|
|
for (i = 0; i < length/4; i++)
|
|
{
|
|
chksum += pick_integer_be(buffer, i*4, 4);
|
|
}
|
|
|
|
return -chksum;
|
|
}
|
|
|
|
|
|
/* Get Amiga floppy data */
|
|
static amiga_floppy *get_amiga_floppy(imgtool::image &image)
|
|
{
|
|
return (amiga_floppy *)image.extra_bytes();
|
|
}
|
|
|
|
|
|
/* Returns the total number of blocks in the image */
|
|
static int get_total_blocks(imgtool::image &img)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
|
|
return 2 * 80 * f->sectors;
|
|
}
|
|
|
|
|
|
/* Returns track, head and sector for a block */
|
|
static void find_block(amiga_floppy *f, int block, int *track,
|
|
int *head, int *sector)
|
|
{
|
|
*track = block / f->sectors;
|
|
*head = (block - *track * f->sectors) / f->sectors;
|
|
*sector = (block - *track * f->sectors) % f->sectors;
|
|
}
|
|
|
|
|
|
/* Generic read block */
|
|
static imgtoolerr_t read_block(imgtool::image &img, int block, uint8_t *buffer)
|
|
{
|
|
imgtoolerr_t ret;
|
|
int track, head, sector;
|
|
|
|
find_block(get_amiga_floppy(img), block, &track, &head, §or);
|
|
|
|
/* get block from image */
|
|
ret = amiga_image_read_sector(img, track, head, sector, buffer, BSIZE);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Generic write block */
|
|
static imgtoolerr_t write_block(imgtool::image &img, int block, const uint8_t *buffer)
|
|
{
|
|
imgtoolerr_t ret;
|
|
int track, head, sector;
|
|
|
|
find_block(get_amiga_floppy(img), block, &track, &head, §or);
|
|
|
|
/* write block to image */
|
|
ret = amiga_image_write_sector(img, track, head, sector, buffer, BSIZE, 0);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Return the type a block */
|
|
static sec_type get_block_type(imgtool::image &img, int block)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* get data */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ST_INVALID;
|
|
|
|
/* return type */
|
|
switch ((int32_t) pick_integer_be(buffer, BSIZE-4, 4))
|
|
{
|
|
case 1: return ST_ROOT;
|
|
case 2: return ST_USERDIR;
|
|
case -3: return ST_FILE;
|
|
case -4: return ST_LINKFILE;
|
|
case 4: return ST_LINKDIR;
|
|
case 3: return ST_SOFTLINK;
|
|
default: return ST_INVALID;
|
|
}
|
|
}
|
|
|
|
|
|
/* Read a bitmap block */
|
|
static imgtoolerr_t read_bitmap_block(imgtool::image &img, int block, bitmap_block *bm)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
bm->chksum = pick_integer_be(buffer, 0, 4);
|
|
copy_integer_array_be(bm->map, (uint32_t *) &buffer[4], MSIZE);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Write a bitmap block */
|
|
static imgtoolerr_t write_bitmap_block(imgtool::image &img, int block, const bitmap_block *bm)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* Setup buffer */
|
|
place_integer_be(buffer, 0, 4, bm->chksum);
|
|
copy_integer_array_be((uint32_t *) &buffer[4], bm->map, MSIZE);
|
|
|
|
/* write block */
|
|
ret = write_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
#ifdef UNUSED_FUNCTION
|
|
/* Read a bitmap extended block */
|
|
static imgtoolerr_t read_bitmap_ext_block(imgtool::image *img, int block, bitmap_ext_block *bm)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
copy_integer_array_be(bm->map, (uint32_t *) &buffer, MSIZE);
|
|
bm->next = pick_integer_be(buffer, BSIZE-4, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
#endif
|
|
|
|
|
|
/* Read the root block */
|
|
static imgtoolerr_t read_root_block(imgtool::image &img, root_block *root)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* get raw root block from image */
|
|
ret = read_block(img, get_total_blocks(img)/2, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* copy data to root_block */
|
|
memset(root, 0, sizeof(root_block));
|
|
|
|
root->ht_size = pick_integer_be(buffer, 12, 4);
|
|
root->chksum = pick_integer_be(buffer, 20, 4);
|
|
copy_integer_array_be(root->ht, (uint32_t *) &buffer[24], TSIZE);
|
|
root->bm_flag = pick_integer_be(buffer, BSIZE-200, 4);
|
|
copy_integer_array_be(root->bm_pages, (uint32_t *) &buffer[BSIZE-196], 25);
|
|
copy_date_be(&root->r, (uint32_t *) &buffer[BSIZE-92]);
|
|
root->name_len = pick_integer_be(buffer, BSIZE-80, 1);
|
|
memcpy(root->diskname, &buffer[BSIZE-79], 30);
|
|
copy_date_be(&root->v, (uint32_t *) &buffer[BSIZE-40]);
|
|
copy_date_be(&root->c, (uint32_t *) &buffer[BSIZE-28]);
|
|
root->extension = pick_integer_be(buffer, BSIZE-8, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t write_root_block(imgtool::image &img, const root_block *root)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* Setup buffer */
|
|
memset(buffer, 0, BSIZE);
|
|
|
|
place_integer_be(buffer, 0, 4, T_HEADER);
|
|
place_integer_be(buffer, 12, 4, root->ht_size);
|
|
place_integer_be(buffer, 20, 4, root->chksum);
|
|
copy_integer_array_be((uint32_t *) &buffer[24], root->ht, TSIZE);
|
|
place_integer_be(buffer, BSIZE-200, 4, root->bm_flag);
|
|
copy_integer_array_be((uint32_t *) &buffer[BSIZE-196], root->bm_pages, 25);
|
|
place_integer_be(buffer, BSIZE-92, 4, root->r.days);
|
|
place_integer_be(buffer, BSIZE-88, 4, root->r.mins);
|
|
place_integer_be(buffer, BSIZE-84, 4, root->r.ticks);
|
|
place_integer_be(buffer, BSIZE-80, 1, root->name_len);
|
|
memcpy(&buffer[BSIZE-79], root->diskname, root->name_len);
|
|
place_integer_be(buffer, BSIZE-40, 4, root->v.days);
|
|
place_integer_be(buffer, BSIZE-36, 4, root->v.mins);
|
|
place_integer_be(buffer, BSIZE-32, 4, root->v.ticks);
|
|
place_integer_be(buffer, BSIZE-28, 4, root->c.days);
|
|
place_integer_be(buffer, BSIZE-24, 4, root->c.mins);
|
|
place_integer_be(buffer, BSIZE-20, 4, root->c.ticks);
|
|
place_integer_be(buffer, BSIZE-8, 4, root->extension);
|
|
place_integer_be(buffer, BSIZE-4, 4, ST_ROOT);
|
|
|
|
/* write root block to image */
|
|
ret = write_block(img, get_total_blocks(img)/2, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Read a file block */
|
|
static imgtoolerr_t read_file_block(imgtool::image &img, int block, file_block *fb)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
fb->header_key = pick_integer_be(buffer, 4, 4);
|
|
fb->high_seq = pick_integer_be(buffer, 8, 4);
|
|
fb->first_data = pick_integer_be(buffer, 16, 4);
|
|
fb->chksum = pick_integer_be(buffer, 20, 4);
|
|
copy_integer_array_be(fb->data_blocks, (uint32_t *) &buffer[24], TSIZE);
|
|
fb->uid = pick_integer_be(buffer, BSIZE-196, 2);
|
|
fb->gid = pick_integer_be(buffer, BSIZE-194, 2);
|
|
fb->protect = pick_integer_be(buffer, BSIZE-192, 4);
|
|
fb->byte_size = pick_integer_be(buffer, BSIZE-188, 4);
|
|
fb->comm_len = pick_integer_be(buffer, BSIZE-184, 1);
|
|
memcpy(fb->comment, &buffer[BSIZE-183], 79);
|
|
copy_date_be(&fb->date, (uint32_t *) &buffer[BSIZE-92]);
|
|
fb->name_len = pick_integer_be(buffer, BSIZE-80, 1);
|
|
memcpy(fb->filename, (uint32_t *) &buffer[BSIZE-79], 30);
|
|
fb->real_entry = pick_integer_be(buffer, BSIZE-44, 4);
|
|
fb->next_link = pick_integer_be(buffer, BSIZE-40, 4);
|
|
fb->hash_chain = pick_integer_be(buffer, BSIZE-16, 4);
|
|
fb->parent = pick_integer_be(buffer, BSIZE-12, 4);
|
|
fb->extension = pick_integer_be(buffer, BSIZE-8, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t read_file_ext_block(imgtool::image &img, int block, file_ext_block *fe)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
fe->header_key = pick_integer_be(buffer, 4, 4);
|
|
fe->high_seq = pick_integer_be(buffer, 8, 4);
|
|
fe->chksum = pick_integer_be(buffer, 20, 4);
|
|
copy_integer_array_be(fe->data_blocks, (uint32_t *) &buffer[24], TSIZE);
|
|
fe->parent = pick_integer_be(buffer, BSIZE-12, 4);
|
|
fe->extension = pick_integer_be(buffer, BSIZE-8, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t read_data_block(imgtool::image &img, int block, data_block *d)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
d->header_key = pick_integer_be(buffer, 4, 4);
|
|
d->seq_num = pick_integer_be(buffer, 8, 4);
|
|
d->data_size = pick_integer_be(buffer, 12, 4);
|
|
d->next_data = pick_integer_be(buffer, 16, 4);
|
|
d->chksum = pick_integer_be(buffer, 20, 4);
|
|
memcpy(d->data, &buffer[24], BSIZE-24);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Read a directory block */
|
|
static imgtoolerr_t read_dir_block(imgtool::image &img, int block, dir_block *db)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
db->header_key = pick_integer_be(buffer, 4, 4);
|
|
db->chksum = pick_integer_be(buffer, 20, 4);
|
|
copy_integer_array_be(db->ht, (uint32_t *) &buffer[24], TSIZE);
|
|
db->uid = pick_integer_be(buffer, BSIZE-196, 2);
|
|
db->gid = pick_integer_be(buffer, BSIZE-194, 2);
|
|
db->protect = pick_integer_be(buffer, BSIZE-192, 4);
|
|
db->comm_len = pick_integer_be(buffer, BSIZE-184, 1);
|
|
memcpy(db->comment, &buffer[BSIZE-183], 79);
|
|
copy_date_be(&db->date, (uint32_t *) &buffer[BSIZE-92]);
|
|
db->name_len = pick_integer_be(buffer, BSIZE-80, 1);
|
|
memcpy(db->dirname, (uint32_t *) &buffer[BSIZE-79], 30);
|
|
db->next_link = pick_integer_be(buffer, BSIZE-40, 4);
|
|
db->hash_chain = pick_integer_be(buffer, BSIZE-16, 4);
|
|
db->parent = pick_integer_be(buffer, BSIZE-12, 4);
|
|
db->extension = pick_integer_be(buffer, BSIZE-8, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t write_dir_block(imgtool::image &img, int block, const dir_block *db)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* Setup buffer */
|
|
memset(buffer, 0, BSIZE);
|
|
|
|
/* Copy data */
|
|
place_integer_be(buffer, 0, 4, T_HEADER);
|
|
place_integer_be(buffer, 4, 4, db->header_key);
|
|
place_integer_be(buffer, 20, 4, db->chksum);
|
|
copy_integer_array_be((uint32_t *) &buffer[24], db->ht, TSIZE);
|
|
place_integer_be(buffer, BSIZE-196, 2, db->uid);
|
|
place_integer_be(buffer, BSIZE-194, 2, db->gid);
|
|
place_integer_be(buffer, BSIZE-192, 4, db->protect);
|
|
place_integer_be(buffer, BSIZE-184, 1, db->comm_len);
|
|
memcpy((uint32_t *) &buffer[BSIZE-183], db->comment, db->comm_len);
|
|
place_integer_be(buffer, BSIZE-92, 4, db->date.days);
|
|
place_integer_be(buffer, BSIZE-88, 4, db->date.mins);
|
|
place_integer_be(buffer, BSIZE-84, 4, db->date.ticks);
|
|
place_integer_be(buffer, BSIZE-80, 1, db->name_len);
|
|
memcpy((uint32_t *) &buffer[BSIZE-79], db->dirname, db->name_len);
|
|
place_integer_be(buffer, BSIZE-40, 4, db->next_link);
|
|
place_integer_be(buffer, BSIZE-16, 4, db->hash_chain);
|
|
place_integer_be(buffer, BSIZE-12, 4, db->parent);
|
|
place_integer_be(buffer, BSIZE-8, 4, db->extension);
|
|
place_integer_be(buffer, BSIZE-4, 4, ST_USERDIR);
|
|
|
|
/* Write block to disk */
|
|
return write_block(img, block, buffer);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t read_hardlink_block(imgtool::image &img, int block, hardlink_block *hl)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
hl->header_key = pick_integer_be(buffer, 4, 4);
|
|
hl->chksum = pick_integer_be(buffer, 20, 4);
|
|
hl->protect = pick_integer_be(buffer, BSIZE-192, 4);
|
|
hl->comm_len = pick_integer_be(buffer, BSIZE-184, 1);
|
|
memcpy(hl->comment, &buffer[BSIZE-183], 79);
|
|
copy_date_be(&hl->date, (uint32_t *) &buffer[BSIZE-92]);
|
|
hl->name_len = pick_integer_be(buffer, BSIZE-80, 1);
|
|
memcpy(hl->hlname, (uint32_t *) &buffer[BSIZE-79], 30);
|
|
hl->real_entry = pick_integer_be(buffer, BSIZE-44, 4);
|
|
hl->next_link = pick_integer_be(buffer, BSIZE-40, 4);
|
|
hl->hash_chain = pick_integer_be(buffer, BSIZE-16, 4);
|
|
hl->parent = pick_integer_be(buffer, BSIZE-12, 4);
|
|
hl->sec_type = pick_integer_be(buffer, BSIZE-4, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t read_softlink_block(imgtool::image &img, int block, softlink_block *sl)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* fill in data */
|
|
sl->header_key = pick_integer_be(buffer, 4, 4);
|
|
sl->chksum = pick_integer_be(buffer, 20, 4);
|
|
memcpy(sl->symbolic_name, &buffer[24], BSIZE-224);
|
|
sl->protect = pick_integer_be(buffer, BSIZE-192, 4);
|
|
sl->comm_len = pick_integer_be(buffer, BSIZE-184, 1);
|
|
memcpy(sl->comment, &buffer[BSIZE-183], 79);
|
|
copy_date_be(&sl->date, (uint32_t *) &buffer[BSIZE-92]);
|
|
sl->name_len = pick_integer_be(buffer, BSIZE-80, 1);
|
|
memcpy(sl->slname, (uint32_t *) &buffer[BSIZE-79], 30);
|
|
sl->hash_chain = pick_integer_be(buffer, BSIZE-16, 4);
|
|
sl->parent = pick_integer_be(buffer, BSIZE-12, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Returns the disk type */
|
|
static disk_type get_disk_type(imgtool::image &img)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
ret = read_block(img, 0, buffer);
|
|
if (ret) return DT_UNKNOWN;
|
|
|
|
switch(buffer[3])
|
|
{
|
|
case 0: return DT_OFS;
|
|
case 1: return DT_FFS;
|
|
case 2: return DT_OFS_INTL;
|
|
case 3: return DT_FFS_INTL;
|
|
case 4: return DT_OFS_INTL_DIRC;
|
|
case 5: return DT_FFS_INTL_DIRC;
|
|
default: return DT_UNKNOWN;
|
|
}
|
|
}
|
|
|
|
|
|
/* Returns true if the disk is formatted with the FastFileSystem */
|
|
static int is_ffs(imgtool::image &img)
|
|
{
|
|
disk_type t = get_disk_type(img);
|
|
|
|
return ((t == DT_FFS ||
|
|
t == DT_FFS_INTL ||
|
|
t == DT_FFS_INTL_DIRC) ? true : false);
|
|
}
|
|
|
|
|
|
/* Returns true if the disk uses the international mode */
|
|
static int is_intl(imgtool::image &img)
|
|
{
|
|
disk_type t = get_disk_type(img);
|
|
|
|
return ((t == DT_OFS_INTL ||
|
|
t == DT_FFS_INTL ||
|
|
t == DT_OFS_INTL_DIRC ||
|
|
t == DT_FFS_INTL_DIRC) ? true : false);
|
|
}
|
|
|
|
#ifdef UNUSED_FUNCTION
|
|
/* Returns true if the disk uses the directory cache mode */
|
|
static int is_dirc(imgtool::image *img)
|
|
{
|
|
disk_type t = get_disk_type(img);
|
|
|
|
return ((t == DT_OFS_INTL_DIRC ||
|
|
t == DT_FFS_INTL_DIRC) ? true : false);
|
|
}
|
|
#endif
|
|
|
|
static imgtoolerr_t get_hash_table(imgtool::image &img, int block, uint32_t *ht)
|
|
{
|
|
imgtoolerr_t ret;
|
|
|
|
switch (get_block_type(img, block))
|
|
{
|
|
case ST_USERDIR:
|
|
{
|
|
dir_block dir;
|
|
|
|
/* get the directory block */
|
|
ret = read_dir_block(img, block, &dir);
|
|
if (ret) return ret;
|
|
|
|
/* copy data */
|
|
memcpy(ht, &dir.ht, TSIZE*4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
case ST_ROOT:
|
|
{
|
|
root_block root;
|
|
|
|
/* get the root block */
|
|
ret = read_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* copy data */
|
|
memcpy(ht, &root.ht, TSIZE*4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
default:
|
|
return IMGTOOLERR_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
|
|
static imgtoolerr_t set_hash_table(imgtool::image &img, int block, const uint32_t *ht)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
imgtoolerr_t ret;
|
|
|
|
/* Read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* Copy new hash table into it */
|
|
copy_integer_array_be((uint32_t *) &buffer[24], ht, TSIZE);
|
|
|
|
/* Write it back again */
|
|
ret = write_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
#ifdef UNUSED_FUNCTION
|
|
static imgtoolerr_t get_root_hash_table(imgtool::image *img, uint32_t *ht)
|
|
{
|
|
return get_hash_table(img, get_total_blocks(img)/2, ht);
|
|
}
|
|
#endif
|
|
|
|
static imgtoolerr_t get_blockname(imgtool::image &img, int block, char *dest)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
imgtoolerr_t ret;
|
|
|
|
/* Read the block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* Copy filename out of the buffer */
|
|
memset(dest, 0, 31);
|
|
memcpy(dest, &buffer[BSIZE-79], buffer[BSIZE-80]);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t get_hash_chain(imgtool::image &img, int block, uint32_t *chain)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
imgtoolerr_t ret;
|
|
|
|
/* Read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* Get chain value */
|
|
*chain = pick_integer_be(buffer, BSIZE-16, 4);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t set_hash_chain(imgtool::image &img, int block, uint32_t chain)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
imgtoolerr_t ret;
|
|
|
|
/* Read block */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* Copy new hash chain value into it */
|
|
place_integer_be(buffer, BSIZE-16, 4, chain);
|
|
|
|
/* Write it back again */
|
|
ret = write_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t walk_hash_chain(imgtool::image &img, const char *path, int start_block, int *prev_block, int *block)
|
|
{
|
|
imgtoolerr_t err;
|
|
uint32_t hash_chain;
|
|
char name[31];
|
|
|
|
/* choose compare function depending on intl mode */
|
|
int (*cmp)(const char *, const char *) = is_intl(img) ? &intl_stricmp : &core_stricmp;
|
|
|
|
/* initialize filenames */
|
|
memset(name, 0, sizeof(name));
|
|
|
|
switch (get_block_type(img, start_block))
|
|
{
|
|
case ST_USERDIR:
|
|
{
|
|
dir_block dir;
|
|
|
|
/* read block */
|
|
err = read_dir_block(img, start_block, &dir);
|
|
if (err) return err;
|
|
|
|
/* copy filename string and next hash */
|
|
memcpy(name, dir.dirname, dir.name_len);
|
|
hash_chain = dir.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_FILE:
|
|
{
|
|
file_block file;
|
|
|
|
/* read block */
|
|
err = read_file_block(img, start_block, &file);
|
|
if (err) return err;
|
|
|
|
/* copy filename string and next hash */
|
|
memcpy(name, file.filename, file.name_len);
|
|
hash_chain = file.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_SOFTLINK:
|
|
{
|
|
softlink_block sl;
|
|
|
|
/* read block */
|
|
err = read_softlink_block(img, start_block, &sl);
|
|
if (err) return err;
|
|
|
|
/* copy filename string and next hash */
|
|
memcpy(name, sl.slname, sl.name_len);
|
|
hash_chain = sl.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_LINKDIR:
|
|
case ST_LINKFILE:
|
|
{
|
|
hardlink_block hl;
|
|
|
|
/* read block */
|
|
err = read_hardlink_block(img, start_block, &hl);
|
|
if (err) return err;
|
|
|
|
/* copy filename string and next hash */
|
|
memcpy(name, hl.hlname, hl.name_len);
|
|
hash_chain = hl.hash_chain;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
return IMGTOOLERR_UNEXPECTED;
|
|
|
|
}
|
|
|
|
/* if we haven't found the right filename but there are linked entries,
|
|
* walk up the chain */
|
|
if ((*cmp)(name, path) != 0 && hash_chain != 0)
|
|
{
|
|
*prev_block = start_block;
|
|
return walk_hash_chain(img, path, hash_chain, prev_block, block);
|
|
}
|
|
|
|
/* found the correct block, return */
|
|
if ((*cmp)(name, path) == 0)
|
|
{
|
|
*block = start_block;
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
/* we should never get here */
|
|
return IMGTOOLERR_UNEXPECTED;
|
|
}
|
|
|
|
|
|
/* Returns the block number for a dir/file/link entry given as NUL delimited
|
|
* list of path parts, for example "dir1\0dir2\0dir3" returns the block number
|
|
* for directory "dir3" */
|
|
static imgtoolerr_t find_entry(imgtool::image &img, const char *path, int start_block, int *block)
|
|
{
|
|
imgtoolerr_t ret;
|
|
const char *next_path;
|
|
int current_block, prev;
|
|
uint32_t ht[TSIZE];
|
|
|
|
/* get the hash table */
|
|
ret = get_hash_table(img, start_block, ht);
|
|
if (ret) return ret;
|
|
|
|
/* calculate hash and get block for initial entry */
|
|
current_block = ht[hash_name(path, is_intl(img))];
|
|
|
|
/* check if there was a match in the hash table */
|
|
if (current_block == 0)
|
|
{
|
|
return IMGTOOLERR_PATHNOTFOUND;
|
|
}
|
|
|
|
/* walk the linked hash list */
|
|
ret = walk_hash_chain(img, path, current_block, &prev, block);
|
|
if (ret) return ret;
|
|
|
|
/* follow links */
|
|
switch (get_block_type(img, *block))
|
|
{
|
|
case ST_SOFTLINK:
|
|
|
|
/* TODO: Softlink support */
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
|
|
case ST_LINKDIR:
|
|
case ST_LINKFILE:
|
|
{
|
|
hardlink_block hl;
|
|
|
|
ret = read_hardlink_block(img, *block, &hl);
|
|
if (ret) return ret;
|
|
|
|
*block = hl.real_entry;
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
/* get next path part */
|
|
next_path = path + strlen(path) + 1;
|
|
|
|
/* if there are more path parts, search the next block */
|
|
if (next_path[0])
|
|
{
|
|
return find_entry(img, next_path, *block, block);
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t get_block_chksum(imgtool::image &img, int block, uint32_t *chksum, int bitmap)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* get block data */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* clear old checksum */
|
|
if (bitmap)
|
|
{
|
|
memset(buffer, 0, 4);
|
|
}
|
|
else
|
|
{
|
|
memset(&buffer[20], 0, 4);
|
|
}
|
|
|
|
/* calulate checksum */
|
|
*chksum = block_checksum(buffer, BSIZE);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t fix_chksum(imgtool::image &img, int block, int bitmap)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
uint32_t chksum;
|
|
|
|
/* calculate block checksum */
|
|
ret = get_block_chksum(img, block, &chksum, bitmap);
|
|
if (ret) return ret;
|
|
|
|
/* read block data */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* update checksum */
|
|
if (bitmap)
|
|
{
|
|
place_integer_be(buffer, 0, 4, chksum);
|
|
}
|
|
else
|
|
{
|
|
place_integer_be(buffer, 20, 4, chksum);
|
|
}
|
|
|
|
/* write back new block data */
|
|
ret = write_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t fix_block_chksum(imgtool::image &img, int block)
|
|
{
|
|
return fix_chksum(img, block, false);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t fix_bitmap_chksum(imgtool::image &img, int block)
|
|
{
|
|
return fix_chksum(img, block, true);
|
|
}
|
|
|
|
|
|
/* Set a block as used */
|
|
static imgtoolerr_t bitmap_mark(imgtool::image &img, int block, int used)
|
|
{
|
|
imgtoolerr_t ret;
|
|
bitmap_block bm;
|
|
root_block root;
|
|
int page;
|
|
|
|
block -= 2; /* subtract boot block sectors, 2 only for floppies! */
|
|
|
|
ret = read_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* figure out bitmap block location */
|
|
page = root.bm_pages[block / (MSIZE * 32)];
|
|
|
|
/* get bitmap */
|
|
ret = read_bitmap_block(img, page, &bm);
|
|
if (ret) return ret;
|
|
|
|
/* subtract pages we skip */
|
|
block -= MSIZE * 32 * (block / (MSIZE * 32));
|
|
|
|
/* mark as used or free */
|
|
if (used)
|
|
{
|
|
bm.map[block/32] &= ~(1 << block % 32);
|
|
}
|
|
else
|
|
{
|
|
bm.map[block/32] |= (1 << block % 32);
|
|
}
|
|
|
|
/* write changed bitmap block back to disk */
|
|
ret = write_bitmap_block(img, page, &bm);
|
|
if (ret) return ret;
|
|
|
|
/* update checksum */
|
|
ret = fix_bitmap_chksum(img, page);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t bitmap_mark_used(imgtool::image &img, int block)
|
|
{
|
|
return bitmap_mark(img, block, true);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t bitmap_mark_free(imgtool::image &img, int block)
|
|
{
|
|
return bitmap_mark(img, block, false);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t update_block_modified_date(imgtool::image &img, int block)
|
|
{
|
|
uint8_t buffer[BSIZE];
|
|
imgtoolerr_t ret;
|
|
amiga_date date;
|
|
time_t now;
|
|
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* Set new time */
|
|
time(&now);
|
|
amiga_setup_time(now, &date);
|
|
|
|
/* Write new time into block */
|
|
place_integer_be(buffer, BSIZE-92, 4, date.days);
|
|
place_integer_be(buffer, BSIZE-88, 4, date.mins);
|
|
place_integer_be(buffer, BSIZE-84, 4, date.ticks);
|
|
|
|
/* Write block back to disk */
|
|
ret = write_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t clear_hash_table_entry(imgtool::image &img, int parent, char *filename)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint32_t ht[TSIZE], chain;
|
|
int index, entry, prev, block;
|
|
|
|
ret = get_hash_table(img, parent, ht);
|
|
if (ret) return ret;
|
|
|
|
/* Calculate hash and get block for initial entry */
|
|
index = hash_name(filename, is_intl(img));
|
|
entry = ht[index];
|
|
|
|
/* Walk the hash chain to get the real entry */
|
|
ret = walk_hash_chain(img, filename, entry, &prev, &block);
|
|
if (ret) return ret;
|
|
|
|
/* Get chained value from block */
|
|
ret = get_hash_chain(img, block, &chain);
|
|
if (ret) return ret;
|
|
|
|
/* Check if we need to change the hash table */
|
|
if (entry == block)
|
|
{
|
|
/* Set new value (might be 0 if there were no linked entries) */
|
|
ht[index] = chain;
|
|
|
|
/* Save changed hash table */
|
|
ret = set_hash_table(img, parent, ht);
|
|
if (ret) return ret;
|
|
|
|
/* Update last modified date */
|
|
ret = update_block_modified_date(img, parent);
|
|
if (ret) return ret;
|
|
|
|
/* Calculate new checksum */
|
|
ret = fix_block_chksum(img, parent);
|
|
if (ret) return ret;
|
|
}
|
|
else
|
|
{
|
|
/* Save chained value to previous chain element */
|
|
ret = set_hash_chain(img, prev, chain);
|
|
if (ret) return ret;
|
|
|
|
/* Calculate new checksum */
|
|
ret = fix_block_chksum(img, prev);
|
|
if (ret) return ret;
|
|
}
|
|
|
|
/* Mark our block as free */
|
|
ret = bitmap_mark_free(img, block);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Returns the number of the first bit that is set in the array */
|
|
static int get_first_bit(uint32_t *array, int size)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < size; i++)
|
|
{
|
|
if (array[i] != 0)
|
|
{
|
|
uint32_t v = array[i];
|
|
int c;
|
|
|
|
/* get first bit that is set */
|
|
for (c = 0; (v & 1) == 0; c++) v >>= 1;
|
|
|
|
/* return bit number */
|
|
return i * 32 + c;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
#ifdef UNUSED_FUNCTION
|
|
static imgtoolerr_t walk_bitmap_ext_blocks(imgtool::image *img, int start, int *block)
|
|
{
|
|
imgtoolerr_t ret;
|
|
bitmap_ext_block bm_ext;
|
|
int bit;
|
|
|
|
/* if we don't have a valid block, bail out */
|
|
if (start == 0)
|
|
{
|
|
return IMGTOOLERR_NOSPACE;
|
|
}
|
|
|
|
/* get the extended bitmap block */
|
|
ret = read_bitmap_ext_block(img, start, &bm_ext);
|
|
if (ret) return ret;
|
|
|
|
/* get the first bit that is set in the map */
|
|
bit = get_first_bit(bm_ext.map, MSIZE);
|
|
|
|
/* if we found one, return */
|
|
if (bit != -1)
|
|
{
|
|
*block += bit;
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
/* increase on each call */
|
|
*block += MSIZE * 32;
|
|
|
|
/* else continue walking the list */
|
|
return walk_bitmap_ext_blocks(img, bm_ext.next, block);
|
|
}
|
|
#endif
|
|
|
|
|
|
/* Searches for a block marked as free
|
|
* TODO: bm_ext support for HDs */
|
|
static imgtoolerr_t find_free_block(imgtool::image &img, int *block)
|
|
{
|
|
imgtoolerr_t ret;
|
|
root_block root;
|
|
int i;
|
|
|
|
/* get root block */
|
|
ret = read_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* iterate bitmap pointers */
|
|
for (i = 0; i < 25; i++)
|
|
{
|
|
bitmap_block bm;
|
|
|
|
/* get bitmap block pointed to */
|
|
ret = read_bitmap_block(img, root.bm_pages[i], &bm);
|
|
if (ret) return ret;
|
|
|
|
*block = i * 32 * MSIZE + get_first_bit(bm.map, MSIZE);
|
|
|
|
if (*block != -1)
|
|
{
|
|
*block += 2;
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
/* if we get here we haven't found a free block */
|
|
return IMGTOOLERR_NOSPACE;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t add_entry(imgtool::image &img, int parent, int block)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint32_t ht[TSIZE];
|
|
char name[31];
|
|
int hash;
|
|
|
|
ret = get_blockname(img, block, name);
|
|
if (ret) return ret;
|
|
|
|
ret = get_hash_table(img, parent, ht);
|
|
if (ret) return ret;
|
|
|
|
hash = hash_name(name, is_intl(img));
|
|
|
|
/* Check if there is already an entry with that name */
|
|
if (ht[hash] != 0)
|
|
{
|
|
/* Save the old value to our hash chain */
|
|
ret = set_hash_chain(img, block, ht[hash]);
|
|
if (ret) return ret;
|
|
}
|
|
|
|
/* Write our block number into the table */
|
|
ht[hash] = block;
|
|
|
|
/* Write table back to disk */
|
|
ret = set_hash_table(img, parent, ht);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* Recursively create new directory entries */
|
|
static imgtoolerr_t makedir(imgtool::image &img, const char *path, int parent)
|
|
{
|
|
imgtoolerr_t ret;
|
|
dir_block dir;
|
|
time_t now;
|
|
int block;
|
|
|
|
if (!path)
|
|
{
|
|
return IMGTOOLERR_PARAMNEEDED;
|
|
}
|
|
|
|
if (!path[0])
|
|
{
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
/* Get a free block */
|
|
ret = find_free_block(img, &block);
|
|
if (ret) return ret;
|
|
|
|
/* Initialize entry */
|
|
memset(&dir, 0, sizeof(dir_block));
|
|
|
|
/* Copy data */
|
|
time(&now);
|
|
amiga_setup_time(now, &dir.date);
|
|
dir.name_len = strlen(path);
|
|
memcpy(dir.dirname, path, dir.name_len);
|
|
dir.header_key = block;
|
|
dir.parent = parent;
|
|
|
|
/* Write block */
|
|
ret = write_dir_block(img, block, &dir);
|
|
if (ret) return ret;
|
|
|
|
/* Fix checksum */
|
|
ret = fix_block_chksum(img, block);
|
|
if (ret) return ret;
|
|
|
|
/* Link the new entry in the parent */
|
|
ret = add_entry(img, parent, block);
|
|
if (ret) return ret;
|
|
|
|
/* Mark it as used */
|
|
ret = bitmap_mark_used(img, block);
|
|
if (ret) return ret;
|
|
|
|
/* Create the next entry */
|
|
return makedir(img, path + strlen(path) + 1, block);
|
|
}
|
|
|
|
|
|
/* Recursively checks the path parts and creates directories for them */
|
|
static imgtoolerr_t checkdir(imgtool::image &img, const char *path, int parent)
|
|
{
|
|
imgtoolerr_t ret;
|
|
int block;
|
|
char first_part[31];
|
|
|
|
memset(first_part, 0, sizeof(first_part));
|
|
strcpy(first_part, path);
|
|
|
|
/* Directories all the way down, bail out */
|
|
if (!path[0])
|
|
{
|
|
return IMGTOOLERR_CANNOTUSEPATH;
|
|
}
|
|
|
|
/* Search for the entry */
|
|
ret = find_entry(img, first_part, parent, &block);
|
|
|
|
switch (ret)
|
|
{
|
|
case IMGTOOLERR_PATHNOTFOUND:
|
|
|
|
/* There is no entry with this name yet, so we can just create them */
|
|
return makedir(img, path, parent);
|
|
|
|
case IMGTOOLERR_SUCCESS:
|
|
|
|
if (get_block_type(img, block) == ST_USERDIR)
|
|
{
|
|
/* Go down one level */
|
|
return checkdir(img, path + strlen(path) + 1, block);
|
|
}
|
|
else
|
|
{
|
|
/* There is an entry but it's not a directory, create it */
|
|
return makedir(img, path, parent);
|
|
}
|
|
|
|
default: return ret;
|
|
}
|
|
}
|
|
|
|
|
|
/* Writes the file data from the specified block into the stream */
|
|
static imgtoolerr_t write_file_block_data(imgtool::image &img, int block, int size, imgtool::stream &destf)
|
|
{
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
|
|
/* Check if we even need to write something */
|
|
if (size == 0)
|
|
{
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
if (is_ffs(img))
|
|
{
|
|
/* Get block and read directly into buffer */
|
|
ret = read_block(img, block, buffer);
|
|
if (ret) return ret;
|
|
}
|
|
else
|
|
{
|
|
data_block db;
|
|
uint32_t chksum;
|
|
|
|
ret = read_data_block(img, block, &db);
|
|
if (ret) return ret;
|
|
|
|
/* Verify data checksum */
|
|
ret = get_block_chksum(img, block, &chksum, false);
|
|
if (ret) return ret;
|
|
|
|
if (db.chksum != chksum)
|
|
{
|
|
return IMGTOOLERR_CORRUPTFILE;
|
|
}
|
|
|
|
/* Copy data to buffer */
|
|
memcpy(buffer, db.data, size);
|
|
}
|
|
|
|
/* Write data to stream */
|
|
if (destf.write(buffer, size) != size)
|
|
{
|
|
return IMGTOOLERR_WRITEERROR;
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t walk_data_block_ptr(imgtool::image &img, uint32_t *ptr, int *filesize, imgtool::stream *destf, bool write)
|
|
{
|
|
int i, blocksize = is_ffs(img) ? BSIZE : BSIZE-24;
|
|
imgtoolerr_t ret;
|
|
|
|
for (i = TSIZE-1; i >= 0; i--)
|
|
{
|
|
/* We write either blocksize bytes or whats remaining */
|
|
int bytes_left = (*filesize >= blocksize) ? blocksize : *filesize;
|
|
|
|
if (write)
|
|
{
|
|
ret = write_file_block_data(img, ptr[i], bytes_left, *destf);
|
|
if (ret) return ret;
|
|
}
|
|
else
|
|
{
|
|
ret = bitmap_mark_free(img, ptr[i]);
|
|
if (ret) return ret;
|
|
}
|
|
|
|
*filesize -= bytes_left;
|
|
|
|
/* Check if we are finished early */
|
|
if (*filesize == 0) break;
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t write_data_block_ptr(imgtool::image &img, uint32_t *ptr, int *filesize, imgtool::stream &destf)
|
|
{
|
|
return walk_data_block_ptr(img, ptr, filesize, &destf, true);
|
|
}
|
|
|
|
|
|
/* Marks all blocks pointed to by the data block pointers as free */
|
|
static imgtoolerr_t clear_data_block_ptr(imgtool::image &img, uint32_t *ptr, int *filesize)
|
|
{
|
|
return walk_data_block_ptr(img, ptr, filesize, nullptr, false);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t walk_file_ext_data(imgtool::image &img, int block, int *filesize, imgtool::stream *destf, int write)
|
|
{
|
|
file_ext_block file_ext;
|
|
imgtoolerr_t ret;
|
|
|
|
/* Get file extension block */
|
|
ret = read_file_ext_block(img, block, &file_ext);
|
|
if (ret) return ret;
|
|
|
|
/* Write all data pointers in the table */
|
|
ret = walk_data_block_ptr(img, file_ext.data_blocks, filesize, destf, write);
|
|
if (ret) return ret;
|
|
|
|
/* Check if we are finished */
|
|
if (*filesize != 0)
|
|
{
|
|
if (file_ext.extension == 0)
|
|
{
|
|
/* We are not finished, but there are no more extension blocks */
|
|
return IMGTOOLERR_CORRUPTFILE;
|
|
}
|
|
else
|
|
{
|
|
/* Write the next file extension block */
|
|
return walk_file_ext_data(img, file_ext.extension, filesize, destf, write);
|
|
}
|
|
}
|
|
|
|
/* Mark ourself as free if we not writing */
|
|
if (!write)
|
|
{
|
|
ret = bitmap_mark_free(img, block);
|
|
if (ret) return ret;
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t write_file_ext_data(imgtool::image &img, int block, int *filesize, imgtool::stream *destf)
|
|
{
|
|
return walk_file_ext_data(img, block, filesize, destf, true);
|
|
}
|
|
|
|
|
|
static imgtoolerr_t clear_file_ext_data(imgtool::image &img, int block, int *filesize)
|
|
{
|
|
return walk_file_ext_data(img, block, filesize, NULL, false);
|
|
}
|
|
|
|
|
|
/* Updates the disk alteration date stored in the root block */
|
|
static imgtoolerr_t update_disk_alteration_date(imgtool::image &img)
|
|
{
|
|
imgtoolerr_t ret;
|
|
root_block root;
|
|
time_t now;
|
|
|
|
/* Get root block */
|
|
ret = read_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* Get current time */
|
|
time(&now);
|
|
amiga_setup_time(now, &root.v);
|
|
|
|
/* Write back new root block */
|
|
ret = write_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* And update its checksum */
|
|
ret = fix_block_chksum(img, get_total_blocks(img)/2);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/*****************************************************************************
|
|
Imgtool functions
|
|
*****************************************************************************/
|
|
|
|
|
|
static imgtoolerr_t amiga_image_open(imgtool::image &img, imgtool::stream::ptr &&stream)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
uint64_t size = stream->size();
|
|
|
|
f->sectors = size/BSIZE/80/2;
|
|
|
|
if (f->sectors != 11 && f->sectors != 22)
|
|
{
|
|
return IMGTOOLERR_CORRUPTIMAGE;
|
|
}
|
|
|
|
f->stream = stream.release();
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static void amiga_image_exit(imgtool::image &img)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
if (f->stream)
|
|
delete f->stream;
|
|
}
|
|
|
|
|
|
static void amiga_image_info(imgtool::image &img, std::ostream &stream)
|
|
{
|
|
imgtoolerr_t ret;
|
|
root_block root;
|
|
char info[255];
|
|
time_t t_c, t_v, t_r;
|
|
char c[19], v[19], r[19];
|
|
|
|
ret = read_root_block(img, &root);
|
|
if (ret) return;
|
|
|
|
t_c = amiga_crack_time(&root.c).to_time_t();
|
|
t_v = amiga_crack_time(&root.v).to_time_t();
|
|
t_r = amiga_crack_time(&root.r).to_time_t();
|
|
|
|
strftime(c, sizeof(c), "%d-%b-%y %H:%M:%S", localtime(&t_c));
|
|
strftime(v, sizeof(v), "%d-%b-%y %H:%M:%S", localtime(&t_v));
|
|
strftime(r, sizeof(r), "%d-%b-%y %H:%M:%S", localtime(&t_r));
|
|
|
|
strcpy(info, "Volume name: ");
|
|
strncat(info, (char *)root.diskname, root.name_len);
|
|
strcat(info, "\nVolume created: ");
|
|
strcat(info, c);
|
|
strcat(info, "\nVolume modified: ");
|
|
strcat(info, v);
|
|
strcat(info, "\n Root modified: ");
|
|
strcat(info, r);
|
|
|
|
stream << info;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_read_sector(imgtool::image &img, uint32_t track, uint32_t head, uint32_t sector, void *buf, size_t len)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
|
|
/* skip ahead to the area we want to read */
|
|
f->stream->seek(track * (head+1) * f->sectors * BSIZE + sector * BSIZE, SEEK_CUR);
|
|
|
|
if (f->stream->read(buf, len) != len)
|
|
{
|
|
return IMGTOOLERR_READERROR;
|
|
}
|
|
|
|
/* reset stream */
|
|
f->stream->seek(0, 0);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_read_sector(imgtool::image &img,
|
|
uint32_t track, uint32_t head, uint32_t sector, std::vector<uint8_t> &buffer)
|
|
{
|
|
try { buffer.resize(BSIZE); }
|
|
catch (std::bad_alloc const &) { return IMGTOOLERR_OUTOFMEMORY; }
|
|
|
|
return amiga_image_read_sector(img, track, head, sector, &buffer[0], buffer.size());
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_write_sector(imgtool::image &img, uint32_t track, uint32_t head, uint32_t sector, const void *buf, size_t len, int ddam)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
|
|
/* skip ahead to the area we want to write */
|
|
f->stream->seek(track * (head+1) * f->sectors * BSIZE + sector * BSIZE, SEEK_CUR);
|
|
|
|
/* write data */
|
|
if (f->stream->write(buf, len) != len)
|
|
{
|
|
return IMGTOOLERR_WRITEERROR;
|
|
}
|
|
|
|
/* reset stream */
|
|
f->stream->seek(0, 0);
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_beginenum(imgtool::directory &enumeration, const char *path)
|
|
{
|
|
int blocks = get_total_blocks(enumeration.image());
|
|
imgtoolerr_t ret;
|
|
amiga_iterator *iter;
|
|
|
|
iter = (amiga_iterator *) enumeration.extra_bytes();
|
|
if (!iter) return IMGTOOLERR_OUTOFMEMORY;
|
|
|
|
iter->index = 1;
|
|
iter->ht_index = 0;
|
|
iter->eof = 0;
|
|
|
|
if (path[0])
|
|
{
|
|
/* search for the directory block, start with the root block */
|
|
ret = find_entry(enumeration.image(), path, blocks/2, &iter->block);
|
|
if (ret) return ret;
|
|
}
|
|
else
|
|
{
|
|
/* we didn't get a path, use the root directory */
|
|
iter->block = blocks / 2;
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_nextenum(imgtool::directory &enumeration, imgtool_dirent &ent)
|
|
{
|
|
amiga_iterator *iter = (amiga_iterator *) enumeration.extra_bytes();
|
|
imgtoolerr_t ret;
|
|
uint32_t ht[TSIZE];
|
|
int block;
|
|
|
|
/* finished listing all entries? */
|
|
if (iter->eof == 1 || iter->ht_index == TSIZE)
|
|
{
|
|
ent.eof = 1;
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
/* get hash table */
|
|
ret = get_hash_table(enumeration.image(), iter->block, ht);
|
|
if (ret) return ret;
|
|
|
|
/* skip empty hash table entries */
|
|
while (ht[iter->ht_index] == 0)
|
|
{
|
|
iter->ht_index++;
|
|
/* check if we are already at the end */
|
|
if (iter->ht_index == TSIZE)
|
|
{
|
|
ent.eof = 1;
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
/* get block number */
|
|
block = (iter->next_block == 0) ? ht[iter->ht_index] : iter->next_block;
|
|
|
|
switch (get_block_type(enumeration.image(), block))
|
|
{
|
|
case ST_FILE:
|
|
{
|
|
file_block file;
|
|
|
|
/* get block */
|
|
ret = read_file_block(enumeration.image(), block, &file);
|
|
if (ret) return ret;
|
|
|
|
/* fill directory entry */
|
|
strncpyz(ent.filename, (char *)file.filename, file.name_len + 1);
|
|
ent.filesize = file.byte_size;
|
|
ent.lastmodified_time = amiga_crack_time(&file.date);
|
|
amiga_decode_flags(file.protect, ent.attr);
|
|
strncpyz(ent.comment, (char *)file.comment, file.comm_len + 1);
|
|
|
|
iter->next_block = file.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_USERDIR:
|
|
{
|
|
dir_block dir;
|
|
|
|
/* get block */
|
|
ret = read_dir_block(enumeration.image(), block, &dir);
|
|
if (ret) return ret;
|
|
|
|
/* fill directory entry */
|
|
strncpyz(ent.filename, (char *)dir.dirname, dir.name_len + 1);
|
|
ent.lastmodified_time = amiga_crack_time(&dir.date);
|
|
amiga_decode_flags(dir.protect, ent.attr);
|
|
strncpyz(ent.comment, (char *)dir.comment, dir.comm_len + 1);
|
|
ent.directory = 1;
|
|
|
|
iter->next_block = dir.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_SOFTLINK:
|
|
{
|
|
softlink_block sl;
|
|
|
|
/* get block */
|
|
ret = read_softlink_block(enumeration.image(), block, &sl);
|
|
if (ret) return ret;
|
|
|
|
/* fill directory entry */
|
|
strncpyz(ent.filename, (char *)sl.slname, sl.name_len + 1);
|
|
ent.lastmodified_time = amiga_crack_time(&sl.date);
|
|
amiga_decode_flags(sl.protect, ent.attr);
|
|
strncpyz(ent.comment, (char *)sl.comment, sl.comm_len + 1);
|
|
strcpy(ent.softlink, (char *)sl.symbolic_name);
|
|
|
|
iter->next_block = sl.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_LINKDIR:
|
|
|
|
ent.directory = 1;
|
|
|
|
case ST_LINKFILE:
|
|
{
|
|
hardlink_block hl;
|
|
|
|
/* get block */
|
|
ret = read_hardlink_block(enumeration.image(), block, &hl);
|
|
if (ret) return ret;
|
|
|
|
/* get filesize from linked file */
|
|
if (!ent.directory)
|
|
{
|
|
file_block file;
|
|
ret = read_file_block(enumeration.image(), hl.real_entry, &file);
|
|
if (ret) return ret;
|
|
ent.filesize = file.byte_size;
|
|
}
|
|
|
|
/* fill directory entry */
|
|
strncpyz(ent.filename, (char *)hl.hlname, hl.name_len + 1);
|
|
ent.lastmodified_time = amiga_crack_time(&hl.date);
|
|
amiga_decode_flags(hl.protect, ent.attr);
|
|
strncpyz(ent.comment, (char *)hl.comment, hl.comm_len + 1);
|
|
ent.hardlink = 1;
|
|
|
|
iter->next_block = hl.hash_chain;
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
/* if there are no linked entries, go to the next hash table entry */
|
|
if (iter->next_block == 0)
|
|
{
|
|
iter->ht_index++;
|
|
}
|
|
|
|
/* jump to next index */
|
|
iter->index++;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_freespace(imgtool::partition &partition, uint64_t *size)
|
|
{
|
|
imgtoolerr_t ret;
|
|
imgtool::image &image(partition.image());
|
|
const int data_size = is_ffs(image) ? BSIZE : BSIZE-24;
|
|
root_block root;
|
|
bitmap_block bm;
|
|
int blocks, blocks_processed = 0, pages, i, c;
|
|
uint32_t v;
|
|
|
|
/* initialize size */
|
|
*size = 0;
|
|
|
|
/* get root block */
|
|
ret = read_root_block(image, &root);
|
|
if (ret) return ret;
|
|
|
|
/* get total number of blocks in the image */
|
|
blocks = get_total_blocks(image);
|
|
|
|
/* subtract the two bootblock blocks (only for floppies!) */
|
|
blocks -= 2;
|
|
|
|
/* iterate all bitmap pages */
|
|
for (pages = 0; pages < 25; pages++)
|
|
{
|
|
ret = read_bitmap_block(image, root.bm_pages[pages], &bm);
|
|
if (ret) return ret;
|
|
|
|
for (i = 0; i < MSIZE; i++)
|
|
{
|
|
v = bm.map[i];
|
|
|
|
/* clear half used value */
|
|
if ((blocks_processed + 32) > blocks)
|
|
v &= ~(~0 << (blocks - blocks_processed));
|
|
|
|
/* count bits */
|
|
for (c = 0; v; c++)
|
|
v &= v - 1;
|
|
|
|
*size += c * data_size;
|
|
|
|
blocks_processed += 32;
|
|
|
|
if (blocks_processed >= blocks)
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_readfile(imgtool::partition &partition, const char *filename, const char *fork, imgtool::stream &destf)
|
|
{
|
|
imgtool::image &img(partition.image());
|
|
imgtoolerr_t ret;
|
|
file_block file;
|
|
int filesize, block;
|
|
|
|
/* Search for the block number */
|
|
ret = find_entry(img, filename, get_total_blocks(img)/2, &block);
|
|
if (ret == IMGTOOLERR_PATHNOTFOUND) return IMGTOOLERR_FILENOTFOUND;
|
|
if (ret) return ret; /* Other errors */
|
|
|
|
/* Phase 1: Follow data pointers */
|
|
ret = read_file_block(img, block, &file);
|
|
if (ret) return ret;
|
|
|
|
filesize = file.byte_size;
|
|
|
|
/* Write out file data pointed to by data block pointers */
|
|
ret = write_data_block_ptr(img, file.data_blocks, &filesize, destf);
|
|
if (ret) return ret;
|
|
|
|
/* Check if we are done already */
|
|
if (filesize == 0) return IMGTOOLERR_SUCCESS;
|
|
|
|
/* Phase 2: Follow file extension blocks */
|
|
ret = write_file_ext_data(img, file.extension, &filesize, &destf);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
/* When a file is deleted, only its File header block number is cleared from
|
|
* the Directory block (or from the same-hash-value list) and the bitmap is
|
|
* updated. File header block, Data blocks and File extension blocks are not
|
|
* cleared, but the bitmap blocks are updated. */
|
|
static imgtoolerr_t amiga_image_deletefile(imgtool::partition &partition, const char *fname)
|
|
{
|
|
imgtool::image &img(partition.image());
|
|
imgtoolerr_t ret;
|
|
int parent, block;
|
|
char filename[31];
|
|
|
|
/* Initialize filename */
|
|
memset(filename, 0, sizeof(filename));
|
|
|
|
/* Search for the block number */
|
|
ret = find_entry(img, fname, get_total_blocks(img)/2, &block);
|
|
if (ret == IMGTOOLERR_PATHNOTFOUND) return IMGTOOLERR_FILENOTFOUND;
|
|
if (ret) return ret;
|
|
|
|
/* Get the parent block, where we need to clear the hash */
|
|
switch (get_block_type(img, block))
|
|
{
|
|
case ST_FILE:
|
|
{
|
|
file_block file;
|
|
int filesize;
|
|
|
|
ret = read_file_block(img, block, &file);
|
|
if (ret) return ret;
|
|
|
|
filesize = file.byte_size;
|
|
parent = file.parent;
|
|
memcpy(filename, file.filename, file.name_len);
|
|
|
|
/* Clear all linked data sectors */
|
|
ret = clear_data_block_ptr(img, file.data_blocks, &filesize);
|
|
if (ret) return ret;
|
|
|
|
/* Clear extended file data sectors */
|
|
if (filesize != 0)
|
|
{
|
|
ret = clear_file_ext_data(img, file.extension, &filesize);
|
|
if (ret) return ret;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case ST_LINKFILE:
|
|
{
|
|
softlink_block link;
|
|
|
|
ret = read_softlink_block(img, block, &link);
|
|
if (ret) return ret;
|
|
|
|
parent = link.parent;
|
|
memcpy(filename, link.slname, link.name_len);
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
return IMGTOOLERR_UNEXPECTED;
|
|
}
|
|
|
|
/* Clear hash table entry */
|
|
ret = clear_hash_table_entry(img, parent, filename);
|
|
if (ret) return ret;
|
|
|
|
/* Update disk alteration date */
|
|
ret = update_disk_alteration_date(img);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_writefile(imgtool::partition &partition, const char *filename, const char *fork, imgtool::stream &sourcef, util::option_resolution *opts)
|
|
{
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_create(imgtool::image &img, imgtool::stream::ptr &&stream, util::option_resolution *opts)
|
|
{
|
|
amiga_floppy *f = get_amiga_floppy(img);
|
|
const std::string &dskname = opts->lookup_string('N');
|
|
imgtoolerr_t ret;
|
|
uint8_t buffer[BSIZE];
|
|
root_block root;
|
|
bitmap_block bm;
|
|
time_t now;
|
|
int blocks;
|
|
|
|
f->stream = stream.get();
|
|
|
|
switch (opts->lookup_int('S'))
|
|
{
|
|
case 0: f->sectors = 11; break;
|
|
case 1: f->sectors = 22; break;
|
|
default: return IMGTOOLERR_PARAMCORRUPT;
|
|
}
|
|
|
|
/* initialize with zeros */
|
|
memset(buffer, 0, BSIZE);
|
|
|
|
/* add DOS magic string and flags */
|
|
buffer[0] = 'D';
|
|
buffer[1] = 'O';
|
|
buffer[2] = 'S';
|
|
buffer[3] = 0;
|
|
|
|
/* File system */
|
|
buffer[3] += (opts->lookup_int('F'));
|
|
|
|
/* File system mode */
|
|
buffer[3] += (opts->lookup_int('M'));
|
|
|
|
/* write first bootblock sector */
|
|
ret = write_block(img, 0, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* reset with zeros */
|
|
memset(buffer, 0, BSIZE);
|
|
|
|
/* write second bootblock sector */
|
|
ret = write_block(img, 1, buffer);
|
|
if (ret) return ret;
|
|
|
|
/* rootblock */
|
|
memset(&root, 0, sizeof(root_block));
|
|
|
|
blocks = get_total_blocks(img);
|
|
|
|
root.chksum = 0;
|
|
root.ht_size = TSIZE;
|
|
root.bm_flag = -1;
|
|
root.bm_pages[0] = blocks/2 + 1; /* generally it's located here */
|
|
|
|
time(&now);
|
|
amiga_setup_time(now, &root.r);
|
|
amiga_setup_time(now, &root.v);
|
|
amiga_setup_time(now, &root.c);
|
|
|
|
/* volume name */
|
|
if (!dskname.empty())
|
|
{
|
|
root.name_len = dskname.length();
|
|
memcpy(&root.diskname, dskname.c_str(), root.name_len);
|
|
}
|
|
else
|
|
{
|
|
root.name_len = strlen("Empty");
|
|
memcpy(&root.diskname, "Empty", root.name_len);
|
|
}
|
|
|
|
/* write root block to disk */
|
|
ret = write_root_block(img, &root);
|
|
if (ret) return ret;
|
|
|
|
/* calculate block checksum */
|
|
ret = fix_block_chksum(img, blocks/2);
|
|
if (ret) return ret;
|
|
|
|
/* bitmap block */
|
|
memset(&bm, 0xff, sizeof(bitmap_block));
|
|
|
|
/* write bitmap block to disk */
|
|
ret = write_bitmap_block(img, root.bm_pages[0], &bm);
|
|
if (ret) return ret;
|
|
|
|
/* set root and bitmap block as used */
|
|
ret = bitmap_mark_used(img, blocks/2);
|
|
if (ret) return ret;
|
|
|
|
ret = bitmap_mark_used(img, root.bm_pages[0]);
|
|
if (ret) return ret;
|
|
|
|
/* write empty last block so that we don't get a truncated image */
|
|
memset(buffer, 0, BSIZE);
|
|
|
|
ret = write_block(img, blocks - 1, buffer);
|
|
if (ret) return ret;
|
|
|
|
f->stream = stream.release();
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_createdir(imgtool::partition &partition, const char *path)
|
|
{
|
|
imgtool::image &img(partition.image());
|
|
imgtoolerr_t ret;
|
|
|
|
/* Create directories */
|
|
ret = checkdir(img, path, get_total_blocks(img)/2);
|
|
if (ret) return ret;
|
|
|
|
/* Update disk alteration date */
|
|
ret = update_disk_alteration_date(img);
|
|
if (ret) return ret;
|
|
|
|
return IMGTOOLERR_SUCCESS;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_getattrs(imgtool::partition &partition, const char *path, const uint32_t *attrs, imgtool_attribute *values)
|
|
{
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_setattrs(imgtool::partition &partition, const char *path, const uint32_t *attrs, const imgtool_attribute *values)
|
|
{
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_geticoninfo(imgtool::partition &partition, const char *path, imgtool_iconinfo *iconinfo)
|
|
{
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
|
|
static imgtoolerr_t amiga_image_suggesttransfer(imgtool::partition &partition, const char *fname, imgtool_transfer_suggestion *suggestions, size_t suggestions_length)
|
|
{
|
|
return IMGTOOLERR_UNIMPLEMENTED;
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************
|
|
Create image options
|
|
*****************************************************************************/
|
|
|
|
|
|
OPTION_GUIDE_START(amiga_createimage_optionguide)
|
|
OPTION_STRING( 'N', "name", "Volume name" )
|
|
OPTION_ENUM_START( 'S', "density", "Density" )
|
|
OPTION_ENUM( 0, "dd", "Double Density" )
|
|
OPTION_ENUM( 1, "hd", "High Density" )
|
|
OPTION_ENUM_END
|
|
OPTION_ENUM_START( 'F', "filesystem", "File system" )
|
|
OPTION_ENUM( 0, "ofs", "OldFileSystem" )
|
|
OPTION_ENUM( 1, "ffs", "FastFileSystem" )
|
|
OPTION_ENUM_END
|
|
OPTION_ENUM_START( 'M', "mode", "File system options" )
|
|
OPTION_ENUM( 0, "none", "None" )
|
|
OPTION_ENUM( 2, "intl", "International Mode" )
|
|
OPTION_ENUM( 4, "dirc", "International Mode with Directory Cache" )
|
|
OPTION_ENUM_END
|
|
OPTION_GUIDE_END
|
|
|
|
|
|
|
|
/*****************************************************************************
|
|
Imgtool module declaration
|
|
*****************************************************************************/
|
|
|
|
|
|
/* Amiga floppy disk attributes */
|
|
void amiga_floppy_get_info(const imgtool_class *imgclass, uint32_t state, union imgtoolinfo *info)
|
|
{
|
|
switch(state)
|
|
{
|
|
/* --- the following bits of info are returned as 64-bit signed integers --- */
|
|
case IMGTOOLINFO_INT_IMAGE_EXTRA_BYTES: info->i = sizeof(amiga_floppy); break;
|
|
case IMGTOOLINFO_INT_DIRECTORY_EXTRA_BYTES: info->i = sizeof(amiga_iterator); break;
|
|
case IMGTOOLINFO_INT_SUPPORTS_LASTMODIFIED_TIME: info->i = 1; break;
|
|
case IMGTOOLINFO_INT_PATH_SEPARATOR: info->i = '/'; break;
|
|
|
|
/* --- the following bits of info are returned as NULL-terminated strings --- */
|
|
case IMGTOOLINFO_STR_NAME: strcpy(info->s = imgtool_temp_str(), "amiga_floppy"); break;
|
|
case IMGTOOLINFO_STR_DESCRIPTION: strcpy(info->s = imgtool_temp_str(), "Amiga floppy disk image (OFS/FFS format)"); break;
|
|
case IMGTOOLINFO_STR_FILE_EXTENSIONS: strcpy(info->s = imgtool_temp_str(), "adf"); break;
|
|
case IMGTOOLINFO_STR_FILE: strcpy(info->s = imgtool_temp_str(), __FILE__); break;
|
|
case IMGTOOLINFO_STR_EOLN: strcpy(info->s = imgtool_temp_str(), EOLN_LF); break;
|
|
case IMGTOOLINFO_STR_CREATEIMAGE_OPTSPEC: strcpy(info->s = imgtool_temp_str(), "S[0]-1;F[0]-1;M[0]/2/4"); break;
|
|
|
|
/* --- the following bits of info are returned as pointers to data or functions --- */
|
|
case IMGTOOLINFO_PTR_OPEN: info->open = amiga_image_open; break;
|
|
case IMGTOOLINFO_PTR_CLOSE: info->close = amiga_image_exit; break;
|
|
case IMGTOOLINFO_PTR_READ_SECTOR: info->read_sector = amiga_image_read_sector; break;
|
|
case IMGTOOLINFO_PTR_WRITE_SECTOR: info->write_sector = amiga_image_write_sector; break;
|
|
case IMGTOOLINFO_PTR_CREATE: info->create = amiga_image_create; break;
|
|
case IMGTOOLINFO_PTR_INFO: info->info = amiga_image_info; break;
|
|
case IMGTOOLINFO_PTR_BEGIN_ENUM: info->begin_enum = amiga_image_beginenum; break;
|
|
case IMGTOOLINFO_PTR_NEXT_ENUM: info->next_enum = amiga_image_nextenum; break;
|
|
case IMGTOOLINFO_PTR_FREE_SPACE: info->free_space = amiga_image_freespace; break;
|
|
case IMGTOOLINFO_PTR_READ_FILE: info->read_file = amiga_image_readfile; break;
|
|
case IMGTOOLINFO_PTR_WRITE_FILE: info->write_file = amiga_image_writefile; break;
|
|
case IMGTOOLINFO_PTR_DELETE_FILE: info->delete_file = amiga_image_deletefile; break;
|
|
case IMGTOOLINFO_PTR_CREATE_DIR: info->create_dir = amiga_image_createdir; break;
|
|
case IMGTOOLINFO_PTR_GET_ATTRS: info->get_attrs = amiga_image_getattrs; break;
|
|
case IMGTOOLINFO_PTR_SET_ATTRS: info->set_attrs = amiga_image_setattrs; break;
|
|
case IMGTOOLINFO_PTR_GET_ICON_INFO: info->get_iconinfo = amiga_image_geticoninfo; break;
|
|
case IMGTOOLINFO_PTR_SUGGEST_TRANSFER: info->suggest_transfer = amiga_image_suggesttransfer; break;
|
|
case IMGTOOLINFO_PTR_CREATEIMAGE_OPTGUIDE: info->createimage_optguide = &amiga_createimage_optionguide; break;
|
|
case IMGTOOLINFO_PTR_CHARCONVERTER: info->charconverter = &imgtool::charconverter_iso_8859_1; break;
|
|
}
|
|
}
|