/*   
 * Copyright (c) 2004, Intel Corporation. All rights reserved.
 * Created by:  crystal.xiong REMOVE-THIS AT intel DOT com
 * This file is licensed under the GPL license.  For the full content
 * of this license, see the COPYING file at the top level of this 
 * source tree.

 * Test pthread_attr_setstack()
 * 
 * Steps:
 * 1.  Initialize pthread_attr_t object (attr) 
 * 2.  set the stackaddr and stacksize to attr
 * 3.  create a thread with the attr
 * 4.  In the created thread, read the stacksize and stackaddr
 *
 * NOTE: pthread_getattr_np is not a POSIX compliant API. It is 
 *       provided by nptl, which is developed by Ulrich Drepper.
 */

#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif 
#include <pthread.h>
#include <limits.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/param.h>
#include <errno.h>
#include <unistd.h>
#include "posixtest.h"

#define TEST "2-1"
#define FUNCTION "pthread_attr_setstack"
#define ERROR_PREFIX "unexpected error: " FUNCTION " " TEST ": "

static void *stack_addr;
size_t stack_size;

void *thread_func()
{
	pthread_attr_t attr;
	void *saddr;
	size_t ssize;
	int rc;

        /* pthread_getattr_np is not POSIX Compliant API*/
	rc = pthread_getattr_np(pthread_self(), &attr);
	if (rc != 0)
    	{
      		perror(ERROR_PREFIX "pthread_getattr_np");
      		exit(PTS_UNRESOLVED);
	}

	pthread_attr_getstack(&attr, &saddr, &ssize);
	if (ssize != stack_size || saddr != stack_addr)
	{	
		perror(ERROR_PREFIX "got the wrong stacksize or stackaddr");
		exit(PTS_FAIL);
	}	
	/* printf("saddr = %p, ssize = %u\n", saddr, ssize); */

	pthread_exit(0);
	return NULL;
}

int main()
{
	pthread_t new_th;
	pthread_attr_t attr;
	size_t ssize;
	void *saddr;
	int rc;

	/* Initialize attr */
	rc = pthread_attr_init(&attr);
	if( rc != 0) {
		perror(ERROR_PREFIX "pthread_attr_init");
		exit(PTS_UNRESOLVED);
	}
	
	/* Get the default stack_addr and stack_size value */	
	rc = pthread_attr_getstack(&attr, &stack_addr, &stack_size); 	
	if( rc != 0) {
		perror(ERROR_PREFIX "pthread_attr_getstack");
		exit(PTS_UNRESOLVED);
	}
	/* printf("stack_addr = %p, stack_size = %u\n", stack_addr, stack_size); */

	stack_size = PTHREAD_STACK_MIN;

	if (posix_memalign (&stack_addr, sysconf(_SC_PAGE_SIZE), 
            stack_size) != 0)
    	{
      		perror (ERROR_PREFIX "out of memory while "
                        "allocating the stack memory");
      		exit(PTS_UNRESOLVED);
    	}
	/* printf("stack_addr = %p, stack_size = %u\n", stack_addr, stack_size); */

	rc = pthread_attr_setstack(&attr, stack_addr, stack_size);
        if (rc != 0 ) {
                perror(ERROR_PREFIX "pthread_attr_setstack");
                exit(PTS_UNRESOLVED);
        }

	rc = pthread_attr_getstack(&attr, &saddr, &ssize);
        if (rc != 0 ) {
                perror(ERROR_PREFIX "pthread_attr_getstack");
                exit(PTS_UNRESOLVED);
        }
	/* printf("saddr = %p, ssize = %u\n", saddr, ssize); */

	rc = pthread_create(&new_th, &attr, thread_func, NULL);
	if (rc !=0 ) {
		perror(ERROR_PREFIX "failed to create a thread");
                exit(PTS_FAIL);
        }

	rc = pthread_join(new_th, NULL);
	if(rc != 0)
        {
                perror(ERROR_PREFIX "pthread_join");
		exit(PTS_UNRESOLVED);
        }

	rc = pthread_attr_destroy(&attr);
	if(rc != 0)
        {
                perror(ERROR_PREFIX "pthread_attr_destroy");
		exit(PTS_UNRESOLVED);
        }
	
	printf("Test PASSED\n");
	return PTS_PASS;
}


