tests/regression/tools/metadata/utils.sh

   1 #!/bin/bash
   2 #
   3 # Copyright (C) 2019 Jonathan Rajotte <jonathan.rajotte-julien@efficios.com>
   4 #
   5 # SPDX-License-Identifier: LGPL-2.1-only
   6
   7 function get_env_value_ctf1 ()
   8 {
   9         local env_file=$1
  10         local key=$2
  11         local result
  12         local ret
  13
  14         result=$(grep "$key" < "$env_file")
  15         ret=$?
  16         if [ $ret -eq 1 ]; then
  17                 echo "invalid_value_extraction"
  18                 return 1
  19         else
  20                 # Strip the key using bash substring removal.
  21                 # This remove all leading chars until the actual value.
  22                 result=${result#* = }
  23
  24                 # Remove the trailing ';'
  25                 result=${result:0:-1}
  26
  27                 # Remove enclosing '"' if present
  28                 if [ "${result:0:1}" == '"' ]; then
  29                         result=${result:1:-1}
  30                 fi
  31
  32                 echo "$result"
  33                 return 0
  34         fi
  35 }
  36
  37 # Expects a json representing the environment of a trace_class fragment.
  38 function get_env_value_ctf2 ()
  39 {
  40         local env_file=$1
  41         local key=$2
  42         local result
  43         local ret
  44
  45         if [[ "$key" == "vpid =" ]]; then
  46                 # The "vpid =" key value is only because the ctf1 extractor uses
  47                 # grep. jq use the complete key.
  48                 key="vpid"
  49         fi
  50
  51
  52         result=$(jq -r ".${key}" < "$env_file")
  53         ret=$?
  54         if [ $ret -eq 1 ]; then
  55                 echo "invalid_value_extraction"
  56                 return 1
  57         else
  58                 echo "$result"
  59                 return 0
  60         fi
  61 }
  62
  63 function get_env_value ()
  64 {
  65         if [ "$TRACE_FORMAT_OUTPUT" == "ctf1" ]; then
  66                 get_env_value_ctf1 "$@"
  67         elif [ "$TRACE_FORMAT_OUTPUT" == "ctf2" ]; then
  68                 get_env_value_ctf2 "$@"
  69         else
  70                 return 1
  71         fi
  72 }
  73
  74 function extract_env_to_file_ctf1 ()
  75 {
  76         local source_dir=$1
  77         local destination=$2
  78         local env
  79
  80         env=$($BABELTRACE_BIN --output-format=ctf-metadata "${source_dir}")
  81
  82         # Extract "env" scope
  83         awk '/env {/,/};/' <<< "$env" > "$destination"
  84 }
  85
  86 function extract_env_to_file_ctf2 ()
  87 {
  88         local source_file="$1/metadata"
  89         local destination=$2
  90         local query="select(.type == \"trace-class\").environment"
  91         jq  "$query" "$source_file" > "$destination"
  92 }
  93
  94 function extract_env_to_file ()
  95 {
  96         if [ "$TRACE_FORMAT_OUTPUT" == "ctf1" ]; then
  97                 extract_env_to_file_ctf1 "$@"
  98                 return $?
  99         elif [ "$TRACE_FORMAT_OUTPUT" == "ctf2" ]; then
 100                 extract_env_to_file_ctf2 "$@"
 101                 return $?
 102         else
 103                 return 1
 104         fi
 105 }
 106
 107 function iso8601_to_lttng_dir_datetime ()
 108 {
 109         local result=$1
 110
 111         result=${result/T/-}
 112
 113         # Remove trailing timezone information including the '-'.
 114         result=${result:0:-5}
 115
 116         echo "$result"
 117         return 0
 118 }