/*

* Copyright (c) 1997-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description:

* THELLOU.C

* A variant of "Hello World" which reports various bits of its environment

* and returns an interesting exit status

* 

*

*/

#include <stdio.h>

#include <stdlib.h>

#include <unistd.h>

#include <string.h>

#include <sys/wait.h>

#include "CTEST.H"

#ifdef __WINS__

int triangle_recurse(wchar_t* prog, wchar_t* num, int progress)

	{

	printf("\nBut not under WINS where I'll be told 0\n");

	return 0;

	}

int triangle_parallel(wchar_t* prog, wchar_t* num)

	{

	printf("\nBut not under WINS where I'll be told 0\n");

	return 0;

	}

#else

int triangle_recurse(wchar_t* prog, wchar_t* num, int progress)

	{

	int n;

	int pid, ret;

	char cmd[100];

	wchar_t wcmd[100];

	int fids[3];

	char tmpbuf[100];

	wcstombs(tmpbuf, num, 50);

	n = atol(tmpbuf);

	if (n<1)

		return 0;

	wcstombs(tmpbuf, prog, 50);

	sprintf(cmd, "%s do_triangle %d", tmpbuf, n-1);

	mbstowcs(wcmd, cmd, 100);

	if (progress)

		{

		pid=wpopen3(wcmd, (wchar_t*)L"", 0, fids);

		if (pid<0)

			{

			fprintf(stderr, "Executing %s, ", cmd);

			perror("wpopen3 failed");

			return -1999999;

			}

		while (1)

			{

			int n=waitpid(pid, &ret, WNOHANG);

			if (n==pid)

				break;

			printf(".");

			fflush(stdout);

			sleep(1);

			}

		}

	else

		ret=wsystem(wcmd);

	return n+ret;

	}

int triangle_parallel(wchar_t* prog, wchar_t* num)

	{

	int n;

	int pid, pid1, pid2, ret1, ret2, base, split;

	char cmd[100];

	wchar_t wcmd[100];

	int fids1[3], fids2[3];

	char* basep;

	char tmpbuf[100];

	wcstombs(tmpbuf, num, 50);

	n = atol(tmpbuf);

	if (n<1)

		return 0;

	basep=getenv("TRIANGLE_BASE");

	if (basep==0)

		return 0;

	base=atol(basep);

	/* we have to add up the numbers base..n inclusive

	 */

	if (base==n)

		return n;

	if (base+1==n)

		return base+n;

	/* At least 3 numbers, so split it into subtasks for child processes

	 */

	split = (n-base)/2;

	wcstombs(tmpbuf, prog, 100);

	sprintf(cmd, "%s do_trianglep %d", tmpbuf, base+split);

	mbstowcs(wcmd, cmd, 100);

	pid1=wpopen3(wcmd, (wchar_t*)L"", 0, fids1);

	if (pid1<0)

		{

		fprintf(stderr, "Doing %d..%d of %d..%d", base+split, base, n, base);

		perror("popen3 failed");

		return -1999999;

		}

	sprintf(cmd, "%d", base+split+1);

	mbstowcs(wcmd, cmd, 100);

	wsetenv((wchar_t*)L"TRIANGLE_BASE", wcmd, 1);

	sprintf(cmd, "%s do_trianglep %d", prog, n);

	mbstowcs(wcmd, cmd, 100);

	pid2=wpopen3(wcmd, (wchar_t*)L"", 0, fids2);

	if (pid2<0)

		{

		fprintf(stderr, "Doing %d..%d of %d..%d", n, base+split+1, n, base);

		perror("wpopen3 failed");

		return -1999999;

		}

	/* Now collect the results */

	ret1=-10000;

	ret2=-10000;

	for (n=0; n<2; n++)

		{

		int ret=-3999999;

		pid=wait(&ret);

		if (pid<0)

			{

			perror("waitpid failed");

			return -2999999;

			}

		if (pid==pid1)

			{

			pid1=-1;

			ret1=ret;

			}

		else if (pid==pid2)

			{

			pid2=-1;

			ret2=ret;

			}

		else

			printf("Unexpected pid %d (not %d or %d)\n", pid, pid1, pid2);

		}

	return ret1+ret2;

	}

#endif

/**

Usage:  THELLOU [triangle[p] <n>]

@SYMTestCaseID          SYSLIB-STDLIB-CT-1096

@SYMTestCaseDesc	    Tests for printing to standard output stream

@SYMTestPriority 	    High

@SYMTestActions  	    Prints some stuff (looks at args, getenv("USER"), cwd) to stdout and stderr.

						Will compute triangle numbers by calling itself recursively, or parallel subdivison

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/		

int wmain (int argc, wchar_t *argv[])

	{

	wchar_t *user=wgetenv((wchar_t*)L"USER");

	char cwd[MAXPATHLEN];

	int ret=3, i;

	char tmpbuf[100];

	start_redirection_server();

	if (argc==3 && wcscmp(argv[1],(wchar_t*)L"do_triangle")==0)

		return triangle_recurse(argv[0], argv[2], 0);

	if (argc==3 && wcscmp(argv[1],(wchar_t*)L"do_trianglep")==0)

		return triangle_parallel(argv[0], argv[2]);

	if (user)

		{

		char tmp[100];

		wcstombs(tmp,user,100);

		printf("Hello %s\n", tmp);

		ret=4;

		}

	else

		printf("Greetings.\n");

	printf("I am process %d\n", getpid());

	if (getcwd(cwd,sizeof(cwd))!=0)

		printf("I am speaking to you from %s\n", cwd);

	printf("I have %d arguments: ", argc);

	for (i=0; i<argc; i++)

		{

		wcstombs(tmpbuf, argv[i], sizeof(tmpbuf));

		printf(">%s< ", tmpbuf);

		}

	printf("\r\n");

	printf("In a few moments I shall say %c to stderr:\n", 'a'+argc);

	fflush(stdout);

	fprintf(stderr, "%c\n", 'a'+argc);

	fflush(stderr);

	if (argc==3 && wcscmp(argv[1],(wchar_t*)L"triangle")==0)

		{

		printf("For my next trick I shall compute the triangle of %s: ", argv[2]);

		fflush(stdout);

		ret=triangle_recurse(argv[0], argv[2], 1);

		printf("it's %d\n", ret);

		}

	if (argc==3 && wcscmp(argv[1],(wchar_t*)L"trianglep")==0)

		{

		printf("I shall now compute the triangle of %s using a parallel algorithm: ", argv[2]);

		fflush(stdout);

		wsetenv((wchar_t*)L"TRIANGLE_BASE", (wchar_t*)L"1", 0);

		ret=triangle_parallel(argv[0], argv[2]);

		printf("it's %d\n", ret);

		}

	printf("Farewell...\nPress a key");

	getchar();

	return ret;

	}