Giter Club home page Giter Club logo

sftlearning's Introduction

SFT Learning Algorithm

I aim to learn models of sanitizers (by using automata learning algorithms) in a black-box manner, so that I can reason about the correctness of sanitizers. The code in this repository is based on symbolicautomata, which is a library for automata. This repository contains implementations for:

  • SFA (Symbolic Finite Automata) learning algorithm
  • SFT (Symbolic Finite Transducer) learning algorithm

These algorithms can ask a sanitizer two types of questions: a membership query or an equivalence query. Based on the results, it will then derive a model (SFA/SFT). This research has been done for my Master's Thesis (Computer Science, University of Twente) which is titled "Reasoning about the Correctness of Sanitizers".

Running the program

If you want to learn and compare a model to its specification, then execute the class CompareToSpec. The program will prompt the user for any necessary information via the standard input.

If you only want to learn a model, then execute the class TestMembershipOracleStream. The learned model will be stored in SVPAlib/src/sftlearning/learned/.

If you want to analyze a model, for example check for idempotency, then call the desired methods in SpecificationChecking.

When learning a model, the program will prompt the user for some settings regarding the equivalence oracle. If you are unsure which settings would be acceptable, try the following settings:

  • Predicate coverage (7)
  • 3000 tests per predicate
  • lower bound of the alphabet is 32
  • upper bound of the alphabet is 126
  • a maximum of 30 minutes to run.

This will derive a model for the Basic Latin Alphabet, if you want to derive a model for a larger alphabet, increase the upper bound. These integers correspond to the number of the HTML code of a character. This means that 32 (HTML-code:  ) corresponds to " " (whitespace) and 126 (HTML-code: ~) corresponds to "~".

Membership Oracle

The membership oracle is used to pose membership queries. These queries give input to some sanitizer and then observe the output of the sanitizer. To be able to pose membership queries, we require a command from the user. This command may contain options/flags. The program should call the sanitizer for each input that is written to the standard input. If the command is run, it should print the output of the program to the standard output The program should continuously wait for input on the standard input and print the corresponding output when it is presented with an input. Note that this program may need to be compiled beforehand, depending on the programming language which was used. We recommend trying to execute the command beforehand from the command line and using the full path to reference a specific program. Note that any libraries, languages or compilers that are used by the sanitizer should be installed on the computer on which SFTLearning is run.

Here are some examples of commands (all the programs mentioned below can be found in here):

  • node Sanitizers/encode/heEncodeStreams.js
  • php Sanitizers/src/filterSanitizeEmailStreams.php
  • python Sanitizers/src/replaceLTStreams.py
  • ruby Sanitizers/encode/htmlEntitiesEncodeStreams.rb

Equivalence Oracle

The equivalence oracle is used to pose equivalence queries. These queries try to discover whether the hypothesis automaton is a correct model of the sanitizer. If the equivalence oracle does not provide correct counterexamples, then the algorithm will be unable to learn the correct model of the program. The equivalence oracle can be implemented in many different ways. The following have already been implemented in SFTLearning:

  • Random testing
  • Random prefix selection
  • State coverage
  • Transition (or branch) coverage
  • Predicate (or condition) coverage
  • History-based

After choosing a specific oracle, the user will be asked to provide some parameters such as number of tests in total or number of tests per state.

Specifications

This research aimed to be able to reason about the correctness of sanitizers by comparing a derived model to a specification. This specification needs to be written by the user in the form of an SFA or SFT. It should be provided in a DOT file with a specific structure. The program will then ask which type of specification the user wants to check. The following types of specifications can be checked:

  • Equality
  • Blacklist (input/output)
  • Whitelist (input/output, equal/subset)
  • Length (input/output, equal/unequal)
  • Idempotency
  • Commutativity
  • Bad output
Structure of the specification

The specifications should be written in a DOT file which adheres to the following structure:

  • State declaration: number[label=x, peripheries=y] where number denotes the state's number, x should be replaced by the label and y should be replaced by the number of peripheries (2 if it is a final state or 1 if it is not a final state).
  • Initial state declaration: XXnumber [attributes]XXnumber -> number where number should be replaced by the state's number and attributes can be replaced by some attributes that the user wants to define.
  • Transition declaration: fromState -> toState [label="[x]/y where fromState should be replaced by the state where the transition starts, toState should be replaced by the state where the transition leads to, x should be replaced by the guard and y should be replaced by the term functions. Guards should be denoted in unicode references "\u0000". Multiple guards should not be separated by anything. The range a to b can be denoted as a guard using a-b. Terms should be written as "x+0" for the identity function, or "y" where y is the specific constant which is outputted. Multiple term functions should be separated with a whitespace.

An example of a specification which defines a sanitizer that represents CyberChef's "Remove whitespace" function:

digraph spec{
 rankdir=LR;
0[label=0,peripheries=2]
XX0 [color=white, label=""]XX0 -> 0
0 -> 0 [label="[\r]/"]
0 -> 0 [label="[.]/"]
0 -> 0 [label="[\t]/"]
0 -> 0 [label="[\n]/"]
0 -> 0 [label="[ ]/"]
0 -> 0 [label="[\f]/"]
0 -> 0 [label="[\u0000-\b\u000b\u000e-\u001f!-\-/-\uffff]/x+0"]
}

sftlearning's People

Contributors

arjunradhakrishna avatar daniellehuisman avatar lorisdanto avatar orionxu avatar pohmann avatar qinheping avatar shxuy avatar sophietje avatar terry27 avatar zkincaid avatar

Stargazers

 avatar  avatar

Watchers

 avatar  avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.