thanthus 0.2.3

data/lib/xanthus/repository.rb

@@ -0,0 +1,28 @@
+module Xanthus
+  class Repository
+
+    def initialize
+    end
+
+    def prepare_xanthus_file
+      script = ''
+      File.readlines('../../.xanthus').each do |line|
+        script += line unless line.include?('github.token') || line.include?('dataverse.token')
+
+        # remove github token
+        script += "\t\t# github.token = 'REMOVED'\n" unless !line.include? 'github.token'
+        # remove dataverse token
+        script += "\t\t# dataverse.token = 'REMOVED'\n" unless !line.include? 'dataverse.token'
+      end
+      File.open('.xanthus', 'w+') do |f|
+        f.write(script)
+      end
+    end
+
+    def prepare_readme_file config
+      File.open('README.md', 'w+') do |f|
+        f.write(config.to_readme_md)
+      end
+    end
+  end
+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

@@ -0,0 +1,7 @@
+module Xanthus
+  VERSION = "0.2.3"
+
+  def self.version
+    puts VERSION
+  end
+end

data/lib/xanthus/virtual_machine.rb

@@ -0,0 +1,119 @@
+require_relative '../utils/os'
+
+module Xanthus
+  class VirtualMachine
+    attr_accessor :name
+    attr_accessor :box
+    attr_accessor :version
+    attr_accessor :cpus
+    attr_accessor :cpu_cap
+    attr_accessor :memory
+    attr_accessor :ip
+    attr_accessor :gui
+    attr_accessor :boxing
+    attr_accessor :ssh_username
+    attr_accessor :ssh_key_path
+    attr_accessor :on_aws
+    attr_accessor :aws_env_key_id
+    attr_accessor :aws_env_key_secret
+    attr_accessor :aws_key_pair_name
+    attr_accessor :aws_region
+    attr_accessor :aws_ami
+    attr_accessor :aws_instance_type
+    attr_accessor :aws_security_group
+
+    def initialize
+      @name = :default
+      @box = 'jhcook/fedora27'
+      @version = '4.13.12.300'
+      @ip = '192.168.33.8'
+      @memory = 4096
+      @cpus = 2
+      @cpu_cap = 70
+      @gui = false
+      @boxing = nil
+      @ssh_username = nil
+      @ssh_key_path = nil
+    end
+
+    def to_vagrant
+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{
+if Vagrant.has_plugin?("vagrant-vbguest")
+  config.vbguest.auto_update = false 
+end
+}
+script += %Q{
+  config.vm.synced_folder ".", "/vagrant", disabled: false, type: 'rsync'
+} unless !@on_aws
+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, override|
+   vb.gui = #{@gui}
+   vb.memory = #{@memory}
+   vb.customize ["modifyvm", :id, "--cpuexecutioncap", "#{@cpu_cap}"]
+   vb.cpus = #{@cpus}
+   vb.name = "#{@name}"
+  end
+} unless @on_aws
+script += %Q{
+  config.vm.provider "aws" do |aws, override|
+   aws.access_key_id = ENV['#{@aws_env_key_id}']
+   aws.secret_access_key = ENV['#{@aws_env_key_secret}']
+   aws.keypair_name = '#{@aws_key_pair_name}'
+   aws.region = '#{@aws_region}'
+   aws.ami =  '#{@aws_ami}'
+   aws.instance_type = '#{@aws_instance_type}'
+   aws.security_groups = ['#{@aws_security_group}']
+  end
+} unless !@on_aws
+
+# add simple support for Windows
+script += %Q{  config.vm.provision "shell", path: "provision.sh"
+
+  config.trigger.before :halt do |trigger|
+    trigger.info = "Retrieving data before halt..."
+    trigger.run = {path: "before_halt.#{sys_script_ext}"}
+  end
+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

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification
+name: thanthus
+version: !ruby/object:Gem::Version
+  version: 0.2.3
+platform: ruby
+authors:
+- Thomas Pasquier
+- Xueyuan "Michael" Han
+- Yilun Sun
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2021-01-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13'
+description: This version is forked from xanthus. As I finished this work when studying
+  in THU, I name this version `thanthus`
+email:
+- syl1887415157@126.com
+executables:
+- xanthus
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- bin/xanthus
+- lib/utils/os.rb
+- lib/xanthus.rb
+- lib/xanthus/configuration.rb
+- lib/xanthus/dataverse.rb
+- lib/xanthus/default.rb
+- lib/xanthus/github.rb
+- lib/xanthus/init.rb
+- lib/xanthus/job.rb
+- lib/xanthus/repository.rb
+- lib/xanthus/script.rb
+- lib/xanthus/version.rb
+- lib/xanthus/virtual_machine.rb
+homepage: http://camflow.org
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.4
+signing_key:
+specification_version: 4
+summary: Automated intrusion detection dataset generation framework.
+test_files: []