From: Florian Fainelli Date: Wed, 17 Dec 2008 13:50:04 +0000 (+0000) Subject: Also move jungo-image.py to scripts/flashing X-Git-Tag: reboot~24965 X-Git-Url: http://git.lede-project.org./?a=commitdiff_plain;h=d8df4c38a8316719f6bc9baa6b132efb50115e70;p=openwrt%2Fstaging%2Fxback.git Also move jungo-image.py to scripts/flashing SVN-Revision: 13675 --- diff --git a/scripts/flashing/jungo-image.py b/scripts/flashing/jungo-image.py new file mode 100644 index 0000000000..0630b4a77e --- /dev/null +++ b/scripts/flashing/jungo-image.py @@ -0,0 +1,258 @@ +#!/usr/bin/env python +# +# Copyright 2008 (C) Jose Vasconcellos +# +# A script that can communicate with jungo-based routers +# (such as MI424-WR, USR8200 and WRV54G) to backup the installed +# firmware and replace the boot loader. +# +# Tested with Python 2.5 on Linux and Windows +# +"""Usage: %s [options] [redboot.bin] +Valid options: +\t-h | --help: usage statement +\t-d | --no-dump: don't create a flash dump +\t-f | --file: use to store dump contents +\t-u | --user: provide username (default admin) +\t-p | --pass: provide password (default password1) +\t-P | --proto: set transfer protocol (default http) +\t --port: set port for http (default 8080) +\t-s | --server: IP address of tftp server +\t-w | --write: initiate loading of redboot (default no modification to flash) +\t-q | --quiet: don't display unnecessary information +\t-v | --verbose: display progress information +\t-V | --version: display version information +""" + +import os +import sys +import getopt +import getpass +import telnetlib +import string +import binascii +import socket +import thread +import SocketServer +import SimpleHTTPServer + +server = "" +HOST = "192.168.1.1" +PORT = 8080 +user = "admin" +#password = getpass.getpass() +password = "password1" +proto = "http" +imagefile = "redboot.bin" +dumpfile = "" +verbose = 1 +no_dump = 0 +dumplen = 0x10000 +write_image = 0 +flashsize=4*1024*1024 +#device="br0" +device="ixp0" + +#################### + +def start_server(): + httpd = SocketServer.TCPServer((server,PORT),SimpleHTTPServer.SimpleHTTPRequestHandler) + thread.start_new_thread(httpd.serve_forever,()) + +#################### + +def get_flash_size(): + global flashsize + tn.write("cat /proc/mtd\n") + # wait for prompt + buf = tn.read_until("Returned 0", 3) + if buf: + i = buf.find('mtd0:') + if i > 0: + flashsize = int(buf[i+6:].split()[0],16) + else: + print "Can't find mtd0!" + else: + print "Can't access /proc/mtd!" + +def image_dump(tn, dumpfile): + if not dumpfile: + tn.write("ver\n"); + buf = tn.read_until("Returned 0") + i = buf.find("Platform:") + if i < 0: + platform="jungo" + else: + line=buf[i+9:] + i=line.find('\n') + platform=line[:i].split()[-1] + + tn.write("ifconfig -v %s\n" % device); + buf = tn.read_until("Returned 0") + + i = buf.find("mac = 0") + if i > 0: + i += 6 + else: + print "No MAC address found! (use -f option)" + sys.exit(1) + dumpfile = "%s-%s.bin" % (platform, buf[i:i+17].replace(':','')) + else: + tn.write("\n") + + print "Dumping flash contents (%dMB) to %s" % (flashsize/1048576, dumpfile) + f = open(dumpfile, "wb") + + t=flashsize/dumplen + for addr in range(t): + if verbose: + sys.stdout.write('\r%d%%'%(100*addr/t)) + sys.stdout.flush() + + tn.write("flash_dump -r 0x%x -l %d -4\n" % (addr*dumplen, dumplen)) + tn.read_until("\n") + + count = addr*dumplen + while 1: + buf = tn.read_until("\n") + if buf.strip() == "Returned 0": + break + s = buf.split() + if s and s[0][-1] == ':': + a=int(s[0][:-1],16) + if a != count: + print "Format error: %x != %x"%(a,count) + sys.exit(2) + count += 16 + f.write(binascii.a2b_hex(string.join(s[1:],''))) + tn.read_until(">",1) + + f.close() + if verbose: + print "" + +def telnet_option(sock,cmd,option): + #print "Option: %d %d" % (ord(cmd), ord(option)) + if cmd == telnetlib.DO: + c=telnetlib.WILL + elif cmd == telnetlib.WILL: + c=telnetlib.DO + sock.sendall(telnetlib.IAC + c + option) + +def telnet_timeout(): + print "Fatal error: telnet timeout!" + sys.exit(1) + +def usage(): + print __doc__ % os.path.basename(sys.argv[0]) + +#################### + +try: + opts, args = getopt.getopt(sys.argv[1:], "hdf:u:qp:P:s:vVw", \ + ["help", "dump", "file=", "user=", "pass=", "proto=", "proto=", + "quiet=", "server=", "verbose", "version", "write"]) +except getopt.GetoptError: + # print help information and exit: + usage() + sys.exit(1) + +for o, a in opts: + if o in ("-h", "--help"): + usage() + sys.exit(1) + if o in ("-V", "--version"): + print "%s: 0.7" % sys.argv[0] + sys.exit(1) + if o in ("-d", "--no-dump"): + no_dump = 1 + if o in ("-f", "--file"): + dumpfile = a + if o in ("-s", "--server"): + server = a + if o in ("-u", "--user"): + user = a + if o in ("-p", "--pass"): + password = a + if o in ("-P", "--proto"): + proto = a + if o in ("--port"): + PORT = a + if o in ("-w", "--write"): + write_image = 1 + if o in ("-q", "--quiet"): + verbose = 0 + if o in ("-v", "--verbose"): + verbose = 1 + +# make sure we have enough arguments +if len(args) > 0: + HOST = args[0] + +if len(args) == 2: + imagefile = args[1] + +#################### +# create a telnet session to the router +try: + tn = telnetlib.Telnet(HOST) +except socket.error, msg: + print "Unable to establish telnet session to %s: %s" % (HOST, msg) + sys.exit(1) + +tn.set_option_negotiation_callback(telnet_option) + +buf = tn.read_until("Username: ", 3) +if not buf: + telnet_timeout() +tn.write(user+"\n") +if password: + buf = tn.read_until("Password: ", 3) + if not buf: + telnet_timeout() + tn.write(password+"\n") + +# wait for prompt +buf = tn.read_until("> ", 3) +if not buf: + telnet_timeout() + +get_flash_size() + +if not no_dump: + image_dump(tn, dumpfile) + +if write_image: + if not os.access(imagefile, os.R_OK): + print "File access error: %s" % (imagefile) + sys.exit(3) + + splitpath = os.path.split(imagefile) + # make sure we're in the directory where the image is located + if splitpath[0]: + os.chdir(splitpath[0]) + + # write image file image + if not server: + server = tn.get_socket().getsockname()[0] + if proto == "http": + cmd = "load -u %s://%s:%d/%s -r 0\n" % (proto, server, PORT, splitpath[1]) + else: + cmd = "load -u %s://%s/%s -r 0\n" % (proto, server, splitpath[1]) + + if proto == "http": + start_server() + + if verbose: + print "Unlocking flash..." + tn.write("unlock 0 0x%x\n" % flashsize) + buf = tn.read_until("Returned 0") + + if verbose: + print "Writing new image..." + print cmd, + tn.write(cmd) + buf = tn.read_until("Returned 0") + +tn.write("exit\n") +tn.close() diff --git a/scripts/jungo-image.py b/scripts/jungo-image.py deleted file mode 100644 index 0630b4a77e..0000000000 --- a/scripts/jungo-image.py +++ /dev/null @@ -1,258 +0,0 @@ -#!/usr/bin/env python -# -# Copyright 2008 (C) Jose Vasconcellos -# -# A script that can communicate with jungo-based routers -# (such as MI424-WR, USR8200 and WRV54G) to backup the installed -# firmware and replace the boot loader. -# -# Tested with Python 2.5 on Linux and Windows -# -"""Usage: %s [options] [redboot.bin] -Valid options: -\t-h | --help: usage statement -\t-d | --no-dump: don't create a flash dump -\t-f | --file: use to store dump contents -\t-u | --user: provide username (default admin) -\t-p | --pass: provide password (default password1) -\t-P | --proto: set transfer protocol (default http) -\t --port: set port for http (default 8080) -\t-s | --server: IP address of tftp server -\t-w | --write: initiate loading of redboot (default no modification to flash) -\t-q | --quiet: don't display unnecessary information -\t-v | --verbose: display progress information -\t-V | --version: display version information -""" - -import os -import sys -import getopt -import getpass -import telnetlib -import string -import binascii -import socket -import thread -import SocketServer -import SimpleHTTPServer - -server = "" -HOST = "192.168.1.1" -PORT = 8080 -user = "admin" -#password = getpass.getpass() -password = "password1" -proto = "http" -imagefile = "redboot.bin" -dumpfile = "" -verbose = 1 -no_dump = 0 -dumplen = 0x10000 -write_image = 0 -flashsize=4*1024*1024 -#device="br0" -device="ixp0" - -#################### - -def start_server(): - httpd = SocketServer.TCPServer((server,PORT),SimpleHTTPServer.SimpleHTTPRequestHandler) - thread.start_new_thread(httpd.serve_forever,()) - -#################### - -def get_flash_size(): - global flashsize - tn.write("cat /proc/mtd\n") - # wait for prompt - buf = tn.read_until("Returned 0", 3) - if buf: - i = buf.find('mtd0:') - if i > 0: - flashsize = int(buf[i+6:].split()[0],16) - else: - print "Can't find mtd0!" - else: - print "Can't access /proc/mtd!" - -def image_dump(tn, dumpfile): - if not dumpfile: - tn.write("ver\n"); - buf = tn.read_until("Returned 0") - i = buf.find("Platform:") - if i < 0: - platform="jungo" - else: - line=buf[i+9:] - i=line.find('\n') - platform=line[:i].split()[-1] - - tn.write("ifconfig -v %s\n" % device); - buf = tn.read_until("Returned 0") - - i = buf.find("mac = 0") - if i > 0: - i += 6 - else: - print "No MAC address found! (use -f option)" - sys.exit(1) - dumpfile = "%s-%s.bin" % (platform, buf[i:i+17].replace(':','')) - else: - tn.write("\n") - - print "Dumping flash contents (%dMB) to %s" % (flashsize/1048576, dumpfile) - f = open(dumpfile, "wb") - - t=flashsize/dumplen - for addr in range(t): - if verbose: - sys.stdout.write('\r%d%%'%(100*addr/t)) - sys.stdout.flush() - - tn.write("flash_dump -r 0x%x -l %d -4\n" % (addr*dumplen, dumplen)) - tn.read_until("\n") - - count = addr*dumplen - while 1: - buf = tn.read_until("\n") - if buf.strip() == "Returned 0": - break - s = buf.split() - if s and s[0][-1] == ':': - a=int(s[0][:-1],16) - if a != count: - print "Format error: %x != %x"%(a,count) - sys.exit(2) - count += 16 - f.write(binascii.a2b_hex(string.join(s[1:],''))) - tn.read_until(">",1) - - f.close() - if verbose: - print "" - -def telnet_option(sock,cmd,option): - #print "Option: %d %d" % (ord(cmd), ord(option)) - if cmd == telnetlib.DO: - c=telnetlib.WILL - elif cmd == telnetlib.WILL: - c=telnetlib.DO - sock.sendall(telnetlib.IAC + c + option) - -def telnet_timeout(): - print "Fatal error: telnet timeout!" - sys.exit(1) - -def usage(): - print __doc__ % os.path.basename(sys.argv[0]) - -#################### - -try: - opts, args = getopt.getopt(sys.argv[1:], "hdf:u:qp:P:s:vVw", \ - ["help", "dump", "file=", "user=", "pass=", "proto=", "proto=", - "quiet=", "server=", "verbose", "version", "write"]) -except getopt.GetoptError: - # print help information and exit: - usage() - sys.exit(1) - -for o, a in opts: - if o in ("-h", "--help"): - usage() - sys.exit(1) - if o in ("-V", "--version"): - print "%s: 0.7" % sys.argv[0] - sys.exit(1) - if o in ("-d", "--no-dump"): - no_dump = 1 - if o in ("-f", "--file"): - dumpfile = a - if o in ("-s", "--server"): - server = a - if o in ("-u", "--user"): - user = a - if o in ("-p", "--pass"): - password = a - if o in ("-P", "--proto"): - proto = a - if o in ("--port"): - PORT = a - if o in ("-w", "--write"): - write_image = 1 - if o in ("-q", "--quiet"): - verbose = 0 - if o in ("-v", "--verbose"): - verbose = 1 - -# make sure we have enough arguments -if len(args) > 0: - HOST = args[0] - -if len(args) == 2: - imagefile = args[1] - -#################### -# create a telnet session to the router -try: - tn = telnetlib.Telnet(HOST) -except socket.error, msg: - print "Unable to establish telnet session to %s: %s" % (HOST, msg) - sys.exit(1) - -tn.set_option_negotiation_callback(telnet_option) - -buf = tn.read_until("Username: ", 3) -if not buf: - telnet_timeout() -tn.write(user+"\n") -if password: - buf = tn.read_until("Password: ", 3) - if not buf: - telnet_timeout() - tn.write(password+"\n") - -# wait for prompt -buf = tn.read_until("> ", 3) -if not buf: - telnet_timeout() - -get_flash_size() - -if not no_dump: - image_dump(tn, dumpfile) - -if write_image: - if not os.access(imagefile, os.R_OK): - print "File access error: %s" % (imagefile) - sys.exit(3) - - splitpath = os.path.split(imagefile) - # make sure we're in the directory where the image is located - if splitpath[0]: - os.chdir(splitpath[0]) - - # write image file image - if not server: - server = tn.get_socket().getsockname()[0] - if proto == "http": - cmd = "load -u %s://%s:%d/%s -r 0\n" % (proto, server, PORT, splitpath[1]) - else: - cmd = "load -u %s://%s/%s -r 0\n" % (proto, server, splitpath[1]) - - if proto == "http": - start_server() - - if verbose: - print "Unlocking flash..." - tn.write("unlock 0 0x%x\n" % flashsize) - buf = tn.read_until("Returned 0") - - if verbose: - print "Writing new image..." - print cmd, - tn.write(cmd) - buf = tn.read_until("Returned 0") - -tn.write("exit\n") -tn.close()