#! /usr/bin/env ruby
-## -*-Ruby-*- $RCSfile$ $Revision$ $Name$
+## $Id$
-## Copyright (C) 2004 773@2ch
+## Copyright (C) 2004 NABEYA Kenichi (aka nanami@2ch)
+## Copyright (C) 2007-2008 Daigo Moriwaki (daigo at debian dot org)
##
## 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
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
-DEFAULT_TIMEOUT = 10 # for single socket operation
-Total_Time = 1500
-Least_Time_Per_Move = 1
-Watchdog_Time = 30 # time for ping
-Login_Time = 300 # time for LOGIN
+TOP_DIR = File.expand_path(File.dirname(__FILE__))
+$:.unshift File.dirname(__FILE__)
+require 'shogi_server'
-Release = "$Name$".split[1].sub(/\A[^\d]*/, '').gsub(/_/, '.')
-Release.concat("-") if (Release == "")
-Revision = "$Revision$".gsub(/[^\.\d]/, '')
+#################################################
+# MAIN
+#
-STDOUT.sync = true
-STDERR.sync = true
+ShogiServer.reload
-require 'getoptlong'
-require 'thread'
-require 'timeout'
-require 'socket'
-require 'ping'
-
-TCPSocket.do_not_reverse_lookup = true
-
-class TCPSocket
- def gets_timeout(t = DEFAULT_TIMEOUT)
- begin
- timeout(t) do
- return self.gets
- end
- rescue TimeoutError
- return nil
- rescue
- return nil
- end
- end
- def gets_safe
- begin
- return self.gets
- rescue
- return nil
- end
- end
- def write_safe(str)
- begin
- return self.write(str)
- rescue
- return nil
- end
- end
-end
-
-
-class League
- def initialize
- @hash = Hash::new
- end
- attr_accessor :hash
-
- def add(player)
- @hash[player.name] = player
- end
- def delete(player)
- @hash.delete(player.name)
- end
- def duplicated?(player)
- if (@hash[player.name])
- return true
- else
- return false
- end
- end
- def get_player(status, game_name, sente, searcher=nil)
- @hash.each do |name, player|
- if ((player.status == status) &&
- (player.game_name == game_name) &&
- ((player.sente == nil) || (player.sente == sente)) &&
- ((searcher == nil) || (player != searcher)))
- return player
- end
- end
- return nil
- end
- def new_game(game_name, player0, player1)
- game = Game::new(game_name, player0, player1)
- end
-end
-
-
-
-
-class Player
- def initialize(str, socket)
- @name = nil
- @password = nil
- @socket = socket
- @status = "connected" # game_waiting -> agree_waiting -> start_waiting -> game
-
- @protocol = nil # CSA or x1
- @eol = "\m" # favorite eol code
- @game = nil
- @game_name = ""
- @mytime = Total_Time
- @sente = nil
- @watchdog_thread = nil
-
- login(str)
- end
-
- attr_accessor :name, :password, :socket, :status
- attr_accessor :protocol, :eol, :game, :mytime, :watchdog_thread, :game_name, :sente
-
- def finish
- log_message(sprintf("user %s finish", @name))
- Thread::kill(@watchdog_thread) if @watchdog_thread
- @socket.close if (! @socket.closed?)
- end
-
- def watchdog(time)
- while true
- begin
- Ping.pingecho(@socket.addr[3])
- rescue
- end
- sleep(time)
- end
- end
-
- def to_s
- if ((status == "game_waiting") ||
- (status == "agree_waiting") ||
- (status == "game"))
- if (@sente)
- return sprintf("%s %s %s +", @name, @status, @game_name)
- elsif (@sente == false)
- return sprintf("%s %s %s -", @name, @status, @game_name)
- elsif (@sente == nil)
- return sprintf("%s %s %s +-", @name, @status, @game_name)
- end
- else
- return sprintf("%s %s", @name, @status)
- end
- end
-
- def write_help
- @socket.write_safe('##[HELP] available commands "%%WHO", "%%CHAT str", "%%GAME game_name +", "%%GAME game_name -"')
- end
-
- def write_safe(str)
- @socket.write_safe(str.gsub(/[\r\n]+/, @eol))
- end
-
- def login(str)
- str =~ /([\r\n]*)$/
- @eol = $1
- str.chomp!
- (login, @name, @password, ext) = str.split
- if (ext)
- @protocol = "x1"
- else
- @protocol = "CSA"
- end
- @watchdog_thread = Thread::start do
- watchdog(Watchdog_Time)
- end
- end
-
- def run
- if (@protocol != "CSA")
- log_message(sprintf("user %s run in %s mode", @name, @protocol))
- write_safe(sprintf("##[LOGIN] +OK %s\n", @protocol))
- else
- log_message(sprintf("user %s run in CSA mode", @name))
- csa_1st_str = "%%GAME default +-"
- end
-
-
- while (csa_1st_str || (str = @socket.gets_safe))
- begin
- $mutex.lock
- if (csa_1st_str)
- str = csa_1st_str
- csa_1st_str = nil
- end
- str.chomp!
- case str
- when /^[\+\-%][^%]/
- if (@status == "game")
- s = @game.handle_one_move(str, self)
- return if (s && @protocol == "CSA")
- else
- next
- end
- when /^AGREE/
- if (@status == "agree_waiting")
- @status = "start_waiting"
- if ((@game.sente.status == "start_waiting") &&
- (@game.gote.status == "start_waiting"))
- @game.start
- @game.sente.status = "game"
- @game.gote.status = "game"
- end
- else
- write_safe("## you are in %s status. AGREE is valid in agree_waiting status\n", @status)
- next
- end
- when /^%%HELP/
- write_help
- when /^%%GAME\s+(\S+)\s+([\+\-]+)/
- if ((@status == "connected") || (@status == "game_waiting"))
- @status = "game_waiting"
- else
- write_safe("## you are in %s status. GAME is valid in connected or game_waiting status\n", @status)
- next
- end
- @status = "game_waiting"
- @game_name = $1
- sente_str = $2
- if (sente_str == "+")
- @sente = true
- rival_sente = false
- elsif (sente_str == "-")
- @sente = false
- rival_sente = true
- else
- @sente = nil
- rival_sente = nil
- end
- rival = LEAGUE.get_player("game_waiting", @game_name, rival_sente, self)
- rival = LEAGUE.get_player("game_waiting", @game_name, nil, self) if (! rival)
- if (rival)
- if (@sente == nil)
- if (rand(2) == 0)
- @sente = true
- rival_sente = false
- else
- @sente = false
- rival_sente = true
- end
- elsif (rival_sente == nil)
- if (@sente)
- rival_sente = false
- else
- rival_sente = true
- end
- end
- rival.sente = rival_sente
- LEAGUE.new_game(@game_name, self, rival)
- self.status = "agree_waiting"
- rival.status = "agree_waiting"
- end
- when /^%%CHAT\s+(\S+)/
- message = $1
- LEAGUE.hash.each do |name, player|
- if (player.protocol != "CSA")
- s = player.write_safe(sprintf("##[CHAT][%s] %s\n", @name, message))
- player.status = "zombie" if (! s)
- end
- end
- when /^%%WHO/
- buf = Array::new
- LEAGUE.hash.each do |name, player|
- buf.push(sprintf("##[WHO] %s\n", player.to_s))
- end
- buf.push("##[WHO] +OK\n")
- write_safe(buf.join)
- when /^%%LOGOUT/
- finish
- return
- else
- write_safe(sprintf("## unknown command %s\n", str))
- end
- ensure
- $mutex.unlock
- end
- end # enf of while
- end
-end
-
-class Board
-end
-
-class Game
- def initialize(game_name, player0, player1)
- @game_name = game_name
- if (player0.sente)
- @sente = player0
- @gote = player1
- else
- @sente = player1
- @gote = player0
- end
- @current_player = @sente
- @next_player = @gote
-
- @sente.game = self
- @gote.game = self
- @sente.status = "agree_waiting"
- @gote.status = "agree_waiting"
- @id = sprintf("%s-%s-%s-%s", @game_name, @sente.name, @gote.name, Time::new.strftime("%Y%m%d%H%M%S"))
- log_message(sprintf("game created %s %s %s", game_name, sente.name, gote.name))
-
- @logfile = @id + ".csa"
- @board = Board::new
- @start_time = nil
- @fh = nil
-
- propose
- end
- attr_accessor :game_name, :sente, :gote, :id, :board, :current_player, :next_player, :fh
-
- def finish
- log_message(sprintf("game finished %s %s %s", game_name, sente.name, gote.name))
- @fh.printf("'$END_TIME:%s\n", Time::new.strftime("%Y/%m/%d %H:%M:%S"))
- @fh.close
- @sente.status = "connected"
- @gote.status = "connected"
- if (@current_player.protocol == "CSA")
- @current_player.finish
- end
- end
-
- def handle_one_move(str, player)
- finish_flag = false
- if (@current_player == player)
- @end_time = Time::new
- t = @end_time - @start_time
- t = Least_Time_Per_Move if (t < Least_Time_Per_Move)
- @sente.write_safe(sprintf("%s,T%d\n", str, t))
- @gote.write_safe(sprintf("%s,T%d\n", str, t))
- @fh.printf("%s\nT%d\n", str, t)
- @current_player.mytime = @current_player.mytime - t
- if (@current_player.mytime < 0)
- timeout_end()
- finish_flag = true
- elsif (str =~ /%KACHI/)
- kachi_end()
- finish_flag = true
- elsif (str =~ /%TORYO/)
- toryo_end
- finish_flag = true
- end
- (@current_player, @next_player) = [@next_player, @current_player]
- @start_time = Time::new
- finish if (finish_flag)
- return finish_flag
- end
- end
-
- def timeout_end
- @current_player.status = "connected"
- @next_player.status = "connected"
- @current_player.write_safe("#TIME_UP\n#LOSE\n")
- @next_player.write_safe("#TIME_UP\n#WIN\n")
- end
-
- def kachi_end
- @current_player.status = "connected"
- @next_player.status = "connected"
- @current_player.write_safe("#JISHOGI\n#WIN\n")
- @next_player.write_safe("#JISHOGI\n#LOSE\n")
- end
-
- def toryo_end
- @current_player.status = "connected"
- @next_player.status = "connected"
- @current_player.write_safe("#RESIGN\n#LOSE\n")
- @next_player.write_safe("#RESIGN\n#WIN\n")
- end
-
- def start
- log_message(sprintf("game started %s %s %s", game_name, sente.name, gote.name))
- @sente.write_safe(sprintf("START:%s\n", @id))
- @gote.write_safe(sprintf("START:%s\n", @id))
- @start_time = Time::new
- end
-
- def propose
- begin
- @fh = open(@logfile, "w")
- @fh.sync = true
-
- @fh.printf("V2\n")
- @fh.printf("N+%s\n", @sente.name)
- @fh.printf("N-%s\n", @gote.name)
- @fh.printf("$EVENT:%s\n", @id)
-
- @sente.write_safe(propose_message("+"))
- @gote.write_safe(propose_message("-"))
-
- @fh.printf("$START_TIME:%s\n", Time::new.strftime("%Y/%m/%d %H:%M:%S"))
- @fh.print <<EOM
-P1-KY-KE-GI-KI-OU-KI-GI-KE-KY
-P2 * -HI * * * * * -KA *
-P3-FU-FU-FU-FU-FU-FU-FU-FU-FU
-P4 * * * * * * * * *
-P5 * * * * * * * * *
-P6 * * * * * * * * *
-P7+FU+FU+FU+FU+FU+FU+FU+FU+FU
-P8 * +KA * * * * * +HI *
-P9+KY+KE+GI+KI+OU+KI+GI+KE+KY
-+
-EOM
- end
- end
-
- def propose_message(sg_flag)
- str = <<EOM
-Protocol_Mode:Server
-Format:Shogi 1.0
-Game_ID:#{@id}
-Name+:#{@sente.name}
-Name-:#{@gote.name}
-Your_Turn:#{sg_flag}
-Rematch_On_Draw:NO
-To_Move:+
-BEGIN Time
-Time_Unit:1sec
-Total_Time:#{Total_Time}
-Least_Time_Per_Move:#{Least_Time_Per_Move}
-END Time
-BEGIN Position
-Jishogi_Declaration:1.1
-P1-KY-KE-GI-KI-OU-KI-GI-KE-KY
-P2 * -HI * * * * * -KA *
-P3-FU-FU-FU-FU-FU-FU-FU-FU-FU
-P4 * * * * * * * * *
-P5 * * * * * * * * *
-P6 * * * * * * * * *
-P7+FU+FU+FU+FU+FU+FU+FU+FU+FU
-P8 * +KA * * * * * +HI *
-P9+KY+KE+GI+KI+OU+KI+GI+KE+KY
-P+
-P-
-+
-END Position
-END Game_Summary
-EOM
- return str
+def gets_safe(socket, timeout=nil)
+ if r = select([socket], nil, nil, timeout)
+ return r[0].first.gets
+ else
+ return :timeout
end
+rescue Exception => ex
+ log_error("gets_safe: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}")
+ return :exception
end
def usage
shogi-server - server for CSA server protocol
SYNOPSIS
- shogi-server event_name port_number
+ shogi-server [OPTIONS] event_name port_number
DESCRIPTION
server for CSA server protocol
OPTIONS
--pid-file file
specify filename for logging process ID
+ --daemon dir
+ run as a daemon. Log files will be put in dir.
+ --player-log-dir dir
+ log network messages for each player. Log files
+ will be put in the dir.
+ --floodgate_history
+ file name to record Floodgate game history
+ default: './floodgate_history.yaml'
LICENSE
- this file is distributed under GPL version2 and might be compiled by Exerb
+ GPL versoin 2 or later
SEE ALSO
RELEASE
- #{Release}
+ #{ShogiServer::Release}
REVISION
- #{Revision}
+ #{ShogiServer::Revision}
EOM
end
+
+def log_debug(str)
+ $logger.debug(str)
+end
+
def log_message(str)
- printf("%s message: %s\n", Time::new.to_s, str)
+ $logger.info(str)
end
def log_warning(str)
- printf("%s message: %s\n", Time::new.to_s, str)
+ $logger.warn(str)
end
def log_error(str)
- printf("%s error: %s\n", Time::new.to_s, str)
+ $logger.error(str)
end
def parse_command_line
options = Hash::new
- parser = GetoptLong.new
- parser.ordering = GetoptLong::REQUIRE_ORDER
- parser.set_options(
- ["--pid-file", GetoptLong::REQUIRED_ARGUMENT])
-
+ parser = GetoptLong.new(
+ ["--daemon", GetoptLong::REQUIRED_ARGUMENT],
+ ["--pid-file", GetoptLong::REQUIRED_ARGUMENT],
+ ["--player-log-dir", GetoptLong::REQUIRED_ARGUMENT])
parser.quiet = true
begin
parser.each_option do |name, arg|
return options
end
-LEAGUE = League::new
+def write_pid_file(file)
+ open(file, "w") do |fh|
+ fh.puts "#{$$}"
+ end
+end
-def good_login?(str)
- return false if (str !~ /^LOGIN /)
- tokens = str.split
- if ((tokens.length == 3) ||
- ((tokens.length == 4) && tokens[3] == "x1"))
- ## ok
- else
- return false
+def mutex_watchdog(mutex, sec)
+ sec = 1 if sec < 1
+ queue = []
+ while true
+ if mutex.try_lock
+ queue.clear
+ mutex.unlock
+ else
+ queue.push(Object.new)
+ if queue.size > sec
+ # timeout
+ log_error("mutex watchdog timeout: %d sec" % [sec])
+ queue.clear
+ end
+ end
+ sleep(1)
end
- return true
end
-def write_pid_file(file)
- open(file, "w") do |fh|
- fh.print Process::pid, "\n"
+def login_loop(client)
+ player = login = nil
+
+ while r = select([client], nil, nil, ShogiServer::Login_Time) do
+ break unless str = r[0].first.gets
+ $mutex.lock # guards LEAGUE
+ begin
+ str =~ /([\r\n]*)$/
+ eol = $1
+ if (ShogiServer::Login::good_login?(str))
+ player = ShogiServer::Player::new(str, client, eol)
+ login = ShogiServer::Login::factory(str, player)
+ if (current_player = LEAGUE.find(player.name))
+ if (current_player.password == player.password &&
+ current_player.status != "game")
+ log_message(sprintf("user %s login forcely", player.name))
+ current_player.kill
+ else
+ login.incorrect_duplicated_player(str)
+ player = nil
+ break
+ end
+ end
+ LEAGUE.add(player)
+ break
+ else
+ client.write("LOGIN:incorrect" + eol)
+ client.write("type 'LOGIN name password' or 'LOGIN name password x1'" + eol) if (str.split.length >= 4)
+ end
+ ensure
+ $mutex.unlock
+ end
+ end # login loop
+ return [player, login]
+end
+
+def setup_logger(log_file)
+ logger = Logger.new(log_file, 'daily')
+ logger.formatter = ShogiServer::Formatter.new
+ logger.level = $DEBUG ? Logger::DEBUG : Logger::INFO
+ logger.datetime_format = "%Y-%m-%d %H:%M:%S"
+ return logger
+end
+
+def setup_watchdog_for_giant_lock
+ $mutex = Mutex::new
+ Thread::start do
+ Thread.pass
+ mutex_watchdog($mutex, 10)
+ end
+end
+
+def setup_floodgate
+ return Thread.start do
+ Thread.pass
+ floodgate = ShogiServer::League::Floodgate.new(LEAGUE)
+ log_message("Flooddgate reloaded. The next match will start at %s." %
+ [floodgate.next_time])
+
+ while (true)
+ begin
+ diff = floodgate.next_time - Time.now
+ if diff > 0
+ sleep(diff/2)
+ next
+ end
+ LEAGUE.reload
+ floodgate.match_game
+ floodgate.charge
+ next_time = floodgate.next_time
+ $mutex.synchronize do
+ log_message("Reloading source...")
+ ShogiServer.reload
+ end
+ floodgate = ShogiServer::League::Floodgate.new(LEAGUE, next_time)
+ log_message("Floodgate will start the next match at %s." %
+ [floodgate.next_time])
+ rescue Exception => ex
+ # ignore errors
+ log_error("[in Floodgate's thread] #{ex} #{ex.backtrace}")
+ end
+ end
end
end
def main
- $mutex = Mutex::new
+
$options = parse_command_line
if (ARGV.length != 2)
usage
exit 2
end
- event = ARGV.shift
+ if $options["player-log-dir"]
+ $options["player-log-dir"] = File.expand_path($options["player-log-dir"])
+ end
+ if $options["player-log-dir"] &&
+ !File.directory?($options["player-log-dir"])
+ usage
+ exit 3
+ end
+ if $options["pid-file"]
+ $options["pid-file"] = File.expand_path($options["pid-file"])
+ end
+ $options["floodgate-history"] ||= File.join(File.dirname(__FILE__), "floodgate_history.yaml")
+ $options["floodgate-history"] = File.expand_path($options["floodgate-history"])
+
+ LEAGUE.event = ARGV.shift
port = ARGV.shift
- write_pid_file($options["pid-file"]) if ($options["pid-file"])
+ dir = $options["daemon"]
+ dir = File.expand_path(dir) if dir
+ if dir && ! File.exist?(dir)
+ FileUtils.mkdir(dir)
+ end
+ log_file = dir ? File.join(dir, "shogi-server.log") : STDOUT
+ $logger = setup_logger(log_file)
- Thread.abort_on_exception = true
+ LEAGUE.dir = dir || TOP_DIR
- server = TCPserver.open(port)
- log_message("server started")
+ config = {}
+ config[:Port] = port
+ config[:ServerType] = WEBrick::Daemon if $options["daemon"]
+ config[:Logger] = $logger
- while true
- Thread::start(server.accept) do |client|
- client.sync = true
- player = nil
- while (str = client.gets_timeout(Login_Time))
- begin
- $mutex.lock
- Thread::kill(Thread::current) if (! str) # disconnected
- str =~ /([\r\n]*)$/
- eol = $1
- if (good_login?(str))
- player = Player::new(str, client)
- if (LEAGUE.duplicated?(player))
- client.write_safe(sprintf("username %s is already connected%s", player.name, eol))
- client.close
- Thread::kill(Thread::current)
- end
- LEAGUE.add(player)
- break
- else
- client.write_safe("type 'LOGIN name password' or 'LOGIN name password x1'" + eol)
- client.close
- Thread::kill(Thread::current)
- end
- ensure
- $mutex.unlock
- end
- end # login loop
- log_message(sprintf("user %s login", player.name))
- player.run
- LEAGUE.delete(player)
- log_message(sprintf("user %s logout", player.name))
+ fg_thread = nil
+
+ config[:StartCallback] = Proc.new do
+ srand
+ if $options["pid-file"]
+ write_pid_file($options["pid-file"])
+ end
+ setup_watchdog_for_giant_lock
+ LEAGUE.setup_players_database
+ fg_thread = setup_floodgate
+ end
+
+ config[:StopCallback] = Proc.new do
+ if $options["pid-file"]
+ FileUtils.rm($options["pid-file"], :force => true)
end
end
+
+ server = WEBrick::GenericServer.new(config)
+ ["INT", "TERM"].each do |signal|
+ trap(signal) do
+ server.shutdown
+ fg_thread.kill if fg_thread
+ end
+ end
+ trap("HUP") do
+ Dependencies.clear
+ end
+ $stderr.puts("server started as a deamon [Revision: #{ShogiServer::Revision}]") if $options["daemon"]
+ log_message("server started [Revision: #{ShogiServer::Revision}]")
+
+ server.start do |client|
+ # client.sync = true # this is already set in WEBrick
+ client.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
+ # Keepalive time can be set by /proc/sys/net/ipv4/tcp_keepalive_time
+ player, login = login_loop(client) # loop
+ next unless player
+
+ log_message(sprintf("user %s login", player.name))
+ login.process
+ player.setup_logger($options["player-log-dir"]) if $options["player-log-dir"]
+ player.run(login.csa_1st_str) # loop
+ $mutex.lock
+ begin
+ if (player.game)
+ player.game.kill(player)
+ end
+ player.finish # socket has been closed
+ LEAGUE.delete(player)
+ log_message(sprintf("user %s logout", player.name))
+ ensure
+ $mutex.unlock
+ end
+ end
end
+
if ($0 == __FILE__)
- main
+ STDOUT.sync = true
+ STDERR.sync = true
+ TCPSocket.do_not_reverse_lookup = true
+ Thread.abort_on_exception = $DEBUG ? true : false
+
+ begin
+ LEAGUE = ShogiServer::League.new(TOP_DIR)
+ main
+ rescue Exception => ex
+ log_error("main: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}")
+ end
end
OSDN Git Service
