usbloadergx/source/mload/mload.c

/* mload.c (for PPC) (c) 2009, Hermes

  This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#include "mload.h"
#include "dip_plugin.h"
#include <malloc.h>

#include "gecko.h"

static const char mload_fs[] ATTRIBUTE_ALIGN( 32 ) = "/dev/mload";
//static u32 patch_datas[8] ATTRIBUTE_ALIGN(32);

static s32 mload_fd = -1;
static s32 hid = -1;
//static data_elf my_data_elf;
//static int thread_id = -1;
//void *external_ehcmodule= NULL;
//int size_external_ehcmodule=0;

/*--------------------------------------------------------------------------------------------------------------*/

// to init/test if the device is running

int mload_init()
{
    int n;

    if ( hid < 0 ) hid = iosCreateHeap( 0x800 );

    if ( hid < 0 )
    {
        if ( mload_fd >= 0 )
            IOS_Close( mload_fd );

        mload_fd = -1;

        return hid;
    }

    if ( mload_fd >= 0 )
    {
        return 0;
    }

    for ( n = 0; n < 20; n++ ) // try 5 seconds
    {
        mload_fd = IOS_Open( mload_fs, 0 );

        if ( mload_fd >= 0 ) break;

        usleep( 250*1000 );
    }

    if ( mload_fd < 0 )
    {

        if ( hid >= 0 )
        {
            iosDestroyHeap( hid );
            hid = -1;
        }
    }

    return mload_fd;
}

/*--------------------------------------------------------------------------------------------------------------*/

// to close the device (remember call it when rebooting the IOS!)

int mload_close()
{
    int ret;

    if ( hid >= 0 )
    {
        iosDestroyHeap( hid );
        hid = -1;
    }

    if ( mload_fd < 0 ) return -1;

    ret = IOS_Close( mload_fd );

    mload_fd = -1;

    return ret;
}

/*--------------------------------------------------------------------------------------------------------------*/

// to get the thread id of mload

int mload_get_thread_id()
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_MLOAD_THREAD_ID, ":" );

    return ret;

}

/*--------------------------------------------------------------------------------------------------------------*/

// get the base and the size of the memory readable/writable to load modules

int mload_get_load_base( u32 *starlet_base, int *size )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GET_LOAD_BASE, ":ii", starlet_base, size );

    return ret;

}

/*--------------------------------------------------------------------------------------------------------------*/

// load and run a module from starlet (it need to allocate MEM2 to send the elf file)
// the module must be a elf made with stripios

int mload_module( void *addr, int len )
{
    int ret;
    void *buf = NULL;


    if ( mload_init() < 0 ) return -1;

    if ( hid >= 0 )
    {
        iosDestroyHeap( hid );
        hid = -1;
    }

    hid = iosCreateHeap( len + 0x800 );

    if ( hid < 0 ) return hid;

    buf = iosAlloc( hid, len );

    if ( !buf ) {ret = -1; goto out;}


    memcpy( buf, addr, len );

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_LOAD_MODULE, ":d", buf, len );

    if ( ret < 0 ) goto out;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_RUN_MODULE, ":" );

    if ( ret < 0 ) {ret = -666; goto out;}

out:
    if ( hid >= 0 )
    {
        iosDestroyHeap( hid );
        hid = -1;
    }

    return ret;

}

/*--------------------------------------------------------------------------------------------------------------*/

// load a module from the PPC
// the module must be a elf made with stripios

int mload_elf( void *my_elf, data_elf *data_elf )
{
    int n, m;
    int p;
    u8 *adr;
    u32 elf = ( u32 ) my_elf;

    if ( elf & 3 ) return -1; // aligned to 4 please!

    elfheader *head = ( void * ) elf;
    elfphentry *entries;

    if ( head->ident0 != 0x7F454C46 ) return -1;
    if ( head->ident1 != 0x01020161 ) return -1;
    if ( head->ident2 != 0x01000000 ) return -1;

    p = head->phoff;

    data_elf->start = ( void * )  head->entry;

    for ( n = 0; n < head->phnum; n++ )
    {
        entries = ( void * ) ( elf + p );
        p += sizeof( elfphentry );

        if ( entries->type == 4 )
        {
            adr = ( void * ) ( elf + entries->offset );

            if ( getbe32( 0 ) != 0 ) return -2; // bad info (sure)

            for ( m = 4; m < entries->memsz; m += 8 )
            {
                switch ( getbe32( m ) )
                {
                    case 0x9:
                        data_elf->start = ( void * ) getbe32( m + 4 );
                        break;
                    case 0x7D:
                        data_elf->prio = getbe32( m + 4 );
                        break;
                    case 0x7E:
                        data_elf->size_stack = getbe32( m + 4 );
                        break;
                    case 0x7F:
                        data_elf->stack = ( void * ) ( getbe32( m + 4 ) );
                        break;

                }

            }

        }
        else if ( entries->type == 1  && entries->memsz != 0 && entries->vaddr != 0 )
        {

            if ( mload_memset( ( void * ) entries->vaddr, 0, entries->memsz ) < 0 ) return -1;
            if ( mload_seek( entries->vaddr, SEEK_SET ) < 0 ) return -1;
            if ( mload_write( ( void * ) ( elf + entries->offset ), entries->filesz ) < 0 ) return -1;

        }
    }

    return 0;
}

/*--------------------------------------------------------------------------------------------------------------*/

// run one thread (you can use to load modules or binary files)

int mload_run_thread( void *starlet_addr, void *starlet_top_stack, int stack_size, int priority )
{
    int ret;


    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_RUN_THREAD, "iiii:", starlet_addr, starlet_top_stack, stack_size, priority );


    return ret;
}

/*--------------------------------------------------------------------------------------------------------------*/

// stops one starlet thread

int mload_stop_thread( int id )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_STOP_THREAD, "i:", id );

    return ret;

}

/*--------------------------------------------------------------------------------------------------------------*/

// continue one stopped starlet thread

int mload_continue_thread( int id )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_CONTINUE_THREAD, "i:", id );

    return ret;

}
/*--------------------------------------------------------------------------------------------------------------*/

// fix starlet address to read/write (uses SEEK_SET, etc as mode)

int mload_seek( int offset, int mode )
{
    if ( mload_init() < 0 ) return -1;

    return IOS_Seek( mload_fd, offset, mode );
}

/*--------------------------------------------------------------------------------------------------------------*/

// read bytes from starlet (it update the offset)

int mload_read( void* buf, u32 size )
{
    if ( mload_init() < 0 ) return -1;

    return IOS_Read( mload_fd, buf, size );
}

/*--------------------------------------------------------------------------------------------------------------*/

// write bytes from starlet (it update the offset)

int mload_write( const void * buf, u32 size )
{
    if ( mload_init() < 0 ) return -1;

    return IOS_Write( mload_fd, buf, size );
}

/*--------------------------------------------------------------------------------------------------------------*/

// fill a block (similar to memset)

int mload_memset( void *starlet_addr, int set, int len )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_MEMSET, "iii:", starlet_addr, set, len );


    return ret;
}

/*--------------------------------------------------------------------------------------------------------------*/

// get the ehci datas ( ehcmodule.elf uses this address)

void * mload_get_ehci_data()
{
    int ret;

    if ( mload_init() < 0 ) return NULL;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GET_EHCI_DATA, ":" );
    if ( ret < 0 ) return NULL;

    return ( void * ) ret;
}

/*--------------------------------------------------------------------------------------------------------------*/

// set the dev/es ioctlv in routine

int mload_set_ES_ioctlv_vector( void *starlet_addr )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_SET_ES_IOCTLV, "i:", starlet_addr );

    return ret;
}



int mload_getw( const void * addr, u32 *dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GETW, "i:i", addr, dat );

    return ret;
}

int mload_geth( const void * addr, u16 *dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GETH, "i:h", addr, dat );

    return ret;
}

int mload_getb( const void * addr, u8 *dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GETB, "i:b", addr, dat );

    return ret;
}

int mload_setw( const void * addr, u32 dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_SETW, "ii:", addr, dat );

    return ret;
}

int mload_seth( const void * addr, u16 dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_SETH, "ih:", addr, dat );

    return ret;
}

int mload_setb( const void * addr, u8 dat )
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_SETB, "ib:", addr, dat );

    return ret;
}

/*--------------------------------------------------------------------------------------------------------------*/

// to get log buffer
// this function return the size of the log buffer and prepare it to read with mload_read() the datas

int mload_get_log()
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GET_LOG, ":" );

    return ret;

}


/*--------------------------------------------------------------------------------------------------------------*/

// to get IOS base for dev/es  to create the cIOS

int mload_get_IOS_base()
{
    int ret;

    if ( mload_init() < 0 ) return -1;

    ret = IOS_IoctlvFormat( hid, mload_fd, MLOAD_GET_IOS_BASE, ":" );

    return ret;

}

/*
int load_ehc_module()
{
    int is_ios=0;
    FILE *fp;

    if(!external_ehcmodule)
        {

        fp=fopen("SD:/apps/usbloader_gx/ehcmodule.elf","rb");
        if(fp==NULL)
            fp=fopen("SD:/apps/usbloadergx/ehcmodule.elf","rb");

        if(fp!=NULL)
            {

            fseek(fp, 0, SEEK_END);
            size_external_ehcmodule = ftell(fp);
            external_ehcmodule = memalign(32, size_external_ehcmodule);
            if(!external_ehcmodule)
                {fclose(fp);}
            else
                {
                fseek(fp, 0, SEEK_SET);

                if(fread(external_ehcmodule,1, size_external_ehcmodule ,fp)!=size_external_ehcmodule)
                    {free(external_ehcmodule); external_ehcmodule=NULL;}

                fclose(fp);
                }
            }
        }

    if(!external_ehcmodule)
        {
        if(mload_init()<0) return -1;

        if (IOS_GetRevision() == 4) {
            gprintf("Loading ehcmodule v4\n");
            mload_elf((void *) ehcmodule_frag_v4_bin, &my_data_elf);
        } else if (IOS_GetRevision() == 0) { // 0? Strange value for v5 ehcmodule
            gprintf("Loading ehcmodule v5\n");
            mload_elf((void *) ehcmodule_frag_v5_bin, &my_data_elf);
        }
        thread_id = mload_run_thread(my_data_elf.start, my_data_elf.stack, my_data_elf.size_stack, my_data_elf.prio);
        if(thread_id < 0) return -1;
        }
    else
        {
        if(mload_init()<0) return -1;
        mload_elf((void *) external_ehcmodule, &my_data_elf);
        thread_id = mload_run_thread(my_data_elf.start, my_data_elf.stack, my_data_elf.size_stack, my_data_elf.prio);
        if(thread_id<0) return -1;
        }
    usleep(350*1000);


    // Test for IOS
    mload_seek(0x20207c84, SEEK_SET);
    mload_read(patch_datas, 4);
    if(patch_datas[0]==0x6e657665)
        {
        is_ios=38;
        }
    else
        {
        is_ios=36;
        }

    if(is_ios==36)
        {
        // IOS 36
        memcpy(ios_36, dip_plugin, 4);      // copy the entry_point
        memcpy((void *) dip_plugin, ios_36, 4*10);  // copy the adresses from the array

        mload_seek(0x1377E000, SEEK_SET);   // copy dip_plugin in the starlet
        mload_write(dip_plugin,size_dip_plugin);

        // enables DIP plugin
        mload_seek(0x20209040, SEEK_SET);
        mload_write(ios_36, 4);

        }
    if(is_ios==38)
        {
        // IOS 38

        memcpy(ios_38, dip_plugin, 4);      // copy the entry_point
        memcpy((void *) dip_plugin, ios_38, 4*10);   // copy the adresses from the array

        mload_seek(0x1377E000, SEEK_SET);   // copy dip_plugin in the starlet
        mload_write(dip_plugin,size_dip_plugin);

        // enables DIP plugin
        mload_seek(0x20208030, SEEK_SET);
        mload_write(ios_38, 4);

        }

    mload_close();

return 0;
}

int patch_cios_data() {
    patch_datas[0]=*((u32 *) (dip_plugin+16*4));
    mload_set_ES_ioctlv_vector((void *) patch_datas[0]);
    return 1;
}
*/