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);
}