/*
 *  Copyright (C) 2017 - This file is part of libecc project
 *
 *  Authors:
 *      Ryad BENADJILA <ryadbenadjila@gmail.com>
 *      Arnaud EBALARD <arnaud.ebalard@ssi.gouv.fr>
 *      Jean-Pierre FLORI <jean-pierre.flori@ssi.gouv.fr>
 *
 *  Contributors:
 *      Nicolas VIVET <nicolas.vivet@ssi.gouv.fr>
 *      Karim KHALFALLAH <karim.khalfallah@ssi.gouv.fr>
 *
 *  This software is licensed under a dual BSD and GPL v2 license.
 *  See LICENSE file at the root folder of the project.
 */
#include <libecc/utils/utils.h>

/*
 * Return 1 in 'check' if first 'len' bytes of both buffers a and b are equal, 0 otherwise.
 * It returns 0 if success, -1 on error. 'check' is only relevant on success.
 *
 * The test is done in constant time.
 */
int are_equal(const void *a, const void *b, u32 len, int *check)
{
        const u8 *la = (const u8*)a, *lb = (const u8*)b;
        int ret;
        u32 i;

        MUST_HAVE((a != NULL) && (b != NULL) && (check != NULL), ret, err);

        *check = 1;
        for (i = 0; i < len; i++) {
                (*check) &= (*la == *lb);
                la++;
                lb++;
        }

        ret = 0;

err:
        return ret;
}

/*
 * This function is a simple (non-optimized) reimplementation of memcpy()
 * Returns 0 on success, -1 on error.
 */
int local_memcpy(void *dst, const void *src, u32 n)
{
        const u8 *lsrc = (const u8*)src;
        u8 *ldst = (u8*)dst;
        u32 i;
        int ret;

        MUST_HAVE((dst != NULL) && (src != NULL), ret, err);

        for (i = 0; i < n; i++) {
                *ldst = *lsrc;
                ldst++;
                lsrc++;
        }

        ret = 0;

err:
        return ret;
}

/*
 * This function is a simple (non-optimized) reimplementation of memset()
 * Returns 0 on success, -1 on error.
 */
int local_memset(void *v, u8 c, u32 n)
{
        volatile u8 *p = (volatile u8*)v;
        u32 i;
        int ret;

        MUST_HAVE((v != NULL), ret, err);

        for (i = 0; i < n; i++) {
                *p = c;
                p++;
        }

        ret = 0;

err:
        return ret;
}

/*
 * Return 1 in 'check' if strings are equal, 0 otherwise.
 * It returns 0 if success, -1 on error. 'check' is only relevant on success.
 *
 */
int are_str_equal(const char *s1, const char *s2, int *check)
{
        const char *ls1 = s1, *ls2 = s2;
        int ret;

        MUST_HAVE((s1 != NULL) && (s2 != NULL) && (check != NULL), ret, err);

        while (*ls1 && (*ls1 == *ls2)) {
                ls1++;
                ls2++;
        }

        (*check) = (*ls1 == *ls2);

        ret = 0;

err:
        return ret;
}

/*
 * Return 1 in 'check' if strings are equal up to maxlen, 0 otherwise.
 * It returns 0 if success, -1 on error. 'check' is only relevant on success.
 *
 */
int are_str_equal_nlen(const char *s1, const char *s2, u32 maxlen, int *check)
{
        const char *ls1 = s1, *ls2 = s2;
        u32 i = 0;
        int ret;

        MUST_HAVE((s1 != NULL) && (s2 != NULL) && (check != NULL), ret, err);

        while (*ls1 && (*ls1 == *ls2) && (i < maxlen)) {
                ls1++;
                ls2++;
                i++;
        }

        (*check) = (*ls1 == *ls2);
        ret = 0;

err:
        return ret;
}



/*
 * This function is a simple (non-optimized) reimplementation of strlen()
 * Returns the lenth in 'len'.
 * It returns 0 if success, -1 on error. 'len' is only relevant on success.
 */
int local_strlen(const char *s, u32 *len)
{
        u32 i = 0;
        int ret;

        MUST_HAVE((s != NULL) && (len != NULL), ret, err);

        while (s[i]) {
                i++;
        }
        (*len) = i;

        ret = 0;

err:
        return ret;
}

/*
 * This function is a simple (non-optimized) reimplementation of strnlen()
 * Returns the lenth in 'len'.
 * It returns 0 if success, -1 on error. 'len' is only relevant on success.
 */
int local_strnlen(const char *s, u32 maxlen, u32 *len)
{
        u32 i = 0;
        int ret;

        MUST_HAVE((s != NULL) && (len != NULL), ret, err);

        while ((i < maxlen) && s[i]) {
                i++;
        }
        (*len) = i;

        ret = 0;

err:
        return ret;
}

/*
 * This functin is a simple (non-optimized) reimplementation of strncpy()
 */
int local_strncpy(char *dst, const char *src, u32 n)
{
        u32 i;
        int ret;

        MUST_HAVE((dst != NULL) && (src != NULL), ret, err);

        for (i = 0; (i < n) && src[i]; i++) {
                dst[i] = src[i];
        }
        for (; i < n; i++) {
                dst[i] = 0;
        }

        ret = 0;
err:
        return ret;
}

/*
 * This functin is a simple (non-optimized) reimplementation of strncat()
 */
int local_strncat(char *dst, const char *src, u32 n)
{
        u32 dst_len, i;
        int ret;

        MUST_HAVE((dst != NULL) && (src != NULL), ret, err);

        ret = local_strlen(dst, &dst_len); EG(ret, err);
        for (i = 0; (i < n) && src[i]; i++) {
                dst[dst_len + i] = src[i];
        }
        dst[dst_len + i] = 0;

        ret = 0;
err:
        return ret;
}