From: Florian Fainelli <florian@openwrt.org>
Date: Wed, 17 Dec 2008 13:50:04 +0000 (+0000)
Subject: Also move jungo-image.py to scripts/flashing
X-Git-Url: http://git.lede-project.org./?a=commitdiff_plain;h=d8df4c38a8316719f6bc9baa6b132efb50115e70;p=openwrt%2Fstaging%2Frobimarko.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 <jvasco@verizon.net>
+#
+# 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] <IP_address> [redboot.bin]
+Valid options:
+\t-h | --help: usage statement
+\t-d | --no-dump: don't create a flash dump
+\t-f | --file: use <filename> 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 <jvasco@verizon.net>
-#
-# 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] <IP_address> [redboot.bin]
-Valid options:
-\t-h | --help: usage statement
-\t-d | --no-dump: don't create a flash dump
-\t-f | --file: use <filename> 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()