nkeynes@26: /**
nkeynes@561:  * $Id$
nkeynes@1:  *
nkeynes@26:  * File loading routines, mostly for loading demos without going through the
nkeynes@26:  * whole procedure of making a CD image for them.
nkeynes@26:  *
nkeynes@26:  * Copyright (c) 2005 Nathan Keynes.
nkeynes@26:  *
nkeynes@26:  * This program is free software; you can redistribute it and/or modify
nkeynes@26:  * it under the terms of the GNU General Public License as published by
nkeynes@26:  * the Free Software Foundation; either version 2 of the License, or
nkeynes@26:  * (at your option) any later version.
nkeynes@26:  *
nkeynes@26:  * This program is distributed in the hope that it will be useful,
nkeynes@26:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
nkeynes@26:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
nkeynes@26:  * GNU General Public License for more details.
nkeynes@1:  */
nkeynes@1: 
nkeynes@481: #include <unistd.h>
nkeynes@1: #include <stdio.h>
nkeynes@736: #include <string.h>
nkeynes@1: #include <fcntl.h>
nkeynes@1: #include <sys/stat.h>
nkeynes@1: #include <errno.h>
nkeynes@1: #include <stdint.h>
nkeynes@105: #include <elf.h>
nkeynes@26: #include "mem.h"
nkeynes@26: #include "bootstrap.h"
nkeynes@171: #include "dreamcast.h"
nkeynes@450: #include "config.h"
nkeynes@427: #include "loader.h"
nkeynes@481: 
nkeynes@26: char bootstrap_magic[32] = "SEGA SEGAKATANA SEGA ENTERPRISES";
nkeynes@1: char iso_magic[6] = "\001CD001";
nkeynes@26: char *file_loader_extensions[][2] = { 
nkeynes@736:         { "sbi", "Self Boot Inducer" },
nkeynes@736:         { "bin", "SH4 Bin file" },
nkeynes@736:         { NULL, NULL } };
nkeynes@26: 
nkeynes@1: #define CDI_V2 0x80000004
nkeynes@1: #define CDI_V3 0x80000005
nkeynes@1: 
nkeynes@543: gboolean file_load_elf_fd( const gchar *filename, int fd );
nkeynes@427: 
nkeynes@427: 
nkeynes@26: gboolean file_load_magic( const gchar *filename )
nkeynes@1: {
nkeynes@1:     char buf[32];
nkeynes@1:     struct stat st;
nkeynes@446:     gboolean result = TRUE;
nkeynes@736: 
nkeynes@1:     int fd = open( filename, O_RDONLY );
nkeynes@1:     if( fd == -1 ) {
nkeynes@26:         return FALSE;
nkeynes@1:     }
nkeynes@736: 
nkeynes@1:     fstat( fd, &st );
nkeynes@736: 
nkeynes@1:     /* begin magic */
nkeynes@1:     if( read( fd, buf, 32 ) != 32 ) {
nkeynes@1:         ERROR( "Unable to read from file '%s'", filename );
nkeynes@1:         close(fd);
nkeynes@26:         return FALSE;
nkeynes@1:     }
nkeynes@26:     if( memcmp( buf, bootstrap_magic, 32 ) == 0 ) {
nkeynes@1:         /* we have a DC bootstrap */
nkeynes@1:         if( st.st_size == BOOTSTRAP_SIZE ) {
nkeynes@502:             sh4ptr_t load = mem_get_region( BOOTSTRAP_LOAD_ADDR );
nkeynes@1:             lseek( fd, 0, SEEK_SET );
nkeynes@1:             read( fd, load, BOOTSTRAP_SIZE );
nkeynes@171:             bootstrap_dump( load, TRUE );
nkeynes@736:             dreamcast_program_loaded( filename, BOOTSTRAP_LOAD_ADDR + 0x300 );
nkeynes@1:         } else {
nkeynes@1:             /* look for a valid ISO9660 header */
nkeynes@1:             lseek( fd, 32768, SEEK_SET );
nkeynes@1:             read( fd, buf, 8 );
nkeynes@1:             if( memcmp( buf, iso_magic, 6 ) == 0 ) {
nkeynes@1:                 /* Alright, got it */
nkeynes@1075:                 WARN( "ISO images not supported yet" );
nkeynes@1:             }
nkeynes@1:         }
nkeynes@26:     } else if( memcmp( buf, "PK\x03\x04", 4 ) == 0 ) {
nkeynes@736:         /* ZIP file, aka SBI file */
nkeynes@736:         WARN( "SBI files not supported yet" );
nkeynes@736:         result = FALSE;
nkeynes@294:     } else if( memcmp( buf, DREAMCAST_SAVE_MAGIC, 16 ) == 0 ) {
nkeynes@736:         /* Save state */
nkeynes@736:         result = (dreamcast_load_state( filename )==0);
nkeynes@105:     } else if( buf[0] == 0x7F && buf[1] == 'E' && 
nkeynes@736:             buf[2] == 'L' && buf[3] == 'F' ) {
nkeynes@736:         /* ELF binary */
nkeynes@736:         lseek( fd, 0, SEEK_SET );
nkeynes@736:         result = file_load_elf_fd( filename, fd );
nkeynes@1:     } else {
nkeynes@736:         result = FALSE;
nkeynes@446:     }
nkeynes@1:     close(fd);
nkeynes@446:     return result;
nkeynes@1: }
nkeynes@19: 
nkeynes@543: void file_load_postload( const gchar *filename, int pc )
nkeynes@110: {
nkeynes@1036:     gchar *bootstrap_file = lxdream_get_global_config_path_value(CONFIG_BOOTSTRAP);
nkeynes@825:     if( bootstrap_file != NULL && bootstrap_file[0] != '\0' ) {
nkeynes@736:         /* Load in a bootstrap before the binary, to initialize everything
nkeynes@736:          * correctly
nkeynes@736:          */
nkeynes@543:         if( mem_load_block( bootstrap_file, BOOTSTRAP_LOAD_ADDR, BOOTSTRAP_SIZE ) == 0 ) {
nkeynes@736:             dreamcast_program_loaded( filename, BOOTSTRAP_LOAD_ADDR+0x300 );
nkeynes@1036:             g_free(bootstrap_file);
nkeynes@736:             return;
nkeynes@736:         }
nkeynes@88:     }
nkeynes@543:     dreamcast_program_loaded( filename, pc );
nkeynes@1036:     g_free(bootstrap_file);
nkeynes@110: }    
nkeynes@110: 
nkeynes@110: 
nkeynes@427: gboolean file_load_binary( const gchar *filename )
nkeynes@110: {
nkeynes@110:     /* Load the binary itself */
nkeynes@427:     if(  mem_load_block( filename, BINARY_LOAD_ADDR, -1 ) == 0 ) {
nkeynes@736:         file_load_postload( filename, BINARY_LOAD_ADDR );
nkeynes@736:         return TRUE;
nkeynes@427:     } else {
nkeynes@736:         return FALSE;
nkeynes@427:     }
nkeynes@19: }
nkeynes@105: 
nkeynes@543: gboolean file_load_elf_fd( const gchar *filename, int fd ) 
nkeynes@105: {
nkeynes@105:     Elf32_Ehdr head;
nkeynes@105:     Elf32_Phdr phdr;
nkeynes@105:     int i;
nkeynes@105: 
nkeynes@105:     if( read( fd, &head, sizeof(head) ) != sizeof(head) )
nkeynes@736:         return FALSE;
nkeynes@105:     if( head.e_ident[EI_CLASS] != ELFCLASS32 ||
nkeynes@736:             head.e_ident[EI_DATA] != ELFDATA2LSB ||
nkeynes@736:             head.e_ident[EI_VERSION] != 1 ||
nkeynes@736:             head.e_type != ET_EXEC ||
nkeynes@736:             head.e_machine != EM_SH ||
nkeynes@736:             head.e_version != 1 ) {
nkeynes@736:         ERROR( "File is not an SH4 ELF executable file" );
nkeynes@736:         return FALSE;
nkeynes@105:     }
nkeynes@105: 
nkeynes@105:     /* Program headers */
nkeynes@105:     for( i=0; i<head.e_phnum; i++ ) {
nkeynes@736:         lseek( fd, head.e_phoff + i*head.e_phentsize, SEEK_SET );
nkeynes@736:         read( fd, &phdr, sizeof(phdr) );
nkeynes@736:         if( phdr.p_type == PT_LOAD ) {
nkeynes@736:             lseek( fd, phdr.p_offset, SEEK_SET );
nkeynes@736:             sh4ptr_t target = mem_get_region( phdr.p_vaddr );
nkeynes@736:             read( fd, target, phdr.p_filesz );
nkeynes@736:             if( phdr.p_memsz > phdr.p_filesz ) {
nkeynes@736:                 memset( target + phdr.p_filesz, 0, phdr.p_memsz - phdr.p_filesz );
nkeynes@736:             }
nkeynes@736:             INFO( "Loaded %d bytes to %08X", phdr.p_filesz, phdr.p_vaddr );
nkeynes@736:         }
nkeynes@105:     }
nkeynes@736: 
nkeynes@543:     file_load_postload( filename, head.e_entry );
nkeynes@446:     return TRUE;
nkeynes@105: }