theforeman/smart-proxy

module Proxy::DHCP
  # Represents Microsoft DHCP Server
  # requires the help of a CGI like script running on a MS IIS Server
  class MS < Server

    def initialize(options = {})
      super options[:server]
      @username = options[:username]
      @password = options[:password]
      @gateway  = options[:gateway]
    end

    def loadSubnets
      super
      cmd = "ListScopeDetailed"
      operand = "enumerate the scopes on #{@name}"

      response = query cmd, operand

      response.each do |line|
        line.chomp
        break if line.match(/^Command completed/)

        # 172.29.216.0   - 255.255.254.0  -Active        -DC BRS               -
        if line =~ /^\s*([\d\.]+)\s*-\s*([\d\.]+)\s*-\s*(Active|Disabled)/
          network = $1
          netmask = $2
          subnet = Proxy::DHCP::Subnet.new(self, network, netmask)
        end
      end
    end

    def loadSubnetData subnet
      raise "invalid Subnet" unless subnet.is_a? Proxy::DHCP::Subnet
      cmd = "ListReservation" + "&ScopeIpAddress=#{subnet.network}"
      operand = "enumerate #{subnet.network} on #{@name}"

      response = query cmd, operand

      subnet.clear

      # Extract the data
      response.each do |line|
        line.chomp
        break if line.match(/^Command completed/)
        #     172.29.216.6      -    00-a0-e7-21-41-00-
        if line =~ /^\s+([\w\.]+)\s+-\s+([-a-f\d]+)/
          ip = $1
          mac = $2.gsub(/-/,":").match(/^(.*?).$/)[1]
          Proxy::DHCP::Record.new(subnet, ip, mac) unless ip.nil? or mac.nil?
        end
      end
      super subnet
    end


    def loadSubnetOptions subnet
      raise "invalid Subnet" unless subnet.is_a? Proxy::DHCP::Subnet
      cmd = "ShowOptionValue&ScopeIPAddress=#{subnet.network}"
      response = query cmd

      subnet.options = parse_options response
      super subnet
    end

    def loadRecordOptions record
      raise "invalid Record" unless record.is_a? Proxy::DHCP::Record
      subnet = record.subnet
      raise "unable to find subnet for #{record}" if subnet.nil?
      cmd = "ShowReservedOptionValue&ScopeIPAddress=#{subnet.network}&ReservedIP=#{record.ip}"
      response = query cmd

      record.options = parse_options response
    end

    def delRecordFor record
      subnet = find_subnet record
      mac    = record.mac.gsub(/:/,"")
      entry  = "ScopeIpAddress=#{subnet.network}&ReservedIP=#{record.ip}"
      cmd = "DeleteReservation&#{entry}&MAC_Address=#{mac}"

      response = query cmd
      logger.info "removed Proxy::DHCP reservation for #{entry}/#{mac}"
      subnet.delete(record)
    end

    private
    def query cmd, operand = nil
      response = invoke cmd
      validate_response cmd, operand, response
      return response
    end

    def invoke cmd
      userAgent = 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.2; SV1; .NET CLR 1.1.4322; .NET CLR 1.0.3705; .NET CLR 2.0.50727; InfoPath.1; MS-RTC LM 8)'
      header  = "https://#{@gateway}/NetShManager/NetShManager.aspx?ServerName=#{@name}&"
      cmd     = "CommandName=/" + cmd
      command = "curl -q -k --silent --user '#{@username}:#{@password}' --user-agent '#{userAgent}' '#{header}#{cmd}'; "
      filtered = command.gsub(@password,"FILTERED")
      puts "Invoking #{filtered}" if @@debug

      response = %x{#{command}}
      puts response if @@debug
      raise Proxy::DHCP::Error, "invalid Response - curl return error code #{$?} - while executing #{filtered}" unless $? == 0
      return $? == 0 ? response.split(/\r\n/) : false
    end

    def validate_response cmd, operand, response
      unless response[-3..-2].join=~/Command completed successfully/
        logger.error "Invoke failed:\n" + response.join("\n")
        scopeIpAddress, ip = cmd.match(/ScopeIpAddress=([^&]+)&ReservedIP=([^&]+)/)[1..2]

        case operand
        when /^create|^delete/
          msg = "Failed to #{operand} for #{ip} in #{scopeIpAddress}"
        when /^enumerate|install/
          msg = "Failed to #{operand}"
        when /^query/
          return response if response[2] =~ /not a reserved client/
        else
          if cmd.match(/Add/)
            msg = "Failed to set #{operand} for #{ip} in #{scopeIpAddress}"
          else
            msg = "Failed to get option values for #{ip} in #{scopeIpAddress}"
          end
        end
      end
      raise Proxy::DHCP::Error.new msg if msg
    rescue
      raise Proxy::DHCP::Error.new "Unknown error while processing #{msg} - #{response}"
    end

    def parse_options response
     optionId = nil
     options = {}
      response.each do |line|
        line.chomp
        break if line.match(/^Command completed/)

        case line
        when /OptionId : (\d+)/
          optionId = $1
        when /Option Element Value = (\S+)/
          options[optionId] = $1
        end
      end
      return options
    end

  end
end