oleorhagen
5 years ago
committed by
GitHub
GitHub
8 changed files with 331 additions and 82 deletions
@ -0,0 +1,47 @@ |
|||||
|
# |
||||
|
# Copyright 2020 Northern.tech AS |
||||
|
# |
||||
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
# you may not use this file except in compliance with the License. |
||||
|
# You may obtain a copy of the License at |
||||
|
# |
||||
|
# http://www.apache.org/licenses/LICENSE-2.0 |
||||
|
# |
||||
|
# Unless required by applicable law or agreed to in writing, software |
||||
|
# distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
# See the License for the specific language governing permissions and |
||||
|
# limitations under the License. |
||||
|
|
||||
|
function parse_cli_options () { |
||||
|
while (( "$#" )); do |
||||
|
case "$1" in |
||||
|
-o | --overlay) |
||||
|
overlays+=("${2}") |
||||
|
shift 2 |
||||
|
;; |
||||
|
-c | --config) |
||||
|
configs+=("${2}") |
||||
|
shift 2 |
||||
|
;; |
||||
|
-d | --disk-image) |
||||
|
disk_image="${2}" |
||||
|
shift 2 |
||||
|
;; |
||||
|
*) |
||||
|
log_fatal "Sorry but the provided option is not supported: $1" |
||||
|
;; |
||||
|
esac |
||||
|
done |
||||
|
|
||||
|
if [ -z "${disk_image}" ]; then |
||||
|
log_warn "Sorry, but '--disk-image' is a mandatory option" |
||||
|
log_warn "See ./mender-convert --help for more information" |
||||
|
exit 1 |
||||
|
fi |
||||
|
|
||||
|
if [ ! -e ${disk_image} ]; then |
||||
|
log_fatal "File not found: ${disk_image}" |
||||
|
fi |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,88 @@ |
|||||
|
# |
||||
|
# Copyright 2020 Northern.tech AS |
||||
|
# |
||||
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
# you may not use this file except in compliance with the License. |
||||
|
# You may obtain a copy of the License at |
||||
|
# |
||||
|
# http://www.apache.org/licenses/LICENSE-2.0 |
||||
|
# |
||||
|
# Unless required by applicable law or agreed to in writing, software |
||||
|
# distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
# See the License for the specific language governing permissions and |
||||
|
# limitations under the License. |
||||
|
|
||||
|
source modules/zip.sh |
||||
|
source modules/log.sh |
||||
|
|
||||
|
# compression_type |
||||
|
# |
||||
|
# $1 - Path to the compressed disk image |
||||
|
# |
||||
|
# @return - The MENDER_COMPRESS_IMAGE compression type |
||||
|
# |
||||
|
function compression_type () { |
||||
|
if [[ $# -ne 1 ]]; then |
||||
|
log_fatal "compression_type() requires one argument" |
||||
|
fi |
||||
|
local -r disk_image="${1}" |
||||
|
case "${disk_image}" in |
||||
|
*.img | *.sdimg) |
||||
|
echo "none" |
||||
|
;; |
||||
|
*.gz) |
||||
|
echo "gzip" |
||||
|
;; |
||||
|
*.zip) |
||||
|
echo "zip" |
||||
|
;; |
||||
|
*.xz ) |
||||
|
echo "lzma" |
||||
|
;; |
||||
|
* ) |
||||
|
log_fatal "Unsupported compression type: ${disk_image}. Please uncompress the image yourself." |
||||
|
;; |
||||
|
esac |
||||
|
} |
||||
|
|
||||
|
# Decompresses the given input image |
||||
|
# |
||||
|
# $1 - Path to the compressed image |
||||
|
# $2 - Path to the output directory |
||||
|
# |
||||
|
# @return - Name of the uncompressed image |
||||
|
# |
||||
|
function decompress_image () { |
||||
|
if [[ $# -ne 2 ]]; then |
||||
|
log_fatal "decompress_image() requires an image argument and an output directory" |
||||
|
fi |
||||
|
local -r input_image="${1}" |
||||
|
local -r output_dir="${2}" |
||||
|
local disk_image="${output_dir}/$(basename ${input_image})" |
||||
|
case "$(compression_type ${disk_image})" in |
||||
|
none ) |
||||
|
: |
||||
|
;; |
||||
|
gzip ) |
||||
|
log_info "Decompressing ${disk_image}..." |
||||
|
disk_image=${disk_image%.gz} |
||||
|
zcat "${input_image}" > "${disk_image}" |
||||
|
;; |
||||
|
zip ) |
||||
|
log_info "Decompressing ${disk_image}..." |
||||
|
filename="$(zip_get_imgname ${input_image})" |
||||
|
unzip "${input_image}" -d "${output_dir}" &>/dev/null |
||||
|
disk_image="$(dirname ${disk_image})/${filename}" |
||||
|
;; |
||||
|
lzma ) |
||||
|
log_info "Decompressing ${disk_image}..." |
||||
|
disk_image=${disk_image%.xz} |
||||
|
xzcat "${input_image}" > "${disk_image}" |
||||
|
;; |
||||
|
* ) |
||||
|
log_fatal "Unsupported input image type: ${input_image}" |
||||
|
;; |
||||
|
esac |
||||
|
echo "${disk_image}" |
||||
|
} |
||||
@ -0,0 +1,34 @@ |
|||||
|
# |
||||
|
# Copyright 2020 Northern.tech AS |
||||
|
# |
||||
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
# you may not use this file except in compliance with the License. |
||||
|
# You may obtain a copy of the License at |
||||
|
# |
||||
|
# http://www.apache.org/licenses/LICENSE-2.0 |
||||
|
# |
||||
|
# Unless required by applicable law or agreed to in writing, software |
||||
|
# distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
# See the License for the specific language governing permissions and |
||||
|
# limitations under the License. |
||||
|
|
||||
|
# |
||||
|
# Parse the filename from the zipped output |
||||
|
# |
||||
|
# $1 - Zip archive path |
||||
|
# |
||||
|
# @return - Name of the img contained in the archive |
||||
|
# |
||||
|
function zip_get_imgname () { |
||||
|
if [[ $# -ne 1 ]]; then |
||||
|
log_fatal "zip_get_imgname requires one argument" |
||||
|
fi |
||||
|
local -r disk_image="${1}" |
||||
|
# Assert that the archive holds only one file |
||||
|
nfiles="$(unzip -l ${disk_image} | awk '{nfiles=$2} END {print nfiles}')" |
||||
|
[[ "$nfiles" -ne 1 ]] && log_fatal "Zip archive has more than one file. Needs to be unzipped by a human. nfiles: $nfiles" |
||||
|
local -r filename="$(unzip -lq ${disk_image} | awk 'NR==3 {filename=$NF} END {print filename}')" |
||||
|
[[ ${filename} == *.img ]] || log_fatal "no img file found in the zip archive ${disk_image}." |
||||
|
echo "$(basename ${filename})" |
||||
|
} |
||||
Loading…
Reference in new issue