nkeynes@138: /**
nkeynes@245:  * $Id: gdrom.c,v 1.11 2006-12-19 09:52:56 nkeynes Exp $
nkeynes@138:  *
nkeynes@138:  * GD-Rom  access functions.
nkeynes@138:  *
nkeynes@138:  * Copyright (c) 2005 Nathan Keynes.
nkeynes@138:  *
nkeynes@138:  * This program is free software; you can redistribute it and/or modify
nkeynes@138:  * it under the terms of the GNU General Public License as published by
nkeynes@138:  * the Free Software Foundation; either version 2 of the License, or
nkeynes@138:  * (at your option) any later version.
nkeynes@138:  *
nkeynes@138:  * This program is distributed in the hope that it will be useful,
nkeynes@138:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
nkeynes@138:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nkeynes@138:  * GNU General Public License for more details.
nkeynes@138:  */
nkeynes@138: 
nkeynes@138: #include <stdio.h>
nkeynes@237: #include <fcntl.h>
nkeynes@168: #include <errno.h>
nkeynes@138: #include "gdrom/ide.h"
nkeynes@138: #include "gdrom/gdrom.h"
nkeynes@143: #include "gdrom/packet.h"
nkeynes@138: #include "dream.h"
nkeynes@138: 
nkeynes@138: static void gdrom_image_destroy( gdrom_disc_t );
nkeynes@152: static gdrom_error_t gdrom_image_read_sectors( gdrom_disc_t, uint32_t, uint32_t, int, char *, uint32_t * );
nkeynes@138: 
nkeynes@237: gdrom_image_class_t gdrom_image_classes[] = { &linux_device_class, &nrg_image_class, &cdi_image_class, NULL };
nkeynes@138: 
nkeynes@138: gdrom_disc_t gdrom_disc = NULL;
nkeynes@138: 
nkeynes@138: char *gdrom_mode_names[] = { "Mode1", "Mode2", "XA 1", "XA2", "Audio", "GD-Rom" };
nkeynes@138: uint32_t gdrom_sector_size[] = { 2048, 2336, 2048, 2324, 2352, 2336 };
nkeynes@138: 
nkeynes@138: gdrom_disc_t gdrom_image_open( const gchar *filename )
nkeynes@138: {
nkeynes@168:     const gchar *ext = strrchr(filename, '.');
nkeynes@168:     gdrom_disc_t disc = NULL;
nkeynes@237: 
nkeynes@237:     int fd = open( filename, O_RDONLY | O_NONBLOCK );
nkeynes@237:     FILE *f;
nkeynes@168:     int i,j;
nkeynes@168:     gdrom_image_class_t extclz = NULL;
nkeynes@168: 
nkeynes@237:     if( fd == -1 ) {
nkeynes@168: 	ERROR("Unable to open file '%s': %s", filename, strerror(errno));
nkeynes@168: 	return NULL;
nkeynes@168:     }
nkeynes@168: 
nkeynes@237:     f = fdopen(fd, "ro");
nkeynes@237: 
nkeynes@237: 
nkeynes@168:     /* try extensions */
nkeynes@168:     if( ext != NULL ) {
nkeynes@168: 	ext++; /* Skip the '.' */
nkeynes@168: 	for( i=0; gdrom_image_classes[i] != NULL; i++ ) {
nkeynes@237: 	    if( gdrom_image_classes[i]->extension != NULL &&
nkeynes@237: 		strcasecmp( gdrom_image_classes[i]->extension, ext ) == 0 ) {
nkeynes@168: 		extclz = gdrom_image_classes[i];
nkeynes@168: 		if( extclz->is_valid_file(f) ) {
nkeynes@168: 		    disc = extclz->open_image_file(filename, f);
nkeynes@168: 		    if( disc != NULL )
nkeynes@168: 			return disc;
nkeynes@168: 		}
nkeynes@168: 		break;
nkeynes@168: 	    }
nkeynes@168: 	}
nkeynes@168:     }
nkeynes@168: 
nkeynes@168:     /* Okay, fall back to magic */
nkeynes@237:     gboolean recognized = FALSE;
nkeynes@168:     for( i=0; gdrom_image_classes[i] != NULL; i++ ) {
nkeynes@168: 	if( gdrom_image_classes[i] != extclz &&
nkeynes@168: 	    gdrom_image_classes[i]->is_valid_file(f) ) {
nkeynes@237: 	    recognized = TRUE;
nkeynes@168: 	    disc = gdrom_image_classes[i]->open_image_file(filename, f);
nkeynes@168: 	    if( disc != NULL )
nkeynes@168: 		return disc;
nkeynes@168: 	}
nkeynes@168:     }
nkeynes@168: 
nkeynes@237:     if( !recognized ) {
nkeynes@237: 	ERROR( "Unable to open disc %s: Unsupported format", filename );
nkeynes@237:     }
nkeynes@168:     fclose(f);
nkeynes@168:     return NULL;
nkeynes@138: }
nkeynes@138: 
nkeynes@138: 
nkeynes@138: gdrom_disc_t gdrom_image_new( FILE *file )
nkeynes@138: {
nkeynes@138:     struct gdrom_disc *disc = (struct gdrom_disc *)calloc(1, sizeof(struct gdrom_disc));
nkeynes@138:     if( disc == NULL )
nkeynes@138: 	return NULL;
nkeynes@138:     disc->read_sectors = gdrom_image_read_sectors;
nkeynes@138:     disc->close = gdrom_image_destroy;
nkeynes@138:     disc->disc_type = IDE_DISC_CDROM;
nkeynes@138:     disc->file = file;
nkeynes@138:     return disc;
nkeynes@138: }
nkeynes@138: 
nkeynes@138: static void gdrom_image_destroy( gdrom_disc_t disc )
nkeynes@138: {
nkeynes@138:     if( disc->file != NULL ) {
nkeynes@138: 	fclose(disc->file);
nkeynes@138: 	disc->file = NULL;
nkeynes@138:     }
nkeynes@138:     free( disc );
nkeynes@138: }
nkeynes@138: 
nkeynes@152: static gdrom_error_t gdrom_image_read_sectors( gdrom_disc_t disc, uint32_t sector,
nkeynes@152: 					       uint32_t sector_count, int mode, char *buf,
nkeynes@152: 					       uint32_t *length )
nkeynes@138: {
nkeynes@152:     int i, file_offset, read_len;
nkeynes@152:     struct gdrom_track *track = NULL;
nkeynes@138: 
nkeynes@138:     for( i=0; i<disc->track_count; i++ ) {
nkeynes@138: 	if( disc->track[i].lba <= sector && 
nkeynes@142: 	    (sector + sector_count) <= (disc->track[i].lba + disc->track[i].sector_count) ) {
nkeynes@152: 	    track = &disc->track[i];
nkeynes@138: 	    break;
nkeynes@138: 	}
nkeynes@138:     }
nkeynes@152:     if( track == NULL )
nkeynes@143: 	return PKT_ERR_BADREAD;
nkeynes@152: 
nkeynes@152:     file_offset = track->offset + track->sector_size * (sector - track->lba);
nkeynes@152:     read_len = track->sector_size * sector_count;
nkeynes@152: 
nkeynes@152:     switch( mode ) {
nkeynes@152:     case GDROM_GD:
nkeynes@158: 	// Temporarily comment this out - it's wrong, but...
nkeynes@158: 	//	if( track->mode != GDROM_GD ) 
nkeynes@158: 	//    return PKT_ERR_BADREADMODE;
nkeynes@158: 	// break;
nkeynes@152:     case GDROM_MODE1:
nkeynes@245:     case GDROM_MODE2_XA1:
nkeynes@152: 	switch( track->mode ) {
nkeynes@152: 	case GDROM_MODE1:
nkeynes@152: 	case GDROM_MODE2_XA1:
nkeynes@154: 	    fseek( disc->file, file_offset, SEEK_SET );
nkeynes@152: 	    fread( buf, track->sector_size, sector_count, disc->file );
nkeynes@152: 	    break;
nkeynes@152: 	case GDROM_MODE2:
nkeynes@152: 	    read_len = sector_count * 2048;
nkeynes@154: 	    file_offset += 8; /* skip the subheader */
nkeynes@152: 	    while( sector_count > 0 ) {
nkeynes@154: 		fseek( disc->file, file_offset, SEEK_SET );
nkeynes@152: 		fread( buf, 2048, 1, disc->file );
nkeynes@152: 		file_offset += track->sector_size;
nkeynes@152: 		buf += 2048;
nkeynes@152: 		sector_count--;
nkeynes@152: 	    }
nkeynes@152: 	    break;
nkeynes@152: 	default:
nkeynes@152: 	    return PKT_ERR_BADREADMODE;
nkeynes@152: 	}
nkeynes@152: 	break;
nkeynes@152:     default:
nkeynes@143: 	return PKT_ERR_BADREADMODE;
nkeynes@152:     }
nkeynes@152: 	    
nkeynes@143:     *length = read_len;
nkeynes@143:     return PKT_ERR_OK;
nkeynes@138: }
nkeynes@138: 
nkeynes@138: uint32_t gdrom_read_sectors( uint32_t sector, uint32_t sector_count,
nkeynes@143: 			     int mode, char *buf, uint32_t *length )
nkeynes@138: {
nkeynes@138:     if( gdrom_disc == NULL )
nkeynes@143: 	return PKT_ERR_NODISC; /* No media */
nkeynes@143:     return gdrom_disc->read_sectors( gdrom_disc, sector, sector_count, mode, buf, length );
nkeynes@138: }
nkeynes@138: 
nkeynes@138: 
nkeynes@167: void gdrom_dump_disc_info( gdrom_disc_t disc ) {
nkeynes@138:     int i;
nkeynes@167:     int last_session = disc->track[disc->track_count-1].session;
nkeynes@167:     gboolean is_bootable = FALSE;
nkeynes@167: 
nkeynes@138:     INFO( "Disc ID: %s, %d tracks in %d sessions", disc->mcn, disc->track_count, 
nkeynes@138: 	  disc->track[disc->track_count-1].session + 1 );
nkeynes@167:     if( last_session > 0 ) {
nkeynes@167: 	/* Boot track is the first data track of the last session, provided that it 
nkeynes@167: 	 * cannot be a single-session disc.
nkeynes@167: 	 */
nkeynes@167: 	int boot_track = -1;
nkeynes@167: 	for( i=disc->track_count-1; i>=0 && disc->track[i].session == last_session; i-- ) {
nkeynes@167: 	    if( disc->track[i].flags & TRACK_DATA ) {
nkeynes@167: 		boot_track = i;
nkeynes@167: 	    }
nkeynes@167: 	}
nkeynes@167: 	if( boot_track != -1 ) {
nkeynes@167: 	    char boot_sector[2048];
nkeynes@167: 	    uint32_t length = sizeof(boot_sector);
nkeynes@167: 	    if( disc->read_sectors( disc, disc->track[boot_track].lba, 1, GDROM_MODE1,
nkeynes@167: 				    boot_sector, &length ) == PKT_ERR_OK ) {
nkeynes@167: 		bootstrap_dump(boot_sector, FALSE);
nkeynes@167: 		is_bootable = TRUE;
nkeynes@167: 	    }
nkeynes@167: 	}
nkeynes@138:     }
nkeynes@167:     if( !is_bootable )
nkeynes@167: 	WARN( "Disc does not appear to be bootable" );
nkeynes@138: }
nkeynes@138: 
nkeynes@149: gdrom_error_t gdrom_get_toc( char *buf ) 
nkeynes@138: {
nkeynes@142:     struct gdrom_toc *toc = (struct gdrom_toc *)buf;
nkeynes@142:     int i;
nkeynes@142: 
nkeynes@138:     if( gdrom_disc == NULL )
nkeynes@149: 	return PKT_ERR_NODISC;
nkeynes@142: 
nkeynes@142:     for( i=0; i<gdrom_disc->track_count; i++ ) {
nkeynes@142: 	toc->track[i] = htonl( gdrom_disc->track[i].lba ) | gdrom_disc->track[i].flags;
nkeynes@142:     }
nkeynes@142:     toc->first = 0x0100 | gdrom_disc->track[0].flags;
nkeynes@142:     toc->last = (gdrom_disc->track_count<<8) | gdrom_disc->track[i-1].flags;
nkeynes@142:     toc->leadout = htonl(gdrom_disc->track[i-1].lba + gdrom_disc->track[i-1].sector_count) |
nkeynes@142: 	gdrom_disc->track[i-1].flags;
nkeynes@142:     for( ;i<99; i++ )
nkeynes@142: 	toc->track[i] = 0xFFFFFFFF;
nkeynes@149:     return PKT_ERR_OK;
nkeynes@149: }
nkeynes@149: 
nkeynes@152: gdrom_error_t gdrom_get_info( char *buf, int session )
nkeynes@149: {
nkeynes@149:     if( gdrom_disc == NULL )
nkeynes@149: 	return PKT_ERR_NODISC;
nkeynes@149:     struct gdrom_track *last_track = &gdrom_disc->track[gdrom_disc->track_count-1];
nkeynes@149:     unsigned int end_of_disc = last_track->lba + last_track->sector_count;
nkeynes@152:     int i;
nkeynes@152:     buf[0] = 0x01; /* Disc status? */
nkeynes@149:     buf[1] = 0;
nkeynes@152: 
nkeynes@152:     if( session == 0 ) {
nkeynes@152: 	buf[2] = last_track->session+1; /* last session */
nkeynes@152: 	buf[3] = (end_of_disc >> 16) & 0xFF;
nkeynes@152: 	buf[4] = (end_of_disc >> 8) & 0xFF;
nkeynes@152: 	buf[5] = end_of_disc & 0xFF;
nkeynes@152: 	return PKT_ERR_OK;
nkeynes@152:     } else {
nkeynes@152: 	session--;
nkeynes@152: 	for( i=0; i<gdrom_disc->track_count; i++ ) {
nkeynes@152: 	    if( gdrom_disc->track[i].session == session ) {
nkeynes@152: 		buf[2] = i+1; /* first track of session */
nkeynes@152: 		buf[3] = (gdrom_disc->track[i].lba >> 16) & 0xFF;
nkeynes@152: 		buf[4] = (gdrom_disc->track[i].lba >> 8) & 0xFF;
nkeynes@152: 		buf[5] = gdrom_disc->track[i].lba & 0xFF;
nkeynes@152: 		return PKT_ERR_OK;
nkeynes@152: 	    }
nkeynes@152: 	}
nkeynes@152: 	return PKT_ERR_BADFIELD; /* No such session */
nkeynes@152:     }
nkeynes@152: 	
nkeynes@138: }
nkeynes@138: 
nkeynes@245: gdrom_track_t gdrom_get_track( int trackno ) {
nkeynes@245:     if( gdrom_disc == NULL || trackno < 1 || trackno > 99 ) {
nkeynes@245: 	return NULL;
nkeynes@245:     } else {
nkeynes@245: 	return &gdrom_disc->track[trackno-1];
nkeynes@245:     }
nkeynes@245: }
nkeynes@245: 
nkeynes@245: uint8_t gdrom_get_track_no_by_lba( uint32_t lba ) {
nkeynes@245:     int i;
nkeynes@245:     if( gdrom_disc != NULL ) {
nkeynes@245: 	for( i=0; i<gdrom_disc->track_count; i++ ) {
nkeynes@245: 	    if( gdrom_disc->track[i].lba <= lba && 
nkeynes@245: 		lba <= (gdrom_disc->track[i].lba + gdrom_disc->track[i].sector_count) ) {
nkeynes@245: 		return i+1;
nkeynes@245: 	    }
nkeynes@245: 	}
nkeynes@245:     }
nkeynes@245:     return -1;
nkeynes@245: }
nkeynes@245: 
nkeynes@138: void gdrom_mount_disc( gdrom_disc_t disc ) 
nkeynes@138: {
nkeynes@138:     gdrom_unmount_disc();
nkeynes@138:     gdrom_disc = disc;
nkeynes@138:     idereg.disc = disc->disc_type | IDE_DISC_READY;
nkeynes@167:     gdrom_dump_disc_info( disc );
nkeynes@138: }
nkeynes@138: 
nkeynes@138: gdrom_disc_t gdrom_mount_image( const gchar *filename )
nkeynes@138: {
nkeynes@138:     gdrom_disc_t disc = gdrom_image_open(filename);
nkeynes@138:     if( disc != NULL )
nkeynes@138: 	gdrom_mount_disc( disc );
nkeynes@138:     return disc;
nkeynes@138: }
nkeynes@138: 
nkeynes@138: void gdrom_unmount_disc( ) 
nkeynes@138: {
nkeynes@138:     if( gdrom_disc != NULL ) {
nkeynes@138: 	gdrom_disc->close(gdrom_disc);
nkeynes@138:     }
nkeynes@138:     gdrom_disc = NULL;
nkeynes@138:     idereg.disc = IDE_DISC_NONE;
nkeynes@138: }
nkeynes@138: 
nkeynes@138: gboolean gdrom_is_mounted( void ) 
nkeynes@138: {
nkeynes@138:     return gdrom_disc != NULL;
nkeynes@138: }