File: align_ptr.h

package info (click to toggle)
scrypt 1.3.3-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,368 kB
  • sloc: ansic: 5,763; sh: 5,504; makefile: 257
file content (52 lines) | stat: -rw-r--r-- 1,697 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#ifndef ALIGN_PTR_H_
#define ALIGN_PTR_H_

#include <stddef.h>
#include <stdint.h>

/**
 * ALIGN_PTR_DECL(type, name, num, alignsize):
 * Declare a pointer called ${name}, which points to an array large enough to
 * contain ${num} values of ${type} and is aligned to ${alignsize} bytes.  The
 * pointer must not be used until ALIGN_PTR_INIT(${name}, ${alignsize}) has
 * been called.  This macro may also create an additional variable called
 * "${name}_buf".
 */
#define ALIGN_PTR_DECL(type, name, num, alignsize)			\
	uint8_t name##_buf[num * sizeof(type) + (alignsize - 1)];	\
	type * name

/**
 * ALIGN_PTR_INIT(name, alignsize):
 * Initialize the variable called ${name} to point to an array which is
 * aligned to ${alignsize} bytes.  They must have previously been declared
 * with ALIGN_PTR_DECL(${name}, ${alignsize}).  This macro assumes that
 * casting a pointer to uintptr_t behaves naturally, i.e. yields a byte offset
 * relative to aligned memory.
 */
#define ALIGN_PTR_INIT(name, alignsize)					\
	name = align_ptr(name##_buf, alignsize)

/**
 * align_ptr(arr, alignment):
 * Return a pointer to the first memory location within ${arr} which is
 * aligned to ${alignsize} bytes.  (It is expected that this function will
 * only be called via the ALIGN_PTR_INIT macro).
 */
static inline void *
align_ptr(uint8_t * arr, size_t alignment)
{
	size_t offset;

	/*
	 * This assumes that casting a pointer to uintptr_t behaves naturally,
	 * i.e. yields a byte offset relative to aligned memory.
	 */
	offset = (uintptr_t)(&arr[0]) % alignment;
	offset = (alignment - offset) % alignment;

	/* Return the aligned pointer. */
	return ((void *)&arr[offset]);
}

#endif /* !ALIGN_PTR_H_ */