4 # Author:: NABEYA Kenichi, Daigo Moriwaki
5 # Homepage:: http://sourceforge.jp/projects/shogi-server/
8 # Copyright (C) 2004 NABEYA Kenichi (aka nanami@2ch)
9 # Copyright (C) 2007-2008 Daigo Moriwaki (daigo at debian dot org)
11 # This program is free software; you can redistribute it and/or modify
12 # it under the terms of the GNU General Public License as published by
13 # the Free Software Foundation; either version 2 of the License, or
14 # (at your option) any later version.
16 # This program is distributed in the hope that it will be useful,
17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 # GNU General Public License for more details.
21 # You should have received a copy of the GNU General Public License
22 # along with this program; if not, write to the Free Software
23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
32 $:.unshift File.dirname(__FILE__)
33 require 'shogi_server'
34 require 'shogi_server/config'
35 require 'shogi_server/util'
38 #################################################
48 # - nil when a socket is closed
50 def gets_safe(socket, timeout=nil)
51 if r = select([socket], nil, nil, timeout)
52 return r[0].first.gets
56 rescue Exception => ex
57 log_error("gets_safe: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}")
64 shogi-server - server for CSA server protocol
67 shogi-server [OPTIONS] event_name port_number
70 server for CSA server protocol
74 specify filename for logging process ID
76 run as a daemon. Log files will be put in dir.
78 log network messages for each player. Log files
79 will be put in the dir.
82 GPL versoin 2 or later
87 #{ShogiServer::Release}
90 #{ShogiServer::Revision}
116 # Parse command line options. Return a hash containing the option strings
117 # where a key is the option name without the first two slashes. For example,
118 # {"pid-file" => "foo.pid"}.
120 def parse_command_line
122 parser = GetoptLong.new(
123 ["--daemon", GetoptLong::REQUIRED_ARGUMENT],
124 ["--pid-file", GetoptLong::REQUIRED_ARGUMENT],
125 ["--player-log-dir", GetoptLong::REQUIRED_ARGUMENT])
128 parser.each_option do |name, arg|
130 options[name] = arg.dup
134 raise parser.error_message
139 # Check command line options.
140 # If any of them is invalid, exit the process.
142 def check_command_line
143 if (ARGV.length != 2)
148 if $options["daemon"]
149 $options["daemon"] = File.expand_path($options["daemon"], File.dirname(__FILE__))
150 unless is_writable_dir? $options["daemon"]
152 $stderr.puts "Can not create a file in the daemon directory: %s" % [$options["daemon"]]
157 $topdir = $options["daemon"] || File.expand_path(File.dirname(__FILE__))
159 if $options["player-log-dir"]
160 $options["player-log-dir"] = File.expand_path($options["player-log-dir"], $topdir)
161 unless is_writable_dir?($options["player-log-dir"])
163 $stderr.puts "Can not write a file in the player log dir: %s" % [$options["player-log-dir"]]
168 if $options["pid-file"]
169 $options["pid-file"] = File.expand_path($options["pid-file"], $topdir)
170 unless ShogiServer::is_writable_file? $options["pid-file"]
172 $stderr.puts "Can not create the pid file: %s" % [$options["pid-file"]]
177 if $options["floodgate-history"]
178 $stderr.puts "WARNING: --floodgate-history has been deprecated."
179 $options["floodgate-history"] = nil
183 # See if a file can be created in the directory.
184 # Return true if a file is writable in the directory, otherwise false.
186 def is_writable_dir?(dir)
187 unless File.directory? dir
194 temp_file = Tempfile.new("dummy-shogi-server", dir)
203 def write_pid_file(file)
204 open(file, "w") do |fh|
209 def mutex_watchdog(mutex, sec)
217 queue.push(Object.new)
220 log_error("mutex watchdog timeout: %d sec" % [sec])
228 def login_loop(client)
231 while r = select([client], nil, nil, ShogiServer::Login_Time) do
234 break unless str = r[0].first.gets
235 rescue Exception => ex
236 # It is posssible that the socket causes an error (ex. Errno::ECONNRESET)
237 log_error("login_loop: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}")
240 $mutex.lock # guards $league
244 if (ShogiServer::Login::good_login?(str))
245 player = ShogiServer::Player::new(str, client, eol)
246 login = ShogiServer::Login::factory(str, player)
247 if (current_player = $league.find(player.name))
248 if (current_player.password == player.password &&
249 current_player.status != "game")
250 log_message(sprintf("user %s login forcely", player.name))
253 login.incorrect_duplicated_player(str)
261 client.write("LOGIN:incorrect" + eol)
262 client.write("type 'LOGIN name password' or 'LOGIN name password x1'" + eol) if (str.split.length >= 4)
268 return [player, login]
271 def setup_logger(log_file)
272 logger = ShogiServer::Logger.new(log_file, 'daily')
273 logger.formatter = ShogiServer::Formatter.new
274 logger.level = $DEBUG ? Logger::DEBUG : Logger::INFO
275 logger.datetime_format = "%Y-%m-%d %H:%M:%S"
279 def setup_watchdog_for_giant_lock
283 mutex_watchdog($mutex, 10)
287 def floodgate_reload_log(leagues)
288 floodgate = leagues.min {|a,b| a.next_time <=> b.next_time}
289 diff = floodgate.next_time - Time.now
290 log_message("Floodgate reloaded. The next match will start at %s in %d seconds" %
291 [floodgate.next_time, diff])
294 def setup_floodgate(game_names)
295 return Thread.start(game_names) do |game_names|
297 leagues = game_names.collect do |game_name|
298 ShogiServer::League::Floodgate.new($league,
299 {:game_name => game_name})
301 leagues.delete_if do |floodgate|
303 unless floodgate.next_time
304 log_error("Unsupported game name: %s" % floodgate.game_name)
310 log_error("No valid Floodgate game names found")
311 return # exit from this thread
313 floodgate_reload_log(leagues)
317 floodgate = leagues.min {|a,b| a.next_time <=> b.next_time}
318 diff = floodgate.next_time - Time.now
320 floodgate_reload_log(leagues) if $DEBUG
324 next_array = leagues.collect do |floodgate|
325 if (floodgate.next_time - Time.now) > 0
326 [floodgate.game_name, floodgate.next_time]
328 log_message("Starting Floodgate games...: %s" % [floodgate.game_name])
332 [floodgate.game_name, floodgate.next_time] # next_time has been updated
335 $mutex.synchronize do
336 log_message("Reloading source...")
339 # Regenerate floodgate instances after ShogiServer.realod
340 leagues = next_array.collect do |game_name, next_time|
341 floodgate = ShogiServer::League::Floodgate.new($league,
342 {:game_name => game_name,
343 :next_time => next_time})
345 floodgate_reload_log(leagues)
346 rescue Exception => ex
348 log_error("[in Floodgate's thread] #{ex} #{ex.backtrace}")
356 $options = parse_command_line
358 $config = ShogiServer::Config.new $options
360 $league = ShogiServer::League.new($topdir)
362 $league.event = ARGV.shift
365 log_file = $options["daemon"] ? File.join($options["daemon"], "shogi-server.log") : STDOUT
366 $logger = setup_logger(log_file)
368 $league.dir = $topdir
372 config[:ServerType] = WEBrick::Daemon if $options["daemon"]
373 config[:Logger] = $logger
377 config[:StartCallback] = Proc.new do
379 if $options["pid-file"]
380 write_pid_file($options["pid-file"])
382 setup_watchdog_for_giant_lock
383 $league.setup_players_database
384 fg_thread = setup_floodgate(["floodgate-900-0", "floodgate-3600-0"])
387 config[:StopCallback] = Proc.new do
388 if $options["pid-file"]
389 FileUtils.rm($options["pid-file"], :force => true)
394 server = WEBrick::GenericServer.new(config)
395 ["INT", "TERM"].each do |signal|
398 fg_thread.kill if fg_thread
401 unless (RUBY_PLATFORM.downcase =~ /mswin|mingw|cygwin|bccwin/)
406 $stderr.puts("server started as a deamon [Revision: #{ShogiServer::Revision}]") if $options["daemon"]
407 log_message("server started [Revision: #{ShogiServer::Revision}]")
409 server.start do |client|
410 # client.sync = true # this is already set in WEBrick
411 client.setsockopt(Socket::SOL_SOCKET, Socket::SO_KEEPALIVE, true)
412 # Keepalive time can be set by /proc/sys/net/ipv4/tcp_keepalive_time
413 player, login = login_loop(client) # loop
416 log_message(sprintf("user %s login", player.name))
418 player.setup_logger($options["player-log-dir"]) if $options["player-log-dir"]
419 player.run(login.csa_1st_str) # loop
423 player.game.kill(player)
425 player.finish # socket has been closed
426 $league.delete(player)
427 log_message(sprintf("user %s logout", player.name))
438 TCPSocket.do_not_reverse_lookup = true
439 Thread.abort_on_exception = $DEBUG ? true : false
443 rescue Exception => ex
445 log_error("main: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}")
447 $stderr.puts "main: #{ex.class}: #{ex.message}\n\t#{ex.backtrace[0]}"