Search
lxdream.org :: lxdream/src/gdrom/nrg.c
lxdream 0.9.1
released Jun 29
Download Now
filename src/gdrom/nrg.c
changeset 736:a02d1475ccfd
prev678:35eb00945316
next832:40e5bb525c4e
author nkeynes
date Mon Jul 14 07:44:42 2008 +0000 (13 years ago)
permissions -rw-r--r--
last change Re-indent everything consistently
Fix include guards for consistency as well
file annotate diff log raw
nkeynes@138
     1
/**
nkeynes@561
     2
 * $Id$
nkeynes@138
     3
 *
nkeynes@138
     4
 * Nero (NRG) CD file format. File information stolen shamelessly from
nkeynes@138
     5
 * libcdio.
nkeynes@138
     6
 *
nkeynes@138
     7
 * Copyright (c) 2005 Nathan Keynes.
nkeynes@138
     8
 *
nkeynes@138
     9
 * This program is free software; you can redistribute it and/or modify
nkeynes@138
    10
 * it under the terms of the GNU General Public License as published by
nkeynes@138
    11
 * the Free Software Foundation; either version 2 of the License, or
nkeynes@138
    12
 * (at your option) any later version.
nkeynes@138
    13
 *
nkeynes@138
    14
 * This program is distributed in the hope that it will be useful,
nkeynes@138
    15
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
nkeynes@138
    16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nkeynes@138
    17
 * GNU General Public License for more details.
nkeynes@138
    18
 */
nkeynes@138
    19
nkeynes@138
    20
#include <assert.h>
nkeynes@138
    21
#include <stdio.h>
nkeynes@138
    22
#include <errno.h>
nkeynes@514
    23
#include <glib/gtypes.h>
nkeynes@678
    24
#include "gdrom/gddriver.h"
nkeynes@138
    25
#include "dream.h"
nkeynes@138
    26
nkeynes@168
    27
static gboolean nrg_image_is_valid( FILE *f );
nkeynes@168
    28
static gdrom_disc_t nrg_image_open( const gchar *filename, FILE *f );
nkeynes@168
    29
nkeynes@168
    30
struct gdrom_image_class nrg_image_class = { "Nero", "nrg", 
nkeynes@736
    31
        nrg_image_is_valid, nrg_image_open };
nkeynes@168
    32
nkeynes@138
    33
#define NERO_V55_ID  0x4e455235 
nkeynes@138
    34
#define NERO_V50_ID  0x4e45524f 
nkeynes@138
    35
nkeynes@138
    36
/* Courtesy of libcdio */
nkeynes@138
    37
/* 5.0 or earlier */
nkeynes@138
    38
#define NERO_ID  0x4e45524f  /* Nero pre 5.5.x */
nkeynes@138
    39
#define CUES_ID  0x43554553  /* Nero pre version 5.5.x-6.x */
nkeynes@138
    40
#define DAOI_ID  0x44414f49
nkeynes@138
    41
#define ETNF_ID  0x45544e46
nkeynes@138
    42
#define SINF_ID  0x53494e46  /* Session information */
nkeynes@138
    43
#define END_ID  0x454e4421
nkeynes@138
    44
/* 5.5+ only */
nkeynes@138
    45
#define NER5_ID  0x4e455235  /* Nero version 5.5.x */
nkeynes@138
    46
#define CDTX_ID  0x43445458  /* CD TEXT */
nkeynes@138
    47
#define CUEX_ID  0x43554558  /* Nero version 5.5.x-6.x */
nkeynes@138
    48
#define DAOX_ID  0x44414f58  /* Nero version 5.5.x-6.x */
nkeynes@138
    49
#define ETN2_ID  0x45544e32
nkeynes@138
    50
#define MTYP_ID  0x4d545950  /* Disc Media type? */
nkeynes@138
    51
nkeynes@138
    52
nkeynes@138
    53
union nrg_footer {
nkeynes@138
    54
    struct nrg_footer_v50 {
nkeynes@736
    55
        uint32_t dummy;
nkeynes@736
    56
        uint32_t id;
nkeynes@736
    57
        uint32_t offset;
nkeynes@138
    58
    } v50;
nkeynes@138
    59
    struct nrg_footer_v55 {
nkeynes@736
    60
        uint32_t id;
nkeynes@736
    61
        uint64_t offset;
nkeynes@138
    62
    } v55;
nkeynes@138
    63
};
nkeynes@138
    64
nkeynes@138
    65
struct nrg_chunk {
nkeynes@138
    66
    uint32_t id;
nkeynes@138
    67
    uint32_t length;
nkeynes@138
    68
};
nkeynes@138
    69
nkeynes@138
    70
struct nrg_etnf {
nkeynes@138
    71
    uint32_t offset;
nkeynes@138
    72
    uint32_t length;
nkeynes@138
    73
    uint32_t mode;
nkeynes@138
    74
    uint32_t lba;
nkeynes@138
    75
    uint32_t padding;
nkeynes@138
    76
};
nkeynes@138
    77
nkeynes@514
    78
struct nrg_etn2 {
nkeynes@514
    79
    uint64_t offset;
nkeynes@514
    80
    uint64_t length;
nkeynes@514
    81
    uint32_t mode;
nkeynes@514
    82
    uint32_t lba;
nkeynes@514
    83
    uint64_t padding;
nkeynes@514
    84
};
nkeynes@514
    85
nkeynes@138
    86
struct nrg_cues {
nkeynes@138
    87
    uint8_t type;
nkeynes@138
    88
    uint8_t track;
nkeynes@138
    89
    uint8_t control;
nkeynes@138
    90
    uint8_t pad;
nkeynes@138
    91
    uint32_t addr;
nkeynes@138
    92
};
nkeynes@138
    93
nkeynes@138
    94
struct nrg_daoi {
nkeynes@138
    95
    uint32_t length;
nkeynes@138
    96
    char mcn[14];
nkeynes@138
    97
    uint8_t disc_mode;
nkeynes@138
    98
    uint8_t unknown[2]; /* always 01 01? */
nkeynes@138
    99
    uint8_t track_count;
nkeynes@138
   100
    struct nrg_daoi_track {
nkeynes@736
   101
        char unknown[10];
nkeynes@736
   102
        uint32_t sector_size __attribute__((packed)); /* Always 0? */
nkeynes@736
   103
        uint8_t mode;
nkeynes@736
   104
        uint8_t unknown2[3]; /* Always 00 00 01? */
nkeynes@736
   105
        uint32_t pregap __attribute__((packed));
nkeynes@736
   106
        uint32_t offset __attribute__((packed));
nkeynes@736
   107
        uint32_t end __attribute__((packed));
nkeynes@138
   108
    } track[0];
nkeynes@138
   109
} __attribute__((packed));
nkeynes@138
   110
nkeynes@514
   111
struct nrg_daox {
nkeynes@514
   112
    uint32_t length;
nkeynes@514
   113
    char mcn[14];
nkeynes@514
   114
    uint8_t disc_mode;
nkeynes@514
   115
    uint8_t unknown[2]; /* always 01 01? */
nkeynes@514
   116
    uint8_t track_count;
nkeynes@514
   117
    struct nrg_daox_track {
nkeynes@736
   118
        char unknown[10];
nkeynes@736
   119
        uint32_t sector_size __attribute__((packed)); /* Always 0? */
nkeynes@736
   120
        uint8_t mode;
nkeynes@736
   121
        uint8_t unknown2[3]; /* Always 00 00 01? */
nkeynes@736
   122
        uint64_t pregap __attribute__((packed));
nkeynes@736
   123
        uint64_t offset __attribute__((packed));
nkeynes@736
   124
        uint64_t end __attribute__((packed));
nkeynes@514
   125
    } track[0];
nkeynes@514
   126
} __attribute__((packed));
nkeynes@514
   127
nkeynes@138
   128
/**
nkeynes@138
   129
 * Convert an 8-bit BCD number to normal integer form. 
nkeynes@138
   130
 * Eg, 0x79 => 79
nkeynes@138
   131
 */
nkeynes@138
   132
uint8_t static bcd_to_uint8( uint8_t bcd )
nkeynes@138
   133
{
nkeynes@138
   134
    return (bcd & 0x0F) + (((bcd & 0xF0)>>4)*10);
nkeynes@138
   135
}
nkeynes@138
   136
nkeynes@138
   137
nkeynes@138
   138
/**
nkeynes@138
   139
 * Convert a 32 bit MSF address (BCD coded) to the
nkeynes@138
   140
 * equivalent LBA form. 
nkeynes@138
   141
 * Eg, 0x
nkeynes@138
   142
 */
nkeynes@138
   143
uint32_t static msf_to_lba( uint32_t msf )
nkeynes@138
   144
{
nkeynes@514
   145
    msf = GUINT32_FROM_BE(msf);
nkeynes@138
   146
    int f = bcd_to_uint8(msf);
nkeynes@138
   147
    int s = bcd_to_uint8(msf>>8);
nkeynes@138
   148
    int m = bcd_to_uint8(msf>>16);
nkeynes@138
   149
    return (m * 60 + s) * 75 + f;
nkeynes@138
   150
nkeynes@138
   151
}
nkeynes@138
   152
nkeynes@138
   153
uint32_t static nrg_track_mode( uint8_t mode )
nkeynes@138
   154
{
nkeynes@138
   155
    switch( mode ) {
nkeynes@138
   156
    case 0: return GDROM_MODE1;
nkeynes@644
   157
    case 2: return GDROM_MODE2_FORM1;
nkeynes@644
   158
    case 3: return GDROM_SEMIRAW_MODE2;
nkeynes@138
   159
    case 7: return GDROM_CDDA;
nkeynes@138
   160
    default: 
nkeynes@736
   161
        ERROR( "Unrecognized track mode %d in Nero image", mode );
nkeynes@736
   162
        return -1;
nkeynes@138
   163
    }
nkeynes@138
   164
}
nkeynes@138
   165
nkeynes@168
   166
static gboolean nrg_image_is_valid( FILE *f )
nkeynes@138
   167
{
nkeynes@168
   168
    union nrg_footer footer;
nkeynes@168
   169
nkeynes@168
   170
    fseek( f, -12, SEEK_END );
nkeynes@168
   171
    fread( &footer, sizeof(footer), 1, f );
nkeynes@514
   172
    if( GUINT32_FROM_BE(footer.v50.id) == NERO_V50_ID ||
nkeynes@736
   173
            GUINT32_FROM_BE(footer.v55.id) == NERO_V55_ID ) {
nkeynes@736
   174
        return TRUE;
nkeynes@168
   175
    } else {
nkeynes@736
   176
        return FALSE;
nkeynes@168
   177
    }
nkeynes@168
   178
}
nkeynes@168
   179
nkeynes@168
   180
static gdrom_disc_t nrg_image_open( const gchar *filename, FILE *f )
nkeynes@168
   181
{
nkeynes@138
   182
    union nrg_footer footer;
nkeynes@138
   183
    struct nrg_chunk chunk;
nkeynes@138
   184
    struct nrg_daoi *dao;
nkeynes@514
   185
    struct nrg_daox *daox;
nkeynes@514
   186
    struct nrg_etnf *etnf;
nkeynes@514
   187
    struct nrg_etn2 *etn2;
nkeynes@138
   188
    gdrom_disc_t disc;
nkeynes@342
   189
    gdrom_image_t image;
nkeynes@138
   190
    gboolean end = FALSE;
nkeynes@514
   191
    uint32_t chunk_id;
nkeynes@138
   192
    int session_id = 0;
nkeynes@138
   193
    int session_track_id = 0;
nkeynes@138
   194
    int track_id = 0;
nkeynes@138
   195
    int cue_track_id = 0, cue_track_count = 0;
nkeynes@514
   196
    int i, count;
nkeynes@138
   197
nkeynes@138
   198
    fseek( f, -12, SEEK_END );
nkeynes@138
   199
    fread( &footer, sizeof(footer), 1, f );
nkeynes@514
   200
    if( GUINT32_FROM_BE(footer.v50.id) == NERO_V50_ID ) {
nkeynes@736
   201
        INFO( "Loading Nero 5.0 image" );
nkeynes@736
   202
        fseek( f, GUINT32_FROM_BE(footer.v50.offset), SEEK_SET );
nkeynes@514
   203
    } else if( GUINT32_FROM_BE(footer.v55.id) == NERO_V55_ID ) {
nkeynes@736
   204
        INFO( "Loading Nero 5.5+ image" );
nkeynes@736
   205
        fseek( f, (uint32_t)GUINT64_FROM_BE(footer.v55.offset), SEEK_SET );
nkeynes@138
   206
    } else {
nkeynes@736
   207
        /* Not a (recognized) Nero image */
nkeynes@736
   208
        return NULL;
nkeynes@138
   209
    }
nkeynes@736
   210
nkeynes@464
   211
    disc = gdrom_image_new(filename, f);
nkeynes@138
   212
    if( disc == NULL ) {
nkeynes@736
   213
        ERROR("Unable to allocate memory!");
nkeynes@736
   214
        return NULL;
nkeynes@138
   215
    }
nkeynes@342
   216
    image = (gdrom_image_t)disc;
nkeynes@138
   217
nkeynes@138
   218
    do {
nkeynes@736
   219
        fread( &chunk, sizeof(chunk), 1, f );
nkeynes@736
   220
        chunk.length = GUINT32_FROM_BE(chunk.length);
nkeynes@736
   221
        char data[chunk.length];
nkeynes@736
   222
        fread( data, chunk.length, 1, f );
nkeynes@736
   223
        chunk_id = GUINT32_FROM_BE(chunk.id);
nkeynes@736
   224
        switch( chunk_id ) {
nkeynes@736
   225
        case CUES_ID:
nkeynes@736
   226
        case CUEX_ID:
nkeynes@736
   227
            cue_track_id = track_id;
nkeynes@736
   228
            cue_track_count = ((chunk.length / sizeof(struct nrg_cues)) >> 1) - 1;
nkeynes@736
   229
            track_id += cue_track_count;
nkeynes@736
   230
            for( i=0; i<chunk.length; i+= sizeof(struct nrg_cues) ) {
nkeynes@736
   231
                struct nrg_cues *cue = (struct nrg_cues *)(data+i);
nkeynes@736
   232
                int track = 0;
nkeynes@736
   233
                uint32_t lba;
nkeynes@736
   234
                if( chunk_id == CUEX_ID ) {
nkeynes@736
   235
                    lba = GUINT32_FROM_BE( cue->addr ) + GDROM_PREGAP;
nkeynes@736
   236
                } else {
nkeynes@736
   237
                    lba = msf_to_lba( cue->addr );
nkeynes@736
   238
                }
nkeynes@736
   239
                if( cue->track == 0 )
nkeynes@736
   240
                    continue; /* Track 0. Leadin? always 0? */
nkeynes@736
   241
                if( cue->track == 0xAA ) { /* end of disc */
nkeynes@736
   242
                    image->track[track_id-1].sector_count =
nkeynes@736
   243
                        lba - image->track[track_id-1].lba;
nkeynes@736
   244
                } else {
nkeynes@736
   245
                    track = cue_track_id + bcd_to_uint8(cue->track) - 1;
nkeynes@736
   246
                    if( (cue->control & 0x01) == 0 ) { 
nkeynes@736
   247
                        /* Pre-gap address. */
nkeynes@736
   248
                        if( track != 0 ) {
nkeynes@736
   249
                            image->track[track-1].sector_count = 
nkeynes@736
   250
                                lba - image->track[track-1].lba;
nkeynes@736
   251
                        }
nkeynes@736
   252
                    } else { /* Track-start address */
nkeynes@736
   253
                        image->track[track].lba = lba;
nkeynes@736
   254
                        image->track[track].flags = cue->type;
nkeynes@736
   255
                    }
nkeynes@736
   256
                }
nkeynes@736
   257
            }
nkeynes@736
   258
            break;
nkeynes@736
   259
        case DAOI_ID:
nkeynes@736
   260
            dao = (struct nrg_daoi *)data;
nkeynes@736
   261
            memcpy( image->mcn, dao->mcn, 13 );
nkeynes@736
   262
            image->mcn[13] = '\0';
nkeynes@736
   263
            assert( dao->track_count * 30 + 22 == chunk.length );
nkeynes@736
   264
            assert( dao->track_count == cue_track_count );
nkeynes@736
   265
            for( i=0; i<dao->track_count; i++ ) {
nkeynes@736
   266
                image->track[cue_track_id].sector_size = GUINT32_FROM_BE(dao->track[i].sector_size);
nkeynes@736
   267
                image->track[cue_track_id].offset = GUINT32_FROM_BE(dao->track[i].offset);
nkeynes@736
   268
                image->track[cue_track_id].mode = nrg_track_mode( dao->track[i].mode );
nkeynes@736
   269
                image->track[cue_track_id].sector_count =
nkeynes@736
   270
                    (GUINT32_FROM_BE(dao->track[i].end) - GUINT32_FROM_BE(dao->track[i].offset))/
nkeynes@736
   271
                    GUINT32_FROM_BE(dao->track[i].sector_size);
nkeynes@736
   272
                cue_track_id++;
nkeynes@736
   273
            }
nkeynes@736
   274
            break;
nkeynes@736
   275
        case DAOX_ID:
nkeynes@736
   276
            daox = (struct nrg_daox *)data;
nkeynes@736
   277
            memcpy( image->mcn, daox->mcn, 13 );
nkeynes@736
   278
            image->mcn[13] = '\0';
nkeynes@736
   279
            assert( daox->track_count * 42 + 22 == chunk.length );
nkeynes@736
   280
            assert( daox->track_count == cue_track_count );
nkeynes@736
   281
            for( i=0; i<daox->track_count; i++ ) {
nkeynes@736
   282
                image->track[cue_track_id].sector_size = GUINT32_FROM_BE(daox->track[i].sector_size);
nkeynes@736
   283
                image->track[cue_track_id].offset = GUINT64_FROM_BE(daox->track[i].offset);
nkeynes@736
   284
                image->track[cue_track_id].mode = nrg_track_mode( daox->track[i].mode );
nkeynes@736
   285
                image->track[cue_track_id].sector_count =
nkeynes@736
   286
                    (GUINT64_FROM_BE(daox->track[i].end) - GUINT64_FROM_BE(daox->track[i].offset))/
nkeynes@736
   287
                    GUINT32_FROM_BE(daox->track[i].sector_size);
nkeynes@736
   288
                cue_track_id++;
nkeynes@736
   289
            }
nkeynes@736
   290
            break;
nkeynes@514
   291
nkeynes@736
   292
        case SINF_ID: 
nkeynes@736
   293
            /* Data is a single 32-bit number representing number of tracks in session */
nkeynes@736
   294
            i = GUINT32_FROM_BE( *(uint32_t *)data );
nkeynes@736
   295
            while( i-- > 0 )
nkeynes@736
   296
                image->track[session_track_id++].session = session_id;
nkeynes@736
   297
            session_id++;
nkeynes@736
   298
            break;
nkeynes@736
   299
        case ETNF_ID:
nkeynes@736
   300
            etnf = (struct nrg_etnf *)data;
nkeynes@736
   301
            count = chunk.length / sizeof(struct nrg_etnf);
nkeynes@736
   302
            for( i=0; i < count; i++, etnf++ ) {
nkeynes@736
   303
                image->track[track_id].offset = GUINT32_FROM_BE(etnf->offset);
nkeynes@736
   304
                image->track[track_id].lba = GUINT32_FROM_BE(etnf->lba) + (i+1)*GDROM_PREGAP;
nkeynes@736
   305
                image->track[track_id].mode = nrg_track_mode( GUINT32_FROM_BE(etnf->mode) );
nkeynes@736
   306
                if( image->track[track_id].mode == -1 ) {
nkeynes@736
   307
                    gdrom_image_destroy_no_close(disc);
nkeynes@736
   308
                    return NULL;
nkeynes@736
   309
                }
nkeynes@736
   310
                if( image->track[track_id].mode == GDROM_CDDA )
nkeynes@736
   311
                    image->track[track_id].flags = 0x01;
nkeynes@736
   312
                else
nkeynes@736
   313
                    image->track[track_id].flags = 0x01 | TRACK_DATA;
nkeynes@736
   314
                image->track[track_id].sector_size = GDROM_SECTOR_SIZE(image->track[track_id].mode);
nkeynes@736
   315
                image->track[track_id].sector_count = GUINT32_FROM_BE(etnf->length) / 
nkeynes@736
   316
                image->track[track_id].sector_size;
nkeynes@736
   317
                track_id++;
nkeynes@736
   318
            }
nkeynes@736
   319
            break;
nkeynes@736
   320
        case ETN2_ID:
nkeynes@736
   321
            etn2 = (struct nrg_etn2 *)data;
nkeynes@736
   322
            count = chunk.length / sizeof(struct nrg_etn2);
nkeynes@736
   323
            for( i=0; i < count; i++, etn2++ ) {
nkeynes@736
   324
                image->track[track_id].offset = (uint32_t)GUINT64_FROM_BE(etn2->offset);
nkeynes@736
   325
                image->track[track_id].lba = GUINT32_FROM_BE(etn2->lba) + (i+1)*GDROM_PREGAP;
nkeynes@736
   326
                image->track[track_id].mode = nrg_track_mode( GUINT32_FROM_BE(etn2->mode) );
nkeynes@736
   327
                if( image->track[track_id].mode == -1 ) {
nkeynes@736
   328
                    gdrom_image_destroy_no_close(disc);
nkeynes@736
   329
                    return NULL;
nkeynes@736
   330
                }
nkeynes@736
   331
                if( image->track[track_id].mode == GDROM_CDDA )
nkeynes@736
   332
                    image->track[track_id].flags = 0x01;
nkeynes@736
   333
                else
nkeynes@736
   334
                    image->track[track_id].flags = 0x01 | TRACK_DATA;
nkeynes@736
   335
                image->track[track_id].sector_size = GDROM_SECTOR_SIZE(image->track[track_id].mode);
nkeynes@736
   336
                image->track[track_id].sector_count = (uint32_t)(GUINT64_FROM_BE(etn2->length) / 
nkeynes@736
   337
                        image->track[track_id].sector_size);
nkeynes@736
   338
                track_id++;
nkeynes@736
   339
            }
nkeynes@736
   340
            break;
nkeynes@736
   341
nkeynes@736
   342
        case END_ID:
nkeynes@736
   343
            end = TRUE;
nkeynes@736
   344
            break;
nkeynes@736
   345
        }
nkeynes@138
   346
    } while( !end );
nkeynes@342
   347
    image->track_count = track_id;
nkeynes@138
   348
    return disc;
nkeynes@138
   349
}
nkeynes@138
   350
nkeynes@138
   351
nkeynes@342
   352
.