502 lines
14 KiB
C
502 lines
14 KiB
C
/* $NetBSD: cd9660_write.c,v 1.1 2005/08/13 01:53:01 fvdl Exp $ */
|
|
|
|
/*
|
|
* Copyright (c) 2005 Daniel Watt, Walter Deignan, Ryan Gabrys, Alan
|
|
* Perez-Rathke and Ram Vedam. All rights reserved.
|
|
*
|
|
* This code was written by Daniel Watt, Walter Deignan, Ryan Gabrys,
|
|
* Alan Perez-Rathke and Ram Vedam.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or
|
|
* without modification, are permitted provided that the following
|
|
* conditions are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above
|
|
* copyright notice, this list of conditions and the following
|
|
* disclaimer in the documentation and/or other materials provided
|
|
* with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY DANIEL WATT, WALTER DEIGNAN, RYAN
|
|
* GABRYS, ALAN PEREZ-RATHKE AND RAM VEDAM ``AS IS'' AND ANY EXPRESS OR
|
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
* DISCLAIMED. IN NO EVENT SHALL DANIEL WATT, WALTER DEIGNAN, RYAN
|
|
* GABRYS, ALAN PEREZ-RATHKE AND RAM VEDAM BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
|
* USE,DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
|
|
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
|
|
* OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "cd9660.h"
|
|
#include "iso9660_rrip.h"
|
|
|
|
#include <sys/cdefs.h>
|
|
#if defined(__RCSID) && !defined(__lint)
|
|
__RCSID("$NetBSD: cd9660_write.c,v 1.1 2005/08/13 01:53:01 fvdl Exp $");
|
|
#endif /* !__lint */
|
|
|
|
static int cd9660_write_volume_descriptors(FILE *);
|
|
static int cd9660_write_path_table(FILE *, int, int);
|
|
static int cd9660_write_path_tables(FILE *);
|
|
static int cd9660_write_file(FILE *, cd9660node *);
|
|
static int cd9660_write_filedata(FILE *, int, const unsigned char *, int);
|
|
#if 0
|
|
static int cd9660_write_buffered(FILE *, int, int, const unsigned char*);
|
|
#endif
|
|
static int cd9660_write_rr(FILE *, cd9660node *, int, int);
|
|
|
|
/*
|
|
* Write the image
|
|
* Writes the entire image
|
|
* @param const char* The filename for the image
|
|
* @returns int 1 on success, 0 on failure
|
|
*/
|
|
int
|
|
cd9660_write_image(const char* image)
|
|
{
|
|
FILE *fd;
|
|
int status;
|
|
char buf[2048];
|
|
|
|
if ((fd = fopen(image, "w+")) == NULL)
|
|
err(1, "Error: Can't open `%s' for writing", image);
|
|
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Writing image\n");
|
|
|
|
/* Write the volume descriptors */
|
|
status = cd9660_write_volume_descriptors(fd);
|
|
if (status == 0) {
|
|
warnx("cd9660_write_image: Error writing volume "
|
|
"descriptors to image");
|
|
goto cleanup_bad_image;
|
|
}
|
|
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Volume descriptors written\n");
|
|
|
|
/*
|
|
* Write the path tables: there are actually four, but right
|
|
* now we are only concearned with two.
|
|
*/
|
|
status = cd9660_write_path_tables(fd);
|
|
if (status == 0) {
|
|
warnx("cd9660_write_image: Error writing path tables to image");
|
|
goto cleanup_bad_image;
|
|
}
|
|
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Path tables written\n");
|
|
|
|
/* Write the directories and files */
|
|
status = cd9660_write_file(fd, diskStructure.rootNode);
|
|
if (status == 0) {
|
|
warnx("cd9660_write_image: Error writing files to image");
|
|
goto cleanup_bad_image;
|
|
}
|
|
|
|
if (diskStructure.is_bootable) {
|
|
cd9660_write_boot(fd);
|
|
}
|
|
|
|
/* Write padding bits. This is temporary */
|
|
memset(buf, 0, 2048);
|
|
cd9660_write_filedata(fd, diskStructure.totalSectors - 1, buf, 1);
|
|
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Files written\n");
|
|
fclose(fd);
|
|
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Image closed\n");
|
|
return 1;
|
|
|
|
cleanup_bad_image:
|
|
fclose(fd);
|
|
if (!diskStructure.keep_bad_images)
|
|
unlink(image);
|
|
if (diskStructure.verbose_level > 0)
|
|
printf("Bad image cleaned up\n");
|
|
return 0;
|
|
}
|
|
|
|
static int
|
|
cd9660_write_volume_descriptors(FILE *fd)
|
|
{
|
|
volume_descriptor *vd_temp = diskStructure.firstVolumeDescriptor;
|
|
int pos;
|
|
|
|
while (vd_temp != NULL) {
|
|
pos = vd_temp->sector*diskStructure.sectorSize;
|
|
cd9660_write_filedata(fd, vd_temp->sector,
|
|
vd_temp->volumeDescriptorData, 1);
|
|
vd_temp = vd_temp->next;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Write out an individual path table
|
|
* Used just to keep redundant code to a minimum
|
|
* @param FILE *fd Valid file pointer
|
|
* @param int Sector to start writing path table to
|
|
* @param int Endian mode : BIG_ENDIAN or LITTLE_ENDIAN
|
|
* @returns int 1 on success, 0 on failure
|
|
*/
|
|
static int
|
|
cd9660_write_path_table(FILE *fd, int sector, int mode)
|
|
{
|
|
int path_table_sectors = CD9660_BLOCKS(diskStructure.sectorSize,
|
|
diskStructure.pathTableLength);
|
|
unsigned char *buffer;
|
|
unsigned char *buffer_head;
|
|
int len;
|
|
path_table_entry temp_entry;
|
|
cd9660node *ptcur;
|
|
|
|
buffer = malloc(diskStructure.sectorSize * path_table_sectors);
|
|
if (buffer == NULL) {
|
|
warnx("cd9660_write_path_table: Memory allocation error "
|
|
"allocating buffer");
|
|
return 0;
|
|
}
|
|
buffer_head = buffer;
|
|
memset(buffer, 0, diskStructure.sectorSize * path_table_sectors);
|
|
|
|
ptcur = diskStructure.rootNode;
|
|
|
|
while (ptcur != NULL) {
|
|
memset(&temp_entry, 0, sizeof(path_table_entry));
|
|
temp_entry.length[0] = ptcur->isoDirRecord->name_len[0];
|
|
temp_entry.extended_attribute_length[0] =
|
|
ptcur->isoDirRecord->ext_attr_length[0];
|
|
memcpy(temp_entry.name, ptcur->isoDirRecord->name,
|
|
temp_entry.length[0] + 1);
|
|
|
|
/* round up */
|
|
len = temp_entry.length[0] + 8 + (temp_entry.length[0] & 0x01);
|
|
|
|
/* todo: function pointers instead */
|
|
if (mode == LITTLE_ENDIAN) {
|
|
cd9660_731(ptcur->fileDataSector,
|
|
temp_entry.first_sector);
|
|
cd9660_721((ptcur->parent == NULL ?
|
|
1 : ptcur->parent->ptnumber),
|
|
temp_entry.parent_number);
|
|
} else {
|
|
cd9660_732(ptcur->fileDataSector,
|
|
temp_entry.first_sector);
|
|
cd9660_722((ptcur->parent == NULL ?
|
|
1 : ptcur->parent->ptnumber),
|
|
temp_entry.parent_number);
|
|
}
|
|
|
|
|
|
memcpy(buffer, &temp_entry, len);
|
|
buffer += len;
|
|
|
|
ptcur = ptcur->ptnext;
|
|
}
|
|
|
|
return cd9660_write_filedata(fd, sector, buffer_head,
|
|
path_table_sectors);
|
|
}
|
|
|
|
|
|
/*
|
|
* Write out the path tables to disk
|
|
* Each file descriptor should be pointed to by the PVD, so we know which
|
|
* sector to copy them to. One thing to watch out for: the only path tables
|
|
* stored are in the endian mode that the application is compiled for. So,
|
|
* the first thing to do is write out that path table, then to write the one
|
|
* in the other endian mode requires to convert the endianness of each entry
|
|
* in the table. The best way to do this would be to create a temporary
|
|
* path_table_entry structure, then for each path table entry, copy it to
|
|
* the temporary entry, translate, then copy that to disk.
|
|
*
|
|
* @param FILE* Valid file descriptor
|
|
* @returns int 0 on failure, 1 on success
|
|
*/
|
|
static int
|
|
cd9660_write_path_tables(FILE *fd)
|
|
{
|
|
if (cd9660_write_path_table(fd,
|
|
diskStructure.primaryLittleEndianTableSector, LITTLE_ENDIAN) == 0)
|
|
return 0;
|
|
|
|
if (cd9660_write_path_table(fd,
|
|
diskStructure.primaryBigEndianTableSector, BIG_ENDIAN) == 0)
|
|
return 0;
|
|
|
|
/* @TODO: handle remaining two path tables */
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Write a file to disk
|
|
* Writes a file, its directory record, and its data to disk
|
|
* This file is designed to be called RECURSIVELY, so initially call it
|
|
* with the root node. All of the records should store what sector the
|
|
* file goes in, so no computation should be necessary.
|
|
*
|
|
* @param int fd Valid file descriptor
|
|
* @param struct cd9660node* writenode Pointer to the file to be written
|
|
* @returns int 0 on failure, 1 on success
|
|
*/
|
|
static int
|
|
cd9660_write_file(FILE *fd, cd9660node *writenode)
|
|
{
|
|
char *buf;
|
|
char *temp_file_name;
|
|
int ret;
|
|
int working_sector;
|
|
int cur_sector_offset;
|
|
int written;
|
|
iso_directory_record_cd9660 temp_record;
|
|
cd9660node *temp;
|
|
int ca = 0;
|
|
|
|
/* Todo : clean up variables */
|
|
|
|
temp_file_name = malloc(CD9660MAXPATH + 1);
|
|
if (temp_file_name == NULL)
|
|
errx(1, "cd9660_write_file: failed to allocate filename space");
|
|
|
|
memset(temp_file_name, 0, CD9660MAXPATH + 1);
|
|
|
|
buf = malloc(diskStructure.sectorSize);
|
|
if (buf == NULL)
|
|
errx(1, "cd9660_write_file: Failed to allocate buffer memory");
|
|
|
|
if ((writenode->level != 0) &&
|
|
!(writenode->node->type & S_IFDIR)) {
|
|
/* Only attempt to write files that have length */
|
|
if (writenode->fileDataLength > 0) {
|
|
cd9660_compute_full_filename(writenode,
|
|
temp_file_name, 0);
|
|
ret = cd9660_copy_file(fd, writenode->fileDataSector,
|
|
temp_file_name);
|
|
if (ret == 0) {
|
|
free(temp_file_name);
|
|
return 0;
|
|
}
|
|
}
|
|
} else {
|
|
/*
|
|
* Here is a new revelation that ECMA didnt explain
|
|
* (at least not well).
|
|
* ALL . and .. records store the name "\0" and "\1"
|
|
* resepctively. So, for each directory, we have to
|
|
* make a new node.
|
|
*
|
|
* This is where it gets kinda messy, since we have to
|
|
* be careful of sector boundaries
|
|
*/
|
|
cur_sector_offset = 0;
|
|
working_sector = writenode->fileDataSector;
|
|
fseek(fd, working_sector * diskStructure.sectorSize, SEEK_SET);
|
|
|
|
/*
|
|
* Now loop over children, writing out their directory
|
|
* records - beware of sector boundaries
|
|
*/
|
|
temp = writenode->child;
|
|
while (temp != NULL) {
|
|
|
|
/*
|
|
* Copy the temporary record and adjust its size
|
|
* if necessary
|
|
*/
|
|
memcpy(&temp_record, temp->isoDirRecord,
|
|
sizeof(iso_directory_record_cd9660));
|
|
|
|
temp_record.length[0] =
|
|
cd9660_compute_record_size(temp);
|
|
|
|
if (temp_record.length[0] + cur_sector_offset >=
|
|
diskStructure.sectorSize) {
|
|
cur_sector_offset = 0;
|
|
working_sector++;
|
|
|
|
/* Seek to the next sector. */
|
|
fseek(fd,
|
|
working_sector * diskStructure.sectorSize,
|
|
SEEK_SET);
|
|
}
|
|
|
|
written = fwrite(&temp_record, 1,
|
|
temp_record.length[0], fd);
|
|
ca = 0;
|
|
if (diskStructure.rock_ridge_enabled) {
|
|
ca = cd9660_write_rr(fd, temp,
|
|
cur_sector_offset, working_sector);
|
|
}
|
|
|
|
if (ferror(fd)) {
|
|
warnx("Write error at %i", __LINE__);
|
|
free(temp_file_name);
|
|
return 0;
|
|
}
|
|
cur_sector_offset += temp_record.length[0];
|
|
|
|
/*
|
|
* If we had to go the the continuation area,
|
|
* head back to where we should be.
|
|
*/
|
|
if (ca) {
|
|
fseek(fd,
|
|
working_sector * diskStructure.sectorSize +
|
|
cur_sector_offset,
|
|
SEEK_SET);
|
|
}
|
|
temp = temp->next;
|
|
}
|
|
|
|
/*
|
|
* Now do the second loop, this time its recursive
|
|
*/
|
|
temp = writenode->child;
|
|
while (temp != NULL) {
|
|
ret = cd9660_write_file(fd, temp);
|
|
if (ret == 0) {
|
|
free(temp_file_name);
|
|
return 0;
|
|
}
|
|
temp = temp->next;
|
|
}
|
|
}
|
|
|
|
free(temp_file_name);
|
|
return 1;
|
|
}
|
|
|
|
/*
|
|
* Wrapper function to write a buffer (one sector) to disk.
|
|
* Seeks and writes the buffer.
|
|
* NOTE: You dont NEED to use this function, but it might make your
|
|
* life easier if you have to write things that align to a sector
|
|
* (such as volume descriptors).
|
|
*
|
|
* @param int fd Valid file descriptor
|
|
* @param int sector Sector number to write to
|
|
* @param const unsigned char* Buffer to write. This should be the
|
|
* size of a sector, and if only a portion
|
|
* is written, the rest should be set to 0.
|
|
*/
|
|
static int
|
|
cd9660_write_filedata(FILE *fd, int sector, const unsigned char *buf,
|
|
int numsecs)
|
|
{
|
|
off_t curpos;
|
|
size_t success;
|
|
|
|
curpos = ftello(fd);
|
|
|
|
fseek(fd, sector * diskStructure.sectorSize, SEEK_SET);
|
|
|
|
success = fwrite(buf, diskStructure.sectorSize * numsecs, 1, fd);
|
|
|
|
fseek(fd, curpos, SEEK_SET);
|
|
|
|
if (success == 1)
|
|
success = diskStructure.sectorSize * numsecs;
|
|
return success;
|
|
}
|
|
|
|
#if 0
|
|
static int
|
|
cd9660_write_buffered(FILE *fd, int offset, int buff_len,
|
|
const unsigned char* buffer)
|
|
{
|
|
static int working_sector = -1;
|
|
static char buf[2048];
|
|
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
int
|
|
cd9660_copy_file(FILE *fd, int start_sector, const char *filename)
|
|
{
|
|
FILE *rf;
|
|
int bytes_read;
|
|
int sector = start_sector;
|
|
int buf_size = diskStructure.sectorSize;
|
|
char *buf;
|
|
|
|
buf = malloc(buf_size);
|
|
if (buf == NULL)
|
|
errx(1, "cd9660_copy_file: memory allocation error "
|
|
"allocating buffer");
|
|
|
|
if ((rf = fopen(filename, "rb")) == NULL) {
|
|
warnx("Cant open file %s",filename);
|
|
return 0;
|
|
}
|
|
|
|
if (diskStructure.verbose_level > 1)
|
|
printf("Writing file: %s\n",filename);
|
|
|
|
fseek(fd, start_sector * diskStructure.sectorSize, SEEK_SET);
|
|
|
|
while (!feof(rf)) {
|
|
bytes_read = fread(buf,1,buf_size,rf);
|
|
if (ferror(rf)) {
|
|
warnx("cd9660_write_file: File read error");
|
|
return 0;
|
|
}
|
|
|
|
fwrite(buf,1,bytes_read,fd);
|
|
if (ferror(fd)) {
|
|
warnx("cd9660_write_file: File write error");
|
|
return 0;
|
|
}
|
|
sector++;
|
|
}
|
|
|
|
fclose(rf);
|
|
free(buf);
|
|
return 1;
|
|
}
|
|
|
|
static int
|
|
cd9660_write_rr(FILE *fd, cd9660node *writenode, int offset, int sector)
|
|
{
|
|
int in_ca = 0;
|
|
struct ISO_SUSP_ATTRIBUTES *myattr;
|
|
|
|
offset += writenode->isoDirRecord->length[0];
|
|
|
|
/* Offset now points at the end of the record */
|
|
for (myattr = writenode->head.lh_first; myattr != NULL;
|
|
myattr = myattr->rr_ll.le_next) {
|
|
fseek(fd,
|
|
in_ca ? offset : sector*diskStructure.sectorSize + offset,
|
|
SEEK_SET);
|
|
fwrite(&(myattr->attr), CD9660_SUSP_ENTRY_SIZE(myattr), 1, fd);
|
|
|
|
offset += CD9660_SUSP_ENTRY_SIZE(myattr);
|
|
if (!in_ca) {
|
|
if ((myattr->susp_type == SUSP_TYPE_SUSP) &&
|
|
(myattr->entry_type == SUSP_ENTRY_SUSP_CE)) {
|
|
/*
|
|
* Point the offset to the start of this
|
|
* record's CE area
|
|
*/
|
|
offset = (diskStructure.
|
|
susp_continuation_area_start_sector *
|
|
diskStructure.sectorSize)
|
|
+ writenode->susp_entry_ce_start;
|
|
in_ca = 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
return in_ca;
|
|
}
|