Allow specifying arbitrary number of listener hosts

This commit is contained in:
XANTRONIX Development 2024-12-04 09:08:39 -05:00
parent 9cb8b2e0e4
commit 456d563f92

View file

@ -1,3 +1,4 @@
import re
import enum
import threading
import socket
@ -6,7 +7,8 @@ import ssl
from configparser import ConfigParser
from nntp.tiny.config import ConfigException
from nntp.tiny.config import (
ConfigException, ConfigSectionException, ConfigValueException)
from nntp.tiny.db import Database
from nntp.tiny.newsgroup import Newsgroup
from nntp.tiny.session import Session
@ -50,34 +52,44 @@ class Server():
return listener
def accept(self, listener):
sock, addr = listener.accept()
sock, addr = None, None
try:
sock, addr = listener.accept()
except ssl.SSLError as e:
return
def spawn():
session = Session(self, sock)
try:
session.handle()
except (ssl.SSLError, ssl.SSLEOFError) as e:
except ssl.SSLEOFError as e:
pass
thread = threading.Thread(target=spawn)
thread.start()
def run(self):
port = int(self.config['listen']['port'])
if not self.config.has_section('listen'):
raise ConfigSectionException('listen')
if not self.config.has_option('listen', 'host'):
raise ConfigValueException('listen', 'host')
hosts = re.split(r'\s*,\s*', self.config['listen']['host'])
port = int(self.config['listen']['port'])
listeners = list()
if self.config.has_option('listen', 'host_inet'):
host = self.config.get('listen', 'host_inet')
listeners.append(self.listen(host, port, socket.AF_INET))
if self.config.has_option('listen', 'host_inet6'):
host = self.config.get('listen', 'host_inet6')
listeners.append(self.listen(host, port, socket.AF_INET6))
for host in hosts:
if host.find(':') < 0:
listeners.append(self.listen(host, port, socket.AF_INET))
else:
listeners.append(self.listen(host, port, socket.AF_INET6))
if len(listeners) == 0:
raise ConfigException("No listener hosts specified")
raise ConfigException('No listener hosts specified')
sel = selectors.DefaultSelector()