ronin-sql 0.1.0

data/lib/ronin/code/sql/create_view.rb

@@ -0,0 +1,65 @@
+#
+#--
+# 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/statement'
+require 'ronin/code/sql/select'
+
+module Ronin
+  module Code
+    module SQL
+      class CreateView < Statement
+
+        option :temp, "TEMP"
+        option :if_not_exists, "IF NOT EXISTS"
+
+        def initialize(style,view=nil,query=nil,&block)
+          @view = view
+          @query = query
+
+          super(style,&block)
+        end
+
+        def view(field)
+          @view = field
+          return self
+        end
+
+        def query(table=nil,opts={:fields => nil, :where => nil},&block)
+          @query = Select.new(@style,table,opts,&block)
+          return self
+        end
+
+        def compile
+          compile_expr(keyword_create,temp?,keyword_view,if_not_exists?,@view,keyword_as,@query)
+        end
+
+        protected
+
+        keyword :create
+        keyword :view
+        keyword :as
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/delete.rb

@@ -0,0 +1,64 @@
+#
+#--
+# 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/statement'
+
+module Ronin
+  module Code
+    module SQL
+      class Delete < Statement
+
+        def initialize(style,table=nil,where_expr=nil,&block)
+          @table = table || everything
+          @where = where_expr
+
+          super(style,&block)
+        end
+
+        def from(table)
+          @table = table
+          return self
+        end
+
+        def where(expr)
+          @where = expr
+          return self
+        end
+
+        def compile
+          compile_expr(keyword_delete,@table,where?)
+        end
+
+        protected
+
+        keyword :delete, 'DELETE FROM'
+        keyword :where
+
+        def where?
+          compile_expr(keyword_where,@where) if @where
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/dialect.rb

@@ -0,0 +1,162 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007 Hal Brodigan (postmodern at users.sourceforge.net)
+#
+# 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/exceptions/unknown_dialect'
+require 'ronin/code/sql/function'
+require 'ronin/extensions/meta'
+
+module Ronin
+  module Code
+    module SQL
+      class Dialect
+
+        # The style to use
+        attr_reader :style
+
+        def initialize(style)
+          @style = style
+        end
+
+        def Dialect.dialects
+          @@dialects ||= {}
+        end
+
+        def Dialect.has_dialect?(name)
+          Dialect.dialects.has_key?(name.to_sym)
+        end
+
+        def Dialect.get_dialect(name)
+          name = name.to_sym
+
+          unless Dialect.has_dialect?(name)
+            raise(UnknownDialect,"unknown dialect #{name.dump}",caller)
+          end
+
+          return Dialect.dialects[name]
+        end
+
+        def expresses?(name)
+          public_methods.include?(name.to_s)
+        end
+
+        def express(name,*args,&block)
+          unless expresses?(name)
+            raise(NameError,"undefined method '#{name}' for #{self}",caller)
+          end
+
+          return send(name,*args,&block)
+        end
+
+        def field(name)
+          field_cache[name.to_sym]
+        end
+
+        protected
+
+        def self.dialect(name)
+          name = name.to_sym
+
+          class_def(:name) { name }
+
+          Dialect.dialects[name] = self
+          return self
+        end
+
+        def self.keyword(name,value=name.to_s.upcase)
+          name = name.to_s.downcase
+
+          class_def("keyword_#{name}") { keyword(value) }
+          return self
+        end
+
+        def self.primitives(*names)
+          names.each do |name|
+            name = name.to_s.downcase
+
+            class_def(name) { keyword(name) }
+          end
+
+          return self
+        end
+
+        def self.data_type(name,options={})
+          name = name.to_s.downcase
+          type_name = name.upcase.to_sym
+
+          if options[:length]==true
+            class_def(name) do |length|
+              if length
+                "#{type_name}(#{length})"
+              else
+                type_name
+              end
+            end
+          else
+            class_def(name) { type_name }
+          end
+
+          return self
+        end
+
+        def self.function(*names)
+          names.each do |name|
+            class_def(name) do |field|
+              Function.new(@style,name,field)
+            end
+          end
+
+          return self
+        end
+
+        def self.aggregators(*names)
+          function(*names)
+        end
+
+        def self.command(name,base)
+          class_eval %{
+            def #{name}(*args,&block)
+              #{base}.new(@style,*args,&block)
+            end
+          }
+
+          return self
+        end
+
+        def keyword(value)
+          keyword_cache[value.to_sym]
+        end
+
+        private
+
+        def keyword_cache
+          @keyword_cache ||= Hash.new { |hash,key| hash[key] = Keyword.new(@style,key) }
+        end
+
+        def field_cache
+          @field_cache ||= Hash.new { |hash,key| hash[key] = Field.new(@style,key) }
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/drop_table.rb

@@ -0,0 +1,51 @@
+#
+#--
+# 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/statement'
+
+module Ronin
+  module Code
+    module SQL
+      class DropTable < Statement
+
+        option :if_exists, "IF EXISTS"
+
+        def initialize(style,table=nil,&block)
+          @table = table
+          @exists = false
+
+          super(style,&block)
+        end
+
+        def compile
+          compile_expr(keyword_drop,if_exists?,@table)
+        end
+
+        protected
+
+        keyword :drop, 'DROP TABLE'
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/exceptions.rb

@@ -0,0 +1,24 @@
+#
+#--
+# 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/exceptions/unknown_dialect'

data/lib/ronin/code/sql/exceptions/unknown_dialect.rb

@@ -0,0 +1,31 @@
+#
+#--
+# 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
+#++
+#
+
+module Ronin
+  module Code
+    module SQL
+      class DialectNotFound < RuntimeError
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/expr.rb

@@ -0,0 +1,193 @@
+#
+#--
+# 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/keyword'
+require 'ronin/extensions/meta'
+
+module Ronin
+  module Code
+    module SQL
+      class Expr
+
+        # The style to use
+        attr_reader :style
+
+        def initialize(style)
+          @style = style
+        end
+
+        def in?(*range)
+          In.new(@style,self,*range)
+        end
+
+        def ===(*range)
+          in?(*range)
+        end
+
+        def not_in?(*range)
+          in?(*range).not!
+        end
+
+        def compile
+          # place holder
+        end
+
+        def to_s
+          compile
+        end
+
+        protected
+
+        def keyword(value)
+          keyword_cache[value.to_sym]
+        end
+
+        def keywords(*values)
+          values.map { |value| keyword(value) }
+        end
+
+        def self.keyword(name,value=name.to_s.upcase)
+          name = name.to_s.downcase
+
+          class_def("keyword_#{name}") do
+            keyword(value)
+          end
+
+          return self
+        end
+
+        def self.binary_op(op,*names)
+          names.each do |name|
+            class_def(name) do |expr|
+              BinaryExpr.new(@style,op,self,expr)
+            end
+          end
+
+          return self
+        end
+
+        binary_op '=', '==', :equals?
+        binary_op '!=', :not_equals?
+        binary_op '<>', '<=>', :different?
+        binary_op '>', '>', :greater?
+        binary_op '>=', '>=', :greater_equal?
+        binary_op '<', '<', :less?
+        binary_op '<=', '<=', :less_equal?
+        binary_op 'IS', :is?
+        binary_op 'IS NOT', :is_not?
+        binary_op 'AS', :as
+        binary_op 'CAST', :cast
+        binary_op 'OR', :or
+        binary_op 'XOR', :xor
+        binary_op 'AND', :and
+
+        def self.like_op(op,*names)
+          names.each do |name|
+            class_def(name) do |expr,escape|
+              LikeExpr.new(@style,op,self,expr,escape)
+            end
+          end
+
+          return self
+        end
+
+        like_op 'LIKE', :like
+        like_op 'GLOB', :glob
+        like_op 'REGEXP', :regexp
+        like_op 'MATCH', :match
+
+        def self.unary_op(op,*names)
+          names.each do |name|
+            class_def(name) do
+              UnaryExpr.new(@style,op,self)
+            end
+          end
+
+          return self
+        end
+
+        unary_op 'NOT', :not!
+        unary_op 'EXISTS', :exists?
+
+        def compile_space
+          @style.compile_space
+        end
+
+        def preappend_space(str)
+          @style.preappend_space(str)
+        end
+
+        def append_space(str)
+          @style.append_space(str)
+        end
+
+        def space(*str)
+          @style.space(*str)
+        end
+
+        def compile_newline
+          @style.compile_newline
+        end
+
+        def quote_string(data)
+          @style.quote_string(data)
+        end
+
+        def compile_keyword(name)
+          @style.compile_keyword(name)
+        end
+
+        def compile_list(*expr)
+          @style.compile_list(*expr)
+        end
+
+        def compile_datalist(*expr)
+          @style.compile_list(*expr)
+        end
+
+        def compile_row(*expr)
+          @style.compile_row(*expr)
+        end
+
+        def compile_data(data)
+          @style.compile_data(data)
+        end
+
+        def compile_expr(*expr)
+          @style.compile_expr(*expr)
+        end
+
+        def compile_statements(*statements)
+          @style.compile_statements(*statements)
+        end
+
+        private
+
+        def keyword_cache
+          @keyword_cache ||= Hash.new { |hash,key| hash[key] = Keyword.new(@style,key) }
+        end
+
+      end
+    end
+  end
+end