2 * Copyright (C) 2015 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License, version 2 only, as
6 * published by the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 * You should have received a copy of the GNU General Public License along with
14 * this program; if not, write to the Free Software Foundation, Inc., 51
15 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 #include <common/common.h>
21 #include <common/utils.h>
22 #include <common/defaults.h>
24 #include "tracefile-array.h"
26 struct tracefile_array
*tracefile_array_create(size_t count
)
28 struct tracefile_array
*tfa
= NULL
;
31 tfa
= zmalloc(sizeof(*tfa
));
35 tfa
->tf
= zmalloc(sizeof(*tfa
->tf
) * count
);
40 for (i
= 0; i
< count
; i
++) {
41 tfa
->tf
[i
].seq_head
= -1ULL;
42 tfa
->tf
[i
].seq_tail
= -1ULL;
44 tfa
->seq_head
= -1ULL;
45 tfa
->seq_tail
= -1ULL;
56 void tracefile_array_destroy(struct tracefile_array
*tfa
)
65 void tracefile_array_file_rotate(struct tracefile_array
*tfa
,
66 enum tracefile_rotate_type type
)
68 uint64_t *headp
, *tailp
;
71 /* Not in tracefile rotation mode. */
75 case TRACEFILE_ROTATE_READ
:
77 * Rotate read head to write head position, thus allowing
78 * reader to consume the newly rotated head file.
80 tfa
->file_head_read
= tfa
->file_head_write
;
82 case TRACEFILE_ROTATE_WRITE
:
83 /* Rotate write head to next file, pushing tail if needed. */
84 tfa
->file_head_write
= (tfa
->file_head_write
+ 1) % tfa
->count
;
85 if (tfa
->file_head_write
== tfa
->file_tail
) {
87 tfa
->file_tail
= (tfa
->file_tail
+ 1) % tfa
->count
;
89 headp
= &tfa
->tf
[tfa
->file_head_write
].seq_head
;
90 tailp
= &tfa
->tf
[tfa
->file_head_write
].seq_tail
;
92 * If we overwrite a file with content, we need to push the tail
93 * to the position following the content we are overwriting.
95 if (*headp
!= -1ULL) {
96 tfa
->seq_tail
= tfa
->tf
[tfa
->file_tail
].seq_tail
;
98 /* Reset this file head/tail (overwrite). */
107 void tracefile_array_commit_seq(struct tracefile_array
*tfa
)
109 uint64_t *headp
, *tailp
;
111 /* Increment overall head. */
113 /* If we are committing our first index overall, set tail to 0. */
114 if (tfa
->seq_tail
== -1ULL) {
118 /* Not in tracefile rotation mode. */
121 headp
= &tfa
->tf
[tfa
->file_head_write
].seq_head
;
122 tailp
= &tfa
->tf
[tfa
->file_head_write
].seq_tail
;
123 /* Update head tracefile seq_head. */
124 *headp
= tfa
->seq_head
;
126 * If we are committing our first index in this packet, set tail
127 * to this index seq count.
129 if (*tailp
== -1ULL) {
130 *tailp
= tfa
->seq_head
;
134 uint64_t tracefile_array_get_read_file_index_head(struct tracefile_array
*tfa
)
136 return tfa
->file_head_read
;
139 uint64_t tracefile_array_get_seq_head(struct tracefile_array
*tfa
)
141 return tfa
->seq_head
;
144 uint64_t tracefile_array_get_file_index_tail(struct tracefile_array
*tfa
)
146 return tfa
->file_tail
;
149 uint64_t tracefile_array_get_seq_tail(struct tracefile_array
*tfa
)
151 return tfa
->seq_tail
;
154 bool tracefile_array_seq_in_file(struct tracefile_array
*tfa
,
155 uint64_t file_index
, uint64_t seq
)
159 * Not in tracefile rotation mode; we are guaranteed to have the
160 * index in this file.
164 assert(file_index
< tfa
->count
);
168 if (seq
>= tfa
->tf
[file_index
].seq_tail
169 && seq
<= tfa
->tf
[file_index
].seq_head
) {