usbloadergx/source/network/HTML_Stream.cpp

197 lines
4.1 KiB
C++
Raw Normal View History

/***************************************************************************
* Copyright (C) 2010
* by Dimok
*
* This software is provided 'as-is', without any express or implied
* warranty. In no event will the authors be held liable for any
* damages arising from the use of this software.
*
* Permission is granted to anyone to use this software for any
* purpose, including commercial applications, and to alter it and
* redistribute it freely, subject to the following restrictions:
*
* 1. The origin of this software must not be misrepresented; you
* must not claim that you wrote the original software. If you use
* this software in a product, an acknowledgment in the product
* documentation would be appreciated but is not required.
*
* 2. Altered source versions must be plainly marked as such, and
* must not be misrepresented as being the original software.
*
* 3. This notice may not be removed or altered from any source
* distribution.
*
* HTML_Stream Class
*
* for WiiXplorer 2010
***************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "HTML_Stream.h"
#include "networkops.h"
#include "http.h"
#define htmlstringcompare(text, cmp, pos) strncasecmp((const char*) &text[pos], (const char*) cmp, strlen((const char*) cmp))
HTML_Stream::HTML_Stream()
{
HTML_File = NULL;
position = 0;
filesize = 0;
}
HTML_Stream::HTML_Stream(const char * url)
{
HTML_File = NULL;
position = 0;
filesize = 0;
LoadLink(url);
}
HTML_Stream::~HTML_Stream()
{
if(HTML_File)
free(HTML_File);
}
bool HTML_Stream::LoadLink(const char * url)
{
if (!IsNetworkInit())
return false;
struct block file = downloadfile(url);
if (!file.data || !file.size)
return false;
if(HTML_File)
free(HTML_File);
HTML_File = (char *) file.data;
filesize = file.size;
position = 0;
return true;
}
const char * HTML_Stream::FindStringStart(const char * string)
{
if (!HTML_File)
return NULL;
while((u32) position < filesize)
{
if(htmlstringcompare(HTML_File, string, position) == 0)
break;
position++;
}
return &HTML_File[position];
}
const char * HTML_Stream::FindStringEnd(const char * string)
{
if (!HTML_File)
return NULL;
while((u32) position < filesize)
{
if(htmlstringcompare(HTML_File, string, position) == 0)
break;
position++;
}
if((u32) position >= filesize)
{
return NULL;
}
position += strlen(string);
return &HTML_File[position];
}
char * HTML_Stream::CopyString(const char * stopat)
{
if(!stopat || !HTML_File)
return NULL;
u32 blocksize = 1024;
u32 counter = 0;
u32 allocatedsize = 0;
char * outtext = (char*) malloc(blocksize);
if(!outtext)
return NULL;
allocatedsize = blocksize;
memset(outtext, 0, blocksize);
while ((htmlstringcompare(HTML_File, stopat, position) != 0) && (position+strlen(stopat) < filesize))
{
if(counter > blocksize)
{
blocksize += 1024;
char * tmpblock = (char*) realloc(outtext, blocksize);
if(!tmpblock)
{
free(outtext);
outtext = NULL;
free(tmpblock);
return NULL;
}
outtext = tmpblock;
}
outtext[counter] = HTML_File[position];
position++;
counter++;
}
outtext[counter] = '\0';
outtext = (char*) realloc(outtext, counter+1);
return outtext;
}
int HTML_Stream::Seek(u32 pos, int origin)
{
if(!HTML_File)
return -1;
switch(origin)
{
case SEEK_SET:
position = pos;
break;
case SEEK_CUR:
position += pos;
break;
case SEEK_END:
position = filesize+pos;
break;
}
return 0;
}
void HTML_Stream::Rewind()
{
if(!HTML_File)
return;
position = 0;
}
int HTML_Stream::GetPosition()
{
return position;
}