1#!/bin/bash
2# Copyright (C) 2016 The Android Open Source Project
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# Include some functions from common.sh.
17SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
18source ${SCRIPT_DIR}/common.sh
19
20OUT_DATA_DIR="test_out"
21
22DEBUG=false
23
24
25apply_overlay() {
26  local overaly="$1"
27  local base_dts="$2"
28  local overlay_dts="$3"
29  local merged_dts="$4"
30
31  ./apply_overlay.sh "--$overaly" "$base_dts" "$overlay_dts" "$merged_dts"
32}
33
34run_stress_test() {
35  # see ufdt_gen_test_dts.c for detail
36  local node_depth="$1"
37  local node_unused="$2"
38  local node_count="$3"
39  local append_count="$4"
40  local override_count="$5"
41
42  mkdir -p "$OUT_DATA_DIR" >& /dev/null
43
44  #
45  # Prepare dtb and dtbo files
46  #
47  local base_dts="${OUT_DATA_DIR}/base.dts"
48  echo "  Base DT: depth=$node_depth unused=$node_unused nodes=$node_count"
49  ufdt_gen_test_dts -n $node_count -d $node_depth -u $node_unused \
50    -o "$base_dts"
51  if $DEBUG; then
52    cat "$base_dts"
53  fi
54
55  local overlay_dts="${OUT_DATA_DIR}/overlay.dts"
56  echo "  Overlay DT: append=$append_count override=$override_count"
57  ufdt_gen_test_dts -p -a $append_count -w $override_count \
58    -o "$overlay_dts"
59  if $DEBUG; then
60    cat "$overlay_dts"
61  fi
62
63  local merged_dts="${OUT_DATA_DIR}/overlay_merged.dts"
64  apply_overlay ufdt $base_dts $overlay_dts $merged_dts
65
66  rm -rf "$OUT_DATA_DIR"
67}
68
69main() {
70  alert "========== Running Stress Tests =========="
71
72  if [ -z "$ANDROID_BUILD_TOP" ]; then
73    die "Run envsetup.sh / lunch yet?"
74  fi
75
76  if ! command_exists ufdt_gen_test_dts ||
77     ! command_exists dtc; then
78    die "Run mmma $(dirname $SCRIPT_DIR) yet?"
79  fi
80
81  (
82
83  # cd to ${SCRIPT_DIR} in a subshell because gen_test.sh uses relative
84  # paths for dependent files.
85  cd "${SCRIPT_DIR}"
86
87  # node_depth      = 2
88  # node_unused     = 0
89  # node_count      = 5000
90  # append_count    = 2500
91  # override_count  = 2500
92  run_stress_test 2 0 5000 2500 2500
93
94  )
95
96  if [ $? -ne 0 ]; then
97    die "Some test cases failed, please check error message..."
98  fi
99}
100
101main "$@"
102