tochka 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 40dc9ab742d3ff31b82292795a7b905782210488
+  data.tar.gz: 771f090404e1262e0f0e174ac961e60f0cf099f9
+SHA512:
+  metadata.gz: 4b6b2639eecb4a961fa4de138a456b71d0f03167e67549a37ed8e4a3a3697b85288295923ac1f24207e56e50d11005d1f566f654a98f04f41f4bd47f16d970ad
+  data.tar.gz: a7a73e80b4536856fdb045d4ab911f598a1f5acf28db7ea5a5676053970bb5d171ceecd401b88ae2909313fc8c6d3b54eebd72ea8855a0babe9d46b2b8df7dd4

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,4 @@
+language: ruby
+rvm:
+  - 1.9.3
+before_install: gem install bundler -v 1.10.6

data/Gemfile

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

data/README.md

@@ -0,0 +1,64 @@
+# Tochka
+
+"Tochka" is softwares to build Wi-Fi frame capturing box with Raspberry Pi +  PiTFT.
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/tochka`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'tochka'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install tochka
+
+### install init.d scripts
+
+1. copy ext/tochkad, ext/tochka-miniui into /etc/init.d
+2. do chmod +x onto both
+3. do 'update-rc.d <name> defaults' to both
+
+## Usage
+
+### PiTFT UI
+
+Tochka provides with user interface to start/stop capture on top of PiTFT.
+
+![tochka-miniui](http://amber.glenda9.org/~enukane/images/tochka-miniui.jpg)
+
+TOUCHSCREEN is currently of NO USE. Just make yourself happy with moving cursor :).
+Use buttons instead: 4 buttons are "START", "STOP", "MODE1", "MODE2".
+
+### Starting capture
+
+Press "START" button and "State" turns into "running"
+
+### Stop capture
+
+Press "STOP", "MODE1" and "MODE2" buttons in an order. Check that "stop count" is now 3.
+Then press "STOP" button again will turn off capturing. The "State" will turn into "stop".
+
+### collecting captured data
+
+Captured data (pcapng files) are stored in /cap directory
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment. Run `bundle exec tochka` to use the gem in this directory, ignoring other installed copies of this gem.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/enukane/tochka.
+

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "tochka"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start

data/bin/setup

@@ -0,0 +1,7 @@
+#!/bin/bash
+set -euo pipefail
+IFS=$'\n\t'
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/exe/tochka-miniui

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+
+require "tochka"
+require "optparse"
+
+opt = OptionParser.new
+OPTS=Tochka::TochkaMiniUI.default_options
+
+opt.on('-f', "--font [FONT_PATH]", "font to use (default: #{Tochka::TochkaMiniUI.default_options[:font_path]})") {|v|
+  OPTS[:font_path] = v
+}
+
+opt.on('-l', "--log-file [LOG_FILE=STDOUT]",
+       "log output file (accepts 'STDOUT', 'STDERR')") {|v|
+  OPTS[:log_file] = v
+}
+
+(class<<self;self;end).module_eval do
+  define_method(:usage) do |msg|
+    puts opt.to_s
+    puts "error: #{msg}" if msg
+    exit 1
+  end
+end
+
+begin
+  res = opt.parse(ARGV)
+  if res.length != 0
+    usage nil
+  end
+rescue
+  usage $!.to_s
+end
+
+$log = Log.new(:output => OPTS[:log_file])
+Tochka::TochkaMiniUI.new(OPTS[:font_path]).run

data/exe/tochkad

@@ -0,0 +1,44 @@
+#!/usr/bin/env ruby
+
+require "tochka"
+require "optparse"
+
+opt = OptionParser.new
+OPTS=Tochka::Daemon.default_options
+
+opt.on('-i', "--interface [IFNAME=#{Tochka::Daemon::DEFAULT_IFNAME}]",
+       "interface to capture") {|v|
+  OPTS[:ifname] = v || Tochka::Daemon::DEFAULT_IFNAME
+}
+
+opt.on('-c', "--cap-dir [CAP_PATH=#{Tochka::Daemon::DEFAULT_CAP_PATH}]",
+       "capture directory") {|v|
+  OPTS[:cap_path] = v || TOchka::Daemon::DEFAULT_CAP_PATH
+}
+
+opt.on('-l', "--log-file [LOG_FILE=STDOUT]",
+       "log output file (accepts 'STDOUT', 'STDERR')") {|v|
+  OPTS[:log_file] = v
+}
+
+(class<<self;self;end).module_eval do
+  define_method(:usage) do |msg|
+    puts opt.to_s
+    puts "error: #{msg}" if msg
+    exit 1
+  end
+end
+
+begin
+  rest = opt.parse(ARGV)
+  if rest.length != 0
+    usage nil
+  end
+rescue
+  usage $!.to_s
+end
+
+# check root priviledge
+
+$log = Log.new(:output => OPTS[:log_file])
+Tochka::Daemon.new(OPTS[:ifname], OPTS[:cap_path]).run

data/ext/tochka-miniui

@@ -0,0 +1,62 @@
+#!/bin/sh
+
+### BEGIN INIT INFO
+# Provides:          tochka-mini
+# Required-Start:    $remote_fs $syslog
+# Required-Stop:     $remote_fs $syslog
+# Default-Start:     2 3 4 5
+# Default-Stop:      0 1 6
+# Short-Description: Put a short description of the service here
+# Description:       Put a long description of the service here
+### END INIT INFO
+
+# Change the next 3 lines to suit where you install your script and what you want to call it
+DIR=/usr/local/bin/
+DAEMON=$DIR/tochka-miniui
+DAEMON_NAME=tochka-miniui
+
+# Add any command line options for your daemon here
+DAEMON_OPTS="-l /tmp/hoge"
+
+# This next line determines what user the script runs as.
+# Root generally not recommended but necessary if you are using the Raspberry Pi GPIO from Python.
+DAEMON_USER=root
+
+# The process ID of the script when it runs is stored here:
+PIDFILE=/var/run/$DAEMON_NAME.pid
+
+. /lib/lsb/init-functions
+
+do_start () {
+    log_daemon_msg "Starting system $DAEMON_NAME daemon"
+    start-stop-daemon --start --background --pidfile $PIDFILE --user $DAEMON_USER --chuid $DAEMON_USER --startas $DAEMON -- $DAEMON_OPTS
+    log_end_msg $?
+}
+do_stop () {
+    log_daemon_msg "Stopping system $DAEMON_NAME daemon"
+    start-stop-daemon --stop --pidfile $PIDFILE --retry 10
+    log_end_msg $?
+}
+
+case "$1" in
+
+    start|stop)
+        do_${1}
+        ;;
+
+    restart|reload|force-reload)
+        do_stop
+        do_start
+        ;;
+
+    status)
+        status_of_proc "$DAEMON_NAME" "$DAEMON" && exit 0 || exit $?
+        ;;
+
+    *)
+        echo "Usage: /etc/init.d/$DAEMON_NAME {start|stop|restart|status}"
+        exit 1
+        ;;
+
+esac
+exit 0

data/ext/tochkad

@@ -0,0 +1,62 @@
+#!/bin/sh
+
+### BEGIN INIT INFO
+# Provides:          tochkad
+# Required-Start:    $remote_fs $syslog
+# Required-Stop:     $remote_fs $syslog
+# Default-Start:     2 3 4 5
+# Default-Stop:      0 1 6
+# Short-Description: Put a short description of the service here
+# Description:       Put a long description of the service here
+### END INIT INFO
+
+# Change the next 3 lines to suit where you install your script and what you want to call it
+DIR=/usr/local/bin/
+DAEMON=$DIR/tochkad
+DAEMON_NAME=tochkad
+
+# Add any command line options for your daemon here
+DAEMON_OPTS=""
+
+# This next line determines what user the script runs as.
+# Root generally not recommended but necessary if you are using the Raspberry Pi GPIO from Python.
+DAEMON_USER=root
+
+# The process ID of the script when it runs is stored here:
+PIDFILE=/var/run/$DAEMON_NAME.pid
+
+. /lib/lsb/init-functions
+
+do_start () {
+    log_daemon_msg "Starting system $DAEMON_NAME daemon"
+    start-stop-daemon --start --background --pidfile $PIDFILE --user $DAEMON_USER --chuid $DAEMON_USER --startas $DAEMON -- $DAEMON_OPTS
+    log_end_msg $?
+}
+do_stop () {
+    log_daemon_msg "Stopping system $DAEMON_NAME daemon"
+    start-stop-daemon --stop --pidfile $PIDFILE --retry 10
+    log_end_msg $?
+}
+
+case "$1" in
+
+    start|stop)
+        do_${1}
+        ;;
+
+    restart|reload|force-reload)
+        do_stop
+        do_start
+        ;;
+
+    status)
+        status_of_proc "$DAEMON_NAME" "$DAEMON" && exit 0 || exit $?
+        ;;
+
+    *)
+        echo "Usage: /etc/init.d/$DAEMON_NAME {start|stop|restart|status}"
+        exit 1
+        ;;
+
+esac
+exit 0

data/lib/tochka/agent.rb

@@ -0,0 +1,116 @@
+module Tochka
+  require "json"
+  require "socket"
+  require "tochka/log"
+
+  class Agent
+    attr_reader :state, :file_name, :file_size, :duration, :current_channel, :frame_count, :channel_walk, :utilization, :utilization_channel
+    SOCKPATH="/var/run/captured.sock"
+
+    CMD_GET_STATUS="get_status"
+    CMD_START_CAPTURE="start_capture"
+    CMD_STOP_CAPTURE="stop_capture"
+
+    STATE_INIT="init"
+    STATE_RUNNING="running"
+    STATE_STOP="stop"
+    STATE_UNKNOWN="unknown"
+
+    def initialize
+      @state = STATE_INIT
+      @file_name = "<None>"
+      @file_size = 0
+      @duration = 0
+      @current_channel = 0
+      @channel_walk = 0
+      @frame_count = 0
+      @utilization = 0
+      @utilization_channel = 0
+    end
+
+    def get_status
+      resp = do_rpc(get_msg(CMD_GET_STATUS))
+      if resp["state"] != nil
+        update_state(resp)
+        return resp
+      end
+
+      # not responded?
+      @state = STATE_UNKNOWN
+      return {}
+    rescue => e
+      $log.err("agent detected unknown error in get_status (#{e})")
+      return {}
+    end
+
+    def start_capture
+      $log.info("requesting start_capture")
+      resp = do_rpc(get_msg(CMD_START_CAPTURE))
+      resp = get_status()
+      if resp["state"] == STATE_RUNNING
+        update_state(resp)
+        return true
+      end
+
+      # not responded or failed
+      @state = STATE_UNKNOWN
+      return false
+    rescue => e
+      $log.err("agent detected unknown error in start_capture (#{e})")
+      return false
+    end
+
+    def stop_capture
+      $log.info("requesting stop_capture")
+      resp = do_rpc(get_msg(CMD_STOP_CAPTURE))
+      resp = get_status()
+      if resp["state"] == STATE_STOP
+        update_state(resp)
+        return true
+      end
+
+      return false
+    rescue => e
+      $log.err("agent detected unknown error in stop_capture (#{e})")
+      return false
+    end
+
+    private
+    def get_msg type=CMD_GET_STATUS
+      JSON.dump(
+        {"command"=> type}
+      )
+    end
+
+    def do_rpc req_json_msg
+      data = {}
+      UNIXSocket.open(SOCKPATH) do |sock|
+        sock.write(req_json_msg+"\n")
+        data = sock.gets
+      end
+      return JSON.parse(data)
+    rescue Errno::EPIPE => e
+      $log.err("RPC failed in do_rpc (EPIPE)")
+      return {}
+    rescue JSON::ParserError => e0
+      $log.err("RPC failed in do_rpc (JSON parser error)")
+      return {}
+    rescue Errno::ENOENT => e1
+      $log.err("RPC failed in do_rpc (no sock file, #{e1})")
+        return {}
+    end
+
+    def update_state msg
+      @state = msg["state"]
+      @file_name = msg["file_name"]
+      @file_size = msg["file_size"]
+      @duration = msg["duration"]
+      @current_channel = msg["current_channel"]
+      @channel_walk = msg["channel_walk"]
+      @frame_count = msg["frame_count"]
+      @utilization = msg["utilization"]
+      @utilization_channel = msg["utilization_channel"]
+    end
+  end
+
+end

data/lib/tochka/athsurvey.rb

@@ -0,0 +1,109 @@
+module Tochka
+  class Athsurvey
+    FREQ2CHAN = {
+      0     => 0, # why radiotap.channel.freq can be 0?
+      # 2.4GHz
+      2412  => 1,
+      2417  => 2,
+      2422  => 3,
+      2427  => 4,
+      2432  => 5,
+      2437  => 6,
+      2442  => 7,
+      2447  => 8,
+      2452  => 9,
+      2457  => 10,
+      2462  => 11,
+      2467  => 12,
+      2472  => 13,
+      2484  => 14,
+
+      # japan client only
+      5170  => 34,
+      5190  => 38,
+      5210  => 42,
+      5230  => 46,
+
+      # W52
+      5180  => 36,
+      5200  => 40,
+      5220  => 44,
+      5240  => 48,
+
+      # W53
+      5260  => 52,
+      5280  => 56,
+      5300  => 60,
+      5320  => 64,
+
+      # W56
+      5500  => 100,
+      5520  => 104,
+      5540  => 108,
+      5560  => 112,
+      5580  => 116,
+      5600  => 120,
+      5620  => 124,
+      5640  => 128,
+      5660  => 132,
+      5680  => 136,
+      5700  => 140,
+      5745  => 149,
+      5765  => 153,
+      5785  => 157,
+      5805  => 161,
+      5825  => 165,
+
+
+      # 802.11j
+      4920  => 184,
+      4940  => 188,
+      4960  => 192,
+      4980  => 194,
+    }
+
+    REG_FREQ=/^\s*frequency:\s+(\d+) /
+    REG_ACTIVE=/^\s*channel active time:\s+(\d+) (|m)s$/
+    REG_BUSY=/^\s*channel busy time:\s+(\d+) (|m)s$/
+
+    def initialize ifname
+      @ifname = ifname
+    end
+
+    def current_data
+      str = get_survey_dump()
+      active = 0
+      busy = 0
+      channel = 0
+
+      str.split("\n").map{|line| line.strip}.each do |line|
+        case line
+        when REG_FREQ
+          channel = FREQ2CHAN[$1.to_i]
+        when REG_ACTIVE
+          active = $1.to_i
+          active *= 1000 if $2 != "m"
+        when REG_BUSY
+          busy = $1.to_i
+          busy *= 1000 if $2 != "m"
+        end
+      end
+      return [0, 0, 0, 0] if active == 0
+
+      return [channel, active, busy, fto2f(busy.to_f * 100 / active.to_f) ]
+    end
+
+    def get_survey_dump
+      return `iw #{@ifname} survey dump | grep -A 4 "in use"`
+    end
+
+    def fto2f f
+      return (f * 100).to_i.to_f / 100
+    end
+  end
+end
+
+if __FILE__ == $0
+aths = Tochka::Athsurvey.new("wlan0")
+p aths.current_data
+end

data/lib/tochka/channel.rb

@@ -0,0 +1,39 @@
+module Tochka
+  require "socket"
+
+  class Channel
+    def initialize recv_handler, sock_data
+    end
+
+    def start
+    end
+
+    def stop
+    end
+  end
+
+  class UnixSocketChannel < Channel
+    SOCKPATH="/var/run/captured.sock"
+
+    def initialize recv_handler, sock_path=SOCKPATH
+      @recv_handler = recv_handler
+      @sock_path = sock_path || SOCKPATH
+      @th = nil
+    end
+
+    def start
+      @th = Thread.new do
+        Socket.unix_server_loop(@sock_path) do |sock, addr|
+          data = sock.gets
+          resp = @recv_handler.call(data)
+          sock.write(resp+"\n")
+        end
+      end
+    end
+
+    def stop
+      @th.kill
+    end
+  end
+
+end