X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=bin%2FkillSalomeWithPort.py;h=fec9e4aa356850f7a1794f4bc39e881283a9febf;hb=ee44cdc8da140e8f0aabaa1b9d9485939899cfdf;hp=23b70628f6b19e33517830e88010c1175b44af46;hpb=e4ec3029fd75194c73491ea6893929cb146d6c02;p=modules%2Fkernel.git diff --git a/bin/killSalomeWithPort.py b/bin/killSalomeWithPort.py index 23b70628f..fec9e4aa3 100755 --- a/bin/killSalomeWithPort.py +++ b/bin/killSalomeWithPort.py @@ -1,6 +1,6 @@ -#! /usr/bin/env python +#! /usr/bin/env python3 # -*- coding: iso-8859-1 -*- -# Copyright (C) 2007-2013 CEA/DEN, EDF R&D, OPEN CASCADE +# Copyright (C) 2007-2017 CEA/DEN, EDF R&D, OPEN CASCADE # # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN, # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS @@ -8,7 +8,7 @@ # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either -# version 2.1 of the License. +# version 2.1 of the License, or (at your option) any later version. # # This library is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -31,9 +31,11 @@ # \endcode # -import os, sys, pickle, signal, commands,glob +import os, sys, pickle, signal, subprocess,glob +import subprocess +import shlex from salome_utils import verbose -import salome_utils + def getPiDict(port,appname='salome',full=True,hidden=True,hostname=None): """ @@ -52,7 +54,14 @@ def getPiDict(port,appname='salome',full=True,hidden=True,hostname=None): - hidden : if True, file name is prefixed with . (dot) symbol; this internal parameter is used to support compatibility with older versions of SALOME """ - from salome_utils import generateFileName, getTmpDir + # bug fix: ensure port is an integer + # Note: this function is also called with port='#####' !!! + try: + port = int(port) + except: + pass + + from salome_utils import generateFileName, getLogDir dir = "" if not hostname: hostname = os.getenv("NSHOST") @@ -63,13 +72,14 @@ def getPiDict(port,appname='salome',full=True,hidden=True,hostname=None): if hidden: # new-style dot-prefixed pidict files # are in the system-dependant temporary diretory - dir = getTmpDir() + dir = getLogDir() else: # old-style non-dot-prefixed pidict files # are in the user's home directory dir = os.getenv("HOME") pass pass + return generateFileName(dir, suffix="pidict", hidden=hidden, @@ -85,6 +95,9 @@ def appliCleanOmniOrbConfig(port): - ${OMNIORB_USER_PATH}/.omniORB_${USER}_last.cfg the last is removed only if the link points to the first file. """ + if verbose(): + print("clean OmniOrb config for port %s"%port) + from salome_utils import generateFileName, getUserName omniorbUserPath = os.getenv("OMNIORB_USER_PATH") if omniorbUserPath is None: @@ -113,9 +126,9 @@ def appliCleanOmniOrbConfig(port): pass pass else: - os.remove(last_running_config) + os.remove(last_running_config) pass - pass + pass if os.access(omniorb_config,os.F_OK): os.remove(omniorb_config) @@ -153,6 +166,8 @@ def shutdownMyPort(port, cleanup=True): - port - port number """ if not port: return + # bug fix: ensure port is an integer + port = int(port) try: from PortManager import releasePort @@ -182,14 +197,16 @@ def shutdownMyPort(port, cleanup=True): try: import time from omniORB import CORBA + from LifeCycleCORBA import LifeCycleCORBA # shutdown all orb = CORBA.ORB_init([''], CORBA.ORB_ID) - lcc = LifeCycleCORBA(orb) + lcc = LifeCycleCORBA(orb) # see (1) + print("Terminating SALOME on port %s..."%(port)) lcc.shutdownServers() # give some time to shutdown to complete time.sleep(1) - # shutdown omniNames and notifd + # shutdown omniNames if cleanup: lcc.killOmniNames() time.sleep(1) @@ -197,7 +214,72 @@ def shutdownMyPort(port, cleanup=True): pass except: pass + sys.exit(0) # see (1) + pass +# (1) If --shutdown-servers option is set to 1, session close procedure is +# called twice: first explicitly by salome command, second by automatic +# atexit to handle Ctrl-C. During second call, LCC does not exist anymore and +# a RuntimeError is raised; we explicitly exit this function with code 0 to +# prevent parent thread from crashing. + +def __killMyPort(port, filedict): + # bug fix: ensure port is an integer + if port: + port = int(port) + + try: + with open(filedict, 'rb') as fpid: + process_ids=pickle.load(fpid) + for process_id in process_ids: + for pid, cmd in list(process_id.items()): + if verbose(): print("stop process %s : %s"% (pid, cmd[0])) + try: + from salome_utils import killpid + killpid(int(pid)) + except: + if verbose(): print(" ------------------ process %s : %s not found"% (pid, cmd[0])) + pass + pass # for pid ... + pass # for process_id ... + # end with + except: + print("Cannot find or open SALOME PIDs file for port", port) + pass + os.remove(filedict) pass +# + +def __guessPiDictFilename(port): + from salome_utils import getShortHostName, getHostName + filedicts = [ + # new-style dot-prefixed pidict file + getPiDict(port, hidden=True), + # provide compatibility with old-style pidict file (not dot-prefixed) + getPiDict(port, hidden=False), + # provide compatibility with old-style pidict file (short hostname) + getPiDict(port, hidden=True, hostname=getShortHostName()), + # provide compatibility with old-style pidict file (not dot-prefixed, short hostname + getPiDict(port, hidden=False, hostname=getShortHostName()), + # provide compatibility with old-style pidict file (long hostname) + getPiDict(port, hidden=True, hostname=getHostName()), + # provide compatibility with old-style pidict file (not dot-prefixed, long hostname) + getPiDict(port, hidden=False, hostname=getHostName()) + ] + + log_msg = "" + for filedict in filedicts: + log_msg += "Trying %s..."%filedict + if os.path.exists(filedict): + log_msg += " ... OK\n" + break + else: + log_msg += " ... not found\n" + + if verbose(): + print(log_msg) + + return filedict +# def killMyPort(port): """ @@ -205,153 +287,79 @@ def killMyPort(port): Parameters: - port - port number """ + # bug fix: ensure port is an integer + if port: + port = int(port) + try: - import PortManager - PortManager.releasePort(port) - except ImportError: + import PortManager # do not remove! Test for PortManager availability! + filedict = getPiDict(port) + if not os.path.isfile(filedict): # removed by previous call, see (1) + if verbose(): + print("SALOME on port %s: already removed by previous call"%port) + # Remove port from PortManager config file + try: + from PortManager import releasePort + if verbose(): + print("Removing port from PortManager configuration file") + releasePort(port) + except ImportError: + pass + return + except: pass - from salome_utils import getShortHostName, getHostName - - # try to shutdown session nomally + # try to shutdown session normally import threading, time - threading.Thread(target=shutdownMyPort, args=(port,False)).start() + threading.Thread(target=shutdownMyPort, args=(port,True)).start() time.sleep(3) # wait a little, then kill processes (should be done if shutdown procedure hangs up) - # new-style dot-prefixed pidict file - filedict = getPiDict(port, hidden=True) - # provide compatibility with old-style pidict file (not dot-prefixed) - if not os.path.exists(filedict): filedict = getPiDict(port, hidden=False) - # provide compatibility with old-style pidict file (short hostname) - if not os.path.exists(filedict): filedict = getPiDict(port, hidden=True, hostname=getShortHostName()) - # provide compatibility with old-style pidict file (not dot-prefixed, short hostname) - if not os.path.exists(filedict): filedict = getPiDict(port, hidden=False, hostname=getShortHostName()) - # provide compatibility with old-style pidict file (long hostname) - if not os.path.exists(filedict): filedict = getPiDict(port, hidden=True, hostname=getHostName()) - # provide compatibility with old-style pidict file (not dot-prefixed, long hostname) - if not os.path.exists(filedict): filedict = getPiDict(port, hidden=False, hostname=getHostName()) - # try: - fpid = open(filedict, 'r') - # - from salome_utils import generateFileName - if sys.platform == "win32": - username = os.getenv( "USERNAME" ) - else: - username = os.getenv('USER') - path = os.path.join('/tmp/logs', username) - fpidomniNames = generateFileName(path, - prefix="", - suffix="Pid_omniNames", - extension="log", - with_port=port) - if not sys.platform == 'win32': - cmd = 'pid=`ps -eo pid,command | egrep "[0-9] omniNames -start %s"` ; echo $pid > %s' % ( str(port), fpidomniNames ) - a = os.system(cmd) - pass - try: - fpidomniNamesFile = open(fpidomniNames) - lines = fpidomniNamesFile.readlines() - fpidomniNamesFile.close() - os.remove(fpidomniNames) - for l in lines: - try: - pidfield = l.split()[0] # pid should be at the first position - if sys.platform == "win32": - import win32pm - if verbose(): print 'stop process '+pidfield+' : omniNames' - win32pm.killpid(int(pidfield),0) - else: - if verbose(): print 'stop process '+pidfield+' : omniNames' - os.kill(int(pidfield),signal.SIGKILL) - pass - pass - except: - pass - pass - pass - except: - pass - # - try: - process_ids=pickle.load(fpid) - fpid.close() - for process_id in process_ids: - for pid, cmd in process_id.items(): - if verbose(): print "stop process %s : %s"% (pid, cmd[0]) - try: - if sys.platform == "win32": - import win32pm - win32pm.killpid(int(pid),0) - else: - os.kill(int(pid),signal.SIGKILL) - pass - pass - except: - if verbose(): print " ------------------ process %s : %s not found"% (pid, cmd[0]) - pass - pass # for pid, cmd ... - pass # for process_id ... - pass # try... - except: - pass - # - os.remove(filedict) - cmd='ps -eo pid,command | egrep "[0-9] omniNames -start '+str(port)+'" | sed -e "s%[^0-9]*\([0-9]*\) .*%\\1%g"' - pid = commands.getoutput(cmd) - a = "" - while pid and len(a.split()) < 2: - a = commands.getoutput("kill -9 " + pid) - pid = commands.getoutput(cmd) - #print pid - pass - pass - except: - print "Cannot find or open SALOME PIDs file for port", port - pass + import PortManager # do not remove! Test for PortManager availability! + filedict = getPiDict(port) + #filedict = __guessPiDictFilename(port) + import glob + all_files = glob.glob("%s*"%filedict) + for f in all_files: + __killMyPort(port, f) + except ImportError: + filedict = __guessPiDictFilename(port) + __killMyPort(port, filedict) # + appliCleanOmniOrbConfig(port) pass -def killNotifdAndClean(port): +def cleanApplication(port): """ - Kill notifd daemon and clean application running on the specified port. + Clean application running on the specified port. Parameters: - port - port number """ + # bug fix: ensure port is an integer + if port: + port = int(port) + try: - filedict=getPiDict(port) - f=open(filedict, 'r') - pids=pickle.load(f) - for d in pids: - for pid,process in d.items(): - if 'notifd' in process: - cmd='kill -9 %d'% pid - os.system(cmd) - os.remove(filedict) + filedict=getPiDict(port) + os.remove(filedict) except: #import traceback #traceback.print_exc() pass appliCleanOmniOrbConfig(port) + pass def killMyPortSpy(pid, port): dt = 1.0 while 1: - if sys.platform == "win32": - from win32pm import killpid - if killpid(int(pid), 0) != 0: - return - else: - from os import kill - try: - kill(int(pid), 0) - except OSError, e: - if e.errno != 3: - return - break - pass + from salome_utils import killpid + ret = killpid(int(pid), 0) + if ret == 0: + break + elif ret < 0: + return from time import sleep sleep(dt) pass @@ -363,7 +371,7 @@ def killMyPortSpy(pid, port): orb = omniORB.CORBA.ORB_init(sys.argv, omniORB.CORBA.ORB_ID) import SALOME_NamingServicePy ns = SALOME_NamingServicePy.SALOME_NamingServicePy_i(orb) - import SALOME + import SALOME #@UnresolvedImport @UnusedImport session = ns.Resolve("/Kernel/Session") assert session except: @@ -383,10 +391,10 @@ def killMyPortSpy(pid, port): if __name__ == "__main__": if len(sys.argv) < 2: - print "Usage: " - print " %s " % os.path.basename(sys.argv[0]) - print - print "Kills SALOME session running on specified ." + print("Usage: ") + print(" %s " % os.path.basename(sys.argv[0])) + print() + print("Kills SALOME session running on specified .") sys.exit(1) pass if sys.argv[1] == "--spy": @@ -397,6 +405,12 @@ if __name__ == "__main__": pass sys.exit(0) pass + try: + from salomeContextUtils import setOmniOrbUserPath #@UnresolvedImport + setOmniOrbUserPath() + except Exception as e: + print(e) + sys.exit(1) for port in sys.argv[1:]: killMyPort(port) pass