ronin-sql 0.1.0

data/Rakefile

@@ -0,0 +1,14 @@
+# -*- ruby -*-
+
+require 'rubygems'
+require 'hoe'
+require './tasks/spec.rb'
+require './lib/ronin/sql/version.rb'
+
+Hoe.new('ronin-sql', Ronin::SQL::VERSION) do |p|
+  p.rubyforge_name = 'ronin'
+  p.developer('Postmodern Modulus III','postmodern.mod3@gmail.com')
+  p.extra_deps = [['ronin', '>=0.0.9']]
+end
+
+# vim: syntax=Ruby

data/lib/ronin/code/sql.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/code'

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

@@ -0,0 +1,62 @@
+#
+#--
+# 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/expr'
+
+module Ronin
+  module Code
+    module SQL
+      class Between < Expr
+
+        def initialize(expr,lower,higher)
+          super(expr.style)
+
+          @expr = expr
+          @lower = lower
+          @higher = higher
+          @negated = false
+        end
+
+        def not!
+          @negated = true
+          return self
+        end
+
+        def compile
+          compile_expr(@expr,negated?,keyword_between,@lower,keyword_and,@higher)
+        end
+
+        protected
+
+        keyword :between
+        keyword :and
+        keyword :not
+
+        def negated?
+          keyword_not if @negated
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,46 @@
+#
+#--
+# 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/expr'
+
+module Ronin
+  module Code
+    module SQL
+      class BinaryExpr < Expr
+
+        def initialize(style,op,left,right)
+          super(style)
+
+          @op = op
+          @left = left
+          @right = right
+        end
+
+        def compile
+          compile_expr(compile_data(@left),compile_keyword(@op),compile_data(@right))
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,61 @@
+#
+#--
+# 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/dialect'
+require 'ronin/code/sql/common_dialect'
+require 'ronin/code/sql/style'
+
+module Ronin
+  module Code
+    module SQL
+      class Builder < Statement
+
+        def initialize(style,options={},&block)
+          @commands = []
+
+          super(style,&block)
+        end
+
+        def compile
+          @style.compile_statements(@commands)
+        end
+
+        protected
+
+        def command(cmd)
+          @commands << cmd
+          return self
+        end
+
+        def method_missing(sym,*args,&block)
+          result = super(sym,*args,&block)
+
+          @commands << result if result.kind_of?(Statement)
+          return result
+        end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+#
+# 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/program'
+require 'ronin/code/sql/injection'
+
+module Ronin
+  module Code
+    def Code.sql(options={},&block)
+      SQL::Program.new(options,&block)
+    end
+
+    def Code.sql_injection(options={},&block)
+      SQL::Injection.new(options,&block)
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+#
+#--
+# 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/dialect'
+require 'ronin/code/sql/create_table'
+require 'ronin/code/sql/create_index'
+require 'ronin/code/sql/create_view'
+require 'ronin/code/sql/insert'
+require 'ronin/code/sql/select'
+require 'ronin/code/sql/update'
+require 'ronin/code/sql/delete'
+require 'ronin/code/sql/drop_table'
+
+module Ronin
+  module Code
+    module SQL
+      class CommonDialect < Dialect
+
+        dialect :common
+
+        primitives :yes, :no, :on, :off, :null
+
+        data_type :int
+        data_type :varchar, :length => true
+        data_type :text
+        data_type :record
+
+        aggregators :count, :min, :max, :sum, :avg
+
+        command :create_type, CreateTable
+        command :create_index, CreateIndex
+        command :create_view, CreateView
+        command :insert, Insert
+        command :select_from, Select
+        command :update, Update
+        command :delete, Delete
+        command :drop_table, DropTable
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,76 @@
+#
+#--
+# 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 CreateIndex < Statement
+
+        option :unqiue, "UNIQUE"
+        option :if_not_exists, "IF NOT EXISTS"
+
+        def initialize(style,index=nil,table=nil,columns={},&block)
+          @index = index
+          @table = table
+          @columns = columns
+
+          super(style,&block)
+        end
+
+        def index(field)
+          @index = field
+          return self
+        end
+
+        def table(field)
+          @table = field
+          return self
+        end
+
+        def column(name,type)
+          @columns[name.to_s] = type.to_s
+          return self
+        end
+
+        def compile(dialect=nil,multiline=false)
+          format_columns = lambda {
+            @columns.map { |name,type|
+        "#{name} #{type}"
+          }
+          }
+
+          return compile_expr(keyword_create,unique?,keyword_index,if_not_exists?,@index,keyword_on,@table,compile_row(format_columns.call))
+        end
+
+        protected
+
+        keyword :create
+        keyword :index
+        keyword :on
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,93 @@
+#
+#--
+# 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 CreateTable < Statement
+
+        option :temp, "TEMP"
+        option :if_not_exists, "IF NOT EXISTS"
+
+        def initialize(style,table=nil,opts={:columns => {}, :not_null => {}, :as => nil},&block)
+          @table = table
+          @columns = opts[:columns]
+          @not_null = opts[:not_null]
+          @as = opts[:as]
+
+          super(style,&block)
+        end
+
+        def table(field)
+          @table = field
+          return self
+        end
+
+        def as(table=nil,opts={:fields => nil, :where => nil},&block)
+          @as = Select.new(@style,table,opts,&block)
+          return self
+        end
+
+        def column(name,type,null=false)
+          name = name.to_s
+          @columns[name] = type.to_s
+          @not_null[name] = null
+          return self
+        end
+
+        def primary_key(field)
+          @primary_key = field
+          return self
+        end
+
+        def compile
+          format_columns = lambda {
+            @columns.map { |name,type|
+            if @not_null[name]
+          "#{name} #{type} NOT NULL"
+            else
+          "#{name} #{type}"
+            end
+          }
+          }
+
+          return compile_expr(keyword_create,temp?,keyword_table,if_not_exists?,@table,compile_row(format_columns.call))
+        end
+
+        protected
+
+        keyword :create
+        keyword :table
+        keyword :primary_key
+
+        def primary_key?
+          compile_expr(keyword_primary_key,@primary_key) if @primary_key
+        end
+
+      end
+    end
+  end
+end