diff --git a/tools/espota.py b/tools/espota.py index 9913f0045c..d3b1254929 100755 --- a/tools/espota.py +++ b/tools/espota.py @@ -5,11 +5,12 @@ # # Modified since 2015-09-18 from Pascal Gollor (https://github.com/pgollor) # Modified since 2015-11-09 from Hristo Gochkov (https://github.com/me-no-dev) +# Modified since 2016-01-03 from Matthew O'Gorman (https://githumb.com/mogorman) # # This script will push an OTA update to the ESP -# use it like: python espota.py -i -p [-a password] -f +# use it like: python espota.py -i -I -p -P [-a password] -f # Or to upload SPIFFS image: -# python espota.py -i -p [-a password] -s -f +# python espota.py -i -I -p -P [-a password] -s -f # # Changes # 2015-09-18: @@ -22,6 +23,10 @@ # - Added digest authentication # - Enchanced error tracking and reporting # +# Changes +# 2016-01-03: +# - Added more options to parser. +# from __future__ import print_function import socket @@ -64,11 +69,10 @@ def update_progress(progress): sys.stderr.write('.') sys.stderr.flush() -def serve(remoteAddr, remotePort, password, filename, command = FLASH): +def serve(remoteAddr, localAddr, remotePort, localPort, password, filename, command = FLASH): # Create a TCP/IP socket sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - serverPort = random.randint(10000,60000) - server_address = ('0.0.0.0', serverPort) + server_address = (localAddr, localPort) logging.info('Starting on %s:%s', str(server_address[0]), str(server_address[1])) try: sock.bind(server_address) @@ -82,7 +86,7 @@ def serve(remoteAddr, remotePort, password, filename, command = FLASH): file_md5 = hashlib.md5(f.read()).hexdigest() f.close() logging.info('Upload size: %d', content_size) - message = '%d %d %d %s\n' % (command, serverPort, content_size, file_md5) + message = '%d %d %d %s\n' % (command, localPort, content_size, file_md5) # Wait for a connection logging.info('Sending invitation to: %s', remoteAddr) @@ -209,12 +213,24 @@ def parser(): help = "ESP8266 IP Address.", default = False ) + group.add_option("-I", "--host_ip", + dest = "host_ip", + action = "store", + help = "Host IP Address.", + default = "0.0.0.0" + ) group.add_option("-p", "--port", dest = "esp_port", type = "int", help = "ESP8266 ota Port. Default 8266", default = 8266 ) + group.add_option("-P", "--host_port", + dest = "host_port", + type = "int", + help = "Host server ota Port. Default random 10000-60000", + default = random.randint(10000,60000) + ) parser.add_option_group(group) # auth @@ -294,7 +310,7 @@ def main(args): command = SPIFFS # end if - return serve(options.esp_ip, options.esp_port, options.auth, options.image, command) + return serve(options.esp_ip, options.host_ip, options.esp_port, options.host_port, options.auth, options.image, command) # end main