10 #include <linux/compiler.h>
12 #define min(a,b) (((a)<(b))?a:b)
13 #define max(a,b) (((a)>(b))?a:b)
14 #define BUG_ON(a) assert(!(a))
16 // Useful outsize __KERNEL__. Not used here because inline is already redefined.
17 #define force_inline inline __attribute__((always_inline))
19 /* Calculate the offset needed to align the type */
20 static inline unsigned int ltt_align(size_t align_drift
,
23 size_t alignment
= min(sizeof(void*), size_of_type
);
25 return ((alignment
- align_drift
) & (alignment
-1));
31 enum lttng_tasklet_priority
{
41 struct lttng_mystruct2
{
43 enum lttng_irq_mode mode
;
44 //struct lttng_mystruct teststr1;
48 static inline size_t lttng_get_size_mystruct2(
49 struct lttng_mystruct2
* obj
)
51 size_t size
=0, locsize
;
53 locsize
= sizeof(unsigned int);
54 size
+= ltt_align(size
, locsize
) + locsize
;
56 locsize
= sizeof(enum lttng_irq_mode
);
57 size
+= ltt_align(size
, locsize
) + locsize
;
59 BUG_ON(sizeof(struct lttng_mystruct2
) != size
);
61 return sizeof(struct lttng_mystruct2
);
64 static inline size_t lttng_get_alignment_mystruct2(
65 struct lttng_mystruct2
*obj
)
67 size_t align
=0, localign
;
69 localign
= sizeof(unsigned int);
70 align
= max(align
, localign
);
72 localign
= sizeof(enum lttng_irq_mode
);
73 align
= max(align
, localign
);
78 static inline void lttng_write_mystruct2(
83 struct lttng_mystruct2
*obj
)
87 align
= lttng_get_alignment_mystruct2(obj
);
88 size
= lttng_get_size_mystruct2(obj
);
91 *to
+= ltt_align(*to
, align
); /* align output */
93 *len
+= ltt_align(*to
+*len
, align
); /* C alignment, ok to do a memcpy of it */
101 #define LTTNG_ARRAY_SIZE_mystruct_myarray 10
102 typedef uint64_t lttng_array_mystruct_myarray
[LTTNG_ARRAY_SIZE_mystruct_myarray
];
104 static inline size_t lttng_get_size_array_mystruct_myarray(
105 lttng_array_mystruct_myarray obj
)
107 size_t size
=0, locsize
;
109 locsize
= sizeof(uint64_t);
110 /* ltt_align == 0 always*/
111 //size += ltt_align(size, locsize) + (LTTNG_ARRAY_SIZE_mystruct_myarray * locsize);
112 BUG_ON(ltt_align(size
, locsize
) != 0);
113 size
+= LTTNG_ARRAY_SIZE_mystruct_myarray
* locsize
;
115 BUG_ON(size
!= LTTNG_ARRAY_SIZE_mystruct_myarray
* sizeof(uint64_t));
120 static inline size_t lttng_get_alignment_array_mystruct_myarray(
121 lttng_array_mystruct_myarray obj
)
123 size_t align
=0, localign
;
125 localign
= sizeof(uint64_t);
126 align
= max(align
, localign
);
132 static inline void lttng_write_array_mystruct_myarray(
137 lttng_array_mystruct_myarray obj
)
141 align
= lttng_get_alignment_array_mystruct_myarray(obj
);
142 size
= lttng_get_size_array_mystruct_myarray(obj
);
145 *to
+= ltt_align(*to
, align
); /* align output */
147 *len
+= ltt_align(*to
+*len
, align
); /* C alignment, ok to do a memcpy of it */
154 typedef struct lttng_sequence_mystruct_mysequence lttng_sequence_mystruct_mysequence
;
155 struct lttng_sequence_mystruct_mysequence
{
161 static inline size_t lttng_get_size_sequence_mystruct_mysequence(
162 lttng_sequence_mystruct_mysequence
*obj
)
164 size_t size
=0, locsize
;
166 locsize
= sizeof(unsigned int);
167 size
+= ltt_align(size
, locsize
) + locsize
;
169 locsize
= sizeof(double);
170 size
+= ltt_align(size
, locsize
) + (obj
->len
* locsize
);
175 static inline size_t lttng_get_alignment_sequence_mystruct_mysequence(
176 lttng_sequence_mystruct_mysequence
*obj
)
178 size_t align
=0, localign
;
180 localign
= sizeof(unsigned int);
181 align
= max(align
, localign
);
183 localign
= sizeof(double);
184 align
= max(align
, localign
);
190 static inline void lttng_write_sequence_mystruct_mysequence(
195 lttng_sequence_mystruct_mysequence
*obj
)
201 /* Flush pending memcpy */
203 memcpy(*to_base
+*to
, *from
, *len
);
208 align
= lttng_get_alignment_sequence_mystruct_mysequence(obj
);
209 //no need size = lttng_get_size_sequence_mystruct_mysequence(obj);
212 *to
+= ltt_align(*to
, align
); /* *len = 0 in this function */
215 *to
+= ltt_align(*to
, sizeof(unsigned int));
217 varlen
+= sizeof(unsigned int);
218 memcpy(*to_base
+*to
, varfrom
, varlen
);
222 *to
+= ltt_align(*to
, sizeof(double));
223 varfrom
= obj
->array
;
224 varlen
+= obj
->len
* sizeof(double);
225 memcpy(*to_base
+*to
, varfrom
, varlen
);
229 /* Realign the *to_base on arch size, set *to to 0 */
230 *to
= ltt_align(*to
, sizeof(void *));
231 *to_base
= *to_base
+*to
;
234 /* Put source *from just after the C sequence */
240 union lttng_mystruct_myunion
{
242 unsigned long myulong
;
246 static inline size_t lttng_get_size_mystruct_myunion(
247 union lttng_mystruct_myunion
*obj
)
249 size_t size
=0, locsize
;
251 locsize
= sizeof(double);
252 size
= max(size
, locsize
);
254 locsize
= sizeof(unsigned long);
255 size
= max(size
, locsize
);
257 BUG_ON(size
!= sizeof(union lttng_mystruct_myunion
));
263 static inline size_t lttng_get_alignment_mystruct_myunion(
264 union lttng_mystruct_myunion
*obj
)
266 size_t align
=0, localign
;
268 localign
= sizeof(double);
269 align
= max(align
, localign
);
271 localign
= sizeof(unsigned long);
272 align
= max(align
, localign
);
278 static inline void lttng_write_mystruct_myunion(
283 union lttng_mystruct_myunion
*obj
)
287 align
= lttng_get_alignment_mystruct_myunion(obj
);
288 size
= lttng_get_size_mystruct_myunion(obj
);
291 *to
+= ltt_align(*to
, align
); /* align output */
293 *len
+= ltt_align(*to
+*len
, align
); /* C alignment, ok to do a memcpy of it */
300 struct lttng_mystruct
{
302 enum lttng_irq_mode mode
;
303 struct lttng_mystruct2 teststr
;
304 lttng_array_mystruct_myarray myarray
;
305 lttng_sequence_mystruct_mysequence mysequence
;
306 union lttng_mystruct_myunion myunion
;
309 static inline size_t lttng_get_size_mystruct(
310 struct lttng_mystruct
*obj
)
312 size_t size
=0, locsize
, localign
;
314 locsize
= sizeof(unsigned int);
315 size
+= ltt_align(size
, locsize
) + locsize
;
317 locsize
= sizeof(enum lttng_irq_mode
);
318 size
+= ltt_align(size
, locsize
) + locsize
;
320 localign
= lttng_get_alignment_mystruct2(&obj
->teststr
);
321 locsize
= lttng_get_size_mystruct2(&obj
->teststr
);
322 size
+= ltt_align(size
, localign
) + locsize
;
324 localign
= lttng_get_alignment_array_mystruct_myarray(obj
->myarray
);
325 locsize
= lttng_get_size_array_mystruct_myarray(obj
->myarray
);
326 size
+= ltt_align(size
, localign
) + locsize
;
328 localign
= lttng_get_alignment_sequence_mystruct_mysequence(&obj
->mysequence
);
329 locsize
= lttng_get_size_sequence_mystruct_mysequence(&obj
->mysequence
);
330 size
+= ltt_align(size
, localign
) + locsize
;
332 localign
= lttng_get_alignment_mystruct_myunion(&obj
->myunion
);
333 locsize
= lttng_get_size_mystruct_myunion(&obj
->myunion
);
334 size
+= ltt_align(size
, localign
) + locsize
;
340 static inline size_t lttng_get_alignment_mystruct(
341 struct lttng_mystruct
*obj
)
343 size_t align
=0, localign
;
345 localign
= sizeof(unsigned int);
346 align
= max(align
, localign
);
348 localign
= sizeof(enum lttng_irq_mode
);
349 align
= max(align
, localign
);
351 localign
= lttng_get_alignment_mystruct2(&obj
->teststr
);
352 align
= max(align
, localign
);
354 localign
= lttng_get_alignment_array_mystruct_myarray(obj
->myarray
);
355 align
= max(align
, localign
);
357 localign
= lttng_get_alignment_sequence_mystruct_mysequence(&obj
->mysequence
);
358 align
= max(align
, localign
);
360 localign
= lttng_get_alignment_mystruct_myunion(&obj
->myunion
);
361 align
= max(align
, localign
);
366 static inline void lttng_write_mystruct(
371 struct lttng_mystruct
*obj
)
375 align
= lttng_get_alignment_mystruct(obj
);
376 // no need : contains variable size fields.
377 // locsize = lttng_get_size_mystruct(obj);
380 *to
+= ltt_align(*to
, align
); /* align output */
382 *len
+= ltt_align(*to
+*len
, align
); /* C alignment, ok to do a memcpy of it */
385 /* Contains variable sized fields : must explode the structure */
387 size
= sizeof(unsigned int);
388 size
+= ltt_align(*to
+*len
, size
) + size
;
391 size
= sizeof(enum lttng_irq_mode
);
392 size
+= ltt_align(*to
+*len
, size
) + size
;
395 lttng_write_mystruct2(to_base
, to
, from
, len
, &obj
->teststr
);
397 lttng_write_array_mystruct_myarray(to_base
, to
, from
, len
, obj
->myarray
);
399 /* Variable length field */
400 lttng_write_sequence_mystruct_mysequence(to_base
, to
, from
, len
, &obj
->mysequence
);
401 *to
= 0; /* Force the compiler to know it's 0 */
402 /* After this previous write, we are sure that *to is 0, and *to_base is
403 * aligned on the architecture size : to rest of alignment will be calculated
406 lttng_write_mystruct_myunion(to_base
, to
, from
, len
, &obj
->myunion
);
408 /* Don't forget to flush last write..... */
419 struct lttng_mystruct test
;
420 test
.mysequence
.len
= 20;
421 test
.mysequence
.array
= malloc(20);
423 size_t size
= lttng_get_size_mystruct(&test
);
424 size_t align
= lttng_get_alignment_mystruct(&test
);
426 void *buf
= malloc(align
+ size
);
427 void *to_base
= buf
; /* the buffer is allocated on arch_size alignment */
432 lttng_write_mystruct(&to_base
, &to
, &from
, &len
, &test
);
434 /* Flush pending memcpy */
436 memcpy(to_base
+to
, from
, len
);
442 free(test
.mysequence
.array
);