usbloadergx/source/FileOperations/File.cpp

#include <cstring>
#include "File.hpp"

CFile::CFile()
{
    file_fd = NULL;
    mem_file = NULL;
    filesize = 0;
    Pos = 0;
}

CFile::CFile(const char * filepath, const char * mode)
{
    file_fd = NULL;
    open(filepath, mode);
}

CFile::CFile(const u8 * mem, int size)
{
    file_fd = NULL;
    open(mem, size);
}

CFile::~CFile()
{
    close();
}

int CFile::open(const char * filepath, const char * mode)
{
    close();

    file_fd = fopen(filepath, mode);
    if(!file_fd)
        return -1;

    fseek(file_fd, 0, SEEK_END);
    filesize = ftell(file_fd);
    rewind();

    return 0;
}

int CFile::open(const u8 * mem, int size)
{
    close();

    mem_file = mem;
    filesize = size;

    return 0;
}

void CFile::close()
{
    if(file_fd)
        fclose(file_fd);

    file_fd = NULL;
    mem_file = NULL;
    filesize = 0;
    Pos = 0;
}

int CFile::read(u8 * ptr, size_t size)
{
    if(file_fd)
    {
        int ret = fread(ptr, 1, size, file_fd);
        if(ret > 0)
            Pos += ret;
        return ret;
    }

    int readsize = size;

    if(readsize > (long int) filesize-Pos)
        readsize = filesize-Pos;

    if(readsize <= 0)
        return readsize;

    if(mem_file != NULL)
    {
        memcpy(ptr, mem_file+Pos, readsize);
        Pos += readsize;
        return readsize;
    }

    return -1;
}

int CFile::write(const u8 * ptr, size_t size)
{
    if(file_fd)
    {
        int ret = fwrite(ptr, 1, size, file_fd);
        if(ret > 0)
            Pos += ret;
        return ret;
    }

    return -1;
}

int CFile::seek(long int offset, int origin)
{
	int ret = 0;

    if(origin == SEEK_SET)
    {
        Pos = offset;
    }
    else if(origin == SEEK_CUR)
    {
        Pos += offset;
    }
    else if(origin == SEEK_END)
    {
        Pos = filesize+offset;
    }
    if(Pos < 0)
    {
        Pos = 0;
        return -1;
    }

    if(file_fd)
        ret = fseek(file_fd, Pos, SEEK_SET);

    if(mem_file != NULL)
    {
        if(Pos > (long int) filesize)
        {
            Pos = filesize;
            return -1;
        }
    }

	return ret;
}