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