tochka 0.1.1

data/lib/tochka/pitft_button.rb

@@ -0,0 +1,105 @@
+module Tochka
+  require "wiringpi"
+
+  class PiTFTButton
+    PIN1 = 4
+    PIN2 = 3
+    PIN3 = 2
+    PIN4 = 1
+
+    PIN_MODE_INPUT=0
+
+    def initialize
+      if ENV['BUTTON_MOCK'] != nil
+        dummy_init
+      else
+        real_init
+      end
+
+      @prev = [false, false, false, false]
+
+    end
+
+    def dummy_init
+      # do nothing
+    end
+
+    def real_init
+      @io = WiringPi::GPIO.new
+
+      [PIN1, PIN2, PIN3, PIN4].each do |pin|
+        @io.pin_mode(pin, PIN_MODE_INPUT)
+        @io.pull_up_dn_control(pin, WiringPi::PUD_UP)
+      end
+
+      backlight_on()
+    end
+
+    def button_all
+      return [button1, button2, button3, button4]
+    end
+
+    def button1
+      read_button(PIN1)
+    end
+
+    def button2
+      read_button(PIN2)
+    end
+
+    def button3
+      read_button(PIN3)
+    end
+
+    def button4
+      read_button(PIN4)
+    end
+
+    def button_all_edge
+      buttons = button_all()
+      edges = []
+      0.upto(3) do |n|
+        if buttons[n] == true and buttons[n] != @prev[n]
+          edges[n] = true
+        else
+          edges[n] = false
+        end
+      end
+
+      @prev = buttons
+      return edges
+    end
+
+    def backlight_on
+      set_backlight(1)
+    end
+
+    def backlight_off
+      set_backlight(0)
+    end
+
+    private
+    def read_button pin
+      return @io.digital_read(pin) == 0
+    end
+
+    def set_backlight val
+      File.open("/sys/class/gpio/gpio508/value", "w") do |f|
+        f.write(val.to_s)
+      end
+    end
+  end
+end
+
+if __FILE__ == $0
+  pitftb = Tochka::PiTFTButton.new
+  while true
+    buttons = pitftb.button_all_edge
+    0.upto(3) do |n|
+      bn = n + 1
+      if buttons[n] == true
+        print "Button #{bn} is On"
+      end
+    end
+  end
+end

data/lib/tochka/version.rb

@@ -0,0 +1,3 @@
+module Tochka
+  VERSION = "0.1.1"
+end

data/lib/tochka/wlan.rb

@@ -0,0 +1,113 @@
+module Tochka
+  require "open3"
+
+  require "tochka/athsurvey"
+
+  class Wlan
+    attr_reader :duration, :current_channel, :file_size, :channel_walk, :utilization, :utilization_channel
+    CHAN = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13,
+      34, 36, 38, 40, 42, 44, 46, 48,
+      52, 56, 60, 64,
+      100, 104, 108, 112, 116,
+      120, 124, 128, 132, 136, 140,
+      149, 153, 157, 161, 165]
+    LIMIT_IDX=CHAN.length
+
+    def initialize ifname=DEFAULT_IFNAME
+      @th_shark = nil
+      @ifname = ifname || DEFAULT_IFNAME
+      init_status()
+      @black_list = []
+      @athsurvey = Tochka::Athsurvey.new(@ifname)
+    end
+
+    def run_capture fpath
+      init_device()
+      init_status()
+      start_time = Time.now.to_i
+
+      stdin, stdout, stderr, @th_tshark = *Open3.popen3(
+        "tshark -i #{@ifname} -F pcapng -w #{fpath}")
+
+        while @th_tshark.alive?
+          sleep 1
+
+          # update status
+          @duration = Time.now.to_i - start_time
+          @file_size = File.size?(fpath) || 0
+
+          # do something here to run before channel transition
+          ary = @athsurvey.current_data()
+          @utilization_channel = ary[0]
+          @utilization = ary[3]
+
+          prev_channel = @current_channel
+          @current_chanenl = move_channel(@current_channel)
+          @channel_walk += 1
+          $log.debug("channel moved to #{@current_channel} from #{prev_channel} (dur=#{@duration}, size=#{@file_size}, walk=#{@channel_walk}, utilization=#{@utilization} uch=#{@utilization_channel})")
+          end
+        rescue => e
+          $log.warn("run_capture detected unknown error (#{e})")
+        end
+
+        def stop_capture
+          if @th_tshark == nil
+            $log.err("tried to kill tshark, but it's not executed? (or already dead?)")
+            return
+          end
+
+          Process.kill("INT", @th_tshark.pid)
+        end
+
+        private
+        def init_status
+          @duration = 0
+          @current_channel = 1
+          @file_size = 0
+          @channel_walk = 0
+        end
+
+        def init_device
+          unless system("ip link set #{@ifname} down")
+            raise "failed to turn down #{@ifname}"
+          end
+          unless system("iw #{@ifname} set monitor fcsfail otherbss control")
+            raise "failed to set #{@ifname} to monitor mode"
+          end
+          unless system("ip link set #{@ifname} up")
+            raise "failed to turn up #{@ifname}"
+          end
+          unless system("iw wlan0 set channel #{@current_channel}")
+            raise "failed to set channel #{@current_channel} on #{@ifname}"
+          end
+        end
+
+        def move_channel current
+          next_channel = pick_channel(current)
+
+          while !system("iw #{@ifname} set channel #{next_channel}")
+            # what if we got unplugged ifname? => should we die?
+            $log.debug("channel transition failed, added to black list (channel=#{next_channel})")
+              @black_list << next_channel
+            sleep 1
+            next_channel = pick_channel(next_channel)
+          end
+
+          @current_channel = next_channel
+        end
+
+        def pick_channel current
+          idx = CHAN.index(current)
+          idx += 1
+          next_channel = CHAN[idx % LIMIT_IDX]
+
+          # we have black list
+          while @black_list.include?(next_channel)
+            idx += 1
+            next_channel = CHAN[idx % LIMIT_IDX]
+          end
+
+          return next_channel
+        end
+    end
+  end

data/lib/tochka.rb

@@ -0,0 +1,11 @@
+require "tochka/version"
+require "tochka/agent"
+require "tochka/channel"
+require "tochka/log"
+require "tochka/pitft_button"
+require "tochka/wlan"
+require "tochka/command/tochka-miniui"
+require "tochka/command/tochkad"
+
+module Tochka
+end

data/tochka.gemspec

@@ -0,0 +1,35 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'tochka/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "tochka"
+  spec.version       = Tochka::VERSION
+  spec.authors       = ["enukane"]
+  spec.email         = ["enukane@glenda9.org"]
+
+  spec.summary       = %q{Wi-Fi packet capture software set on Raspberry Pi.}
+  spec.homepage      = "https://github.com/enukane/tochka"
+
+  # Prevent pushing this gem to RubyGems.org by setting 'allowed_push_host', or
+  # delete this section to allow pushing this gem to any host.
+  #if spec.respond_to?(:metadata)
+  #  spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'"
+  #else
+  #  raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
+  #end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.10"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec"
+
+  spec.add_dependency 'wiringpi', "~> 2.0.0"
+  spec.add_dependency 'rubysdl', "~> 2.2.0" # requires sge, ttf enabled
+  spec.add_dependency 'pidfile', "~> 0.3.0"
+end

metadata

@@ -0,0 +1,152 @@
+--- !ruby/object:Gem::Specification
+name: tochka
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- enukane
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-12-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: wiringpi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.0.0
+- !ruby/object:Gem::Dependency
+  name: rubysdl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.2.0
+- !ruby/object:Gem::Dependency
+  name: pidfile
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.0
+description: 
+email:
+- enukane@glenda9.org
+executables:
+- tochka-miniui
+- tochkad
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/tochka-miniui
+- exe/tochkad
+- ext/tochka-miniui
+- ext/tochkad
+- lib/tochka.rb
+- lib/tochka/agent.rb
+- lib/tochka/athsurvey.rb
+- lib/tochka/channel.rb
+- lib/tochka/command/tochka-miniui.rb
+- lib/tochka/command/tochkad.rb
+- lib/tochka/log.rb
+- lib/tochka/pitft_button.rb
+- lib/tochka/version.rb
+- lib/tochka/wlan.rb
+- tochka.gemspec
+homepage: https://github.com/enukane/tochka
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Wi-Fi packet capture software set on Raspberry Pi.
+test_files: []