X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=bin%2FkillSalomeWithPort.py;h=fec9e4aa356850f7a1794f4bc39e881283a9febf;hb=ee44cdc8da140e8f0aabaa1b9d9485939899cfdf;hp=bf5ca5ef7954c06376dd974063cae713047ecd20;hpb=1a9c06c1b6b7bb7a2a9787229e1c113f4c81cb0c;p=modules%2Fkernel.git diff --git a/bin/killSalomeWithPort.py b/bin/killSalomeWithPort.py index bf5ca5ef7..fec9e4aa3 100755 --- a/bin/killSalomeWithPort.py +++ b/bin/killSalomeWithPort.py @@ -1,118 +1,417 @@ -#!/usr/bin/env python -import os, sys, pickle, signal, commands - -def getPiDict(port,appname='salome',full=True): - from Utils_Identity import getShortHostName - - host = os.getenv("HOSTNAME") - if not host: - host = os.getenv("HOST") - if not host: - host = getShortHostName() - - filedict = [] - filedict.append( os.getenv('USER') ) # user name - filedict.append( host ) # host name - filedict.append( str(port) ) # port number - filedict.append( appname.upper() ) # application name - filedict.append( 'pidict' ) # constant part - - filedict = '_'.join(filedict) +#! /usr/bin/env python3 +# -*- coding: iso-8859-1 -*- +# 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 +# +# 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, 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 +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +# +# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com +# + +## \file killSalomeWithPort.py +# Stop all %SALOME servers from given sessions by killing them +# +# The sessions are indicated by their ports on the command line as in : +# \code +# killSalomeWithPort.py 2811 2815 +# \endcode +# + +import os, sys, pickle, signal, subprocess,glob +import subprocess +import shlex +from salome_utils import verbose + + +def getPiDict(port,appname='salome',full=True,hidden=True,hostname=None): + """ + Get file with list of SALOME processes. + This file is located in the user's home directory + and named .___SALOME_pidict + where + is user name + is host name + is port number + + Parameters: + - port : port number + - appname : application name (default is 'SALOME') + - full : if True, full path to the file is returned, otherwise only file name is returned + - hidden : if True, file name is prefixed with . (dot) symbol; this internal parameter is used + to support compatibility with older versions of SALOME + """ + # 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") + if hostname: hostname = hostname.split(".")[0] + pass if full: - filedict = os.getenv("HOME") + '/' + filedict - return filedict + # full path to the pidict file is requested + if hidden: + # new-style dot-prefixed pidict files + # are in the system-dependant temporary diretory + 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, + with_username=True, + with_hostname=hostname or True, + with_port=port, + with_app=appname.upper()) def appliCleanOmniOrbConfig(port): """ - remove omniorb config files related to the port in SALOME application: - - ${HOME}/${APPLI}/.omniORB_${HOSTNAME}_${NSPORT}.cfg - - ${HOME}/${APPLI}/.omniORB_${HOSTNAME}_last.cfg + Remove omniorb config files related to the port in SALOME application: + - ${OMNIORB_USER_PATH}/.omniORB_${USER}_${HOSTNAME}_${NSPORT}.cfg + - ${OMNIORB_USER_PATH}/.omniORB_${USER}_last.cfg the last is removed only if the link points to the first file. """ - from Utils_Identity import getShortHostName - appli=os.environ.get("APPLI") - if appli is None: + 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: #Run outside application context pass else: - home = os.environ['HOME'] - home='%s/%s'%(home,appli) - hostname=getShortHostName() - omniorb_config = '%s/.omniORB_%s_%s.cfg'%(home,hostname, str(port)) - last_running_config = '%s/.omniORB_%s_last.cfg'%(home, hostname) + omniorb_config = generateFileName(omniorbUserPath, prefix="omniORB", + extension="cfg", + hidden=True, + with_username=True, + with_hostname=True, + with_port=port) + last_running_config = generateFileName(omniorbUserPath, prefix="omniORB", + with_username=True, + suffix="last", + extension="cfg", + hidden=True) if os.access(last_running_config,os.F_OK): - pointedPath = os.readlink(last_running_config) - if pointedPath[0] != '/': - pointedPath=os.path.join(os.path.dirname(last_running_config), pointedPath) - if pointedPath == omniorb_config: - os.unlink(last_running_config) + if not sys.platform == 'win32': + pointedPath = os.readlink(last_running_config) + if pointedPath[0] != '/': + pointedPath=os.path.join(os.path.dirname(last_running_config), pointedPath) + pass + if pointedPath == omniorb_config: + os.unlink(last_running_config) + pass + pass + else: + os.remove(last_running_config) pass pass + if os.access(omniorb_config,os.F_OK): os.remove(omniorb_config) pass + + if os.path.lexists(last_running_config):return + + #try to relink last.cfg to an existing config file if any + files = glob.glob(os.path.join(omniorbUserPath,".omniORB_"+getUserName()+"_*.cfg")) + current_config=None + current=0 + for f in files: + stat=os.stat(f) + if stat.st_atime > current: + current=stat.st_atime + current_config=f + if current_config: + if sys.platform == "win32": + import shutil + shutil.copyfile(os.path.normpath(current_config), last_running_config) + pass + else: + os.symlink(os.path.normpath(current_config), last_running_config) + pass + pass pass + pass ########## kills all salome processes with the given port ########## -def killMyPort(port): - filedict=getPiDict(port) - found = 0 +def shutdownMyPort(port, cleanup=True): + """ + Shutdown SALOME session running on the specified port. + Parameters: + - port - port number + """ + if not port: return + # bug fix: ensure port is an integer + port = int(port) + try: - fpid=open(filedict, 'r') - found = 1 - except: - print "file %s giving SALOME process id is not readable"% filedict - pass - - if found: - cmd = 'pid=`ps -eo pid,command | egrep "[0-9] omniNames -start '+str(port)+'"` ; echo $pid > /tmp/logs/'+os.getenv('USER')+"/_"+port+'_Pid_omniNames.log' - a = os.system(cmd) - try: - fpidomniNames=open('/tmp/logs/'+os.getenv('USER')+"/_"+port+'_Pid_omniNames.log') - prc = fpidomniNames.read() - fpidomniNames.close() - if prc != None : - for field in prc.split(" ") : - if field == "omniNames" : - if pidfield != "egrep" : - print 'stop process '+pidfield+' : omniNames' - os.system('kill -9 '+pidfield) - pidfield = field - except: + from PortManager import releasePort + releasePort(port) + except ImportError: + pass + + from salome_utils import generateFileName + + # set OMNIORB_CONFIG variable to the proper file + omniorbUserPath = os.getenv("OMNIORB_USER_PATH") + kwargs = {} + if omniorbUserPath is not None: + kwargs["with_username"]=True + else: + omniorbUserPath = os.path.realpath(os.path.expanduser('~')) + omniorb_config = generateFileName(omniorbUserPath, prefix="omniORB", + extension="cfg", + hidden=True, + with_hostname=True, + with_port=port, + **kwargs) + os.environ['OMNIORB_CONFIG'] = omniorb_config + os.environ['NSPORT'] = str(port) + + # give the chance to the servers to shutdown properly + try: + import time + from omniORB import CORBA + + from LifeCycleCORBA import LifeCycleCORBA + # shutdown all + orb = CORBA.ORB_init([''], CORBA.ORB_ID) + 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 + if cleanup: + lcc.killOmniNames() + time.sleep(1) pass - - try: + 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) - fpid.close() for process_id in process_ids: - - for pid, cmd in process_id.items(): - print "stop process %s : %s"% (pid, cmd[0]) + for pid, cmd in list(process_id.items()): + if verbose(): print("stop process %s : %s"% (pid, cmd[0])) try: - os.kill(int(pid),signal.SIGKILL) + from salome_utils import killpid + killpid(int(pid)) except: - print " ------------------ process %s : %s not found"% (pid, cmd[0]) + 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): + """ + Kill SALOME session running on the specified port. + Parameters: + - port - port number + """ + # bug fix: ensure port is an integer + if port: + port = int(port) + + try: + 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 - 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("pid=`ps -eo pid,command | egrep \"[0-9] omniNames -start "+str(port)+"\" | sed -e \"s%[^0-9]*\([0-9]*\) .*%\\1%g\"`; kill -9 $pid") - pid = commands.getoutput("ps -eo pid,command | egrep \"[0-9] omniNames -start "+str(port)+"\" | sed -e \"s%[^0-9]*\([0-9]*\) .*%\\1%g\"") - print pid - + return + except: pass + # try to shutdown session normally + import threading, time + 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) + + try: + 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 cleanApplication(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) + os.remove(filedict) + except: + #import traceback + #traceback.print_exc() + pass + appliCleanOmniOrbConfig(port) pass - + +def killMyPortSpy(pid, port): + dt = 1.0 + while 1: + 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 + filedict = getPiDict(port, hidden=True) + if not os.path.exists(filedict): + return + try: + import omniORB + orb = omniORB.CORBA.ORB_init(sys.argv, omniORB.CORBA.ORB_ID) + import SALOME_NamingServicePy + ns = SALOME_NamingServicePy.SALOME_NamingServicePy_i(orb) + import SALOME #@UnresolvedImport @UnusedImport + session = ns.Resolve("/Kernel/Session") + assert session + except: + return + try: + status = session.GetStatSession() + except: + # -- session is in naming service but has crash + status = None + pass + if status: + if not status.activeGUI: + return + pass + killMyPort(port) + return 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 .") + sys.exit(1) + pass + if sys.argv[1] == "--spy": + if len(sys.argv) > 3: + pid = sys.argv[2] + port = sys.argv[3] + killMyPortSpy(pid, port) + 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 + pass