One of the main tools that I've found useful in pen. testing is the Dradis Framework, it's a good way of keeping track of findings and notes during a test and I've also found it's template feature is good for keeping a list of things to remember during a test.
One of the features available in Dradis is import plugins. This lets you create a link to an external information source, such as a OSVDB or a database of vulnerabilities.
Having a database of vulnerabilities or findings can be pretty useful in cutting down the time required for reporting on a test as you can keep standard wordings in place (who really wants to write the same section about preventing XSS more than once!).
So recently I knocked up a simple vulnerability database to link in to Dradis and I thought it might be of use, so here's the process.
Creating the App
We're going to use Ruby on Rails for this as it's nice and easy to develop for (as you'll see) and also that's what Dradis is based on, so makes sense to keep all the coding in the same underlying language. Also rails apps are very portable, they're basically contained within a single directory structure, so it's relatively easy to move them from place to place.
Before starting the application, there's the usual pre-req's. I'm using Ruby 1.9.2 and Rails 3 so having those installed is a good thing. If you're using Linux then it's helpful to have RVM working as some distros don't have ruby 1.9.2 packaged up as yet.
once you've got the pre-req's working, we can start by creating a rails app

rails new vulnlist

This creates a new application called vulnlist and adds all the standard rails files in.
Creating the Scaffold
Once we've got the app created, we can use rails scaffolding to quickly create the basic structure for our app. The web pages that scaffolding creates aren't the most pretty, but they'll do for now.
With the scaffold we can specify what fields we want to create in the database and also what data types those fields are.

rails generate scaffold Vulnerability title:string test_type:string description:text remediation:text technical_notes:text severity:string owasp_reference:string

Once we've completed this we can look at the basic app by setting up the database with

rake db:migrate

Ensuring that all our gems are installed ok with

bundle install

and launching the app

rails server

At this point browsing to should show a blank page with our fields in it. From this page we can create new vulnerabilities and edit or delete existing ones.
Now that we've got this basic structure setup it's worth using git to keep a handle on the source code. On Linux the procedure for this is pretty easy
If you've not already got it installed

sudo apt-get install git-core

then in the root of the application

git init
git add .
git commit -m "Initial Commit with Scaffold"

Having git running on the app will make it pretty easy to revert any mistakes made along the way, as long as we've done regular commits.
Setting up the Searches
So far we've got a basic structure in place and can do the basic Create, Read, Update, Delete cycle on our data. However for the dradis integration, what'd be useful is if we could search for vulnerabilities using various criteria and have the results returned to Dradis.
This turns out to be relatively straightforward. First what we need is a new action in our controller. Opening vulnlist/app/controllers/vulnerabilities_controller.rb we can see the existing actions that we've got for the application.
What we need to do now is add a new action to allow for vulnerability searches.

def vuln_search
case params[:search_type]
when "description"
@vulnerabilities = Vulnerability.where("description like ?", "%"+params[:query]+"%")
when "owasp"
@vulnerabilities = Vulnerability.where("owasp_reference like ?",params[:query]+"%")
when "severity"
@vulnerabilities = Vulnerability.find_all_by_severity(params[:query])
when "test_type"
@vulnerabilities = Vulnerability.find_all_by_test_type(params[:query])
respond_to do |format|
format.xml {render :xml => @vulnerabilities}
format.json {render :json => @vulnerabilities}

This defines a new method called "vuln_search" which takes two parameters, search_type and query. The search type parameter lets us pick from different finders. Rails provides access into the application database via ActiveRecord and this just uses a couple of the finder types for different parameters. Where the query is going to be one of a number of fixed values like "severity" which will be something like high, medium or low, we can just use a standard find_all_by_ approach, but where it's a more free text style search, we use Vulnerability.where and pass in the query parameter that way.
The respond_to block is a really nice feature of rails. By adding in the two lines for :xml and :json rails wires up responses so that we can get the data out in those format, no additional code required.
Now that we've got the basic code in place, we just need to modify the rails routes so that the application knows how to access our new method.
This is done by modifying the vulnlist/config/routes.rb file, and adding the following code

controller :vulnerabilities do
get 'vuln_search' => :vuln_search

At this point, we've got the application basically working. If you put in a couple of test findings, then you should be able to go to for example and get some XML data back.
Tidying up
Now that we've got the basics working, there's a couple of additional steps that it's worth looking at to tidy some things up.
First off, we'd like some of our fields (OWASP Reference, Severity and test type) to be one of a number of defined values. The "proper" way to do this would be to create additional models for these and link them in to the main vulnerabilities controller, but there's a quicker way which is probably going to work well enough for our purposes.
Opening up vulnlist/app/models/vulnerability.rb we can specify some Constant values for these settings

TEST_TYPES = ["Web Application","Windows Server","Unix Server","Wireless","Web Server","Oracle","MS SQL","MySQL","DB2"]
SEVERITY_LEVELS = ["Critical","High","Medium","Low","No Impact"]
OWASP_TOP_10 = ["A1 - Injection","A2 - Cross Site Scripting (XSS)","A3 - Broken Authentication and Session Management","A4 - Insecure Direct Object Reference","A5 - Cross-Site Request Forgery (CSRF)","A6 - Security Misconfiguration","A7 - Insecure Cryptographic Storage","A8 - Failure to Restrict URL Access","A9 - Insufficient Transport Layer Protection","A10 - Unvalidated Redirects and Forwards"]

Then we can modify the form that the scaffolding process created to use these arrays as a select list. The form is found in vulnlist/app/views/vulnerabilities/_form.html.erb. In that file we just need to replace the "text_field" lines for those three fields with the following select lines

 "Select the test type" %>
 "Select the severity level" %>
 "Select the appropriate OWASP top 10 reference" %>

This picks up the Constants from our model and helps keep the data consistent.
Localhost Only
As you'll have noticed with this application, there's pretty much no security whatsoever. At the moment it's setup as a personal database only and isn't suitable to be on any kind of network. Adding that security isn't too difficult with rails, however it's not really a problem for the basic use case that we have here. Both the vulnerability list and the dradis installation only need to listen on the localhost.
Configuring rails to only listen on the localhost (as opposed to specifying it on the command line) is a bit hacky, but here's a way to do it based on this post and this dradis change . We need to modify the vulnlist/script/rails file and add the following lines

require 'rubygems'
require 'rails/commands/server'
require 'rack'
require 'webrick'
module Rails
class Server  3003,
:Host => "",
:environment => (ENV['RAILS_ENV'] || "development").dup,
:daemonize => false,
:debugger => false,
:pid => File.expand_path("tmp/pids/"),
:config => File.expand_path("")

This also moves the application off the default port of 3000, to a new one of 3003 which hopefully shouldn't clash with other services.
Default Routes
At the moment if we visit the root page of our application, now at we get the default rails welcome page. What'd be nicer is if we were re-directed to the vulnerability listing automatically.
That's easily done with two steps. First edit the vulnlist/config/routes.rb file and add the line

root :to => "vulnerabilities#index"

then delete the file vulnlist/public/index.html file.
So at the end of this first part we've created a basic vulnerability database which we can search easily on a number of parameters.
The next step is to create the dradis plugin to hook the two together, which as I'll cover next time is a reasonably easy thing to do.


Security Geek, Kubernetes, Docker, Ruby, Hillwalking