[wxWidgets.git] / src / msdos / utilsdos.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        utils.cpp
// Purpose:     DOS implementations of utility functions
// Author:      Vaclav Slavik, M.J.Wetherell
// Id:          $Id$
// Copyright:   (c) 2001-2002 SciTech Software, Inc. (www.scitechsoft.com)
//              (c) 2005 M.J.Wetherell
// Licence:     wxWindows licence
/////////////////////////////////////////////////////////////////////////////

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
    #pragma hdrstop
#endif

#include "wx/utils.h"
#include "wx/string.h"

#include "wx/intl.h"
#include "wx/apptrait.h"
#include "wx/log.h"
#include "wx/process.h"
#include "wx/confbase.h"        // for wxExpandEnvVars()
#include "wx/app.h"
#include "wx/cmdline.h"
#include "wx/filename.h"
#include "wx/wfstream.h"

#include <stdarg.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <signal.h>
#include <time.h>
#include <dos.h>
#include <process.h>

//----------------------------------------------------------------------------
// Sleep
//----------------------------------------------------------------------------

void wxSleep(int nSecs)
{
    wxMilliSleep(1000 * nSecs);
}

void wxMilliSleep(unsigned long milliseconds)
{
#if HAVE_USLEEP || defined __DJGPP__
    usleep(milliseconds * 1000);
#elif defined __WATCOMC__
    delay(milliseconds);
#else
    clock_t start = clock();
    while ((clock() - start) * 1000 / CLOCKS_PER_SEC < (clock_t)milliseconds)
    {
        // yield if in a multitasking environment
        // "Release Current Virtual Machine's Time Slice" in DPMI 1.0
        REGS r;
        memset(&r, 0, sizeof(r));
        r.x.ax = 0x1680;
        int386(0x2f, &r, &r);
    }
#endif
}

void wxMicroSleep(unsigned long microseconds)
{
#if HAVE_USLEEP || defined __DJGPP__
    usleep(microseconds);
#else
    wxMilliSleep(microseconds/1000);
#endif
}

//----------------------------------------------------------------------------
// Get/Set environment variables
//----------------------------------------------------------------------------

bool wxGetEnv(const wxString& var, wxString *value)
{
    // wxGetenv is defined as getenv()
    wxChar *p = wxGetenv(var);
    if ( !p )
        return FALSE;

    if ( value )
        *value = p;

    return TRUE;
}

bool wxSetEnv(const wxString& variable, const wxChar *value)
{
    wxString s = variable;
    if ( value )
        s << _T('=') << value;

    // transform to ANSI
    const char *p = s.mb_str();

    // the string will be free()d by libc
    char *buf = (char *)malloc(strlen(p) + 1);
    strcpy(buf, p);

    return putenv(buf) == 0;
}

//----------------------------------------------------------------------------
// Hostname, username, home directory
//----------------------------------------------------------------------------

// Based on the MSW implementation
//
// Respects the following environment variables in this order: %HomeDrive% +
// %HomePath%, %UserProfile%, $HOME. Otherwise takes program's directory if
// wxApp has been initialised, otherwise returns ".".
//
const wxChar* wxGetHomeDir(wxString *home)
{
    wxString& strDir = *home;

    strDir.clear();

    // try HOMEDRIVE/PATH
    const wxChar *szHome = wxGetenv(wxT("HOMEDRIVE"));
    if ( szHome != NULL )
        strDir << szHome;
    szHome = wxGetenv(wxT("HOMEPATH"));

    if ( szHome != NULL )
    {
        strDir << szHome;

        // the idea is that under NT these variables have default values of
        // "%systemdrive%:" and "\\". As we don't want to create our config
        // files in the root directory of the system drive, we will create it
        // in our program's dir. However, if the user took care to set
        // HOMEPATH to something other than "\\", we suppose that he knows
        // what he is doing and use the supplied value.
        if ( wxStrcmp(szHome, wxT("\\")) == 0 )
            strDir.clear();
    }

    if ( strDir.empty() )
    {
        // If we have a valid USERPROFILE directory, as is the case in
        // Windows NT, 2000 and XP, we should use that as our home directory.
        szHome = wxGetenv(wxT("USERPROFILE"));

        if ( szHome != NULL )
            strDir = szHome;
    }

    if ( strDir.empty() )
    {
        // If we have a valid HOME directory, as is used on many machines
        // that have unix utilities on them, we should use that.
        szHome = wxGetenv(wxT("HOME"));

        if ( szHome != NULL )
        {
            strDir = szHome;
            // when msys sets %HOME% it uses '/' (cygwin uses '\\')
            strDir.Replace(_T("/"), _T("\\"));
        }
    }

    if ( !strDir.empty() )
    {
        // sometimes the value of HOME may be "%USERPROFILE%", so reexpand the
        // value once again, it shouldn't hurt anyhow
        strDir = wxExpandEnvVars(strDir);
    }
    else // fall back to the program directory
    {
        if ( wxTheApp )
        {
            wxString prog(wxTheApp->argv[0]);
#ifdef __DJGPP__
            // djgpp startup code switches the slashes around, so restore them
            prog.Replace(_T("/"), _T("\\"));
#endif
            // it needs to be a full path to be usable
            if ( prog.compare(1, 2, _T(":\\")) == 0 )
                wxSplitPath(prog, &strDir, NULL, NULL);
        }
        if ( strDir.empty() )
        {
            strDir = _T(".");
        }
    }

    return strDir.c_str();
}

wxChar *wxGetUserHome(const wxString& user)
{
    static wxString home;

    if (user.empty() || user == wxGetUserId())
        return wx_const_cast(wxChar*, wxGetHomeDir(&home));
    else
        return _T("");
}

#if WXWIN_COMPATIBILITY_2_2
void wxFatalError(const wxString &msg, const wxString &title)
{
    wxFprintf( stderr, _("Error ") );
    if (!title.IsNull()) wxFprintf( stderr, wxT("%s "), WXSTRINGCAST(title) );
    if (!msg.IsNull()) wxFprintf( stderr, wxT(": %s"), WXSTRINGCAST(msg) );
    wxFprintf( stderr, wxT(".\n") );
    exit(3); // the same exit code as for abort()
}
#endif // WXWIN_COMPATIBILITY_2_2

// returns %UserName%, $USER or just "user"
//
bool wxGetUserId(wxChar *buf, int n)
{
    const wxChar *user = wxGetenv(_T("UserName"));

    if (!user)
        user = wxGetenv(_T("USER"));

    if (!user)
        user = _T("user");

    wxStrncpy(buf, user, n);
    return true;
}

bool wxGetUserName(wxChar *buf, int n)
{
    return wxGetUserId(buf, n);
}

// returns %ComputerName%, or $HOSTNAME, or "host"
//
bool wxGetHostName(wxChar *buf, int n)
{
    const wxChar *host = wxGetenv(_T("ComputerName"));
    
    if (!host)
        host = wxGetenv(_T("HOSTNAME"));

    if (!host)
        host = _T("host");

    wxStrncpy(buf, host, n);
    return true;
}

// adds %UserDnsDomain% to wxGetHostName()
//
bool wxGetFullHostName(wxChar *buf, int n)
{
    wxGetHostName(buf, n);

    const wxChar *domain = wxGetenv(_T("UserDnsDomain"));

    if (domain)
        wxStrncat(wxStrncat(buf, _T("."), n), domain, n);

    return true;
}

//----------------------------------------------------------------------------
// Processes
//----------------------------------------------------------------------------

unsigned long wxGetProcessId()
{
    return (unsigned long)getpid();
}

int wxKill(long pid, wxSignal sig, wxKillError *rc, int WXUNUSED(flags))
{
    int result = -1;

    if (pid != (long)wxGetProcessId())
    {
        result = raise(sig);
        if (rc)
            *rc = result == 0 ? wxKILL_OK : wxKILL_BAD_SIGNAL;
    }
    else
    {
        wxLogDebug(_T("wxKill can only send signals to the current process under MSDOS"));
        if (rc)
            *rc = wxKILL_NO_PROCESS;
    }

    return result;
}

bool wxShell(const wxString& command /*=wxEmptyString*/)
{
    // FIXME: suspend/resume gui
    int result = system(command);

    if (result == -1)
        wxLogSysError(_("can't execute '%s'"), command.c_str());

    return result == 0;
}

long wxExecute(const wxString& command, int flags, wxProcess *process)
{
    // FIXME: shouldn't depend on wxCmdLineParser
    wxArrayString args(wxCmdLineParser::ConvertStringToArgs(command));
    size_t n = args.size();
    wxChar **argv = new wxChar*[n + 1];

    argv[n] = NULL;
    while (n-- > 0)
        argv[n] = wx_const_cast(wxChar*, args[n].c_str());

    long result = wxExecute(argv, flags, process);

    delete [] argv;
    return result;
}

#if wxUSE_STREAMS

// A wxFFileInputStream that deletes the file in it's destructor
// 
class wxTempFileInStream : public wxFFileInputStream
{
public:
    wxTempFileInStream(const wxString& name)
        : wxFFileInputStream(name, _T("rt"))
    { }

    ~wxTempFileInStream()
    {
        m_file->Close();
        wxRemoveFile(m_file->GetName());
    }
};

// A file descriptor that can be redirected to a file
//
class wxRedirectableFd
{
public:
    wxRedirectableFd(int fd) : m_fd(fd), m_dup(-1) { }
    ~wxRedirectableFd();

    // Redirect the descriptor to a file, similar to ANSI C's freopen, but
    // for low level descriptors. The desctructor un-redirects. If O_CREAT
    // is in the flags then the destructor will delete the file unless it is
    // given away with Release().
    bool Reopen(const wxString& name, int flags);

    // un-redirect the redirected file descriptor, closing the file, and give
    // away the filename without deleting it
    wxString Release();

private:
    // un-redirect the descriptor, closing the file
    void Restore();

    int m_fd;
    int m_dup;
    wxString m_name;
};

wxRedirectableFd::~wxRedirectableFd()
{
    Restore();
    if (!m_name.empty())
        wxRemoveFile(m_name);
}

bool wxRedirectableFd::Reopen(const wxString& name, int flags)
{
    wxASSERT(m_dup == -1);
    bool result = false;

    // save a duplicate so that the descriptor can be closed now and
    // restored later
    m_dup = dup(m_fd);

    if (m_dup != -1)
    {
        int tmp = open(name.mb_str(), flags);

        if (tmp != -1)
        {
            close(m_fd);

            if (flags & O_CREAT)
                m_name = name;

            result = dup2(tmp, m_fd) == m_fd;
            close(tmp);
        }
    }

    if (!result)
        wxLogSysError(_("error opening '%s'"), name.c_str());

    return result;
}

void wxRedirectableFd::Restore()
{
    if (m_dup != -1)
    {
        close(m_fd);
        dup2(m_dup, m_fd);
        close(m_dup);
        m_dup = -1;
    }
}

wxString wxRedirectableFd::Release()
{
    Restore();
    wxString name = m_name;
    m_name.clear();
    return name;
}

#endif // wxUSE_STREAMS

// wxExecute implementation
//
long wxExecute(wxChar **argv, int flags, wxProcess *process)
{
#if wxUSE_STREAMS
    const int STDIN = 0;
    const int STDOUT = 1;
    const int STDERR = 2;

    wxRedirectableFd in(STDIN), out(STDOUT), err(STDERR);
    bool redirect = process && process->IsRedirected() && (flags & wxEXEC_SYNC);

    if (redirect)
    {
        // close stdin/out/err and reopen them as files
        if (!in.Reopen(_T("NUL"), O_RDONLY | O_TEXT))
            return -1;

        if (!out.Reopen(wxFileName::CreateTempFileName(_T("out")),
                        O_CREAT | O_WRONLY | O_TRUNC | O_TEXT))
            return -1;

        if (!err.Reopen(wxFileName::CreateTempFileName(_T("err")),
                        O_CREAT | O_WRONLY | O_TRUNC | O_TEXT))
            return -1;
    }
#endif // wxUSE_STREAMS

    // FIXME: suspend/resume gui
    int mode = flags & wxEXEC_SYNC ? P_WAIT : P_NOWAIT;
    int result = spawnvp(mode, argv[0], argv);

    if (result == -1)
        wxLogSysError(_("can't execute '%s'"), argv[0]);

#if wxUSE_STREAMS
    if (redirect)
        process->SetPipeStreams(new wxTempFileInStream(out.Release()),
                                new wxFFileOutputStream(_T("NUL"), _T("wt")),
                                new wxTempFileInStream(err.Release()));
#endif // wxUSE_STREAMS

    return result;
}

//----------------------------------------------------------------------------
// Traits for console apps
//----------------------------------------------------------------------------

wxToolkitInfo& wxConsoleAppTraits::GetToolkitInfo()
{
    static wxToolkitInfo info;
    info.versionMajor = _osmajor;
    info.versionMinor = _osminor;
    info.name = _T("wxBase");
    info.os = wxDOS;
    return info;
}
Commit	Line	Data
81d9e625 MW	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: utils.cpp
	3	// Purpose: DOS implementations of utility functions
a3c0d768	4	// Author: Vaclav Slavik, M.J.Wetherell
81d9e625 MW	5	// Id: $Id$
81d9e625 MW	6	// Copyright: (c) 2001-2002 SciTech Software, Inc. (www.scitechsoft.com)
a3c0d768	7	// (c) 2005 M.J.Wetherell
81d9e625 MW	8	// Licence: wxWindows licence
	9	/////////////////////////////////////////////////////////////////////////////
	10
	11	// For compilers that support precompilation, includes "wx.h".
	12	#include "wx/wxprec.h"
	13
	14	#ifdef __BORLANDC__
	15	#pragma hdrstop
	16	#endif
	17
	18	#include "wx/utils.h"
	19	#include "wx/string.h"
	20
	21	#include "wx/intl.h"
	22	#include "wx/apptrait.h"
	23	#include "wx/log.h"
	24	#include "wx/process.h"
a3c0d768 MW	25	#include "wx/confbase.h" // for wxExpandEnvVars()
	26	#include "wx/app.h"
	27	#include "wx/cmdline.h"
	28	#include "wx/filename.h"
	29	#include "wx/wfstream.h"
81d9e625 MW	30
	31	#include <stdarg.h>
	32	#include <string.h>
	33	#include <sys/stat.h>
	34	#include <sys/types.h>
	35	#include <unistd.h>
a3c0d768	36	#include <signal.h>
81d9e625	37	#include <time.h>
a3c0d768 MW	38	#include <dos.h>
a3c0d768 MW	39	#include <process.h>
81d9e625 MW	40
81d9e625 MW	41	//----------------------------------------------------------------------------
a3c0d768	42	// Sleep
81d9e625 MW	43	//----------------------------------------------------------------------------
	44
	45	void wxSleep(int nSecs)
	46	{
	47	wxMilliSleep(1000 * nSecs);
	48	}
	49
	50	void wxMilliSleep(unsigned long milliseconds)
	51	{
	52	#if HAVE_USLEEP \|\| defined __DJGPP__
	53	usleep(milliseconds * 1000);
	54	#elif defined __WATCOMC__
	55	delay(milliseconds);
	56	#else
	57	clock_t start = clock();
	58	while ((clock() - start) * 1000 / CLOCKS_PER_SEC < (clock_t)milliseconds)
a3c0d768 MW	59	{
	60	// yield if in a multitasking environment
	61	// "Release Current Virtual Machine's Time Slice" in DPMI 1.0
	62	REGS r;
	63	memset(&r, 0, sizeof(r));
	64	r.x.ax = 0x1680;
	65	int386(0x2f, &r, &r);
	66	}
81d9e625 MW	67	#endif
	68	}
	69
	70	void wxMicroSleep(unsigned long microseconds)
	71	{
	72	#if HAVE_USLEEP \|\| defined __DJGPP__
	73	usleep(microseconds);
	74	#else
	75	wxMilliSleep(microseconds/1000);
	76	#endif
	77	}
	78
a3c0d768 MW	79	//----------------------------------------------------------------------------
	80	// Get/Set environment variables
	81	//----------------------------------------------------------------------------
81d9e625 MW	82
	83	bool wxGetEnv(const wxString& var, wxString *value)
	84	{
	85	// wxGetenv is defined as getenv()
	86	wxChar *p = wxGetenv(var);
	87	if ( !p )
	88	return FALSE;
	89
	90	if ( value )
	91	*value = p;
	92
	93	return TRUE;
	94	}
	95
	96	bool wxSetEnv(const wxString& variable, const wxChar *value)
	97	{
	98	wxString s = variable;
	99	if ( value )
	100	s << _T('=') << value;
	101
	102	// transform to ANSI
	103	const char *p = s.mb_str();
	104
	105	// the string will be free()d by libc
	106	char buf = (char )malloc(strlen(p) + 1);
	107	strcpy(buf, p);
	108
	109	return putenv(buf) == 0;
	110	}
	111
a3c0d768 MW	112	//----------------------------------------------------------------------------
	113	// Hostname, username, home directory
	114	//----------------------------------------------------------------------------
	115
	116	// Based on the MSW implementation
	117	//
	118	// Respects the following environment variables in this order: %HomeDrive% +
	119	// %HomePath%, %UserProfile%, $HOME. Otherwise takes program's directory if
	120	// wxApp has been initialised, otherwise returns ".".
	121	//
81d9e625 MW	122	const wxChar* wxGetHomeDir(wxString *home)
81d9e625 MW	123	{
a3c0d768	124	wxString& strDir = *home;
81d9e625	125
a3c0d768 MW	126	strDir.clear();
	127
	128	// try HOMEDRIVE/PATH
	129	const wxChar *szHome = wxGetenv(wxT("HOMEDRIVE"));
	130	if ( szHome != NULL )
	131	strDir << szHome;
	132	szHome = wxGetenv(wxT("HOMEPATH"));
	133
	134	if ( szHome != NULL )
	135	{
	136	strDir << szHome;
	137
	138	// the idea is that under NT these variables have default values of
	139	// "%systemdrive%:" and "\\". As we don't want to create our config
	140	// files in the root directory of the system drive, we will create it
	141	// in our program's dir. However, if the user took care to set
	142	// HOMEPATH to something other than "\\", we suppose that he knows
	143	// what he is doing and use the supplied value.
	144	if ( wxStrcmp(szHome, wxT("\\")) == 0 )
	145	strDir.clear();
	146	}
	147
	148	if ( strDir.empty() )
	149	{
	150	// If we have a valid USERPROFILE directory, as is the case in
	151	// Windows NT, 2000 and XP, we should use that as our home directory.
	152	szHome = wxGetenv(wxT("USERPROFILE"));
	153
	154	if ( szHome != NULL )
	155	strDir = szHome;
	156	}
	157
	158	if ( strDir.empty() )
	159	{
	160	// If we have a valid HOME directory, as is used on many machines
	161	// that have unix utilities on them, we should use that.
	162	szHome = wxGetenv(wxT("HOME"));
	163
	164	if ( szHome != NULL )
	165	{
	166	strDir = szHome;
	167	// when msys sets %HOME% it uses '/' (cygwin uses '\\')
	168	strDir.Replace(_T("/"), _T("\\"));
	169	}
	170	}
	171
	172	if ( !strDir.empty() )
	173	{
	174	// sometimes the value of HOME may be "%USERPROFILE%", so reexpand the
	175	// value once again, it shouldn't hurt anyhow
	176	strDir = wxExpandEnvVars(strDir);
	177	}
	178	else // fall back to the program directory
	179	{
	180	if ( wxTheApp )
	181	{
	182	wxString prog(wxTheApp->argv[0]);
	183	#ifdef __DJGPP__
	184	// djgpp startup code switches the slashes around, so restore them
	185	prog.Replace(_T("/"), _T("\\"));
	186	#endif
	187	// it needs to be a full path to be usable
	188	if ( prog.compare(1, 2, _T(":\\")) == 0 )
	189	wxSplitPath(prog, &strDir, NULL, NULL);
190	}
191	if ( strDir.empty() )
192	{
193	strDir = _T(".");
194	}
195	}
196
197	return strDir.c_str();
81d9e625 MW	198	}
81d9e625 MW	199
a3c0d768	200	wxChar *wxGetUserHome(const wxString& user)
81d9e625	201	{
a3c0d768 MW	202	static wxString home;
	203
	204	if (user.empty() \|\| user == wxGetUserId())
	205	return wx_const_cast(wxChar*, wxGetHomeDir(&home));
	206	else
	207	return _T("");
81d9e625 MW	208	}
	209
	210	#if WXWIN_COMPATIBILITY_2_2
	211	void wxFatalError(const wxString &msg, const wxString &title)
	212	{
	213	wxFprintf( stderr, _("Error ") );
	214	if (!title.IsNull()) wxFprintf( stderr, wxT("%s "), WXSTRINGCAST(title) );
	215	if (!msg.IsNull()) wxFprintf( stderr, wxT(": %s"), WXSTRINGCAST(msg) );
	216	wxFprintf( stderr, wxT(".\n") );
	217	exit(3); // the same exit code as for abort()
	218	}
	219	#endif // WXWIN_COMPATIBILITY_2_2
	220
a3c0d768 MW	221	// returns %UserName%, $USER or just "user"
	222	//
	223	bool wxGetUserId(wxChar *buf, int n)
81d9e625	224	{
a3c0d768 MW	225	const wxChar *user = wxGetenv(_T("UserName"));
	226
	227	if (!user)
	228	user = wxGetenv(_T("USER"));
	229
	230	if (!user)
	231	user = _T("user");
	232
	233	wxStrncpy(buf, user, n);
	234	return true;
81d9e625 MW	235	}
81d9e625 MW	236
a3c0d768	237	bool wxGetUserName(wxChar *buf, int n)
81d9e625	238	{
a3c0d768	239	return wxGetUserId(buf, n);
81d9e625 MW	240	}
81d9e625 MW	241
a3c0d768 MW	242	// returns %ComputerName%, or $HOSTNAME, or "host"
	243	//
	244	bool wxGetHostName(wxChar *buf, int n)
81d9e625	245	{
a3c0d768 MW	246	const wxChar *host = wxGetenv(_T("ComputerName"));
	247
	248	if (!host)
	249	host = wxGetenv(_T("HOSTNAME"));
	250
	251	if (!host)
	252	host = _T("host");
	253
	254	wxStrncpy(buf, host, n);
	255	return true;
81d9e625 MW	256	}
81d9e625 MW	257
a3c0d768 MW	258	// adds %UserDnsDomain% to wxGetHostName()
	259	//
	260	bool wxGetFullHostName(wxChar *buf, int n)
	261	{
	262	wxGetHostName(buf, n);
	263
	264	const wxChar *domain = wxGetenv(_T("UserDnsDomain"));
	265
	266	if (domain)
	267	wxStrncat(wxStrncat(buf, _T("."), n), domain, n);
	268
	269	return true;
	270	}
	271
	272	//----------------------------------------------------------------------------
	273	// Processes
	274	//----------------------------------------------------------------------------
	275
	276	unsigned long wxGetProcessId()
81d9e625	277	{
a3c0d768	278	return (unsigned long)getpid();
81d9e625 MW	279	}
81d9e625 MW	280
a3c0d768	281	int wxKill(long pid, wxSignal sig, wxKillError *rc, int WXUNUSED(flags))
81d9e625	282	{
a3c0d768 MW	283	int result = -1;
	284
	285	if (pid != (long)wxGetProcessId())
	286	{
	287	result = raise(sig);
	288	if (rc)
	289	*rc = result == 0 ? wxKILL_OK : wxKILL_BAD_SIGNAL;
	290	}
	291	else
	292	{
	293	wxLogDebug(_T("wxKill can only send signals to the current process under MSDOS"));
	294	if (rc)
	295	*rc = wxKILL_NO_PROCESS;
	296	}
	297
	298	return result;
81d9e625 MW	299	}
81d9e625 MW	300
a3c0d768	301	bool wxShell(const wxString& command /=wxEmptyString/)
81d9e625	302	{
a3c0d768 MW	303	// FIXME: suspend/resume gui
	304	int result = system(command);
	305
	306	if (result == -1)
	307	wxLogSysError(_("can't execute '%s'"), command.c_str());
	308
	309	return result == 0;
81d9e625 MW	310	}
81d9e625 MW	311
a3c0d768	312	long wxExecute(const wxString& command, int flags, wxProcess *process)
81d9e625	313	{
a3c0d768 MW	314	// FIXME: shouldn't depend on wxCmdLineParser
	315	wxArrayString args(wxCmdLineParser::ConvertStringToArgs(command));
	316	size_t n = args.size();
	317	wxChar *argv = new wxChar[n + 1];
	318
	319	argv[n] = NULL;
	320	while (n-- > 0)
	321	argv[n] = wx_const_cast(wxChar*, args[n].c_str());
	322
	323	long result = wxExecute(argv, flags, process);
	324
	325	delete [] argv;
	326	return result;
81d9e625 MW	327	}
81d9e625 MW	328
a3c0d768 MW	329	#if wxUSE_STREAMS
	330
	331	// A wxFFileInputStream that deletes the file in it's destructor
	332	//
	333	class wxTempFileInStream : public wxFFileInputStream
	334	{
	335	public:
	336	wxTempFileInStream(const wxString& name)
	337	: wxFFileInputStream(name, _T("rt"))
	338	{ }
	339
	340	~wxTempFileInStream()
	341	{
	342	m_file->Close();
	343	wxRemoveFile(m_file->GetName());
	344	}
	345	};
	346
	347	// A file descriptor that can be redirected to a file
	348	//
	349	class wxRedirectableFd
	350	{
	351	public:
	352	wxRedirectableFd(int fd) : m_fd(fd), m_dup(-1) { }
	353	~wxRedirectableFd();
	354
	355	// Redirect the descriptor to a file, similar to ANSI C's freopen, but
	356	// for low level descriptors. The desctructor un-redirects. If O_CREAT
	357	// is in the flags then the destructor will delete the file unless it is
	358	// given away with Release().
	359	bool Reopen(const wxString& name, int flags);
	360
	361	// un-redirect the redirected file descriptor, closing the file, and give
	362	// away the filename without deleting it
	363	wxString Release();
	364
	365	private:
	366	// un-redirect the descriptor, closing the file
	367	void Restore();
	368
	369	int m_fd;
	370	int m_dup;
	371	wxString m_name;
	372	};
	373
	374	wxRedirectableFd::~wxRedirectableFd()
	375	{
	376	Restore();
	377	if (!m_name.empty())
	378	wxRemoveFile(m_name);
	379	}
	380
	381	bool wxRedirectableFd::Reopen(const wxString& name, int flags)
	382	{
	383	wxASSERT(m_dup == -1);
	384	bool result = false;
	385
	386	// save a duplicate so that the descriptor can be closed now and
	387	// restored later
	388	m_dup = dup(m_fd);
	389
	390	if (m_dup != -1)
	391	{
	392	int tmp = open(name.mb_str(), flags);
393
394	if (tmp != -1)
395	{
396	close(m_fd);
397
398	if (flags & O_CREAT)
399	m_name = name;
400
401	result = dup2(tmp, m_fd) == m_fd;
402	close(tmp);
403	}
404	}
405
406	if (!result)
407	wxLogSysError(_("error opening '%s'"), name.c_str());
408
409	return result;
410	}
411
412	void wxRedirectableFd::Restore()
413	{
414	if (m_dup != -1)
415	{
416	close(m_fd);
417	dup2(m_dup, m_fd);
418	close(m_dup);
419	m_dup = -1;
420	}
421	}
422
423	wxString wxRedirectableFd::Release()
424	{
425	Restore();
426	wxString name = m_name;
427	m_name.clear();
428	return name;
429	}
430
431	#endif // wxUSE_STREAMS
432
433	// wxExecute implementation
434	//
435	long wxExecute(wxChar *argv, int flags, wxProcess process)
436	{
437	#if wxUSE_STREAMS
438	const int STDIN = 0;
439	const int STDOUT = 1;
440	const int STDERR = 2;
441
442	wxRedirectableFd in(STDIN), out(STDOUT), err(STDERR);
443	bool redirect = process && process->IsRedirected() && (flags & wxEXEC_SYNC);
444
445	if (redirect)
446	{
447	// close stdin/out/err and reopen them as files
448	if (!in.Reopen(_T("NUL"), O_RDONLY \| O_TEXT))
449	return -1;
450
451	if (!out.Reopen(wxFileName::CreateTempFileName(_T("out")),
452	O_CREAT \| O_WRONLY \| O_TRUNC \| O_TEXT))
453	return -1;
454
455	if (!err.Reopen(wxFileName::CreateTempFileName(_T("err")),
456	O_CREAT \| O_WRONLY \| O_TRUNC \| O_TEXT))
457	return -1;
458	}
459	#endif // wxUSE_STREAMS
460
461	// FIXME: suspend/resume gui
462	int mode = flags & wxEXEC_SYNC ? P_WAIT : P_NOWAIT;
463	int result = spawnvp(mode, argv[0], argv);
464
465	if (result == -1)
466	wxLogSysError(_("can't execute '%s'"), argv[0]);
467
468	#if wxUSE_STREAMS
469	if (redirect)
470	process->SetPipeStreams(new wxTempFileInStream(out.Release()),
471	new wxFFileOutputStream(_T("NUL"), _T("wt")),
472	new wxTempFileInStream(err.Release()));
473	#endif // wxUSE_STREAMS
474
475	return result;
476	}
477
478	//----------------------------------------------------------------------------
479	// Traits for console apps
480	//----------------------------------------------------------------------------
481
81d9e625 MW	482	wxToolkitInfo& wxConsoleAppTraits::GetToolkitInfo()
	483	{
	484	static wxToolkitInfo info;
a3c0d768 MW	485	info.versionMajor = _osmajor;
a3c0d768 MW	486	info.versionMinor = _osminor;
81d9e625 MW	487	info.name = _T("wxBase");
	488	info.os = wxDOS;
	489	return info;
	490	}