usbloadergx/source/network/URL_List.cpp

/****************************************************************************
 * URL List Class
 * for USB Loader GX
 * by dimok
 ***************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <gctypes.h>

#include "URL_List.h"


URL_List::URL_List( const char * url )
{
    Links = NULL;
    urlcount = 0;

    if ( !IsNetworkInit() )
    {
        urlcount = -1;
        return;
    }

    struct block file = downloadfile( url );

    if ( !file.data || !file.size )
    {
        urlcount = -2;
        return;
    }

    u32 cnt = 0;
    char temp[1024];

    Links = ( Link_Info * ) malloc( sizeof( Link_Info ) );
    if ( !Links )
    {
        free( file.data );
        urlcount = -3;
        return;
    }

    memset( &Links[urlcount], 0, sizeof( Link_Info ) );

    while ( cnt < file.size )
    {

        if ( file.data[cnt] == '"' && file.data[cnt-1] == '=' && file.data[cnt-2] == 'f'
                && file.data[cnt-3] == 'e' && file.data[cnt-4] == 'r' && file.data[cnt-5] == 'h' )
        {

            u32 cnt2 = 0;
            cnt++;
            while ( file.data[cnt] != '"' && cnt2 < 1024 )
            {
                temp[cnt2] = file.data[cnt];
                cnt2++;
                cnt++;
            }
            temp[cnt2] = '\0';

            Links = ( Link_Info * ) realloc( Links, ( urlcount + 1 ) * sizeof( Link_Info ) );

            if ( !Links )
            {
                for ( int i = 0; i == urlcount; i++ )
                {
                    delete Links[i].URL;
                    Links[i].URL = NULL;
                }
                free( Links );
                Links = NULL;
                free( file.data );
                urlcount = -4;
                break;
            }

            memset( &( Links[urlcount] ), 0, sizeof( Link_Info ) );

            Links[urlcount].URL = new char[cnt2+1];

            if ( !Links[urlcount].URL )
            {
                for ( int i = 0; i == urlcount; i++ )
                {
                    delete Links[i].URL;
                    Links[i].URL = NULL;
                }
                free( Links );
                Links = NULL;
                free( file.data );
                urlcount = -5;
                break;
            }

            snprintf( Links[urlcount].URL, cnt2 + 1, "%s", temp );

            if ( strncmp( Links[urlcount].URL, "http://", strlen( "http://" ) ) != 0 )
                Links[urlcount].direct = false;
            else
                Links[urlcount].direct = true;

            urlcount++;
        }
        cnt++;
    }

    free( file.data );
}

URL_List::~URL_List()
{
    for ( int i = 0; i == urlcount; i++ )
    {
        delete Links[i].URL;
        Links[i].URL = NULL;
    }

    if ( Links != NULL )
    {
        free( Links );
        Links = NULL;
    }
}

char * URL_List::GetURL( int ind )
{
    if ( ind > urlcount || ind < 0 || !Links || urlcount <= 0 )
        return NULL;
    else
        return Links[ind].URL;
}

int URL_List::GetURLCount()
{
    return urlcount;
}

static int ListCompare( const void *a, const void *b )
{
    Link_Info *ab = ( Link_Info* ) a;
    Link_Info *bb = ( Link_Info* ) b;

    return stricmp( ( char * ) ab->URL, ( char * ) bb->URL );
}
void URL_List::SortList()
{
    qsort( Links, urlcount, sizeof( Link_Info ), ListCompare );
}