feat: first commit

h2non · h2non · commit 1d9df94a6505 · 2015-07-29T23:52:10.000+01:00
diff --git a/.codeclimate.yml b/.codeclimate.yml
@@ -0,0 +1,4 @@
+languages:
+  JavaScript: true
+exclude_paths:
+  - "examples/*.js"
diff --git a/.editorconfig b/.editorconfig
@@ -0,0 +1,20 @@
+root = true
+
+[*]
+charset = utf-8
+indent_style = space
+indent_size = 2
+end_of_line = lf
+trim_trailing_whitespace = true
+insert_final_newline = true
+
+[Makefile]
+charset = utf-8
+indent_style = tabs
+indent_size = 2
+end_of_line = lf
+trim_trailing_whitespace = true
+insert_final_newline = true
+
+[*.sh]
+insert_final_newline = false
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,5 @@
+node_modules
+.DS_Store
+npm-debug.log
+test/.tmp
+*.nar
diff --git a/.npmignore b/.npmignore
@@ -0,0 +1,2 @@
+test/fixtures
+*.nar
diff --git a/.travis.yml b/.travis.yml
@@ -0,0 +1,6 @@
+language: node_js
+node_js:
+  - "0.12"
+  - "iojs"
+  - "iojs-v1.6.0"
+
diff --git a/LICENSE b/LICENSE
@@ -0,0 +1,24 @@
+The MIT License
+
+Copyright (c) Tomas Aparicio and contributors
+
+Permission is hereby granted, free of charge, to any person
+obtaining a copy of this software and associated documentation
+files (the "Software"), to deal in the Software without
+restriction, including without limitation the rights to use,
+copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the
+Software is furnished to do so, subject to the following
+conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
+OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.
diff --git a/README.md b/README.md
@@ -0,0 +1,21 @@
+# toxify
+
+Hackable HTTP proxy to simulate system failure conditions. Built for [node.js](http://nodejs.org)/[io.js](https://iojs.org)
+
+**Work in progress**
+
+## Built-in poisons
+
+- [x] Delay
+- [_] Server error
+- [_] Close socket
+- [_] Bandwidth
+- [_] Throttle
+- [_] Debounce
+- [_] Rate limit
+- [_] Slow close
+- [_] Slicer
+
+## License
+
+MIT - Tomas Aparicio
diff --git a/bin/toxify b/bin/toxify
@@ -0,0 +1,5 @@
+#!/usr/bin/env node
+
+const toxify = require('..')
+
+toxify({})
diff --git a/examples/server.js b/examples/server.js
@@ -0,0 +1,13 @@
+const toxify = require('..')
+
+const proxy = toxify()
+
+proxy.percentage(80)
+proxy.delay({ max: 1000, min: 100 })
+proxy.throttle(100)
+
+proxy
+  .forward('http://httpbin.org')
+  .all('/*')
+
+proxy.listen(8089)
diff --git a/index.js b/index.js
@@ -0,0 +1,10 @@
+const Toxify = require('./lib/toxify')
+
+module.exports = toxify
+
+function toxify(opts) {
+  return new Toxify(opts)
+}
+
+toxify.poisons = require('./lib/poisons')
+toxify.VERSION = require('./package.json').version
diff --git a/lib/poisons/abort.js b/lib/poisons/abort.js
@@ -0,0 +1,5 @@
+module.exports = function (filter) {
+  return function (req, res, next) {
+    req.socket.destroy()
+  }
+}
diff --git a/lib/poisons/delay.js b/lib/poisons/delay.js
@@ -0,0 +1,39 @@
+module.exports = function delay(opts, filter) {
+  opts = opts || {}
+  var max = +opts.max || 1000
+  var min = +opts.min || 100
+
+  return function (req, res, next) {
+    var defer = delay(req, next)
+
+    if (filter) {
+      return filter(req, res, defer)
+    }
+
+    defer()
+  }
+
+  function delay(req, next) {
+    return function (err, invalid) {
+      if (err) return next(err)
+      if (invalid) return next()
+
+      var timeout = setTimeout(next, calculateDelay())
+      req.once('close', cleanTimeout)
+
+      function cleanTimeout() {
+        clearTimeout(timeout)
+      }
+
+      function clean() {
+        req.removeEventListener('close', cleanTimeout)
+        next()
+      }
+    }
+  }
+
+  function calculateDelay() {
+    if (opts.jitter) return opts.jitter
+    return Math.random() * (max - min) + min
+  }
+}
diff --git a/lib/poisons/error.js b/lib/poisons/error.js
@@ -0,0 +1,6 @@
+module.exports = function (opts, filter) {
+  return function (req, res, next) {
+    res.writeHead(+opts.code || 500, opts.headers)
+    res.end()
+  }
+}
diff --git a/lib/poisons/index.js b/lib/poisons/index.js
@@ -0,0 +1,6 @@
+module.exports = {
+  delay: require('./delay'),
+  error: require('./error'),
+  abort: require('./abort'),
+  throttle: require('./throttle')
+}
diff --git a/lib/poisons/throttle.js b/lib/poisons/throttle.js
@@ -0,0 +1,8 @@
+const Throttle = require('throttle')
+
+module.exports = function throttle(opts, filter) {
+  return function (req, res, next) {
+    res.pipe(new Throttle(opts))
+    next()
+  }
+}
diff --git a/lib/toxify.js b/lib/toxify.js
@@ -0,0 +1,42 @@
+const rocky = require('rocky')
+const Base = rocky.Base
+const Rocky = rocky.Rocky
+const poisons = require('./poisons')
+
+module.exports = Toxify
+
+function Toxify(opts) {
+  Rocky.call(this, opts)
+}
+
+Toxify.prototype = Object.create(Rocky.prototype)
+
+Base.prototype.percentage = function (percent) {
+  this.opts.percentage = Math.max(percent, 100)
+  return this
+}
+
+Base.prototype.usePoison = function (poison) {
+  var percent = this.opts.percentage
+
+  if (!percent) {
+    this.use(poison)
+    return this
+  }
+
+  this.use(function (req, res, next) {
+    var rand = (Math.random() * 10) | 0
+    var perc = (percent * 0.1) | 0
+    if (rand > perc) return next()
+    poison(req, res, next)
+  })
+
+  return this
+}
+
+Object.keys(poisons).forEach(function (poison) {
+  var fn = poisons[poison]
+  Base.prototype[fn.name || poison] = function () {
+    this.usePoison(fn.apply(null, arguments))
+  }
+})
diff --git a/package.json b/package.json
@@ -0,0 +1,38 @@
+{
+  "name": "toxify",
+  "version": "0.1.0",
+  "description": "Hackable HTTP proxy to simulate system failure conditions",
+  "repository": "h2non/rocky",
+  "author": "Tomas Aparicio",
+  "license": "MIT",
+  "keywords": [
+    "http",
+    "proxy",
+    "toxic",
+    "fault",
+    "tolerant",
+    "testing",
+    "resilency",
+    "fuzz",
+    "evil",
+    "testing"
+  ],
+  "engines": {
+    "node": ">= 0.12"
+  },
+  "scripts": {
+    "test": "./node_modules/.bin/mocha --timeout 5000 --reporter spec --ui tdd test/*"
+  },
+  "dependencies": {
+    "rocky": "^0.3.2",
+    "throttle": "^1.0.3",
+    "yargs": "^3.15.0"
+  },
+  "devDependencies": {
+    "chai": "^3.0.0",
+    "mocha": "^2.2.5",
+    "request": "^2.58.0",
+    "sinon": "^1.15.3",
+    "supertest": "^1.0.1"
+  }
+}
diff --git a/test/tofixy.js b/test/tofixy.js
