/*
 * Copyright (c) 2004, Bull S.A..  All rights reserved.
 * Created by: Sebastien Decugis

 * This program is free software; you can redistribute it and/or modify it
 * under the terms of version 2 of the GNU General Public License as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it would be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write the Free Software Foundation, Inc., 59
 * Temple Place - Suite 330, Boston MA 02111-1307, USA.
 *
 
 
 * This file is a scalability test for the pthread_cond_init function.

 * The steps are:
 * -> Restrict the memory to 32Mb * SCALABILITY_FACTOR
 * -> While there is free memory
 *      -> allocate memory for 10 cond vars
 *      -> time = 0
 *      -> init the 10 cond vars with different attributes
 *      -> output time
 * -> When memory is full; undo everything:
 *      -> time=0
 *      -> destroy the 10 cond vars
 *      -> output time
 *      -> free memory
 */

 /* We are testing conformance to IEEE Std 1003.1, 2003 Edition */
 #define _POSIX_C_SOURCE 200112L
 
 /* We need XSI conformance for memory limitation */
#ifndef WITHOUT_XOPEN
 #define _XOPEN_SOURCE	600
 
/********************************************************************************************/
/****************************** standard includes *****************************************/
/********************************************************************************************/
 #include <pthread.h>
 #include <errno.h>
 #include <unistd.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <stdarg.h>
 #include <sys/resource.h>
 #include <sys/time.h>
 
/********************************************************************************************/
/******************************   Test framework   *****************************************/
/********************************************************************************************/
 #include "testfrmw.h"
 #include "testfrmw.c"
 /* This header is responsible for defining the following macros:
  * UNRESOLVED(ret, descr);  
  *    where descr is a description of the error and ret is an int (error code for example)
  * FAILED(descr);
  *    where descr is a short text saying why the test has failed.
  * PASSED();
  *    No parameter.
  * 
  * Both three macros shall terminate the calling process.
  * The testcase shall not terminate in any other maneer.
  * 
  * The other file defines the functions
  * void output_init()
  * void output(char * string, ...)
  * 
  * Those may be used to output information.
  */

/********************************************************************************************/
/********************************** Configuration ******************************************/
/********************************************************************************************/
#ifndef SCALABILITY_FACTOR
#define SCALABILITY_FACTOR 1
#endif
#ifndef VERBOSE
#define VERBOSE 1
#endif

/********************************************************************************************/
/***********************************    Test case   *****************************************/
/********************************************************************************************/

typedef struct _teststruct
{
	pthread_cond_t cnd[10 * SCALABILITY_FACTOR];
	pthread_condattr_t ca[4];
	pthread_condattr_t *pca[10 * SCALABILITY_FACTOR];
	struct _teststruct * prev;
} teststruct_t;

int main(int argc, char * argv[])
{
	struct rlimit rl;
	int ret;
	int i;
	teststruct_t *cur, *prev;
 	struct timeval  time_zero, time_cour, time_res, time_sav[8];
 	long sav= 0;
	
	long monotonic, pshared;
	
	pshared = sysconf(_SC_THREAD_PROCESS_SHARED);
	monotonic = sysconf(_SC_MONOTONIC_CLOCK);
	#if VERBOSE > 1
	output("Support for process-shared condvars: %li\n", pshared);
	output("Support for monotonic clock        : %li\n", monotonic);
	#endif
	
	/* Limit the process memory to a small value (32Mb for example). */
	rl.rlim_max=1024 * 1024 * 32 * SCALABILITY_FACTOR;
	rl.rlim_cur=rl.rlim_max;
	if ((ret = setrlimit(RLIMIT_AS,  &rl)))
	{ UNRESOLVED(ret, "Memory limitation failed"); }
	#if VERBOSE > 1
	output(";Memory is now limited to %dMb\n", 	rl.rlim_max >> 20);
	#endif
	
	prev = NULL;
	cur = NULL;
	
	/* Loop while we have memory left */
	while (1) 
	{
		/* Allocate memory for 10 mutex and related stuff */
		cur = malloc(sizeof(teststruct_t)); 
		if (cur == NULL) /* No memory left */
			break;
		
		/* Link to the previous so we are able to free memory */
		cur->prev = prev;
		prev = cur;

		/* Initialize the mutex attributes */
			/* We will have:
			 * pca[0] = NULL
			 * pca[1] = Default cond attribute
			 * pca[2] = (if supported) pshared cond attribute
			 * pca[3] = (if supported) monotonic clock cond attribute
			 * pca[4] = (if supported) pshared + monotonic
			 * pca[5] = pca[0]...
			 */
		for (i=0; i<4; i++)
		{
			ret = pthread_condattr_init(&(cur->ca[i]));
			if (ret != 0)
			{ UNRESOLVED(ret, "Cond attribute init failed"); }
			
			if ((monotonic > 0) && ((i == 2) || (i == 3)))
			{
				ret = pthread_condattr_setclock(&(cur->ca[i]), CLOCK_MONOTONIC);
				if (ret != 0)
				{  UNRESOLVED(ret, "Set monotonic clock failed");  }
			}
			if ((pshared > 0) && ((i == 1) || (i == 3)))
			{
				ret = pthread_condattr_setpshared(&(cur->ca[i]), PTHREAD_PROCESS_SHARED);
				if (ret != 0)
				{  UNRESOLVED(ret, "Set process shared attribute failed");  }
			}
			
			
		}
		
		for (i=0; i<(10 * SCALABILITY_FACTOR); i++)
		{
			cur->pca[i]=(i%5)?&(cur->ca[i % 4]):NULL;
		} /* The mutex attributes are now initialized */
		
		/* Save the time */
		gettimeofday(&time_zero, NULL);
		
		/* For each condvar, we will:
		 * - init the condvar
		 * - destroy the condvar
		 * - init the condvar
		 * - destroy the condvar
		 * - init the condvar
		 */
		for (i=0; i<10 * SCALABILITY_FACTOR; i++)
		{
			ret = pthread_cond_init(&(cur->cnd[i]), cur->pca[i]);
			if (ret)
			{ UNRESOLVED(ret, "Cond 1st init failed"); }
			ret = pthread_cond_destroy(&(cur->cnd[i]));
			if (ret)
			{ UNRESOLVED(ret, "Cond 1st destroy failed"); }
			ret = pthread_cond_init(&(cur->cnd[i]), cur->pca[i]);
			if (ret)
			{ UNRESOLVED(ret, "Cond 2nd init failed"); }
			ret = pthread_cond_destroy(&(cur->cnd[i]));
			if (ret)
			{ UNRESOLVED(ret, "Cond 2nd destroy failed"); }
			ret = pthread_cond_init(&(cur->cnd[i]), cur->pca[i]);
			if (ret)
			{ UNRESOLVED(ret, "Cond 3rd init failed"); }
		}
		/* Compute the operation duration */		
		gettimeofday(&time_cour, NULL);
		time_res.tv_usec = time_cour.tv_usec + 1000000 - time_zero.tv_usec;
		if (time_res.tv_usec < 1000000)
		{
			time_res.tv_sec = time_cour.tv_sec - 1 - time_zero.tv_sec;
		}
		else
		{
			time_res.tv_sec = time_cour.tv_sec - time_zero.tv_sec;
			time_res.tv_usec -= 1000000;
		}
		
		if (sav >3)
		{
			time_sav[4].tv_sec = time_sav[5].tv_sec;
			time_sav[4].tv_usec = time_sav[5].tv_usec;
			time_sav[5].tv_sec = time_sav[6].tv_sec;
			time_sav[5].tv_usec = time_sav[6].tv_usec;
			time_sav[6].tv_sec = time_sav[7].tv_sec;
			time_sav[6].tv_usec = time_sav[7].tv_usec;
			time_sav[7].tv_sec = time_res.tv_sec;
			time_sav[7].tv_usec = time_res.tv_usec;
		} else {
			time_sav[sav].tv_sec = time_res.tv_sec;
			time_sav[sav].tv_usec = time_res.tv_usec;
		}
		sav++;
	}
	if (errno != ENOMEM)
	{  UNRESOLVED(errno, "Memory not full"); }
	
	/* Now we just have to cleanup everything. */
	while (prev != NULL)
	{
		cur = prev;
		prev = cur->prev;
		
		/* Free the condvar resources in the cur element */
		for (i=0; i<10 * SCALABILITY_FACTOR; i++)
		{
			ret = pthread_cond_destroy(&(cur->cnd[i]));
			if (ret)
			{ UNRESOLVED(ret, "Cond final destroy failed"); }
		}
		/* Free the cond attributes resources in the cur element */
		for (i=0; i<4; i++)
		{
			ret = pthread_condattr_destroy(&(cur->ca[i]));
			if (ret != 0)
			{ UNRESOLVED(ret, "Cond attribute destroy failed"); }
		}
		 /* Free the element memory */
		 free(cur);
	}
	#if VERBOSE > 0
	if (sav < 8)
	{
		output("Not enough iterations to build statistics\n");
	}
	else
	{
		output("Duration for the operations:\n");
		output(" %8i : %2i.%06i s\n", 0, time_sav[0].tv_sec, time_sav[0].tv_usec);
		output(" %8i : %2i.%06i s\n", 1, time_sav[1].tv_sec, time_sav[1].tv_usec);
		output(" %8i : %2i.%06i s\n", 2, time_sav[2].tv_sec, time_sav[2].tv_usec);
		output(" %8i : %2i.%06i s\n", 3, time_sav[3].tv_sec, time_sav[3].tv_usec);
		output(" [...]\n");
		output(" %8i : %2i.%06i s\n", sav-3, time_sav[4].tv_sec, time_sav[4].tv_usec);
		output(" %8i : %2i.%06i s\n", sav-2, time_sav[5].tv_sec, time_sav[5].tv_usec);
		output(" %8i : %2i.%06i s\n", sav-1, time_sav[6].tv_sec, time_sav[6].tv_usec);
		output(" %8i : %2i.%06i s\n", sav   , time_sav[7].tv_sec, time_sav[7].tv_usec);
	}
	#endif

	PASSED;
}



#else /* WITHOUT_XOPEN */
int main (int argc, char * argv[])
{
	output_init();
	UNRESOLVED(0, "This test requires XSI features");
}
#endif
