Function main

ddserver/recursor/__main__.py:200–242

(logger)

@require(logger='ddserver.utils.logger:Logger')
def main(logger):
  messages = receiver()

  # Handle messages until HELO was received
  for message in messages:
    # Handle HELO message
    if message.tag == 'HELO':
      # Expecting ABI version 1
      if message.version != 1:
        logger.error('recursor: Unappropriate ABI version: %s', message.version)
        send(formatter.FAIL)

      send(formatter.OK,
           banner='ddserver')
      break

    else:
      logger.error('recursor: Missing HELO before command: %s', message.tag)
      send(formatter.FAIL)

  # Handle all messages after HELO
  for message in messages:
    if message.tag == 'HELO':
      logger.error('recursor: Duplicated HELO')
      send(formatter.FAIL)

    elif message.tag == 'Q':
      # Handle query
      answer(query=message)

      send(formatter.END)

    elif message.tag == 'AXFR':
      # We do not support transfer by now
      send(formatter.END)

    elif message.tag == 'PING':
      # Ping does not require any data response
      send(formatter.END)

    else:
      logger.error('recursor: Unhandled message tag: %s', message)
      send(formatter.FAIL)


if __name__ == '__main__':