/* Copyright (c) 2007-2009, Stanford University
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*     * Redistributions of source code must retain the above copyright
*       notice, this list of conditions and the following disclaimer.
*     * Redistributions in binary form must reproduce the above copyright
*       notice, this list of conditions and the following disclaimer in the
*       documentation and/or other materials provided with the distribution.
*     * Neither the name of Stanford University nor the
*       names of its contributors may be used to endorse or promote products
*       derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY STANFORD UNIVERSITY ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL STANFORD UNIVERSITY BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef STDDEFINES_H_
#define STDDEFINES_H_

#include <assert.h>
#include <stdlib.h>
#include <sys/time.h>
#include <stdint.h>
#include <stdio.h>

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <sys/wait.h>
//#include "../../jemalloc/include/jemalloc/jemalloc.h"

//#define TIMING

#define FILENAME "/dev/contigmem"

/* Debug printf */
#define dprintf(...) fprintf(stdout, __VA_ARGS__)

/* Wrapper to check for errors */
#define CHECK_ERROR(a)                                       \
   if (a)                                                    \
   {                                                         \
      perror("Error at line\n\t" #a "\nSystem Msg");         \
      assert ((a) == 0);                                     \
   }

// static inline void *MALLOC(size_t size)
// {
// //    void * temp = malloc(size);
// //    assert(temp);
// //    return temp;

// // Ensuring malloc this is mmap and phyiscally contigous
//   return MALLOCCONTIGOUS(size);
// }

static inline void *CALLOC(size_t num, size_t size)
{
    void * temp = calloc(num, size);
    assert(temp);
    return temp;
}

static inline void *REALLOC(void *ptr, size_t size)
{
    void * temp = realloc(ptr, size);
    assert(temp);
    return temp;
}

static inline char *GETENV(char *envstr)
{
    char *env = getenv(envstr);
    if (!env) return "0";
    else return env;
}

static inline void *FREE(void *ptr)
{
    // int *pt = ptr;
    // size_t size;
    // --pt;
    // size = *pt;
    // if(munmap(pt, size) == -1)
    // {
    //     perror("munmap");
    //     exit(EXIT_FAILURE);
    // }
    // free(ptr);
}

#define GET_TIME(start, end, duration)                                     \
   duration.tv_sec = (end.tv_sec - start.tv_sec);                         \
   if (end.tv_nsec >= start.tv_nsec) {                                     \
      duration.tv_nsec = (end.tv_nsec - start.tv_nsec);                   \
   }                                                                       \
   else {                                                                  \
      duration.tv_nsec = (1000000000L - (start.tv_nsec - end.tv_nsec));   \
      duration.tv_sec--;                                                   \
   }                                                                       \
   if (duration.tv_nsec >= 1000000000L) {                                  \
      duration.tv_sec++;                                                   \
      duration.tv_nsec -= 1000000000L;                                     \
   }

static inline unsigned int time_diff (
        struct timeval *end, struct timeval *begin)
{
#ifdef TIMING
    uint64_t result;

    result = end->tv_sec - begin->tv_sec;
    result *= 1000000;     /* usec */
    result += end->tv_usec - begin->tv_usec;

    return result;
#else
    return 0;
#endif
}

static inline void get_time (struct timeval *t)
{
#ifdef TIMING
    gettimeofday (t, NULL);
#endif
}

static inline void *MALLOCCONTIGOUS(size_t size)
{
    dprintf("Getting called\n");
//    void * temp = malloc(size);
//    assert(temp);
//    return temp;

// Ensuring malloc this is mmap and phyiscally contigous
    int fd = open
            (FILENAME, O_RDWR, 0600);
    if (fd < 0) {
        perror("open");
        exit(EXIT_FAILURE);
    }

    off_t offset = 0; // offset to seek to.

    if (ftruncate(fd, size) < 0) {
        perror("ftruncate");
        close(fd);
        exit(EXIT_FAILURE);
    }

    void *ptr = mmap(NULL, size,
                    PROT_READ | PROT_WRITE, MAP_SHARED,
                    fd, offset);
    if(ptr == MAP_FAILED)
    {
        perror("mmap");
        exit(EXIT_FAILURE);
    }

    // *ptr = size;

    return ptr;
    // void *ptr = malloc(size);
    // return ptr;
}


static inline void REGULARMMAP(size_t size)
{
    dprintf("Regular mmap\n");

    void *ptr = mmap(NULL, size,
                    PROT_READ | PROT_WRITE, MAP_SHARED,
                    -1, 0);
}
#endif // STDDEFINES_H_