#pragma once

#include <initializer_list>
#include <stdio.h>
#include <sys/mman.h>
#include <unistd.h>

void *malloc_hugetlbfs(size_t size) {
void *r;
  char tmpName[] = "/mnt/hugetlbfs/mmallocXXXXXX";
  int fd = mkstemp(tmpName);
  if (fd < 0) goto fail1;

  if (ftruncate(fd, size) != 0) goto fail2;

  r = mmap(NULL,
		 size,
		 PROT_READ | PROT_WRITE,
		 MAP_SHARED | MAP_POPULATE,
		 fd, 0);
  if (r == MAP_FAILED) goto fail2;

  return r;

 fail2:
  close(fd);
 fail1:
  perror(NULL);
  return MAP_FAILED;  
}

void *malloc_mmap(size_t size) {
  void *r;
  for (auto flag : {MAP_POPULATE | MAP_HUGETLB, MAP_POPULATE, 0}) {
    fprintf(stderr, "try %i", flag);
    r = mmap(NULL, size,
	     PROT_READ | PROT_WRITE,
	     flag | MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
    if (r != MAP_FAILED) break;
    perror("mmap");
  }
  return r;
}

template<typename T>
T* mmalloc() {
  void *r;
  if ((r = malloc_hugetlbfs(sizeof(T))) == MAP_FAILED)
    if ((r = malloc_mmap(sizeof(T))) == MAP_FAILED)
      exit(EXIT_FAILURE);
  return (T*) r;
}