ios4r2.rb

require 'strscan'
require 'yaml'
require 'pp'
require 'rubygems'
require 'treetop'
require 'storage'

class Parser

	attr_accessor :config, :token, :tokens_list, :parsed_hashes, :token_instruction, :token_instruction_subconf, :index, :grammar
	@parser
	
	def initialize
	@config = IO.readlines(ARGV[0])
	@tokens_list = Array.new
	@parsed_hashes = Hash.new
	@token_instruction_subconf = Hash.new
	
	end
	
	def find_lines
		token_regexp = Regexp.new(@token)
		@config.each_index {|index|
			scan = StringScanner.new(@config[index])
			if scan.scan(token_regexp) == @token 
				@tokens_list << @config[index]
			end
		}
	end
	
	def find_token_instruction_subconf
			token_instruction_regexp = Regexp.new(@token_instruction)
			index = 1
			until index ==  @config.length do
			scan = StringScanner.new(@config[index])
				if scan.scan(token_instruction_regexp) == @token_instruction
					pointer = index
					until index == @config.length do					
						pointer +=1
							token_regexp = Regexp.new(@token)
							scan2 = StringScanner.new(@config[pointer])
							until scan2.scan(token_regexp) == @token or @config[pointer].include? "!" do
							@token_instruction_subconf.update({pointer+1 => @config[pointer]})
							pointer += 1
							scan2 = StringScanner.new(@config[pointer])
							end
							break
					end
					index = pointer
				else
				index += 1
				end
			end
	end
					
	
	def load_grammar
		dir = File.dirname(__FILE__)
		Treetop.load dir  + "/" + grammar
		parsername = grammar.capitalize + "Parser"
		@parser = Object.module_eval("::#{parsername}").new
	end
	
	def use_grammar_on(an_array)
		an_array.each_pair {|key, value|
		result = @parser.parse(value)
		values = Hash.new
		result.singleton_methods.each {|method| 
			begin
				value = {method => result.send(method).text_value} 
				values.update(value)
			end
		}
		@parsed_hashes.update({key => values})
		}
	
	end
	
	def find_grammar
		if @token.include? " " or  @token.include? "-"
			# limitations de TreeTop
			@grammar = @token.gsub(/ /,'_').gsub(/-/,'_')
		else
			@grammar = @token
		end
	end
	
	def grammar_fail
		@parsed_hashes.each_pair{|key, value|
		if value.to_s == ""
			p "can't parse '#{@token_instruction_subconf[key]}' at line number '#{key}'"
		end	
		}
	end
		

end




s = Storage.new('ios.tct')

z = Parser.new
z.token = "ip access-list extended"
z.find_lines
z.tokens_list.each{|token|
z.token_instruction = token
#z.token_instruction = z.tokens_list[7]
#p z.token_instruction
z.find_token_instruction_subconf
#pp z.token_instruction_subconf
#z.find_grammar
#z.load_grammar
#z.use_grammar_on(z.tokens_list[2])
z.grammar = "ios_acl"
z.load_grammar
#z.token_instruction = z.tokens_list[4]
#p z.token_instruction_subconf
z.use_grammar_on(z.token_instruction_subconf) 
z.grammar_fail

#pp z.parsed_hashes



@temp1 = Hash.new

z.parsed_hashes.each_pair{|key, value|
	@temp1.update({"index" => key})
	value.each_pair{|key2, value2|
		@temp1.update({key2 => value2})
		}
		@temp1.update({'parent' => z.token_instruction})
		@temp1.update({'type' => 'acl'})
		@temp1.update({'time' => Time.now.to_i})
		@temp1.update({'file' => ARGV[0]})
	s.add(@temp1)
	}



}
  pp s.db.query { |q|
   q.add_condition 'type', :equals, 'acl'
    q.order_by 'destination_ip'
}
s.db.close