/*
 * 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_getschedpolicy()
 *
 * Steps:
 * 1.  Initialize a pthread_attr_t object using pthread_attr_init()
 * 2.  Call pthread_attr_setschedpolicy with policy parameter
 * 3.  Call pthread_attr_getschedpolicy to get the policy
 *
 */

#include <pthread.h>
#include <stdio.h>
#include <stdlib.h>
#include "posixtest.h"

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

#define FIFOPOLICY SCHED_FIFO
#define RRPOLICY SCHED_RR
#define OTHERPOLICY SCHED_OTHER

static int verify_policy(pthread_attr_t * attr, int policytype)
{
	int rc;
	int policy;

	rc = pthread_attr_getschedpolicy(attr, &policy);
	if (rc != 0) {
		printf(ERROR_PREFIX "pthread_attr_getschedpolicy\n");
		exit(PTS_UNRESOLVED);
	}
	switch (policytype) {
	case SCHED_FIFO:
		if (policy != FIFOPOLICY) {
			printf(ERROR_PREFIX "got wrong policy param\n");
			exit(PTS_FAIL);
		}
		break;
	case SCHED_RR:
		if (policy != RRPOLICY) {
			printf(ERROR_PREFIX "got wrong policy param\n");
			exit(PTS_FAIL);
		}
		break;
	case SCHED_OTHER:
		if (policy != OTHERPOLICY) {
			printf(ERROR_PREFIX "got wrong policy param\n");
			exit(PTS_FAIL);
		}
		break;
	}
	return 0;
}

int main(void)
{
	int rc = 0;
	pthread_attr_t attr;

	rc = pthread_attr_init(&attr);
	if (rc != 0) {
		printf(ERROR_PREFIX "pthread_attr_init\n");
		exit(PTS_UNRESOLVED);
	}

	rc = pthread_attr_setschedpolicy(&attr, FIFOPOLICY);
	if (rc != 0) {
		printf(ERROR_PREFIX "pthread_attr_setschedpolicy\n");
		exit(PTS_UNRESOLVED);
	}
	verify_policy(&attr, FIFOPOLICY);

	rc = pthread_attr_setschedpolicy(&attr, RRPOLICY);
	if (rc != 0) {
		printf(ERROR_PREFIX "pthread_attr_setschedpolicy\n");
		exit(PTS_UNRESOLVED);
	}
	verify_policy(&attr, RRPOLICY);

	rc = pthread_attr_setschedpolicy(&attr, OTHERPOLICY);
	if (rc != 0) {
		printf(ERROR_PREFIX "pthread_attr_setschedpolicy\n");
		exit(PTS_UNRESOLVED);
	}
	verify_policy(&attr, OTHERPOLICY);

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