forked from flipperdevices/flipperzero-firmware
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'dev' into release-candidate
- Loading branch information
Showing
11 changed files
with
229 additions
and
17 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| #!/usr/bin/env python3 | ||
|
|
||
| from flipper.storage import FlipperStorage | ||
|
|
||
| import logging | ||
| import argparse | ||
| import os | ||
| import sys | ||
| import pathlib | ||
| import serial.tools.list_ports as list_ports | ||
|
|
||
|
|
||
| class Main: | ||
| def __init__(self): | ||
| # command args | ||
| self.parser = argparse.ArgumentParser() | ||
| self.parser.add_argument("-d", "--debug", action="store_true", help="Debug") | ||
| self.parser.add_argument("-p", "--port", help="CDC Port", default="auto") | ||
| self.parser.add_argument( | ||
| "-b", | ||
| "--baud", | ||
| help="Port Baud rate", | ||
| required=False, | ||
| default=115200 * 4, | ||
| type=int, | ||
| ) | ||
|
|
||
| self.subparsers = self.parser.add_subparsers(help="sub-command help") | ||
|
|
||
| self.parser_install = self.subparsers.add_parser( | ||
| "install", help="Install OTA package" | ||
| ) | ||
| self.parser_install.add_argument("manifest_path", help="Manifest path") | ||
| self.parser_install.add_argument( | ||
| "--pkg_dir_name", help="Update dir name", default="pcbundle", required=False | ||
| ) | ||
| self.parser_install.set_defaults(func=self.install) | ||
|
|
||
| # logging | ||
| self.logger = logging.getLogger() | ||
|
|
||
| def __call__(self): | ||
| self.args = self.parser.parse_args() | ||
| if "func" not in self.args: | ||
| self.parser.error("Choose something to do") | ||
| # configure log output | ||
| self.log_level = logging.DEBUG if self.args.debug else logging.INFO | ||
| self.logger.setLevel(self.log_level) | ||
| self.handler = logging.StreamHandler(sys.stdout) | ||
| self.handler.setLevel(self.log_level) | ||
| self.formatter = logging.Formatter("%(asctime)s [%(levelname)s] %(message)s") | ||
| self.handler.setFormatter(self.formatter) | ||
| self.logger.addHandler(self.handler) | ||
| # execute requested function | ||
| self.args.func() | ||
|
|
||
| # make directory with exist check | ||
| def mkdir_on_storage(self, storage, flipper_dir_path): | ||
| if not storage.exist_dir(flipper_dir_path): | ||
| self.logger.debug(f'"{flipper_dir_path}" does not exist, creating') | ||
| if not storage.mkdir(flipper_dir_path): | ||
| self.logger.error(f"Error: {storage.last_error}") | ||
| return False | ||
| else: | ||
| self.logger.debug(f'"{flipper_dir_path}" already exists') | ||
| return True | ||
|
|
||
| # send file with exist check and hash check | ||
| def send_file_to_storage(self, storage, flipper_file_path, local_file_path, force): | ||
| exists = storage.exist_file(flipper_file_path) | ||
| do_upload = not exists | ||
| if exists: | ||
| hash_local = storage.hash_local(local_file_path) | ||
| hash_flipper = storage.hash_flipper(flipper_file_path) | ||
| self.logger.debug(f"hash check: local {hash_local}, flipper {hash_flipper}") | ||
| do_upload = force or (hash_local != hash_flipper) | ||
|
|
||
| if do_upload: | ||
| self.logger.info(f'Sending "{local_file_path}" to "{flipper_file_path}"') | ||
| if not storage.send_file(local_file_path, flipper_file_path): | ||
| self.logger.error(f"Error: {storage.last_error}") | ||
| return False | ||
| return True | ||
|
|
||
| def _get_port(self): | ||
| if self.args.port != "auto": | ||
| return self.args.port | ||
| # Try guessing | ||
| flippers = list(list_ports.grep("flip")) | ||
| if len(flippers) == 1: | ||
| flipper = flippers[0] | ||
| self.logger.info(f"Using {flipper.serial_number} on {flipper.device}") | ||
| return flipper.device | ||
| elif len(flippers) == 0: | ||
| self.logger.error("Failed to find connected Flipper") | ||
| elif len(flippers) > 1: | ||
| self.logger.error("More than one Flipper is attached") | ||
| self.logger.error("Failed to guess which port to use. Specify --port") | ||
|
|
||
| def install(self): | ||
| if not (port := self._get_port()): | ||
| return 1 | ||
|
|
||
| storage = FlipperStorage(port, self.args.baud) | ||
| storage.start() | ||
|
|
||
| if not os.path.isfile(self.args.manifest_path): | ||
| self.logger.error("Error: manifest not found") | ||
| return 2 | ||
|
|
||
| manifest_path = pathlib.Path(os.path.abspath(self.args.manifest_path)) | ||
| manifest_name, pkg_name = manifest_path.parts[-1], manifest_path.parts[-2] | ||
|
|
||
| pkg_dir_name = self.args.pkg_dir_name or pkg_name | ||
| flipper_update_path = f"/ext/update/{pkg_dir_name}" | ||
|
|
||
| self.logger.info(f'Installing "{pkg_name}" from {flipper_update_path}') | ||
| # if not os.path.exists(self.args.manifest_path): | ||
| # self.logger.error("Error: package not found") | ||
| if not self.mkdir_on_storage(storage, flipper_update_path): | ||
| self.logger.error(f"Error: cannot create {storage.last_error}") | ||
| return -2 | ||
|
|
||
| for dirpath, dirnames, filenames in os.walk(manifest_path.parents[0]): | ||
| for fname in filenames: | ||
| self.logger.debug(f"Uploading {fname}") | ||
| local_file_path = os.path.join(dirpath, fname) | ||
| flipper_file_path = f"{flipper_update_path}/{fname}" | ||
| if not self.send_file_to_storage( | ||
| storage, flipper_file_path, local_file_path, False | ||
| ): | ||
| self.logger.error(f"Error: {storage.last_error}") | ||
| return -3 | ||
|
|
||
| storage.send_and_wait_eol( | ||
| f"update install {flipper_update_path}/{manifest_name}\r" | ||
| ) | ||
| break | ||
| storage.stop() | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| Main()() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.