1 #! /usr/bin/env python3
2 # -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2007-2020 CEA/DEN, EDF R&D, OPEN CASCADE
5 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
6 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
8 # This library is free software; you can redistribute it and/or
9 # modify it under the terms of the GNU Lesser General Public
10 # License as published by the Free Software Foundation; either
11 # version 2.1 of the License, or (at your option) any later version.
13 # This library is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 # Lesser General Public License for more details.
18 # You should have received a copy of the GNU Lesser General Public
19 # License along with this library; if not, write to the Free Software
20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
25 ## \file killSalomeWithPort.py
26 # Stop all %SALOME servers from given sessions by killing them
28 # The sessions are indicated by their ports on the command line as in :
30 # killSalomeWithPort.py 2811 2815
34 import re, os, sys, pickle, signal, glob
37 from salome_utils import verbose
40 def getPiDict(port,appname='salome',full=True,hidden=True,hostname=None):
42 Get file with list of SALOME processes.
43 This file is located in the user's home directory
44 and named .<user>_<host>_<port>_SALOME_pidict
52 - appname : application name (default is 'SALOME')
53 - full : if True, full path to the file is returned, otherwise only file name is returned
54 - hidden : if True, file name is prefixed with . (dot) symbol; this internal parameter is used
55 to support compatibility with older versions of SALOME
57 # bug fix: ensure port is an integer
58 # Note: this function is also called with port='#####' !!!
64 from salome_utils import generateFileName, getLogDir
67 hostname = os.getenv("NSHOST")
68 if hostname: hostname = hostname.split(".")[0]
71 # full path to the pidict file is requested
73 # new-style dot-prefixed pidict files
74 # are in the system-dependant temporary directory
77 # old-style non-dot-prefixed pidict files
78 # are in the user's home directory
79 dir = os.path.expanduser("~")
83 return generateFileName(dir,
87 with_hostname=hostname or True,
89 with_app=appname.upper())
91 def appliCleanOmniOrbConfig(port):
93 Remove omniorb config files related to the port in SALOME application:
94 - ${OMNIORB_USER_PATH}/.omniORB_${USER}_${HOSTNAME}_${NSPORT}.cfg
95 - ${OMNIORB_USER_PATH}/.omniORB_${USER}_last.cfg
96 the last is removed only if the link points to the first file.
99 print("clean OmniOrb config for port %s"%port)
101 from salome_utils import generateFileName, getUserName
102 omniorbUserPath = os.getenv("OMNIORB_USER_PATH")
103 if omniorbUserPath is None:
104 #Run outside application context
107 omniorb_config = generateFileName(omniorbUserPath, prefix="omniORB",
113 last_running_config = generateFileName(omniorbUserPath, prefix="omniORB",
118 if os.access(last_running_config,os.F_OK):
119 if not sys.platform == 'win32':
120 pointedPath = os.readlink(last_running_config)
121 if pointedPath[0] != '/':
122 pointedPath=os.path.join(os.path.dirname(last_running_config), pointedPath)
124 if pointedPath == omniorb_config:
125 os.unlink(last_running_config)
129 os.remove(last_running_config)
133 if os.access(omniorb_config,os.F_OK):
134 os.remove(omniorb_config)
137 if os.path.lexists(last_running_config):return
139 #try to relink last.cfg to an existing config file if any
140 files = glob.glob(os.path.join(omniorbUserPath,".omniORB_"+getUserName()+"_*.cfg"))
145 if stat.st_atime > current:
146 current=stat.st_atime
149 if sys.platform == "win32":
151 shutil.copyfile(os.path.normpath(current_config), last_running_config)
154 os.symlink(os.path.normpath(current_config), last_running_config)
160 ########## kills all salome processes with the given port ##########
162 def shutdownMyPort(port, cleanup=True):
164 Shutdown SALOME session running on the specified port.
169 # bug fix: ensure port is an integer
173 from PortManager import releasePort
178 from salome_utils import generateFileName
180 # set OMNIORB_CONFIG variable to the proper file
181 omniorbUserPath = os.getenv("OMNIORB_USER_PATH")
183 if omniorbUserPath is not None:
184 kwargs["with_username"]=True
186 omniorbUserPath = os.path.realpath(os.path.expanduser('~'))
187 omniorb_config = generateFileName(omniorbUserPath, prefix="omniORB",
193 os.environ['OMNIORB_CONFIG'] = omniorb_config
194 os.environ['NSPORT'] = str(port)
196 # give the chance to the servers to shutdown properly
199 from omniORB import CORBA
201 from LifeCycleCORBA import LifeCycleCORBA
203 orb = CORBA.ORB_init([''], CORBA.ORB_ID)
204 lcc = LifeCycleCORBA(orb) # see (1)
205 print("Terminating SALOME on port %s..."%(port))
206 lcc.shutdownServers()
207 # give some time to shutdown to complete
211 from salome_utils import killOmniNames
213 filedict=getPiDict(port)
214 __killMyPort(port, filedict)
215 from PortManager import releasePort
222 sys.exit(0) # see (1)
224 # (1) If --shutdown-servers option is set to 1, session close procedure is
225 # called twice: first explicitly by salome command, second by automatic
226 # atexit to handle Ctrl-C. During second call, LCC does not exist anymore and
227 # a RuntimeError is raised; we explicitly exit this function with code 0 to
228 # prevent parent thread from crashing.
230 def __killMyPort(port, filedict):
231 # bug fix: ensure port is an integer
236 with open(filedict, 'rb') as fpid:
237 process_ids=pickle.load(fpid)
238 for process_id in process_ids:
239 for pid, cmd in list(process_id.items()):
241 if verbose(): print("stop process %s : %s"% (pid, cmd[0]))
243 from salome_utils import killpid
246 if verbose(): print(" ------------------ process %s : %s not found"% (pid, cmd[0]))
247 if pid in checkUnkilledProcess():
254 pass # for process_id ...
257 print("Cannot find or open SALOME PIDs file for port", port)
263 def __guessPiDictFilename(port):
264 from salome_utils import getShortHostName, getHostName
266 # new-style dot-prefixed pidict file
267 getPiDict(port, hidden=True),
268 # provide compatibility with old-style pidict file (not dot-prefixed)
269 getPiDict(port, hidden=False),
270 # provide compatibility with old-style pidict file (short hostname)
271 getPiDict(port, hidden=True, hostname=getShortHostName()),
272 # provide compatibility with old-style pidict file (not dot-prefixed, short hostname
273 getPiDict(port, hidden=False, hostname=getShortHostName()),
274 # provide compatibility with old-style pidict file (long hostname)
275 getPiDict(port, hidden=True, hostname=getHostName()),
276 # provide compatibility with old-style pidict file (not dot-prefixed, long hostname)
277 getPiDict(port, hidden=False, hostname=getHostName())
281 for filedict in filedicts:
282 log_msg += "Trying %s..."%filedict
283 if os.path.exists(filedict):
284 log_msg += " ... OK\n"
287 log_msg += " ... not found\n"
295 def killMyPort(port):
297 Kill SALOME session running on the specified port.
301 # bug fix: ensure port is an integer
306 import PortManager # do not remove! Test for PortManager availability!
307 filedict = getPiDict(port)
308 if not os.path.isfile(filedict): # removed by previous call, see (1)
310 print("SALOME on port %s: already removed by previous call"%port)
311 # Remove port from PortManager config file
313 from PortManager import releasePort
315 print("Removing port from PortManager configuration file")
323 # try to shutdown session normally
324 import threading, time
325 threading.Thread(target=shutdownMyPort, args=(port,True)).start()
326 time.sleep(3) # wait a little, then kill processes (should be done if shutdown procedure hangs up)
329 import PortManager # do not remove! Test for PortManager availability!
330 filedict = getPiDict(port)
331 #filedict = __guessPiDictFilename(port)
333 all_files = glob.glob("%s*"%filedict)
335 __killMyPort(port, f)
337 filedict = __guessPiDictFilename(port)
338 __killMyPort(port, filedict)
341 appliCleanOmniOrbConfig(port)
344 def cleanApplication(port):
346 Clean application running on the specified port.
350 # bug fix: ensure port is an integer
355 filedict=getPiDict(port)
359 #traceback.print_exc()
362 appliCleanOmniOrbConfig(port)
365 def killMyPortSpy(pid, port):
368 from salome_utils import killpid
369 ret = killpid(pid, 0)
374 from time import sleep
377 filedict = getPiDict(port, hidden=True)
378 if not os.path.exists(filedict):
382 orb = omniORB.CORBA.ORB_init(sys.argv, omniORB.CORBA.ORB_ID)
383 import SALOME_NamingServicePy
384 ns = SALOME_NamingServicePy.SALOME_NamingServicePy_i(orb)
385 import SALOME #@UnresolvedImport @UnusedImport
386 session = ns.Resolve("/Kernel/Session")
391 status = session.GetStatSession()
393 # -- session is in naming service but has crash
397 if not status.activeGUI:
403 def checkUnkilledProcess():
404 #check processes in system after kill
405 from salome_utils import getUserName
409 if sys.platform != 'win32':
411 def _getDictfromOutput(_output, _dic, _cond = None):
412 _pids = dict(zip(list(map(int, _output[::2])), _output[1::2]))
414 _pids = {pid:cmd for pid,cmd in _pids.items() if re.match(_cond, cmd)}
417 # 1. SALOME servers plus omniNames
418 cmd = 'ps --noheading -U {user} -o pid,cmd | awk \'{{printf("%s %s\\n", $1, $2)}}\''.format(user=user)
419 _getDictfromOutput(subprocess.getoutput(cmd).split(), processes, '^(SALOME_|omniNames)')
421 cmd = 'ps -fea | grep \'{user}\' | grep \'ghs3d\' | grep \'f /tmp/GHS3D_\' | grep -v \'grep\' | awk \'{{print("%s %s\\n", $2, $8)}}\''.format(user=user)
422 _getDictfromOutput(subprocess.getoutput(cmd).split(), processes)
424 cmd = 'ps -fea | grep \'{user}\' | grep \'ompi-server\' | grep -v \'grep\' | awk \'{{print("%s %s\\n", $2, $8)}}\''.format(user=user)
425 _getDictfromOutput(subprocess.getoutput(cmd).split(), processes)
427 cmd = 'tasklist /fo csv | findstr /i "SALOME_ omniNames"'
428 prc = subprocess.getoutput(cmd)
431 prc = [prc[i].split(',') for i in range(0, len(prc)) if i % 2 == 0]
432 prc = dict([(int(prc[j][1].replace('"', '')), prc[j][0].replace('"', '')) for j in range(0, len(prc))])
433 processes.update(prc)
438 if __name__ == "__main__":
439 if len(sys.argv) < 2:
441 print(" %s <port>" % os.path.basename(sys.argv[0]))
443 print("Kills SALOME session running on specified <port>.")
446 if sys.argv[1] == "--spy":
447 if len(sys.argv) > 3:
450 killMyPortSpy(pid, port)
454 elif sys.argv[1] == "--find":
455 processes = checkUnkilledProcess()
457 print("Unkilled processes: ")
458 print(" --------------------")
459 print(" PID : Process name")
460 print(" --------------------")
461 for pair in processes.items():
462 print('%6d : %s' % pair)
465 from salomeContextUtils import setOmniOrbUserPath #@UnresolvedImport
467 except Exception as e:
470 for port in sys.argv[1:]: