]>
git.proxmox.com Git - mirror_ovs.git/blob - python/ovs/socket_util.py
1 # Copyright (c) 2010 Nicira Networks
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at:
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
22 import ovs
.fatal_signal
25 def make_unix_socket(style
, nonblock
, bind_path
, connect_path
):
26 """Creates a Unix domain socket in the given 'style' (either
27 socket.SOCK_DGRAM or socket.SOCK_STREAM) that is bound to 'bind_path' (if
28 'bind_path' is not None) and connected to 'connect_path' (if 'connect_path'
29 is not None). If 'nonblock' is true, the socket is made non-blocking.
31 Returns (error, socket): on success 'error' is 0 and 'socket' is a new
32 socket object, on failure 'error' is a positive errno value and 'socket' is
36 sock
= socket
.socket(socket
.AF_UNIX
, style
)
37 except socket
.error
, e
:
38 return get_exception_errno(e
), None
43 if bind_path
is not None:
44 # Delete bind_path but ignore ENOENT.
48 if e
.errno
!= errno
.ENOENT
:
51 ovs
.fatal_signal
.add_file_to_unlink(bind_path
)
55 if sys
.hexversion
>= 0x02060000:
56 os
.fchmod(sock
.fileno(), 0700)
58 os
.chmod("/dev/fd/%d" % sock
.fileno(), 0700)
61 if connect_path
is not None:
63 sock
.connect(connect_path
)
64 except socket
.error
, e
:
65 if get_exception_errno(e
) != errno
.EINPROGRESS
:
68 except socket
.error
, e
:
74 if bind_path
is not None:
75 ovs
.fatal_signal
.add_file_to_unlink(bind_path
)
76 return get_exception_errno(e
), None
79 def check_connection_completion(sock
):
81 p
.register(sock
, select
.POLLOUT
)
82 if len(p
.poll(0)) == 1:
83 return get_socket_error(sock
)
88 def get_socket_error(sock
):
89 """Returns the errno value associated with 'socket' (0 if no error) and
90 resets the socket's error status."""
91 return sock
.getsockopt(socket
.SOL_SOCKET
, socket
.SO_ERROR
)
94 def get_exception_errno(e
):
95 """A lot of methods on Python socket objects raise socket.error, but that
96 exception is documented as having two completely different forms of
97 arguments: either a string or a (errno, string) tuple. We only want the
99 if type(e
.args
) == tuple:
109 """Returns a readable and writable fd for /dev/null, if successful,
110 otherwise a negative errno value. The caller must not close the returned
111 fd (because the same fd will be handed out to subsequent callers)."""
115 null_fd
= os
.open("/dev/null", os
.O_RDWR
)
117 logging
.error("could not open /dev/null: %s"
118 % os
.strerror(e
.errno
))
123 def write_fully(fd
, buf
):
124 """Returns an (error, bytes_written) tuple where 'error' is 0 on success,
125 otherwise a positive errno value, and 'bytes_written' is the number of
126 bytes that were written before the error occurred. 'error' is 0 if and
127 only if 'bytes_written' is len(buf)."""
133 retval
= os
.write(fd
, buf
)
135 if retval
== len(buf
):
136 return 0, bytes_written
+ len(buf
)
138 logging
.warning("write returned 0")
139 return errno
.EPROTO
, bytes_written
141 bytes_written
+= retval
144 return e
.errno
, bytes_written
147 def set_nonblocking(sock
):
150 except socket
.error
, e
:
151 logging
.error("could not set nonblocking mode on socket: %s"
152 % os
.strerror(get_socket_error(e
)))