@@ -62,7 +62,7 @@ struct tracecmd_output {
bool quiet;
unsigned long file_state;
unsigned long file_version;
- size_t options_start;
+ tsize_t options_start;
bool big_endian;
struct list_head options;
@@ -85,6 +85,8 @@ struct list_event_system {
#define HAS_SECTIONS(H) ((H)->file_version >= FILE_VERSION_SECTIONS)
+static int write_options_v7(struct tracecmd_output *handle);
+
static stsize_t
do_write_check(struct tracecmd_output *handle, const void *data, tsize_t size)
{
@@ -194,6 +196,10 @@ void tracecmd_output_close(struct tracecmd_output *handle)
if (!handle)
return;
+ /* write any unsaved options at the end of trace files with sections */
+ if (handle->file_version >= FILE_VERSION_SECTIONS)
+ write_options_v7(handle);
+
if (handle->fd >= 0) {
close(handle->fd);
handle->fd = -1;
@@ -1224,6 +1230,7 @@ int tracecmd_output_set_version(struct tracecmd_output *handle, int file_version
*/
static int output_write_init(struct tracecmd_output *handle)
{
+ unsigned long long offset;
char buf[BUFSIZ];
int endian4;
@@ -1257,6 +1264,14 @@ static int output_write_init(struct tracecmd_output *handle)
endian4 = convert_endian_4(handle, handle->page_size);
if (do_write_check(handle, &endian4, 4))
return -1;
+ if (HAS_SECTIONS(handle)) {
+ /* Write 0 as options offset and save its location */
+ offset = 0;
+ handle->options_start = do_lseek(handle, 0, SEEK_CUR);
+ if (do_write_check(handle, &offset, 8))
+ return -1;
+ }
+
handle->file_state = TRACECMD_FILE_INIT;
return 0;
}
@@ -1325,7 +1340,7 @@ tracecmd_add_option_v(struct tracecmd_output *handle,
* We can only add options before tracing data were written.
* This may change in the future.
*/
- if (handle->file_state > TRACECMD_FILE_OPTIONS)
+ if (!HAS_SECTIONS(handle) && handle->file_state > TRACECMD_FILE_OPTIONS)
return NULL;
for (i = 0; i < count; i++)
@@ -1338,8 +1353,7 @@ tracecmd_add_option_v(struct tracecmd_output *handle,
return NULL;
}
}
-
- option = malloc(sizeof(*option));
+ option = calloc(1, sizeof(*option));
if (!option) {
tracecmd_warning("Could not allocate space for option");
free(data);
@@ -1402,7 +1416,7 @@ int tracecmd_write_cpus(struct tracecmd_output *handle, int cpus)
return 0;
}
-int tracecmd_write_options(struct tracecmd_output *handle)
+static int write_options_v6(struct tracecmd_output *handle)
{
struct tracecmd_option *options;
unsigned short option;
@@ -1420,7 +1434,7 @@ int tracecmd_write_options(struct tracecmd_output *handle)
if (do_write_check(handle, "options ", 10))
return -1;
-
+ handle->options_start = do_lseek(handle, 0, SEEK_CUR);
list_for_each_entry(options, &handle->options, list) {
endian2 = convert_endian_2(handle, options->id);
if (do_write_check(handle, &endian2, 2))
@@ -1444,11 +1458,83 @@ int tracecmd_write_options(struct tracecmd_output *handle)
return -1;
handle->file_state = TRACECMD_FILE_OPTIONS;
+ return 0;
+}
+
+static int write_options_v7(struct tracecmd_output *handle)
+{
+ struct tracecmd_option *options;
+ unsigned long long endian8;
+ unsigned short endian2;
+ unsigned int endian4;
+ bool new = false;
+ tsize_t offset;
+
+ /* Check if there are unsaved options */
+ list_for_each_entry(options, &handle->options, list) {
+ if (!options->offset) {
+ new = true;
+ break;
+ }
+ }
+
+ if (!new)
+ return 0;
+ offset = do_lseek(handle, 0, SEEK_CUR);
+ /* Append to the previous options section, if any */
+ if (handle->options_start) {
+ if (do_lseek(handle, handle->options_start, SEEK_SET) == (off64_t)-1)
+ return -1;
+ endian8 = convert_endian_8(handle, offset);
+ if (do_write_check(handle, &endian8, 8))
+ return -1;
+ if (do_lseek(handle, offset, SEEK_SET) == (off_t)-1)
+ return -1;
+ }
+ offset = out_write_section_header(handle, TRACECMD_OPTION_DONE, "options", 0, false);
+ if (offset == (off_t)-1)
+ return -1;
+ list_for_each_entry(options, &handle->options, list) {
+ /* Option is already saved, skip it */
+ if (options->offset)
+ continue;
+ endian2 = convert_endian_2(handle, options->id);
+ if (do_write_check(handle, &endian2, 2))
+ return -1;
+ endian4 = convert_endian_4(handle, options->size);
+ if (do_write_check(handle, &endian4, 4))
+ return -1;
+ /* Save the data location */
+ options->offset = do_lseek(handle, 0, SEEK_CUR);
+ if (do_write_check(handle, options->data, options->size))
+ return -1;
+ }
+
+ endian2 = convert_endian_2(handle, TRACECMD_OPTION_DONE);
+ if (do_write_check(handle, &endian2, 2))
+ return -1;
+ endian4 = convert_endian_4(handle, 8);
+ if (do_write_check(handle, &endian4, 4))
+ return -1;
+ endian8 = 0;
+ handle->options_start = do_lseek(handle, 0, SEEK_CUR);
+ if (do_write_check(handle, &endian8, 8))
+ return -1;
+ if (out_update_section_header(handle, offset))
+ return -1;
return 0;
}
-int tracecmd_append_options(struct tracecmd_output *handle)
+int tracecmd_write_options(struct tracecmd_output *handle)
+{
+ if (!HAS_SECTIONS(handle))
+ return write_options_v6(handle);
+ return write_options_v7(handle);
+}
+
+
+static int append_options_v6(struct tracecmd_output *handle)
{
struct tracecmd_option *options;
unsigned short option;
@@ -1499,6 +1585,13 @@ int tracecmd_append_options(struct tracecmd_output *handle)
return 0;
}
+int tracecmd_append_options(struct tracecmd_output *handle)
+{
+ if (!HAS_SECTIONS(handle))
+ return append_options_v6(handle);
+ return write_options_v7(handle);
+}
+
static struct tracecmd_option *
add_buffer_option(struct tracecmd_output *handle, const char *name, int cpus)
{
Introduced chain of options sections in trace file version 7. Extended the "DONE" option to hold the offset into the file to the next options section. Format of trace file version 7 is extended with a new mandatory field after the compression algorithm header: <8 bytes>, unsigned long long integer - offset into the trace file where the first options section is located. This allows to place this section anywhere in the file. As all other sections have corresponding options, describing their offsets into the trace file, this change makes the structure of trace file version 7 flexible. Signed-off-by: Tzvetomir Stoyanov (VMware) <tz.stoyanov@gmail.com> --- lib/trace-cmd/trace-output.c | 107 ++++++++++++++++++++++++++++++++--- 1 file changed, 100 insertions(+), 7 deletions(-)