Adding pamela but I'm dumb and messed up routing

This commit is contained in:
Will Bradley 2013-01-31 20:33:40 -07:00
parent 6673573e36
commit a7e999614c
21 changed files with 364 additions and 1 deletions

View File

@ -0,0 +1,3 @@
# Place all the behaviors and hooks related to the matching controller here.
# All this logic will automatically be available in application.js.
# You can use CoffeeScript in this file: http://jashkenas.github.com/coffee-script/

View File

@ -0,0 +1,3 @@
// Place all the styles related to the pamela controller here.
// They will automatically be included in application.css.
// You can use Sass (SCSS) here: http://sass-lang.com/

View File

@ -0,0 +1,189 @@
class PamelaController < ApplicationController
#require "active_record"
require "optparse"
#require "rubygems"
def index
@macs = Mac.all
end
# GET /macs/1
# GET /macs/1.json
def show
@mac = Mac.find(params[:id])
respond_to do |format|
format.html # show.html.erb
format.json { render :json => @macs }
end
end
# GET /macs/new
# GET /macs/new.json
def new
@mac = Mac.new
@users = User.all.sort_by(&:name)
respond_to do |format|
format.html # new.html.erb
format.json { render :json => @mac }
end
end
# GET /macs/1/edit
def edit
@mac = Mac.find(params[:id])
@users = User.all.sort_by(&:name)
end
# POST /macs
# POST /user
def create
@mac = Mac.new(params[:mac])
respond_to do |format|
if @mac.save
format.html { redirect_to "/pamela", :notice => 'Mac was successfully created.' }
format.json { render :json => @mac, :status => :created, :location => @mac }
else
format.html { render :action => "new" }
format.json { render :json => @mac.errors, :status => :unprocessable_entity }
end
end
end
# PUT /macs/1
# PUT /macs/1.json
def update
#Log who updated this
@mac = Mac.find(params[:id])
respond_to do |format|
if @mac.update_attributes(params[:mac])
format.html { redirect_to "/pamela", :notice => 'User certification was successfully updated.' }
format.json { head :no_content }
else
format.html { render :action => "edit" }
format.json { render :json => @mac.errors, :status => :unprocessable_entity }
end
end
end
def scan
Rails.logger.info "starting scan..."
# Command line arguments
options = {};
OptionParser.new { |opts|
opts.banner = "Usage: pamela-scanner.rb --interface=en0"
options[:verbose] = true
opts.on("v", "--verbose", "Run verbosely") { |verbose|
options[:verbose] = verbose
}
options[:interface] = "eth0"
opts.on("i", "--interface=interface", "Network Interface") { |interface|
options[:interface] = interface
}
options[:max_age] = 20
opts.on("a", "--max-age=minutes", "Minutes to keep expired macs active") { |max_age|
options[:max_age] = max_age.to_i
}
options[:db_host] = "configure_me"
opts.on("r", "--db-host=host", "Database Host") { |host|
options[:db_host] = host
}
options[:db_name] = "configure_me"
opts.on("n", "--db-name=name", "Database Name") { |name|
options[:db_name] = name
}
options[:db_user] = "configure_me"
opts.on("u", "--db-user=user", "Database User") { |user|
options[:db_user] = user
}
options[:db_password] = "configure_me"
opts.on("p", "--db-password=password", "Database Password") { |password|
options[:db_password] = password
}
}.parse!
# Open the database
#ActiveRecord::Base::establish_connection(
# :adapter => "mysql2",
# :host => options[:db_host],
# :database => options[:db_name],
# :username => options[:db_user],
# :password => options[:db_password])
#class Mac < ActiveRecord::Base
#end
#class MacLog < ActiveRecord::Base
#end
# Scan the network for mac addresses
macs = {};
command = sprintf("arp-scan -R --interface=%s --localnet", options[:interface])
if options[:verbose]
Rails.logger.info "Running [#{command}]"
end
IO.popen(command) { |stdin|
Rails.logger.info "Reading stdin: "+stdin.inspect
stdin.each { |line|
next if line !~ /^([\d\.]+)\s+([[:xdigit:]:]+)\s/;
macs[$2] = $1;
}
}
# Scan the existing macs and update each record as necessary
Mac.find(:all).each { |entry|
Rails.logger.info "Existing MAC: "+entry.inspect
mac = entry.mac
ip = entry.ip
if macs.has_key?(mac)
if ! entry.active || ! entry.since
Rails.logger.info "Activating #{mac} at #{ip}" if options[:verbose]
entry.since = Time.now
MacLog.new(:mac => mac, :ip => ip, :action => "activate").save
end
entry.active = 1
entry.ip = ip
entry.refreshed = Time.now
entry.save
macs.delete(mac)
next
end
# Entry is no longer current
if entry.active
ageMinutes = ((Time.now - entry.refreshed)/60).to_i
next if ageMinutes < options[:max_age]
Rails.logger.info "Deactivating #{mac}, #{ageMinutes} minutes old" if options[:verbose]
entry.active = 0
entry.save
MacLog.new(:mac => mac, :ip => ip, :action => "deactivate").save
end
}
# Add entries for any macs not already in the db
macs.each { |mac, ip|
Rails.logger.info "Activating new entry #{mac} at #{ip}" if options[:verbose]
Mac.new(:mac => mac, :ip => ip, :active => 1, :since => Time.now, :refreshed => Time.now).save
MacLog.new(:mac => mac, :ip => ip, :action => "activate").save
}
@log = MacLog.all
end #def scan
end

View File

@ -0,0 +1,2 @@
module PamelaHelper
end

4
app/models/mac.rb Normal file
View File

@ -0,0 +1,4 @@
class Mac < ActiveRecord::Base
belongs_to :user
attr_accessible :active, :ip, :mac, :refreshed, :since
end

3
app/models/mac_log.rb Normal file
View File

@ -0,0 +1,3 @@
class MacLog < ActiveRecord::Base
attr_accessible :action, :ip, :mac
end

View File

@ -0,0 +1,25 @@
<%= form_for(@mac) do |f| %>
<% if @mac.errors.any? %>
<div id="error_explanation">
<h2><%= pluralize(@mac.errors.count, "error") %> prohibited this Mac from being saved:</h2>
<ul>
<% @mac.errors.full_messages.each do |msg| %>
<li><%= msg %></li>
<% end %>
</ul>
</div>
<% end %>
<div class="field">
<%= f.label :user_id, "User" %><br />
<%= collection_select(:mac, :user_id, @users, :id, :name) %>
</div>
<div class="field">
<%= f.label :mac, "Mac" %><br />
<%= f.text_field :mac %>
</div>
<div class="actions">
<%= f.submit %>
</div>
<% end %>

View File

@ -0,0 +1,6 @@
<h1>Editing User Certification</h1>
<%= render 'form' %>
<%= link_to 'Show', @user_certification %> |
<%= link_to 'Back', user_certifications_path %>

View File

@ -0,0 +1,9 @@
<h2>What machines are on our network?</h2>
<% @macs.each do |mac| %>
<%= mac.user.name unless mac.user.blank? %>
<%= mac.mac %> <!--
<%= mac.since %>,
<%= mac.refreshed %>,
<%= mac.active %>--> |
<%= link_to 'Edit', edit_mac_path(mac) %> <br/>
<% end %>

View File

@ -0,0 +1,5 @@
<h1>New User Certification</h1>
<%= render 'form' %>
<%= link_to 'Back', user_certifications_path %>

View File

@ -0,0 +1,5 @@
Scanning...
<% @log.each do |log| %>
<%= log.mac %> =
<%= log.ip %><br/>
<% end %>

View File

@ -27,6 +27,15 @@ Dooraccess::Application.routes.draw do
match 'door_logs/download' => 'door_logs#download', :as => :download
match 'door_logs/auto_download' => 'door_logs#auto_download', :as => :auto_download
match 'pamela/scan' => 'pamela#scan'
match 'pamela' => 'pamela#index', :as => :macs
match 'pamela/:id' => 'pamela#show', :as => :mac
match 'pamela/:id/edit' => 'pamela#edit', :as => :edit_mac
put 'pamela/:id' => 'pamela#update' #PUT
match 'pamela/new' => 'pamela#new', :as => :new_mac
post 'pamela' => 'pamela#create' #POST
root :to => "home#index"
# The priority is based upon order of creation:

View File

@ -0,0 +1,15 @@
class CreateMacs < ActiveRecord::Migration
def change
create_table :macs do |t|
t.references :user
t.string :mac
t.string :ip
t.datetime :since
t.datetime :refreshed
t.boolean :active
t.timestamps
end
add_index :macs, :user_id
end
end

View File

@ -0,0 +1,11 @@
class CreateMacLogs < ActiveRecord::Migration
def change
create_table :mac_logs do |t|
t.string :mac
t.string :ip
t.string :action
t.timestamps
end
end
end

View File

@ -10,7 +10,7 @@
#
# It's strongly recommended to check this file into your version control system.
ActiveRecord::Schema.define(:version => 20130125185724) do
ActiveRecord::Schema.define(:version => 20130201022153) do
create_table "cards", :force => true do |t|
t.string "card_number"
@ -35,6 +35,27 @@ ActiveRecord::Schema.define(:version => 20130125185724) do
t.datetime "updated_at", :null => false
end
create_table "mac_logs", :force => true do |t|
t.string "mac"
t.string "ip"
t.string "action"
t.datetime "created_at", :null => false
t.datetime "updated_at", :null => false
end
create_table "macs", :force => true do |t|
t.integer "user_id"
t.string "mac"
t.string "ip"
t.datetime "since"
t.datetime "refreshed"
t.boolean "active"
t.datetime "created_at", :null => false
t.datetime "updated_at", :null => false
end
add_index "macs", ["user_id"], :name => "index_macs_on_user_id"
create_table "user_certifications", :force => true do |t|
t.integer "user_id"
t.integer "certification_id"

11
test/fixtures/mac_logs.yml vendored Normal file
View File

@ -0,0 +1,11 @@
# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
one:
mac: MyString
ip: MyString
action: MyString
two:
mac: MyString
ip: MyString
action: MyString

17
test/fixtures/macs.yml vendored Normal file
View File

@ -0,0 +1,17 @@
# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
one:
user:
mac: MyString
ip: MyString
since: 2013-01-31 19:19:49
refreshed: 2013-01-31 19:19:49
active: false
two:
user:
mac: MyString
ip: MyString
since: 2013-01-31 19:19:49
refreshed: 2013-01-31 19:19:49
active: false

View File

@ -0,0 +1,7 @@
require 'test_helper'
class PamelaControllerTest < ActionController::TestCase
# test "the truth" do
# assert true
# end
end

View File

@ -0,0 +1,4 @@
require 'test_helper'
class PamelaHelperTest < ActionView::TestCase
end

View File

@ -0,0 +1,7 @@
require 'test_helper'
class MacLogTest < ActiveSupport::TestCase
# test "the truth" do
# assert true
# end
end

7
test/unit/mac_test.rb Normal file
View File

@ -0,0 +1,7 @@
require 'test_helper'
class MacTest < ActiveSupport::TestCase
# test "the truth" do
# assert true
# end
end