blob: fb6bb4d1afae4a2b124bc019f7db5870b9426a0e [file] [log] [blame]
William A. Kennington III32e440d2021-12-03 18:19:07 -08001#!/bin/bash
2# Copyright 2021 Google LLC
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8# http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15
16[ -z "${gbmc_upgrade-}" ] || exit
17
18: "${GBMC_UPGRADE_SIG=/tmp/bmc.sig}"
19
20gbmc_upgrade_hook() {
21 [ -n "${bootfile_url-}" ] || return
22
23 local tmpdir
24 tmpdir="$(mktemp -d)" || return
25 gbmc_upgrade_internal || true
26 # SC doesn't know our variable is defined elsewhere
27 # shellcheck disable=SC2153
28 rm -rf -- "$tmpdir" "$GBMC_UPGRADE_SIG" "$GBMC_UPGRADE_IMG"
29}
30
31gbmc_upgrade_fetch() (
32 echo "Fetching $bootfile_url" >&2
33
William A. Kennington IIIfea3baf2022-01-26 15:09:11 -080034 # We only support tarballs at the moment, our URLs will always denote
35 # this with a URI query param of `format=TAR`.
36 if ! [[ "$bootfile_url" =~ [\&?]format=TAR(&|$) ]]; then
William A. Kennington III32e440d2021-12-03 18:19:07 -080037 echo "Unknown upgrade unpack method: $bootfile_url" >&2
38 return 1
39 fi
40
41 # Ensure some sane output file limit
42 # Currently no BMC image is larger than 64M
43 ulimit -H -f $((96 * 1024 * 1024)) || return
William A. Kennington IIIfea3baf2022-01-26 15:09:11 -080044 wget -q -O - "$bootfile_url" | tar -xC "$tmpdir" || true
William A. Kennington III32e440d2021-12-03 18:19:07 -080045
46 local sig
47 sig="$(find "$tmpdir" -name 'image-*.sig')" || return
48 local img="${sig%.sig}"
49 mv "$sig" "$GBMC_UPGRADE_SIG" || return
50 mv "$img" "$GBMC_UPGRADE_IMG" || return
51
52 # Regular packages have a VERSION file with the image
53 local imgdir="${sig%/*}"
54 if [ -f "$imgdir/VERSION" ]; then
55 cat "$imgdir/VERSION" || return
56 fi
57
58 # Staging packages have a directory named after the version
59 local vdir="${imgdir##*/}"
60 if [[ "$vdir" =~ ([0-9]+[.]){3}[0-9]+ ]]; then
61 echo "$vdir"
62 fi
63)
64
65GBMC_BR_DHCP_HOOKS+=(gbmc_upgrade_hook)
66
67gbmc_upgrade=1