'}),Ze=l({},Be.DefaultType,{content:"(string|element|function)"}),tn="fade",en="show",nn=".popover-header",on=".popover-body",rn={HIDE:"hide"+ze,HIDDEN:"hidden"+ze,SHOW:"show"+ze,SHOWN:"shown"+ze,INSERTED:"inserted"+ze,CLICK:"click"+ze,FOCUSIN:"focusin"+ze,FOCUSOUT:"focusout"+ze,MOUSEENTER:"mouseenter"+ze,MOUSELEAVE:"mouseleave"+ze},sn=function(t){var e,n;function i(){return t.apply(this,arguments)||this}n=t,(e=i).prototype=Object.create(n.prototype),(e.prototype.constructor=e).__proto__=n;var o=i.prototype;return o.isWithContent=function(){return this.getTitle()||this._getContent()},o.addAttachmentClass=function(t){g(this.getTipElement()).addClass($e+"-"+t)},o.getTipElement=function(){return this.tip=this.tip||g(this.config.template)[0],this.tip},o.setContent=function(){var t=g(this.getTipElement());this.setElementContent(t.find(nn),this.getTitle());var e=this._getContent();"function"==typeof e&&(e=e.call(this.element)),this.setElementContent(t.find(on),e),t.removeClass(tn+" "+en)},o._getContent=function(){return this.element.getAttribute("data-content")||this.config.content},o._cleanTipClass=function(){var t=g(this.getTipElement()),e=t.attr("class").match(Ge);null!==e&&0
+
+<%= @title %>
+
+
+
+
diff --git a/admin-interface/docroot/views/daemons.erb b/admin-interface/docroot/views/daemons.erb
new file mode 100644
index 0000000..0f35c28
--- /dev/null
+++ b/admin-interface/docroot/views/daemons.erb
@@ -0,0 +1,11 @@
+<% @daemons.reject{|_,d| d.nil? }.select{|id, daemon| daemon.server_running? }.sort_by {|_, daemon| daemon.config['server-config-name'] }.each do |id, daemon| %>
+
+<% end %>
diff --git a/admin-interface/docroot/views/homepage.erb b/admin-interface/docroot/views/homepage.erb
new file mode 100644
index 0000000..26ab26f
--- /dev/null
+++ b/admin-interface/docroot/views/homepage.erb
@@ -0,0 +1,5 @@
+
diff --git a/admin-interface/docroot/views/layout-main.erb b/admin-interface/docroot/views/layout-main.erb
new file mode 100644
index 0000000..6017afc
--- /dev/null
+++ b/admin-interface/docroot/views/layout-main.erb
@@ -0,0 +1,170 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/login.erb b/admin-interface/docroot/views/login.erb
new file mode 100644
index 0000000..3c941fc
--- /dev/null
+++ b/admin-interface/docroot/views/login.erb
@@ -0,0 +1,70 @@
+
+
+
+
+
+
+
+
+
+
+ socket
+ put_socket socket.remote_address.inspect_sockaddr, socket
+ return socket
+ end
+ rescue => e
+ log "Failed to open socket to #{server_ip}:#{port} (#{e.class}) #{e.message}", level: :warn
+ raise
+ end
+
+ def authenticate(socket, password)
+ # "If the rcon_password cvar is not set, or if it is set to empty string, all SERVERDATA_AUTH requests will be refused."
+ # Currently we assume there is always a password.
+ log "#{socket.remote_host.ljust(21)} Authenticating RCON connection"
+ packet = build_packet(password, 3, hide_content: true) # Type 3 for auth packet
+ id = send_receive_wrapper socket, packet, sensitive: true
+ raise Errno::ENOTCONN if id.nil?
+ raise "Failed to authenticate. Bad RCON password?" if id == -1
+ log "#{socket.remote_host.ljust(21)} Successfully authenticated."
+ true
+ rescue Errno::ENOTCONN => e
+ log "#{socket.remote_host.ljust(21)} Failed to connect with RCON: (#{e.class}) #{e.message}", :warn
+ delete_socket socket
+ raise
+ end
+
+ def close
+ @sockets.each do |host, socket|
+ log "Closing socket to #{host}"
+ delete_socket socket
+ end
+ log "All sockets closed."
+ end
+
+ def get_socket_for_host(server_ip, port, password)
+ socket = get_socket "#{server_ip}:#{port}"
+ socket = open_socket server_ip, port, password if socket.nil?
+ begin
+ socket.remote_address # see if we're connected
+ rescue Errno::ENOTCONN
+ socket = reopen_socket socket
+ end
+ socket
+ end
+
+ def build_packet(body, type=2, id=1, hide_content: false)
+ # packet = [0xFF, 0xFF, 0xFF, 0xFF, a2s_info_header, content].pack('c5Z*')
+ # packet = [0xFF, 0xFF, 0xFF, 0xFF, 0x55, -1].pack('cccccl_')
+ packet_id = id # This can be any positive integer and can be used to match responses with requests if desired. Need not be unique.
+ packet_type = type
+ packet_body = body
+ packet_empty_string = ''
+ packet = [packet_id, packet_type, packet_body, packet_empty_string] # Skip size for now
+ size = packet.pack(@@pack_string[1..]).length # Get packet size by packing what we have and checking length.
+ packet.unshift(size)
+ # log "Packet: " << packet.inspect
+ packed = packet.pack(@@pack_string)
+ # log "Packet: " << packet.inspect.utf8 unless hide_content
+ # log "Packed: " << packed.inspect.utf8 unless hide_content
+ packed
+ end
+
+ def delete_socket(socket)
+ if socket.nil?
+ log "Tried to delete nil socket!", level: :warn
+ return
+ end
+ log "#{socket.remote_host.ljust(21)} Removing socket"
+ socket.close rescue nil
+ @sockets.delete @sockets.key(socket)
+ end
+
+ def reopen_socket(socket)
+ ip, port, pass = socket.info
+ log "#{socket.remote_host.ljust(21)} Reopening socket", level: :warn unless socket.nil?
+ delete_socket socket unless socket.nil?
+ socket = get_socket_for_host ip, port, pass
+ socket
+ end
+
+ def send_receive_wrapper(socket, packet, timeout=2, retries=1, sensitive: false)
+ log "#{socket.remote_host.ljust(21)} Attempting to gain mutex lock"
+ synchronize(socket.mutex) do
+ log "#{socket.remote_host.ljust(21)} Sucessfully gained mutex lock"
+ log "#{socket.remote_host.ljust(21)} Sending packet: #{sensitive ? 'REDACTED' : packet.inspect.utf8}"
+ response = send_receive(socket, packet, timeout, retries, sensitive: sensitive)
+ response.gsub!("\x00", '') if response.is_a? String # Int for auth
+ log "#{socket.remote_host.ljust(21)} Got response: #{response.inspect.utf8}"
+ log "#{socket.remote_host.ljust(21)} Ending mutex lock"
+ response
+ end
+ end
+
+ def send_receive(socket, packet, timeout=3, retries=1, sensitive: false)
+ begin
+ socket.write packet
+ rescue Errno::EPIPE # broken pipe
+ log "#{socket.remote_host.ljust(21)} Broken pipe. Reopening socket.", level: :warn
+ socket = reopen_socket socket
+ socket.write packet
+ end
+ log "#{socket.remote_host.ljust(21)} Receiving response"
+ header, _ = if IO.select([socket], nil, nil, timeout)
+ socket.recv 12
+ end
+ if header.to_s.empty?
+ # delete_socket socket
+ # raise "#{socket.remote_host.ljust(21)} Nil response for RCON."
+ # Unlike Insurgency: Sandstorm, some commands have a nil/empty header
+ # such as AddBluforBots/AddOpforBots.
+ return nil
+ end
+ log "#{socket.remote_host.ljust(21)} Header: #{header.inspect}"
+ size, id, type = header.unpack('l 0
+ log "#{socket.remote_host.ljust(21)} No response; retrying"
+ return send_receive(socket, packet, timeout, retries - 1)
+ else
+ log "#{socket.remote_host.ljust(21)} No response; no retries remain"
+ # Reopen socket? Could harm server if this happens often.
+ raise NoTCPResponseError.new socket.remote_host
+ end
+ end
+
+ log "#{socket.remote_host.ljust(21)} Read initial response: #{response.inspect}"
+
+ # Ensure no paginated responses waiting
+ log "#{socket.remote_host.ljust(21)} Writing empty packet: #{@@empty_packet.inspect}"
+ socket.write @@empty_packet
+ additional_responses = false
+ loop do
+ additional_header, _ = if IO.select([socket], nil, nil, timeout)
+ socket.recv 12
+ end
+ log "#{socket.remote_host.ljust(21)} Additional header: #{additional_header.inspect}"
+ break if additional_header.nil?
+ size, id, type = additional_header.unpack('l e
+ log "#{socket.remote_host.ljust(21)} Remote host closed/reset the connection. #{e.class}: #{e.message}", level: :warn
+ log "#{socket.remote_host.ljust(21)} Removing closed socket", level: :warn
+ delete_socket socket
+ raise e
+ end
+
+ # User interface
+ def send(server_ip, port, password, command, buffer: nil, outcome_buffer: nil, no_rx: false, ignore_status: false, ignore_message: false, timeout: 2, retries: 1)
+ command.strip!
+ socket = get_socket_for_host server_ip, port, password
+ raise "Couldn't get socket for #{server_ip}:#{port}!" if socket.nil?
+ packet = build_packet command
+ if buffer
+ formatted_output = "#{datetime} | RCON #{server_ip}:#{port} (TX >>) #{command}"
+ buffer[:filters].each { |filter| filter.call(formatted_output) }
+ buffer.synchronize { buffer.push formatted_output }
+ end
+ response = send_receive_wrapper socket, packet, timeout, retries
+ if command == 'exit'
+ # The connection will be broken. Start fresh next time.
+ log "#{socket.remote_host.ljust(21)} Exit command detected. Deleting socket."
+ delete_socket socket
+ end
+ if buffer && !no_rx
+ buffer.synchronize do
+ formatted_output = "#{datetime} | RCON #{server_ip}:#{port} (RX <<) #{response}"
+ buffer[:filters].each { |filter| filter.call(formatted_output) }
+ buffer.push formatted_output
+ end
+ end
+ if outcome_buffer
+ outcome_buffer.synchronize do
+ outcome_buffer[:status] = true unless ignore_status
+ outcome_buffer[:message] = "RCON response received from [#{server_ip}:#{port}]." unless ignore_message
+ end
+ end
+ response
+ rescue => e
+ log "Error while sending RCON", e
+ if outcome_buffer
+ outcome_buffer.synchronize do
+ buffer[:status] = false unless ignore_status
+ buffer[:message] = e.message unless ignore_message
+ end
+ end
+ raise e
+ end
+
+ def get_players_and_bots(server_ip, port, password, buffer: nil, ignore_status: false, ignore_message: false, timeout: 2, retries: 1)
+ resp = send(server_ip, port, password, 'status', buffer: buffer, ignore_status: ignore_status, ignore_message: ignore_message, timeout: timeout, retries: retries)
+ raise "#{server_ip}:#{port} Response was nil!" if resp.nil?
+ raise "#{server_ip}:#{port} Map is changing! (Invalid gamestate)" if resp.strip == "Invalid gamestate"
+ _, info, players_and_bots, _ = resp.split("\n========================================\n")
+ info = info.split("\n")
+ return [], [] if players_and_bots.nil?
+ players_and_bots = players_and_bots.split("\n")
+ players_and_bots.shift # Remove header row
+ players = []
+ bots = []
+ players_and_bots.each do |pb|
+ id, name_and_unique_id = pb.split("\t")
+ parts = name_and_unique_id.split(/\s+/)
+ unique_id = parts.pop
+ name = parts.join
+ is_invalid_entry = id.to_s.empty? || unique_id.to_s.empty? || name.to_s.empty?
+ throw "Invalid player or bot: #{pb}" if is_invalid_entry
+ pb = {
+ 'id' => id,
+ 'name' => name,
+ 'unique_id' => unique_id
+ }
+ if (unique_id === "BOT")
+ bots.push(pb)
+ else
+ players.push(pb)
+ end
+ end
+ return players, bots
+ end
+end
+
+if __FILE__ == $0
+ require 'pp'
+ require 'pry'
+ server_ip = ARGV[0]
+ port = ARGV[1]
+ password = ARGV[2]
+ client = RconClient.new
+ players, bots = client.get_players_and_bots(server_ip, port, password)
+ binding.pry
+end
diff --git a/admin-interface/lib/self-updater.rb b/admin-interface/lib/self-updater.rb
new file mode 100644
index 0000000..54cd8ed
--- /dev/null
+++ b/admin-interface/lib/self-updater.rb
@@ -0,0 +1,56 @@
+require 'fileutils'
+require 'json'
+require 'net/http'
+require 'open-uri'
+require 'pathname'
+require 'zip'
+
+class SelfUpdater
+ UPDATE_ZIP = File.join WRAPPER_ROOT, 'update.zip'
+ IGNORE_FILES = [
+ 'config/config.toml'
+ ]
+
+ def self.download(url, path)
+ case io = URI.open(url)
+ when StringIO
+ File.open(path, 'w') { |f| f.write(io) }
+ when Tempfile
+ io.close
+ FileUtils.mv(io.path, path)
+ else
+ log "Unhandled IO type: #{io}", :warn
+ end
+ end
+
+ def self.update_to_latest
+ log "Determining latest release"
+ releases = JSON.parse Net::HTTP.get(URI('https://api.github.com/repos/Joe-Klauza/ohd-admin-wrapper/releases'))
+ latest = releases.first
+ version = latest['name']
+ zip_download_url = latest['zipball_url']
+ log "Downloading #{version}: #{zip_download_url}"
+
+ Dir.chdir WRAPPER_ROOT do
+ SelfUpdater.download(zip_download_url, UPDATE_ZIP)
+ log "Downloaded #{version} to update.zip"
+ log "Extracting update.zip"
+ Zip::File.open(UPDATE_ZIP) do |zip_file|
+ zip_file.each do |f|
+ filename = File.join Pathname(f.name).each_filename.to_a[1..] # Ignore project dir
+ filepath = File.join(WRAPPER_ROOT, filename)
+ next if IGNORE_FILES.include?(filename)
+ zip_file.extract(f, filepath) { true } # Overwrite
+ end
+ end
+ end
+ log "Extracted update.zip"
+ log "Deleting update.zip"
+ FileUtils.rm(UPDATE_ZIP)
+ log "Deleted update.zip"
+ version
+ rescue => e
+ log "Failed to update to latest", e
+ raise "Failed to update. (#{e.class}): #{e.message}"
+ end
+end
diff --git a/admin-interface/lib/server-monitor.rb b/admin-interface/lib/server-monitor.rb
new file mode 100644
index 0000000..c41ea6b
--- /dev/null
+++ b/admin-interface/lib/server-monitor.rb
@@ -0,0 +1,386 @@
+require 'benchmark'
+# require 'geocoder'
+require_relative 'logger'
+require_relative 'rcon-client'
+require_relative 'server-query'
+require_relative 'steam-api-client'
+
+# Geocoder.configure(ip_lookup: :ipapi_com) # :ipinfo_io (default, 1,000/day) :ipapi_com (150/min) #:geoip2 (no actual lookup; not reliable)
+
+class ServerMonitor
+ attr_reader :info
+ attr_reader :name
+ attr_reader :ip
+ attr_reader :query_port
+ attr_reader :rcon_port
+ attr_reader :rcon_pass
+ attr_reader :rcon_buffer
+
+ def initialize(ip, query_port, rcon_port, rcon_pass, interval: 15.0, delay: 5, rcon_fail_limit: 60, query_fail_limit: 30, name: '', rcon_buffer: nil, daemon_handle: nil, welcome_message_delay: 20)
+ @stop = false
+ @ip = ip
+ @query_port = query_port
+ @rcon_port = rcon_port
+ @rcon_pass = rcon_pass
+ @interval = interval
+ @rcon_fail_limit = rcon_fail_limit
+ @query_fail_limit = query_fail_limit
+ @name = name
+ @rcon_buffer = rcon_buffer
+ @rcon_buffer[:persistent] = true
+ @daemon_handle = daemon_handle
+ @steam_api_client = SteamApiClient.new(@daemon_handle.steam_api_key) unless @daemon_handle.nil? || @daemon_handle.steam_api_key.to_s.empty?
+
+ @rcon_client = RconClient.new
+ @info = {
+ a2s_connection_problem: true,
+ rcon_connection_problem: true,
+ server_down: true,
+ rcon_players: [],
+ rcon_bots: nil,
+ rcon_last_success: Time.now.to_i,
+ a2s_info: nil,
+ a2s_player: nil,
+ a2s_rules: nil,
+ a2s_last_success: Time.now.to_i
+ }
+ @welcome_message_delay = welcome_message_delay
+ @host = "#{ip}:[#{query_port},#{rcon_port}]"
+ @thread = Thread.new do
+ if delay > 0
+ log "Waiting #{delay} seconds to monitor"
+ sleep delay
+ end
+ @thread = monitor unless @stop
+ end
+ log "Initialized monitor"
+ end
+
+ def [](thing)
+ @info[thing]
+ end
+
+ def log(message, exception=nil, level: nil)
+ super("#{@host} Monitor | #{message}", exception, level: level) # Call the log function created by logger
+ end
+
+ def get_uptime(original_start, now=Time.now.to_i)
+ seconds_lapsed = now - original_start
+ uptime = ''
+ hours = seconds_lapsed / (60 * 60)
+ minutes = (seconds_lapsed / 60) % 60
+ seconds = seconds_lapsed % 60
+ uptime << ("%d hours " % hours) if hours > 0
+ uptime << ("%d minutes " % minutes) if hours > 0 || minutes > 0
+ uptime << ("%d seconds" % seconds)
+ uptime
+ end
+
+ def steam_duration_to_seconds(duration)
+ duration.split(':').map { |a| a.to_i }.inject(0) do |total, current|
+ total * 60 + current # Convert hours to minutes, add minutes, convert minutes to seconds, add seconds
+ end
+ end
+
+ def seconds_to_steam_duration(seconds)
+ hours = seconds / (60 * 60)
+ minutes = (seconds / 60) % 60
+ seconds = seconds % 60
+ "%02d:%02d:%02d" % [hours, minutes, seconds]
+ end
+
+ def time_ago(timestamp)
+ delta = Time.now.utc.to_i - timestamp.to_i
+ value, unit = case delta
+ when -1..45 then [nil, 'just now']
+ when 46..60 then [nil, 'about a minute ago']
+ when 61..3599 then [(delta / 60).round, 'minute']
+ when 3600..86399 then [(delta / 3600.0), 'hour']
+ when 86400..31535999 then [(delta / 86400.0), 'day']
+ else [(delta / 31536000.0), 'year']
+ end
+ if value.nil?
+ unit
+ else
+ if unit == 'minute'
+ "%d minute#{'s' if value > 1} ago" % value
+ else
+ "%.1f #{unit}#{'s' if ("%.1f" % value).to_f >= 1.1} ago" % value
+ end
+ end
+ end
+
+ def process_rcon_players(rcon_players, prev_rcon_players)
+ return if [rcon_players, prev_rcon_players].any?(&:nil?)
+ players_gone = prev_rcon_players.reject { |prev| rcon_players.map { |current| current['unique_id'] }.include?(prev['unique_id']) }
+ players_joined = rcon_players.reject { |current| prev_rcon_players.map { |prev| prev['unique_id'] }.include?(current['unique_id']) }
+
+ now = Time.now.to_i
+ # Look up players we haven't in the past 24 hours
+ if @daemon_handle && (@steam_api_client || !@daemon_handle.steam_api_key.empty?)
+ @steam_api_client = SteamApiClient.new(@daemon_handle.steam_api_key) if @steam_api_client.nil?
+ ids_to_query = rcon_players.select { |p| now - $config_handler.players.dig(p['unique_id'], 'last_steam_lookup').to_i > 60 * 60 * 24 }.map { |p| p['unique_id'] }
+ steam_users = @steam_api_client.get_info(ids_to_query)
+ steam_bans = @steam_api_client.get_bans(ids_to_query)
+ steam_users.each do |user|
+ next unless user['steamid']
+ # Add their Steam data to the RCON player to merge into all players later
+ matching_player = rcon_players.select { |player| player['unique_id'] == user['steamid'] }.first
+ matching_player['steam_info'] = {
+ 'steamid' => user['steamid'],
+ 'name' => user['personaname'] || '',
+ 'created' => user['timecreated'].to_s || '',
+ 'avatar' => user['avatar'] || '',
+ 'avatar_medium' => user['avatarmedium'] || '',
+ 'avatar_full' => user['avatarfull'] || '',
+ 'persona_state' => user['personastate'] || '',
+ 'primary_clan_id' => user['primaryclanid'] || '',
+ 'community_visibility_state' => user['communityvisibilitystate'] || '',
+ 'last_logoff' => user['lastlogoff'] || '',
+ 'comment_permission' => user['commentpermission'] || '',
+ 'profile_url' => user['profileurl'] || '',
+ 'persona_state_flags' => user['personastateflags'] || '',
+ 'game_server_ip' => user['gameserverip'] || '',
+ 'game_server_unique_id' => user['gameserversteamid'] || '',
+ 'game_extra_info' => user['gameextrainfo'] || '',
+ 'game_id' => user['gameid'] || ''
+ }
+ ban_info = steam_bans.select { |u| u['SteamId'] == user['steamid'] }.first
+ matching_player['steam_info']['bans'] = {
+ 'community' => ban_info['CommunityBanned'], # Bool
+ 'vac' => ban_info['VACBanned'], # Bool
+ 'vac_number' => ban_info['NumberOfVACBans'].to_s,
+ 'days_since_last' => ban_info['DaysSinceLastBan'].to_s,
+ 'game_number' => ban_info['NumberOfGameBans'].to_s,
+ 'economy' => ban_info['EconomyBan'] # String
+ }
+ matching_player['last_steam_lookup'] = now
+ end
+ end
+
+ # Welcome joined players
+ players_joined.each do |player|
+ if @daemon_handle
+ is_admin = @daemon_handle.is_ohd_admin?(player['unique_id'])
+ message_option = is_admin ? 'admin_join_message' : 'join_message'
+ unless @daemon_handle.config[message_option].empty?
+ Thread.new(message_option) do |message_option|
+ message = @daemon_handle.config[message_option]
+ message = message.gsub('${player_name}', player.dig('steam_info', 'name') || player['name']).gsub('${player_id}', player['unique_id'] || 'NULL')
+ sleep @welcome_message_delay # Allow time to select a loadout and see the chat
+ @rcon_client.send(@ip, @rcon_port, @rcon_pass, "say #{message}")
+ end
+ end
+ log "Player joined: #{player['name']} (#{player['unique_id']})#{' (admin)' if is_admin}", level: :info
+ else
+ log "Player joined: #{player['name']} (#{player['unique_id']})", level: :info
+ end
+ end
+
+ # players_needing_ip_query = rcon_players.select do |p|
+ # now - $config_handler.players.dig(p['unique_id'], 'last_ip_lookup', p['ip']).to_i > 60 * 60 * 24
+ # end
+ # players_needing_ip_query.each do |player|
+ # if LOCAL_IP_PREFIXES.any?{ |prefix| player['ip'].start_with?(prefix) }
+ # # The user is likely within the same network as the server, so use the external IP
+ # if @daemon_handle
+ # # Local server external IP
+ # ip = EXTERNAL_IP
+ # else
+ # # Remote monitor external IP
+ # ip = @ip.dup
+ # end
+ # next if ip.empty?
+ # else
+ # ip = player['ip']
+ # end
+ # begin
+ # log "Looking up IP #{ip} for player #{player['name']} (#{player['unique_id']})"
+ # ip_info = Geocoder.search(ip).first || Geocoder.search(ip).first # Often fails on the first try with "Geocoding API's response was not valid JSON"
+ # if ip_info.nil? || !ip_info.respond_to?(:data)
+ # log "Couldn't find IP info for #{ip} for player #{player['name']} (#{player['unique_id']})"
+ # next
+ # end
+ # ip_info = ip_info.data
+ # log "Got info about IP #{ip} for player #{player['name']} (#{player['unique_id']}): #{ip_info}"
+ # $config_handler.players[player['unique_id']]['last_ip_lookup'] = {} if $config_handler.players[player['unique_id']]['last_ip_lookup'].nil?
+ # $config_handler.players[player['unique_id']]['ip_info'] = {} if $config_handler.players[player['unique_id']]['ip_info'].nil?
+ # $config_handler.players[player['unique_id']]['last_ip_lookup'][player['ip']] = now
+ # $config_handler.players[player['unique_id']]['ip_info'][player['ip']] = ip_info
+ # rescue => e
+ # log "Failed to look up IP #{player['ip']} (#{player['unique_id']})"
+ # end
+ # end
+
+ # Merge the current data
+ rcon_players.map do |player|
+ if $config_handler.players[player['unique_id']].nil?
+ $config_handler.players[player['unique_id']] = player
+ else
+ $config_handler.players[player['unique_id']].merge!(player) # Merge the current data
+ player.merge!($config_handler.players[player['unique_id']]) # Merge back the Steam info in case we don't have it
+ end
+ player
+ end
+
+ players_gone.each do |player|
+ if @daemon_handle
+ is_admin = @daemon_handle.is_ohd_admin?(player['unique_id'])
+ log "Player left: #{player['name']} (#{player['unique_id']})#{' (admin)' if is_admin}", level: :info
+ unless @daemon_handle.config['leave_message'].empty?
+ Thread.new do
+ message = @daemon_handle.config['leave_message'].gsub('${player_name}', player.dig('steam_info', 'name') || player['name']).gsub('${player_id}', player['unique_id'] || 'NULL')
+ @rcon_client.send(@ip, @rcon_port, @rcon_pass, "say #{message}")
+ end
+ end
+ end
+ saved_player = $config_handler.players[player['unique_id']]
+ score = player['score'].to_i
+ saved_player['total_score'] = saved_player['total_score'].to_i + score
+ saved_player['high_score'] = score if saved_player['high_score'].to_i <= score
+ saved_player['last_seen'] = now
+ saved_player['last_server'] = @info.dig(:a2s_info, 'name') || (@daemon_handle && @daemon_handle.name)
+ begin
+ matching_a2s_player = @info[:a2s_player].select { |p| p['name'] == player['name'] }.first
+ if matching_a2s_player.nil? && @info[:a2s_player].map { |p| p['name'].empty? }.size == 1
+ # Sometimes A2S_PLAYER has one or more blank player names for a while (bug)
+ # If there's only one, we can assume that this is the player we're unable to match by name
+ matching_a2s_player = @info[:a2s_player].map { |p| p['name'].empty? }.first
+ end
+ duration_steam = matching_a2s_player['duration'] rescue nil
+ raise "Couldn't match by player name (#{player['name']}) to get duration. A2S_PLAYER: #{@info[:a2s_player]}" if duration_steam.nil?
+ duration_seconds = steam_duration_to_seconds(duration_steam) + (@interval / 2.0).floor # Assume they left ~halfway between the last check and now
+ saved_player['last_duration'] = seconds_to_steam_duration(duration_seconds)
+ saved_player['longest_duration'] = seconds_to_steam_duration(duration_seconds) if saved_player['longest_duration'].to_i <= duration_seconds
+ saved_player['total_duration'] = saved_player['total_duration'].to_i + duration_seconds
+ rescue => e
+ log "Failed to calculate duration_seconds for saved player info", e
+ end
+ saved_player['known_ips'] = saved_player['known_ips'].to_a.push(player['ip']).uniq
+ end
+ rescue => e
+ log "Error occurred during RCON player processing", e
+ end
+
+ def do_rcon_query
+ rcon_players, rcon_bots = @rcon_client.get_players_and_bots(@ip, @rcon_port, @rcon_pass, buffer: @rcon_buffer, ignore_status: true, ignore_message: true)
+ process_rcon_players(rcon_players, @info[:rcon_players])
+ log "Got RCON players: #{rcon_players}"
+ @info.merge!({
+ rcon_connection_problem: false,
+ server_down: false,
+ rcon_players: rcon_players,
+ rcon_bots: rcon_bots,
+ rcon_last_success: Time.now.to_i
+ })
+ rescue => e
+ if e.message.include?('Invalid gamestate')
+ log "Skipping RCON status parsing (map is changing)"
+ return
+ end
+ log "RCON query failed", e
+ @info[:rcon_connection_problem] = true
+ rcon_fail_time = Time.now.to_i - @info[:rcon_last_success]
+ log "Time since last RCON success: #{rcon_fail_time.to_s << 's'}", level: rcon_fail_time > @rcon_fail_limit ? :error : :warn
+ if rcon_fail_time > @rcon_fail_limit
+ @info[:server_down] = true
+ if @daemon_handle && @daemon_handle.frozen_config['hang_recovery'].to_s.casecmp('true').zero?
+ Thread.new do
+ log "Restarting server due to repeated RCON failure", level: :warn
+ response = @daemon_handle.do_restart_server
+ log "Daemon response: #{response}"
+ end
+ end
+ end
+ end
+
+ def do_server_query
+ a2s_info = ServerQuery::a2s_info(@ip, @query_port)
+ log "Got A2S_INFO: #{a2s_info}"
+ a2s_player = ServerQuery::a2s_player(@ip, @query_port)
+ log "Got A2S_PLAYER: #{a2s_player}"
+ a2s_rules = ServerQuery::a2s_rules(@ip, @query_port)
+ log "Got A2S_RULES: #{a2s_rules}"
+
+ prev_map = @info.dig(:a2s_info, 'map')
+ if a2s_info['map'] != prev_map
+ log "Map changed: #{prev_map} => #{a2s_info['map']}", level: :info
+ end
+
+ # Sometimes the server can be in a zombie state where server query succeeds
+ # but nothing else works (including RCON); i.e. we shouldn't set server_down: false
+ # based on a successful server query response if RCON is reporting as working (but
+ # failure is more often indicative of an issue than RCON, which is more buggy)
+ @info.merge!({
+ a2s_connection_problem: false,
+ a2s_info: a2s_info,
+ a2s_player: a2s_player,
+ a2s_rules: a2s_rules,
+ a2s_last_success: Time.now.to_i
+ })
+ @info[:server_down] = @info[:rcon_connection_problem]
+ rescue => e
+ log "Server query failed", e
+ @info[:a2s_connection_problem] = true
+ query_fail_time = Time.now.to_i - @info[:a2s_last_success]
+ log "Time since last server query success: #{query_fail_time.to_s << 's'}", level: query_fail_time > @query_fail_limit ? :error : :warn
+ if query_fail_time > @query_fail_limit
+ @info[:server_down] = true
+ if @daemon_handle && @daemon_handle.frozen_config['query_recovery'].to_s.casecmp('true').zero?
+ Thread.new do
+ log "Restarting server due to repeated Server Query failure", level: :warn
+ response = @daemon_handle.do_restart_server
+ log "Daemon response: #{response}"
+ end
+ end
+ end
+ end
+
+ def stop
+ @stop = true
+ @rcon_buffer[:status] = true
+ @rcon_buffer[:message] = "#{@host} Monitor stopped"
+ @thread.kill if @thread.respond_to?('kill')
+ end
+
+ def all_green?
+ # RCON and Query successfully reached
+ !@info[:a2s_connection_problem] && !@info[:rcon_connection_problem]
+ end
+
+ def monitor
+ return nil if @stop
+ Thread.new do
+ @rcon_buffer.reset
+ original_start = Time.now.to_i
+ loop do
+ lapsed = Benchmark.realtime do
+ begin
+ start = Time.now.to_i
+ log "Retrieving RCON players"
+ time_taken = Benchmark.realtime { do_rcon_query }
+ log "Took #{"%.3f" % time_taken}s (Retrieving RCON players)"
+ log "Retrieving Server Query info, players, and rules"
+ time_taken = Benchmark.realtime { do_server_query }
+ log "Took #{"%.3f" % time_taken}s (Retrieving Server Query info, players, and rules)"
+ rescue => e
+ log "error during monitoring!", e
+ break
+ end
+ end
+ log "Uptime: #{get_uptime(original_start)}"
+ sleep_seconds = [@interval - lapsed, 0.0].max # Ensure we don't try to sleep with a negative value
+ log "Server monitoring took #{"%.1f" % lapsed}s. Sleeping #{"%.1f" % sleep_seconds}s."
+ sleep sleep_seconds
+ if @stop
+ log "Stopping monitor"
+ next
+ end
+ end
+ rescue => e
+ log "Error while monitoring", e
+ ensure
+ log "Monitoring stopped."
+ end
+ end
+end
diff --git a/admin-interface/lib/server-query.rb b/admin-interface/lib/server-query.rb
new file mode 100644
index 0000000..2081862
--- /dev/null
+++ b/admin-interface/lib/server-query.rb
@@ -0,0 +1,134 @@
+#!/usr/bin/env ruby
+
+require 'socket'
+require_relative 'logger'
+require_relative '../ext/string'
+
+# https://developer.valvesoftware.com/wiki/Server_queries
+# https://apidock.com/ruby/Array/pack
+# https://apidock.com/ruby/String/unpack
+
+class NoUDPResponseError < StandardError
+ def initialize(host=nil)
+ super "Could not read a UDP response from server#{host ? " #{host}" : '!'}"
+ end
+end
+
+class ServerQuery
+ def self.send_recv_udp(packet, server_ip, server_port, socket_opts: 0, timeout: 2, retries: 1)
+ s = UDPSocket.new
+ s.send(packet, socket_opts, server_ip, server_port)
+ resp, from = if IO.select([s], nil, nil, timeout)
+ s.recvfrom(60000)
+ end
+ if resp.nil?
+ if retries > 0
+ s.close
+ return send_recv_udp(packet, server_ip, server_port, socket_opts: socket_opts, timeout: timeout, retries: retries - 1)
+ else
+ raise NoUDPResponseError.new "#{server_ip}:#{server_port}"
+ end
+ end
+ return resp, from
+ ensure
+ s.close
+ end
+
+ def self.a2s_info(server_ip, server_port = 27015)
+ # https://developer.valvesoftware.com/wiki/Server_queries#A2S_INFO
+ a2s_info_header = 0x54
+ content = "Source Engine Query\0"
+ packet = [0xFF, 0xFF, 0xFF, 0xFF, a2s_info_header, content].pack('c5Z*')
+ resp, _ = send_recv_udp(packet, server_ip, server_port)
+ log "[#{server_ip}:#{server_port} A2S_INFO response: #{resp.inspect}"
+ # "\xFF\xFF\xFF\xFFAH\x85]\xBF"
+ info = {}
+ log "[#{server_ip}:#{server_port} A2S_INFO parsed: #{info}"
+ info
+ rescue NoUDPResponseError => e
+ log "[#{server_ip}:#{server_port}] Couldn't get UDP response (#{e.class} raised)", level: :warn
+ raise
+ rescue => e
+ log "[#{server_ip}:#{server_port}] Rescued error while querying info", e, level: :warn
+ raise
+ end
+
+ def self.a2s_player(server_ip, server_port = 27015)
+ # https://developer.valvesoftware.com/wiki/Server_queries#A2S_PLAYER
+ players = []
+ packet = [0xFF, 0xFF, 0xFF, 0xFF, 0x55, -1].pack('cccccl_')
+ resp, _ = send_recv_udp(packet, server_ip, server_port)
+ if resp.unpack('xxxxcl').first == 65 # Challenge detected ('A' response); request again with given long
+ packet = [0xFF, 0xFF, 0xFF, 0xFF, 0x55, resp.unpack('xxxxcl').last].pack('cccccl')
+ resp, _ = send_recv_udp(packet, server_ip, server_port)
+ end
+ log "[#{server_ip}:#{server_port} A2S_PLAYER response: #{resp.inspect}"
+ resp = resp[6..] # Remove padding, header, player count
+ until resp.empty?
+ begin
+ pack_string = 'xZ*lf'
+ player_data = resp.unpack(pack_string)
+ resp = resp[player_data.pack(pack_string).length..] # Remove the read player info so we can iterate through the response (arbitrary length)
+ player_info = {
+ 'name' => player_data[0].to_s.utf8.strip, # Can be empty
+ 'score' => player_data[1],
+ 'duration' => Time.at(player_data[2]).utc.strftime("%H:%M:%S"),
+ }
+ players << player_info
+ rescue StandardError => e
+ log "[#{server_ip}:#{server_port}] Rescued error while iterating players", e
+ break
+ end
+ end
+ log "[#{server_ip}:#{server_port} A2S_PLAYER parsed: #{players}"
+ players
+ rescue NoUDPResponseError => e
+ log "[#{server_ip}:#{server_port}] Couldn't get UDP response (#{e.class} raised)", level: :warn
+ raise
+ rescue => e
+ log "[#{server_ip}:#{server_port}] Rescued error while querying players", e, level: :warn
+ raise
+ end
+
+ def self.a2s_rules(server_ip, server_port = 27015)
+ # https://developer.valvesoftware.com/wiki/Server_queries#A2S_RULES
+ packet = [0xFF, 0xFF, 0xFF, 0xFF, 0x56, -1].pack('cccccl_')
+ resp, _ = send_recv_udp(packet, server_ip, server_port)
+ if resp.unpack('xxxxcl').first == 65 # Challenge detected ('A' response); request again with given long
+ packet = [0xFF, 0xFF, 0xFF, 0xFF, 0x56, resp.unpack('xxxxcl').last].pack('cccccl')
+ resp, _ = send_recv_udp(packet, server_ip, server_port)
+ end
+ log "[#{server_ip}:#{server_port} A2S_RULES response: #{resp.inspect}"
+ data = resp.unpack('xxxxxxxA*').first
+ rules = data.split("\x00").each_slice(2).to_h
+ raise "Unable to parse rules from response: #{resp.inspect}" unless rules.length > 1
+ # {"Coop_b"=>"true", # For mutators this can incorrectly report as false after a player joins
+ # "GameMode_s"=>"Checkpoint", # For mutators this can incorrectly report as regular Checkpoint until a player joins
+ # "MatchServer_b"=>"false",
+ # "Mutated_b"=>"false", # For mutators this can incorrectly report as false until a player joins
+ # "OfficialRuleset_b"=>"false",
+ # "PlrC_i"=>"0",
+ # "PlrM_i"=>"8",
+ # "Pwd_b"=>"false",
+ # "RankedServer_b"=>"false",
+ # "S"=>"523",
+ # "Versus_b"=>"false"}
+ log "[#{server_ip}:#{server_port} A2S_RULES parsed: #{rules}"
+ rules
+ rescue NoUDPResponseError => e
+ log "[#{server_ip}:#{server_port}] Couldn't get UDP response (#{e.class} raised)", level: :warn
+ raise
+ rescue => e
+ log "[#{server_ip}:#{server_port}] Rescued error while querying rules: #{resp.inspect}", e, level: :warn
+ raise
+ end
+end
+
+if __FILE__ == $0
+ require 'pp'
+ # Parse ip/port from args. Allows 'ip:port' or 'ip' 'port'
+ ip, port = ARGV[1].nil? ? ARGV[0].split(':') : [ARGV[0], ARGV[1]]
+ pp ServerQuery::a2s_info(ip, port)
+ pp ServerQuery::a2s_rules(ip, port)
+ pp ServerQuery::a2s_player(ip, port)
+end
diff --git a/admin-interface/lib/server-updater.rb b/admin-interface/lib/server-updater.rb
new file mode 100644
index 0000000..94522b1
--- /dev/null
+++ b/admin-interface/lib/server-updater.rb
@@ -0,0 +1,164 @@
+#!/usr/bin/env ruby
+
+require 'fileutils'
+require 'open3'
+require_relative 'subprocess'
+
+class ServerUpdater
+ attr_reader :update_available
+ attr_reader :thread
+
+ def initialize(server_root_dir, steamcmd_path, steam_appinfovdf_path)
+ @server_root_dir = server_root_dir
+ @app_manifest = File.join server_root_dir, 'steamapps', 'appmanifest_950900.acf'
+ # raise Errno::ENOENT, @app_manifest unless File.exist? @app_manifest
+ @steamcmd_path = steamcmd_path
+ # raise Errno::ENOENT, @steamcmd_path unless File.exist? @steamcmd_path
+ @steam_appinfovdf_path = steam_appinfovdf_path
+
+ @update_available = false
+ @installed_build_id = nil
+ @available_build_id = nil
+ Thread.new { update_available? } # Populate update availability (still pretty slow sometimes)
+ end
+
+ def back_up_app_cache
+ FileUtils.mv(@steam_appinfovdf_path, @steam_appinfovdf_path + '.bak') if File.exist?(@steam_appinfovdf_path)
+ rescue => e
+ log "Error backing up app cache", e
+ end
+
+ def restore_app_cache
+ FileUtils.cp(@steam_appinfovdf_path + '.bak', @steam_appinfovdf_path) if File.exist?(@steam_appinfovdf_path + '.bak') && !File.exist?(@steam_appinfovdf_path)
+ rescue => e
+ log "Error restoring app cache", e
+ end
+
+ def get_update_info
+ return [@update_available, @installed_build_id, @available_build_id]
+ end
+
+ def get_latest_build_id
+ back_up_app_cache
+ stdout, stderr, status = Open3.capture3(@steamcmd_path,
+ '+login anonymous',
+ '+app_info_print 950900',
+ '+exit'
+ )
+ # Convert SteamCMD output to JSON for better traversal
+ output = stdout[stdout.index('"950900"')..stdout.index(/^\}$/)]
+ # Strip irrelevant characters, add colon separators for JSON
+ lines = output.gsub("\t", '').gsub('""', '":"').gsub(/"(\s*)\{/, '":\1{').split("\n")
+ # Add comma unless next line is an object start/end
+ converted = lines.each_with_index.map do |line, index|
+ # Don't add a comma to any line containing { or when {/} is on the following line
+ (line[/[{]/] || lines[index + 1].nil? || lines[index + 1][/[{}]/]) ? line : "#{line},"
+ end.join("\n")
+ json = JSON.parse "{#{converted}}"
+ build_id = json.dig('950900', 'depots', 'branches', 'public', 'buildid')
+ if build_id.nil?
+ log "Unable to get build ID from converted JSON. See log file for debug information.", level: :warn
+ log "SteamCMD STDOUT: #{stdout}\nSteamCMD STDERR: #{stderr}"
+ log "Converted JSON: #{json}"
+ return nil
+ end
+ log "Got latest build ID: #{build_id}"
+ @available_build_id = build_id
+ rescue => e
+ log "Rescued error while getting latest build ID. See log file for SteamCMD output", e
+ log "SteamCMD STDOUT: #{stdout}\nSteamCMD STDERR: #{stderr}"
+ nil
+ ensure
+ restore_app_cache
+ end
+
+ def get_installed_build_id
+ open(@app_manifest) do |file|
+ build_id = file.grep(/buildid/).first
+ build_id = build_id[/\d+/] unless build_id.nil?
+ log "Got installed build ID: #{build_id}"
+ @installed_build_id = build_id
+ end
+ rescue => e
+ log "Rescued error while getting installed build ID", e
+ nil
+ end
+
+ # Returns true if update is required, false otherwise (including when checks failed)
+ def update_available?(installed_build_id=get_installed_build_id, latest_build_id=get_latest_build_id)
+ if installed_build_id.nil? || latest_build_id.nil?
+ available = false
+ else
+ available = installed_build_id == latest_build_id ? false : true
+ end
+ log (available ? 'Update is available!' : 'Server is up-to-date!')
+ @update_available = available
+ end
+
+ def run_steamcmd(command, buffer: nil, ignore_status: false, ignore_message: false)
+ command.unshift @steamcmd_path
+ thread = Thread.new do
+ # Start a thread so we can change the environment...
+ log "Running SteamCMD command: #{command}", level: :info
+ SubprocessRunner.run(
+ command,
+ buffer: buffer,
+ ignore_status: ignore_status,
+ ignore_message: ignore_message
+ )
+ end
+ thread.join
+ output = if buffer.nil?
+ thread.value
+ else
+ buffer[:data].join("\n")
+ end
+ update_text = nil
+ updated = ['fully installed.', 'already up to date.'].any? do |it|
+ if output.include? it
+ update_text = it
+ update_text.prepend 'update ' if update_text.include? 'fully'
+ true
+ else
+ false
+ end
+ end
+ response = "Server #{updated ? update_text : 'failed to update!' }"
+ log response, level: updated ? :info : :error
+ if buffer
+ buffer.synchronize do
+ buffer[:status] = updated
+ buffer[:message] = response
+ end
+ end
+ [updated, response]
+ end
+
+ # Returns true if an update was performed, false if already up-to-date
+ def update_server(buffer=nil, validate: nil, ignore_status: true, ignore_message: true)
+ log 'Updating server', level: :info
+ command = [
+ '+login anonymous',
+ "+force_install_dir \"#{@server_root_dir}\"",
+ "+app_update 950900#{' validate' if validate}",
+ '+exit'
+ ]
+ updated, response = run_steamcmd(command, buffer: buffer, ignore_status: ignore_status, ignore_message: ignore_message)
+ update_available?
+ [updated, response]
+ end
+
+ def monitor_update(minutes_between_checks: 3)
+ while true
+ begin
+ if update_available?
+ log 'A new server update is available!', level: :info
+ yield if block_given?
+ end
+ rescue => e
+ log "Rescued error during update check", e
+ end
+ sleep 60 * minutes_between_checks
+ end
+ end
+end
diff --git a/admin-interface/lib/steam-api-client.rb b/admin-interface/lib/steam-api-client.rb
new file mode 100644
index 0000000..eda5e49
--- /dev/null
+++ b/admin-interface/lib/steam-api-client.rb
@@ -0,0 +1,48 @@
+# encoding: UTF-8
+
+require 'json'
+require 'net/http'
+
+class SteamApiClient
+
+ def initialize(api_key, timeout: 4)
+ @api_key = api_key
+ @timeout = timeout
+ end
+
+ def call_api(uri, timeout: @timeout)
+ uri = URI(uri)
+ Net::HTTP.start(uri.host, uri.port, read_timeout: timeout, open_timeout: timeout) do |http|
+ request = Net::HTTP::Get.new uri
+ http.request request
+ end.body
+ end
+
+ def get(path, ids)
+ if ids.empty?
+ log "Skipping Steam check (no IDs provided)"
+ return {}
+ end
+ log "Getting Steam #{path.include?('GetPlayerBans') ? 'bans' : 'info'} for IDs: #{ids.join(', ')}"
+ raw_response = call_api "http://api.steampowered.com/#{path}?key=#{@api_key}&steamids=#{ids.join(',')}"
+ JSON.parse raw_response
+ rescue => e
+ if e.is_a?(JSON::ParserError) && raw_response.include?('429 Too Many Requests')
+ log "Failed to get ban information; too many Steam requests. (IDs: #{ids})", level: :warn
+ else
+ log "Failed to get ban information for ID(s) #{ids}: #{raw_response}", e
+ end
+ {}
+ end
+
+ def get_info(ids)
+ path = "ISteamUser/GetPlayerSummaries/v0002/"
+ get(path, ids).dig('response', 'players') || []
+ end
+
+ def get_bans(ids)
+ path = "ISteamUser/GetPlayerBans/v1/"
+ get(path, ids).dig('players') || []
+ end
+
+end
diff --git a/admin-interface/lib/subprocess.rb b/admin-interface/lib/subprocess.rb
new file mode 100644
index 0000000..ae09ca4
--- /dev/null
+++ b/admin-interface/lib/subprocess.rb
@@ -0,0 +1,92 @@
+# PTY doesn't work on windows, but on Linux it gets us concurrent progress output from steamcmd since steamcmd detects shell interactivity for this
+# https://github.com/ValveSoftware/Source-1-Games/issues/1684
+PTY_AVAILABLE = WINDOWS ? false : (require 'pty')
+require_relative 'logger'
+
+class SubprocessRunner
+ def self.run_via_pty(command)
+ read, write, pid = PTY.spawn(*command)
+ return [pid, read, nil, write]
+ end
+
+ def self.run_via_process(command, out, err, input=nil)
+ opts = {
+ out: out,
+ err: err,
+ in: input
+ }.reject { |_, v| v.nil? }
+ pid = Process.spawn(*command, opts)
+ end
+
+ def self.run(command, buffer: nil, ignore_status: false, ignore_message: false, out: IO.pipe, err: IO.pipe, input: nil, shell: [], pty: PTY_AVAILABLE, no_prefix: false, formatter: nil)
+ origin = File.basename command.first
+ origin = "#{File.basename shell.first}(#{origin})" unless shell.empty?
+ command = command.split(' ') unless command.is_a? Array
+
+ log "Running subprocess via (#{pty ? 'PTY' : 'Process'}): #{command}"
+ command = shell.concat command
+
+ pid, stdout, stderr, stdin = if pty
+ [out, err, input].each { |pipe| pipe.close if pipe.respond_to?('close') }
+ run_via_pty(command)
+ else
+ pid = run_via_process(command, out.last, err.last, input)
+ [pid, out.first, err.first]
+ end
+ yield(pid) if block_given?
+ captured_output = []
+ [stdout, stderr].reject(&:nil?).each do |stream|
+ Thread.new do
+ loop do
+ output = stream.gets.utf8
+ break if output.nil?
+ formatted_output = if formatter
+ formatter.call(output, origin)
+ else
+ no_prefix ? output.chomp : "#{datetime} | #{origin} | #{output.chomp}"
+ end
+ if buffer.nil?
+ captured_output.push output
+ else
+ buffer[:filters].each { |filter| filter.call(formatted_output) }
+ buffer[:mutex].synchronize do # Synchronize with the reading thread to avoid mismatched indices when truncating, etc.
+ buffer.push formatted_output
+ end
+ end
+ log formatted_output # Ensure this is after filters to avoid NWI's lack of color code resets
+ rescue Errno::EIO, IOError, EOFError => e
+ break
+ end
+ end
+ end
+ pid, process_status = Process.wait2(pid)
+
+ log "Checking exit status"
+ status, message = if process_status.exitstatus
+ [process_status.exitstatus.zero? ? true : false, "PID (#{pid}) #{process_status.exitstatus.zero? ? 'completed successfully' : 'exited unsuccessfully'} (#{process_status.exitstatus})."]
+ else
+ if process_status.signaled?
+ signal = process_status.termsig || process_status.stopsig
+ [true, "PID (#{pid}) stopped with signal #{Signal.list.key signal} (#{signal})."]
+ else
+ [false, "PID (#{pid}) stopped for an unknown reason."]
+ end
+ end
+
+ log "Exit status/message: #{status} - #{message}"
+
+ if buffer.nil?
+ captured_output.join("\n")
+ else
+ buffer.synchronize do
+ buffer[:status] = status unless ignore_status
+ buffer[:message] = message unless ignore_message
+ end
+ end
+ rescue => e
+ log "Error running subprocess: #{command}", e
+ raise e
+ ensure
+ [out.first, out.last, err.first, err.last, input, stdout, stderr, stdin].each { |pipe| pipe.close if pipe.respond_to?('close') }
+ end
+end
diff --git a/admin-interface/lib/user.rb b/admin-interface/lib/user.rb
new file mode 100644
index 0000000..5eaaa91
--- /dev/null
+++ b/admin-interface/lib/user.rb
@@ -0,0 +1,61 @@
+require 'bcrypt'
+require 'sysrandom'
+
+class User
+ attr_accessor :name
+ attr_reader :role
+ attr_reader :id
+ attr_reader :initial_password
+
+ ROLES = {
+ host: 4,
+ admin: 3,
+ moderator: 2,
+ user: 1
+ }
+
+ def initialize(name, role, password: nil, initial_password: nil, id: nil)
+ @name = name # Name will also be our GUID, since having duplicate user names is confusing without emails
+ @role = role
+ @initial_password = initial_password
+ @id = id || Sysrandom.uuid
+ if password
+ # User info is being read from file
+ @password = BCrypt::Password.new(password).to_s
+ else
+ generate_initial_password
+ end
+ end
+
+ def first_login?
+ return false if @initial_password.nil?
+ password_matches?(@initial_password)
+ end
+
+ def generate_initial_password
+ @initial_password = ConfigHandler.generate_password
+ @password = BCrypt::Password.create(@initial_password).to_s
+ end
+
+ def password # So we can user.password == 'password'
+ BCrypt::Password.new(@password)
+ end
+
+ def password=(new_password)
+ @initial_password = nil
+ @password = BCrypt::Password.create(new_password).to_s
+ end
+
+ def password_matches?(password)
+ BCrypt::Password.new(@password) == password
+ end
+
+ def role=(new_role)
+ new_role = new_role.to_sym
+ if ROLES.include? new_role
+ @role = new_role
+ else
+ raise "Failed to assign role: #{new_role} is not a valid role!"
+ end
+ end
+end
diff --git a/admin-interface/lib/webapp.rb b/admin-interface/lib/webapp.rb
new file mode 100644
index 0000000..ebc8c91
--- /dev/null
+++ b/admin-interface/lib/webapp.rb
@@ -0,0 +1,1381 @@
+WINDOWS = Gem.win_platform?
+
+require 'pry'
+require 'cgi'
+require 'net/http'
+require 'openssl'
+require 'sinatra/base'
+require 'socket'
+require 'sys/proctable'
+require 'sysrandom'
+require 'toml-rb'
+require 'webrick'
+require 'webrick/https'
+require_relative '../ext/thread'
+require_relative 'logger'
+require_relative 'buffer'
+require_relative 'certificate-generator'
+require_relative 'daemon'
+require_relative 'config-handler'
+require_relative 'user'
+require_relative 'rcon-client'
+require_relative 'self-updater'
+require_relative 'server-updater'
+
+Encoding.default_external = "UTF-8"
+LOGGER.threshold :info # Changes STDOUT only
+log "Loading config"
+$config_handler = ConfigHandler.new
+
+class AdminWrapperSite < Sinatra::Base
+ def self.set_up
+ log "Initializing webserver"
+ @@config = load_webapp_config
+ @@daemons = {}
+ @@daemons_mutex = Mutex.new
+ @@monitors = {}
+ @@buffers = {}
+ @@buffer_mutex = Mutex.new
+ @@wrapper_user_log = create_buffer.last
+ @@wrapper_connection_log = create_buffer.last
+ @@rcon_client = RconClient.new
+ @@server_updater = ServerUpdater.new(SERVER_ROOT, STEAMCMD_EXE, STEAM_APPINFO_VDF)
+ @@update_thread = nil
+ @@prereqs_complete = false
+ @@lan_access_bind_ip = Socket.ip_address_list.detect{ |intf| intf.ipv4_private? }.ip_address rescue '?'
+ @@loaded_wrapper_version = @@config['wrapper_version']
+ trap 'EXIT' do
+ Thread.new do
+ if @@daemons.any?
+ log "Stopping daemons", level: :info
+ @@daemons.each { |_, daemon| daemon.implode }
+ end
+ end.join
+ end
+ handle_arguments unless ARGV.empty?
+ end
+
+ def self.load_webapp_config
+ FileUtils.cp(WEBAPP_CONFIG_SAMPLE, WEBAPP_CONFIG) unless File.exist?(WEBAPP_CONFIG)
+ log "Loading wrapper config from #{File.basename WEBAPP_CONFIG}"
+ config = TomlRB.load_file WEBAPP_CONFIG
+ config['automatic_updates'] = true if config['automatic_updates'].nil?
+ config['steam_api_key'] = '' if config['steam_api_key'].nil?
+ config
+ rescue => e
+ log "Couldn't load config from #{config_file}", e
+ raise
+ end
+
+ def self.save_webapp_config(config, config_file = WEBAPP_CONFIG)
+ log "Saving wrapper config"
+ File.write(config_file, TomlRB.dump(config))
+ end
+
+ def self.handle_arguments
+ args = ARGV.dup
+ while true
+ arg = args.shift
+ break if arg.nil?
+ case arg
+ when '--start', '-s'
+ config_name_or_id = args.shift
+ break if config_name_or_id.nil?
+ @@daemons_mutex.synchronize do
+ Thread.new(config_name_or_id) do |config_name_or_id|
+ config = $config_handler.server_configs.values.select{|conf| conf['id'] == config_name_or_id || conf['server-config-name'] == config_name_or_id }.first.dup rescue nil
+ sleep 2 # Allow webrick SSL cert to be logged before starting servers
+ if config
+ log "Starting daemon for #{config_name_or_id}", level: :info
+ init_daemon(config, start: true)
+ else
+ log "Unknown server config name/ID: #{config_name_or_id}", level: :warn
+ end
+ end
+ end
+ when '--log-level', '-l'
+ val = args.shift.to_s.upcase.to_sym
+ if Logger::Severity.constants.include? val
+ log "Setting STDOUT logger threshold to #{val}", level: :info
+ LOGGER.threshold val
+ else
+ log "Unknown log level: #{val}. Try one of these: #{Logger::Severity.constants.map(&:to_s).join(', ')}", level: :warn
+ end
+ else
+ log "Unknown argument: #{arg}", level: :warn
+ end
+ end
+ end
+
+ def self.init_daemon(config, start: false)
+ id = config['id']
+ @@daemons_mutex.synchronize do
+ if @@daemons[id]
+ log "Daemon with config ID #{id} already exists", level: :info
+ else
+ log "Initializing daemon with config ID #{id}", level: :info
+ @@daemons[id] = ServerDaemon.new(
+ config,
+ @@daemons,
+ @@daemons_mutex,
+ @@rcon_client,
+ create_buffer.last, # Server log
+ create_buffer.last, # RCON
+ create_buffer.last, # Chat
+ steam_api_key: @@config['steam_api_key']
+ )
+ end
+ end
+ @@daemons[id].do_start_server if start
+ @@daemons[id]
+ end
+
+ def self.create_buffer(uuid=nil)
+ @@buffer_mutex.synchronize do
+ return [uuid, @@buffers[uuid]] unless @@buffers[uuid].nil?
+ uuid = Sysrandom.uuid if uuid.nil?
+ buffer = Buffer.new(uuid)
+ @@buffers[uuid] = buffer
+ [uuid, buffer]
+ end
+ end
+
+ # def which(command)
+ # exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
+ # ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
+ # exts.each do |ext|
+ # exe = File.join(path, "#{command}#{ext}")
+ # return exe if File.executable?(exe) && !File.directory?(exe)
+ # end
+ # end
+ # nil
+ # end
+
+ def steamcmd_installed?
+ # on_path = WINDOWS ? which("steamcmd.exe") : (which("steamcmd") || which("steamcmd.sh"))
+ # return on_path unless on_path.nil?
+ STEAMCMD_EXE if File.exist? STEAMCMD_EXE
+ end
+
+ def game_server_installed?
+ BINARY if File.exist? BINARY
+ end
+
+ def get_server_status(config_id)
+ daemon = @@daemons[config_id]
+ return 'OFF' if daemon.nil?
+ daemon.server_running? ? 'ON' : 'OFF'
+ end
+
+ def get_update_status
+ return 'OFF' if @@update_thread.nil?
+ @@update_thread.alive? ? 'ON' : 'OFF'
+ end
+
+ def check_prereqs
+ steamcmd_path = steamcmd_installed?
+ game_server_path = game_server_installed?
+ @@prereqs_complete = steamcmd_path && game_server_path
+ @@update_thread = get_game_update_thread if @@prereqs_complete && @@update_thread.nil? && @@config['automatic_updates']
+ return [steamcmd_path, game_server_path]
+ end
+
+ def get_update_info
+ @@server_updater.get_update_info
+ end
+
+ def install_server(buffer=nil, validate: nil)
+ log "Installing server..."
+ success, message = @@server_updater.update_server(buffer, validate: validate)
+ message
+ end
+
+ def update_server(buffer=nil, validate: nil)
+ was_running = []
+ if WINDOWS
+ @@daemons.each do |id, daemon|
+ if daemon.server_running?
+ daemon.do_pre_update_warning
+ daemon.do_stop_server
+ was_running << id
+ end
+ end
+ end
+ success, message = @@server_updater.update_server(buffer, validate: validate)
+ @update_pending = !success
+ @@daemons.each do |id, daemon|
+ if WINDOWS
+ next unless was_running.include?(id)
+ daemon.do_start_server
+ else
+ next unless daemon.server_running?
+ daemon.do_pre_update_warning
+ daemon.do_restart_server
+ end
+ end
+ message
+ end
+
+ def run_steamcmd(command, buffer: nil)
+ @@server_updater.run_steamcmd(command, buffer: buffer)
+ end
+
+ def get_game_update_thread
+ Thread.new do
+ begin
+ @@server_updater.monitor_update do
+ update_server if @@config['automatic_updates']
+ end
+ rescue => e
+ log "Game update thread failed", e
+ end
+ end
+ end
+
+ def start_updates
+ if @@update_thread
+ log "Restarting updates", level: :info
+ @@update_thread.kill
+ @@update_thread = get_game_update_thread
+ "Automated updates restarted"
+ else
+ log "Starting updates", level: :info
+ @@update_thread = get_game_update_thread
+ "Automated updates started"
+ end
+ end
+
+ def stop_updates
+ if @@update_thread
+ log "Stopping updates", level: :info
+ @@update_thread.kill
+ @@update_thread = nil
+ "Automated updates stopped"
+ else
+ "Automated updates not running"
+ end
+ end
+
+ def zip_files(zip_path, glob)
+ log "Zipping logs from #{glob} into #{zip_path.sub(USER_HOME, '~')}"
+ files_to_zip = Dir.glob(glob).map { |f| File.expand_path f }
+ Zip::File.open(zip_path, Zip::File::CREATE) do |zipfile|
+ files_to_zip.each do |f|
+ log "Zipping #{f.sub(USER_HOME, '~')}"
+ zipfile.add(File.basename(f), f)
+ end
+ end
+ end
+
+ def get_rcon_info(id: nil)
+ if @@daemons[id]
+ ['127.0.0.1', @@daemons[id].active_rcon_port, @@daemons[id].active_rcon_pass]
+ elsif @@monitors[id]
+ [@@monitors[id].ip, @@monitors[id].rcon_port, @@monitors[id].rcon_pass]
+ else
+ nil
+ end
+ end
+
+ set_up
+
+ configure do
+ working_directory = File.expand_path File.join File.dirname(__FILE__), '..', 'docroot'
+ log "Setting webserver root dir to: #{working_directory.sub(USER_HOME, '~')}", level: :info
+ set :root, working_directory
+ enable :sessions, :dump_errors #, :raise_errors, :show_exceptions #, :logging
+ set :raise_errors, false
+ set :show_exceptions, false
+ set :session_secret, @@config['admin_interface_session_secret'] unless @@config['admin_interface_session_secret'].empty?
+ end
+
+ # Add a condition that we can use for various routes
+ register do
+ def auth(role)
+ condition do
+ begin
+ unless has_role?(role)
+ message = "IP: #{request.ip} | User unauthorized: #{"'#{@user.name}' [#{@user.role}] (#{session[:user_id]})" rescue '[logged out]'} requesting #{role}-privileged content: #{request.path_info}"
+ @@wrapper_user_log << "#{datetime} | #{message}"
+ log message, level: :warn
+ status 403
+ redirect "/login#{CGI.escape(request.path_info) unless request.path_info.casecmp('/').zero?}" if request.request_method == 'GET'
+ error = "403 Unauthorized - You don't have permission to do this."
+ redirect "/error?status=403&message=#{CGI.escape error}"
+ false
+ end
+ rescue => e
+ log "#{request.ip} | Error checking auth status for role: #{role}", e
+ status 500
+ redirect "/login/#{CGI.escape(request.path_info) unless request.path_info.casecmp('/').zero?}" if request.request_method == 'GET'
+ error = "500 Error - Unknown role: #{role}"
+ redirect "/error?status=500&message=#{CGI.escape error}"
+ false
+ end
+ end
+ end
+ end
+
+ # Add helper functions for user authentication, etc.
+ helpers do
+ def logged_in?
+ !@user.nil?
+ end
+
+ def has_role?(role)
+ raise "Invalid role: #{role}" unless User::ROLES.keys.include?(role)
+ logged_in? && User::ROLES[@user.role] >= User::ROLES[role]
+ end
+
+ def is_host?
+ has_role?(:host)
+ end
+
+ def is_admin?
+ has_role?(:admin)
+ end
+
+ def is_moderator?
+ has_role?(:moderator)
+ end
+
+ def is_user?
+ has_role?(:user)
+ end
+
+ def get_active_config
+ # log "Getting active server config using ID #{session[:active_server_config]}", level: :fatal
+ if $config_handler.server_configs.empty?
+ $config_handler.add_server_config(get_default_config)
+ end
+ # binding.pry
+ active_config = $config_handler.server_configs.select { |id, _| id == session[:active_server_config] }.to_h.values.first || $config_handler.server_configs.values.last
+ # log "Got active server config with name #{active_config['server-config-name']} (ID: #{active_config['id']})", level: :fatal
+ active_config
+ end
+
+ def get_config(config_id)
+ halt 400, 'Invalid config ID format' unless config_id.to_s.is_uuid?
+ config = $config_handler.server_configs[config_id]
+ halt 400, "Unable to find config for ID #{config_id}" if config.nil?
+ config
+ end
+
+ def redirect_to_config(config_id, path)
+ if config_id && config_id.is_uuid? && $config_handler.server_configs.keys.include?(config_id)
+ session[:active_server_config] = config_id
+ else
+ config_id = session[:active_server_config]
+ if config_id && config_id.is_uuid? && $config_handler.server_configs.keys.include?(config_id)
+ redirect "#{path}?config-id=#{session[:active_server_config]}"
+ else
+ session[:active_server_config] = nil
+ end
+ end
+ end
+ end
+
+ error do
+ e = env['sinatra.error']
+ log "#{request.ip} | Error occurred in route #{request.path}", e
+ if is_host?
+ "An error occurred (#{CGI.escapeHTML(e.class.to_s)}). Only hosts can see the following message and backtrace: #{CGI.escapeHTML(e.message)} | Backtrace:
#{e.backtrace.map{|l| CGI.escapeHTML(l.gsub(USER_HOME, '~')) }.join('
')}" + else + CGI.escapeHTML "An error occurred (#{e.class.to_s}). Please view the logs or contact the maintainer." + end + end + + before do + env['rack.logger'] = LOGGER + env['rack.errors'] = LOGGER + @user = $config_handler.users[session[:user_id]] if session[:user_id] + check_prereqs unless @@prereqs_complete + request.body.rewind + # body = request.body.read + # request.body.rewind # Be kind + timestamp = datetime + message = "IP: #{request.ip} | #{"User: #{@user.name} | " if @user}Request: #{request.request_method}#{' ' << request.script_name unless request.script_name.strip.empty? } #{request.path_info}#{'?' << request.query_string unless request.query_string.strip.empty?}" # #{(' | Body: ' << body) unless body.strip.empty?}" + log message + @@wrapper_user_log.push "#{timestamp} | #{message}" if request.request_method != 'GET' + @@wrapper_connection_log.push "#{timestamp} | #{message} | Agent: #{request.user_agent}" + nil + end + + # after do + # log "Responding: #{response.body}" + # end + + %i(get post put).each do |method| + send method, '/error' do + status params['status'] + params['message'] + end + end + + get '/confirm' do + @yes = CGI.unescape params[:yes] + @no = CGI.unescape params[:no] + @title = CGI.unescape params[:title] + @body = CGI.unescape params[:body] + @input_label = CGI.unescape params[:input_label] + erb :confirm + end + + get %r{/login(/)?(.*)} do + @destination = '/' + params['captures'].last.sub('login', '') + @destination = nil if @destination == '/' + erb :login + end + + post '/login' do + data = Oj.load(request.body.read) + request.body.rewind + user = data['user'] + password = data['pass'] + destination = data['destination'] + if user && password + known_user = $config_handler.users.values.select { |u| u.name == user }.first + if known_user && known_user.password == password # BCrypt::Password == string comparison + message = "IP: #{request.ip} | #{known_user.name} [#{known_user.role}] logged in (#{known_user.id})" + log message, level: :info + @@wrapper_user_log << "#{datetime} | #{message}" + session[:user_id] = known_user.id + return known_user.first_login? ? "/change-password#{destination}" : destination + end + end + status 401 + "Failed to log in." + end + + get '/logout' do + user = $config_handler.users[session[:user_id]] + halt 400, 'Not logged in' if user.nil? + message = "IP: #{request.ip} | #{user.name} [#{user.role}] logged out (#{user.id})" + log message, level: :info + @@wrapper_user_log << "#{datetime} | #{message}" + session[:user_id] = nil + redirect '/login' + end + + get '/status', auth: :user do + @daemons = @@daemons + @monitors = @@monitors + erb :'server-status', layout: :'layout-main' do + erb :'server-list' + end + end + + get '/server-list', auth: :user do + @daemons = @@daemons + @monitors = @@monitors + erb :'server-list' + end + + get '/server-daemons', auth: :admin do + @daemons = @@daemons + erb :'server-daemons', layout: :'layout-main' + end + + get '/change-password(/:destination)?', auth: :user do + @destination = "/#{params['destination']}".sub('/change-password', '') + @first_login = !@user.initial_password.nil? + erb :'change-password', layout: :'layout-main' + end + + post '/change-password', auth: :user do + data = Oj.load(request.body.read) + request.body.rewind + password = data['pass'] + destination = "#{data['destination']}" + if password.strip.empty? || @user.password_matches?(password) + status 400 + return "Invalid password. The password must be new and not blank!" + end + @user.password = password + $config_handler.write_user_config + destination + end + + get '/wrapper-log', auth: :host do + @user_log_id = @@wrapper_user_log[:uuid] + @conn_log_id = @@wrapper_connection_log[:uuid] + erb :'wrapper-log', layout: :'layout-main' + end + + get '/wrapper-config', auth: :host do + @config = @@config + @lan_access_bind_ip = @@lan_access_bind_ip + @wrapper_version = @@loaded_wrapper_version + erb :'wrapper-config', layout: :'layout-main' + end + + put '/wrapper-config/:action', auth: :host do + wrapper_config = @@config + variable = params['variable'] + value = params['value'] + value = value == 'true' if ['true', 'false'].include?(value) + if params['action'] == 'set' + begin + if wrapper_config.keys.include?(variable) + sensitive = ['admin_interface_session_secret', 'steam_api_key'].include? variable + old = sensitive ? '[REDACTED]' : wrapper_config[variable] + wrapper_config[variable] = value + value = sensitive ? '[REDACTED]' : wrapper_config[variable] + self.class.save_webapp_config(wrapper_config) + @@daemons.values.each { |d| d.steam_api_key = wrapper_config[variable] } if variable == 'steam_api_key' + "Changed #{variable}: #{old} => #{value}" + else + status 400 + msg + end + rescue => e + msg = "Failed to set #{variable + ' -> ' + variable == 'admin_interface_session_secret' ? '[REDACTED]' : value}" + log msg, e + status 400 + "#{msg} | #{e.class}: #{e.message}" + end + elsif params['action'] == 'get' + value = wrapper_config[variable] + if value.nil? + status 404 + "Could not find value for #{variable}" + else + value.to_s + end + else + status 400 + 'Unknown action' + end + end + + get '/wrapper-users-list', auth: :host do + @users = $config_handler.users + erb :'wrapper-users-list' + end + + get '/wrapper-users', auth: :host do + @users = $config_handler.users + erb :'wrapper-users', layout: :'layout-main' do + erb :'wrapper-users-list' + end + end + + post '/wrapper-users/:action', auth: :host do + action = params['action'] + data = Oj.load(request.body.read) + request.body.rewind + id = data['id'] + name = data['name'] + role = data['role'].to_s.downcase.to_sym + + case action + when 'create' + if name.to_s.empty? || role.to_s.empty? + status 400 + 'Missing name/role' + elsif $config_handler.users.values.select { |u| u.name == name }.size > 0 + status 400 + 'A user with that name already exists' + else + user = User.new(name, role) + $config_handler.users[user.id] = user + $config_handler.write_user_config + 'User created' + end + when 'delete' + if id.to_s.empty? + status 400 + 'Missing ID' + elsif $config_handler.users[id].nil? + status 400 + 'No such user' + elsif $config_handler.users[id].role == :host && $config_handler.users.values.select { |u| u.role == :host }.size == 1 + status 400 + 'Cannot delete last Host' + else + $config_handler.users.delete id + $config_handler.write_user_config + 'User deleted' + end + when 'save' + if name.to_s.empty? || role.to_s.empty? + status 400 + 'Missing name/role' + elsif $config_handler.users[id].nil? + status 400 + 'No such user' + elsif !User::ROLES.keys.include?(role) + status 400 + 'No such role' + elsif role != :host && $config_handler.users[id].role == :host && $config_handler.users.values.select { |u| u.role == :host }.size == 1 + status 400 + 'Cannot change role of last Host' + else + user = $config_handler.users[id] + if user.name == name && user.role == role + status 400 + 'Nothing was changed' + else + user.name = name + user.role = role + $config_handler.write_user_config + 'User saved' + end + end + else + status 400 + 'Unknown action' + end + end + + get '/pry', auth: :host do + @buffers = @@buffers + prev_threshold = LOGGER.threshold + Thread.new do + LOGGER.threshold :fatal # Turn down STDOUT logging + `stty echo` unless WINDOWS || !$stdout.isatty # Ensure we have echoing enabled; something from logging turns it off... + + # Stuff + + binding.pry + ensure + `stty echo` unless WINDOWS || !$stdout.isatty + LOGGER.threshold prev_threshold + end.join + nil + rescue Interrupt + end + + get '/threads/:id', auth: :user do + daemon = @@daemons[params[:id]] + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + monitor = daemon.monitor rescue nil + @info = monitor.info rescue nil + erb :threads + end + + get '/players/:id', auth: :user do + daemon = @@daemons[params[:id]] + @monitor = daemon.monitor rescue @@monitors[params[:id]] + @info = @monitor.info rescue nil + erb :'players' + end + + get '/update-info', auth: :user do + @update_available, @old_build, @new_build = get_update_info + erb(:'update-info') + end + + post '/restart-wrapper', auth: :host do + @@daemons.each { |_, daemon| daemon.do_stop_server } + ENV['HOME'] = USER_HOME + # Exit with a particular code, to be handled by the start script + at_exit do + exit 2 + end + log "Restart invoked", level: :info + Thread.new do + log "Replacing the current process...", level: :info + sleep 0.2 + exit + end + '' + end + + post '/update-wrapper', auth: :host do + version = SelfUpdater.update_to_latest + @@config['wrapper_version'] = version + self.class.save_webapp_config @@config + "Updated to #{version}! Restart the wrapper to apply." + rescue => e + status 500 + e.message + end + + get '/', auth: :user do + @steamcmd_path, @game_server_path = check_prereqs + if is_admin? + redirect '/setup' unless @steamcmd_path && @game_server_path + redirect '/control' + else + redirect '/status' + end + end + + get '/about', auth: :user do + erb(:about, layout: :'layout-main') + end + + get '/setup', auth: :admin do + @steamcmd_path, @game_server_path = check_prereqs + @automatic_updates = @@config['automatic_updates'] + erb(:'server-setup', layout: :'layout-main') + end + + get '/maplist', auth: :admin do + @config = get_config(params['config-id']) + config_id = @config['id'] # Needed in ERB binding + erb(:maplist) + end + + get '/config', auth: :admin do + redirect '/setup' unless @@prereqs_complete + config_id = params['config-id'] + redirect_to_config(config_id, '/config') + active_config = get_active_config + # We _should_ always have a config, as config_handler should ensure that at least the Default config exists at all times + @config = $config_handler.server_configs[active_config['id']] || $config_handler.server_configs.values.last + unless @config + status 400 + return "Unable to find config" + end + config_id = @config['id'] + $config_handler.init_server_config_files config_id + @game_ini = File.read ERB.new(CONFIG_FILES[:game_ini][:local_erb]).result(binding) + @engine_ini = File.read ERB.new(CONFIG_FILES[:engine_ini][:local_erb]).result(binding) + @game_user_settings_ini = File.read ERB.new(CONFIG_FILES[:game_user_settings_ini][:local_erb]).result(binding) + @admins_cfg = File.read ERB.new(CONFIG_FILES[:admins_cfg][:local_erb]).result(binding) + @mapcycle_cfg = File.read ERB.new(CONFIG_FILES[:mapcycle_cfg][:local_erb]).result(binding) + @bans_cfg = File.read ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + @motd_txt = File.read ERB.new(CONFIG_FILES[:motd_txt][:local_erb]).result(binding) + @notes_txt = File.read ERB.new(CONFIG_FILES[:notes_txt][:local_erb]).result(binding) + erb(:'server-config', layout: :'layout-main') + end + + get '/control', auth: :admin do + redirect '/setup' unless @@prereqs_complete + config_id = params['config-id'] + redirect_to_config(config_id, '/control') + @config = get_active_config + @config_id = @config['id'] + daemon = @@daemons[@config_id] + if daemon + @config = daemon.frozen_config + end + @game_port = daemon.server_running? ? daemon.active_game_port : @config['server_game_port'] rescue @config['server_game_port'] + @rcon_port = daemon.server_running? ? daemon.active_rcon_port : @config['server_rcon_port'] rescue @config['server_rcon_port'] + @query_port = daemon.server_running? ? daemon.active_query_port : @config['server_query_port'] rescue @config['server_query_port'] + @server_status = get_server_status(@config_id) + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + @monitor = daemon.monitor rescue nil + @info = @monitor.info rescue nil + erb(:'server-control', layout: :'layout-main') + end + + get '/tools/:resource', auth: :admin do + @config = get_active_config + resource = params['resource'] + case resource + when 'rcon' + erb(:'rcon-tool', layout: :'layout-main') + when 'steamcmd' + redirect '/setup' unless steamcmd_installed? + @server_root_dir = SERVER_ROOT + erb(:'steamcmd-tool', layout: :'layout-main') + when 'monitor' + config = $config_handler.monitor_configs.first + @name, @config = config ? config : [@config['server-config-name'], @config] + erb(:'monitor-tool', layout: :'layout-main') + else + status 500 + "Unknown resource: #{resource}" + end + end + + post '/tools/:resource', auth: :admin do + resource = params['resource'] + body = request.body.read + options = Oj.load body, symbol_keys: true + request.body.rewind + case resource + when 'steamcmd' + uuid, buffer = self.class.create_buffer + Thread.new { run_steamcmd(options[:command].split("\n"), buffer: buffer) } + uuid + else + status 400 + "Unknown resource: #{resource}" + end + end + + get '/server-control-status' do + @config = get_active_config + @id = @config['id'] + daemon = @@daemons[@id] + if daemon + @config = daemon.frozen_config + log "/server-control-status working with daemon ID #{daemon.config['id']}" + else + log "/server-control-status Failed to get running daemon for config with id #{@id}" + end + @game_port = daemon.server_running? ? daemon.active_game_port : @config['server_game_port'] rescue @config['server_game_port'] + @rcon_port = daemon.server_running? ? daemon.active_rcon_port : @config['server_rcon_port'] rescue @config['server_rcon_port'] + @query_port = daemon.server_running? ? daemon.active_query_port : @config['server_query_port'] rescue @config['server_query_port'] + @server_status = get_server_status(@id) + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + @monitor = daemon.monitor rescue nil + @info = @monitor.info rescue nil + erb :'server-control-status' + end + + get '/server-status/:id', auth: :user do + id = params[:id] + get_server_status(id) + end + + get '/buffer/:uuid(/:bookmark)?', auth: :admin do # Sensitive information could be contained in these buffers (passwords, paths); admins only. + uuid = params['uuid'] + unless @@buffers.keys.include? uuid + status 400 + return "Unknown UUID: #{uuid}" + end + buffer = @@buffers[uuid] + @bookmark_uuid = params['bookmark'] + limit = buffer[:limit] + + buffer[:mutex].synchronize do # Synchronize with the writing thread to avoid mismatched indices when truncating, etc. + buffer[:iterator] = buffer[:iterator].nil? ? 1 : buffer[:iterator] + 1 + # log "Buffer [#{uuid}][#{buffer[:iterator]}] arrived with bookmark: [#{@bookmark_uuid}] -> #{buffer[:bookmarks][@bookmark_uuid]}" + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Bookmarks: #{buffer[:bookmarks]}" + if buffer[:bookmarks].length > 100 + log "Buffer has many bookmarks! Are that many clients connected? #{buffer[:uuid]} (#{buffer[:bookmarks].length} bookmarks)", level: :warn + log "Purging first 20 bookmarks", level: :warn + 20.times { buffer[:bookmarks].shift } + end + unless buffer[:bookmarks].keys.include?(@bookmark_uuid) || @bookmark_uuid.nil? + status 400 + return "Failed to tail log (Unknown Bookmark UUID: #{@bookmark_uuid})." + end + if buffer[:bookmarks][@bookmark_uuid] == -1 # Client read everything; ready for outcome + obj = { + :status => buffer[:status].dup, + :message => buffer[:message].dup + } + log "Responding status/message: #{obj[:status]} - #{obj[:message]}" + @@buffer_mutex.synchronize do + buffer[:bookmarks].delete @bookmark_uuid + @@buffers[uuid][:persistent] ? @@buffers[uuid].reset : @@buffers.delete(uuid) + end + return Oj.dump(obj, mode: :compat) + end + + start_index = buffer[:bookmarks][@bookmark_uuid] || 0 + start_index = 0 if start_index > buffer.size + # log "Buffer [#{uuid}][#{buffer[:iterator]}] start_index unexpectedly negative" if start_index < 0 + new_bookmark_uuid = Sysrandom.uuid + new_bookmark = (buffer.size - start_index > limit) ? (start_index + limit) : buffer.size + # log "Buffer - Calculated new_bookmark: #{new_bookmark}" + if new_bookmark == buffer.size && !buffer[:status].nil? + # Process is done. The next bookmark will be an indicator to send status and message next time + # as well as slice to the end of the array below + new_bookmark = -1 + end + if new_bookmark == start_index && @bookmark_uuid # No data will be given; have them request the same bookmark if we were given one + new_bookmark_uuid = @bookmark_uuid + else + buffer[:bookmarks][new_bookmark_uuid] = new_bookmark + buffer[:bookmarks].delete @bookmark_uuid unless @bookmark_uuid.nil? + end + # log "Buffer - Command done. Moved to : #{new_bookmark}" if new_bookmark == -1 + + data_response = start_index == new_bookmark ? [] : buffer[:data][start_index..new_bookmark] + response_object = { + :bookmark => new_bookmark_uuid, + :data => data_response + } + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Responding #{start_index}..#{new_bookmark}: #{response_object}" + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Bookmarks: #{buffer[:bookmarks]}" + Oj.dump(response_object, mode: :compat) + end + end + + post '/control/server/:action(/:id)?', auth: :admin do + id = params['id'] + action = params['action'] + body = request.body.read + request.body.rewind + options = Oj.load body, symbol_keys: true + daemon = @@daemons[id] + if daemon.nil? && !['install', 'update', 'start'].include?(action) + status 400 + return "Could not find an active daemon for ID #{id}!" + end + validate = params['validate'].to_s == "true" + case action + when 'install' + uuid, buffer = self.class.create_buffer + log "Installing server" + Thread.new do + install_server(buffer, validate: validate) + end + log "Server install thread started. Returning UUID: #{uuid}" + uuid + when 'update' + uuid, buffer = self.class.create_buffer + log "Updating server" + Thread.new do + update_server(buffer, validate: validate) + end + log "Server update thread started. Returning UUID: #{uuid}" + uuid + when 'start' + config = $config_handler.server_configs.select{ |_, config| config['id'] == id }.first.last rescue nil + if config.nil? + status 400 + return "Unknown config ID: #{id}" + end + daemon = self.class.init_daemon config # Returns existing daemon if present + session[:active_daemon_uuid] = daemon.buffer[:uuid] + daemon.do_start_server + when 'stop' + body daemon.do_stop_server + session[:active_daemon_uuid] = nil if daemon.buffer[:uuid] == session[:active_daemon_uuid] + when 'delete' + @@daemons.delete(@@daemons.key daemon).implode + when 'restart' + daemon.config.merge!($config_handler.server_configs[daemon.id]) if $config_handler.server_configs[daemon.id] + if daemon.server_running? + daemon.do_restart_server + else + status 400 + "Server is not running" + end + when 'rcon' + begin + uuid, buffer = self.class.create_buffer + Thread.new { daemon.do_send_rcon(options[:command], host: options[:host] || '127.0.0.1', port: options[:port], pass: options[:pass], buffer: daemon.rcon_buffer, outcome_buffer: buffer) } + uuid + rescue => e + log "Error while sending RCON", e + status 500 + e.message + end + when 'chat' + begin + uuid, buffer = self.class.create_buffer + Thread.new { daemon.do_send_rcon("say #{options[:command]}", host: options[:host], port: options[:port], pass: options[:pass], buffer: daemon.chat_buffer, outcome_buffer: buffer, no_rx: true) } + uuid + rescue => e + log "Error while sending chat RCON", e + status 500 + e.message + end + else + status 400 + "Unknown action: #{action}" + end + end + + get '/config/file/:file', auth: :admin do + config = get_config(params['config-id']) + file_type = params['file'] + halt(400, "File type not found: #{file_type}") unless CONFIG_FILES[file_type.to_sym] + file = ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + halt(400, "File not found: #{file.sub(USER_HOME, '~')}") unless File.exist? file + File.read(file) + end + + post '/config/file', auth: :admin do + config_id = params['config_id'] + config = get_config(config_id) + file_type = params['file'] + halt(400, "File type not found: #{file_type}") unless CONFIG_FILES[file_type.to_sym] + file = ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + content = params['content'] + content << "\n" unless content.end_with? "\n" + FileUtils.mkdir_p(File.dirname(file)) + File.write(file, content) + "Wrote #{file.sub(USER_HOME, '~')}." + end + + put '/config/:action', auth: :admin do + if params['action'] == 'set' + begin + value = case params['variable'] + when 'server_mutators' + params['value'].split(',') + else + params['value'] + end + config_id = params['config-id'] + config = get_config(config_id) + status, msg, current, old = $config_handler.set config['id'], params['variable'], value + if status + daemon = @@daemons.select { |id, _| id == config['id'] }.first.last rescue nil + if daemon + daemon.config.merge!($config_handler.server_configs[config_id]) # Merge the current config for next restart + log "Merged config to active daemon for config ID #{daemon.config['id']}" + end + "Changed #{params['variable']}: #{old} => #{current}" + else + status 400 + msg + end + rescue => e + msg = "Failed to set #{params['variable'] + ' -> ' + params['value']}" + log msg, e + status 400 + "#{msg} | #{e.class}: #{e.message}" + end + elsif params['action'] == 'get' + value = $config_handler.get params['config-id'], params['variable'] + if value.nil? + status 404 + "Could not find value for #{params['variable']}" + else + value.to_s + end + else + halt 400, 'Unknown action' + end + end + + get '/daemons', auth: :admin do + @daemons = @@daemons + erb :daemons + end + + get '/daemon/:config_id', auth: :admin do + daemon = @@daemons[params[:config_id]] + if daemon.nil? + status 400 + "Could not find server daemon with ID #{params[:config_id]}" + else + session[:active_server_config] = daemon.config['id'] + nil + end + end + + get '/monitors', auth: :admin do + @monitors = @@monitors + erb :monitors + end + + get '/monitor-config/:name', auth: :admin do + name = params['name'] + config = $config_handler.monitor_configs[name] + running = !@@monitors[config['id']].nil? + Oj.dump({'running' => running}.merge(config)) + end + + post '/monitor-config/:name', auth: :admin do + name = params['name'] + data = Oj.load(request.body.read) + request.body.rewind + config = data['config'] + if config['ip'] && config['query_port'] && config['rcon_port'] && config['rcon_password'] + $config_handler.monitor_configs[name] = { + 'id' => $config_handler.monitor_configs.dig(name, 'id') || Sysrandom.uuid, + 'ip' => config['ip'], + 'name' => name, + 'query_port' => config['query_port'], + 'rcon_port' => config['rcon_port'], + 'rcon_password' => config['rcon_password'], + } + $config_handler.write_monitor_configs + Oj.dump({message: 'Monitor config saved'}.merge($config_handler.monitor_configs[name])) + else + status 400 + 'Missing parameter(s)' + end + end + + delete '/monitor-config/:name', auth: :admin do + name = params['name'] + if $config_handler.monitor_configs[name] + $config_handler.monitor_configs.delete name + "Monitor config deleted" + else + code 400 + "Unknown monitor config" + end + end + + get '/monitor-configs', auth: :admin do + @configs = $config_handler.monitor_configs + erb :'monitor-configs' + end + + get '/monitor/:id', auth: :admin do + # Get UUID for RCON buffer for tailing + if @@monitors[params['id']] + @@monitors[params['id']].rcon_buffer[:uuid] + else + status 400 + 'Unknown monitor' + end + end + + post '/monitor/:id/:action', auth: :admin do + data = Oj.load(request.body.read) + request.body.rewind + id = params['id'] + action = params['action'] + monitor_config = $config_handler.monitor_configs.select { |_, config| config['id'] == id }.first.last rescue nil + if monitor_config.nil? + status 404 + return "Could not find monitor with ID: #{id}" + end + case action + when 'start' + if @@monitors[id] + status 409 + "Monitor with ID #{id} already exists" + else + @@monitors[id] = ServerMonitor.new(monitor_config['ip'], monitor_config['query_port'], monitor_config['rcon_port'], monitor_config['rcon_password'], rcon_buffer: self.class.create_buffer.last, interval: 10, delay: 0, name: monitor_config['name']) + "Started monitor with ID #{id}" + end + when 'stop' + if @@monitors[id] + @@monitors[id].stop + @@monitors.delete id + "Monitor stopped" + else + status 400 + "Monitor with ID #{id} doesn't exist" + end + else + status 400 + 'Unknown action' + end + end + + get '/monitoring-details/:id', auth: :admin do + @info = {} + @id = params[:id] + matching_daemons = @@daemons[@id] + @monitor = @@daemons[@id] && @@daemons[@id].monitor || @@monitors[@id] + @info = @monitor.info rescue nil + erb :'monitoring-details' + end + + post '/moderator/:action/:unique_id', auth: :moderator do + data = Oj.load(request.body.read) + request.body.rewind + reason = data['reason'].gsub('"', '\"') + unique_id = params[:unique_id] + ip, port, password = get_rcon_info(id: data['id']) + unless password + halt 400, "Could not find an RCON password for ID: #{data['id']}" + end + case params[:action] + when 'ban' + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(ip, port, password, "banid #{unique_id} #{reason}", buffer: buffer) } + uuid + when 'kick' + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(ip, port, password, "kickid #{unique_id} #{reason}", buffer: buffer) } + uuid + else + status 400 + 'Unknown action' + end + end + + post '/rcon', auth: :admin do + body = request.body.read + request.body.rewind + options = Oj.load body, symbol_keys: true + log "Calling RCON client for command: [#{options[:host]}:#{options[:port]}] (TX >>) #{options[:command]}" + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(options[:host], options[:port], options[:pass], options[:command], buffer: buffer) } + uuid + end + + get '/server-configs', auth: :admin do + @configs = $config_handler.server_configs + erb :'server-configs' + end + + get '/server-config', auth: :admin do + config_id = params['config-id'] + config = get_config(config_id) + session[:active_server_config] = config_id + Oj.dump(config) + end + + post '/server-config', auth: :admin do + config = Oj.load(request.body.read) + config_id = config['id'] + if config_id && !config_id.is_uuid? + halt 400, "Invalid config ID format" + end + config_name = config['server-config-name'] + request.body.rewind + new_config = $config_handler.create_server_config config + overwrote = new_config['id'] == config_id + Oj.dump({ + message: "#{overwrote ? 'Overwrote' : 'Created'} #{new_config['id']}", + id: new_config['id'], + }, mode: :compat) + end + + delete '/server-config', auth: :admin do + config_id = params['config-id'] + halt 400, "Server config not found for ID #{config_id}" unless $config_handler.server_configs[config_id] + $config_handler.delete_server_config config_id + "Deleted #{params['name']}" + end + + get '/get-buffer/:id/:type', auth: :admin do + id = params[:id] + daemon = @@daemons[id] + if daemon.nil? + status 400 + return "No daemon exists for config ID #{id}" + end + type = params[:type] + case type + when 'server' + daemon.buffer[:uuid] + when 'rcon' + daemon.rcon_buffer[:uuid] + when 'chat' + daemon.chat_buffer[:uuid] + else + status 400 + "Unknown buffer type: #{type}" + end + end + + post '/automatic-updates/:action', auth: :admin do + action = params['action'] + case action + when 'disable' + @@config['automatic_updates'] = false + self.class.save_webapp_config @@config + stop_updates + when 'enable' + @@config['automatic_updates'] = true + self.class.save_webapp_config @@config + start_updates + else + status 400 + "Unknown action: #{action}" + end + end + + get '/generate-password', auth: :admin do + ConfigHandler.generate_password + end + + get '/download-server-config', auth: :admin do + config_id = params['config-id'] + unless config_id.is_uuid? + halt 400, "Invalid config ID format" + end + config_dir = File.join CONFIG_FILES_DIR, config_id + unless File.directory?(config_dir) + halt 400, "No files found for this config ID" + end + zip_path = File.expand_path File.join CONFIG_FILES_DIR, "ohdaw-config_#{config_id}_#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + FileUtils.touch zip_path + zip_files zip_path, File.join(config_dir, '**/*') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end + + get '/download-server-log/:config_id', auth: :admin do + log_file = File.expand_path File.join SERVER_LOG_DIR, "#{params['config_id']}.log" + if File.exist? log_file + send_file log_file, filename: File.basename(log_file), disposition: :attachment + else + status 400 + "Unable to find log file: #{params['config_name']}.log" + end + end + + get '/download-server-logs(/:config_id)?', auth: :admin do + config_id = params['config_id'] + zip_path = File.expand_path File.join SERVER_LOG_DIR, "#{config_id + '-' if config_id}ohd-logs-#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + zip_files zip_path, File.join(SERVER_LOG_DIR, config_id ? "#{config_id}*.log" : '*.log') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end + + get '/download-wrapper-logs', auth: :host do + zip_path = File.expand_path File.join WEBAPP_ROOT, 'log', "ohdaw-logs-#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + zip_files zip_path, File.join(WEBAPP_ROOT, 'log', '*.log*') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end +end + +def get_webrick_options(config = AdminWrapperSite.load_webapp_config) + webrick_options = { + AccessLog: [[LOGGER, WEBrick::AccessLog::COMMON_LOG_FORMAT], [LOGGER, WEBrick::AccessLog::REFERER_LOG_FORMAT]], + Host: config['admin_interface_bind_ip'], + Port: config['admin_interface_bind_port'], + Logger: LOGGER, # Logger.new(IO::NULL), + app: AdminWrapperSite, # app must be lowercase + SSLEnable: config['admin_interface_use_ssl'] + } + + if config['admin_interface_use_ssl'] + webrick_options.merge!({ + SSLVerifyClient: config['admin_interface_verify_ssl'] ? OpenSSL::SSL::VERIFY_PEER : OpenSSL::SSL::VERIFY_NONE, + SSLCertName: [[ 'CN', WEBrick::Utils::getservername ]] + }) + if File.exist?(config['admin_interface_ssl_cert']) && File.exist?(config['admin_interface_ssl_key']) + webrick_options.merge!({ + SSLCertificate: OpenSSL::X509::Certificate.new(File.read config['admin_interface_ssl_cert']), + SSLPrivateKey: OpenSSL::PKey::RSA.new(File.read config['admin_interface_ssl_key']) + }) + else + # Make temporary cert/key, since WEBrick's defaults won't work with modern browsers + cert, key = CertificateGenerator.generate + log "Writing generated certs to file", level: :info + config['admin_interface_ssl_cert'] = GENERATED_SSL_CERT + config['admin_interface_ssl_key'] = GENERATED_SSL_KEY + AdminWrapperSite.save_webapp_config(config) + File.write GENERATED_SSL_CERT, cert + File.write GENERATED_SSL_KEY, key + webrick_options.merge!({ + SSLCertificate: cert, + SSLPrivateKey: key, + }) + end + end + webrick_options +rescue => e + log "Couldn't create webrick options", e + raise +end + +begin + config = AdminWrapperSite.load_webapp_config + options = get_webrick_options(config) + log "Starting webserver with options: #{options}" + $server = Rack::Server.new(options) + Thread.new { sleep 1; log "Webserver initialized! Visit: http#{'s' if config['admin_interface_use_ssl']}://#{config['admin_interface_bind_ip'] == '0.0.0.0' ? '127.0.0.1' : config['admin_interface_bind_ip']}:#{config['admin_interface_bind_port']}", level: :info } + $server.start +rescue => e + unless e.is_a? SystemExit + log "Exception caused webserver exit", e + at_exit { exit 1 } + end +ensure + log "Webserver stopped", level: :info + $config_handler.write_config + log "Config written", level: :info + `stty echo` unless WINDOWS || !$stdout.isatty +end diff --git a/admin-interface/log/.gitignore b/admin-interface/log/.gitignore new file mode 100644 index 0000000..55f8b93 --- /dev/null +++ b/admin-interface/log/.gitignore @@ -0,0 +1,2 @@ +*.log* + diff --git a/extras/systemd/sandstorm-admin-wrapper.service b/extras/systemd/sandstorm-admin-wrapper.service new file mode 100644 index 0000000..9533450 --- /dev/null +++ b/extras/systemd/sandstorm-admin-wrapper.service @@ -0,0 +1,13 @@ +[Install] +WantedBy=multi-user.target + +[Unit] +Description=OHD Admin Wrapper + +[Service] +User=wrapper-user +Type=simple +WorkingDirectory=/home/wrapper-user/ohd-admin-wrapper/admin-interface +ExecStartPre=/home/wrapper-user/.rbenv/bin/rbenv exec bundle install +ExecStart=/home/wrapper-user/.rbenv/bin/rbenv exec bundle exec ruby lib/webapp.rb +Restart=always diff --git a/linux_start.sh b/linux_start.sh new file mode 100755 index 0000000..1d40dd4 --- /dev/null +++ b/linux_start.sh @@ -0,0 +1,12 @@ +#!/usr/bin/env bash + +wrapper_root="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )" + +cd "$wrapper_root"/admin-interface + +while true; do + bundle && bundle exec ruby "$wrapper_root"/admin-interface/lib/webapp.rb "$@" + CODE=$? + [ $CODE -eq 2 ] || exit $CODE # Restart if exit code 2 +done + diff --git a/windows_start.bat b/windows_start.bat new file mode 100644 index 0000000..47788f0 --- /dev/null +++ b/windows_start.bat @@ -0,0 +1,32 @@ +@echo off +cd %~dp0\admin-interface + +:bundle +echo Bundling +call bundle +if ERRORLEVEL 1 goto bundle_error +echo Starting webapp + +:run_webapp +call bundle exec ruby lib/webapp.rb %* +echo Error level: %errorlevel% +if errorlevel 2 goto :bundle +if errorlevel 1 goto exit_error +if errorlevel 0 goto exit_success + +:webapp_error +echo Failed to run OHD Admin Wrapper. Ensure Ruby is version 2.6.3+. Logs are located in admin-interface/log. +goto exit_error + +:bundle_error +echo Failed to run Ruby Bundler. Ensure Ruby is installed with the bundler gem. https://rubyinstaller.org/downloads/ +goto exit_error + +:exit_error +pause +exit 1 + +:exit_success +echo OHD Admin Wrapper exited successfully. +pause +exit 0
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/maplist.erb b/admin-interface/docroot/views/maplist.erb
new file mode 100644
index 0000000..74b7f89
--- /dev/null
+++ b/admin-interface/docroot/views/maplist.erb
@@ -0,0 +1,5 @@
+<% MAPMAP.values.sort.each do |map| %>
+<%= map %>
+<% end %>
+
+Random
\ No newline at end of file
diff --git a/admin-interface/docroot/views/monitor-configs.erb b/admin-interface/docroot/views/monitor-configs.erb
new file mode 100644
index 0000000..97c219f
--- /dev/null
+++ b/admin-interface/docroot/views/monitor-configs.erb
@@ -0,0 +1,10 @@
+<% @configs.sort_by { |name, _| name }.each do |name, config| %>
+
+<% end %>
\ No newline at end of file
diff --git a/admin-interface/docroot/views/monitor-tool.erb b/admin-interface/docroot/views/monitor-tool.erb
new file mode 100644
index 0000000..22abe03
--- /dev/null
+++ b/admin-interface/docroot/views/monitor-tool.erb
@@ -0,0 +1,113 @@
+
diff --git a/admin-interface/docroot/views/monitoring-details.erb b/admin-interface/docroot/views/monitoring-details.erb
new file mode 100644
index 0000000..78ac5e8
--- /dev/null
+++ b/admin-interface/docroot/views/monitoring-details.erb
@@ -0,0 +1,26 @@
+<% server_down = @info[:server_down].nil? ? true : @info[:server_down] rescue true %>
+<% rcon_down = @info[:rcon_connection_problem].nil? ? true : @info[:rcon_connection_problem] rescue true %>
+<% query_down = @info[:a2s_connection_problem].nil? ? true : @info[:a2s_connection_problem] rescue true %>
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/monitors.erb b/admin-interface/docroot/views/monitors.erb
new file mode 100644
index 0000000..6eaebb4
--- /dev/null
+++ b/admin-interface/docroot/views/monitors.erb
@@ -0,0 +1,11 @@
+<% @monitors.sort_by {|_, monitor| monitor.name }.each do |id, monitor| %>
+
+<% end %>
diff --git a/admin-interface/docroot/views/players.erb b/admin-interface/docroot/views/players.erb
new file mode 100644
index 0000000..81948eb
--- /dev/null
+++ b/admin-interface/docroot/views/players.erb
@@ -0,0 +1,41 @@
+<% players = @info.nil? ? [] : @info[:rcon_players] || [] %>
+<% players.each do |player| %>
+<% player['display_name'] = player.dig('steam_info', 'name') || player['name'] %>
+<% matches = (@info[:a2s_player] || []).select{ |p| p['name'] == player['name'] } %>
+<% next if matches.empty? %>
+<% player['duration'] = matches.first['duration'] if matches.size == 1 %>
+<% end %>
+Login
+
+
+ <% players.sort_by { |p| [p['score'].to_i, (p['duration'] || ''), p['name']] }.reverse.each_with_index do |player, i| %>
+
diff --git a/admin-interface/docroot/views/rcon-tool.erb b/admin-interface/docroot/views/rcon-tool.erb
new file mode 100644
index 0000000..0534828
--- /dev/null
+++ b/admin-interface/docroot/views/rcon-tool.erb
@@ -0,0 +1,62 @@
+
diff --git a/admin-interface/docroot/views/server-config.erb b/admin-interface/docroot/views/server-config.erb
new file mode 100644
index 0000000..e5c33af
--- /dev/null
+++ b/admin-interface/docroot/views/server-config.erb
@@ -0,0 +1,456 @@
+
diff --git a/admin-interface/docroot/views/server-configs.erb b/admin-interface/docroot/views/server-configs.erb
new file mode 100644
index 0000000..46e67e5
--- /dev/null
+++ b/admin-interface/docroot/views/server-configs.erb
@@ -0,0 +1,7 @@
+<% @configs.sort_by {|_, config| config['server-config-name'] }.each do |key, config| %>
+
+<% end %>
diff --git a/admin-interface/docroot/views/server-control-status.erb b/admin-interface/docroot/views/server-control-status.erb
new file mode 100644
index 0000000..e5468e8
--- /dev/null
+++ b/admin-interface/docroot/views/server-control-status.erb
@@ -0,0 +1,13 @@
+
diff --git a/admin-interface/docroot/views/server-control.erb b/admin-interface/docroot/views/server-control.erb
new file mode 100644
index 0000000..6312235
--- /dev/null
+++ b/admin-interface/docroot/views/server-control.erb
@@ -0,0 +1,127 @@
+
diff --git a/admin-interface/docroot/views/server-daemons.erb b/admin-interface/docroot/views/server-daemons.erb
new file mode 100644
index 0000000..bbc797b
--- /dev/null
+++ b/admin-interface/docroot/views/server-daemons.erb
@@ -0,0 +1,30 @@
+
+
diff --git a/admin-interface/docroot/views/server-list.erb b/admin-interface/docroot/views/server-list.erb
new file mode 100644
index 0000000..b067626
--- /dev/null
+++ b/admin-interface/docroot/views/server-list.erb
@@ -0,0 +1,66 @@
+<% at_least_one_server = false %>
+<% at_least_one_server = true unless @daemons.values.compact.empty? && @monitors.compact.empty? %>
+<% @daemons.reject{|_,d| d.nil? }.sort_by{ |_, d| d.config['server_name'] }.each do |id, daemon| %>
+ <% @id = id %>
+ <% server_running = daemon.server_running? %>
+ <% @monitor = daemon.monitor rescue nil %>
+ <% @info = @monitor.info rescue nil %>
+
+
+
+
+<% end %>
+
+<% @monitors.each do |id, monitor| %>
+ <% @id = id %>
+ <% @monitor = monitor %>
+ <% @info = @monitor.info rescue nil %>
+
+
+
+<% end %>
+
+<% unless at_least_one_server %>
+
+<% end %>
diff --git a/admin-interface/docroot/views/server-setup.erb b/admin-interface/docroot/views/server-setup.erb
new file mode 100644
index 0000000..6039d8f
--- /dev/null
+++ b/admin-interface/docroot/views/server-setup.erb
@@ -0,0 +1,92 @@
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/server-status.erb b/admin-interface/docroot/views/server-status.erb
new file mode 100644
index 0000000..1c1a3ab
--- /dev/null
+++ b/admin-interface/docroot/views/server-status.erb
@@ -0,0 +1,13 @@
+
+
+
+
diff --git a/admin-interface/docroot/views/steamcmd-tool.erb b/admin-interface/docroot/views/steamcmd-tool.erb
new file mode 100644
index 0000000..24b7b4b
--- /dev/null
+++ b/admin-interface/docroot/views/steamcmd-tool.erb
@@ -0,0 +1,35 @@
+
diff --git a/admin-interface/docroot/views/threads.erb b/admin-interface/docroot/views/threads.erb
new file mode 100644
index 0000000..355d13b
--- /dev/null
+++ b/admin-interface/docroot/views/threads.erb
@@ -0,0 +1,49 @@
+
diff --git a/admin-interface/docroot/views/update-info.erb b/admin-interface/docroot/views/update-info.erb
new file mode 100644
index 0000000..5827c0a
--- /dev/null
+++ b/admin-interface/docroot/views/update-info.erb
@@ -0,0 +1,21 @@
+
+
+ <% end %>
+
+ <%# <% ip_info = player.dig('ip_info', player['ip']).to_h %>
+ <%# <% country_code = ip_info['countryCode'] %>
+ <%# <% region_name = ip_info['regionName'] %>
+ <%# <% city = ip_info['city'] %>
+ <%# <%= (' ') if country_code %>
+ <<% if player['unique_id'].nil? %>span<% else %>a target="<%= !@use_steam_urls ? '_blank' : '' %>" href="<%= !@use_steam_urls ? 'https://steamcommunity.com/profiles/' : 'steam://url/SteamIDPage/' %><%= player['unique_id'] %>" <% end %>>
+  || '/custom/missing.png' %>)
+ <%= CGI.escapeHTML player['display_name'] %>
+ span<% else %>a<% end %>>
+
+
+ <%= CGI.escapeHTML player['display_name'] %>
+ span<% else %>a<% end %>>
+ <%= player['score'] %>
+ <%= player['duration'] %>
+
+ <% if is_moderator? %>
+
+
+
+ <%# <%= "#{country_code} ➤ #{region_name} ➤ #{city}" %>
+
+ <% end %>
+
+ Server Update
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/wrapper-config.erb b/admin-interface/docroot/views/wrapper-config.erb
new file mode 100644
index 0000000..72e9f1a
--- /dev/null
+++ b/admin-interface/docroot/views/wrapper-config.erb
@@ -0,0 +1,160 @@
+
diff --git a/admin-interface/docroot/views/wrapper-log.erb b/admin-interface/docroot/views/wrapper-log.erb
new file mode 100644
index 0000000..6d4509a
--- /dev/null
+++ b/admin-interface/docroot/views/wrapper-log.erb
@@ -0,0 +1,37 @@
+
+
+
\ No newline at end of file
diff --git a/admin-interface/docroot/views/wrapper-users-list.erb b/admin-interface/docroot/views/wrapper-users-list.erb
new file mode 100644
index 0000000..8ec4f86
--- /dev/null
+++ b/admin-interface/docroot/views/wrapper-users-list.erb
@@ -0,0 +1,29 @@
+<% @users.each do |id, user| %>
+
+
+<% end %>
\ No newline at end of file
diff --git a/admin-interface/docroot/views/wrapper-users.erb b/admin-interface/docroot/views/wrapper-users.erb
new file mode 100644
index 0000000..ceae2ed
--- /dev/null
+++ b/admin-interface/docroot/views/wrapper-users.erb
@@ -0,0 +1,35 @@
+
diff --git a/admin-interface/ext/string.rb b/admin-interface/ext/string.rb
new file mode 100644
index 0000000..8401af9
--- /dev/null
+++ b/admin-interface/ext/string.rb
@@ -0,0 +1,50 @@
+# encoding: UTF-8
+
+require 'unicode/display_width'
+
+# Certain UTF-8 characters display as wider than they report with the standard Ruby length method
+# This can cause issues when justifying text containing these characters. Use the methods below instead.
+class String
+ def length_utf8
+ Unicode::DisplayWidth.of(self)
+ end
+
+ def ljust_utf8(length, padstr=' ')
+ if Unicode::DisplayWidth.of(self) < length
+ self + (padstr * (length - Unicode::DisplayWidth.of(self)))
+ else
+ self
+ end
+ end
+
+ def rjust_utf8(length, padstr=' ')
+ if Unicode::DisplayWidth.of(self) < length
+ (padstr * (length - Unicode::DisplayWidth.of(self))) + self
+ else
+ self
+ end
+ end
+
+ def center_utf8(length, padstr=' ')
+ if Unicode::DisplayWidth.of(self) < length
+ prefix = padstr * ((length - Unicode::DisplayWidth.of(self))/2.to_f).floor # Floor here to keep text left of center if uneven
+ postfix = padstr * ((length - Unicode::DisplayWidth.of(self))/2.to_f).ceil
+ prefix + self + postfix
+ else
+ self
+ end
+ end
+
+ def utf8
+ s = encode("UTF-16be", :invalid=>:replace, :replace=>"?").encode('UTF-8') #force_encoding('UTF-8')
+ s.strip # Don't strip; test that it can be stripped (else we fall back to encode())
+ s
+ rescue => e
+ log("Failed to encode string as UTF-8: #{self.inspect}", level: :warn)
+ encode('UTF-8', invalid: :replace, undef: :replace)
+ end
+
+ def is_uuid?
+ self =~ /^[{]?[0-9a-fA-F]{8}-([0-9a-fA-F]{4}-){3}[0-9a-fA-F]{12}[}]?$/
+ end
+end
diff --git a/admin-interface/ext/thread.rb b/admin-interface/ext/thread.rb
new file mode 100644
index 0000000..63851f1
--- /dev/null
+++ b/admin-interface/ext/thread.rb
@@ -0,0 +1,27 @@
+
+# Threads which terminate unexpectedly are not logged
+# so we monkey-patch a rescue in for StandardErrors in order
+# to log and raise the exception.
+
+Thread.report_on_exception = false
+
+module ThreadExtensions
+ def initialize(*args)
+ begin
+ super(*args) do |args|
+ begin
+ yield(args)
+ rescue => e
+ level = :error
+ level = :debug if e.is_a?(IOError) && e.message == 'stream closed in another thread'
+ log "(#{self.inspect}#{' ' << self[:name] if self[:name]}) StandardError in thread", e, level: level
+ raise e
+ end
+ end
+ end
+ end
+end
+
+class Thread
+ prepend ThreadExtensions
+end
diff --git a/admin-interface/lib/buffer.rb b/admin-interface/lib/buffer.rb
new file mode 100644
index 0000000..3738995
--- /dev/null
+++ b/admin-interface/lib/buffer.rb
@@ -0,0 +1,57 @@
+require_relative 'logger'
+
+class Buffer < Hash
+ def initialize(uuid, opts={})
+ self.merge!({
+ uuid: uuid,
+ data: [],
+ bookmarks: {},
+ status: nil,
+ message: nil,
+ mutex: Mutex.new,
+ limit: 500,
+ filters: [],
+ persistent: false
+ })
+ self.merge!(opts)
+ end
+
+ def synchronize
+ self[:mutex].synchronize { yield }
+ end
+
+ def <<(string)
+ self[:data] << string
+ truncate
+ self[:data]
+ end
+
+ def push(string)
+ self << string
+ end
+
+ def truncate(limit: self[:limit])
+ current_size = size
+ return 0 unless current_size > limit
+ over_limit = current_size - limit
+ self[:data].shift over_limit # Remove the first (oldest) n elements
+ self[:bookmarks].each do |uuid, value|
+ next if value == -1
+ self[:bookmarks][uuid] = (value - over_limit).floor 0
+ end
+ over_limit
+ end
+
+ def reset
+ unless self[:persistent]
+ self[:data].clear
+ self[:bookmarks].clear
+ end
+ self[:status] = nil
+ self[:message] = nil
+ end
+
+ def size
+ self[:data].size
+ end
+end
diff --git a/admin-interface/lib/certificate-generator.rb b/admin-interface/lib/certificate-generator.rb
new file mode 100644
index 0000000..9d5b9e7
--- /dev/null
+++ b/admin-interface/lib/certificate-generator.rb
@@ -0,0 +1,26 @@
+require_relative 'logger'
+
+class CertificateGenerator
+ def self.generate
+ key = OpenSSL::PKey::RSA.new(2048)
+ public_key = key.public_key
+ subject = "/CN=localhost"
+ cert = OpenSSL::X509::Certificate.new
+ cert.subject = cert.issuer = OpenSSL::X509::Name.parse(subject)
+ cert.not_before = Time.now
+ cert.not_after = Time.now + 365 * 24 * 60 * 60
+ cert.public_key = public_key
+ cert.serial = 0x0
+ cert.version = 2
+ ef = OpenSSL::X509::ExtensionFactory.new
+ ef.subject_certificate = cert
+ ef.issuer_certificate = cert
+ cert.extensions = [
+ ef.create_extension("basicConstraints", "CA:TRUE", true),
+ ef.create_extension("subjectKeyIdentifier", "hash")
+ ]
+ cert.add_extension ef.create_extension("authorityKeyIdentifier", "keyid:always,issuer:always")
+ cert.sign key, OpenSSL::Digest::SHA256.new
+ return cert, key
+ end
+end
\ No newline at end of file
diff --git a/admin-interface/lib/config-handler.rb b/admin-interface/lib/config-handler.rb
new file mode 100644
index 0000000..bd82a01
--- /dev/null
+++ b/admin-interface/lib/config-handler.rb
@@ -0,0 +1,696 @@
+require 'bcrypt'
+require 'erb'
+require 'fileutils'
+require 'inifile'
+require 'net/http'
+require 'oj'
+require 'socket'
+require 'sysrandom'
+require_relative 'logger'
+
+# Used in server-monitor to avoid showing internal IPs for local players
+# but OHD does not (yet?) provide player IP addresses like Insurgency: Sandstorm does
+# LOCAL_IP_PREFIXES = ['10.', '0.', '127.','192.168.'] # https://en.wikipedia.org/wiki/Reserved_IP_addresses
+# Thread.new { EXTERNAL_IP = Net::HTTP.get URI "https://api.ipify.org" rescue nil }
+
+CONFIG_PATH = File.expand_path File.join File.dirname(__FILE__), '..', 'config'
+CONFIG_FILE = File.join CONFIG_PATH, 'config.json'
+USERS_CONFIG_FILE = File.join CONFIG_PATH, 'users.json'
+MONITOR_CONFIGS_FILE = File.join CONFIG_PATH, 'monitor-configs.json'
+SERVER_CONFIGS_FILE = File.join CONFIG_PATH, 'server-configs.json'
+PLAYER_INFO_FILE = File.join CONFIG_PATH, 'players.json'
+
+WRAPPER_ROOT = File.expand_path(File.join(File.dirname(__FILE__), '..', '..')).freeze
+WRAPPER_CONFIG = File.join WRAPPER_ROOT, 'config'
+WEBAPP_ROOT = File.join WRAPPER_ROOT, 'admin-interface'
+WEBAPP_CONFIG = File.join(WRAPPER_CONFIG, 'config.toml')
+WEBAPP_CONFIG_SAMPLE = File.join(WRAPPER_CONFIG, 'config.toml.sample')
+GENERATED_SSL_CERT = File.join(WRAPPER_CONFIG, 'generated_ssl_cert.pem')
+GENERATED_SSL_KEY = File.join(WRAPPER_CONFIG, 'generated_ssl_key.pem')
+SERVER_ROOT = File.join WRAPPER_ROOT, 'ohd-server'
+STEAMCMD_ROOT = File.join WRAPPER_ROOT, 'steamcmd'
+STEAMCMD_EXE = File.join STEAMCMD_ROOT, 'installation', (WINDOWS ? "steamcmd.exe" : "steamcmd.sh")
+STEAM_APPINFO_VDF = File.join STEAMCMD_ROOT, (WINDOWS ? "installation" : 'Steam'), 'appcache', 'appinfo.vdf'
+USER_HOME = ENV['HOME'] unless USER_HOME
+ENV['HOME'] = STEAMCMD_ROOT # Steam will pollute the user directory otherwise on Linux.
+
+BINARY = File.join SERVER_ROOT, 'HarshDoorstop', 'Binaries', (WINDOWS ? 'Win64\HarshDoorstopServer-Win64-Shipping.exe' : 'Linux/HarshDoorstopServer-Linux-Shipping')
+SERVER_LOG_DIR = File.join SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Logs'
+RCON_LOG_FILE = File.join SERVER_LOG_DIR, 'HarshDoorstop.log'
+CONFIG_FILES_DIR = File.join WRAPPER_ROOT, 'server-config'
+FileUtils.mkdir_p CONFIG_FILES_DIR
+CONFIG_FILES = {
+ game_ini: {
+ type: :ini,
+ actual: File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Game.ini'),
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'Game.ini')%>"
+ },
+ engine_ini: {
+ type: :ini,
+ actual: File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Engine.ini'),
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'Engine.ini')%>"
+ },
+ game_user_settings_ini: {
+ type: :ini,
+ actual: File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'GameUserSettings.ini'),
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'GameUserSettings.ini')%>"
+ },
+ admins_cfg: {
+ type: :txt,
+ actual_erb: "<%=File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Admins_' + config_id + '.cfg')%>",
+ local_erb: "<%=File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Admins_' + config_id + '.cfg')%>"
+ },
+ mapcycle_cfg: {
+ type: :txt,
+ actual_erb: "<%=File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'MapCycle_' + config_id + '.cfg')%>",
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'MapCycle.cfg')%>"
+ },
+ bans_cfg: {
+ type: :txt,
+ actual: File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Bans.cfg'),
+ local_erb: "<%=File.join(SERVER_ROOT, 'HarshDoorstop', 'Saved', 'Config', (WINDOWS ? 'WindowsServer' : 'LinuxServer'), 'Bans.cfg')%>"
+ },
+ motd_txt: {
+ type: :txt,
+ actual_erb: "<%=File.join(SERVER_ROOT, 'HarshDoorstop', 'Config', 'Server', 'Motd_' + config_id + '.txt')%>",
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'Motd.txt')%>"
+ },
+ notes_txt: {
+ type: :txt,
+ actual: nil,
+ local_erb: "<%=File.join(CONFIG_FILES_DIR, config_id, 'Notes.txt')%>"
+ },
+}
+
+MAPMAP = {
+ 'Argonne' => 'Argonne',
+ 'Montecassino' => ' Monte Cassino',
+ 'LamDong' => 'Lam Dong',
+ 'Khafji_P' => 'Khafji_P',
+ 'Risala' => 'Risala',
+ 'AAS-TestMap' => 'Test Map (AAS)'
+ # 'SaintQuentin' => 'Saint Quentin',
+ # 'OmahaBeach' => 'OmahaBeach',
+ # 'Carentan' => 'Carentan',
+ # 'Tan_Binh' => 'Tan_Binh'
+}
+MAPMAP_INVERTED = MAPMAP.invert
+SIDES = [
+ 'Insurgents',
+ 'Security'
+]
+SCENARIO_MODES = [
+]
+GAME_MODES = [
+ 'Ambush',
+ 'Domination',
+ 'Firefight',
+ 'Frontline',
+ 'Occupy',
+ 'Skirmish',
+ 'CaptureTheBase',
+ 'TeamDeathmatch',
+ 'Filming',
+ 'Mission',
+ 'Checkpoint',
+ 'CheckpointHardcore',
+ 'CheckpointTutorial',
+ 'Operations',
+ 'Outpost',
+ 'Survival'
+]
+RULE_SETS = [
+ 'CheckpointFrenzy',
+ 'CompetitiveFirefight',
+ 'CompetitiveTheater',
+ 'MatchmakingCasual',
+ 'OfficialRules'
+]
+MUTATORS = {
+ 'AllYouCanEat' => { 'name' => 'All You Can Eat', 'description' => 'Start with 100 supply points.' },
+ 'AntiMaterielRiflesOnly' => { 'name' => 'Anti-Materiel Only', 'description' => 'Only anti-materiel rifles are available along with normal equipment and explosives.' },
+ 'BoltActionsOnly' => { 'name' => 'Bolt-Actions Only', 'description' =>'Only bolt-action rifles are available along with normal equipment and explosives.' },
+ 'Broke' => { 'name' => 'Broke', 'description' => 'Start with 0 supply points.' },
+ 'BudgetAntiquing' => {'name' => 'Budget Antiquing', 'description' => 'Only allows only old, cheap weapons along with normal equipment and explosives.' },
+ 'BulletSponge' => { 'name' => 'Bullet Sponge', 'description' => 'Health is increased.' },
+ 'Competitive' => { 'name' => 'Competitive', 'description' => 'Equipment is more expensive, rounds are shorter, and capturing objectives is faster.' },
+ 'CompetitiveLoadouts' => { 'name' => 'Competitive Loadouts', 'description' => 'Player classes are replaced with those from Competitive.' },
+ 'FastMovement' => { 'name' => 'Fast Movement', 'description' => 'Move faster.' },
+ 'Frenzy' => { 'name' => 'Frenzy', 'description' => 'Fight against AI enemies who only use melee attacks. Watch out for special enemies.' },
+ 'FullyLoaded' => {'name' => 'Fully Loaded', 'description' => 'All weapons, equipment, and explosives in the game are available in the Loadout Menu.' },
+ 'Guerrillas' => { 'name' => 'Guerrillas', 'description' => 'Start with 5 supply points.' },
+ 'Gunslingers' => {'name' => 'Gunslingers', 'description' => 'Players are equipped with the MR 73 revolver, normal equipment, and explosives.' },
+ 'Hardcore' => { 'name' => 'Hardcore', 'description' => 'Mutator featuring slower movement speeds and longer capture times.' },
+ 'HeadshotOnly' => { 'name' => 'Headshots Only', 'description' => 'Players only take damage when shot in the head.' },
+ 'HotPotato' => { 'name' => 'Hot Potato', 'description' => 'A live fragmentation grenade is dropped on death.' },
+ 'LockedAim' => { 'name' => 'Locked Aim', 'description' => 'Weapons always point to the center of the screen.' },
+ 'MakarovsOnly' => {'name' => 'Makarovs Only', 'description' => 'Makarov pistols only.' },
+ 'NoAim' => { 'name' => 'No Aim Down Sights', 'description' => 'Aiming down sights is disabled.' },
+ 'NoDrops' => {'name' => 'No Drops', 'description' => 'Weapons, explosives, and equipment are not dropped by dead players or enemy AI.' },
+ 'PistolsOnly' => { 'name' => 'Pistols Only', 'description' => 'Only pistols are available along with normal equipment and explosives.' },
+ 'Poor' => {'name' => 'Poor', 'description' => 'Players start with only 2 supply points' },
+ 'ShotgunsOnly' => { 'name' => 'Shotguns Only', 'description' => 'Only Shotguns are available along with normal equipment and explosives.' },
+ 'SlowCaptureTimes' => { 'name' => 'Slow Capture Times', 'description' => 'Objectives will take longer to capture.' },
+ 'SlowMovement' => { 'name' => 'Slow Movement', 'description' => 'Move slower.' },
+ 'SoldierOfFortune' => { 'name' => 'Soldier of Fortune', 'description' => 'Gain supply points as your score increases.' },
+ 'SpecialOperations' => { 'name' => 'Special Operations', 'description' => 'Start with 30 supply points.' },
+ 'Strapped' => { 'name' => 'Strapped', 'description' => 'Start with 1 supply point.' },
+ 'Ultralethal' => { 'name' => 'Ultralethal', 'description' => 'Everyone dies with one shot.' },
+ 'Vampirism' => { 'name' => 'Vampirism', 'description' => 'Receive health when dealing damage to enemies equal to the amount of damage dealt.' },
+ 'Warlords' => { 'name' => 'Warlords', 'description' => 'Start with 10 supply points.' }
+}
+
+
+class ConfigHandler
+ attr_reader :monitor_configs
+ attr_reader :server_configs
+ attr_reader :users
+ attr_reader :bans_thread
+ attr_accessor :players
+
+ def self.generate_password
+ Sysrandom.base64(32 + Sysrandom.random_number(32)).gsub("\n", '')
+ end
+
+ CONFIG_VARIABLES = {
+ 'server-config-name' => {
+ 'default' => 'Default',
+ 'validation' => Proc.new { true }
+ },
+ 'server_default_map' => {
+ 'default' => MAPMAP.keys.sample,
+ 'random' => false,
+ 'validation' => Proc.new { |map| !map.to_s.empty? },
+ 'type' => :map
+ },
+ 'server_lighting_day' => {
+ 'default' => 'true',
+ 'validation' => Proc.new { |val| ['true', 'false'].include? val }
+ },
+ 'server_default_side' => {
+ 'default' => SIDES.sample,
+ 'random' => false,
+ 'validation' => Proc.new { |side| SIDES.include?(side) || side == 'Random' }
+ },
+ 'server_max_players' => {
+ 'default' => '8',
+ 'type' => :argument,
+ 'validation' => Proc.new do |num|
+ num.to_s =~ /\A\d+\Z/
+ rescue
+ false
+ end
+ },
+ 'server_max_players_override' => {
+ 'default' => '10',
+ 'type' => :engine_ini,
+ 'validation' => Proc.new do |num|
+ num.to_s =~ /\A\d+\Z/
+ rescue
+ false
+ end,
+ 'getter' => Proc.new { |engine_ini| engine_ini['SystemSettings']['net.MaxPlayersOverride'] },
+ 'setter' => Proc.new { |engine_ini, players| engine_ini['SystemSettings']['net.MaxPlayersOverride'] = players }
+ },
+ 'server_game_mode' => {
+ 'default' => 'None',
+ 'validation' => Proc.new { |mode| ['None'].concat(GAME_MODES).include? mode }
+ },
+ 'server_scenario_mode' => {
+ 'default' => 'Checkpoint',
+ 'validation' => Proc.new { |mode| SCENARIO_MODES.include? mode }
+ },
+ 'server_rule_set' => {
+ 'default' => 'None',
+ 'validation' => Proc.new { |rule_set| RULE_SETS.include?(rule_set) || rule_set = 'None' }
+ },
+ 'server_mutators_custom' => {
+ 'default' => '',
+ 'validation' => Proc.new { true }
+ },
+ 'server_mutators' => {
+ 'default' => [],
+ 'validation' => Proc.new { |mutators| mutators.all? { |mutator| MUTATORS.keys.include?(mutator) } }
+ },
+ 'server_cheats' => {
+ 'default' => 'false',
+ 'validation' => Proc.new { |val| ['true', 'false'].include? val }
+ },
+ 'server_name' => {
+ 'default' => 'OHD Admin Wrapper',
+ 'type' => :argument,
+ 'template' => '-hostname=<%= it %>',
+ 'validation' => Proc.new { true },
+ },
+ 'server_password' => {
+ 'default' => '',
+ 'type' => :argument,
+ 'template' => '-password=<%= it %>',
+ 'validation' => Proc.new { true },
+ 'sensitive' => true
+ },
+ 'server_game_port' => {
+ 'default' => '7777',
+ 'type' => :argument,
+ 'template' => '-Port=<%= it %>',
+ 'validation' => Proc.new { |port| ConfigHandler.valid_port? port }
+ },
+ 'server_query_port' => {
+ 'default' => '27131',
+ 'type' => :argument,
+ 'template' => '-QueryPort=<%= it %>',
+ 'validation' => Proc.new { |port| ConfigHandler.valid_port? port }
+ },
+ 'server_rcon_enabled' => {
+ 'default' => 'true',
+ 'type' => :game_ini,
+ 'getter' => Proc.new { |game_ini| game_ini['/Script/RCON.RCONServerSystem']['bEnabled'] },
+ 'setter' => Proc.new { |game_ini, bool| game_ini['/Script/RCON.RCONServerSystem']['bEnabled'] = bool.to_s.casecmp('true').zero? ? 'True' : 'False' },
+ 'validation' => Proc.new { true }
+ },
+ 'server_rcon_port' => {
+ 'default' => '27015',
+ 'type' => :game_ini,
+ 'getter' => Proc.new { |game_ini| game_ini['/Script/RCON.RCONServerSystem']['ListenPort'] },
+ 'setter' => Proc.new { |game_ini, port| game_ini['/Script/RCON.RCONServerSystem']['ListenPort'] = port },
+ 'validation' => Proc.new { |port| ConfigHandler.valid_port? port }
+ },
+ 'server_rcon_password' => {
+ 'default' => ConfigHandler.generate_password,
+ 'type' => :game_ini,
+ 'getter' => Proc.new { |game_ini| game_ini['/Script/RCON.RCONServerSystem']['Password'] },
+ 'setter' => Proc.new { |game_ini, password| game_ini['/Script/RCON.RCONServerSystem']['Password'] = password },
+ 'validation' => Proc.new { true },
+ 'sensitive' => true
+ },
+ 'server_gst' => {
+ 'default' => '',
+ 'type' => :argument,
+ 'validation' => Proc.new { |token| token =~ /\A[ABCDEF0-9]+\Z/ || token.empty? },
+ 'sensitive' => true
+ },
+ 'server_gslt' => {
+ 'default' => '',
+ 'type' => :argument,
+ 'validation' => Proc.new { |token| token =~ /\A[ABCDEF0-9]+\Z/ || token.empty? },
+ 'sensitive' => true
+ },
+ 'server_custom_server_args' => {
+ 'default' => '',
+ 'validation' => Proc.new { true }
+ },
+ 'server_custom_travel_args' => {
+ 'default' => '',
+ 'validation' => Proc.new { true }
+ },
+ 'hang_recovery' => {
+ 'default' => 'true',
+ 'validation' => Proc.new { |val| ['true', 'false'].include? val }
+ },
+ 'query_recovery' => {
+ 'default' => 'true',
+ 'validation' => Proc.new { |val| ['true', 'false'].include? val }
+ },
+ 'join_message' => {
+ 'default' => 'Welcome, ${player_name}!',
+ 'validation' => Proc.new { true }
+ },
+ 'leave_message' => {
+ 'default' => 'See you later, ${player_name}!',
+ 'validation' => Proc.new { true }
+ },
+ 'admin_join_message' => {
+ 'default' => 'Welcome, ${player_name}! (admin)',
+ 'validation' => Proc.new { true }
+ },
+ 'admin_leave_message' => {
+ 'default' => 'See you later, ${player_name}! (admin)',
+ 'validation' => Proc.new { true }
+ }
+ }
+
+ def initialize
+ @users = load_user_config
+ @monitor_configs = load_monitor_configs
+ @server_configs = load_server_configs
+ @players = load_player_config
+ end
+
+ def self.valid_port?(port)
+ port = port.to_i
+ port >= 1 && port <= 65535
+ rescue
+ false
+ end
+
+ def self.sanitize_directory(directory_name)
+ directory_name.gsub(/[^ 0-9A-Za-z.\-'"]/, '')
+ end
+
+ def get_uuid(uuid=Sysrandom.uuid)
+ return uuid if @server_configs.nil? || @server_configs.empty?
+ while @server_configs.keys.include?(uuid)
+ uuid = Sysrandom.uuid
+ end
+ uuid
+ end
+
+ def get_default_config
+ defaults = {'id' => get_uuid}
+ CONFIG_VARIABLES.each { |k, v| defaults[k] = v['default'] }
+ defaults
+ end
+
+ def get_default_user_config
+ default_admin_user = User.new('admin', :host, password: BCrypt::Password.create('password').to_s, initial_password: 'password')
+ {
+ default_admin_user.id => default_admin_user
+ }
+ end
+
+ def load_user_config
+ @users = Oj.load(File.read(USERS_CONFIG_FILE))
+ @users = get_default_user_config if @users.to_s.strip.empty?
+ @users
+ rescue Errno::ENOENT
+ get_default_user_config
+ rescue => e
+ log "Failed to load user config from #{USERS_CONFIG_FILE}. Using default user config.", e
+ raise
+ end
+
+ def load_monitor_configs
+ @monitor_configs = Oj.load(File.read(MONITOR_CONFIGS_FILE))
+ @monitor_configs = {} if @monitor_configs.to_s.empty?
+ @monitor_configs.each { |_, config| config['id'] ||= get_uuid }
+ @monitor_configs
+ rescue Errno::ENOENT
+ {}
+ rescue => e
+ log "Failed to load monitor configs from #{MONITOR_CONFIGS_FILE}. Using empty config.", e
+ raise
+ end
+
+ def add_server_config(config)
+ @server_configs[config['id']] = config
+ end
+
+ def get_server_configs_skeleton
+ skeleton = {}
+ default_config = get_default_config
+ skeleton[default_config['id']] = default_config
+ end
+
+ def load_server_configs
+ @server_configs = Oj.load(File.read(SERVER_CONFIGS_FILE))
+ if @server_configs.nil? || @server_configs.empty?
+ @server_configs = {}
+ add_server_config(get_default_config)
+ end
+ # Ensure all keys are IDs, else convert automatically
+ @server_configs.select { |k, _| !k.is_uuid? }.keys.each do |key|
+ log "Replacing config key '#{key}' with an ID"
+ config = @server_configs[key]
+ id = config['id'] || get_uuid
+ config['id'] = id
+ @server_configs[id] = config
+ @server_configs.delete(key)
+ end
+ # Ensure all configs have latest config params
+ @server_configs.each do |id, config|
+ @server_configs[id] = get_default_config.merge(config)
+ end
+ init_server_config_files
+ @server_configs
+ rescue Errno::ENOENT
+ {'Default' => get_default_config}
+ rescue => e
+ log "Failed to load server configs from #{SERVER_CONFIGS_FILE}. Using empty config.", e
+ raise
+ end
+
+ def load_player_config
+ Oj.load(File.read(PLAYER_INFO_FILE))
+ rescue Errno::ENOENT
+ {}
+ rescue => e
+ log "Failed to load user config from #{USERS_CONFIG_FILE}. Using default user config.", e
+ raise
+ end
+
+ def write_user_config
+ log "Writing user config"
+ File.write(USERS_CONFIG_FILE, Oj.dump(@users, indent: 2))
+ end
+
+ def write_monitor_configs
+ log "Writing monitor configs"
+ File.write(MONITOR_CONFIGS_FILE, Oj.dump(@monitor_configs, indent: 2))
+ end
+
+ def create_server_config(incoming_config)
+ config_id = incoming_config['id']
+ config_name = incoming_config['server-config-name']
+ overwrote = false
+ outgoing_config = if config_id && @server_configs[config_id] && @server_configs[config_id]['server-config-name'] == config_name
+ overwrote = true
+ log "Writing server config: #{config_id} #{config_name}", level: :info
+ @server_configs[config_id] = get_default_config.merge(incoming_config)
+ else
+ new_config = get_default_config
+ config_id = new_config['id']
+ log "Creating server config: #{config_id} #{config_name}", level: :info
+ @server_configs[config_id] = get_default_config.merge(incoming_config)
+ .merge({
+ 'id' => config_id, # Ignore provided config ID
+ })
+ end
+ write_server_configs
+ init_server_config_files(config_id)
+ outgoing_config
+ end
+
+ def delete_server_config(config_id)
+ config = @server_configs[config_id]
+ config_name = config['server-config-name']
+ log "Deleting server config: #{config_id} #{config_name}", level: :info
+ @server_configs.delete config_id
+ CONFIG_FILES.reject { |f, _| [:bans_cfg, :admins_cfg].include?(f) }.values.each do |it|
+ local = ERB.new(it[:local_erb]).result(binding)
+ FileUtils.rm local
+ rescue Errno::ENOENT
+ rescue => e
+ log "Error while trying to delete file (#{local})", e
+ end
+ begin
+ dir = File.join(CONFIG_FILES_DIR, config_id)
+ FileUtils.rmdir dir
+ rescue Errno::ENOENT
+ rescue => e
+ log "Error while trying to delete directory (#{dir})", e
+ end
+ add_server_config(get_default_config) if @server_configs.empty?
+ write_server_configs
+ nil
+ end
+
+ def write_server_configs
+ log "Writing server configs"
+ File.write(SERVER_CONFIGS_FILE, Oj.dump(@server_configs, indent: 2))
+ nil
+ end
+
+ def write_player_info
+ log "Writing player info"
+ File.write(PLAYER_INFO_FILE, Oj.dump(@players, indent: 2))
+ nil
+ end
+
+ def write_config
+ write_user_config
+ write_monitor_configs
+ write_server_configs
+ write_player_info
+ true
+ rescue => e
+ log 'Failed to write config', e
+ false
+ end
+
+ def init_server_config_files(config_id=nil)
+ config_ids = @server_configs.values.map { |c| c['id'] }
+ config_ids = config_id.nil? ? config_ids : [config_id]
+ config_ids.reject{ |id| !config_ids.include?(id) }.each do |config_id|
+ CONFIG_FILES.values.each do |it|
+ [ERB.new(it[:local_erb]).result(binding), it[:actual], it[:actual_erb] && ERB.new(it[:actual_erb]).result(binding)].each do |path|
+ next if path.nil?
+ FileUtils.mkdir_p File.dirname path
+ FileUtils.touch path
+ end
+ end
+ end
+ nil
+ end
+
+ def apply_server_config_files(config)
+ config_id = config['id'] # Needed for bindings
+ # Apply values in case any in memory aren't in the file
+ apply_game_ini_local config
+ apply_engine_ini_local config
+
+ # Remove :delete files so that the server doesn't use any the user may have manually set outside of OHDAW
+ CONFIG_FILES.reject {|_,i| i[:delete].nil? }.values.each do |it|
+ local = ERB.new(it[:delete]).result(binding)
+ log "Deleting #{local}"
+ FileUtils.rm local
+ rescue Errno::ENOENT
+ rescue => e
+ log "Error while trying to delete file (#{local})", e
+ end
+
+ CONFIG_FILES.reject {|_,i| i[:actual].nil? && i[:actual_erb].nil? }.values.each do |it|
+ local = ERB.new(it[:local_erb]).result(binding) # relies on config_id
+ actual = it[:actual_erb] ? ERB.new(it[:actual_erb]).result(binding) : it[:actual] # relies on config_id for actual_erb if present
+ if local == actual
+ log "Skipping #{local} -> #{actual} (same file)"
+ return
+ end
+ log "Applying #{local} -> #{actual}"
+ FileUtils.cp local, actual
+ rescue => e
+ log "Error while trying to copy file (#{local} -> #{actual})", e
+ end
+ nil
+ end
+
+ def get_server_config_file_content(config_file, config_id)
+ local = ERB.new(CONFIG_FILES[config_file][:local_erb]).result(binding)
+ File.read(local)
+ end
+
+ def set(config_id, variable, value)
+ return [false, "Variable not in config: #{variable}"] unless CONFIG_VARIABLES.keys.include? variable
+ return [false, "Variable has no validation: #{variable}"] unless CONFIG_VARIABLES[variable]['validation'].respond_to?('call')
+ config = (@server_configs[config_id] = get_default_config.merge(@server_configs[config_id] || {}))
+ old_value = config[variable]
+ return [false, "Variable #{variable} is already set to #{CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : value}."] if value.to_s == old_value.to_s
+ status = false
+ msg = "Failed to set #{variable.inspect} to #{CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : value.inspect}"
+ old_value = old_value || CONFIG_VARIABLES[variable]['default'] || 'Unknown'
+
+ if CONFIG_VARIABLES[variable]['validation'].call(value)
+ log "Value passed validation: #{CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : value.inspect}"
+ config[variable] = value
+ write_server_configs
+ apply_game_ini_local(@server_configs[config_id]) if CONFIG_VARIABLES[variable]['type'] == :game_ini
+ apply_engine_ini_local(@server_configs[config_id]) if CONFIG_VARIABLES[variable]['type'] == :engine_ini
+ status = true
+ else
+ msg = "Value failed validation for variable #{variable.inspect}: #{CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : value.inspect}"
+ log msg, level: :warn
+ end
+
+ return [status, msg, CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : config[variable], CONFIG_VARIABLES[variable]['sensitive'] ? '[REDACTED]' : old_value]
+ end
+
+ def get(config_id, variable)
+ @server_configs[config_id][variable]
+ end
+
+ def get_log_file(config_id)
+ log_file = File.join(SERVER_LOG_DIR, "#{config_id}.log")
+ FileUtils.mkdir_p File.dirname log_file
+ begin
+ FileUtils.touch log_file
+ rescue Errno::EACCES
+ log "Unable to access log file: #{log_file}"
+ end
+ log_file
+ end
+
+ def get_scenario(map, mode, side)
+ filtered_map = MAPMAP[map] || map
+ "Scenario_#{filtered_map}_#{mode}#{'_' << side if ['Checkpoint', 'Push'].include?(mode)}"
+ end
+
+ def get_additional_travel_args(config)
+ # Split using regex for unquoted strings
+ config['server_custom_travel_args'].dup.to_s.split(/[ ]+(?=[^"]*(?:"[^"]*"[^"]*)*$)/)
+ end
+
+ def get_query_string(config, map: nil, side: nil, game_mode: nil, max_players: nil, password: nil)
+ map = config['server_default_map'] == 'Random' ? MAPMAP.keys.sample : config['server_default_map'].dup if map.nil?
+ game_mode = config['server_game_mode'] == 'Random' ? GAME_MODES.sample : config['server_game_mode'].dup if game_mode.nil?
+ max_players ||= config['server_max_players'].dup
+ password ||= config['server_password'].dup
+ query = map.dup
+ query << "?MaxPlayers=#{max_players}"
+ query << "?Game=#{game_mode}" unless game_mode == 'None'
+ query << "?Password=#{password}" unless password.empty?
+ get_additional_travel_args(config).each do |arg|
+ query << arg
+ end
+ query
+ end
+
+ def get_additional_server_args(config)
+ # Split using regex for unquoted strings
+ config['server_custom_server_args'].dup.split(/[ ]+(?=[^"]*(?:"[^"]*"[^"]*)*$)/)
+ end
+
+ def get_server_arguments(config)
+ arguments = []
+ map = config['server_default_map'].dup
+ starting_map = map == 'Random' ? MAPMAP.keys.sample : map
+ starting_map = MAPMAP.keys.include?(starting_map) ? starting_map : (MAPMAP.keys.include?(MAPMAP.key(starting_map)) ? MAPMAP.key(starting_map) : MAPMAP.keys.first )
+ arguments.push(
+ get_query_string(config, map: starting_map),
+ "-ServerName=#{config['server_name']}",
+ "-MaxPlayers=#{config['server_max_players']}",
+ "-Port=#{config['server_game_port']}",
+ "-QueryPort=#{config['server_query_port']}",
+ "-log=#{config['id']}.log",
+ "-LogCmds=LogGameplayEvents Log",
+ "-LOCALLOGTIMES",
+ "-AdminList=Admins_#{config['id']}.cfg",
+ "-MapCycle=MapCycle_#{config['id']}.cfg"
+ )
+ custom_args = get_additional_server_args(config)
+ arguments.concat(custom_args)
+ arguments
+ end
+
+ def apply_game_ini_local(config)
+ # config_id needed in binding
+ config_id = config['id']
+ apply_ini(config, ERB.new(CONFIG_FILES[:game_ini][:local_erb]).result(binding))
+ end
+
+ def apply_engine_ini_local(config)
+ # config_id needed in binding
+ config_id = config['id']
+ apply_ini(config, ERB.new(CONFIG_FILES[:engine_ini][:local_erb]).result(binding))
+ end
+
+ def apply_ini(config, path)
+ type = File.basename(path).downcase.sub('.', '_').to_sym
+ log "Applying #{type} values to #{path}"
+ ini = IniFile.load(path)
+ changed = false
+ CONFIG_VARIABLES.select { |_, metadata| metadata['type'] == type }.each do |option, metadata|
+ next if (metadata['getter'].call(ini).to_s) == config[option].to_s
+ changed = true
+ log "Applying #{option}"
+ metadata['setter'].call(ini, config[option])
+ end
+ if changed
+ log "Saving #{type}"
+ ini.save
+ else
+ log "No #{type} changes"
+ end
+ rescue => e
+ log "Error while applying #{type} settings", e
+ raise
+ end
+end
diff --git a/admin-interface/lib/daemon.rb b/admin-interface/lib/daemon.rb
new file mode 100644
index 0000000..e209d36
--- /dev/null
+++ b/admin-interface/lib/daemon.rb
@@ -0,0 +1,369 @@
+require 'file-tail'
+require_relative 'rcon-client'
+require_relative 'server-monitor'
+require_relative 'server-updater'
+require_relative 'subprocess'
+
+include Process
+
+class ServerDaemon
+ attr_accessor :config
+ attr_accessor :executable
+ attr_accessor :server_root_dir
+ attr_accessor :arguments
+ attr_accessor :rcon_ip
+ attr_accessor :rcon_port
+ attr_accessor :rcon_pass
+ attr_accessor :player_feed
+ attr_accessor :steam_api_key
+ attr_reader :frozen_config
+ attr_reader :name
+ attr_reader :id
+ attr_reader :active_game_port
+ attr_reader :active_rcon_port
+ attr_reader :active_query_port
+ attr_reader :active_rcon_pass
+ attr_reader :buffer
+ attr_reader :rcon_buffer
+ attr_reader :chat_buffer
+ attr_reader :rcon_client
+ attr_reader :game_pid
+ attr_reader :threads
+ attr_reader :monitor
+ attr_reader :log_file
+ attr_reader :rcon_listening
+
+ def initialize(config, daemons, mutex, rcon_client, server_buffer, rcon_buffer, chat_buffer, steam_api_key: '')
+ @config = config
+ @name = @config['server-config-name']
+ @id = @config['id']
+ @daemons = daemons
+ @daemons_mutex = mutex
+ @monitor_mutex = Mutex.new
+ @rcon_ip = '127.0.0.1'
+ @buffer = server_buffer
+ @rcon_buffer = rcon_buffer
+ @chat_buffer = chat_buffer
+ @rcon_client = rcon_client
+ @rcon_listening = false
+ @game_pid = nil
+ @monitor = nil
+ @threads = {}
+ @buffer[:persistent] = true
+ @rcon_buffer[:persistent] = true
+ @chat_buffer[:persistent] = true
+ @chat_buffer[:filters] = [
+ Proc.new do |line|
+ line.gsub!(/\x1b\[[0-9;]*m/, '') # Remove color codes
+ line.gsub!(/^(\d{4}\/\d{2}\/\d{2}) (\d{2}:\d{2}:\d{2}).*TX >>\) say/, '\1 \2 ADMIN:') # Cut down RCON messages (TX)
+ line.gsub!(/^\[(\d{4})\.(\d{2})\.(\d{2})-(\d{2}).(\d{2}).(\d{2}).*LogChat: Display: /, '\1/\2/\3 \4:\5:\6 ') # Cut down server log messages (RX)
+ end
+ ]
+ @buffer[:filters] = [
+ Proc.new do |line|
+ line.gsub!(/\x1b\[[0-9;]*m/, '') # Remove color codes
+ line.prepend "#{get_server_id} | "
+ end
+ ]
+ @rcon_buffer[:filters] = [
+ Proc.new { |line| line.prepend "#{get_server_id} | " }
+ ]
+ @log_file = nil
+ @player_feed = []
+ @admin_ids = nil
+ @steam_api_key = steam_api_key
+ @exit_requested = false
+ start_daemon
+ log "Daemon initialized"
+ end
+
+ def log(message, exception=nil, level: nil)
+ super("#{get_server_id} | #{message}", exception, level: level) # Call the log function created by logger
+ end
+
+ def get_server_id
+ conf = @frozen_config || @config
+ "[PID #{@game_pid || '(N/A)'} Ports #{[conf['server_game_port'], conf['server_query_port'], conf['server_rcon_port']].join(',')}]"
+ end
+
+ def server_running?
+ if @game_pid.nil?
+ return false
+ end
+ Process.kill(0, @game_pid)
+ true
+ rescue Errno::ESRCH
+ false
+ end
+
+ def do_pre_update_warning(sleep_length: 5)
+ log "Sending restart warning to server", level: :info
+ message = 'say This server is restarting in 5 seconds to apply a new server update.'
+ message << ' This may take some time.' if WINDOWS # Since we have to stop the server before downloading the update
+ do_blast_message(message)
+ sleep sleep_length
+ rescue => e
+ log "Error while trying to message players", e
+ end
+
+ def do_blast_message(message, amount: 3, interval: 0.2)
+ amount.times do |i|
+ do_send_rcon message
+ sleep interval unless i + 1 == amount
+ end
+ end
+
+ def is_ohd_admin?(unique_id)
+ @admin_ids.include?(unique_id.to_s)
+ end
+
+ def do_send_rcon(command, host: nil, port: nil, pass: nil, buffer: nil, outcome_buffer: nil, no_rx: false)
+ host ||= @rcon_ip
+ port ||= @active_rcon_port || @config['server_rcon_port']
+ port ||= @active_rcon_pass
+ pass ||= @active_rcon_pass || @config['server_rcon_password']
+ buffer ||= @rcon_buffer
+ outcome_buffer ||= buffer
+ log "Calling RCON client for command: #{command}"
+ @rcon_client.send(host, port, pass, command, buffer: buffer, outcome_buffer: outcome_buffer, no_rx: no_rx)
+ end
+
+ def do_start_server
+ @exit_requested = false
+ # log "start daemons mutex wait"
+ @daemons_mutex.synchronize do
+ # log "start daemons mutex start"
+ return "Exiting" if @exit_requested
+ if server_running? || (@threads[:game_server] && @threads[:game_server].alive?)
+ "Server is already running. PID: #{@game_pid}"
+ else
+ new_game_port = @config['server_game_port']
+ if @active_game_port && @active_game_port != new_game_port
+ # We need to move the daemon for future access
+ log "Moving daemon #{@active_game_port} -> #{new_game_port}"
+ former_tenant = @daemons[new_game_port]
+ if former_tenant
+ log "Stopping daemon #{former_tenant.name} using desired game port #{new_game_port}", level: :warn
+ former_tenant.implode
+ end
+ @daemons[new_game_port] = @daemons.delete @active_game_port
+ end
+ log "Starting server", level: :info
+ @server_started = false
+ @server_failed = false
+ @game_pid = nil
+ @threads[:game_server] = get_game_server_thread
+ # Keep the mutex lock until we see RCON listening or the server fails to start
+ sleep 0.1 until (@game_pid && @rcon_listening) || @server_failed
+ @game_pid ? "Server is starting. PID: #{@game_pid}" : "Server failed to start!"
+ end
+ # log "start daemons mutex end"
+ end
+ # log "start daemons mutex clear"
+ end
+
+ def do_restart_server
+ log "Restarting server", level: :info
+ do_stop_server
+ do_start_server
+ end
+
+ def do_stop_server
+ @exit_requested = true
+ # log "stop daemons mutex wait"
+ @daemons_mutex.synchronize do
+ # log "stop daemons mutex start"
+ return 'Server not running.' unless server_running?
+ log "Stopping server", level: :info
+ # No need to do anything besides remove it from monitoring
+ # We want the signal to be sent to the thread's subprocess
+ # so that the thread has time to set the status/message in the buffer
+ @server_thread_exited = false
+ @threads.delete(:game_server)
+ msg = kill_server_process
+ log "Waiting for server thread to exit and clean up"
+ sleep 0.2 until @server_thread_exited || @server_failed
+ log "Server thread #{@server_failed ? "failed" : "exited and cleaned up"}"
+ msg
+ # log "stop daemons mutex end"
+ end
+ # log "stop daemons mutex clear"
+ end
+
+ def implode
+ log "Daemon for server #{@name} (#{@config['id']}) imploding", level: :info
+ @exit_requested = true
+ @game_pid = nil
+ @buffer.reset
+ @buffer = nil
+ @rcon_buffer.reset
+ @rcon_buffer = nil
+ Thread.new { @monitor.stop if @monitor }
+ game_server_thread = @threads.delete :game_server
+ kill_server_process
+ game_server_thread.join unless game_server_thread.nil?
+ @threads.keys.each do |thread_name|
+ thread = @threads.delete thread_name
+ thread.kill if thread.respond_to? :kill
+ end
+ end
+
+ def kill_server_process(signal: nil)
+ signal = 'KILL' if signal.nil? # TERM can hang shutting down EAC. KILL doesn't, but might not disconnect players (instead they time out).
+ return "Unable to send #{signal} (#{Signal.list[signal]}) signal to server; no known PID!" unless @game_pid
+ return "Server isn't running!" unless server_running?
+ begin
+ Process.kill(signal, @game_pid)
+ rescue Errno::ESRCH
+ end
+ msg = "Sent #{signal} (#{Signal.list[signal]}) signal to PID #{@game_pid}."
+ log msg, level: :info
+ msg
+ end
+
+ def create_monitor
+ @monitor_mutex.synchronize do
+ if @monitor.nil?
+ Thread.new { @monitor = ServerMonitor.new('127.0.0.1', @active_query_port, @active_rcon_port, @active_rcon_pass, name: @name, rcon_buffer: @rcon_buffer, interval: 5, daemon_handle: self) }
+ end
+ sleep 0.5 while @monitor.nil? && !(@exit_requested)
+ end
+ end
+
+ def run_game_server
+ log "Applying config"
+ @frozen_config = @config.dup
+ $config_handler.apply_server_config_files @frozen_config
+ @admin_ids = $config_handler.get_server_config_file_content(:admins_cfg, @frozen_config['id']).split("\n").map { |l| l[/\d{17}/] }.compact
+ executable = BINARY
+ arguments = $config_handler.get_server_arguments(@frozen_config)
+ @active_game_port = @frozen_config['server_game_port']
+ @active_query_port = @frozen_config['server_query_port']
+ @active_rcon_port = @frozen_config['server_rcon_port']
+ @active_rcon_pass = @frozen_config['server_rcon_password']
+ @log_file = $config_handler.get_log_file(@frozen_config['id'])
+ log "Spawning game process: #{[executable, *arguments].inspect}", level: :info
+ SubprocessRunner.run(
+ [executable, *arguments],
+ buffer: @buffer,
+ pty: false,
+ no_prefix: true,
+ formatter: Proc.new { |output, _| WINDOWS ? "#{datetime} | #{output.chomp}" : output.chomp } # Windows doesn't have the timestamp, so we'll add our own to make it look nice.
+ ) do |pid|
+ @game_pid = pid
+ log "Game process spawned. Starting self-monitoring after detecting RCON listening message.", level: :info
+ @rcon_tail_thread = Thread.new do
+ # last_modified_log_time = File.mtime(Dir[File.join(SERVER_LOG_DIR, '*.log')].sort_by{|f| File.mtime(f) }.last).to_i rescue 0
+ # other_used_logs = @daemons.map { |_, daemon| daemon.log_file }
+ # @rcon_buffer[:data] << "[PID: #{@game_pid} Game Port: #{@active_game_port}] Waiting to detect log file in use"
+ log "Waiting to ensure log file is in use"
+ earlier = Time.now.to_i
+ loop do
+ last_updated = File.mtime(@log_file).to_i
+ break if last_updated > earlier
+ sleep 0.5
+ end
+ log "Log file is in use. Proceeding with log tailing."
+ begin
+ File.open(@log_file) do |log|
+ log.extend(File::Tail)
+ log.interval = 0.1
+ log.backward(0)
+ last_line_was_rcon = false
+ log.tail do |line|
+ Thread.exit if @exit_requested
+ next if line.nil?
+ if line.include? 'LogRCON'
+ last_line_was_rcon = true
+ if line[/LogRcon: Error: Failed to create TcpListener at .* for rcon support/]
+ log "RCON failed to initialize: #{line}", level: :warn
+ kill_server_process
+ elsif !@rcon_listening && line.include?('LogRCON: RCON server listening') && @monitor.nil?
+ @rcon_listening = true
+ log "RCON listening - Ending server start/stop lock", level: :info
+ create_monitor
+ Thread.new do
+ sleep 0.5 until @exit_requested || (@monitor && @monitor.all_green?)
+ log "Server is ready (RCON and Query connected)", level: :info
+ @server_started = true
+ end
+ end
+ elsif last_line_was_rcon
+ if line =~ /^\[\d{4}\.\d{2}\.\d{2}-\d{2}\.\d{2}\.\d{2}:/ || line =~ /^Log/
+ last_line_was_rcon = false
+ end
+ end
+ if line.include? 'LogChat'
+ @chat_buffer[:filters].each { |filter| filter.call(line) } # Remove color codes; add server ID
+ @chat_buffer.synchronize { @chat_buffer.push line.chomp }
+ elsif last_line_was_rcon
+ @buffer[:filters].each { |filter| filter.call(line) } # Remove color codes; add server ID
+ @rcon_buffer.synchronize { @rcon_buffer.push line.chomp }
+ end
+ end
+ end
+ rescue EOFError
+ rescue => e
+ log "Error in RCON tail thread!", e
+ raise e
+ ensure
+ log "RCON tail thread stopped"
+ end
+ end
+ log "RCON tailing thread started."
+ end
+ log 'Game process exited', level: :info
+ ensure
+ @rcon_tail_thread.kill rescue nil
+ end
+
+ def get_game_server_thread
+ Thread.new do
+ if !@exit_requested
+ run_game_server
+ end
+ rescue => e
+ @server_failed = true
+ log "Game server failed", e
+ Thread.new { @monitor.stop if @monitor }
+ @threads.delete :game_server unless @server_started # If we can't even start the server, don't keep trying
+ kill_server_process
+ ensure
+ @server_thread_exited = true
+ begin
+ @monitor.stop unless @monitor.nil?
+ @monitor = nil
+ @game_pid = nil
+ @log_file = nil
+ @rcon_listening = false
+ socket = @rcon_client.sockets["127.0.0.1:#{@active_rcon_port}"]
+ @rcon_client.delete_socket(socket) unless socket.nil?
+ rescue => e
+ log "Error while cleaning up game server thread", e
+ end
+ log "Game server thread exiting"
+ end
+ end
+
+ def start_daemon(thread_check_interval: 2)
+ @daemon_thread = Thread.new do
+ while true
+ while @threads.empty? || @threads.values.all?(&:alive?)
+ sleep thread_check_interval
+ end
+ dead_threads = @threads.select { |_, t| !t.alive? }.keys
+ log "Dead daemon thread(s) detected: #{dead_threads.join(' ')}"
+ dead_threads.each do |key|
+ if @threads[key]
+ log "Starting #{key} thread", level: :info
+ @threads[key] = public_send("get_#{key.to_s}_thread")
+ end
+ end
+ sleep thread_check_interval
+ end
+ rescue => e
+ log "Error in daemon's self-monitoring thread", e
+ raise
+ end
+ end
+end
diff --git a/admin-interface/lib/logger.rb b/admin-interface/lib/logger.rb
new file mode 100644
index 0000000..989c655
--- /dev/null
+++ b/admin-interface/lib/logger.rb
@@ -0,0 +1,91 @@
+# encoding: UTF-8
+
+require 'logger'
+
+Dir.mkdir 'log' unless Dir.exist? 'log'
+LOG_FILE = File.join('log', 'ohd-admin-wrapper.log')
+SEVERITY_JUSTIFY = Logger::Severity.constants.map(&:length).max # [:DEBUG, :INFO, :WARN, :ERROR, :FATAL, :UNKNOWN]
+DATETIME_FORMAT = "%Y/%m/%d %H:%M:%S.%L %z"
+USER_HOME ||= ENV['HOME']
+$caller_justification = 25
+
+class MultiTargetLogger
+ attr_reader :level
+ attr_reader :loggers
+
+ def initialize(loggers)
+ @level = Logger::DEBUG
+ loggers.each do |name, logger|
+ logger.formatter = proc do |severity, datetime, progname, message|
+ "#{datetime} | #{severity.ljust SEVERITY_JUSTIFY}#{" | #{progname}" if progname} | #{message}\n"
+ end
+ logger.level = @level
+ end
+ @loggers = loggers
+ end
+
+ Logger::Severity.constants.each do |severity|
+ severity = severity.downcase
+ define_method(severity) do |*args|
+ # Suppress FATAL error log on intentional exit
+ if severity == :fatal
+ severity = :debug if args.first.is_a? SystemExit
+ end
+ # Suppress SSL error for self-generated certificate
+ if args.first.to_s.end_with? ': sslv3 alert certificate unknown'
+ severity = :debug
+ args[0] = args.first.to_s
+ end
+ @loggers.each { |_, logger| logger.send(severity, *args) if Logger.const_get(severity.to_s.upcase) >= logger.level }
+ if args.first.is_a? Exception
+ @loggers.each { |_, logger| logger.send(:error, "Backtrace:#{args.first.backtrace.join("\n ").prepend("\n ")}") if Logger.const_get(severity.to_s.upcase) >= logger.level }
+ end
+
+ nil
+ end
+ define_method("#{severity}?") do |*args|
+ Logger.const_get(severity.to_s.upcase) >= @level
+ end
+ end
+
+ def puts(*args)
+ log(:debug, *args)
+ end
+
+ def log(severity, *args)
+ severity = severity.downcase
+ @loggers.each { |_, logger| logger.send(severity, *args) if Logger.const_get(severity.to_s.upcase) >= logger.level }
+ nil
+ end
+
+ def threshold(severity=nil, logger: :stdout)
+ return @loggers[:stdout].level if severity.nil?
+ level = severity.is_a?(Integer) ? severity : Logger.const_get(severity.to_s.upcase)
+ @loggers[:stdout].level = level
+ prev_level = @level
+ @level = level
+ prev_level
+ end
+
+ def <<(message, severity: :debug)
+ log(severity, message.chomp) # Thanks, WEBrick AccessLog...
+ nil
+ end
+end
+
+LOGGER_FILE = Logger.new(LOG_FILE, 10, 1024 * 5000)
+LOGGER_STDOUT = Logger.new($stdout)
+LOGGER = MultiTargetLogger.new({stdout: LOGGER_STDOUT, file: LOGGER_FILE})
+
+def datetime # Helper function used elsewhere
+ Time.now.strftime(DATETIME_FORMAT)
+end
+
+def log(message, exception=nil, level: nil)
+ unless exception.nil?
+ message = message + " | Exception occurred (#{exception.class}): #{exception.message}\n #{exception.backtrace.to_a.map { |l| l.sub USER_HOME, '~'}.join("\n ")}"
+ level ||= :error
+ end
+ level ||= :debug
+ LOGGER.log level, message
+end
diff --git a/admin-interface/lib/rcon-client.rb b/admin-interface/lib/rcon-client.rb
new file mode 100644
index 0000000..17003cf
--- /dev/null
+++ b/admin-interface/lib/rcon-client.rb
@@ -0,0 +1,396 @@
+#!/usr/bin/env ruby
+
+require 'socket'
+require 'monitor'
+require_relative '../ext/string'
+require_relative 'logger'
+
+# https://developer.valvesoftware.com/wiki/Source_RCON_Protocol
+# https://apidock.com/ruby/Array/pack
+# https://apidock.com/ruby/String/unpack
+
+# Packet structure:
+# Field | Type | Value
+# ----------------------------------------------------------------------
+# Size | 32-bit little-endian Signed Integer (l<) | Varies, see below.
+# ID | 32-bit little-endian Signed Integer (l<) | Varies, see below.
+# Type | 32-bit little-endian Signed Integer (l<) | Varies, see below.
+# Body | Null-terminated ASCII String (Z*) | Varies, see below.
+# Empty String | Null-terminated ASCII String (Z*) | 0x00
+
+# Packet size:
+# The packet size field is a 32-bit little endian integer, representing the length of the request in bytes.
+# Note that the packet size field itself is not included when determining the size of the packet, so the value
+# of this field is always 4 less than the packet's actual length. The minimum possible value for packet size is 10:
+
+# Size | Containing
+# 4 Bytes | ID Field
+# 4 Bytes | Type Field
+# At least 1 Byte | Packet body (potentially empty)
+# 1 Bytes | Empty string terminator
+
+class NoTCPResponseError < StandardError
+ def initialize(host=nil)
+ super "Could not read a TCP response from server#{host ? " #{host}" : '!'}"
+ end
+end
+
+class TCPSocketError < StandardError
+ def initialize(msg=nil)
+ super msg
+ end
+end
+
+class RconSocket < Socket
+ attr_reader :mutex
+
+ def initialize(ip, port, password, timeout=3)
+ @mutex = Mutex.new
+ @rcon_ip = ip
+ @rcon_port = port
+ @rcon_password = password
+
+ addr = Socket.getaddrinfo(ip, nil)
+ sockaddr = Socket.pack_sockaddr_in(port, addr[0][3])
+
+ super(Socket.const_get(addr[0][0]), Socket::SOCK_STREAM, 0).tap do |socket|
+ socket.setsockopt(Socket::IPPROTO_TCP, Socket::TCP_NODELAY, 1)
+ begin
+ socket.connect_nonblock(sockaddr)
+ rescue IO::WaitWritable
+ if IO.select(nil, [socket], nil, timeout)
+ begin
+ socket.connect_nonblock(sockaddr)
+ rescue Errno::EISCONN
+ # Connected
+ rescue
+ socket.close
+ raise
+ end
+ else
+ # IO.select returns nil when the socket is not ready before timeout
+ # seconds have elapsed
+ socket.close
+ raise "Connection to #{socket.remote_host} timed out."
+ end
+ end
+ end
+ end
+
+ def info
+ return @rcon_ip, @rcon_port, @rcon_password
+ end
+
+ def remote_host
+ "#{@rcon_ip}:#{@rcon_port}"
+ end
+end
+
+# Handles holding open RCON connections to each server and sending commands
+class RconClient
+ attr_reader :sockets
+
+ @@pack_string = 'l#{e.backtrace.map{|l| CGI.escapeHTML(l.gsub(USER_HOME, '~')) }.join('
')}" + else + CGI.escapeHTML "An error occurred (#{e.class.to_s}). Please view the logs or contact the maintainer." + end + end + + before do + env['rack.logger'] = LOGGER + env['rack.errors'] = LOGGER + @user = $config_handler.users[session[:user_id]] if session[:user_id] + check_prereqs unless @@prereqs_complete + request.body.rewind + # body = request.body.read + # request.body.rewind # Be kind + timestamp = datetime + message = "IP: #{request.ip} | #{"User: #{@user.name} | " if @user}Request: #{request.request_method}#{' ' << request.script_name unless request.script_name.strip.empty? } #{request.path_info}#{'?' << request.query_string unless request.query_string.strip.empty?}" # #{(' | Body: ' << body) unless body.strip.empty?}" + log message + @@wrapper_user_log.push "#{timestamp} | #{message}" if request.request_method != 'GET' + @@wrapper_connection_log.push "#{timestamp} | #{message} | Agent: #{request.user_agent}" + nil + end + + # after do + # log "Responding: #{response.body}" + # end + + %i(get post put).each do |method| + send method, '/error' do + status params['status'] + params['message'] + end + end + + get '/confirm' do + @yes = CGI.unescape params[:yes] + @no = CGI.unescape params[:no] + @title = CGI.unescape params[:title] + @body = CGI.unescape params[:body] + @input_label = CGI.unescape params[:input_label] + erb :confirm + end + + get %r{/login(/)?(.*)} do + @destination = '/' + params['captures'].last.sub('login', '') + @destination = nil if @destination == '/' + erb :login + end + + post '/login' do + data = Oj.load(request.body.read) + request.body.rewind + user = data['user'] + password = data['pass'] + destination = data['destination'] + if user && password + known_user = $config_handler.users.values.select { |u| u.name == user }.first + if known_user && known_user.password == password # BCrypt::Password == string comparison + message = "IP: #{request.ip} | #{known_user.name} [#{known_user.role}] logged in (#{known_user.id})" + log message, level: :info + @@wrapper_user_log << "#{datetime} | #{message}" + session[:user_id] = known_user.id + return known_user.first_login? ? "/change-password#{destination}" : destination + end + end + status 401 + "Failed to log in." + end + + get '/logout' do + user = $config_handler.users[session[:user_id]] + halt 400, 'Not logged in' if user.nil? + message = "IP: #{request.ip} | #{user.name} [#{user.role}] logged out (#{user.id})" + log message, level: :info + @@wrapper_user_log << "#{datetime} | #{message}" + session[:user_id] = nil + redirect '/login' + end + + get '/status', auth: :user do + @daemons = @@daemons + @monitors = @@monitors + erb :'server-status', layout: :'layout-main' do + erb :'server-list' + end + end + + get '/server-list', auth: :user do + @daemons = @@daemons + @monitors = @@monitors + erb :'server-list' + end + + get '/server-daemons', auth: :admin do + @daemons = @@daemons + erb :'server-daemons', layout: :'layout-main' + end + + get '/change-password(/:destination)?', auth: :user do + @destination = "/#{params['destination']}".sub('/change-password', '') + @first_login = !@user.initial_password.nil? + erb :'change-password', layout: :'layout-main' + end + + post '/change-password', auth: :user do + data = Oj.load(request.body.read) + request.body.rewind + password = data['pass'] + destination = "#{data['destination']}" + if password.strip.empty? || @user.password_matches?(password) + status 400 + return "Invalid password. The password must be new and not blank!" + end + @user.password = password + $config_handler.write_user_config + destination + end + + get '/wrapper-log', auth: :host do + @user_log_id = @@wrapper_user_log[:uuid] + @conn_log_id = @@wrapper_connection_log[:uuid] + erb :'wrapper-log', layout: :'layout-main' + end + + get '/wrapper-config', auth: :host do + @config = @@config + @lan_access_bind_ip = @@lan_access_bind_ip + @wrapper_version = @@loaded_wrapper_version + erb :'wrapper-config', layout: :'layout-main' + end + + put '/wrapper-config/:action', auth: :host do + wrapper_config = @@config + variable = params['variable'] + value = params['value'] + value = value == 'true' if ['true', 'false'].include?(value) + if params['action'] == 'set' + begin + if wrapper_config.keys.include?(variable) + sensitive = ['admin_interface_session_secret', 'steam_api_key'].include? variable + old = sensitive ? '[REDACTED]' : wrapper_config[variable] + wrapper_config[variable] = value + value = sensitive ? '[REDACTED]' : wrapper_config[variable] + self.class.save_webapp_config(wrapper_config) + @@daemons.values.each { |d| d.steam_api_key = wrapper_config[variable] } if variable == 'steam_api_key' + "Changed #{variable}: #{old} => #{value}" + else + status 400 + msg + end + rescue => e + msg = "Failed to set #{variable + ' -> ' + variable == 'admin_interface_session_secret' ? '[REDACTED]' : value}" + log msg, e + status 400 + "#{msg} | #{e.class}: #{e.message}" + end + elsif params['action'] == 'get' + value = wrapper_config[variable] + if value.nil? + status 404 + "Could not find value for #{variable}" + else + value.to_s + end + else + status 400 + 'Unknown action' + end + end + + get '/wrapper-users-list', auth: :host do + @users = $config_handler.users + erb :'wrapper-users-list' + end + + get '/wrapper-users', auth: :host do + @users = $config_handler.users + erb :'wrapper-users', layout: :'layout-main' do + erb :'wrapper-users-list' + end + end + + post '/wrapper-users/:action', auth: :host do + action = params['action'] + data = Oj.load(request.body.read) + request.body.rewind + id = data['id'] + name = data['name'] + role = data['role'].to_s.downcase.to_sym + + case action + when 'create' + if name.to_s.empty? || role.to_s.empty? + status 400 + 'Missing name/role' + elsif $config_handler.users.values.select { |u| u.name == name }.size > 0 + status 400 + 'A user with that name already exists' + else + user = User.new(name, role) + $config_handler.users[user.id] = user + $config_handler.write_user_config + 'User created' + end + when 'delete' + if id.to_s.empty? + status 400 + 'Missing ID' + elsif $config_handler.users[id].nil? + status 400 + 'No such user' + elsif $config_handler.users[id].role == :host && $config_handler.users.values.select { |u| u.role == :host }.size == 1 + status 400 + 'Cannot delete last Host' + else + $config_handler.users.delete id + $config_handler.write_user_config + 'User deleted' + end + when 'save' + if name.to_s.empty? || role.to_s.empty? + status 400 + 'Missing name/role' + elsif $config_handler.users[id].nil? + status 400 + 'No such user' + elsif !User::ROLES.keys.include?(role) + status 400 + 'No such role' + elsif role != :host && $config_handler.users[id].role == :host && $config_handler.users.values.select { |u| u.role == :host }.size == 1 + status 400 + 'Cannot change role of last Host' + else + user = $config_handler.users[id] + if user.name == name && user.role == role + status 400 + 'Nothing was changed' + else + user.name = name + user.role = role + $config_handler.write_user_config + 'User saved' + end + end + else + status 400 + 'Unknown action' + end + end + + get '/pry', auth: :host do + @buffers = @@buffers + prev_threshold = LOGGER.threshold + Thread.new do + LOGGER.threshold :fatal # Turn down STDOUT logging + `stty echo` unless WINDOWS || !$stdout.isatty # Ensure we have echoing enabled; something from logging turns it off... + + # Stuff + + binding.pry + ensure + `stty echo` unless WINDOWS || !$stdout.isatty + LOGGER.threshold prev_threshold + end.join + nil + rescue Interrupt + end + + get '/threads/:id', auth: :user do + daemon = @@daemons[params[:id]] + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + monitor = daemon.monitor rescue nil + @info = monitor.info rescue nil + erb :threads + end + + get '/players/:id', auth: :user do + daemon = @@daemons[params[:id]] + @monitor = daemon.monitor rescue @@monitors[params[:id]] + @info = @monitor.info rescue nil + erb :'players' + end + + get '/update-info', auth: :user do + @update_available, @old_build, @new_build = get_update_info + erb(:'update-info') + end + + post '/restart-wrapper', auth: :host do + @@daemons.each { |_, daemon| daemon.do_stop_server } + ENV['HOME'] = USER_HOME + # Exit with a particular code, to be handled by the start script + at_exit do + exit 2 + end + log "Restart invoked", level: :info + Thread.new do + log "Replacing the current process...", level: :info + sleep 0.2 + exit + end + '' + end + + post '/update-wrapper', auth: :host do + version = SelfUpdater.update_to_latest + @@config['wrapper_version'] = version + self.class.save_webapp_config @@config + "Updated to #{version}! Restart the wrapper to apply." + rescue => e + status 500 + e.message + end + + get '/', auth: :user do + @steamcmd_path, @game_server_path = check_prereqs + if is_admin? + redirect '/setup' unless @steamcmd_path && @game_server_path + redirect '/control' + else + redirect '/status' + end + end + + get '/about', auth: :user do + erb(:about, layout: :'layout-main') + end + + get '/setup', auth: :admin do + @steamcmd_path, @game_server_path = check_prereqs + @automatic_updates = @@config['automatic_updates'] + erb(:'server-setup', layout: :'layout-main') + end + + get '/maplist', auth: :admin do + @config = get_config(params['config-id']) + config_id = @config['id'] # Needed in ERB binding + erb(:maplist) + end + + get '/config', auth: :admin do + redirect '/setup' unless @@prereqs_complete + config_id = params['config-id'] + redirect_to_config(config_id, '/config') + active_config = get_active_config + # We _should_ always have a config, as config_handler should ensure that at least the Default config exists at all times + @config = $config_handler.server_configs[active_config['id']] || $config_handler.server_configs.values.last + unless @config + status 400 + return "Unable to find config" + end + config_id = @config['id'] + $config_handler.init_server_config_files config_id + @game_ini = File.read ERB.new(CONFIG_FILES[:game_ini][:local_erb]).result(binding) + @engine_ini = File.read ERB.new(CONFIG_FILES[:engine_ini][:local_erb]).result(binding) + @game_user_settings_ini = File.read ERB.new(CONFIG_FILES[:game_user_settings_ini][:local_erb]).result(binding) + @admins_cfg = File.read ERB.new(CONFIG_FILES[:admins_cfg][:local_erb]).result(binding) + @mapcycle_cfg = File.read ERB.new(CONFIG_FILES[:mapcycle_cfg][:local_erb]).result(binding) + @bans_cfg = File.read ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + @motd_txt = File.read ERB.new(CONFIG_FILES[:motd_txt][:local_erb]).result(binding) + @notes_txt = File.read ERB.new(CONFIG_FILES[:notes_txt][:local_erb]).result(binding) + erb(:'server-config', layout: :'layout-main') + end + + get '/control', auth: :admin do + redirect '/setup' unless @@prereqs_complete + config_id = params['config-id'] + redirect_to_config(config_id, '/control') + @config = get_active_config + @config_id = @config['id'] + daemon = @@daemons[@config_id] + if daemon + @config = daemon.frozen_config + end + @game_port = daemon.server_running? ? daemon.active_game_port : @config['server_game_port'] rescue @config['server_game_port'] + @rcon_port = daemon.server_running? ? daemon.active_rcon_port : @config['server_rcon_port'] rescue @config['server_rcon_port'] + @query_port = daemon.server_running? ? daemon.active_query_port : @config['server_query_port'] rescue @config['server_query_port'] + @server_status = get_server_status(@config_id) + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + @monitor = daemon.monitor rescue nil + @info = @monitor.info rescue nil + erb(:'server-control', layout: :'layout-main') + end + + get '/tools/:resource', auth: :admin do + @config = get_active_config + resource = params['resource'] + case resource + when 'rcon' + erb(:'rcon-tool', layout: :'layout-main') + when 'steamcmd' + redirect '/setup' unless steamcmd_installed? + @server_root_dir = SERVER_ROOT + erb(:'steamcmd-tool', layout: :'layout-main') + when 'monitor' + config = $config_handler.monitor_configs.first + @name, @config = config ? config : [@config['server-config-name'], @config] + erb(:'monitor-tool', layout: :'layout-main') + else + status 500 + "Unknown resource: #{resource}" + end + end + + post '/tools/:resource', auth: :admin do + resource = params['resource'] + body = request.body.read + options = Oj.load body, symbol_keys: true + request.body.rewind + case resource + when 'steamcmd' + uuid, buffer = self.class.create_buffer + Thread.new { run_steamcmd(options[:command].split("\n"), buffer: buffer) } + uuid + else + status 400 + "Unknown resource: #{resource}" + end + end + + get '/server-control-status' do + @config = get_active_config + @id = @config['id'] + daemon = @@daemons[@id] + if daemon + @config = daemon.frozen_config + log "/server-control-status working with daemon ID #{daemon.config['id']}" + else + log "/server-control-status Failed to get running daemon for config with id #{@id}" + end + @game_port = daemon.server_running? ? daemon.active_game_port : @config['server_game_port'] rescue @config['server_game_port'] + @rcon_port = daemon.server_running? ? daemon.active_rcon_port : @config['server_rcon_port'] rescue @config['server_rcon_port'] + @query_port = daemon.server_running? ? daemon.active_query_port : @config['server_query_port'] rescue @config['server_query_port'] + @server_status = get_server_status(@id) + @pid = daemon.game_pid if daemon + process = Sys::ProcTable.ps(pid: @pid) if @pid + threads = (WINDOWS ? process.thread_count : process.nlwp) if process + @threads = threads || 0 + @monitor = daemon.monitor rescue nil + @info = @monitor.info rescue nil + erb :'server-control-status' + end + + get '/server-status/:id', auth: :user do + id = params[:id] + get_server_status(id) + end + + get '/buffer/:uuid(/:bookmark)?', auth: :admin do # Sensitive information could be contained in these buffers (passwords, paths); admins only. + uuid = params['uuid'] + unless @@buffers.keys.include? uuid + status 400 + return "Unknown UUID: #{uuid}" + end + buffer = @@buffers[uuid] + @bookmark_uuid = params['bookmark'] + limit = buffer[:limit] + + buffer[:mutex].synchronize do # Synchronize with the writing thread to avoid mismatched indices when truncating, etc. + buffer[:iterator] = buffer[:iterator].nil? ? 1 : buffer[:iterator] + 1 + # log "Buffer [#{uuid}][#{buffer[:iterator]}] arrived with bookmark: [#{@bookmark_uuid}] -> #{buffer[:bookmarks][@bookmark_uuid]}" + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Bookmarks: #{buffer[:bookmarks]}" + if buffer[:bookmarks].length > 100 + log "Buffer has many bookmarks! Are that many clients connected? #{buffer[:uuid]} (#{buffer[:bookmarks].length} bookmarks)", level: :warn + log "Purging first 20 bookmarks", level: :warn + 20.times { buffer[:bookmarks].shift } + end + unless buffer[:bookmarks].keys.include?(@bookmark_uuid) || @bookmark_uuid.nil? + status 400 + return "Failed to tail log (Unknown Bookmark UUID: #{@bookmark_uuid})." + end + if buffer[:bookmarks][@bookmark_uuid] == -1 # Client read everything; ready for outcome + obj = { + :status => buffer[:status].dup, + :message => buffer[:message].dup + } + log "Responding status/message: #{obj[:status]} - #{obj[:message]}" + @@buffer_mutex.synchronize do + buffer[:bookmarks].delete @bookmark_uuid + @@buffers[uuid][:persistent] ? @@buffers[uuid].reset : @@buffers.delete(uuid) + end + return Oj.dump(obj, mode: :compat) + end + + start_index = buffer[:bookmarks][@bookmark_uuid] || 0 + start_index = 0 if start_index > buffer.size + # log "Buffer [#{uuid}][#{buffer[:iterator]}] start_index unexpectedly negative" if start_index < 0 + new_bookmark_uuid = Sysrandom.uuid + new_bookmark = (buffer.size - start_index > limit) ? (start_index + limit) : buffer.size + # log "Buffer - Calculated new_bookmark: #{new_bookmark}" + if new_bookmark == buffer.size && !buffer[:status].nil? + # Process is done. The next bookmark will be an indicator to send status and message next time + # as well as slice to the end of the array below + new_bookmark = -1 + end + if new_bookmark == start_index && @bookmark_uuid # No data will be given; have them request the same bookmark if we were given one + new_bookmark_uuid = @bookmark_uuid + else + buffer[:bookmarks][new_bookmark_uuid] = new_bookmark + buffer[:bookmarks].delete @bookmark_uuid unless @bookmark_uuid.nil? + end + # log "Buffer - Command done. Moved to : #{new_bookmark}" if new_bookmark == -1 + + data_response = start_index == new_bookmark ? [] : buffer[:data][start_index..new_bookmark] + response_object = { + :bookmark => new_bookmark_uuid, + :data => data_response + } + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Responding #{start_index}..#{new_bookmark}: #{response_object}" + # log "Buffer [#{uuid}][#{buffer[:iterator]}] Bookmarks: #{buffer[:bookmarks]}" + Oj.dump(response_object, mode: :compat) + end + end + + post '/control/server/:action(/:id)?', auth: :admin do + id = params['id'] + action = params['action'] + body = request.body.read + request.body.rewind + options = Oj.load body, symbol_keys: true + daemon = @@daemons[id] + if daemon.nil? && !['install', 'update', 'start'].include?(action) + status 400 + return "Could not find an active daemon for ID #{id}!" + end + validate = params['validate'].to_s == "true" + case action + when 'install' + uuid, buffer = self.class.create_buffer + log "Installing server" + Thread.new do + install_server(buffer, validate: validate) + end + log "Server install thread started. Returning UUID: #{uuid}" + uuid + when 'update' + uuid, buffer = self.class.create_buffer + log "Updating server" + Thread.new do + update_server(buffer, validate: validate) + end + log "Server update thread started. Returning UUID: #{uuid}" + uuid + when 'start' + config = $config_handler.server_configs.select{ |_, config| config['id'] == id }.first.last rescue nil + if config.nil? + status 400 + return "Unknown config ID: #{id}" + end + daemon = self.class.init_daemon config # Returns existing daemon if present + session[:active_daemon_uuid] = daemon.buffer[:uuid] + daemon.do_start_server + when 'stop' + body daemon.do_stop_server + session[:active_daemon_uuid] = nil if daemon.buffer[:uuid] == session[:active_daemon_uuid] + when 'delete' + @@daemons.delete(@@daemons.key daemon).implode + when 'restart' + daemon.config.merge!($config_handler.server_configs[daemon.id]) if $config_handler.server_configs[daemon.id] + if daemon.server_running? + daemon.do_restart_server + else + status 400 + "Server is not running" + end + when 'rcon' + begin + uuid, buffer = self.class.create_buffer + Thread.new { daemon.do_send_rcon(options[:command], host: options[:host] || '127.0.0.1', port: options[:port], pass: options[:pass], buffer: daemon.rcon_buffer, outcome_buffer: buffer) } + uuid + rescue => e + log "Error while sending RCON", e + status 500 + e.message + end + when 'chat' + begin + uuid, buffer = self.class.create_buffer + Thread.new { daemon.do_send_rcon("say #{options[:command]}", host: options[:host], port: options[:port], pass: options[:pass], buffer: daemon.chat_buffer, outcome_buffer: buffer, no_rx: true) } + uuid + rescue => e + log "Error while sending chat RCON", e + status 500 + e.message + end + else + status 400 + "Unknown action: #{action}" + end + end + + get '/config/file/:file', auth: :admin do + config = get_config(params['config-id']) + file_type = params['file'] + halt(400, "File type not found: #{file_type}") unless CONFIG_FILES[file_type.to_sym] + file = ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + halt(400, "File not found: #{file.sub(USER_HOME, '~')}") unless File.exist? file + File.read(file) + end + + post '/config/file', auth: :admin do + config_id = params['config_id'] + config = get_config(config_id) + file_type = params['file'] + halt(400, "File type not found: #{file_type}") unless CONFIG_FILES[file_type.to_sym] + file = ERB.new(CONFIG_FILES[:bans_cfg][:local_erb]).result(binding) + content = params['content'] + content << "\n" unless content.end_with? "\n" + FileUtils.mkdir_p(File.dirname(file)) + File.write(file, content) + "Wrote #{file.sub(USER_HOME, '~')}." + end + + put '/config/:action', auth: :admin do + if params['action'] == 'set' + begin + value = case params['variable'] + when 'server_mutators' + params['value'].split(',') + else + params['value'] + end + config_id = params['config-id'] + config = get_config(config_id) + status, msg, current, old = $config_handler.set config['id'], params['variable'], value + if status + daemon = @@daemons.select { |id, _| id == config['id'] }.first.last rescue nil + if daemon + daemon.config.merge!($config_handler.server_configs[config_id]) # Merge the current config for next restart + log "Merged config to active daemon for config ID #{daemon.config['id']}" + end + "Changed #{params['variable']}: #{old} => #{current}" + else + status 400 + msg + end + rescue => e + msg = "Failed to set #{params['variable'] + ' -> ' + params['value']}" + log msg, e + status 400 + "#{msg} | #{e.class}: #{e.message}" + end + elsif params['action'] == 'get' + value = $config_handler.get params['config-id'], params['variable'] + if value.nil? + status 404 + "Could not find value for #{params['variable']}" + else + value.to_s + end + else + halt 400, 'Unknown action' + end + end + + get '/daemons', auth: :admin do + @daemons = @@daemons + erb :daemons + end + + get '/daemon/:config_id', auth: :admin do + daemon = @@daemons[params[:config_id]] + if daemon.nil? + status 400 + "Could not find server daemon with ID #{params[:config_id]}" + else + session[:active_server_config] = daemon.config['id'] + nil + end + end + + get '/monitors', auth: :admin do + @monitors = @@monitors + erb :monitors + end + + get '/monitor-config/:name', auth: :admin do + name = params['name'] + config = $config_handler.monitor_configs[name] + running = !@@monitors[config['id']].nil? + Oj.dump({'running' => running}.merge(config)) + end + + post '/monitor-config/:name', auth: :admin do + name = params['name'] + data = Oj.load(request.body.read) + request.body.rewind + config = data['config'] + if config['ip'] && config['query_port'] && config['rcon_port'] && config['rcon_password'] + $config_handler.monitor_configs[name] = { + 'id' => $config_handler.monitor_configs.dig(name, 'id') || Sysrandom.uuid, + 'ip' => config['ip'], + 'name' => name, + 'query_port' => config['query_port'], + 'rcon_port' => config['rcon_port'], + 'rcon_password' => config['rcon_password'], + } + $config_handler.write_monitor_configs + Oj.dump({message: 'Monitor config saved'}.merge($config_handler.monitor_configs[name])) + else + status 400 + 'Missing parameter(s)' + end + end + + delete '/monitor-config/:name', auth: :admin do + name = params['name'] + if $config_handler.monitor_configs[name] + $config_handler.monitor_configs.delete name + "Monitor config deleted" + else + code 400 + "Unknown monitor config" + end + end + + get '/monitor-configs', auth: :admin do + @configs = $config_handler.monitor_configs + erb :'monitor-configs' + end + + get '/monitor/:id', auth: :admin do + # Get UUID for RCON buffer for tailing + if @@monitors[params['id']] + @@monitors[params['id']].rcon_buffer[:uuid] + else + status 400 + 'Unknown monitor' + end + end + + post '/monitor/:id/:action', auth: :admin do + data = Oj.load(request.body.read) + request.body.rewind + id = params['id'] + action = params['action'] + monitor_config = $config_handler.monitor_configs.select { |_, config| config['id'] == id }.first.last rescue nil + if monitor_config.nil? + status 404 + return "Could not find monitor with ID: #{id}" + end + case action + when 'start' + if @@monitors[id] + status 409 + "Monitor with ID #{id} already exists" + else + @@monitors[id] = ServerMonitor.new(monitor_config['ip'], monitor_config['query_port'], monitor_config['rcon_port'], monitor_config['rcon_password'], rcon_buffer: self.class.create_buffer.last, interval: 10, delay: 0, name: monitor_config['name']) + "Started monitor with ID #{id}" + end + when 'stop' + if @@monitors[id] + @@monitors[id].stop + @@monitors.delete id + "Monitor stopped" + else + status 400 + "Monitor with ID #{id} doesn't exist" + end + else + status 400 + 'Unknown action' + end + end + + get '/monitoring-details/:id', auth: :admin do + @info = {} + @id = params[:id] + matching_daemons = @@daemons[@id] + @monitor = @@daemons[@id] && @@daemons[@id].monitor || @@monitors[@id] + @info = @monitor.info rescue nil + erb :'monitoring-details' + end + + post '/moderator/:action/:unique_id', auth: :moderator do + data = Oj.load(request.body.read) + request.body.rewind + reason = data['reason'].gsub('"', '\"') + unique_id = params[:unique_id] + ip, port, password = get_rcon_info(id: data['id']) + unless password + halt 400, "Could not find an RCON password for ID: #{data['id']}" + end + case params[:action] + when 'ban' + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(ip, port, password, "banid #{unique_id} #{reason}", buffer: buffer) } + uuid + when 'kick' + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(ip, port, password, "kickid #{unique_id} #{reason}", buffer: buffer) } + uuid + else + status 400 + 'Unknown action' + end + end + + post '/rcon', auth: :admin do + body = request.body.read + request.body.rewind + options = Oj.load body, symbol_keys: true + log "Calling RCON client for command: [#{options[:host]}:#{options[:port]}] (TX >>) #{options[:command]}" + uuid, buffer = self.class.create_buffer + Thread.new { @@rcon_client.send(options[:host], options[:port], options[:pass], options[:command], buffer: buffer) } + uuid + end + + get '/server-configs', auth: :admin do + @configs = $config_handler.server_configs + erb :'server-configs' + end + + get '/server-config', auth: :admin do + config_id = params['config-id'] + config = get_config(config_id) + session[:active_server_config] = config_id + Oj.dump(config) + end + + post '/server-config', auth: :admin do + config = Oj.load(request.body.read) + config_id = config['id'] + if config_id && !config_id.is_uuid? + halt 400, "Invalid config ID format" + end + config_name = config['server-config-name'] + request.body.rewind + new_config = $config_handler.create_server_config config + overwrote = new_config['id'] == config_id + Oj.dump({ + message: "#{overwrote ? 'Overwrote' : 'Created'} #{new_config['id']}", + id: new_config['id'], + }, mode: :compat) + end + + delete '/server-config', auth: :admin do + config_id = params['config-id'] + halt 400, "Server config not found for ID #{config_id}" unless $config_handler.server_configs[config_id] + $config_handler.delete_server_config config_id + "Deleted #{params['name']}" + end + + get '/get-buffer/:id/:type', auth: :admin do + id = params[:id] + daemon = @@daemons[id] + if daemon.nil? + status 400 + return "No daemon exists for config ID #{id}" + end + type = params[:type] + case type + when 'server' + daemon.buffer[:uuid] + when 'rcon' + daemon.rcon_buffer[:uuid] + when 'chat' + daemon.chat_buffer[:uuid] + else + status 400 + "Unknown buffer type: #{type}" + end + end + + post '/automatic-updates/:action', auth: :admin do + action = params['action'] + case action + when 'disable' + @@config['automatic_updates'] = false + self.class.save_webapp_config @@config + stop_updates + when 'enable' + @@config['automatic_updates'] = true + self.class.save_webapp_config @@config + start_updates + else + status 400 + "Unknown action: #{action}" + end + end + + get '/generate-password', auth: :admin do + ConfigHandler.generate_password + end + + get '/download-server-config', auth: :admin do + config_id = params['config-id'] + unless config_id.is_uuid? + halt 400, "Invalid config ID format" + end + config_dir = File.join CONFIG_FILES_DIR, config_id + unless File.directory?(config_dir) + halt 400, "No files found for this config ID" + end + zip_path = File.expand_path File.join CONFIG_FILES_DIR, "ohdaw-config_#{config_id}_#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + FileUtils.touch zip_path + zip_files zip_path, File.join(config_dir, '**/*') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end + + get '/download-server-log/:config_id', auth: :admin do + log_file = File.expand_path File.join SERVER_LOG_DIR, "#{params['config_id']}.log" + if File.exist? log_file + send_file log_file, filename: File.basename(log_file), disposition: :attachment + else + status 400 + "Unable to find log file: #{params['config_name']}.log" + end + end + + get '/download-server-logs(/:config_id)?', auth: :admin do + config_id = params['config_id'] + zip_path = File.expand_path File.join SERVER_LOG_DIR, "#{config_id + '-' if config_id}ohd-logs-#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + zip_files zip_path, File.join(SERVER_LOG_DIR, config_id ? "#{config_id}*.log" : '*.log') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end + + get '/download-wrapper-logs', auth: :host do + zip_path = File.expand_path File.join WEBAPP_ROOT, 'log', "ohdaw-logs-#{DateTime.now.strftime('%Y-%m-%d_%H-%M-%S')}.zip" + zip_files zip_path, File.join(WEBAPP_ROOT, 'log', '*.log*') + begin + send_file zip_path, filename: File.basename(zip_path), disposition: :attachment + ensure + Thread.new { FileUtils.rm zip_path } + end + end +end + +def get_webrick_options(config = AdminWrapperSite.load_webapp_config) + webrick_options = { + AccessLog: [[LOGGER, WEBrick::AccessLog::COMMON_LOG_FORMAT], [LOGGER, WEBrick::AccessLog::REFERER_LOG_FORMAT]], + Host: config['admin_interface_bind_ip'], + Port: config['admin_interface_bind_port'], + Logger: LOGGER, # Logger.new(IO::NULL), + app: AdminWrapperSite, # app must be lowercase + SSLEnable: config['admin_interface_use_ssl'] + } + + if config['admin_interface_use_ssl'] + webrick_options.merge!({ + SSLVerifyClient: config['admin_interface_verify_ssl'] ? OpenSSL::SSL::VERIFY_PEER : OpenSSL::SSL::VERIFY_NONE, + SSLCertName: [[ 'CN', WEBrick::Utils::getservername ]] + }) + if File.exist?(config['admin_interface_ssl_cert']) && File.exist?(config['admin_interface_ssl_key']) + webrick_options.merge!({ + SSLCertificate: OpenSSL::X509::Certificate.new(File.read config['admin_interface_ssl_cert']), + SSLPrivateKey: OpenSSL::PKey::RSA.new(File.read config['admin_interface_ssl_key']) + }) + else + # Make temporary cert/key, since WEBrick's defaults won't work with modern browsers + cert, key = CertificateGenerator.generate + log "Writing generated certs to file", level: :info + config['admin_interface_ssl_cert'] = GENERATED_SSL_CERT + config['admin_interface_ssl_key'] = GENERATED_SSL_KEY + AdminWrapperSite.save_webapp_config(config) + File.write GENERATED_SSL_CERT, cert + File.write GENERATED_SSL_KEY, key + webrick_options.merge!({ + SSLCertificate: cert, + SSLPrivateKey: key, + }) + end + end + webrick_options +rescue => e + log "Couldn't create webrick options", e + raise +end + +begin + config = AdminWrapperSite.load_webapp_config + options = get_webrick_options(config) + log "Starting webserver with options: #{options}" + $server = Rack::Server.new(options) + Thread.new { sleep 1; log "Webserver initialized! Visit: http#{'s' if config['admin_interface_use_ssl']}://#{config['admin_interface_bind_ip'] == '0.0.0.0' ? '127.0.0.1' : config['admin_interface_bind_ip']}:#{config['admin_interface_bind_port']}", level: :info } + $server.start +rescue => e + unless e.is_a? SystemExit + log "Exception caused webserver exit", e + at_exit { exit 1 } + end +ensure + log "Webserver stopped", level: :info + $config_handler.write_config + log "Config written", level: :info + `stty echo` unless WINDOWS || !$stdout.isatty +end diff --git a/admin-interface/log/.gitignore b/admin-interface/log/.gitignore new file mode 100644 index 0000000..55f8b93 --- /dev/null +++ b/admin-interface/log/.gitignore @@ -0,0 +1,2 @@ +*.log* + diff --git a/extras/systemd/sandstorm-admin-wrapper.service b/extras/systemd/sandstorm-admin-wrapper.service new file mode 100644 index 0000000..9533450 --- /dev/null +++ b/extras/systemd/sandstorm-admin-wrapper.service @@ -0,0 +1,13 @@ +[Install] +WantedBy=multi-user.target + +[Unit] +Description=OHD Admin Wrapper + +[Service] +User=wrapper-user +Type=simple +WorkingDirectory=/home/wrapper-user/ohd-admin-wrapper/admin-interface +ExecStartPre=/home/wrapper-user/.rbenv/bin/rbenv exec bundle install +ExecStart=/home/wrapper-user/.rbenv/bin/rbenv exec bundle exec ruby lib/webapp.rb +Restart=always diff --git a/linux_start.sh b/linux_start.sh new file mode 100755 index 0000000..1d40dd4 --- /dev/null +++ b/linux_start.sh @@ -0,0 +1,12 @@ +#!/usr/bin/env bash + +wrapper_root="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )" + +cd "$wrapper_root"/admin-interface + +while true; do + bundle && bundle exec ruby "$wrapper_root"/admin-interface/lib/webapp.rb "$@" + CODE=$? + [ $CODE -eq 2 ] || exit $CODE # Restart if exit code 2 +done + diff --git a/windows_start.bat b/windows_start.bat new file mode 100644 index 0000000..47788f0 --- /dev/null +++ b/windows_start.bat @@ -0,0 +1,32 @@ +@echo off +cd %~dp0\admin-interface + +:bundle +echo Bundling +call bundle +if ERRORLEVEL 1 goto bundle_error +echo Starting webapp + +:run_webapp +call bundle exec ruby lib/webapp.rb %* +echo Error level: %errorlevel% +if errorlevel 2 goto :bundle +if errorlevel 1 goto exit_error +if errorlevel 0 goto exit_success + +:webapp_error +echo Failed to run OHD Admin Wrapper. Ensure Ruby is version 2.6.3+. Logs are located in admin-interface/log. +goto exit_error + +:bundle_error +echo Failed to run Ruby Bundler. Ensure Ruby is installed with the bundler gem. https://rubyinstaller.org/downloads/ +goto exit_error + +:exit_error +pause +exit 1 + +:exit_success +echo OHD Admin Wrapper exited successfully. +pause +exit 0