/*
* 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;
	}