9086b799e1b209d31f314bfabc99db541b7544fc
[occt.git] / src / OSD / OSD_Process.cxx
1 // Copyright (c) 1998-1999 Matra Datavision
2 // Copyright (c) 1999-2012 OPEN CASCADE SAS
3 //
4 // The content of this file is subject to the Open CASCADE Technology Public
5 // License Version 6.5 (the "License"). You may not use the content of this file
6 // except in compliance with the License. Please obtain a copy of the License
7 // at http://www.opencascade.org and read it completely before using this file.
8 //
9 // The Initial Developer of the Original Code is Open CASCADE S.A.S., having its
10 // main offices at: 1, place des Freres Montgolfier, 78280 Guyancourt, France.
11 //
12 // The Original Code and all software distributed under the License is
13 // distributed on an "AS IS" basis, without warranty of any kind, and the
14 // Initial Developer hereby disclaims all such warranties, including without
15 // limitation, any warranties of merchantability, fitness for a particular
16 // purpose or non-infringement. Please see the License for the specific terms
17 // and conditions governing the rights and limitations under the License.
18
19
20 #ifdef HAVE_CONFIG_H
21 # include <config.h>
22 #endif
23
24 #ifndef WNT
25
26 #include <OSD_Process.ixx>
27 #include <OSD_WhoAmI.hxx>
28 #include <OSD_Environment.hxx>
29
30 const OSD_WhoAmI Iam = OSD_WProcess;
31
32 #include <errno.h>
33
34 #ifdef HAVE_UNISTD_H
35 # include <unistd.h>
36 #endif
37
38 #include <stdlib.h>
39
40 #ifdef HAVE_SYS_PARAM_H
41 # include <sys/param.h>
42 #endif
43
44 #if defined(HAVE_TIME_H) || defined(WNT)
45 # include <time.h>
46 #endif
47
48 #ifdef HAVE_SYS_TIME_H
49 # include <sys/time.h>
50 #endif
51
52 #ifdef HAVE_PWD_H
53 # include <pwd.h>       // For command getpwuid
54 #endif
55
56 OSD_Process::OSD_Process(){
57 }
58
59
60 void OSD_Process::Spawn (const TCollection_AsciiString& cmd,
61                          const Standard_Boolean /*ShowWindow*/)
62 {
63  system(cmd.ToCString());
64 }
65
66
67 void OSD_Process::TerminalType(TCollection_AsciiString& Name){
68 TCollection_AsciiString which="TERM";
69 OSD_Environment term (which,"");
70
71  term.Value();
72  which = term.Value();
73  Name = term.Name();
74 }
75
76
77 // Get date of system date
78
79 Quantity_Date  OSD_Process::SystemDate(){
80 Quantity_Date result;
81 Standard_Integer month=0,day=0,year=0,hh=0,mn=0,ss=0;
82 struct tm transfert;
83 struct timeval tval;
84 struct timezone tzone;
85 int status;
86
87  status = gettimeofday( &tval, &tzone );
88  if (status == -1) myError.SetValue (errno, Iam, "GetSystem");
89  else {
90   memcpy(&transfert, localtime((time_t *)&tval.tv_sec), sizeof(struct
91 tm));
92   month = transfert.tm_mon + 1;  // Add to January (month #1)
93   day   = transfert.tm_mday;
94   year  = transfert.tm_year;
95   hh    = transfert.tm_hour;
96   mn    = transfert.tm_min ;
97   ss    = transfert.tm_sec ;
98 }
99
100  result.SetValues ( month, day, year+1900, hh, mn, ss);
101  return (result);
102 }
103
104
105 Standard_Integer OSD_Process::ProcessId(){
106  return (getpid());
107 }
108
109
110 Standard_Integer OSD_Process::UserId(){
111  return (getuid());
112 }
113
114
115 TCollection_AsciiString OSD_Process::UserName(){
116  struct passwd *infos;
117  infos = getpwuid(getuid()); 
118  TCollection_AsciiString result=infos->pw_name;
119
120  return(result);
121 }
122
123 Standard_Boolean OSD_Process::IsSuperUser (){
124   if (getuid()) {
125     return Standard_False;
126   }
127   else {
128     return Standard_True;
129   }
130 }
131
132
133 OSD_Path OSD_Process::CurrentDirectory(){
134 char cwd[MAXPATHLEN+1] ;
135 OSD_Path result;
136 TCollection_AsciiString Name;
137
138  if (!getcwd(cwd,MAXPATHLEN+1))
139    myError.SetValue (errno, Iam, "Where");
140  else {
141    Name = cwd;
142
143 //   JPT : August,20 1993. This code has been replaced by #ifdef ... #endif
144 //   position = Name.SearchFromEnd(".");
145 //   if (position != -1){
146 //     Ext = Name;
147 //     Ext.Remove(1,position);
148 //     Name.Remove( position,Ext.Length()+1);
149 //   }
150 //   result.SetValues("","","","","",Name,Ext);
151 //   End
152
153 #if defined(vax) || defined(__vms)
154    Standard_Integer iDisk = Name.Search(":");
155    if (iDisk){
156      TCollection_AsciiString Disk;
157      TCollection_AsciiString Directory;
158      Disk = Name.SubString(1,iDisk-1);
159      Directory = Name.SubString(iDisk+1,Name.Length());
160      result.SetValues("","","",Disk,Directory,"","");
161    }
162 #else
163    Name += TCollection_AsciiString("/");
164    result = OSD_Path(Name);
165    //      result.SetValues("","","","",Name,"","");
166 #endif
167
168  }
169 return (result);
170 }
171
172
173 void OSD_Process::SetCurrentDirectory(const OSD_Path& where){
174 TCollection_AsciiString Name;
175 int status;
176
177  where.SystemName(Name);
178
179  status = chdir (Name.ToCString());
180  if (status == -1) myError.SetValue(errno, Iam, "Move to directory");
181 }
182
183
184 void OSD_Process::Reset(){
185  myError.Reset();
186 }
187
188 Standard_Boolean OSD_Process::Failed()const{
189  return( myError.Failed());
190 }
191
192 void OSD_Process::Perror() {
193  myError.Perror();
194 }
195
196
197 Standard_Integer OSD_Process::Error()const{
198  return( myError.Error());
199 }
200
201 #else
202
203 //------------------------------------------------------------------------
204 //-------------------  WNT Sources of OSD_Path ---------------------------
205 //------------------------------------------------------------------------
206
207 //it is important to undefine NOUSER and enforce including <windows.h> before
208 //Standard_Macro.hxx defines it and includes <windows.h> causing compilation errors
209 #ifdef NOUSER
210 #undef NOUSER /* we need SW_HIDE from windows.h */
211 #endif
212 #include <windows.h>
213
214 #ifdef SetCurrentDirectory
215 # undef SetCurrentDirectory /* undefine SetCurrentDirectory from <winbase.h> to correctly include <OSD_Process.hxx> */
216 #endif
217 #include <OSD_Process.hxx>
218
219 #include <OSD_Path.hxx>
220 #include <Quantity_Date.hxx>
221
222 #include <OSD_WNT_1.hxx>
223 #include <LMCONS.H> /// pour UNLEN  ( see MSDN about GetUserName() )
224
225
226 #pragma warning( disable : 4700 )
227
228 void _osd_wnt_set_error ( OSD_Error&, OSD_WhoAmI, ... );
229
230 OSD_Process :: OSD_Process () {
231
232 }  // end constructor
233
234 void OSD_Process :: Spawn ( const TCollection_AsciiString& cmd ,
235                             const Standard_Boolean ShowWindow /* = Standard_True */) {
236
237  STARTUPINFO         si;
238  PROCESS_INFORMATION pi;
239
240  ZeroMemory (  &si, sizeof ( STARTUPINFO )  );
241
242  si.cb = sizeof ( STARTUPINFO );
243  //============================================
244  //---> Added by Stephane Routelous ( stephane.routelous@altavista.net )        [16.03.01]
245  //---> Reason : to allow to hide the window
246  if ( !ShowWindow )
247  {
248          si.dwFlags             = STARTF_USESHOWWINDOW;
249          si.wShowWindow = SW_HIDE;      
250  }
251  //<--- End Added by Stephane Routelous ( stephane.routelous@altavista.net )    [16.03.01]
252  //============================================
253
254  if (!CreateProcess (
255       NULL, (char *)cmd.ToCString (), NULL, NULL, TRUE, CREATE_NEW_CONSOLE, NULL, NULL, &si, &pi
256                     )
257  )
258
259   _osd_wnt_set_error ( myError, OSD_WProcess );
260
261  else {
262  
263   CloseHandle ( pi.hThread );
264
265   WaitForSingleObject ( pi.hProcess, INFINITE );
266
267   CloseHandle ( pi.hProcess );
268  
269  }  // end else
270
271 }  // end OSD_Process :: Spawn
272
273 void OSD_Process :: TerminalType ( TCollection_AsciiString& Name ) {
274
275  Name = TEXT( "WIN32 console" );
276
277 }  // end OSD_Process :: TerminalType
278
279 Quantity_Date OSD_Process :: SystemDate () {
280
281  Quantity_Date retVal;
282  SYSTEMTIME    st;
283
284  GetLocalTime ( &st );
285
286  retVal.SetValues (
287          st.wMonth, st.wDay, st.wYear, st.wHour, st.wMinute, st.wSecond, st.wMilliseconds
288         );
289
290  return retVal;
291
292 }  // end OSD_Process :: SystemDate
293
294 Standard_Integer OSD_Process :: UserId () {
295
296  PSID         retVal        = NULL;
297  HANDLE       hProcessToken = INVALID_HANDLE_VALUE;
298  PTOKEN_OWNER pTKowner      = NULL;
299
300  if (  !OpenProcessToken (
301          GetCurrentProcess (),
302          TOKEN_QUERY, &hProcessToken
303         ) ||
304         (  pTKowner = ( PTOKEN_OWNER )GetTokenInformationEx (
305                                        hProcessToken, TokenOwner
306                                       )
307         ) == NULL ||
308         (  retVal   = CopySidEx ( pTKowner -> Owner )  ) == NULL
309  )
310
311   _osd_wnt_set_error ( myError, OSD_WProcess );
312
313  if ( hProcessToken != INVALID_HANDLE_VALUE ) CloseHandle ( hProcessToken );
314  if ( pTKowner      != NULL                 ) FreeTokenInformation ( pTKowner );
315
316  return ( Standard_Integer )retVal;
317
318 }  // end OSD_Process :: UserId
319
320 TCollection_AsciiString OSD_Process :: UserName () 
321 {
322         Standard_PCharacter pBuff = new char[UNLEN + 1];
323         DWORD                   dwSize = UNLEN + 1;
324         TCollection_AsciiString retVal;
325         if (  !GetUserName ( pBuff, &dwSize )  )
326         {
327                 _osd_wnt_set_error ( myError, OSD_WProcess );
328         }
329         else
330         {
331                 TCollection_AsciiString theTmpUserName(pBuff,(int)dwSize -1 );
332                 retVal = theTmpUserName;
333         }
334         delete [] pBuff;
335         return retVal;
336 }  // end OSD_Process :: UserName
337
338 Standard_Boolean OSD_Process :: IsSuperUser () {
339
340  Standard_Boolean retVal = FALSE;
341  PSID             pSIDadmin;
342  HANDLE           hProcessToken = INVALID_HANDLE_VALUE;
343  PTOKEN_GROUPS    pTKgroups;
344
345  if (  !OpenProcessToken (
346          GetCurrentProcess (),
347          TOKEN_QUERY, &hProcessToken
348         ) ||
349         (  pTKgroups = ( PTOKEN_GROUPS )GetTokenInformationEx (
350                                          hProcessToken, TokenGroups
351                                         )
352         ) == NULL
353  )
354
355   _osd_wnt_set_error ( myError, OSD_WProcess );
356
357  else {
358  
359   pSIDadmin = AdminSid ();
360
361   for ( int i = 0; i < ( int )pTKgroups -> GroupCount; ++i )
362
363    if (  EqualSid ( pTKgroups -> Groups[ i ].Sid, pSIDadmin )  ) {
364    
365     retVal = TRUE;
366     break;
367    
368    }  // end if
369  
370  }  // end else
371
372  if ( hProcessToken != INVALID_HANDLE_VALUE ) CloseHandle ( hProcessToken );
373  if ( pTKgroups     != NULL                 ) FreeTokenInformation ( pTKgroups );
374
375  return retVal;
376
377 }  // end OSD_Process :: IsSuperUser
378
379 Standard_Integer OSD_Process :: ProcessId () {
380
381  return ( Standard_Integer )GetCurrentProcessId ();
382
383 }  // end OSD_Process :: ProcessId
384
385 OSD_Path OSD_Process :: CurrentDirectory () {
386
387  Standard_PCharacter pBuff;
388  DWORD            dwSize = 0;
389  OSD_Path         retVal;
390
391  dwSize = GetCurrentDirectory ( dwSize, pBuff );
392  pBuff  = new Standard_Character[ dwSize ];
393
394  if (   (  dwSize = GetCurrentDirectory ( dwSize, pBuff )  ) == NULL   )
395
396   _osd_wnt_set_error ( myError, OSD_WProcess );
397
398  else
399  
400   retVal = OSD_Path ( pBuff );
401  
402  delete[] pBuff;
403
404  return retVal;
405  
406 }  // end OSD_Process :: CurrentDirectory
407
408 void OSD_Process :: SetCurrentDirectory ( const OSD_Path& where ) {
409
410 #ifdef UNICODE
411 # define SetCurrentDirectory  SetCurrentDirectoryW
412 #else
413 # define SetCurrentDirectory  SetCurrentDirectoryA
414 #endif  // UNICODE
415
416  TCollection_AsciiString path;
417
418  where.SystemName ( path );
419
420  if (   !::SetCurrentDirectory (  path.ToCString ()  )   )
421
422   _osd_wnt_set_error ( myError, OSD_WProcess );
423
424 }  // end OSD_Process :: SetCurrentDirectory
425
426 Standard_Boolean OSD_Process :: Failed () const {
427
428  return myError.Failed ();
429
430 }  // end OSD_Process :: Failed
431
432 void OSD_Process :: Reset () {
433
434  myError.Reset ();
435
436 }  // end OSD_Process :: Reset
437
438 void OSD_Process :: Perror () {
439
440  myError.Perror ();
441
442 }  // end OSD_Process :: Perror
443
444 Standard_Integer OSD_Process :: Error () const {
445
446  return myError.Error ();
447
448 }  // end OSD_Process :: Error
449
450 #endif