mirror of
https://github.com/wiidev/usbloadergx.git
synced 2024-11-28 06:04:18 +01:00
1129a26b44
Uses MEM2 for downloads, improves timeout handling and adds proxy support.
149 lines
2.6 KiB
C++
149 lines
2.6 KiB
C++
/****************************************************************************
|
|
* 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 download file = {};
|
|
downloadfile(url, &file);
|
|
|
|
if (file.size <= 0)
|
|
{
|
|
urlcount = -2;
|
|
return;
|
|
}
|
|
|
|
u32 cnt = 0;
|
|
char temp[1024];
|
|
|
|
Links = (Link_Info *) malloc(sizeof(Link_Info));
|
|
if (!Links)
|
|
{
|
|
MEM2_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;
|
|
MEM2_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;
|
|
MEM2_free(file.data);
|
|
urlcount = -5;
|
|
break;
|
|
}
|
|
|
|
snprintf(Links[urlcount].URL, cnt2 + 1, "%s", temp);
|
|
|
|
if (strncmp(Links[urlcount].URL, "https://", strlen("https://")) != 0)
|
|
Links[urlcount].direct = false;
|
|
else Links[urlcount].direct = true;
|
|
|
|
urlcount++;
|
|
}
|
|
cnt++;
|
|
}
|
|
|
|
MEM2_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 strcasecmp((char *) ab->URL, (char *) bb->URL);
|
|
}
|
|
void URL_List::SortList()
|
|
{
|
|
qsort(Links, urlcount, sizeof(Link_Info), ListCompare);
|
|
}
|