#include <stdio.h>
#include <time.h>

#include "funccrypt.h"


static int crypted_fn(int arg0, char *arg1, void *arg2)
{
    CRYPT_PROLOGUE(crypted_fn);
    int i;
    printf("I'm decrypted ..\n");
    for (i = 0; i < 32; ++i)
        printf("%d ", i);
    puts("");
    CRYPT_EPILOGUE(crypted_fn);

    return 0x66;
}

static void crypted_fn2(void)
{
    CRYPT_PROLOGUE(crypted_fn2);
    printf("Another decrypted fn..\n");
    printf(
#ifdef _DEBUG
        "CRYPT_PROLOGUE crpyt_func returned: %s\n",
        crypt_strs[ CRYPT_RETVAL() ]
#else
        "CRYPT_PROLOGUE crpyt_func returned: %d\n",
        CRYPT_RETVAL()
#endif
    );
    CRYPT_EPILOGUE(crypted_fn2);
    printf("This part stays unencrypted all the time.\n");
}

CRYPT_FNDEF(crypted_fn3, void *arg0, int arg1, const char *arg2)
{
    printf("Yet another cryptable fn.. ");
    /* generate some nonsense machine instructions */
    printf("."); printf("."); printf("."); printf(".");
    printf("\n");
}
CRYPT_FNEND(crypted_fn3);

int main(void)
{
    srand(time(NULL));

#ifdef _DEBUG
    printf("Before Encryption:\n");
    printf("crypted_fn:\n");
    printHexBuf((uint8_t *)crypted_fn, 160, 32);
    printf("crypted_fn2:\n");
    printHexBuf((uint8_t *)crypted_fn2, 160, 32);
    printf("crypted_fn3:\n");
    printHexBuf((uint8_t *)crypted_fn3, 160, 32);
#endif
    printf("\nAfter Encryption:\n");
#ifdef _DEBUG
    printf("crypted_fn return val: %s\n",
           crypt_strs[ crypt_func((void *)crypted_fn) ]);
    printf("crypted_fn2 return val: %s\n",
           crypt_strs[ crypt_func((void *)crypted_fn2) ]);
    printf("crypted_fn3 return val: %s\n",
           crypt_strs[ crypt_func((void *)crypted_fn3) ]);
#else
    printf("crypted_fn return val: %d\n",
           crypt_func((void *)crypted_fn));
    printf("crypted_fn2 return val: %d\n",
           crypt_func((void *)crypted_fn2));
    printf("crypted_fn3 return val: %d\n",
           crypt_func((void *)crypted_fn3));
#endif

#ifdef _DEBUG
    printf("crypted_fn:\n");
    printHexBuf((uint8_t *)crypted_fn, 160, 32);
    printf("crypted_fn2:\n");
    printHexBuf((uint8_t *)crypted_fn2, 160, 32);
    printf("crypted_fn3:\n");
    printHexBuf((uint8_t *)crypted_fn3, 160, 32);
#endif
    printf("\noutput:\n");
    printf("crypted_fn: 0x%X\n", crypted_fn(0, NULL, NULL));
    crypted_fn2();
    crypted_fn3(NULL, (unsigned int)-1, "TEST");

    return 0;
}