flash.rb

$: << File.realpath('..', __FILE__)

require 'device'
require 'backend-driver'

if ARGV[1] == '--version'
  $stdout.puts "McHCK Programmer (Input Club edition)"
  exit
end

$stderr.puts "Attaching debugger..."
adiv5 = Adiv5.new(BackendDriver.from_string(ARGV[0]))
k = nil

if ARGV[1] == '--detect'
  k = Device.detect(adiv5, true) # Reset may not work properly for mass erase if chip already flashed
  $stdout.puts k.desc
  exit
elsif ARGV[1] == '--detect2file'
  k = Device.detect(adiv5, true) # Reset may not work properly for mass erase if chip already flashed
  $stdout.puts k.desc
  File.open('/tmp/detectlog.log', 'w') { |file| file.write(k.detect_info) }
  exit
elsif ARGV[1] == '--mass-erase'
  k = Device.detect(adiv5, true) # Reset may not work properly for mass erase if chip already flashed
  $stderr.puts "done."
  $stderr.puts "Mass erasing chip..."
  k.mass_erase
  $stderr.puts "done."
else
  k = Device.detect(adiv5)
  $stderr.puts "done."

  firmware = File.read(ARGV[1], :mode => 'rb')
  address = Integer(ARGV[2])

  $stderr.puts "Programming %d bytes of firmware to address %#x..." % [firmware.bytesize, address]
  k.halt_core!
  k.program(address, firmware) do |address, i, total|
    $stderr.puts "programming %#x, %d of %d" % [address, i, total]
  end
  $stderr.puts "done."
end

$stderr.puts "resetting..."
k.reset_system!
k.disable_debug!
k.continue!
$stderr.puts "done."