//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
//
// copyright            : (C) 2008 by Eran Ifrah                            
// file name            : winprocess.cpp              
//                                                                          
// -------------------------------------------------------------------------
// A                                                                        
//              _____           _      _     _ _                            
//             /  __ \         | |    | |   (_) |                           
//             | /  \/ ___   __| | ___| |    _| |_ ___                      
//             | |    / _ \ / _  |/ _ \ |   | | __/ _ )                     
//             | \__/\ (_) | (_| |  __/ |___| | ||  __/                     
//              \____/\___/ \__,_|\___\_____/_|\__\___|                     
//                                                                          
//                                                  F i l e                 
//                                                                          
//    This program is free software; you can redistribute it and/or modify  
//    it under the terms of the GNU General Public License as published by  
//    the Free Software Foundation; either version 2 of the License, or     
//    (at your option) any later version.                                   
//                                                                          
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
 #include "winprocess.h"
#ifdef __WXMSW__
#include "wx/filefn.h"
#include <memory>

/*static*/ 
WinProcess* WinProcess::Execute(const wxString& cmd, wxString &errMsg, const wxString &workingDir)
{
	SECURITY_ATTRIBUTES saAttr;
	BOOL fSuccess;
	
	wxString wd(workingDir);
	if(workingDir.IsEmpty()){
		wd = wxGetCwd();
	}
	
	// Set the bInheritHandle flag so pipe handles are inherited.
	saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);
	saAttr.bInheritHandle = TRUE;
	saAttr.lpSecurityDescriptor = NULL;

	// The steps for redirecting child process's STDOUT: 
	//     1. Save current STDOUT, to be restored later. 
	//     2. Create anonymous pipe to be STDOUT for child process. 
	//     3. Set STDOUT of the parent process to be write handle to 
	//        the pipe, so it is inherited by the child process. 
	//     4. Create a noninheritable duplicate of the read handle and
	//        close the inheritable read handle.  
	
	WinProcess *prc = new WinProcess();

	// Save the handle to the current STDOUT.  
	prc->hSaveStdout = GetStdHandle(STD_OUTPUT_HANDLE);  
	
	// Create a pipe for the child process's STDOUT.  
	if( !CreatePipe( &prc->hChildStdoutRd, &prc->hChildStdoutWr, &saAttr, 0) )
	{
		delete prc;
		return NULL;
	}

	// Set a write handle to the pipe to be STDOUT.  
	if( !SetStdHandle(STD_OUTPUT_HANDLE, prc->hChildStdoutWr) )
	{
		delete prc;
		return NULL;
	}
   
	// Create noninheritable read handle and close the inheritable read handle. 
    fSuccess = DuplicateHandle( GetCurrentProcess(), prc->hChildStdoutRd,
        GetCurrentProcess(),  &prc->hChildStdoutRdDup , 
		0,  FALSE,
        DUPLICATE_SAME_ACCESS );
	if( !fSuccess )
	{
		delete prc;
        return NULL;
	}
	CloseHandle( prc->hChildStdoutRd );
	
	// The steps for redirecting child process's STDIN: 
	//     1.  Save current STDIN, to be restored later. 
	//     2.  Create anonymous pipe to be STDIN for child process. 
	//     3.  Set STDIN of the parent to be the read handle to the 
	//         pipe, so it is inherited by the child process. 
	//     4.  Create a noninheritable duplicate of the write handle, 
	//         and close the inheritable write handle.  

	// Save the handle to the current STDIN. 
	prc->hSaveStdin = GetStdHandle(STD_INPUT_HANDLE);  

	// Create a pipe for the child process's STDIN.  
	if( !CreatePipe(&prc->hChildStdinRd, &prc->hChildStdinWr, &saAttr, 0) )
	{
		delete prc;
		return NULL;
	}
	// Set a read handle to the pipe to be STDIN.  
	if( !SetStdHandle(STD_INPUT_HANDLE, prc->hChildStdinRd) ) 
	{
		delete prc;
		return NULL;
	}
	// Duplicate the write handle to the pipe so it is not inherited.  
	fSuccess = DuplicateHandle(GetCurrentProcess(), prc->hChildStdinWr, 
		GetCurrentProcess(), &prc->hChildStdinWrDup, 
		0, FALSE,                  // not inherited       
		DUPLICATE_SAME_ACCESS ); 
	if( !fSuccess ) 
	{
		delete prc;
		return NULL;
	}
	CloseHandle(prc->hChildStdinWr);  

	// Execute the child process
	STARTUPINFO siStartInfo;  

	// Set up members of STARTUPINFO structure.  
	ZeroMemory( &siStartInfo, sizeof(STARTUPINFO) );
	siStartInfo.cb = sizeof(STARTUPINFO);  
	
	siStartInfo.dwFlags = STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW; ;
	siStartInfo.hStdInput = prc->hChildStdinRd;
	siStartInfo.hStdOutput = prc->hChildStdoutWr;
	siStartInfo.hStdError = prc->hChildStdoutWr;

	// Set the window to hide
	siStartInfo.wShowWindow = SW_HIDE;
	BOOL ret = CreateProcess( NULL	,
			(WCHAR*)cmd.GetData(),						// shell line execution command
			NULL,										// process security attributes 
			NULL,										// primary thread security attributes 
			TRUE,										// handles are inherited 
			CREATE_NO_WINDOW,							// creation flags 
			NULL,										// use parent's environment 
			wd.c_str(),									// CD to tmp dir
			&siStartInfo,								// STARTUPINFO pointer 
			&prc->piProcInfo);							// receives PROCESS_INFORMATION 
	if( ret )
	{
		prc->dwProcessId = prc->piProcInfo.dwProcessId;
	}
	else
	{
		int err = GetLastError();
		wxUnusedVar(err);
		delete prc;
		return NULL;
	}
	
	// After process creation, restore the saved STDIN and STDOUT.  
	if( !SetStdHandle(STD_INPUT_HANDLE, prc->hSaveStdin) )
	{
		delete prc;
		return NULL;
	}
	if( !SetStdHandle(STD_OUTPUT_HANDLE, prc->hSaveStdout) )
	{
		
		delete prc;
		return NULL;
	}
	return prc;
}

WinProcess::WinProcess()
{
	hChildStdinRd = NULL;
	hChildStdoutWr = NULL;
	hChildStdinWrDup = NULL;
	hChildStdoutRdDup = NULL;
	piProcInfo.hProcess = NULL;
	piProcInfo.hThread = NULL;
}

WinProcess::~WinProcess()
{
}

bool WinProcess::Read(wxString& buff)
{
	DWORD dwRead;
	DWORD dwMode;
	DWORD dwTimeout;
	char *chBuf = new char [65536+1]; 	//64K should be sufficient buffer
	memset(chBuf, 0, 65536+1);
	
	std::auto_ptr<char> sp(chBuf);
	
	// Make the pipe to non-blocking mode
	dwMode = PIPE_READMODE_BYTE | PIPE_NOWAIT;
	dwTimeout = 1000;
	SetNamedPipeHandleState(hChildStdoutRdDup, 
							&dwMode,
							NULL,
							&dwTimeout);	// Timeout of 30 seconds
	if( ReadFile( hChildStdoutRdDup, chBuf, 65536, &dwRead, NULL) || dwRead == 0)
	{
		chBuf[dwRead/sizeof(char)] = 0;
		//printf("%s\n", chBuf);
		buff = wxString(chBuf, wxConvUTF8);
		if(buff.IsEmpty() && dwRead > 0){
			//conversion failed
			buff = wxString::From8BitData(chBuf);
		}
		return true;
	}
	return false;
}

bool WinProcess::Write(const wxString& buff)
{
	DWORD dwMode;
	DWORD dwTimeout;
	char  chBuf[4097]; 

	wxString tmpCmd = buff;
	tmpCmd = tmpCmd.Trim().Trim(false);
	tmpCmd += wxT("\n");

	strcpy(chBuf, tmpCmd.mb_str());

	// Make the pipe to non-blocking mode
	dwMode = PIPE_READMODE_BYTE | PIPE_NOWAIT;
	dwTimeout = 30000;
	SetNamedPipeHandleState(hChildStdinWrDup,
							&dwMode,
							NULL,
							NULL);				// Timeout of 30 seconds
	DWORD dwWritten;
	if(!WriteFile(hChildStdinWrDup, chBuf, (unsigned long)strlen(chBuf), &dwWritten, NULL))
		return false;
	return true;
}

bool WinProcess::IsAlive()
{
	DWORD dwExitCode;
	if(GetExitCodeProcess(piProcInfo.hProcess, &dwExitCode))
	{
		if(dwExitCode == STILL_ACTIVE)
			return true;
	}
	return false;
}


void WinProcess::Cleanup()
{
	if(IsAlive())
		TerminateProcess(piProcInfo.hProcess, 255);
	
	CloseHandle( hChildStdinRd);
	CloseHandle( hChildStdoutWr);
	CloseHandle( hChildStdinWrDup );
	CloseHandle( hChildStdoutRdDup );
	CloseHandle( piProcInfo.hProcess );
	CloseHandle( piProcInfo.hThread );

	hChildStdinRd = NULL;
	hChildStdoutWr = NULL;
	hChildStdinWrDup = NULL;
	hChildStdoutRdDup = NULL;
	piProcInfo.hProcess = NULL;
	piProcInfo.hThread = NULL;
}

#endif //__WXMSW__
