move scripts to new project

scripts/crc16.rb

@@ -1,24 +0,0 @@
-CRC_POLY = 0x8408
-
-def calc_crc(data, poly = CRC_POLY) 
-  state = 0;
-  data.each do |byte|
-    8.times do
-      if ((byte ^ state) & 0x01) == 0x01
-        state = (state >> 1) ^ poly;
-        #printf "^ "
-      else 
-        state = state >> 1;
-        #printf "> "
-      end
-      #printf "%08b %016b // %02X %04X \n", byte, state, byte, state
-      byte = byte >> 1;
-    end
-    #printf "---\n"
-  end
-  state
-end
-
-input = gets.strip
-r = calc_crc([input].pack('H*').bytes)
-printf "%04X (%d)\n", r, r

scripts/tests.rb

@@ -1,233 +0,0 @@
-require_relative 'helpers'
-
-COLOR_S2_BASES = [0x5A, 0x22, 0x30, 0x11]
-KELVIN_S2_BASES = COLOR_S2_BASES.map { |x| x + 0x80 }
-SATURATION_BRIGHTNESS_S2_BASES = [0x9A, 0x62, 0x70, 0x51]
-
-# saturation bases + 0x80
-COMMAND_BASES = [0x1A, 0xE2, 0xF0, 0xD1]
-
-GROUP_BASES = [0xAF, 0x04, 0xDD, 0x07]
-
-OFFSETTING_S2_FN = ->(bases, s, e) do
-  ->(b0) do
-    value = bases[ (b0 % 4) ]
-    if b0 >= s && b0 <= e
-      value = (value + 0x80) % 0x100
-    end
-    
-    [value]
-  end
-end
-
-extra_args = {
-  kelvin: {
-    increments: 2,
-    s1_range: [0x0C],
-    s2_range: OFFSETTING_S2_FN.call(KELVIN_S2_BASES, 0x14, 0x93)
-  },
-  color: {
-    s1_range: [0x15],
-    s2_range: OFFSETTING_S2_FN.call(COLOR_S2_BASES, 0x14, 0x93)
-  },
-  brightness: {
-    s1_range: [0x0F],
-    s2_range: OFFSETTING_S2_FN.call(SATURATION_BRIGHTNESS_S2_BASES, 0x54, 0xD3)
-  },
-  saturation: {
-    invert_key: true,
-    s1_range: [0x0E],
-    s2_range: OFFSETTING_S2_FN.call(SATURATION_BRIGHTNESS_S2_BASES, 0x54, 0xD3)
-  },
-  on_seq: {
-    s1_range: [0x00],
-  },
-  group: {
-    s1_range: [0x00],
-    s2_range: OFFSETTING_S2_FN.call(GROUP_BASES, 0x54, 0xD3)
-  },
-  checksum: {
-    s1_range: ->(x) do
-      lsn = (((x & 0xF)^2) + 7) % 0x10
-      
-      (0..0xF).map { |msn| (msn << 4) + lsn }
-    end,
-    s2_range: [ 0xA1, 0x53, 0x78, 0xA4 ]
-  }
-}
-
-type_overrides = {
-  checksum: 'on_seq'
-}
-
-lengths = {
-  color: 256,
-  brightness: 0x64,
-  kelvin: 0x64,
-  saturation: 0x64,
-  on_seq: 30,
-  checksum: 0xFF,
-  group: 40
-}
-
-columns = {
-  on_seq: 6,
-  group: 7,
-  checksum: 8
-}
-
-extract_fns = {
-  group: ->(b0) do
-    get_group_sequence(type: 'on', col: 7, key: b0, group_range: (0..100))
-  end
-}
-
-# PACKET_TYPES = %w(color brightness saturation kelvin on_seq checksum)
-PACKET_TYPES = %w(color brightness kelvin saturation group)
-
-(0x00..0xFF).each do |b0|
-  k = xor_key(b0)
-  results = {}
-  
-  PACKET_TYPES.each do |type|
-    type_args = extra_args[type.to_sym] || {}
-    file_type = type_overrides[type.to_sym] || type
-    
-    if extract_fn = extract_fns[type.to_sym]
-      packet_str = extract_fn.call(b0)
-    else
-      packet_str = get_sequence(file_type, b0, columns[type.to_sym] || 5)
-    end
-    
-    max_len = lengths[type.to_sym]*2
-    if packet_str.length > max_len
-      packet_str = packet_str[0...max_len]
-    end
-    
-    args = {}.merge(extra_args[type.to_sym] || {})
-    
-    
-    if args[:s1_range].is_a?(Proc)
-      args[:s1_range] = args[:s1_range].call(b0)
-    end
-    
-    if args[:s2_range].is_a?(Proc)
-      args[:s2_range] = args[:s2_range].call(b0)
-    end
-    
-    args = {
-      seq: packet_str,
-      xor_range: [k]
-    }.merge(args)
-    
-    results[type] = search_sequence(args)
-  end
-  
-  num_fns = results.values.map { |x| x[:scramble_fns].length }.max
-  printf "0x%02X", b0
-  
-  (0...num_fns).each do |i|
-    printf "\t\t"
-    
-    PACKET_TYPES.each do |type|
-      result = results[type]
-      
-      fns = result[:scramble_fns]
-      
-      if i < fns.length
-        fn = result[:scramble_fns][i]
-        printf "0x%02X\t0x%02X\t0x%02X\t%d\t", fn[:a], fn[:x], fn[:b], result[:num_misses]
-      else
-        printf "\t\t\t\t"
-      end
-    end
-      
-    printf "\n"
-  end
-end
-
-# Dir.glob("#{CAPTURES_DIR}/**.txt").each do |filename|
-#   puts File.basename(filename)
-#   capture = parse_capture(filename)
-#   
-#   pp first_byte_determines_next(capture, 3)
-#   
-#   # a = capture
-#   #   .group_by { |x| x[0] }
-#   #   .sort
-#   #   .map { |_, packets| [packets.first[0], packets.first[1]] }
-#   # mappings = Hash[a]
-#   # perms = []
-#   # current_perm = []
-#   # elem = a.first
-#   # current_perm = [elem[0]]
-#   # visited = Set.new([elem[0]])
-#   # perm_visited = Set.new([elem[0]])
-#   # 
-#   # while visited.length < mappings.length
-#   #   new_elem = nil
-#   #   
-#   #   puts visited.inspect
-#   #   
-#   #   if !mappings.include?(elem[1]) || perm_visited.include?(mappings[elem[1]])
-#   #     new_elem = (Set.new(mappings.keys) - visited).first
-#   #     perms << current_perm
-#   #     current_perm = []
-#   #     perm_visited = Set.new
-#   #   else
-#   #     new_elem = [elem[1], mappings[elem[1]]]
-#   #   end
-#   #   
-#   #   visited << new_elem[0]
-#   #   perm_visited << new_elem[0]
-#   #   elem = new_elem
-#   # end
-#   # 
-#   # perms << current_perm
-#   # 
-#   # pp perms
-#   
-#   a = capture
-#    .group_by { |x| x[0] }
-#    .sort
-#    .map { |_, packets| [packets.first[0], packets.first[1]] }
-#    .sort { |x,y| x[1] <=> y[1] }
-#   a = a
-#    .each_with_index.map do |x, i|
-#      [x[1],x[0],if a[i-1][1] == x[1]-1
-#        (x[0]-a[i-1][0])
-#      else
-#        nil
-#      end]
-#    end
-#    .to_a
-#   #  # .group_by { |x| x[1] }
-#   #  # .sort
-#   #  # .to_a
-#   
-#   pp a
-#   
-#   # a = capture
-#   #   .group_by { |x| x[0] }
-#   #   .map { |_, packets| packets.first }
-#   #   .sort #{ |x, y| x[1] <=> y[1] }
-#   #   .map { |x| x }
-#   #   .to_a
-#   #   
-# end
-
-# on1 = parse_capture("#{CAPTURES_DIR}/rgbwcct_group1_on.txt")
-# on2 = parse_capture("#{CAPTURES_DIR}/rgbwcct_group2_on.txt")
-# 
-# on1_g = on1.group_by { |x| x[0] }
-# on2_g = on2.group_by { |x| x[0] }
-# 
-# a = (on1 + on2)
-#   .map { |x| x[0] }
-#   .sort
-#   .uniq
-#   .map do |key|
-#     {key: key, vals: {on1: (on1_g[key]||[]).map { |x| x[1] }, on2: (on2_g[key]||[]).map { |x| x[1] } }}
-#   end
-#   .to_a
-# pp a

scripts/wifibox.rb

@@ -1,136 +0,0 @@
-require 'socket'
-require 'set'
-require 'net/http'
-
-STD_COMMAND_PREFIX = [ 0x31, 0x00, 0x00, 0x08 ]
-STD_COMMAND_SUFFIX = [ 0x00, 0x00, 0x00 ]
-
-class Commands
-  def self.std_command(*cmd)
-    ->() { STD_COMMAND_PREFIX + cmd + STD_COMMAND_SUFFIX }
-  end
-
-  def self.arg_command(prefix)
-    ->(val) { STD_COMMAND_PREFIX + [prefix, val] + STD_COMMAND_SUFFIX }
-  end
-
-  VALUES = [
-    LIGHT_ON = std_command(0x04, 0x01),
-    LIGHT_OFF = std_command(0x04, 0x02),
-    SATURATION = arg_command(0x02),
-    BRIGHTNESS = arg_command(0x03),
-    KELVIN = arg_command(0x05),
-    WHITE_ON = std_command(0x05, 0x64),
-    LINK = ->() { [ 0x3D, 0x00, 0x00, 0x08, 0x00, 0x00 ] + STD_COMMAND_SUFFIX },
-    UNLINK = ->() { [ 0x3D, 0x00, 0x00, 0x08, 0x00, 0x00 ] + STD_COMMAND_SUFFIX },
-    COLOR = ->(value) { STD_COMMAND_PREFIX + [0x01] + ([value]*4) }
-  ]
-end
-
-class Milight
-  ADDR = ['<broadcast>', 5987]
-  
-  attr_reader :socket
-  
-  def initialize
-    @socket = UDPSocket.new
-    socket.setsockopt(Socket::SOL_SOCKET, Socket::SO_BROADCAST, true)
-    @sequence = 0
-  end
-  
-  def hex_to_bytes(s)
-    s.strip.split(' ').map { |x| x.to_i(16) }.pack('C*')
-  end
-
-  def send(msg)
-    socket.send(msg, 0, ADDR[0], ADDR[1])
-  end
-
-  def recv
-    socket.recvfrom(1000)[0]
-  end
-
-  def start_session
-    tries = 5
-    begin
-      Timeout.timeout(5) do
-        send(hex_to_bytes("20 00 00 00 16 02 62 3A D5 ED A3 01 AE 08 2D 46 61 41 A7 F6 DC AF D3 E6 00 00 1E"))
-        msg = recv
-        @session = msg[-3..-2].bytes
-      end
-    rescue Exception => e
-      puts "Error: #{e}"
-      retry if (tries -= 1) > 0
-    end
-  end
-
-  def send_command(cmd, zone = 0)
-    start_session if !@session
-    
-    msg = [0x80, 0, 0, 0, 0x11, @session[0], @session[1], 0, @sequence, 0]
-    msg += cmd
-    msg += [zone,0]
-    msg += [msg[-11..-1].reduce(&:+)&0xFF]
-    
-    send(msg.pack('C*'))
-    
-    @sequence = (@sequence + 1) % 0xFF
-    
-    #recv
-  end
-end
-
-def get_file(cmd, value, group)
-  name = "../../packet_captures/sidoh_wifibox1/rgbcct_group#{group}_#{cmd}#{value.nil? ? "" : "_#{value}"}.txt"
-  File.expand_path(File.join(__FILE__, name))
-end
-
-def get_packet
-  Net::HTTP.get('10.133.8.167', '/gateway_traffic/rgb_cct').split("\n").last.strip
-end
-
-milight = Milight.new
-
-(1..4).each do |group|
-  (0..0xFF).each do |value|
-    seen_keys = Set.new
-    last_val = 0
-    
-    file = get_file("color", value, group)
-    
-    if File.exists?(file)
-      File.read(file).split("\n").each { |x| seen_keys << x.split(' ').first }
-    end
-    
-    puts "Processing: #{file}"
-    
-    File.open(file, 'a') do |f|
-      while seen_keys.size < 255
-        t = Thread.new do 
-          packet = get_packet
-          key = packet.split(' ')[0]
-          seen_keys << key
-          f.write "#{packet}\n"
-          f.flush
-        end
-        
-        while %w(sleep run).include?(t.status)
-          milight.send_command(Commands::COLOR.call(value), group)
-          sleep 0.1
-          print "."
-        end
-        
-        print '*'
-        
-        puts "\n#{file} - #{seen_keys.length}" if last_val < seen_keys.length
-        last_val = seen_keys.length
-      end
-    end
-  end
-end
-
-# 10000.times do
-#   milight.send_command(, 1).inspect
-#   print "."
-#   sleep 0.1
-# end