- if self.simulation.node.get("hostname") in ['localhost', '127.0.0.1']:
- sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
- sock.connect(self.simulation.remote_socket)
- sock.send("%s\n" % encoded)
- reply = sock.recv(1024)
- sock.close()
- else:
- command = ( "python -c 'import socket;"
- "sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM);"
- "sock.connect(\"%(socket_addr)s\");"
- "msg = \"%(encoded_message)s\\n\";"
- "sock.send(msg);"
- "reply = sock.recv(1024);"
- "sock.close();"
- "print reply'") % {
- "encoded_message": encoded,
- "socket_addr": self.simulation.remote_socket,
- }
-
- (reply, err), proc = self.simulation.node.execute(command,
- with_lock = True)
-
- if (err and proc.poll()) or reply.strip() == "":
- msg = (" Couldn't connect to remote socket %s - REPLY: %s "
- "- ERROR: %s ") % (
- self.simulation.remote_socket, reply, err)
- self.simulation.error(msg, reply, err)
- raise RuntimeError(msg)
-
+ with self._socket_lock:
+ if self.simulation.node.get("hostname") in ['localhost', '127.0.0.1']:
+ sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+ sock.connect(self.simulation.remote_socket)
+ sock.send("%s\n" % encoded)
+ reply = sock.recv(1024)
+ sock.close()
+ else:
+ command = ( "python -c 'import socket;"
+ "sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM);"
+ "sock.connect(\"%(socket_addr)s\");"
+ "msg = \"%(encoded_message)s\\n\";"
+ "sock.send(msg);"
+ "reply = sock.recv(1024);"
+ "sock.close();"
+ "print reply'") % {
+ "encoded_message": encoded,
+ "socket_addr": self.simulation.remote_socket,
+ }
+
+ (reply, err), proc = self.simulation.node.execute(command,
+ with_lock = True)
+
+ if (err and proc.poll()) or reply.strip() == "":
+ msg = (" Couldn't connect to remote socket %s - REPLY: %s "
+ "- ERROR: %s ") % (
+ self.simulation.remote_socket, reply, err)
+ self.simulation.error(msg, reply, err)
+ raise RuntimeError(msg)
+