fnode 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e3534634c7d91c19df63d20f248ab30d73bd747a
+  data.tar.gz: bcb483783455d17a86cfc875aa95cec0a36eb6a1
+SHA512:
+  metadata.gz: f48ec7638e9c011f11850c76540ed8fd081637e373e98c37a27b2b4d14ac37e1d3f4c45a5944946d25d78531eea2dfe7e1d5ef4c2e5d7a4ca99591a60fa1c818
+  data.tar.gz: 6079d19fb543b5ef03c72e2cf45a51e73652c13496330b628cd09cd8250119bbf8991668c1c08669a9584c30f1180d48edfe7ede94cc8216a9b76707f0ea0914

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+pkg/*
+/log
+/fuzzings
+config.yml

data/.rspec

@@ -0,0 +1,2 @@
+--color
+--require spec_helper

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in fnode.gemspec
+gemspec

data/Gemfile.lock

@@ -0,0 +1,59 @@
+PATH
+  remote: .
+  specs:
+    FNode (0.0.1)
+      rest-client
+      sinatra
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    byebug (9.0.5)
+    diff-lcs (1.2.5)
+    domain_name (0.5.20160615)
+      unf (>= 0.0.5, < 1.0.0)
+    http-cookie (1.0.2)
+      domain_name (~> 0.5)
+    mime-types (3.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2016.0521)
+    netrc (0.11.0)
+    rack (1.6.4)
+    rack-protection (1.5.3)
+      rack
+    rest-client (2.0.0)
+      http-cookie (>= 1.0.2, < 2.0)
+      mime-types (>= 1.16, < 4.0)
+      netrc (~> 0.8)
+    rspec (3.5.0)
+      rspec-core (~> 3.5.0)
+      rspec-expectations (~> 3.5.0)
+      rspec-mocks (~> 3.5.0)
+    rspec-core (3.5.2)
+      rspec-support (~> 3.5.0)
+    rspec-expectations (3.5.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.5.0)
+    rspec-mocks (3.5.0)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.5.0)
+    rspec-support (3.5.0)
+    sinatra (1.4.7)
+      rack (~> 1.5)
+      rack-protection (~> 1.4)
+      tilt (>= 1.3, < 3)
+    tilt (2.0.5)
+    unf (0.1.4)
+      unf_ext
+    unf_ext (0.0.7.2)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  FNode!
+  byebug
+  rspec
+
+BUNDLED WITH
+   1.12.5

data/README.md

@@ -0,0 +1,22 @@
+fnode
+--------
+
+Fuzzing Node.
+
+# Requirements
+
+* Ruby 2.0 +
+* Python 2.7 +
+
+
+# Install
+
+```
+pip install fusil
+gem install fnode
+```
+
+# Usage
+not ready.
+
+# TODO

data/Rakefile

@@ -0,0 +1 @@
+require 'bundler/gem_tasks'

data/bin/fnode

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.push File.expand_path('../lib', __dir__)
+
+require "fnode"
+
+FNode::Server.run!

data/fnode.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$LOAD_PATH.push File.expand_path('../lib', __FILE__)
+require 'fnode/version'
+
+Gem::Specification.new do |s|
+  s.name        = 'fnode'
+  s.version     = FNode::VERSION
+  s.licenses    = ['MIT']
+  s.date        = '2016-08-29'
+  s.summary     = "F Node."
+  s.description = "F Node."
+  s.authors     = ["Aston Fu"]
+  s.email       = 'fudhao@gmail.com'
+  s.files       = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.require_paths = ['lib']
+  s.executables   = %w(fnode)
+  s.homepage    = 'https://github.com/astonfu/fnode'
+
+  s.add_runtime_dependency 'rest-client'
+  s.add_runtime_dependency 'sinatra'
+  s.add_runtime_dependency 'logging'
+
+  s.add_development_dependency 'rspec'
+  s.add_development_dependency 'byebug'
+end

data/fuzzers/fusil_fuzzer.py

@@ -0,0 +1,64 @@
+#!/usr/bin/python
+"""
+fuzzing app templete
+"""
+
+PROGRAM = 'fuzzing'
+PLAY_DURATION = 3
+MAX_FILESIZE = 1024*1024
+
+from fusil.application import Application
+from optparse import OptionGroup
+from fusil.process.mangle import MangleProcess
+from fusil.process.watch import WatchProcess
+from fusil.process.stdout import WatchStdout
+from fusil.auto_mangle import AutoMangle
+from fusil.terminal_echo import TerminalEcho
+
+class Fuzzer(Application):
+    NAME = "fuzzing"
+    USAGE = "%prog [options] filename"
+    NB_ARGUMENTS = 1
+
+    def createFuzzerOptions(self, parser):
+        options = OptionGroup(parser, "fuzzing")
+        options.add_option("--fuzzing", help="fuzzing program path (default: %s)" % PROGRAM,
+            type="str", default=PROGRAM)
+        options.add_option("--duration", help="Playing maximum duration in seconds (default: %s)" % PLAY_DURATION,
+            type="int", default=PLAY_DURATION)
+        options.add_option("--filesize", help="Maximum file size in bytes (default: %s)" % MAX_FILESIZE,
+            type="int", default=MAX_FILESIZE)
+        return options
+
+    def setupProject(self):
+        project = self.project
+        # Command line
+        arguments = [self.options.fuzzing]
+        timeout = self.options.duration + 1.0
+
+        # Create buggy input file
+        orig_filename = self.arguments[0]
+        mangle = AutoMangle(project, orig_filename)
+        mangle.max_size = self.options.filesize
+
+        process = MangleProcess(project,
+            arguments,
+            "<fuzzing>",
+            timeout=timeout)
+
+        process.setupX11()
+        process.env.copy('HOME')
+        watch = WatchProcess(process, timeout_score=0)
+        if watch.cpu:
+            watch.cpu.weight = 0.20
+            watch.cpu.max_load = 0.50
+            watch.cpu.max_duration = min(3, timeout-0.5)
+            watch.cpu.max_score = 0.50
+
+        stdout = WatchStdout(process)
+
+        # Restore terminal state
+        TerminalEcho(project)
+
+if __name__ == "__main__":
+    Fuzzer().main()

data/lib/fnode.rb

@@ -0,0 +1 @@
+require_relative "fnode/server"

data/lib/fnode/node.rb

@@ -0,0 +1,132 @@
+require 'logging'
+require 'fileutils'
+require 'singleton'
+require 'yaml'
+
+
+module FNode
+  class Node
+    include Singleton
+    CONFIG_FILE = "config.yml"
+    FUZZINGS_FOLDER = "fuzzings"
+    ATTRS = %w(name ip port os state pid test_app test_file_path admin_ip admin_port)
+
+    ATTRS.each do |attr|
+      attr_accessor(attr)
+    end
+
+    def load_attrs(yml_file=CONFIG_FILE)
+      attrs = YAML.load_file(yml_file)
+      ATTRS.each do |attr|
+        self.public_send("#{attr}=", attrs[attr])
+      end
+    end
+
+    def dump_attrs(yml_file=CONFIG_FILE)
+      attrs = {}
+      ATTRS.each do |attr|
+        attrs.store attr, self.public_send(attr)
+      end
+
+      open(yml_file, "w") do |f|
+        f << attrs.to_yaml
+      end
+    end
+
+    def set_state(new_state)
+      state = new_state
+      @log.info "Change state: #{state}"
+    end
+
+    def initialize
+      load_attrs CONFIG_FILE
+      setup_logger
+    end
+
+    def run_fuzz_test
+      stop_fuzz_test unless pid.nil?
+
+      self.pid = fork do
+        FileUtils.mkdir_p FUZZINGS_FOLDER
+        Dir.chdir FUZZINGS_FOLDER
+        cmd = "python #{File.expand_path('../../fuzzers/fusil_fuzzer.py', __dir__)} #{test_file_path} --force-unsafe --keep-sessions --fuzzing #{test_app}"
+        begin
+          set_state  "running"
+          @log.info "pid: #{Process.pid}"
+          exec cmd
+        rescue Exception => e
+          @log_error.error "run test error: " + e.to_s
+          stop_fuzz_test
+        end
+      end
+    end
+
+    def stop_fuzz_test
+      unless pid.nil?
+        begin
+          Process.kill('QUIT', pid)
+          self.pid = nil
+          set_state "stop"
+        rescue => e
+          @log_error.error "stop fuzz test error" + e.to_s
+        end
+      end
+    end
+
+    def get_server_file
+      file = Tempfile.new("templete_file")
+      file.binmode
+      file << open("http://#{admin_ip}:#{admin_port}/tasks/#{task.id}/templete_file").read
+      file.close
+      file
+    end
+
+    def self.test
+      n = Node.instance
+      n.test_app = "pluma"
+      n.test_file_path = "/tmp/fuzz.txt"
+      n.run_fuzz_test
+      sleep 10
+      n.stop_fuzz_test
+    end
+
+    private
+      def setup_logger
+        require 'fileutils'
+        FileUtils.mkdir_p 'log'
+
+        @log = Logging.logger['fnode']
+        @log.level = :info
+
+        # here we setup a color scheme called 'bright'
+        Logging.color_scheme( 'bright',
+          :levels => {
+            :info  => :green,
+            :warn  => :yellow,
+            :error => :red,
+            :fatal => [:white, :on_red]
+          },
+          :date => :blue,
+          :logger => :cyan,
+          :message => :magenta
+        )
+
+        Logging.appenders.stdout(
+          'stdout',
+          :layout => Logging.layouts.pattern(
+            :pattern => '[%d] %-5l %c: %m\n',
+            :color_scheme => 'bright'
+          )
+        )
+
+        @log.add_appenders 'stdout', \
+          Logging.appenders.file('log/fnode.log', \
+            :layout => Logging.layouts.pattern(:pattern => '[%d] %-5l %c: %m\n'))
+
+        @log_error = Logging.logger['error']
+        @log_error.level = :error
+        @log_error.add_appenders Logging.appenders.file('log/fnode.error.log', \
+            :layout => Logging.layouts.pattern(:pattern => '[%d] %-5l %c: %m\n'))
+      end
+  end
+end

data/lib/fnode/server.rb

@@ -0,0 +1,20 @@
+
+require 'sinatra'
+
+module FNode
+  class Server < Sinatra::Base
+    get '/' do
+      "OK"
+    end
+
+    get '/run' do
+    end
+
+    get '/stop' do
+
+    end
+
+    
+  end
+
+end

data/lib/fnode/templates/config.example.yml

@@ -0,0 +1,11 @@
+---
+name: name
+ip: 127.0.0.1
+port: 4567
+os: Linux
+state: closed
+pid:
+test_app: app
+test_file_path: "/path/to/file"
+admin_ip: 127.0.0.1
+admin_port: 3000

data/lib/fnode/version.rb

@@ -0,0 +1,3 @@
+module FNode
+  VERSION = '0.0.2'
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: fnode
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Aston Fu
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-08-29 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sinatra
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: logging
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: F Node.
+email: fudhao@gmail.com
+executables:
+- fnode
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- bin/fnode
+- fnode.gemspec
+- fuzzers/fusil_fuzzer.py
+- lib/fnode.rb
+- lib/fnode/node.rb
+- lib/fnode/server.rb
+- lib/fnode/templates/config.example.yml
+- lib/fnode/version.rb
+homepage: https://github.com/astonfu/fnode
+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: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: F Node.
+test_files: []