2 # -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2007-2014 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
28 import cPickle as pickle
34 logger = logging.getLogger(__name__)
35 # logger.setLevel(logging.DEBUG)
36 ch = logging.StreamHandler()
37 ch.setLevel(logging.DEBUG)
38 formatter = logging.Formatter("%(levelname)s:%(threadName)s:%(message)s")
39 ch.setFormatter(formatter)
43 logger = createLogger()
45 #------------------------------------
46 # A file locker (Linux only)
47 def __acquire_lock(lock):
48 if sys.platform == "win32":
50 # lock 1 byte: file is supposed to be zero-byte long
51 msvcrt.locking(lock.fileno(), msvcrt.LK_LOCK, 1)
54 fcntl.flock(lock, fcntl.LOCK_EX)
56 def __release_lock(lock):
57 if sys.platform == "win32":
59 msvcrt.locking(lock.fileno(), msvcrt.LK_UNLCK, 1)
62 fcntl.flock(lock, fcntl.LOCK_UN)
65 def _getConfigurationFilename():
66 omniorbUserPath = os.getenv("OMNIORB_USER_PATH")
68 from salome_utils import generateFileName
69 portmanager_config = generateFileName(omniorbUserPath,
75 temp = tempfile.NamedTemporaryFile()
76 lock_file = os.path.join(os.path.dirname(temp.name), ".omniORB_PortManager.lock")
79 return (portmanager_config, lock_file)
82 def __isPortUsed(port, busy_ports):
83 return (port in busy_ports) or __isNetworkConnectionActiveOnPort(port)
86 def __isNetworkConnectionActiveOnPort(port):
87 # :NOTE: Under windows:
88 # netstat options -l and -t are unavailable
89 # grep command is unavailable
90 from subprocess import Popen, PIPE
91 (stdout, stderr) = Popen(['netstat','-an'], stdout=PIPE).communicate()
93 buf = StringIO.StringIO(stdout)
94 ports = buf.readlines()
95 # search for TCP - LISTEN connections
97 regObj = re.compile( ".*tcp.*:([0-9]+).*:.*listen", re.IGNORECASE );
100 p = int(regObj.match(item).group(1))
101 if p == port: return True
106 def getPort(preferedPort=None):
107 logger.debug("GET PORT")
109 config_file, lock_file = _getConfigurationFilename()
110 oldmask = os.umask(0)
111 with open(lock_file, 'w') as lock:
116 config = {'busy_ports':[]}
117 logger.debug("read configuration file")
119 with open(config_file, 'r') as f:
120 config = pickle.load(f)
121 except IOError: # empty file
124 logger.debug("load busy_ports: %s"%str(config["busy_ports"]))
127 busy_ports = config["busy_ports"]
129 if not port or __isPortUsed(port, busy_ports):
131 while __isPortUsed(port, busy_ports):
134 msg += "Can't find a free port to launch omniNames\n"
135 msg += "Try to kill the running servers and then launch SALOME again.\n"
136 raise RuntimeError, msg
138 logger.debug("found free port: %s"%str(port))
139 config["busy_ports"].append(port)
142 logger.debug("write busy_ports: %s"%str(config["busy_ports"]))
144 with open(config_file, 'w') as f:
145 pickle.dump(config, f)
154 logger.debug("get port: %s"%str(port))
158 def releasePort(port):
160 logger.debug("RELEASE PORT (%s)"%port)
162 config_file, lock_file = _getConfigurationFilename()
163 with open(lock_file, 'w') as lock:
168 config = {'busy_ports':[]}
169 logger.debug("read configuration file")
171 with open(config_file, 'r') as f:
172 config = pickle.load(f)
173 except IOError: # empty file
176 logger.debug("load busy_ports: %s"%str(config["busy_ports"]))
178 # remove port from list
179 busy_ports = config["busy_ports"]
181 if port in busy_ports:
182 busy_ports.remove(port)
183 config["busy_ports"] = busy_ports
186 logger.debug("write busy_ports: %s"%str(config["busy_ports"]))
188 with open(config_file, 'w') as f:
189 pickle.dump(config, f)
196 logger.debug("released port port: %s"%str(port))
200 config_file, lock_file = _getConfigurationFilename()
201 with open(lock_file, 'w') as lock:
206 config = {'busy_ports':[]}
207 logger.debug("read configuration file")
209 with open(config_file, 'r') as f:
210 config = pickle.load(f)
211 except IOError: # empty file
214 logger.debug("load busy_ports: %s"%str(config["busy_ports"]))
216 busy_ports = config["busy_ports"]