class PolicyServer

# File flashpolicyd.rb, line 195
def initialize(port, host, xml, logger, timeout=10, debug=false)
  @logger = logger
  @connections = []
  @@connMutex = Mutex.new
  @@clientsMutex = Mutex.new
  @@bogusclients = 0
  @@totalclients = 0
  @timeout = timeout
  @@starttime = Time.new
  @xml = xml
  @port = port
  @host = host

  if debug
    @logger.level = Logger::DEBUG
    debug("Starting in DEBUG mode")
  else
    @logger.level = Logger::INFO
  end
end

# File flashpolicyd.rb, line 258
def bogusclient(msg, client)
  addr = client.addr

  warn("Client #{addr[2]} #{msg}")

  @@clientsMutex.synchronize {
    @@bogusclients += 1
  }
end

# File flashpolicyd.rb, line 164
def debug(msg)
  log(Logger::DEBUG, msg)
end

# File flashpolicyd.rb, line 228
def dumpconnections
  if (@connections.size == 0)
    info("No active connections to dump")
  else
    connections = @connections

    info("Dumping current #{connections.size} connections:")

    connections.each{ |c|
      addr = c.addr
      info("#{c.thread.object_id} started at #{c.timecreated} currently in #{c.thread.status} status serving #{addr[2]} [#{addr[3]}]")
    }
  end
end

# File flashpolicyd.rb, line 244
def dumpthreads
  Thread.list.each {|t|
    info("Thread: #{t.id} status #{t.status}")
  }
end

# File flashpolicyd.rb, line 174
def error(msg)
  log(Logger::ERROR, msg)
end

# File flashpolicyd.rb, line 169
def fatal(msg)
  log(Logger::FATAL, msg)
end

# File flashpolicyd.rb, line 154
def info(msg)
  log(Logger::INFO, msg)
end

# File flashpolicyd.rb, line 149
def log(severity, msg)
  @logger.add(severity) { "#{Thread.current.object_id}: #{msg}" }
end

# File flashpolicyd.rb, line 251
def printstats
  u = sec2dhms(Time.new - @@starttime)

  info("Had #{@@totalclients} clients and #{@@bogusclients} bogus clients. Uptime #{u[0]} days #{u[1]} hours #{u[2]} min. #{@connections.size} connection(s) in use now.")
end

# File flashpolicyd.rb, line 273
def serve(connection)
  client = connection.client

  # Flash clients send a null terminate request
  $/ = "\000"

  # run this in a timeout block, clients will have --timeout seconds to complete the transaction or go away
  begin
    timeout(@timeout.to_i) do
      loop do
        request = client.gets

        if request =~ /policy-file-request/
          client.puts(@xml)

          debug("Sent xml data to client")
          break
        end
      end
    end
  rescue Timeout::Error
    bogusclient("connection timed out after #{@timeout} seconds", connection)
  rescue Errno::ENOTCONN => e
    warn("Unexpected disconnection while handling request")
  rescue Errno::ECONNRESET => e
    warn("Connection reset by peer")
  rescue Exception => e
    bogusclient("Unexpected #{e.class} exception: #{e}", connection)
  end
end

# File flashpolicyd.rb, line 314
def start
  begin
    # Disable reverse lookups, makes it all slow down
    BasicSocket::do_not_reverse_lookup=true
    server = TCPServer.new(@host, @port)
  rescue Exception => e
    fatal("Can't open server: #{e.class} #{e}")
    exit
  end

  begin
    @serverThread = Thread.new {
      while (session = server.accept)
        Thread.new(session) do |client|
          begin
            debug("Handling new connection from #{client.peeraddr[2]}, #{Thread.list.size} total threads ")

            @@clientsMutex.synchronize {
              @@totalclients += 1
            }

            connection = OpenStruct.new
            connection.client = client
            connection.timecreated = Time.new
            connection.thread = Thread.current
            connection.addr = client.peeraddr

            @@connMutex.synchronize {
              @connections << connection
              debug("Pushed connection thread to @connections, now #{@connections.size} connections")
            }

            debug("Calling serve on connection")
            serve(connection)

            client.close

            @@connMutex.synchronize {
              @connections.delete(connection)
              debug("Removed connection from @connections, now #{@connections.size} connections")
            }

          rescue Errno::ENOTCONN => e
            warn("Unexpected disconnection while handling request")
          rescue Errno::ECONNRESET => e
            warn("Connection reset by peer")
          rescue Exception => e
            error("Unexpected #{e.class} exception while handling client connection: #{e}")
            error("Unexpected #{e.class} exception while handling client connection: #{e.backtrace.join("\n")}")
            client.close
          end # block around main logic
        end # while
      end # around Thread.new for client connections
    } # @serverThread
  rescue Exception => e
    fatal("Got #{e.class} exception in main listening thread: #{e}")
  end
end

# File flashpolicyd.rb, line 217
def toggledebug
  if (@logger.debug?)
    @logger.level = Logger::INFO
    info("Set logging level to INFO")
  else
    @logger.level = Logger::DEBUG
    info("Set logging level to DEBUG")
  end
end

# File flashpolicyd.rb, line 159
def warn(msg)
  log(Logger::WARN, msg)
end