/* Benchmark malloc and free functions.
   Copyright (C) 2019-2021 Free Software Foundation, Inc.
   This file is part of the GNU C Library.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, see
   <https://www.gnu.org/licenses/>.  */

// modified by Daan Leijen to fit the bench suite and add lifo/fifo free order.

// #include <pthread.h>
// #include <stdio.h>
// #include <stdlib.h>
// #include <sys/resource.h>
// #include "malloc.h"

#include <cheriintrin.h>
#include <stdint.h>
#include <stddef.h>

// #define malloc      MALLOCCHERI
// #define free        FREECHERI

// #include "bench-timing.h"
// #include "json-lib.h"

void free(void * __capability ptr);
void * __capability malloc(size_t size);


/* Benchmark the malloc/free performance of a varying number of blocks of a
   given size.  This enables performance tracking of the t-cache and fastbins.
   It tests 3 different scenarios: single-threaded using main arena,
   multi-threaded using thread-arena, and main arena with SINGLE_THREAD_P
   false.  */

// source: https://github.com/daanx/mimalloc-bench/blob/master/bench/glibc-bench/bench-malloc-thread.c

#define NUM_ITERS 20
#define NUM_ALLOCS 4
#define MAX_ALLOCS 1600

// Daan: disable timing
typedef long timing_t;
#define TIMING_NOW(s) 
#define TIMING_DIFF(e,start,stop)


typedef struct
{
  size_t iters;
  size_t size;
  int n;
  timing_t elapsed;
} malloc_args;

static void
do_benchmark(malloc_args *args, char * __capability * __capability arr)
{
  // timing_t start, stop;
  size_t iters = args->iters;
  size_t size = args->size;
  int n = args->n;

//   TIMING_NOW (start);

  for (int j = 0; j < iters; j++)
    {
      for (int i = 0; i < n; i++) {
        arr[i] = malloc (size);
        for(int g = 0; g < size; g++) { arr[i][g] =(char)g; }
      }

      // free half in fifo order
      for (int i = 0; i < n/2; i++) {
	        free (arr[i]);
          arr[i] = NULL;  
      }
    
      // and the other half in lifo order
      for(int i = n-1; i >= n/2; i--) {
          free(arr[i]);
          arr[i] = NULL;
      }
  }      

//   TIMING_NOW (stop);

//   TIMING_DIFF (args->elapsed, start, stop);
}

static malloc_args tests[3][NUM_ALLOCS];
static int allocs[NUM_ALLOCS] = { 25, 100, 400, MAX_ALLOCS };

// static void *
// thread_test (void *p)
// {
//   char **arr = (char**)p;

//   /* Run benchmark multi-threaded.  */
//   for (int i = 0; i < NUM_ALLOCS; i++)
//     do_benchmark (&tests[2][i], arr);

//   return p;
// }

void
bench (unsigned long size)
{
  size_t iters = NUM_ITERS;
  // char**arr = (char**)malloc (MAX_ALLOCS * sizeof (void*));
  char * __capability * __capability arr = malloc(MAX_ALLOCS * sizeof(char * __capability));

  for (int t = 0; t < 3; t++)
    for (int i = 0; i < NUM_ALLOCS; i++)
      {
	tests[t][i].n = allocs[i];
	tests[t][i].size = size;
	tests[t][i].iters = iters / allocs[i];

	/* Do a quick warmup run.  */
	if (t == 0)
	  do_benchmark (&tests[0][i], arr);
      }

  /* Run benchmark single threaded in main_arena.  */
  for (int i = 0; i < NUM_ALLOCS; i++)
    do_benchmark (&tests[0][i], arr);

  /* Run benchmark in a thread_arena.  */
//   pthread_t t;
//   pthread_create (&t, NULL, thread_test, (void*)arr);
//   pthread_join (t, NULL);

  /* Repeat benchmark in main_arena with SINGLE_THREAD_P == false.  */
  // for (int i = 0; i < NUM_ALLOCS; i++)
  //   do_benchmark (&tests[1][i], arr);

  free (arr);
  arr = NULL;

  /*
  json_ctx_t json_ctx;

  json_init (&json_ctx, 0, stdout);

  json_document_begin (&json_ctx);

  json_attr_string (&json_ctx, "timing_type", TIMING_TYPE);

  json_attr_object_begin (&json_ctx, "functions");

  json_attr_object_begin (&json_ctx, "malloc");

  char s[100];
  double iters2 = iters;

  json_attr_object_begin (&json_ctx, "");
  json_attr_double (&json_ctx, "malloc_block_size", size);

  struct rusage usage;
  getrusage (RUSAGE_SELF, &usage);
  json_attr_double (&json_ctx, "max_rss", usage.ru_maxrss);

  for (int i = 0; i < NUM_ALLOCS; i++)
    {
      sprintf (s, "main_arena_st_allocs_%04d_time", allocs[i]);
      json_attr_double (&json_ctx, s, tests[0][i].elapsed / iters2);
    }

  for (int i = 0; i < NUM_ALLOCS; i++)
    {
      sprintf (s, "main_arena_mt_allocs_%04d_time", allocs[i]);
      json_attr_double (&json_ctx, s, tests[1][i].elapsed / iters2);
    }

  for (int i = 0; i < NUM_ALLOCS; i++)
    {
      sprintf (s, "thread_arena__allocs_%04d_time", allocs[i]);
      json_attr_double (&json_ctx, s, tests[2][i].elapsed / iters2);
    }

  json_attr_object_end (&json_ctx);

  json_attr_object_end (&json_ctx);

  json_attr_object_end (&json_ctx);

  json_document_end (&json_ctx);
  */
}

// static void usage (const char *name)
// {
//   fprintf (stderr, "%s: <alloc_size>\n", name);
//   exit (1);
// }

int main (void)
{
//   INITREGULARALLOC();
  long size = 3;
//   if (argc == 2)
//     size = strtol (argv[1], NULL, 0);

//   if (argc > 2 || size <= 0)
//     usage (argv[0]);

  // bench (size);
  // bench (2*size);
  // bench (4*size);
  // bench (8*size);
  // bench (16*size);
  // bench (32*size); 

  return 0;
}