xanthus 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c4832454eeff21b2bf3013e37d38adaf16c2e3b1e71ec902361cba7b25e63be5
-  data.tar.gz: be923875ac163b8d946f36623b0751436d67b0332d1a76880213f9f9470567cb
+  metadata.gz: 3b32bb667df960d2a565943f27d57c28462cfc37e6ffd8e56f1fddb817afa828
+  data.tar.gz: 772ac67b093aae2ee190481888405b97f08639494a86266c4017750b0da89fcb
 SHA512:
-  metadata.gz: ea5ba69726815f5156300bae089e4b996cd724752089656288c039649d917f2c215ff8e9cdd3e499ccc2c09dc01784094838dea0d3b2118d7c7928014a4f0fa9
-  data.tar.gz: 76cb3e1fcdafecb9f4f1fcfe514ea8cebc50e30b0af28e55130cc10eb2980734c29ded9491fb6a212a69ad86d3e1858cbe584e6e98272b3398986ee4b8d96aad
+  metadata.gz: d2f02645f8141275b74adf7b1b70c3a36609fc9172be148a949ed1b0c1acb18bec134eb13e9ed7717cf098f6fda40289a82b38066f650d49c0c2abcaac56a924
+  data.tar.gz: 379c8d188ac86ee7abf532ef43f05458f541033abe4d97a4087c5026a72c01dd1c0db5da1b03bffbe0822cc9315b624db7e3dae30f654c12746851eac7c3ca6d

data/lib/xanthus.rb

@@ -1,5 +1,6 @@
 require "xanthus/version"
 require "xanthus/init"
+require "xanthus/script"
 require "xanthus/virtual_machine"
 require "xanthus/job"
 require "xanthus/default"

data/lib/xanthus/configuration.rb

@@ -61,6 +61,9 @@ Seed: #{@seed}
     yield(config)
     puts "Running experiment #{config.name} with seed #{config.seed}."
     srand config.seed
+    config.vms.each do |k, v|
+      v.generate_box config
+    end
     config.github_conf.init(config) unless config.github_conf.nil?
     config.jobs.each do |name,job|
       for i in 0..(job.iterations-1) do

data/lib/xanthus/init.rb

@@ -110,7 +110,7 @@ It is very cool and interesting!
   config.job :normal_spade do |job|
     job.iterations = 2
     job.tasks = {spade: [:pre, :spade_start, :normal, :spade_stop, :post]}
-    job.outputs = {spade: {trace: '/tmp/audit_cmd.avro'}}
+    job.outputs = {spade: {trace: '/tmp/audit_cdm.avro'}}
   end
 
   config.job :attack_spade do |job|

data/lib/xanthus/job.rb

@@ -7,12 +7,16 @@ module Xanthus
     attr_accessor :tasks
     attr_accessor :outputs
     attr_accessor :inputs
+    attr_accessor :pre_instructions
+    attr_accessor :post_instructions
 
     def initialize
       @iterations = 0
       @tasks = Hash.new
       @outputs = Hash.new
       @inputs = Hash.new
+      @pre_instructions = nil
+      @post_instructions = nil
     end
 
     def output_script outputs
@@ -25,26 +29,6 @@ module Xanthus
 
     def setup_env machine, scripts, config
       puts 'Setting up task on machine '+machine.to_s+'...'
-      script = ''
-      scripts.each do |t|
-        v = eval(config.scripts[t])
-        if v.kind_of?(Array)
-          v.each do |w|
-            script+=w+"\n"
-          end
-        else
-          script+=v
-        end
-      end
-      script += self.output_script(@outputs[machine]) unless  @outputs[machine].nil?
-
-      script_to_clean = script
-      script = ''
-      script_to_clean.each_line do |s|
-        script += s.strip + "\n" unless s=="\n"
-      end
-      script = script.gsub "\n\n", "\n"
-
       FileUtils.mkdir_p machine.to_s
       Dir.chdir machine.to_s do
         if !@inputs[machine].nil?
@@ -57,18 +41,47 @@ module Xanthus
         File.open('Vagrantfile', 'w+') do |f|
           f.write(config.vms[machine].to_vagrant)
         end
+        script = Script.new(scripts, config).to_s
+        script += self.output_script(@outputs[machine]) unless  @outputs[machine].nil?
         File.open('provision.sh', 'w+') do |f|
           f.write(script)
         end
       end
     end
 
+    def host_scripts config
+      puts 'Setting up host scripts...'
+      if !@pre_instructions.nil?
+        script = Script.new(@pre_instructions, config).to_s
+        File.open('pre.sh', 'w+') do |f|
+          f.write(script)
+        end
+      end
+
+      if !@post_instructions.nil?
+        script = Script.new(@post_instructions, config).to_s
+        File.open('post.sh', 'w+') do |f|
+          f.write(script)
+        end
+      end
+    end
+
+    def execute_pre_instructions
+      puts 'Running pre instructions...'
+      system('sh', './pre.sh')
+    end
+
     def run machine
       Dir.chdir machine.to_s do
         system('vagrant', 'up')
       end
     end
 
+    def execute_post_instructions
+      puts 'Running post instructions...'
+      system('sh', './post.sh')
+    end
+
     def halt machine
       Dir.chdir machine.to_s do
         system('vagrant', 'halt')
@@ -86,12 +99,15 @@ module Xanthus
       puts "Running job #{name.to_s}-#{iteration.to_s}..."
       FileUtils.mkdir_p 'tmp'
       Dir.chdir 'tmp' do
+        self.host_scripts config
         @tasks.each do |machine, templates|
           self.setup_env machine, templates, config
         end
+        self.execute_pre_instructions unless @pre_instructions.nil?
         @tasks.each do |machine, templates|
           self.run machine
         end
+        self.execute_post_instructions unless @post_instructions.nil?
         @tasks.each do |machine, templates|
           self.halt machine
         end

data/lib/xanthus/script.rb

@@ -0,0 +1,31 @@
+module Xanthus
+  class Script
+    attr_accessor :script
+
+
+    def initialize array, config
+      script = ''
+      array.each do |t|
+        v = eval(config.scripts[t])
+        if v.kind_of?(Array)
+          v.each do |w|
+            script+=w+"\n"
+          end
+        else
+          script+=v
+        end
+      end
+      script_to_clean = script
+      script = ''
+      script_to_clean.each_line do |s|
+        script += s.strip + "\n" unless s=="\n"
+      end
+      script = script.gsub "\n\n", "\n"
+      @script = script
+    end
+
+    def to_s
+      @script
+    end
+  end
+end

data/lib/xanthus/version.rb

@@ -1,5 +1,5 @@
 module Xanthus
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 
   def self.version
     puts VERSION

data/lib/xanthus/virtual_machine.rb

@@ -8,6 +8,9 @@ module Xanthus
     attr_accessor :memory
     attr_accessor :ip
     attr_accessor :gui
+    attr_accessor :boxing
+    attr_accessor :ssh_username
+    attr_accessor :ssh_key_path
 
     def initialize
       @name = :default
@@ -18,15 +21,25 @@ module Xanthus
       @cpus = 2
       @cpu_cap = 70
       @gui = false
+      @boxing = nil
+      @ssh_username = nil
+      @ssh_key_path = nil
     end
 
     def to_vagrant
-%Q{
+script = %Q{
 Vagrant.configure(2) do |config|
   config.vm.box = "#{@box}"
   config.vm.box_version = "#{@version}"
   config.vm.network "private_network", ip: "#{@ip}"
-
+}
+script += %Q{
+  config.ssh.username = "#{@ssh_username}"
+} unless ssh_username.nil?
+script += %Q{
+  config.ssh.private_key_path = "#{@ssh_key_path}"
+} unless ssh_key_path.nil?
+script += %Q{
   config.vm.provider "virtualbox" do |vb|
    vb.gui = #{@gui}
    vb.memory = #{@memory}
@@ -37,6 +50,33 @@ Vagrant.configure(2) do |config|
   config.vm.provision "shell", path: "provision.sh"
 end
 }
+      return script
+    end
+
+    def generate_box config
+      return unless !boxing.nil?
+      puts 'Generating box...'
+
+      FileUtils.mkdir_p 'boxing'
+      Dir.chdir 'boxing' do
+        File.open('Vagrantfile', 'w+') do |f|
+          f.write(self.to_vagrant)
+        end
+
+        script =  Script.new(boxing, config).to_s
+        File.open('provision.sh', 'w+') do |f|
+          f.write(script)
+        end
+
+        system('vagrant', 'up')
+        system('vagrant', 'halt')
+        system('vagrant', 'package', '--output', "#{name}.box")
+        puts "#{name}.box created."
+        system('vagrant', 'box', 'add', '--force', "local/#{name}", "#{name}.box")
+        system('vagrant', 'destroy', '-f')
+        @box = "local/#{name}"
+        @version = '0'
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: xanthus
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Thomas Pasquier
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-03-06 00:00:00.000000000 Z
+date: 2019-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -56,6 +56,7 @@ files:
 - lib/xanthus/github.rb
 - lib/xanthus/init.rb
 - lib/xanthus/job.rb
+- lib/xanthus/script.rb
 - lib/xanthus/version.rb
 - lib/xanthus/virtual_machine.rb
 homepage: http://camflow.org