litex_server: check socket flags exist before using them

Some flags are only available on certain platforms.  Verify these flags
exist prior to using them when opening a socket.

See
https://stackoverflow.com/questions/14388706/socket-options-so-reuseaddr-and-so-reuseport-how-do-they-differ-do-they-mean-t
for more information

Signed-off-by: Sean Cross <sean@xobs.io>
This commit is contained in:
Sean Cross 2019-04-20 17:28:26 +08:00
parent 9ee6c35b42
commit f71b8d4f57
1 changed files with 6 additions and 1 deletions

View File

@ -21,8 +21,13 @@ class RemoteServer(EtherboneIPC):
def open(self): def open(self):
if hasattr(self, "socket"): if hasattr(self, "socket"):
return return
socket_flags = 0
if hasattr(socket, "SO_REUSEADDR"):
socket_flags = socket_flags | socket.SO_REUSEADDR
if hasattr(socket, "SO_REUSEPORT"):
socket_flags = socket_flags | socket.SO_REUSEPORT
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1) self.socket.setsockopt(socket.SOL_SOCKET, socket_flags, 1)
self.socket.bind((self.bind_ip, self.bind_port)) self.socket.bind((self.bind_ip, self.bind_port))
print("tcp port: {:d}".format(self.bind_port)) print("tcp port: {:d}".format(self.bind_port))
self.socket.listen(1) self.socket.listen(1)