nkeynes@342: /**
nkeynes@561:  * $Id$
nkeynes@342:  *
nkeynes@342:  * GD-Rom image-file common functions. 
nkeynes@342:  *
nkeynes@342:  * Copyright (c) 2005 Nathan Keynes.
nkeynes@342:  *
nkeynes@342:  * This program is free software; you can redistribute it and/or modify
nkeynes@342:  * it under the terms of the GNU General Public License as published by
nkeynes@342:  * the Free Software Foundation; either version 2 of the License, or
nkeynes@342:  * (at your option) any later version.
nkeynes@342:  *
nkeynes@342:  * This program is distributed in the hope that it will be useful,
nkeynes@342:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
nkeynes@342:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nkeynes@342:  * GNU General Public License for more details.
nkeynes@342:  */
nkeynes@342: 
nkeynes@422: #include <netinet/in.h>
nkeynes@422: 
nkeynes@342: #include "gdrom/gdrom.h"
nkeynes@342: #include "gdrom/packet.h"
nkeynes@422: #include "bootstrap.h"
nkeynes@342: 
nkeynes@342: static void gdrom_image_destroy( gdrom_disc_t disc );
nkeynes@342: static gdrom_error_t gdrom_image_read_sector( gdrom_disc_t disc, uint32_t lba, int mode, 
nkeynes@422: 					      unsigned char *buf, uint32_t *readlength );
nkeynes@422: static gdrom_error_t gdrom_image_read_toc( gdrom_disc_t disc, unsigned char *buf );
nkeynes@422: static gdrom_error_t gdrom_image_read_session( gdrom_disc_t disc, int session, unsigned char *buf );
nkeynes@422: static gdrom_error_t gdrom_image_read_position( gdrom_disc_t disc, uint32_t lba, unsigned char *buf );
nkeynes@342: static int gdrom_image_drive_status( gdrom_disc_t disc );
nkeynes@342: 
nkeynes@498: struct cdrom_sector_header {
nkeynes@498:     uint8_t sync[12];
nkeynes@498:     uint8_t msf[3];
nkeynes@498:     uint8_t mode;
nkeynes@498: };
nkeynes@342: 
nkeynes@342: /**
nkeynes@342:  * Initialize a gdrom_disc structure with the gdrom_image_* methods
nkeynes@342:  */
nkeynes@342: void gdrom_image_init( gdrom_disc_t disc )
nkeynes@342: {
nkeynes@342:     memset( disc, 0, sizeof(struct gdrom_disc) ); /* safety */
nkeynes@342:     disc->read_sector = gdrom_image_read_sector;
nkeynes@342:     disc->read_toc = gdrom_image_read_toc;
nkeynes@342:     disc->read_session = gdrom_image_read_session;
nkeynes@342:     disc->read_position = gdrom_image_read_position;
nkeynes@342:     disc->drive_status = gdrom_image_drive_status;
nkeynes@342:     disc->play_audio = NULL; /* not supported yet */
nkeynes@342:     disc->run_time_slice = NULL; /* not needed */
nkeynes@342:     disc->close = gdrom_image_destroy;
nkeynes@342: }
nkeynes@342: 
nkeynes@464: gdrom_disc_t gdrom_image_new( const gchar *filename, FILE *f )
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)calloc(sizeof(struct gdrom_image), 1);
nkeynes@342:     if( image == NULL ) {
nkeynes@342: 	return NULL;
nkeynes@342:     }
nkeynes@342:     image->disc_type = IDE_DISC_CDROM;
nkeynes@342:     image->file = f;
nkeynes@342:     gdrom_disc_t disc = (gdrom_disc_t)image;
nkeynes@342:     gdrom_image_init(disc);
nkeynes@464:     if( filename == NULL ) {
nkeynes@464: 	disc->name = NULL;
nkeynes@464:     } else {
nkeynes@464: 	disc->name = g_strdup(filename);
nkeynes@464:     }
nkeynes@464: 
nkeynes@342:     return disc;
nkeynes@342: }
nkeynes@342: 
nkeynes@342: static void gdrom_image_destroy( gdrom_disc_t disc )
nkeynes@342: {
nkeynes@492:     int i;
nkeynes@492:     FILE *lastfile = NULL;
nkeynes@342:     gdrom_image_t img = (gdrom_image_t)disc;
nkeynes@342:     if( img->file != NULL ) {
nkeynes@342: 	fclose(img->file);
nkeynes@342: 	img->file = NULL;
nkeynes@342:     }
nkeynes@492:     for( i=0; i<img->track_count; i++ ) {
nkeynes@492: 	if( img->track[i].file != NULL && img->track[i].file != lastfile ) {
nkeynes@492: 	    lastfile = img->track[i].file;
nkeynes@492: 	    fclose(lastfile);
nkeynes@492: 	    img->track[i].file = NULL;
nkeynes@492: 	}
nkeynes@492:     }
nkeynes@464:     if( disc->name != NULL ) {
nkeynes@480: 	g_free( (gpointer)disc->name );
nkeynes@464: 	disc->name = NULL;
nkeynes@464:     }
nkeynes@342:     free( disc );
nkeynes@342: }
nkeynes@342: 
nkeynes@468: void gdrom_image_destroy_no_close( gdrom_disc_t disc )
nkeynes@468: {
nkeynes@492:     int i;
nkeynes@492:     FILE *lastfile = NULL;
nkeynes@468:     gdrom_image_t img = (gdrom_image_t)disc;
nkeynes@468:     if( img->file != NULL ) {
nkeynes@468: 	img->file = NULL;
nkeynes@468:     }
nkeynes@492:     for( i=0; i<img->track_count; i++ ) {
nkeynes@492: 	if( img->track[i].file != NULL && img->track[i].file != lastfile ) {
nkeynes@492: 	    lastfile = img->track[i].file;
nkeynes@492: 	    fclose(lastfile);
nkeynes@492: 	    img->track[i].file = NULL;
nkeynes@492: 	}
nkeynes@492:     }
nkeynes@468:     if( disc->name != NULL ) {
nkeynes@480: 	g_free( (gpointer)disc->name );
nkeynes@468: 	disc->name = NULL;
nkeynes@468:     }
nkeynes@468:     free( disc );
nkeynes@468: }
nkeynes@468: 
nkeynes@342: static int gdrom_image_get_track_by_lba( gdrom_image_t image, uint32_t lba )
nkeynes@342: {
nkeynes@342:     int i;
nkeynes@342:     for( i=0; i<image->track_count; i++ ) {
nkeynes@342: 	if( image->track[i].lba <= lba && 
nkeynes@342: 	    lba < (image->track[i].lba + image->track[i].sector_count) ) {
nkeynes@342: 	    return i+1;
nkeynes@342: 	}
nkeynes@342:     }
nkeynes@342:     return -1;
nkeynes@342: }
nkeynes@342: 
nkeynes@492: /**
nkeynes@492:  * Read a block from an image file, handling negative file offsets
nkeynes@492:  * with 0-fill.
nkeynes@492:  */
nkeynes@498: static void gdrom_read_block( unsigned char *buf, int file_offset, int length, FILE *f )
nkeynes@492: {
nkeynes@492:     if( file_offset < 0 ) {
nkeynes@492: 	int size = -file_offset;
nkeynes@492: 	if( size >= length ) {
nkeynes@492: 	    memset( buf, 0, length );
nkeynes@492: 	    return;
nkeynes@492: 	} else {
nkeynes@492: 	    memset( buf, 0, size );
nkeynes@492: 	    file_offset = 0;
nkeynes@492: 	    length -= size;
nkeynes@492: 	}
nkeynes@492:     }
nkeynes@492:     fseek( f, file_offset, SEEK_SET );
nkeynes@492:     fread( buf, length, 1, f );
nkeynes@492: }
nkeynes@492: 
nkeynes@342: static gdrom_error_t gdrom_image_read_sector( gdrom_disc_t disc, uint32_t lba,
nkeynes@422: 					      int mode, unsigned char *buf, uint32_t *length )
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)disc;
nkeynes@498:     struct cdrom_sector_header secthead;
nkeynes@422:     int file_offset, read_len, track_no;
nkeynes@498: 
nkeynes@492:     FILE *f;
nkeynes@342: 
nkeynes@342:     track_no = gdrom_image_get_track_by_lba( image, lba );
nkeynes@342:     if( track_no == -1 ) {
nkeynes@342: 	return PKT_ERR_BADREAD;
nkeynes@342:     }
nkeynes@342:     struct gdrom_track *track = &image->track[track_no-1];
nkeynes@342:     file_offset = track->offset + track->sector_size * (lba - track->lba);
nkeynes@342:     read_len = track->sector_size;
nkeynes@492:     if( track->file != NULL ) {
nkeynes@492: 	f = track->file;
nkeynes@492:     } else {
nkeynes@492: 	f = image->file;
nkeynes@492:     }
nkeynes@492: 
nkeynes@492:     
nkeynes@342: 
nkeynes@342:     switch( mode ) {
nkeynes@640:     case 0x20: /* Actually, read anything, but for now... */
nkeynes@342:     case 0x24:
nkeynes@342:     case 0x28:
nkeynes@498: 	read_len = 2048;
nkeynes@342: 	switch( track->mode ) {
nkeynes@342: 	case GDROM_MODE1:
nkeynes@342: 	case GDROM_MODE2_XA1:
nkeynes@492: 	    gdrom_read_block( buf, file_offset, track->sector_size, f );
nkeynes@342: 	    break;
nkeynes@342: 	case GDROM_MODE2:
nkeynes@342: 	    file_offset += 8; /* skip the subheader */
nkeynes@492: 	    gdrom_read_block( buf, file_offset, 2048, f );
nkeynes@342: 	    break;
nkeynes@498: 	case GDROM_RAW:
nkeynes@498: 	    gdrom_read_block( (unsigned char *)(&secthead), file_offset, sizeof(secthead), f );
nkeynes@498: 	    switch( secthead.mode ) {
nkeynes@498: 	    case 1:
nkeynes@498: 		file_offset += 16;
nkeynes@498: 		break;
nkeynes@498: 	    case 2:
nkeynes@498: 		file_offset += 24;
nkeynes@498: 		break;
nkeynes@498: 	    default:
nkeynes@498: 		return PKT_ERR_BADREADMODE;
nkeynes@498: 	    }
nkeynes@498: 	    gdrom_read_block( buf, file_offset, 2048, f );
nkeynes@498: 	    break;
nkeynes@342: 	default:
nkeynes@342: 	    return PKT_ERR_BADREADMODE;
nkeynes@342: 	}
nkeynes@342: 	break;
nkeynes@342:     default:
nkeynes@342: 	return PKT_ERR_BADREADMODE;
nkeynes@342:     }
nkeynes@342: 	    
nkeynes@342:     *length = read_len;
nkeynes@342:     return PKT_ERR_OK;
nkeynes@342: 
nkeynes@342: }
nkeynes@342: 
nkeynes@422: static gdrom_error_t gdrom_image_read_toc( gdrom_disc_t disc, unsigned char *buf ) 
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)disc;
nkeynes@342:     struct gdrom_toc *toc = (struct gdrom_toc *)buf;
nkeynes@342:     int i;
nkeynes@342: 
nkeynes@342:     for( i=0; i<image->track_count; i++ ) {
nkeynes@342: 	toc->track[i] = htonl( image->track[i].lba ) | image->track[i].flags;
nkeynes@342:     }
nkeynes@342:     toc->first = 0x0100 | image->track[0].flags;
nkeynes@342:     toc->last = (image->track_count<<8) | image->track[i-1].flags;
nkeynes@342:     toc->leadout = htonl(image->track[i-1].lba + image->track[i-1].sector_count) |
nkeynes@342: 	image->track[i-1].flags;
nkeynes@342:     for( ;i<99; i++ )
nkeynes@342: 	toc->track[i] = 0xFFFFFFFF;
nkeynes@342:     return PKT_ERR_OK;
nkeynes@342: }
nkeynes@342: 
nkeynes@422: static gdrom_error_t gdrom_image_read_session( gdrom_disc_t disc, int session, unsigned char *buf )
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)disc;
nkeynes@342:     struct gdrom_track *last_track = &image->track[image->track_count-1];
nkeynes@342:     unsigned int end_of_disc = last_track->lba + last_track->sector_count;
nkeynes@342:     int i;
nkeynes@342:     buf[0] = 0x01; /* Disc status? */
nkeynes@342:     buf[1] = 0;
nkeynes@342: 
nkeynes@342:     if( session == 0 ) {
nkeynes@342: 	buf[2] = last_track->session+1; /* last session */
nkeynes@342: 	buf[3] = (end_of_disc >> 16) & 0xFF;
nkeynes@342: 	buf[4] = (end_of_disc >> 8) & 0xFF;
nkeynes@342: 	buf[5] = end_of_disc & 0xFF;
nkeynes@342: 	return PKT_ERR_OK;
nkeynes@342:     } else {
nkeynes@342: 	session--;
nkeynes@342: 	for( i=0; i<image->track_count; i++ ) {
nkeynes@342: 	    if( image->track[i].session == session ) {
nkeynes@342: 		buf[2] = i+1; /* first track of session */
nkeynes@342: 		buf[3] = (image->track[i].lba >> 16) & 0xFF;
nkeynes@342: 		buf[4] = (image->track[i].lba >> 8) & 0xFF;
nkeynes@342: 		buf[5] = image->track[i].lba & 0xFF;
nkeynes@342: 		return PKT_ERR_OK;
nkeynes@342: 	    }
nkeynes@342: 	}
nkeynes@342: 	return PKT_ERR_BADFIELD; /* No such session */
nkeynes@342:     }
nkeynes@342: }
nkeynes@342: 
nkeynes@422: static gdrom_error_t gdrom_image_read_position( gdrom_disc_t disc, uint32_t lba, unsigned char *buf )
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)disc;
nkeynes@342:     int track_no = gdrom_image_get_track_by_lba( image, lba );
nkeynes@342:     if( track_no == -1 ) {
nkeynes@342: 	track_no = 1;
nkeynes@342: 	lba = 150;
nkeynes@342:     }
nkeynes@342:     struct gdrom_track *track = &image->track[track_no-1];
nkeynes@342:     uint32_t offset = lba - track->lba;
nkeynes@342:     buf[4] = track->flags;
nkeynes@342:     buf[5] = track_no;
nkeynes@342:     buf[6] = 0x01; /* ?? */
nkeynes@342:     buf[7] = (offset >> 16) & 0xFF;
nkeynes@342:     buf[8] = (offset >> 8) & 0xFF;
nkeynes@342:     buf[9] = offset & 0xFF;
nkeynes@342:     buf[10] = 0;
nkeynes@342:     buf[11] = (lba >> 16) & 0xFF;
nkeynes@342:     buf[12] = (lba >> 8) & 0xFF;
nkeynes@342:     buf[13] = lba & 0xFF;
nkeynes@422:     return PKT_ERR_OK;
nkeynes@342: }
nkeynes@342: 
nkeynes@342: static int gdrom_image_drive_status( gdrom_disc_t disc ) 
nkeynes@342: {
nkeynes@342:     gdrom_image_t image = (gdrom_image_t)disc;
nkeynes@342:     return image->disc_type | IDE_DISC_READY;
nkeynes@342: }
nkeynes@342: 
nkeynes@342: void gdrom_image_dump_info( gdrom_disc_t d ) {
nkeynes@342:     gdrom_image_t disc = (gdrom_image_t)d;
nkeynes@342:     int i;
nkeynes@342:     int last_session = disc->track[disc->track_count-1].session;
nkeynes@342:     gboolean is_bootable = FALSE;
nkeynes@342: 
nkeynes@342:     INFO( "Disc ID: %s, %d tracks in %d sessions", disc->mcn, disc->track_count, 
nkeynes@342: 	  disc->track[disc->track_count-1].session + 1 );
nkeynes@342:     if( last_session > 0 ) {
nkeynes@342: 	/* Boot track is the first data track of the last session, provided that it 
nkeynes@342: 	 * cannot be a single-session disc.
nkeynes@342: 	 */
nkeynes@342: 	int boot_track = -1;
nkeynes@342: 	for( i=disc->track_count-1; i>=0 && disc->track[i].session == last_session; i-- ) {
nkeynes@342: 	    if( disc->track[i].flags & TRACK_DATA ) {
nkeynes@342: 		boot_track = i;
nkeynes@342: 	    }
nkeynes@342: 	}
nkeynes@342: 	if( boot_track != -1 ) {
nkeynes@422: 	    unsigned char boot_sector[MAX_SECTOR_SIZE];
nkeynes@342: 	    uint32_t length = sizeof(boot_sector);
nkeynes@342: 	    if( d->read_sector( d, disc->track[boot_track].lba, 0x28,
nkeynes@342: 				   boot_sector, &length ) == PKT_ERR_OK ) {
nkeynes@342: 		bootstrap_dump(boot_sector, FALSE);
nkeynes@342: 		is_bootable = TRUE;
nkeynes@342: 	    }
nkeynes@342: 	}
nkeynes@342:     }
nkeynes@342:     if( !is_bootable ) {
nkeynes@342: 	WARN( "Disc does not appear to be bootable" );
nkeynes@342:     }
nkeynes@342: }
nkeynes@342: