@@ -63,7 +63,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;
bool do_compress;
struct tracecmd_compression *compress;
@@ -1449,6 +1449,7 @@ int tracecmd_output_set_compression(struct tracecmd_output *handler, const char
*/
int tracecmd_output_write_init(struct tracecmd_output *handler)
{
+ unsigned long long offset;
char buf[BUFSIZ];
int endian4;
@@ -1485,6 +1486,11 @@ int tracecmd_output_write_init(struct tracecmd_output *handler)
if (handler->file_version >= 7) {
if (write_compression_header(handler))
return -1;
+ /* Write 0 as options offset and save its location */
+ offset = 0;
+ handler->options_start = do_lseek(handler, 0, SEEK_CUR);
+ if (do_write_check(handler, &offset, 8))
+ return -1;
}
handler->file_state = TRACECMD_FILE_INIT;
@@ -1559,7 +1565,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 (handle->file_version < 7 && handle->file_state > TRACECMD_FILE_OPTIONS)
return NULL;
for (i = 0; i < count; i++)
@@ -1572,8 +1578,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);
@@ -1636,7 +1641,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;
@@ -1654,7 +1659,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))
@@ -1665,7 +1670,7 @@ int tracecmd_write_options(struct tracecmd_output *handle)
return -1;
/* Save the data location in case it needs to be updated */
- options->offset = lseek64(handle->fd, 0, SEEK_CUR);
+ options->offset = do_lseek(handle, 0, SEEK_CUR);
if (do_write_check(handle, options->data,
options->size))
@@ -1678,11 +1683,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 (handle->file_version < 7)
+ 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;
@@ -1697,14 +1774,12 @@ int tracecmd_append_options(struct tracecmd_output *handle)
*/
if (handle->file_state != TRACECMD_FILE_OPTIONS)
return -1;
-
- if (lseek64(handle->fd, 0, SEEK_END) == (off_t)-1)
+ if (do_lseek(handle, 0, SEEK_END) == -1)
return -1;
- offset = lseek64(handle->fd, -2, SEEK_CUR);
+ offset = do_lseek(handle, -2, SEEK_CUR);
if (offset == (off_t)-1)
return -1;
-
- r = pread(handle->fd, &option, 2, offset);
+ r = do_preed(handle, &option, 2, offset);
if (r != 2 || option != TRACECMD_OPTION_DONE)
return -1;
@@ -1718,7 +1793,7 @@ int tracecmd_append_options(struct tracecmd_output *handle)
return -1;
/* Save the data location in case it needs to be updated */
- options->offset = lseek64(handle->fd, 0, SEEK_CUR);
+ options->offset = do_lseek(handle, 0, SEEK_CUR);
if (do_write_check(handle, options->data,
options->size))
@@ -1733,6 +1808,13 @@ int tracecmd_append_options(struct tracecmd_output *handle)
return 0;
}
+int tracecmd_append_options(struct tracecmd_output *handle)
+{
+ if (handle->file_version < 7)
+ 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 | 110 ++++++++++++++++++++++++++++++----- 1 file changed, 96 insertions(+), 14 deletions(-)