nasl 0.0.8 → 0.1.1

data/lib/nasl/parser/reference.rb

@@ -0,0 +1,41 @@
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+################################################################################
+
+require 'nasl/parser/node'
+
+module Nasl
+  class Reference < Node
+    attr_reader :name
+
+    def initialize(tree, *tokens)
+      super
+
+      @name = @tokens[0]
+
+      @children << :name
+    end
+  end
+end

data/lib/nasl/parser/repeat.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/parser/repetition.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/parser/return.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -22,7 +22,7 @@
 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-################################################################################
+###############################################################################
 
 require 'nasl/parser/node'
 

data/lib/nasl/parser/string.rb

@@ -1,5 +1,5 @@
-!################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -28,6 +28,7 @@ require 'nasl/parser/node'
 
 module Nasl
   class String < Node
+    include Comparable
     attr_reader :text, :type
 
     def initialize(tree, *tokens)
@@ -37,6 +38,10 @@ module Nasl
       @type = @tokens.first.type
     end
 
+    def <=>(other)
+      self.text <=> other.text
+    end
+
     def to_xml(xml)
       xml.method_missing(@type.downcase, @text)
     end

data/lib/nasl/parser/tree.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -27,7 +27,7 @@
 require 'builder'
 
 module Nasl
-  class Tree < Array
+  class Tree < ::Array
     def all(cls)
       (@all[Nasl.const_get(cls).to_s] ||= [])
     end

data/lib/nasl/parser/undefined.rb

@@ -1,5 +1,5 @@
-!################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/parser/while.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/test.rb

@@ -1,5 +1,5 @@
 ################################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/token.rb

@@ -1,5 +1,5 @@
-###############################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without

data/lib/nasl/tokenizer.rb

@@ -1,5 +1,5 @@
-###############################################################################
-# Copyright (c) 2011-2012, Mak Kolybabi
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -42,6 +42,7 @@ module Nasl
       'global_var' => :GLOBAL,
       'if'         => :IF,
       'import'     => :IMPORT,
+      'in'         => :IN,
       'include'    => :INCLUDE,
       'local_var'  => :LOCAL,
       'repeat'     => :REPEAT,
@@ -83,7 +84,6 @@ module Nasl
 
       ["|",    :BIT_OR],
       ["^",    :BIT_XOR],
-      ["&",    :BIT_AND],
       [">>>",  :BIT_SRA],
       [">>",   :BIT_SRL],
       ["<<",   :BIT_SLL],
@@ -113,7 +113,10 @@ module Nasl
       ["[",    :LBRACK],
       ["]",    :RBRACK],
       ["{",    :LBRACE],
-      ["}",    :RBRACE]
+      ["}",    :RBRACE],
+
+      ["&",    :AMPERSAND],
+      ["@",    :AT_SIGN]
     ]
 
     @@annotated = [

data/lib/nasl/version.rb

@@ -1,3 +1,29 @@
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+################################################################################
+
 module Nasl
-  VERSION = '0.0.8'
+  VERSION = '0.1.1'
 end

data/nasl.gemspec

@@ -1,4 +1,31 @@
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+################################################################################
+
 # -*- encoding: utf-8 -*-
+
 $:.push File.expand_path('../lib', __FILE__)
 
 require 'nasl/version'
@@ -8,11 +35,11 @@ Gem::Specification.new do |s|
   s.version     = Nasl::VERSION
   s.license     = 'BSD'
   s.homepage    = 'http://github.com/tenable/nasl'
-  s.summary     = 'A parser for the Nessus Attack Scripting Language.'
-  s.description = File.open('README.md').read
+  s.summary     = 'A language parser for the Nessus Attack Scripting Language.'
+  s.description = 'A language parser for the Nessus Attack Scripting Language. Supporting NASL v5.2.'
 
-  s.authors     = ['Mak Kolybabi']
-  s.email       = ['mak@kolybabi.com']
+  s.authors     = ['Mak Kolybabi', 'Alex Weber', 'Jacob Hammack']
+  s.email       = ['mak@kolybabi.com', 'aweber@tenble.com', 'jhammack@tenable.com']
 
   s.rubyforge_project = 'nasl'
 
@@ -21,9 +48,9 @@ Gem::Specification.new do |s|
   s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
   s.require_paths = ['lib']
 
-  s.add_development_dependency 'racc'
-  s.add_development_dependency 'rake'
+  s.add_development_dependency 'racc', '~>1.4'
+  s.add_development_dependency 'rake', '~>10.1'
 
-  s.add_runtime_dependency 'builder', '>= 3.1.4'
-  s.add_runtime_dependency 'rainbow'
+  s.add_runtime_dependency 'builder', '~> 3.1'
+  s.add_runtime_dependency 'rainbow', '~> 2.0'
 end

data/test/unit/parser/test_array.rb

@@ -0,0 +1,174 @@
+################################################################################
+# Copyright (c) 2011-2014, Tenable Network Security
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+# 1. Redistributions of source code must retain the above copyright notice, this
+#    list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright notice,
+#    this list of conditions and the following disclaimer in the documentation
+#    and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+################################################################################
+
+class TestArray < Test::Unit::TestCase
+  include Nasl::Test
+
+  def test_empty
+    tree = parse("foo = {};")
+    assert_not_nil(tree)
+
+    arrays = tree.all(:Array)
+    assert_not_nil(arrays)
+    assert_equal(1, arrays.length)
+
+    array = arrays.first
+    assert_not_nil(array)
+    assert_equal(0, array.pairs.length)
+  end
+
+  def test_integers
+    tree = parse("foo = {1:02, 3:4, 5:0x6};")
+    assert_not_nil(tree)
+
+    arrays = tree.all(:Array)
+    assert_not_nil(arrays)
+    assert_equal(1, arrays.length)
+
+    array = arrays.first
+    assert_not_nil(array)
+    assert_equal(3, array.pairs.length)
+
+    assert_equal(1, array.pairs[0].key.value)
+    assert_equal(array.keys[1], array.pairs[0].value)
+    pair = array.pairs[0]
+    assert(pair.key.is_a? Nasl::Integer)
+    assert(pair.value.is_a? Nasl::Integer)
+    assert_equal(2, pair.value.value)
+
+    assert_equal(3, array.pairs[1].key.value)
+    assert_equal(array.keys[3], array.pairs[1].value)
+    pair = array.pairs[1]
+    assert(pair.key.is_a? Nasl::Integer)
+    assert(pair.value.is_a? Nasl::Integer)
+    assert_equal(4, pair.value.value)
+
+    assert_equal(5, array.pairs[2].key.value)
+    assert_equal(array.keys[5], array.pairs[2].value)
+    pair = array.pairs[2]
+    assert(pair.key.is_a? Nasl::Integer)
+    assert(pair.value.is_a? Nasl::Integer)
+    assert_equal(6, pair.value.value)
+  end
+
+  def test_string
+    tree = parse(%q|foo = {'a':"b", "c":'d'};|)
+    assert_not_nil(tree)
+
+    arrays = tree.all(:Array)
+    assert_not_nil(arrays)
+    assert_equal(1, arrays.length)
+
+    array = arrays.first
+    assert_not_nil(array)
+    assert_equal(2, array.pairs.length)
+
+    pair = array.pairs[0]
+    assert(pair.key.is_a? Nasl::String)
+    assert_equal(:DATA, pair.key.type)
+    assert_equal('a', pair.key.text)
+    assert(pair.value.is_a? Nasl::String)
+    assert_equal(:STRING, pair.value.type)
+    assert_equal('b', pair.value.text)
+    assert_equal(pair.value, array.keys['a'])
+
+    pair = array.pairs[1]
+    assert(pair.key.is_a? Nasl::String)
+    assert_equal(:STRING, pair.key.type)
+    assert_equal('c', pair.key.text)
+    assert(pair.value.is_a? Nasl::String)
+    assert_equal(:DATA, pair.value.type)
+    assert_equal('d', pair.value.text)
+    assert_equal(pair.value, array.keys['c'])
+  end
+
+  def test_mixed
+    tree = parse(%q|foo = {'a':1, 2:"b"};|)
+    assert_not_nil(tree)
+
+    arrays = tree.all(:Array)
+    assert_not_nil(arrays)
+    assert_equal(1, arrays.length)
+
+    array = arrays.first
+    assert_not_nil(array)
+    assert_equal(2, array.pairs.length)
+
+    pair = array.pairs[0]
+    assert(pair.key.is_a? Nasl::String)
+    assert_equal(:DATA, pair.key.type)
+    assert_equal('a', pair.key.text)
+    assert(pair.value.is_a? Nasl::Integer)
+    assert_equal(1, pair.value.value)
+    assert_equal(pair.value, array.keys['a'])
+
+    pair = array.pairs[1]
+    assert(pair.key.is_a? Nasl::Integer)
+    assert_equal(2, pair.key.value)
+    assert(pair.value.is_a? Nasl::String)
+    assert_equal(:STRING, pair.value.type)
+    assert_equal('b', pair.value.text)
+    assert_equal(pair.value, array.keys[2])
+  end
+
+  # A single trailing comma in an array literal is valid, but multiple is not.
+  def test_single_trailing_comma
+    tree = parse(%q|foo = {'a':1, 2:"b",};|)
+    assert_not_nil(tree)
+
+    arrays = tree.all(:Array)
+    assert_not_nil(arrays)
+    assert_equal(1, arrays.length)
+
+    array = arrays.first
+    assert_not_nil(array)
+    assert_equal(2, array.pairs.length)
+
+    pair = array.pairs[0]
+    assert(pair.key.is_a? Nasl::String)
+    assert_equal(:DATA, pair.key.type)
+    assert_equal('a', pair.key.text)
+    assert(pair.value.is_a? Nasl::Integer)
+    assert_equal(1, pair.value.value)
+    assert_equal(pair.value, array.keys['a'])
+
+    pair = array.pairs[1]
+    assert(pair.key.is_a? Nasl::Integer)
+    assert_equal(2, pair.key.value)
+    assert(pair.value.is_a? Nasl::String)
+    assert_equal(:STRING, pair.value.type)
+    assert_equal('b', pair.value.text)
+    assert_equal(pair.value, array.keys[2])
+  end
+
+  def test_multiple_trailing_comma
+    fail_parse(%q|foo = {'a':1, 2:"b",,};|)
+  end
+
+  def test_empty_array_with_comma
+    fail_parse(%q|return {,};|)
+  end
+end