Split master server configuration options into a separate file, refactor
so that the query socket can be bound to a specific address/port. Subversion-branch: /master Subversion-revision: 2229
This commit is contained in:
parent
2687e6ddb5
commit
da3ab15169
|
@ -26,19 +26,7 @@ import struct
|
||||||
import simplejson
|
import simplejson
|
||||||
from select import select
|
from select import select
|
||||||
from time import time, strftime
|
from time import time, strftime
|
||||||
|
from master_config import *
|
||||||
# Filename of log file.
|
|
||||||
|
|
||||||
LOG_FILE = "chocolate-master.log"
|
|
||||||
|
|
||||||
# Servers must refresh themselves periodically. If nothing happens
|
|
||||||
# after this many seconds, remove them from the list.
|
|
||||||
|
|
||||||
SERVER_TIMEOUT = 2 * 60 * 60 # 2 hours
|
|
||||||
|
|
||||||
# How long is metadata valid before we force another query to refresh it?
|
|
||||||
|
|
||||||
METADATA_REFRESH_TIME = 6 * 60 * 60 # 6 hours
|
|
||||||
|
|
||||||
# Maximum length of a query response.
|
# Maximum length of a query response.
|
||||||
|
|
||||||
|
@ -58,10 +46,19 @@ NET_MASTER_PACKET_TYPE_QUERY_RESPONSE = 3
|
||||||
NET_MASTER_PACKET_TYPE_GET_METADATA = 4
|
NET_MASTER_PACKET_TYPE_GET_METADATA = 4
|
||||||
NET_MASTER_PACKET_TYPE_GET_METADATA_RESPONSE = 5
|
NET_MASTER_PACKET_TYPE_GET_METADATA_RESPONSE = 5
|
||||||
|
|
||||||
# Address and port to listen on.
|
def bind_socket_to(sock, config):
|
||||||
|
""" Bind the specified socket to the address/port configuration from
|
||||||
|
the configuration file. """
|
||||||
|
|
||||||
UDP_ADDRESS = socket.inet_ntoa(struct.pack(">l", socket.INADDR_ANY))
|
if config is not None:
|
||||||
UDP_PORT = 2342
|
if config[0] is not None:
|
||||||
|
address = socket.gethostbyname(config[0])
|
||||||
|
else:
|
||||||
|
address = socket.inet_ntoa(struct.pack(">l", socket.INADDR_ANY))
|
||||||
|
|
||||||
|
sock.bind((address, config[1]))
|
||||||
|
|
||||||
|
# Address and port to listen on.
|
||||||
|
|
||||||
def read_string(packet):
|
def read_string(packet):
|
||||||
""" Given binary packet data, read a NUL-terminated string, returning
|
""" Given binary packet data, read a NUL-terminated string, returning
|
||||||
|
@ -113,18 +110,24 @@ class MasterServer:
|
||||||
|
|
||||||
def log_output(self, addr, s):
|
def log_output(self, addr, s):
|
||||||
timestamp = strftime("%b %d %H:%M:%S")
|
timestamp = strftime("%b %d %H:%M:%S")
|
||||||
self.log_file.write("%s %s:%i: %s\n" % ((timestamp,) + addr + (s,)))
|
|
||||||
|
if addr is not None:
|
||||||
|
addr_str = "%s:%i" % addr
|
||||||
|
else:
|
||||||
|
addr_str = "-"
|
||||||
|
|
||||||
|
self.log_file.write("%s %s %s\n" % (timestamp, addr_str, s))
|
||||||
self.log_file.flush()
|
self.log_file.flush()
|
||||||
|
|
||||||
def __init__(self, address, port):
|
def __init__(self, server_address, query_address):
|
||||||
""" Initialise a new master server. """
|
""" Initialise a new master server. """
|
||||||
|
|
||||||
self.servers = {}
|
self.servers = {}
|
||||||
|
|
||||||
self.open_log_file()
|
self.open_log_file()
|
||||||
|
|
||||||
self.server_addr = (address, port)
|
self.sock = self.open_socket(server_address)
|
||||||
self.open_socket()
|
self.query_sock = self.open_socket(query_address)
|
||||||
|
|
||||||
def send_query(self, server):
|
def send_query(self, server):
|
||||||
""" Send a query to the specified server. """
|
""" Send a query to the specified server. """
|
||||||
|
@ -345,21 +348,18 @@ class MasterServer:
|
||||||
self.send_add_response(server, 0)
|
self.send_add_response(server, 0)
|
||||||
del self.servers[server.addr]
|
del self.servers[server.addr]
|
||||||
|
|
||||||
def open_socket(self):
|
def open_socket(self, address):
|
||||||
""" Open the server socket and bind to the listening address. """
|
""" Open a server socket and bind to the specified address. """
|
||||||
|
|
||||||
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||||
self.sock.bind(self.server_addr)
|
bind_socket_to(sock, address)
|
||||||
|
|
||||||
# Query socket, used to send queries to servers to check that
|
return sock
|
||||||
# they are actually accessible.
|
|
||||||
|
|
||||||
self.query_sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
""" Run the server main loop, listening for packets. """
|
""" Run the server main loop, listening for packets. """
|
||||||
|
|
||||||
self.log_output(self.server_addr, "Server started.")
|
self.log_output(None, "Server started.")
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
r, w, x = select([self.sock, self.query_sock], [], [], 5)
|
r, w, x = select([self.sock, self.query_sock], [], [], 5)
|
||||||
|
@ -373,6 +373,6 @@ class MasterServer:
|
||||||
self.rx_packet_query_sock()
|
self.rx_packet_query_sock()
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
server = MasterServer(UDP_ADDRESS, UDP_PORT)
|
server = MasterServer(SERVER_ADDRESS, QUERY_ADDRESS)
|
||||||
server.run()
|
server.run()
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,43 @@
|
||||||
|
#
|
||||||
|
# Copyright(C) 2010 Simon Howard
|
||||||
|
#
|
||||||
|
# This program is free software; you can redistribute it and/or
|
||||||
|
# modify it under the terms of the GNU General Public License
|
||||||
|
# as published by the Free Software Foundation; either version 2
|
||||||
|
# of the License, or (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program; if not, write to the Free Software
|
||||||
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
|
||||||
|
# 02111-1307, USA.
|
||||||
|
#
|
||||||
|
#
|
||||||
|
# Configuration file for master server.
|
||||||
|
#
|
||||||
|
|
||||||
|
# Filename of log file.
|
||||||
|
|
||||||
|
LOG_FILE = "chocolate-master.log"
|
||||||
|
|
||||||
|
# Servers must refresh themselves periodically. If nothing happens
|
||||||
|
# after this many seconds, remove them from the list.
|
||||||
|
|
||||||
|
SERVER_TIMEOUT = 2 * 60 * 60 # 2 hours
|
||||||
|
|
||||||
|
# How long is metadata valid before we force another query to refresh it?
|
||||||
|
|
||||||
|
METADATA_REFRESH_TIME = 6 * 60 * 60 # 6 hours
|
||||||
|
|
||||||
|
# Address and port to listen on.
|
||||||
|
|
||||||
|
SERVER_ADDRESS = (None, 2342)
|
||||||
|
|
||||||
|
# Address and port to bind query socket.
|
||||||
|
|
||||||
|
QUERY_ADDRESS = None
|
||||||
|
|
Loading…
Reference in New Issue