#include "utils.h"

u32 align(u32 offset, u32 alignment)
{
    u32 mask = ~(alignment-1);

    return (offset + (alignment-1)) & mask;
}

u64 align64(u64 offset, u32 alignment)
{
    u64 mask = ~(alignment-1);

    return (offset + (alignment-1)) & mask;
}

u64 getle64(const u8* p)
{
    u64 n = p[0];

    n |= (u64)p[1]<<8;
    n |= (u64)p[2]<<16;
    n |= (u64)p[3]<<24;
    n |= (u64)p[4]<<32;
    n |= (u64)p[5]<<40;
    n |= (u64)p[6]<<48;
    n |= (u64)p[7]<<56;
    return n;
}

u64 getbe64(const u8* p)
{
    u64 n = 0;

    n |= (u64)p[0]<<56;
    n |= (u64)p[1]<<48;
    n |= (u64)p[2]<<40;
    n |= (u64)p[3]<<32;
    n |= (u64)p[4]<<24;
    n |= (u64)p[5]<<16;
    n |= (u64)p[6]<<8;
    n |= (u64)p[7]<<0;
    return n;
}

u32 getle32(const u8* p)
{
    return (p[0]<<0) | (p[1]<<8) | (p[2]<<16) | (p[3]<<24);
}

u32 getbe32(const u8* p)
{
    return (p[0]<<24) | (p[1]<<16) | (p[2]<<8) | (p[3]<<0);
}

u32 getle16(const u8* p)
{
    return (p[0]<<0) | (p[1]<<8);
}

u32 getbe16(const u8* p)
{
    return (p[0]<<8) | (p[1]<<0);
}

void putle16(u8* p, u16 n)
{
    p[0] = (u8) n;
    p[1] = (u8) (n>>8);
}

void putle32(u8* p, u32 n)
{
    p[0] = (u8) n;
    p[1] = (u8) (n>>8);
    p[2] = (u8) (n>>16);
    p[3] = (u8) (n>>24);
}

void putle64(u8* p, u64 n)
{
    p[0] = (u8) n;
    p[1] = (u8) (n >> 8);
    p[2] = (u8) (n >> 16);
    p[3] = (u8) (n >> 24);
    p[4] = (u8) (n >> 32);
    p[5] = (u8) (n >> 40);
    p[6] = (u8) (n >> 48);
    p[7] = (u8) (n >> 56);
}

void putbe16(u8* p, u16 n)
{
    p[1] = (u8) n;
    p[0] = (u8) (n >> 8);
}

void putbe32(u8* p, u32 n)
{
    p[3] = (u8) n;
    p[2] = (u8) (n >> 8);
    p[1] = (u8) (n >> 16);
    p[0] = (u8) (n >> 24);
}

void putbe64(u8* p, u64 n)
{
    p[7] = (u8) n;
    p[6] = (u8) (n >> 8);
    p[5] = (u8) (n >> 16);
    p[4] = (u8) (n >> 24);
    p[3] = (u8) (n >> 32);
    p[2] = (u8) (n >> 40);
    p[1] = (u8) (n >> 48);
    p[0] = (u8) (n >> 56);
}

u32 swap_uint32(u32 val)
{
    val = ((val << 8) & 0xFF00FF00) | ((val >> 8) & 0xFF00FF);
    return (val << 16) | (val >> 16);
}

void reverse_endian(u32* buffer, size_t size)
{
    u32 i = 0;
    if(size % sizeof(u32)) return;

    u32 words = size / sizeof(u32);

    for(i = 0; i < words; i++)
        buffer[i] = swap_uint32(buffer[i]);
}

void reverse_words(u32* buffer, size_t size)
{
    u32 i = 0;
    if(size % sizeof(u32)) return;

    u32 words = size / sizeof(u32);

    u32 temp = 0;

    for (i = 0; i < words; i++)
    {
        temp = buffer[i];
        buffer[i] = buffer[words - 1];
        buffer[words - 1] = temp;
        words--;
    }
}

void reverse(u32* buffer, size_t size)
{
    reverse_words(buffer, size);
    reverse_endian(buffer, size);
}

void hexdump(void *ptr, int buflen)
{
    u8 *buf = (u8*)ptr;
    int i, j;

    for (i=0; i<buflen; i+=16)
    {
        printf("%06x: ", i);
        for (j=0; j<16; j++)
        {
            if (i+j < buflen)
            {
                printf("%02x ", buf[i+j]);
            }
            else
            {
                printf("   ");
            }
        }

        printf(" ");

        for (j=0; j<16; j++)
        {
            if (i+j < buflen)
            {
                printf("%c", (buf[i+j] >= 0x20 && buf[i+j] <= 0x7e) ? buf[i+j] : '.');
            }
        }
        printf("\n");
    }
}

void memdump(FILE* fout, const char* prefix, const u8* data, u32 size)
{
    u32 i;
    u32 prefixlen = strlen(prefix);
    u32 offs = 0;
    u32 line = 0;
    while(size)
    {
        u32 max = 32;

        if (max > size)
            max = size;

        if (line==0)
            fprintf(fout, "%s", prefix);
        else
            fprintf(fout, "%*s", prefixlen, "");


        for(i=0; i<max; i++)
            fprintf(fout, "%02X", data[offs+i]);
        fprintf(fout, "\n");
        line++;
        size -= max;
        offs += max;
    }
}

int makedir(const char* dir)
{
#ifdef _WIN32
    return _mkdir(dir);
#else
    return mkdir(dir, 0777);
#endif
}

bool isnumeric(const char* string)
{
    while(*string)
    {
        if(!isdigit(*string))
            return false;
        string++;
    }
    return true;
}