espressif/esp-idf
1
0
Fork 1
mirror of https://github.com/espressif/esp-idf.git synced 2025-08-05 05:34:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

docs/ota: Allow example_test.py to run a custom https-server

Browse Source
This commit is contained in:
David Cermak
2021-03-02 14:58:11 +01:00
parent 138ab8a3af
commit 79d596bcdc
1 changed files with 26 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
examples/system/ota/simple_ota_example/example_test.py
View File

@@ -3,6 +3,7 @@ import os
import re import re
import socket import socket
import ssl import ssl
import sys
from threading import Thread from threading import Thread
import ttfw_idf import ttfw_idf
@@ -68,18 +69,16 @@ def get_my_ip():
return my_ip return my_ip
def start_https_server(ota_image_dir, server_ip, server_port): def start_https_server(ota_image_dir, server_ip, server_port, server_file=None, key_file=None):
# parser = argparse.ArgumentParser()
# parser.add_argument('-p', '--port', dest='port', type= int,
# help= "Server Port", default= 8000)
# args = parser.parse_args()
os.chdir(ota_image_dir) os.chdir(ota_image_dir)
if server_file is None:
server_file = os.path.join(ota_image_dir, 'server_cert.pem') server_file = os.path.join(ota_image_dir, 'server_cert.pem')
cert_file_handle = open(server_file, 'w+') cert_file_handle = open(server_file, 'w+')
cert_file_handle.write(server_cert) cert_file_handle.write(server_cert)
cert_file_handle.close() cert_file_handle.close()
if key_file is None:
key_file = os.path.join(ota_image_dir, 'server_key.pem') key_file = os.path.join(ota_image_dir, 'server_key.pem')
key_file_handle = open('server_key.pem', 'w+') key_file_handle = open('server_key.pem', 'w+')
key_file_handle.write(server_key) key_file_handle.write(server_key)
@@ -201,6 +200,17 @@ def test_examples_protocol_simple_ota_example_with_flash_encryption(env, extra_d
if __name__ == '__main__': if __name__ == '__main__':
if sys.argv[2:]: # if two or more arguments provided:
# Usage: example_test.py <image_dir> <server_port> [cert_di>]
this_dir = os.path.dirname(os.path.realpath(__file__))
bin_dir = os.path.join(this_dir, sys.argv[1])
port = int(sys.argv[2])
cert_dir = bin_dir if not sys.argv[3:] else os.path.join(this_dir, sys.argv[3]) # optional argument
print('Starting HTTPS server at "https://:{}"'.format(port))
start_https_server(bin_dir, '', port,
server_file=os.path.join(cert_dir, 'ca_cert.pem'),
key_file=os.path.join(cert_dir, 'ca_key.pem'))
else:
test_examples_protocol_simple_ota_example() test_examples_protocol_simple_ota_example()
test_examples_protocol_simple_ota_example_ethernet_with_spiram_config() test_examples_protocol_simple_ota_example_ethernet_with_spiram_config()
test_examples_protocol_simple_ota_example_with_flash_encryption() test_examples_protocol_simple_ota_example_with_flash_encryption()