Salome HOME
Fix for '53560: TC8.1.0:HEXABLOCK' issue.
[modules/kernel.git] / src / Container / Container_i.cxx
1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 //  SALOME Container : implementation of container and engine for Kernel
24 //  File   : Container_i.cxx
25 //  Author : Paul RASCLE, EDF - MARC TAJCHMAN, CEA 
26 //  Module : SALOME
27 //  $Header$
28 //#define private public
29 //
30 #include <string.h>
31 #include <stdio.h>
32 #include <time.h>
33 #include <sys/types.h>
34 #ifndef WIN32
35 #include <sys/time.h>
36 #include <dlfcn.h>
37 #include <unistd.h>
38 #include <sys/wait.h>
39 #else
40 #include <signal.h>
41 #include <process.h>
42 #include <direct.h>
43 int SIGUSR1 = 1000;
44 #endif
45
46 #include "utilities.h"
47 #include <SALOMEconfig.h>
48 #include CORBA_SERVER_HEADER(SALOME_Component)
49 #include CORBA_SERVER_HEADER(SALOME_Exception)
50 #include <pthread.h>  // must be before Python.h !
51 #include "SALOME_Container_i.hxx"
52 #include "SALOME_Component_i.hxx"
53 #include "SALOME_FileRef_i.hxx"
54 #include "SALOME_FileTransfer_i.hxx"
55 #include "Salome_file_i.hxx"
56 #include "SALOME_NamingService.hxx"
57 #include "Basics_Utils.hxx"
58
59 #ifdef _XOPEN_SOURCE
60 #undef _XOPEN_SOURCE
61 #endif
62
63 #include <Python.h>
64 #include "Container_init_python.hxx"
65
66 bool _Sleeping = false ;
67
68 // // Needed by multi-threaded Python --- Supervision
69 int _ArgC ;
70 char ** _ArgV ;
71
72 extern "C" {void ActSigIntHandler() ; }
73 #ifndef WIN32
74 extern "C" {void SigIntHandler(int, siginfo_t *, void *) ; }
75 #else
76 extern "C" {void SigIntHandler( int ) ; }
77 #endif
78
79 #ifndef WIN32
80 #define LIB "lib"
81 #define ENGINESO "Engine.so"
82 #else
83 #define LIB ""
84 #define ENGINESO "Engine.dll"
85 #endif
86
87 #ifdef WIN32
88 #define SEP ';'
89 #define SLASH '\\'
90 #else
91 #define SEP ':'
92 #define SLASH '/'
93 #endif
94
95 std::map<std::string, int> Engines_Container_i::_cntInstances_map;
96 std::map<std::string, void *> Engines_Container_i::_library_map;
97 std::map<std::string, void *> Engines_Container_i::_toRemove_map;
98 omni_mutex Engines_Container_i::_numInstanceMutex ;
99
100 static PyObject* _pyCont;
101
102 int checkifexecutable(const std::string&);
103 int findpathof(const std::string& path, std::string&, const std::string&);
104
105 /*! \class Engines_Container_i
106  *  \brief C++ implementation of Engines::Container interface
107  *
108  */
109
110
111 //=============================================================================
112 /*! 
113 *  Default constructor, not for use
114 */
115 //=============================================================================
116
117 Engines_Container_i::Engines_Container_i () :
118 _numInstance(0),_id(0),_NS(0)
119 {
120 }
121
122 //=============================================================================
123 /*! 
124 *  Construtor to use
125 */
126 //=============================================================================
127
128 Engines_Container_i::Engines_Container_i (CORBA::ORB_ptr orb, 
129                                           PortableServer::POA_ptr poa,
130                                           char *containerName ,
131                                           int argc , char* argv[],
132                                           bool activAndRegist,
133                                           bool isServantAloneInProcess
134                                           ) :
135   _numInstance(0),_isServantAloneInProcess(isServantAloneInProcess),_id(0),_NS(0)
136 {
137   _pid = (long)getpid();
138
139   if(activAndRegist)
140     ActSigIntHandler() ;
141
142   _argc = argc ;
143   _argv = argv ;
144
145   std::string hostname = Kernel_Utils::GetHostname();
146 #ifndef WIN32
147   MESSAGE(hostname << " " << getpid() << 
148     " Engines_Container_i starting argc " <<
149     _argc << " Thread " << pthread_self() ) ;
150 #else
151   MESSAGE(hostname << " " << _getpid() << 
152     " Engines_Container_i starting argc " << _argc<< " Thread " << pthread_self().p ) ;
153 #endif
154
155   int i = 0 ;
156   while ( _argv[ i ] )
157   {
158     MESSAGE("           argv" << i << " " << _argv[ i ]) ;
159     i++ ;
160   }
161
162   if ( argc < 2 )
163   {
164     INFOS("SALOME_Container usage : SALOME_Container ServerName");
165     ASSERT(0) ;
166   }
167   SCRUTE(argv[1]);
168   _isSupervContainer = false;
169
170   _orb = CORBA::ORB::_duplicate(orb) ;
171   _poa = PortableServer::POA::_duplicate(poa) ;
172
173   // Pour les containers paralleles: il ne faut pas enregistrer et activer
174   // le container generique, mais le container specialise
175
176   if(activAndRegist)
177   {
178     _id = _poa->activate_object(this);
179     _NS = new SALOME_NamingService();
180     _NS->init_orb( _orb ) ;
181     CORBA::Object_var obj=_poa->id_to_reference(*_id);
182     Engines::Container_var pCont 
183       = Engines::Container::_narrow(obj);
184     _remove_ref();
185
186     _containerName = _NS->BuildContainerNameForNS(containerName,
187       hostname.c_str());
188     SCRUTE(_containerName);
189     _NS->Register(pCont, _containerName.c_str());
190     MESSAGE("Engines_Container_i::Engines_Container_i : Container name "
191       << _containerName);
192
193     // Python: 
194     // import SALOME_Container
195     // pycont = SALOME_Container.SALOME_Container_i(containerIORStr)
196
197     CORBA::String_var sior =  _orb->object_to_string(pCont);
198     std::string myCommand="pyCont = SALOME_Container.SALOME_Container_i('";
199     myCommand += _containerName + "','";
200     myCommand += sior;
201     myCommand += "')\n";
202     SCRUTE(myCommand);
203
204     //[RNV]: Comment the PyEval_AcquireLock() and PyEval_ReleaseLock() because this 
205     //approach leads to the deadlock of the main thread of the application on Windows platform
206     //in case if cppContainer runs in the standalone mode. The problem with the PyThreadState 
207     //described by ABN seems not reproduced, to be checked carefully later...
208     PyGILState_STATE gstate = PyGILState_Ensure();
209     
210     //// [ABN]: using the PyGILState* API here is unstable. omniORB logic is invoked
211     //// by the Python code executed below, and in some (random) cases, the Python code
212     //// execution ends with a PyThreadState which was not the one we have here.
213     //// (TODO: understand why ...)
214     //// To be on the safe side we get and load the thread state ourselves:    
215     //PyEval_AcquireLock();  // get GIL
216     //PyThreadState * mainThreadState = PyThreadState_Get();
217     //PyThreadState_Swap(mainThreadState);
218
219 #ifdef WIN32
220     // mpv: this is temporary solution: there is a unregular crash if not
221     //Sleep(2000);
222     //
223     // first element is the path to Registry.dll, but it's wrong
224     PyRun_SimpleString("import sys\n");
225     PyRun_SimpleString("sys.path = sys.path[1:]\n");
226 #endif
227     PyRun_SimpleString("import SALOME_Container\n");
228     PyRun_SimpleString((char*)myCommand.c_str());
229     PyObject *mainmod = PyImport_AddModule("__main__");
230     PyObject *globals = PyModule_GetDict(mainmod);
231     _pyCont = PyDict_GetItemString(globals, "pyCont");
232
233     //PyThreadState_Swap(NULL);
234     //PyEval_ReleaseLock();
235     PyGILState_Release(gstate);
236
237     fileTransfer_i* aFileTransfer = new fileTransfer_i();
238     CORBA::Object_var obref=aFileTransfer->_this();
239     _fileTransfer = Engines::fileTransfer::_narrow(obref);
240     aFileTransfer->_remove_ref();
241   }
242 }
243
244 //=============================================================================
245 /*! 
246 *  Destructor
247 */
248 //=============================================================================
249
250 Engines_Container_i::~Engines_Container_i()
251 {
252   MESSAGE("Container_i::~Container_i()");
253   if(_id)
254     delete _id;
255   if(_NS)
256     delete _NS;
257   for(std::map<std::string,Engines::PyNode_var>::iterator it=_dftPyNode.begin();it!=_dftPyNode.end();it++)
258     {
259       Engines::PyNode_var tmpVar((*it).second);
260       if(!CORBA::is_nil(tmpVar))
261         tmpVar->UnRegister();
262     }
263   for(std::map<std::string,Engines::PyScriptNode_var>::iterator it=_dftPyScriptNode.begin();it!=_dftPyScriptNode.end();it++)
264     {
265       Engines::PyScriptNode_var tmpVar((*it).second);
266       if(!CORBA::is_nil(tmpVar))
267         tmpVar->UnRegister();
268     }
269 }
270
271 //=============================================================================
272 //! Get container name
273 /*! 
274 *  CORBA attribute: Container name (see constructor)
275 */
276 //=============================================================================
277
278 char* Engines_Container_i::name()
279 {
280   return CORBA::string_dup(_containerName.c_str()) ;
281 }
282
283 //=============================================================================
284 //! Get container working directory
285 /*! 
286 *  CORBA attribute: Container working directory 
287 */
288 //=============================================================================
289
290 char* Engines_Container_i::workingdir()
291 {
292   char wd[256];
293   getcwd (wd,256);
294   return CORBA::string_dup(wd) ;
295 }
296
297 //=============================================================================
298 //! Get container log file name
299 /*! 
300 *  CORBA attribute: Container log file name
301 */
302 //=============================================================================
303
304 char* Engines_Container_i::logfilename()
305 {
306   return CORBA::string_dup(_logfilename.c_str()) ;
307 }
308
309 //! Set container log file name
310 void Engines_Container_i::logfilename(const char* name)
311 {
312   _logfilename=name;
313 }
314
315 //=============================================================================
316 //! Get container host name
317 /*! 
318 *  CORBA method: Get the hostName of the Container (without domain extensions)
319 */
320 //=============================================================================
321
322 char* Engines_Container_i::getHostName()
323 {
324   std::string s = Kernel_Utils::GetHostname();
325   //  MESSAGE("Engines_Container_i::getHostName " << s);
326   return CORBA::string_dup(s.c_str()) ;
327 }
328
329 //=============================================================================
330 //! Get container PID
331 /*! 
332 *  CORBA method: Get the PID (process identification) of the Container
333 */
334 //=============================================================================
335
336 CORBA::Long Engines_Container_i::getPID()
337 {
338   return (CORBA::Long)getpid();
339 }
340
341 //=============================================================================
342 //! Ping the servant to check it is still alive
343 /*! 
344 *  CORBA method: check if servant is still alive
345 */
346 //=============================================================================
347 void Engines_Container_i::ping()
348 {
349   MESSAGE("Engines_Container_i::ping() pid "<< getpid());
350 }
351
352 //=============================================================================
353 //! Shutdown the container
354 /*! 
355 *  CORBA method, oneway: Server shutdown. 
356 *  - Container name removed from naming service,
357 *  - servant deactivation,
358 *  - orb shutdown if no other servants in the process 
359 */
360 //=============================================================================
361 void Engines_Container_i::Shutdown()
362 {
363   MESSAGE("Engines_Container_i::Shutdown()");
364
365   // Clear registered temporary files
366   clearTemporaryFiles();
367
368   /* For each component contained in this container
369   * tell it to self-destroy
370   */
371   std::map<std::string, Engines::EngineComponent_var>::iterator itm;
372   for (itm = _listInstances_map.begin(); itm != _listInstances_map.end(); itm++)
373     {
374       try
375         {
376           itm->second->destroy();
377         }
378       catch(const CORBA::Exception& e)
379         {
380           // ignore this entry and continue
381         }
382       catch(...)
383         {
384           // ignore this entry and continue
385         }
386     }
387   _listInstances_map.clear();
388
389   _NS->Destroy_FullDirectory(_containerName.c_str());
390   _NS->Destroy_Name(_containerName.c_str());
391   if(_isServantAloneInProcess)
392   {
393     MESSAGE("Effective Shutdown of container Begins...");
394     if(!CORBA::is_nil(_orb))
395       _orb->shutdown(0);
396   }
397 }
398
399 //=============================================================================
400 //! load a component implementation
401 /*! 
402 *  CORBA method
403 *  \param componentName         component name
404 *  \param reason                explains error when load fails
405 *  \return true if dlopen successfull or already done, false otherwise
406 */
407 //=============================================================================
408 bool
409 Engines_Container_i::load_component_Library(const char* componentName, CORBA::String_out reason)
410 {
411
412   //=================================================================
413   // --- C++ implementation section 
414   //=================================================================
415   std::string retso;
416   if(load_component_CppImplementation(componentName,retso))
417     {
418       reason=CORBA::string_dup("");
419       return true;
420     }
421   else if(retso != "ImplementationNotFound")
422     {
423       reason=CORBA::string_dup(retso.c_str());
424       return false;
425     }
426
427   retso="Component ";
428   retso+=componentName;
429   retso+=": Can't find C++ implementation ";
430   retso+=std::string(LIB) + componentName + ENGINESO;
431
432   //=================================================================
433   // --- Python implementation section 
434   //=================================================================
435   std::string retpy;
436   if(load_component_PythonImplementation(componentName,retpy))
437     {
438       reason=CORBA::string_dup("");
439       return true;
440     }
441   else if(retpy != "ImplementationNotFound")
442     {
443       reason=CORBA::string_dup(retpy.c_str());
444       return false;
445     }
446   
447   retpy="Component ";
448   retpy+=componentName;
449   retpy+=": Can't find python implementation ";
450   retpy+=componentName;
451   retpy+="(.py)";
452
453   //=================================================================
454   // -- Executable implementation section
455   //=================================================================
456   std::string retex;
457   if(load_component_ExecutableImplementation(componentName,retex))
458     {
459       reason=CORBA::string_dup("");
460       return true;
461     }
462   else if(retex != "ImplementationNotFound")
463     {
464       reason=CORBA::string_dup(retex.c_str());
465       return false;
466     }
467
468   retex="Component ";
469   retex+=componentName;
470   retex+=": Can't find executable implementation ";
471   retex+=componentName;
472   retex+=".exe";
473
474   std::string ret="Component implementation not found: ";
475   ret += componentName ;
476   ret += '\n' ;
477   ret += retso+ '\n' ;
478   ret += retpy+ '\n' ;
479   ret += retex+ '\n' ;
480
481   std::cerr << ret << std::endl;
482   reason=CORBA::string_dup(ret.c_str());
483
484   return false;
485 }
486
487 //=============================================================================
488 //! try to load a C++ component implementation
489 /*! 
490 *  C++ method: 
491 *  \param componentName      the name of the component (COMPONENT, for example)
492 *  \param reason             explains error when load fails
493 *  \return true if loading is successfull or already done, false otherwise
494 */
495 //=============================================================================
496 bool
497 Engines_Container_i::load_component_CppImplementation(const char* componentName, std::string& reason)
498 {
499   std::string aCompName(componentName);
500   std::string impl_name = std::string(LIB) + aCompName + ENGINESO;
501   SCRUTE(impl_name);
502
503   _numInstanceMutex.lock(); // lock to be alone
504   // (see decInstanceCnt, finalize_removal))
505   if (_toRemove_map.count(impl_name) != 0) _toRemove_map.erase(impl_name);
506   if (_library_map.count(impl_name) != 0)
507     {
508       MESSAGE("Library " << impl_name << " already loaded");
509       _numInstanceMutex.unlock();
510       reason="";
511       return true;
512     }
513   _numInstanceMutex.unlock();
514
515 #ifndef WIN32
516   void* handle;
517   handle = dlopen( impl_name.c_str() , RTLD_NOW ) ;
518   if ( !handle )
519     {
520       //not loadable. Try to find the lib file in LD_LIBRARY_PATH
521       std::string path;
522       char* p=getenv("LD_LIBRARY_PATH");
523       if(p)path=p;
524       path=path+SEP+"/usr/lib"+SEP+"/lib";
525
526       std::string pth;
527       if(findpathof(path, pth, impl_name))
528         {
529           //found but not loadable
530           reason="Component ";
531           reason+=aCompName;
532           reason+=": C++ implementation found ";
533           reason+=pth;
534           reason+=" but it is not loadable. Error:\n";
535           reason+=dlerror();
536           std::cerr << reason << std::endl;
537           return false;
538         }
539       else
540         {
541           //not found
542           //continue with other implementation
543           reason="ImplementationNotFound";
544           return false;
545         }
546     }
547 #else
548   HINSTANCE handle;
549   handle = LoadLibrary( impl_name.c_str() );
550   if ( !handle )
551     {
552       reason="ImplementationNotFound";
553     }
554 #endif
555
556   if ( handle )
557     {
558       _numInstanceMutex.lock();
559       _library_map[impl_name] = handle;
560       _numInstanceMutex.unlock();
561       reason="";
562       return true;
563     }
564
565   return false;
566
567 }
568 //=============================================================================
569 //! try to load a Python component implementation
570 /*! 
571 *  C++ method: 
572 *  \param componentName         name of the component
573 *  \param reason                explains error when load fails
574 *  \return true if loading is successfull or already done, false otherwise
575 */
576 //=============================================================================
577 bool
578 Engines_Container_i::load_component_PythonImplementation(const char* componentName, std::string& reason)
579 {
580   std::string aCompName(componentName);
581
582   _numInstanceMutex.lock() ; // lock to be alone (stl container write)
583   if (_library_map.count(aCompName) != 0)
584     {
585       _numInstanceMutex.unlock() ;
586       reason="";
587       return true; // Python Component, already imported
588     }
589   _numInstanceMutex.unlock() ;
590
591   PyGILState_STATE gstate = PyGILState_Ensure();
592   PyObject *result = PyObject_CallMethod(_pyCont,
593                                          (char*)"import_component",
594                                          (char*)"s",componentName);
595
596   reason=PyString_AsString(result);
597   Py_XDECREF(result);
598   SCRUTE(reason);
599   PyGILState_Release(gstate);
600
601   if (reason=="")
602     {
603       //Python component has been loaded (import componentName)
604       _numInstanceMutex.lock() ; // lock to be alone (stl container write)
605       _library_map[aCompName] = (void *)_pyCont; // any non O value OK
606       _numInstanceMutex.unlock() ;
607       MESSAGE("import Python: "<< aCompName <<" OK");
608       return true;
609     }
610   else if(reason=="ImplementationNotFound")
611     {
612       //Python implementation has not been found. Continue with other implementation
613       reason="ImplementationNotFound";
614     }
615   else
616     {
617       //Python implementation has been found but loading has failed
618       std::cerr << reason << std::endl;
619     }
620   return false;
621
622 }
623 //=============================================================================
624 //! try to load a Executable component implementation
625 /*! 
626 *  C++ method: 
627 *  \param componentName        name of the component
628 *  \param reason               explains error when load fails
629 *  \return true if loading is successfull or already done, false otherwise
630 */
631 //=============================================================================
632 bool
633 Engines_Container_i::load_component_ExecutableImplementation(const char* componentName, std::string& reason)
634 {
635   std::string aCompName(componentName);
636   std::string executable=aCompName+".exe";
637
638   std::string path;
639   std::string pth;
640
641   char* p=getenv("PATH");
642   if(p)path=p;
643
644   if (findpathof(path, pth, executable))
645     {
646       if(checkifexecutable(pth))
647         {
648           _numInstanceMutex.lock() ; // lock to be alone (stl container write)
649           _library_map[executable] = (void *)1; // any non O value OK
650           _numInstanceMutex.unlock() ;
651           MESSAGE("import executable: "<< pth <<" OK");
652           reason="";
653           return true;
654         }
655       reason="Component ";
656       reason+=aCompName;
657       reason+=": implementation found ";
658       reason+=pth;
659       reason+=" but it is not executable";
660       std::cerr << reason << std::endl;
661     }
662   else
663     {
664       reason="ImplementationNotFound";
665     }
666   return false;
667 }
668
669 //=============================================================================
670 //! Create a new component instance
671 /*! 
672 *  CORBA method: Creates a new servant instance of a component.
673 *  The servant registers itself to naming service and Registry.
674 *  \param genericRegisterName  Name of the component instance to register
675 *                         in Registry & Name Service (without _inst_n suffix)
676 *  \param studyId         0 for multiStudy instance, 
677 *                         study Id (>0) otherwise
678 *  \return a loaded component
679 */
680 //=============================================================================
681 Engines::EngineComponent_ptr
682 Engines_Container_i::create_component_instance(const char*genericRegisterName,
683                                                CORBA::Long studyId)
684 {
685   Engines::FieldsDict_var env = new Engines::FieldsDict;
686   char* reason;
687   Engines::EngineComponent_ptr compo =
688     create_component_instance_env(genericRegisterName, studyId, env, reason);
689   CORBA::string_free(reason);
690   return compo;
691 }
692
693 //=============================================================================
694 //! Create a new component instance with environment variables specified
695 /*! 
696 *  CORBA method: Creates a new servant instance of a component.
697 *  The servant registers itself to naming service and Registry.
698 *  \param genericRegisterName  Name of the component instance to register
699 *                         in Registry & Name Service (without _inst_n suffix)
700 *  \param studyId         0 for multiStudy instance, 
701 *                         study Id (>0) otherwise
702 *  \param env             dict of env variables
703 *  \param reason          explains error when create_component_instance_env fails
704 *  \return a loaded component
705 */
706 //=============================================================================
707 Engines::EngineComponent_ptr
708 Engines_Container_i::create_component_instance_env(const char*genericRegisterName,
709                                                    CORBA::Long studyId,
710                                                    const Engines::FieldsDict& env,
711                                                    CORBA::String_out reason)
712 {
713   if (studyId < 0)
714     {
715       INFOS("studyId must be > 0 for mono study instance, =0 for multiStudy");
716       reason=CORBA::string_dup("studyId must be > 0 for mono study instance, =0 for multiStudy");
717       return Engines::EngineComponent::_nil() ;
718     }
719
720   std::string error;
721   if (_library_map.count(genericRegisterName) != 0)
722     {
723       // It's a Python component
724       Engines::EngineComponent_ptr compo = createPythonInstance(genericRegisterName, studyId, error);
725       reason=CORBA::string_dup(error.c_str());
726       return compo;
727     }
728
729   std::string impl_name = std::string(LIB) + genericRegisterName + ENGINESO;
730   if (_library_map.count(impl_name) != 0)
731     {
732       // It's a C++ component
733       void* handle = _library_map[impl_name];
734       Engines::EngineComponent_ptr compo = createInstance(genericRegisterName, handle, studyId, error);
735       reason=CORBA::string_dup(error.c_str());
736       return compo;
737     }
738
739   impl_name = std::string(genericRegisterName) + ".exe";
740   if (_library_map.count(impl_name) != 0)
741     {
742       //It's an executable component
743       Engines::EngineComponent_ptr compo = createExecutableInstance(genericRegisterName, studyId, env, error);
744       reason=CORBA::string_dup(error.c_str());
745       return compo;
746     }
747
748   error="load_component_Library has probably not been called for component: ";
749   error += genericRegisterName;
750   INFOS(error);
751   reason=CORBA::string_dup(error.c_str());
752   return Engines::EngineComponent::_nil() ;
753 }
754
755 //=============================================================================
756 //! Create a new component instance (Executable implementation)
757 /*! 
758 *  \param CompName               Name of the component instance
759 *  \param studyId                0 for multiStudy instance, 
760 *                                study Id (>0) otherwise
761 *  \param env                    dict of env variables
762 *  \param reason                 explains error when creation fails
763 *  \return a loaded component
764 *
765 *   This component is implemented in an executable with name genericRegisterName.exe
766 *   It must register itself in Naming Service. The container waits some time (10 s max)
767 *   it's registration.
768 */
769 //=============================================================================
770 Engines::EngineComponent_ptr
771 Engines_Container_i::createExecutableInstance(std::string CompName, int studyId,
772                                               const Engines::FieldsDict& env,
773                                               std::string& reason)
774 {
775   Engines::EngineComponent_var iobject = Engines::EngineComponent::_nil() ;
776
777   _numInstanceMutex.lock() ; // lock on the instance number
778   _numInstance++ ;
779   int numInstance = _numInstance ;
780   _numInstanceMutex.unlock() ;
781
782   char aNumI[12];
783   sprintf( aNumI , "%d" , numInstance ) ;
784   std::string instanceName = CompName + "_inst_" + aNumI ;
785   std::string component_registerName = _containerName + "/" + instanceName;
786
787   //check if an entry exist in naming service
788   CORBA::Object_var nsobj = _NS->Resolve(component_registerName.c_str());
789   if ( !CORBA::is_nil(nsobj) )
790   {
791     // unregister the registered component
792     _NS->Destroy_Name(component_registerName.c_str());
793     //kill or shutdown it ???
794   }
795
796   // first arg container ior string
797   // second arg container name
798   // third arg instance name
799
800   Engines::Container_var pCont= _this();
801   CORBA::String_var sior =  _orb->object_to_string(pCont);
802
803   std::string command;
804   command="mkdir -p ";
805   command+=instanceName;
806   command+=";cd ";
807   command+=instanceName;
808   command+=";";
809   command+=CompName ;
810   command+=".exe";
811   command+=" ";
812   command+= sior; // container ior string
813   command+=" ";
814   command+=_containerName; //container name
815   command+=" ";
816   command+=instanceName; //instance name
817   command+=" &";
818   MESSAGE("SALOME_Container::create_component_instance command=" << command);
819
820 #ifndef WIN32
821   // use fork/execl instead of system to get finer control on env variables
822   int status;
823   pid_t pid = fork();
824   if(pid == 0) // child
825     {
826       for (CORBA::ULong i=0; i < env.length(); i++)
827         {
828           if (env[i].value.type()->kind() == CORBA::tk_string)
829             {
830               const char* value;
831               env[i].value >>= value;
832               std::string s(env[i].key);
833               s+='=';
834               s+=value;
835               putenv(strdup(s.c_str()));
836             }
837         }
838
839       execl("/bin/sh", "sh", "-c", command.c_str() , (char *)0);
840       status=-1;
841     }
842   else if(pid < 0)       // failed to fork
843     {
844       status=-1;
845     }
846   else            //parent
847     {
848       pid_t tpid;
849       do
850         {
851           tpid = wait(&status);
852         } while (tpid != pid);
853     }
854 #else
855   // launch component with a system call
856   int status=system(command.c_str());
857 #endif
858
859   if (status == -1)
860   {
861     reason="SALOME_Container::create_component_instance system failed (system command status -1)";
862     MESSAGE(reason);
863     return Engines::EngineComponent::_nil();
864   }
865 #ifndef WIN32
866   else if (WEXITSTATUS(status) == 217)
867   {
868     reason="SALOME_Container::create_component_instance system failed (system command status 217)";
869     MESSAGE(reason);
870     return Engines::EngineComponent::_nil();
871   }
872 #endif
873   else
874   {
875     int count=20;
876     if (getenv("TIMEOUT_TO_WAIT_EXE_COMPONENT") != 0)
877     {
878       std::string new_count_str = getenv("TIMEOUT_TO_WAIT_EXE_COMPONENT");
879       int new_count;
880       std::istringstream ss(new_count_str);
881       if (!(ss >> new_count))
882       {
883         INFOS("[Container] TIMEOUT_TO_WAIT_EXE_COMPONENT should be an int");
884       }
885       else
886         count = new_count;
887     }
888     INFOS("[Container] waiting " << count << " second steps exe component ");
889     CORBA::Object_var obj = CORBA::Object::_nil() ;
890     while ( CORBA::is_nil(obj) && count )
891     {
892 #ifndef WIN32
893       sleep( 1 ) ;
894 #else
895       Sleep(1000);
896 #endif
897       count-- ;
898       MESSAGE( count << ". Waiting for component " << CompName);
899       obj = _NS->Resolve(component_registerName.c_str());
900     }
901
902     if(CORBA::is_nil(obj))
903     {
904       reason="SALOME_Container::create_component_instance failed";
905       MESSAGE(reason);
906       return Engines::EngineComponent::_nil();
907     }
908     else
909     {
910       MESSAGE("SALOME_Container::create_component_instance successful");
911       iobject = Engines::EngineComponent::_narrow(obj);
912       _listInstances_map[instanceName] = iobject;
913       return iobject._retn();
914     }
915   }
916 }
917
918
919 //=============================================================================
920 //! Create a new component instance (Python implementation)
921 /*! 
922 *  \param CompName               Name of the component instance
923 *  \param studyId                0 for multiStudy instance, 
924 *                                study Id (>0) otherwise
925 *  \param reason                 explains error when creation fails
926 *  \return a loaded component
927 */
928 //=============================================================================
929 Engines::EngineComponent_ptr
930 Engines_Container_i::createPythonInstance(std::string CompName, int studyId,
931                                           std::string& reason)
932 {
933   Engines::EngineComponent_var iobject = Engines::EngineComponent::_nil() ;
934
935   _numInstanceMutex.lock() ; // lock on the instance number
936   _numInstance++ ;
937   int numInstance = _numInstance ;
938   _numInstanceMutex.unlock() ;
939
940   char aNumI[12];
941   sprintf( aNumI , "%d" , numInstance ) ;
942   std::string instanceName = CompName + "_inst_" + aNumI ;
943   std::string component_registerName = _containerName + "/" + instanceName;
944
945   PyGILState_STATE gstate = PyGILState_Ensure();
946   PyObject *result = PyObject_CallMethod(_pyCont,
947                                          (char*)"create_component_instance",
948                                          (char*)"ssl",
949                                          CompName.c_str(),
950                                          instanceName.c_str(),
951                                          studyId);
952   const char *ior;
953   const char *error;
954   PyArg_ParseTuple(result,"ss", &ior, &error);
955   std::string iors = ior;
956   reason=error;
957   Py_DECREF(result);
958   PyGILState_Release(gstate);
959
960   if( iors!="" )
961     {
962       CORBA::Object_var obj = _orb->string_to_object(iors.c_str());
963       iobject = Engines::EngineComponent::_narrow( obj ) ;
964       _listInstances_map[instanceName] = iobject;
965     }
966   return iobject._retn();
967 }
968
969 char *
970 Engines_Container_i::create_python_service_instance(const char * CompName,
971                                                     CORBA::String_out reason)
972 {
973   CORBA::Object_var object = CORBA::Object::_nil();
974
975   _numInstanceMutex.lock() ; // lock on the instance number
976   _numInstance++ ;
977   int numInstance = _numInstance ;
978   _numInstanceMutex.unlock() ;
979
980   char aNumI[12];
981   sprintf( aNumI , "%d" , numInstance ) ;
982   std::string instanceName = std::string(CompName) + "_inst_" + aNumI ;
983   std::string component_registerName = _containerName + "/" + instanceName;
984
985   PyGILState_STATE gstate = PyGILState_Ensure();
986   PyObject *result = PyObject_CallMethod(_pyCont,
987                                          (char*)"create_component_instance",
988                                          (char*)"ssl",
989                                          CompName,
990                                          instanceName.c_str(),
991                                          0);
992   const char *ior;
993   const char *error;
994   PyArg_ParseTuple(result,"ss", &ior, &error);
995   reason = CORBA::string_dup(error);
996   char * _ior = CORBA::string_dup(ior);
997   Py_DECREF(result);
998   PyGILState_Release(gstate);
999
1000   return _ior;
1001 }
1002
1003
1004 //=============================================================================
1005 //! Create a new component instance (C++ implementation)
1006 /*! 
1007 *  C++ method: create a servant instance of a component.
1008 *  \param genericRegisterName    Name of the component instance to register
1009 *                                in Registry & Name Service,
1010 *                                (without _inst_n suffix, like "COMPONENT")
1011 *  \param handle                 loaded library handle
1012 *  \param studyId                0 for multiStudy instance, 
1013 *                                study Id (>0) otherwise
1014 *  \param reason                 explains error when creation fails
1015 *  \return a loaded component
1016
1017 *  example with names:
1018 *    - aGenRegisterName = COMPONENT (= first argument)
1019 *    - _containerName = /Containers/cli76ce/FactoryServer
1020 *    - factoryName = COMPONENTEngine_factory
1021 *    - component_registerBase = /Containers/cli76ce/FactoryServer/COMPONENT
1022 *    - instanceName = COMPONENT_inst_1
1023 *    - component_registerName = /Containers/cli76ce/FactoryServer/COMPONENT_inst_1
1024 */
1025 //=============================================================================
1026 Engines::EngineComponent_ptr
1027 Engines_Container_i::createInstance(std::string genericRegisterName,
1028                                     void *handle,
1029                                     int studyId,
1030                                     std::string& reason)
1031 {
1032   // --- find the factory
1033
1034   std::string aGenRegisterName = genericRegisterName;
1035   std::string factory_name = aGenRegisterName + std::string("Engine_factory");
1036   SCRUTE(factory_name) ;
1037
1038   typedef PortableServer::ObjectId* (*FACTORY_FUNCTION) (CORBA::ORB_ptr,
1039                                                          PortableServer::POA_ptr, 
1040                                                          PortableServer::ObjectId *, 
1041                                                          const char *, 
1042                                                          const char *) ;
1043
1044 #ifndef WIN32
1045   FACTORY_FUNCTION Component_factory = (FACTORY_FUNCTION)dlsym( handle, factory_name.c_str() );
1046 #else
1047   FACTORY_FUNCTION Component_factory = (FACTORY_FUNCTION)GetProcAddress( (HINSTANCE)handle, factory_name.c_str() );
1048 #endif
1049
1050   if ( !Component_factory )
1051   {
1052     MESSAGE( "Can't resolve symbol: " + factory_name );
1053 #ifndef WIN32
1054     reason=dlerror();
1055     MESSAGE(reason);
1056 #endif
1057     return Engines::EngineComponent::_nil() ;
1058   }
1059
1060   // --- create instance
1061
1062   Engines::EngineComponent_var iobject = Engines::EngineComponent::_nil() ;
1063
1064   try
1065   {
1066     _numInstanceMutex.lock() ; // lock on the instance number
1067     _numInstance++ ;
1068     int numInstance = _numInstance ;
1069     _numInstanceMutex.unlock() ;
1070
1071     char aNumI[12];
1072     sprintf( aNumI , "%d" , numInstance ) ;
1073     std::string instanceName = aGenRegisterName + "_inst_" + aNumI ;
1074     std::string component_registerName =
1075       _containerName + "/" + instanceName;
1076
1077     // --- Instanciate required CORBA object
1078
1079     PortableServer::ObjectId *id ; //not owner, do not delete (nore use var)
1080     id = (Component_factory) ( _orb, _poa, _id, instanceName.c_str(),
1081                                                 aGenRegisterName.c_str() ) ;
1082     if (id == NULL)
1083       {
1084         reason="Can't get ObjectId from factory";
1085         INFOS(reason);
1086         return iobject._retn();
1087       }
1088
1089     // --- get reference & servant from id
1090
1091     CORBA::Object_var obj = _poa->id_to_reference(*id);
1092     iobject = Engines::EngineComponent::_narrow( obj ) ;
1093
1094     Engines_Component_i *servant =
1095       dynamic_cast<Engines_Component_i*>(_poa->reference_to_servant(iobject));
1096     ASSERT(servant);
1097     //SCRUTE(servant->_refcount_value());
1098     _numInstanceMutex.lock() ; // lock to be alone (stl container write)
1099     _listInstances_map[instanceName] = iobject;
1100     _cntInstances_map[aGenRegisterName] += 1;
1101     _numInstanceMutex.unlock() ;
1102     SCRUTE(aGenRegisterName);
1103     SCRUTE(_cntInstances_map[aGenRegisterName]);
1104     servant->setStudyId(studyId);
1105     servant->_remove_ref(); // do not need servant any more (remove ref from reference_to_servant)
1106     //SCRUTE(servant->_refcount_value());
1107
1108     // --- register the engine under the name
1109     //     containerName(.dir)/instanceName(.object)
1110
1111     _NS->Register( iobject , component_registerName.c_str() ) ;
1112     MESSAGE( component_registerName.c_str() << " bound" ) ;
1113   }
1114   catch (...)
1115   {
1116     reason="Container_i::createInstance exception catched";
1117     INFOS(reason) ;
1118   }
1119   return iobject._retn();
1120 }
1121
1122 //=============================================================================
1123 //! Find an existing (in the container) component instance
1124 /*!
1125 *  CORBA method: Finds a servant instance of a component
1126 *  \param registeredName  Name of the component in Registry or Name Service,
1127 *                         without instance suffix number
1128 *  \param studyId         0 if instance is not associated to a study,
1129 *                         >0 otherwise (== study id)
1130 *  \return the first instance found with same studyId
1131 */
1132 //=============================================================================
1133 Engines::EngineComponent_ptr
1134 Engines_Container_i::find_component_instance( const char* registeredName,
1135                                               CORBA::Long studyId)
1136 {
1137   Engines::EngineComponent_var anEngine = Engines::EngineComponent::_nil();
1138   std::map<std::string,Engines::EngineComponent_var>::iterator itm =_listInstances_map.begin();
1139   while (itm != _listInstances_map.end())
1140   {
1141     std::string instance = (*itm).first;
1142     SCRUTE(instance);
1143     if (instance.find(registeredName) == 0)
1144     {
1145       anEngine = (*itm).second;
1146       if (studyId == anEngine->getStudyId())
1147       {
1148         return anEngine._retn();
1149       }
1150     }
1151     itm++;
1152   }
1153   return anEngine._retn();
1154 }
1155
1156 //=============================================================================
1157 //! Remove the component instance from container
1158 /*!
1159 *  CORBA method: Stops the component servant, and deletes all related objects
1160 *  \param component_i     Component to be removed
1161 */
1162 //=============================================================================
1163
1164 void Engines_Container_i::remove_impl(Engines::EngineComponent_ptr component_i)
1165 {
1166   ASSERT(! CORBA::is_nil(component_i));
1167   std::string instanceName = component_i->instanceName() ;
1168   MESSAGE("unload component " << instanceName);
1169   _numInstanceMutex.lock() ; // lock to be alone (stl container write)
1170   _listInstances_map.erase(instanceName);
1171   _numInstanceMutex.unlock() ;
1172   component_i->destroy() ;
1173   _NS->Destroy_Name(instanceName.c_str());
1174 }
1175
1176 //=============================================================================
1177 //! Unload component libraries from the container
1178 /*!
1179 *  CORBA method: Discharges unused libraries from the container.
1180 */
1181 //=============================================================================
1182 void Engines_Container_i::finalize_removal()
1183 {
1184   MESSAGE("finalize unload : dlclose");
1185   _numInstanceMutex.lock(); // lock to be alone
1186   // (see decInstanceCnt, load_component_Library)
1187   std::map<std::string, void *>::iterator ith;
1188   for (ith = _toRemove_map.begin(); ith != _toRemove_map.end(); ith++)
1189   {
1190     void *handle = (*ith).second;
1191     std::string impl_name= (*ith).first;
1192     if (handle)
1193     {
1194       SCRUTE(handle);
1195       SCRUTE(impl_name);
1196       //        dlclose(handle);                // SALOME unstable after ...
1197       //        _library_map.erase(impl_name);
1198     }
1199   }
1200   _toRemove_map.clear();
1201   _numInstanceMutex.unlock();
1202 }
1203
1204 //=============================================================================
1205 //! Decrement component instance reference count
1206 /*!
1207 *
1208 */
1209 //=============================================================================
1210 void Engines_Container_i::decInstanceCnt(std::string genericRegisterName)
1211 {
1212   if(_cntInstances_map.count(genericRegisterName)==0)
1213     return;
1214   std::string aGenRegisterName =genericRegisterName;
1215   MESSAGE("Engines_Container_i::decInstanceCnt " << aGenRegisterName);
1216   ASSERT(_cntInstances_map[aGenRegisterName] > 0);
1217   _numInstanceMutex.lock(); // lock to be alone
1218   // (see finalize_removal, load_component_Library)
1219   _cntInstances_map[aGenRegisterName] -= 1;
1220   SCRUTE(_cntInstances_map[aGenRegisterName]);
1221   if (_cntInstances_map[aGenRegisterName] == 0)
1222   {
1223     std::string impl_name =
1224       Engines_Component_i::GetDynLibraryName(aGenRegisterName.c_str());
1225     SCRUTE(impl_name);
1226     void* handle = _library_map[impl_name];
1227     ASSERT(handle);
1228     _toRemove_map[impl_name] = handle;
1229   }
1230   _numInstanceMutex.unlock();
1231 }
1232
1233 //=============================================================================
1234 //! Find or create a new component instance
1235 /*!
1236 *  CORBA method: find or create an instance of the component (servant),
1237 *  load a new component class (dynamic library) if required,
1238 *
1239 *  ---- FOR COMPATIBILITY WITH 2.2 ----
1240 *
1241 *  ---- USE ONLY FOR MULTISTUDY INSTANCES ! --------
1242 *
1243 *  The servant registers itself to naming service and Registry.
1244 *  \param genericRegisterName  Name of the component to register
1245 *                              in Registry & Name Service
1246 *  \param componentName       Name of the constructed library of the component
1247 *  \return a loaded component
1248 */
1249 //=============================================================================
1250
1251 Engines::EngineComponent_ptr
1252 Engines_Container_i::load_impl( const char* genericRegisterName,
1253                                 const char* componentName )
1254 {
1255   char* reason;
1256   std::string impl_name = std::string(LIB) + genericRegisterName + ENGINESO;
1257   Engines::EngineComponent_var iobject = Engines::EngineComponent::_nil() ;
1258   if (load_component_Library(genericRegisterName,reason))
1259     iobject = find_or_create_instance(genericRegisterName, impl_name);
1260   CORBA::string_free(reason);
1261   return iobject._retn();
1262 }
1263
1264 //=============================================================================
1265 //! Finds an already existing component instance or create a new instance
1266 /*!
1267 *  C++ method: Finds an already existing servant instance of a component, or
1268 *              create an instance.
1269 *  ---- USE ONLY FOR MULTISTUDY INSTANCES ! --------
1270 *  \param genericRegisterName    Name of the component instance to register
1271 *                                in Registry & Name Service,
1272 *                                (without _inst_n suffix, like "COMPONENT")
1273 *  \param componentLibraryName   like "libCOMPONENTEngine.so"
1274 *  \return a loaded component
1275 *
1276 *  example with names:
1277 *    - aGenRegisterName = COMPONENT (= first argument)
1278 *    - impl_name = libCOMPONENTEngine.so (= second argument)
1279 *    - _containerName = /Containers/cli76ce/FactoryServer
1280 *    - factoryName = COMPONENTEngine_factory
1281 *    - component_registerBase = /Containers/cli76ce/FactoryServer/COMPONENT
1282 *    - instanceName = COMPONENT_inst_1
1283 *    - component_registerName = /Containers/cli76ce/FactoryServer/COMPONENT_inst_1
1284 */
1285 //=============================================================================
1286
1287 Engines::EngineComponent_ptr
1288 Engines_Container_i::find_or_create_instance(std::string genericRegisterName,
1289                                              std::string componentLibraryName)
1290 {
1291   std::string aGenRegisterName = genericRegisterName;
1292   std::string impl_name = componentLibraryName;
1293   if (_library_map.count(impl_name) == 0)
1294   {
1295     INFOS("shared library " << impl_name <<" must be loaded before creating instance");
1296     return Engines::EngineComponent::_nil() ;
1297   }
1298   else
1299   {
1300     // --- find a registered instance in naming service, or create
1301
1302     void* handle = _library_map[impl_name];
1303     std::string component_registerBase =
1304       _containerName + "/" + aGenRegisterName;
1305     Engines::EngineComponent_var iobject = Engines::EngineComponent::_nil() ;
1306     std::string reason;
1307     try
1308     {
1309       CORBA::Object_var obj =
1310         _NS->ResolveFirst( component_registerBase.c_str());
1311       if ( CORBA::is_nil( obj ) )
1312       {
1313         iobject = createInstance(genericRegisterName,
1314                                  handle,
1315                                  0,
1316                                  reason); // force multiStudy instance here !
1317       }
1318       else
1319       {
1320         iobject = Engines::EngineComponent::_narrow( obj ) ;
1321         Engines_Component_i *servant =
1322           dynamic_cast<Engines_Component_i*>
1323           (_poa->reference_to_servant(iobject));
1324         ASSERT(servant)
1325           int studyId = servant->getStudyId();
1326         ASSERT (studyId >= 0);
1327         if (studyId == 0) // multiStudy instance, OK
1328         {
1329           // No ReBind !
1330           MESSAGE(component_registerBase.c_str()<<" already bound");
1331         }
1332         else // monoStudy instance: NOK
1333         {
1334           iobject = Engines::EngineComponent::_nil();
1335           INFOS("load_impl & find_component_instance methods "
1336             << "NOT SUITABLE for mono study components");
1337         }
1338       }
1339     }
1340     catch (...)
1341     {
1342       INFOS( "Container_i::load_impl catched" ) ;
1343     }
1344     return iobject._retn();
1345   }
1346 }
1347
1348 //=============================================================================
1349 //! Indicate if container is a python one
1350 /*! 
1351 *  Retrieves only with container naming convention if it is a python container
1352 */
1353 //=============================================================================
1354 bool Engines_Container_i::isPythonContainer(const char* ContainerName)
1355 {
1356   bool ret=false;
1357   int len=strlen(ContainerName);
1358   if(len>=2)
1359     if(strcmp(ContainerName+len-2,"Py")==0)
1360       ret=true;
1361   return ret;
1362 }
1363
1364 //=============================================================================
1365 //! Kill the container
1366 /*!
1367 *  CORBA method: Kill the container process with exit(0).
1368 *  To remove :  never returns !
1369 */
1370 //=============================================================================
1371 bool Engines_Container_i::Kill_impl()
1372 {
1373   MESSAGE("Engines_Container_i::Kill() pid "<< getpid() << " containerName "
1374     << _containerName.c_str() << " machineName "
1375     << Kernel_Utils::GetHostname().c_str());
1376   INFOS("===============================================================");
1377   INFOS("= REMOVE calls to Kill_impl in C++ container                  =");
1378   INFOS("===============================================================");
1379   //_exit( 0 ) ;
1380   ASSERT(0);
1381   return false;
1382 }
1383
1384 //=============================================================================
1385 /*! 
1386 *  
1387 */
1388 //=============================================================================
1389 void ActSigIntHandler()
1390 {
1391 #ifndef WIN32
1392   struct sigaction SigIntAct ;
1393   SigIntAct.sa_sigaction = &SigIntHandler ;
1394   sigemptyset(&SigIntAct.sa_mask);
1395   SigIntAct.sa_flags = SA_SIGINFO ;
1396 #endif
1397
1398   // DEBUG 03.02.2005 : the first parameter of sigaction is not a mask of signals
1399   // (SIGINT | SIGUSR1) :
1400   // it must be only one signal ===> one call for SIGINT 
1401   // and an other one for SIGUSR1
1402
1403 #ifndef WIN32
1404   if ( sigaction( SIGINT , &SigIntAct, NULL ) ) 
1405   {
1406     perror("SALOME_Container main ") ;
1407     exit(0) ;
1408   }
1409   if ( sigaction( SIGUSR1 , &SigIntAct, NULL ) )
1410   {
1411     perror("SALOME_Container main ") ;
1412     exit(0) ;
1413   }
1414   if ( sigaction( SIGUSR2 , &SigIntAct, NULL ) )
1415   {
1416     perror("SALOME_Container main ") ;
1417     exit(0) ;
1418   }
1419
1420   //PAL9042 JR : during the execution of a Signal Handler (and of methods called through Signal Handlers)
1421   //             use of streams (and so on) should never be used because :
1422   //             streams of C++ are naturally thread-safe and use pthread_mutex_lock ===>
1423   //             A stream operation may be interrupted by a signal and if the Handler use stream we
1424   //             may have a "Dead-Lock" ===HangUp
1425   //==INFOS is commented
1426   //  INFOS(pthread_self() << "SigIntHandler activated") ;
1427
1428 #else  
1429   signal( SIGINT, SigIntHandler );
1430 // legacy code required to supervisor. Commented in order to avoid problems on Windows 
1431 //  signal( SIGUSR1, SigIntHandler );
1432 #endif
1433
1434 }
1435
1436 void SetCpuUsed() ;
1437 void CallCancelThread() ;
1438
1439 #ifndef WIN32
1440 void SigIntHandler(int what ,
1441                    siginfo_t * siginfo ,
1442                    void * toto ) 
1443 {
1444   //PAL9042 JR : during the execution of a Signal Handler (and of methods called through Signal Handlers)
1445   //             use of streams (and so on) should never be used because :
1446   //             streams of C++ are naturally thread-safe and use pthread_mutex_lock ===>
1447   //             A stream operation may be interrupted by a signal and if the Handler use stream we
1448   //             may have a "Dead-Lock" ===HangUp
1449   //==MESSAGE is commented
1450   //  MESSAGE(pthread_self() << "SigIntHandler what     " << what << std::endl
1451   //          << "              si_signo " << siginfo->si_signo << std::endl
1452   //          << "              si_code  " << siginfo->si_code << std::endl
1453   //          << "              si_pid   " << siginfo->si_pid) ;
1454
1455   if ( _Sleeping )
1456   {
1457     _Sleeping = false ;
1458     //     MESSAGE("SigIntHandler END sleeping.") ;
1459     return ;
1460   }
1461   else
1462   {
1463     ActSigIntHandler() ;
1464     if ( siginfo->si_signo == SIGUSR1 )
1465     {
1466       SetCpuUsed() ;
1467     }
1468     else if ( siginfo->si_signo == SIGUSR2 )
1469     {
1470       CallCancelThread() ;
1471     }
1472     else 
1473     {
1474       _Sleeping = true ;
1475       //      MESSAGE("SigIntHandler BEGIN sleeping.") ;
1476       int count = 0 ;
1477       while( _Sleeping )
1478       {
1479         sleep( 1 ) ;
1480         count += 1 ;
1481       }
1482       //      MESSAGE("SigIntHandler LEAVE sleeping after " << count << " s.") ;
1483     }
1484     return ;
1485   }
1486 }
1487 #else // Case WIN32
1488 void SigIntHandler( int what )
1489 {
1490 #ifndef WIN32
1491   MESSAGE( pthread_self() << "SigIntHandler what     " << what << std::endl );
1492 #else
1493   MESSAGE( "SigIntHandler what     " << what << std::endl );
1494 #endif
1495   if ( _Sleeping )
1496   {
1497     _Sleeping = false ;
1498     MESSAGE("SigIntHandler END sleeping.") ;
1499     return ;
1500   }
1501   else
1502   {
1503     ActSigIntHandler() ;
1504     if ( what == SIGUSR1 )
1505     {
1506       SetCpuUsed() ;
1507     }
1508     else
1509     {
1510       _Sleeping = true ;
1511       MESSAGE("SigIntHandler BEGIN sleeping.") ;
1512       int count = 0 ;
1513       while( _Sleeping ) 
1514       {
1515         Sleep( 1000 ) ;
1516         count += 1 ;
1517       }
1518       MESSAGE("SigIntHandler LEAVE sleeping after " << count << " s.") ;
1519     }
1520     return ;
1521   }
1522 }
1523 #endif
1524
1525 //=============================================================================
1526 //! Get or create a file reference object associated to a local file (to transfer it)
1527 /*!
1528 *  CORBA method: get or create a fileRef object associated to a local file
1529 *  (a file on the computer on which runs the container server), which stores
1530 *  a list of (machine, localFileName) corresponding to copies already done.
1531 *
1532 *  \param  origFileName absolute path for a local file to copy on other
1533 *          computers
1534 *  \return a fileRef object associated to the file.
1535 */
1536 //=============================================================================
1537 Engines::fileRef_ptr
1538 Engines_Container_i::createFileRef(const char* origFileName)
1539 {
1540   std::string origName(origFileName);
1541   Engines::fileRef_var theFileRef = Engines::fileRef::_nil();
1542
1543   if (origName[0] != '/')
1544   {
1545     INFOS("path of file to copy must be an absolute path begining with '/'");
1546     return Engines::fileRef::_nil();
1547   }
1548
1549   if (CORBA::is_nil(_fileRef_map[origName]))
1550   {
1551     CORBA::Object_var obj=_poa->id_to_reference(*_id);
1552     Engines::Container_var pCont = Engines::Container::_narrow(obj);
1553     fileRef_i* aFileRef = new fileRef_i(pCont, origFileName);
1554     theFileRef = Engines::fileRef::_narrow(aFileRef->_this());
1555     _numInstanceMutex.lock() ; // lock to be alone (stl container write)
1556     _fileRef_map[origName] = theFileRef;
1557     _numInstanceMutex.unlock() ;
1558   }
1559
1560   theFileRef =  Engines::fileRef::_duplicate(_fileRef_map[origName]);
1561   ASSERT(! CORBA::is_nil(theFileRef));
1562   return theFileRef._retn();
1563 }
1564
1565 //=============================================================================
1566 //! Get a fileTransfer reference
1567 /*!
1568 *  CORBA method:
1569 *  \return a reference to the fileTransfer object
1570 */
1571 //=============================================================================
1572 Engines::fileTransfer_ptr
1573 Engines_Container_i::getFileTransfer()
1574 {
1575   Engines::fileTransfer_var aFileTransfer
1576     = Engines::fileTransfer::_duplicate(_fileTransfer);
1577   return aFileTransfer._retn();
1578 }
1579
1580 //=============================================================================
1581 //! Create a Salome file
1582 //=============================================================================
1583 Engines::Salome_file_ptr
1584 Engines_Container_i::createSalome_file(const char* origFileName)
1585 {
1586   std::string origName(origFileName);
1587   if (CORBA::is_nil(_Salome_file_map[origName]))
1588   {
1589     Salome_file_i* aSalome_file = new Salome_file_i();
1590     aSalome_file->setContainer(Engines::Container::_duplicate(this->_this()));
1591     try
1592     {
1593       aSalome_file->setLocalFile(origFileName);
1594       aSalome_file->recvFiles();
1595     }
1596     catch (const SALOME::SALOME_Exception& e)
1597     {
1598       return Engines::Salome_file::_nil();
1599     }
1600
1601     Engines::Salome_file_var theSalome_file = Engines::Salome_file::_nil();
1602     theSalome_file = Engines::Salome_file::_narrow(aSalome_file->_this());
1603     _numInstanceMutex.lock() ; // lock to be alone (stl container write)
1604     _Salome_file_map[origName] = theSalome_file;
1605     _numInstanceMutex.unlock() ;
1606   }
1607
1608   Engines::Salome_file_ptr theSalome_file =
1609     Engines::Salome_file::_duplicate(_Salome_file_map[origName]);
1610   ASSERT(!CORBA::is_nil(theSalome_file));
1611   return theSalome_file;
1612 }
1613
1614 //=============================================================================
1615 /*! \brief copy a file from a remote host (container) to the local host
1616  * \param container the remote container
1617  * \param remoteFile the file to copy locally from the remote host into localFile
1618  * \param localFile the local file
1619  */
1620 //=============================================================================
1621 void Engines_Container_i::copyFile(Engines::Container_ptr container, const char* remoteFile, const char* localFile)
1622 {
1623   Engines::fileTransfer_var fileTransfer = container->getFileTransfer();
1624
1625   FILE* fp;
1626   if ((fp = fopen(localFile,"wb")) == NULL)
1627     {
1628       INFOS("file " << localFile << " cannot be open for writing");
1629       return;
1630     }
1631
1632   CORBA::Long fileId = fileTransfer->open(remoteFile);
1633   if (fileId > 0)
1634     {
1635       Engines::fileBlock* aBlock;
1636       int toFollow = 1;
1637       int ctr=0;
1638       while (toFollow)
1639         {
1640           ctr++;
1641           //SCRUTE(ctr);
1642           aBlock = fileTransfer->getBlock(fileId);
1643           toFollow = aBlock->length();
1644           //SCRUTE(toFollow);
1645           CORBA::Octet *buf = aBlock->get_buffer();
1646           fwrite(buf, sizeof(CORBA::Octet), toFollow, fp);
1647           delete aBlock;
1648         }
1649       fclose(fp);
1650       MESSAGE("end of transfer");
1651       fileTransfer->close(fileId);
1652     }
1653   else
1654     {
1655       INFOS("open reference file for copy impossible");
1656     }
1657 }
1658
1659 //=============================================================================
1660 /*! \brief create a PyNode object to execute remote python code
1661  * \param nodeName the name of the node
1662  * \param code the python code to load
1663  * \return the PyNode
1664  */
1665 //=============================================================================
1666 Engines::PyNode_ptr Engines_Container_i::createPyNode(const char* nodeName, const char* code)
1667 {
1668     Engines::PyNode_var node= Engines::PyNode::_nil();
1669
1670     PyGILState_STATE gstate = PyGILState_Ensure();
1671     PyObject *res = PyObject_CallMethod(_pyCont,
1672       (char*)"create_pynode",
1673       (char*)"ss",
1674       nodeName,
1675       code);
1676     if(res==NULL)
1677       {
1678         //internal error
1679         PyErr_Print();
1680         PyGILState_Release(gstate);
1681         SALOME::ExceptionStruct es;
1682         es.type = SALOME::INTERNAL_ERROR;
1683         es.text = "can not create a python node";
1684         throw SALOME::SALOME_Exception(es);
1685       }
1686     long ierr=PyInt_AsLong(PyTuple_GetItem(res,0));
1687     PyObject* result=PyTuple_GetItem(res,1);
1688     std::string astr=PyString_AsString(result);
1689     Py_DECREF(res);
1690     PyGILState_Release(gstate);
1691     if(ierr==0)
1692       {
1693         Utils_Locker lck(&_mutexForDftPy);
1694         CORBA::Object_var obj=_orb->string_to_object(astr.c_str());
1695         node=Engines::PyNode::_narrow(obj);
1696         std::map<std::string,Engines::PyNode_var>::iterator it(_dftPyNode.find(nodeName));
1697         if(it==_dftPyNode.end())
1698           {
1699             _dftPyNode[nodeName]=node;
1700           }
1701         else
1702           {
1703             Engines::PyNode_var oldNode((*it).second);
1704             if(!CORBA::is_nil(oldNode))
1705               oldNode->UnRegister();
1706             (*it).second=node;
1707           }
1708         if(!CORBA::is_nil(node))
1709           node->Register();
1710         return node._retn();
1711       }
1712     else
1713       {
1714         SALOME::ExceptionStruct es;
1715         es.type = SALOME::INTERNAL_ERROR;
1716         es.text = astr.c_str();
1717         throw SALOME::SALOME_Exception(es);
1718       }
1719 }
1720
1721 //=============================================================================
1722 /*! \brief Retrieves the last created PyNode instance with createPyNode.
1723  *
1724  */
1725 //=============================================================================
1726 Engines::PyNode_ptr  Engines_Container_i::getDefaultPyNode(const char *nodeName)
1727 {
1728   Utils_Locker lck(&_mutexForDftPy);
1729   std::map<std::string,Engines::PyNode_var>::iterator it(_dftPyNode.find(nodeName));
1730   if(it==_dftPyNode.end())
1731     return Engines::PyNode::_nil();
1732   else
1733     {
1734       Engines::PyNode_var tmpVar((*it).second);
1735       if(!CORBA::is_nil(tmpVar))
1736         return Engines::PyNode::_duplicate(tmpVar);
1737       else
1738         return Engines::PyNode::_nil();
1739     }
1740 }
1741
1742 //=============================================================================
1743 /*! \brief create a PyScriptNode object to execute remote python code
1744  * \param nodeName the name of the node
1745  * \param code the python code to load
1746  * \return the PyScriptNode
1747  */
1748 //=============================================================================
1749 Engines::PyScriptNode_ptr Engines_Container_i::createPyScriptNode(const char* nodeName, const char* code)
1750 {
1751     Engines::PyScriptNode_var node= Engines::PyScriptNode::_nil();
1752
1753     PyGILState_STATE gstate = PyGILState_Ensure();
1754     PyObject *res = PyObject_CallMethod(_pyCont,
1755       (char*)"create_pyscriptnode",
1756       (char*)"ss",
1757       nodeName,
1758       code);
1759     if(res==NULL)
1760       {
1761         //internal error
1762         PyErr_Print();
1763         PyGILState_Release(gstate);
1764         SALOME::ExceptionStruct es;
1765         es.type = SALOME::INTERNAL_ERROR;
1766         es.text = "can not create a python node";
1767         throw SALOME::SALOME_Exception(es);
1768       }
1769     long ierr=PyInt_AsLong(PyTuple_GetItem(res,0));
1770     PyObject* result=PyTuple_GetItem(res,1);
1771     std::string astr=PyString_AsString(result);
1772     Py_DECREF(res);
1773     PyGILState_Release(gstate);
1774
1775     if(ierr==0)
1776       {
1777         Utils_Locker lck(&_mutexForDftPy);
1778         CORBA::Object_var obj=_orb->string_to_object(astr.c_str());
1779         node=Engines::PyScriptNode::_narrow(obj);
1780         std::map<std::string,Engines::PyScriptNode_var>::iterator it(_dftPyScriptNode.find(nodeName));
1781         if(it==_dftPyScriptNode.end())
1782           {
1783             _dftPyScriptNode[nodeName]=node;
1784           }
1785         else
1786           {
1787             Engines::PyScriptNode_var oldNode((*it).second);
1788             if(!CORBA::is_nil(oldNode))
1789               oldNode->UnRegister();
1790             (*it).second=node;
1791           }
1792         if(!CORBA::is_nil(node))
1793           node->Register();
1794         return node._retn();
1795       }
1796     else
1797       {
1798         SALOME::ExceptionStruct es;
1799         es.type = SALOME::INTERNAL_ERROR;
1800         es.text = astr.c_str();
1801         throw SALOME::SALOME_Exception(es);
1802       }
1803 }
1804
1805 //=============================================================================
1806 /*! \brief Retrieves the last created PyScriptNode instance with createPyScriptNode.
1807  *
1808  */
1809 //=============================================================================
1810 Engines::PyScriptNode_ptr Engines_Container_i::getDefaultPyScriptNode(const char *nodeName)
1811 {
1812   Utils_Locker lck(&_mutexForDftPy);
1813   std::map<std::string,Engines::PyScriptNode_var>::iterator it(_dftPyScriptNode.find(nodeName));
1814   if(it==_dftPyScriptNode.end())
1815     return Engines::PyScriptNode::_nil();
1816   else
1817     {
1818       Engines::PyScriptNode_var tmpVar((*it).second);
1819       if(!CORBA::is_nil(tmpVar))
1820         return Engines::PyScriptNode::_duplicate(tmpVar);
1821       else
1822         return Engines::PyScriptNode::_nil();
1823     }
1824 }
1825
1826 //=============================================================================
1827 /* int checkifexecutable(const char *filename)
1828 *
1829 * Return non-zero if the name is an executable file, and
1830 * zero if it is not executable, or if it does not exist.
1831 */
1832 //=============================================================================
1833 int checkifexecutable(const std::string& filename)
1834 {
1835   int result;
1836   struct stat statinfo;
1837
1838   result = stat(filename.c_str(), &statinfo);
1839   if (result < 0) return 0;
1840   if (!S_ISREG(statinfo.st_mode)) return 0;
1841
1842 #ifdef WIN32
1843   return 1;
1844 #else
1845   if (statinfo.st_uid == geteuid()) return statinfo.st_mode & S_IXUSR;
1846   if (statinfo.st_gid == getegid()) return statinfo.st_mode & S_IXGRP;
1847   return statinfo.st_mode & S_IXOTH;
1848 #endif
1849 }
1850
1851
1852 //=============================================================================
1853 /*! \brief Find a file by searching in a path
1854  *  \param filename file name to search
1855  *  \param path path to search in
1856  *  \param pth the complete file path if found
1857  *  \return 1 if found 0 if not 
1858 */
1859 //=============================================================================
1860 int findpathof(const std::string& path, std::string& pth, const std::string& filename)
1861 {
1862   if ( path.size() == 0 ) return 0;
1863
1864   std::string::size_type offset = 0;
1865   std::string::size_type pos = 0;
1866   int found = 0;
1867   struct stat statinfo;
1868
1869   while(!found)
1870   {
1871     pos = path.find( SEP, offset );
1872     pth = path.substr( offset, pos - offset );
1873     if ( pth.size() > 0 )
1874     {
1875       if( pth[pth.size()-1] != SLASH ) pth += SLASH;
1876       pth += filename;
1877       int result=stat(pth.c_str(), &statinfo);
1878       if(result == 0) found=1;
1879     }
1880     if (pos == std::string::npos) break;
1881     offset = pos+1;
1882   }
1883   return found;
1884 }
1885
1886 void Engines_Container_i::registerTemporaryFile( const std::string& fileName )
1887 {
1888   _tmp_files.remove( fileName );
1889   _tmp_files.push_back( fileName );
1890 }
1891
1892 void Engines_Container_i::unregisterTemporaryFile( const std::string& fileName )
1893 {
1894   _tmp_files.remove( fileName );
1895 }
1896
1897 void Engines_Container_i::clearTemporaryFiles()
1898 {
1899   std::list<std::string>::const_iterator it;
1900   for ( it = _tmp_files.begin(); it != _tmp_files.end(); ++it ) {
1901 #ifdef WIN32
1902     std::string command = "del /F /P";
1903 #else
1904     std::string command = "rm -rf ";
1905 #endif
1906     command += *it;
1907     system( command.c_str() );
1908   }
1909   _tmp_files.clear();
1910 }