#
#--
# Ronin SQL - A Ronin library providing support for SQL related security
# tasks.
#
# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#++
#

require 'ronin/code/sql/style'

module Ronin
  module Code
    module SQL
      class InjectionStyle < Style

        # Comment-Obfusticate all keywords
        attr_accessor :comment_evasion

        # Swapcase-Obfusciate all keywords
        attr_accessor :case_evasion

        def initialize(options={})
          super(options)

          if options[:comment_evasion].nil?
            @comment_evasion = false
          else
            @comment_evasion = options[:comment_evasion]
          end

          if options[:case_evasion].nil?
            @case_evasion = false
          else
            @case_evasion = options[:case_evasion]
          end
        end

        def compile_space
          if @comment_evasion
            return '/**/'
          else
            return super
          end
        end

        def compile_keyword(name)
          name = name.to_s

          if @case_evasion
            (rand(name.length)+1).times do
              i = rand(name.length-1).to_i
              name[i] = name[i..i].swapcase
            end

            return name
          else
            return super(name)
          end
        end

      end
    end
  end
end