tests/regression/ust/ust-dl/test_ust-dl.py

   1 #!/usr/bin/env python3
   2 #
   3 # Copyright (C) - 2013 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   4 # Copyright (C) - 2015 Antoine Busque <abusque@efficios.com>
   5 #
   6 # This program is free software; you can redistribute it and/or modify it
   7 # under the terms of the GNU General Public License, version 2 only, as
   8 # published by the Free Software Foundation.
   9 #
  10 # This program is distributed in the hope that it will be useful, but WITHOUT
  11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12 # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13 # more details.
  14 #
  15 # You should have received a copy of the GNU General Public License along with
  16 # this program; if not, write to the Free Software Foundation, Inc., 51
  17 # Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18
  19 import os
  20 import subprocess
  21 import re
  22 import shutil
  23 import sys
  24
  25 test_path = os.path.dirname(os.path.abspath(__file__)) + "/"
  26 test_utils_path = test_path
  27 for i in range(4):
  28     test_utils_path = os.path.dirname(test_utils_path)
  29 test_utils_path = test_utils_path + "/utils"
  30 sys.path.append(test_utils_path)
  31 from test_utils import *
  32
  33
  34 NR_TESTS = 6
  35 current_test = 1
  36 print("1..{0}".format(NR_TESTS))
  37
  38 # Check if a sessiond is running... bail out if none found.
  39 if session_daemon_alive() == 0:
  40     bail("""No sessiond running. Please make sure you are running this test
  41     with the "run" shell script and verify that the lttng tools are
  42     properly installed.""")
  43
  44 session_info = create_session()
  45 enable_ust_tracepoint_event(session_info, "*")
  46 start_session(session_info)
  47
  48 test_env = os.environ.copy()
  49 test_env["LD_PRELOAD"] += ":liblttng-ust-dl.so"
  50 test_env["LD_LIBRARY_PATH"] += ":" + test_path
  51 test_process = subprocess.Popen(test_path + "prog",
  52                                 stdout=subprocess.PIPE, stderr=subprocess.PIPE,
  53                                 env=test_env)
  54
  55 if sys.version_info >= (3, 3):
  56     try:
  57         test_process.wait(5)
  58     except subprocess.TimeoutExpired:
  59         test_process.kill()
  60         bail("Failed to run ust-dl test application.", session_info)
  61 else:
  62     test_process.wait()
  63
  64 print_test_result(test_process.returncode == 0, current_test, "Test application exited normally")
  65 current_test += 1
  66
  67 stop_session(session_info)
  68
  69 # Check for dl events in the resulting trace
  70 try:
  71     babeltrace_process = subprocess.Popen(["babeltrace", session_info.trace_path], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  72 except FileNotFoundError:
  73     bail("Could not open babeltrace. Please make sure it is installed.", session_info)
  74
  75 dlopen_event_found = False
  76 build_id_event_found = False
  77 debug_link_event_found = False
  78 dlclose_event_found = False
  79
  80 for event_line in babeltrace_process.stdout:
  81     # Let babeltrace finish to get the return code
  82     if dlopen_event_found and build_id_event_found and \
  83        debug_link_event_found and dlclose_event_found:
  84         continue
  85
  86     event_line = event_line.decode('utf-8').replace("\n", "")
  87     if re.search(r".*lttng_ust_dl:dlopen.*", event_line) is not None:
  88         dlopen_event_found = True
  89     elif re.search(r".*lttng_ust_dl:build_id.*", event_line) is not None:
  90         build_id_event_found = True
  91     elif re.search(r".*lttng_ust_dl:debug_link.*", event_line) is not None:
  92         debug_link_event_found = True
  93     elif re.search(r".*lttng_ust_dl:dlclose.*", event_line) is not None:
  94         dlclose_event_found = True
  95
  96 babeltrace_process.wait()
  97
  98 print_test_result(babeltrace_process.returncode == 0, current_test, "Resulting trace is readable")
  99 current_test += 1
 100
 101 print_test_result(dlopen_event_found, current_test, "lttng_ust_dl:dlopen event found in resulting trace")
 102 current_test += 1
 103
 104 print_test_result(build_id_event_found, current_test, "lttng_ust_dl:build_id event found in resulting trace")
 105 current_test += 1
 106
 107 print_test_result(debug_link_event_found, current_test, "lttng_ust_dl:debug_link event found in resulting trace")
 108 current_test += 1
 109
 110 print_test_result(dlclose_event_found, current_test, "lttng_ust_dl:dlclose event found in resulting trace")
 111 current_test += 1
 112
 113 shutil.rmtree(session_info.tmp_directory)