arachni 1.4 → 1.6.0

data/lib/arachni/parser/nodes/element.rb

@@ -0,0 +1,48 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+require_relative 'base'
+require_relative '../with_children'
+
+module Arachni
+class Parser
+module Nodes
+
+class Element < Base
+    require_relative 'element/with_attributes'
+
+    include WithChildren
+    include WithAttributes
+
+    attr_reader :name
+
+    def initialize( name )
+        @name = name.downcase.to_sym
+    end
+
+    def to_html( indentation = 2, level = 0 )
+        indent = ' ' * (indentation * level)
+
+        html = "#{indent}<#{name}"
+
+        attributes.each do |k, v|
+            html << " #{k}=#{v.inspect}"
+        end
+
+        html << ">\n"
+        children.each do |node|
+            html << node.to_html( indentation, level + 1  )
+        end
+        html << "#{indent}</#{name}>\n"
+    end
+
+end
+
+end
+end
+end

data/lib/arachni/parser/nodes/text.rb

@@ -0,0 +1,32 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+require_relative 'base'
+require_relative 'with_value'
+
+module Arachni
+class Parser
+module Nodes
+
+class Text < Base
+    include WithValue
+
+    def text
+        @value.to_s
+    end
+
+    def to_html( indentation = 2, level = 0 )
+        indent = ' ' * (indentation * level)
+        "#{indent}#{value}\n"
+    end
+
+end
+
+end
+end
+end

data/lib/arachni/parser/nodes/with_value.rb

@@ -0,0 +1,29 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+module Arachni
+class Parser
+module Nodes
+
+module WithValue
+
+    attr_reader :value
+
+    def initialize( value )
+        self.value = value
+    end
+
+    def value=( v )
+        @value = v.recode.strip.freeze
+    end
+
+end
+
+end
+end
+end

data/lib/arachni/parser/sax.rb

@@ -0,0 +1,76 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+require_relative 'document'
+
+module Arachni
+class Parser
+class SAX < Ox::Sax
+
+    class Stop < Exception
+    end
+
+    attr_reader :document
+
+    def initialize( options = {} )
+        super()
+
+        @document = Document.new
+
+        @stop_on_first = Set.new
+        @stop_on_first.merge( options[:stop_on_first] ) if options[:stop_on_first]
+
+        @current_node = @document
+    end
+
+    def start_element( name )
+        # We were instructed to stop on the first sight of the previous element
+        # but came across this one before it closed.
+        fail Stop if @stop
+        @stop = stop?( name )
+
+        e = Nodes::Element.new( name )
+
+        e.document     = @document
+        @current_node << e
+        @current_node  = e
+    end
+
+    def end_element( name )
+        # Finished parsing the desired element, abort.
+        fail Stop if @stop
+
+        @current_node = @current_node.parent
+    end
+
+    def attr( name, value )
+        return if !@current_node.respond_to?( :attributes )
+
+        @current_node.attributes[name] = value
+    end
+
+    def text( value )
+        return if value.strip.empty?
+        @current_node << Nodes::Text.new( value )
+    end
+
+    def comment( value )
+        @current_node << Nodes::Comment.new( value )
+    end
+
+    private
+
+    def stop?( name )
+        return false if @stop_on_first.empty?
+
+        @stop_on_first.include?( name.to_s.downcase )
+    end
+
+end
+end
+end

data/lib/arachni/parser/with_children/search.rb

@@ -0,0 +1,92 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+module Arachni
+class Parser
+module WithChildren
+
+module Search
+
+    def traverse( klass = nil, &block )
+        traverser children, klass, &block
+    end
+
+    def descendants
+        @descendants ||= begin
+            n = []
+            traverse { |e| n << e }
+            n
+        end
+    end
+
+    def nodes_by_class( klass )
+        @nodes_by_name ||= {}
+        @nodes_by_name[name] ||= begin
+            descendants.select do |e|
+                e.kind_of? klass
+            end
+        end
+    end
+
+    def nodes_by_name( name )
+        name = name.to_s.downcase
+
+        @nodes_by_name ||= {}
+        @nodes_by_name[name.hash] ||= begin
+            descendants.select do |e|
+                e.respond_to?( :name ) && e.name == name.to_sym
+            end
+        end
+    end
+
+    def nodes_by_names( *names )
+        names = names.flatten
+
+        @nodes_by_name ||= {}
+        @nodes_by_name[names.hash] ||= names.map { |n| nodes_by_name( n ) }.flatten
+    end
+
+    def nodes_by_attribute_name( name )
+        name = name.to_s.downcase
+
+        @nodes_by_attribute_name ||= {}
+        @nodes_by_attribute_name[name.hash] ||= begin
+            descendants.select do |e|
+                e.respond_to?(:attributes) && e.attributes.include?( name )
+            end
+        end
+    end
+
+    def nodes_by_attribute_name_and_value( name, value )
+        name = name.to_s.downcase
+
+        @nodes_by_attribute_name_and_value ||= {}
+        @nodes_by_attribute_name_and_value[[name, value].hash] ||= begin
+            nodes_by_attribute_name( name ).select do |e|
+                e[name].to_s.downcase == value
+            end
+        end
+    end
+
+    private
+
+    def traverser( nodes, klass = nil, &block )
+        nodes.each do |node|
+            block.call( node ) if !klass || node.is_a?( klass )
+
+            next if !node.respond_to?( :children )
+
+            traverser node.children, klass, &block
+        end
+    end
+
+end
+
+end
+end
+end

data/lib/arachni/parser/with_children.rb

@@ -0,0 +1,35 @@
+=begin
+    Copyright 2010-2022 Ecsypno <http://www.ecsypno.com>
+
+    This file is part of the Arachni Framework project and is subject to
+    redistribution and commercial restrictions. Please see the Arachni Framework
+    web site for more information on licensing and terms of use.
+=end
+
+require_relative 'with_children/search'
+
+module Arachni
+class Parser
+module WithChildren
+
+    include Search
+
+    def children
+        @children ||= []
+    end
+
+    def text
+        txt = children.find { |n| n.is_a? Parser::Nodes::Text }
+        return '' if !txt
+
+        txt.value
+    end
+
+    def <<( child )
+        child.parent = self
+        children << child
+    end
+
+end
+end
+end