More error checking for trace folder (Fix #306) (#307)

* More error checking for trace folder * don't be stupid with the variable... --------- Co-authored-by: Thomas Applencourt <[email protected]>
argonne-lcf · Jan 8, 2025 · d8324e6 · d8324e6
1 parent 09613ea
commit d8324e6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/utils/babeltrace_thapi.in b/utils/babeltrace_thapi.in
@@ -260,7 +260,13 @@ THAPI_METADATA_FILE = 'thapi_metadata.yaml'
 
 def thapi_metadata(trace)
   @thapi_metadata ||= {}
-  @thapi_metadata[trace] ||= YAML.load_file(File.join(trace, THAPI_METADATA_FILE))
+  full_path =  File.join(trace, THAPI_METADATA_FILE)
+  unless File.file?(full_path)
+    STDERR.puts("Error: #{trace} isn't a valid trace (missing #{THAPI_METADATA_FILE})")
+    exit(1)
+  end
+
+  @thapi_metadata[trace] ||= YAML.load_file(full_path)
 end
 
 def modify_metadata(command, trace)