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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
#ifndef _PCILIB_KMEM_H
#define _PCILIB_KMEM_H
#include "pcilib.h"
typedef struct pcilib_kmem_list_s pcilib_kmem_list_t;
#include "tools.h"
typedef enum {
PCILIB_KMEM_FLAG_REUSE = KMEM_FLAG_REUSE,
PCILIB_KMEM_FLAG_EXCLUSIVE = KMEM_FLAG_EXCLUSIVE,
PCILIB_KMEM_FLAG_PERSISTENT = KMEM_FLAG_PERSISTENT,
PCILIB_KMEM_FLAG_HARDWARE = KMEM_FLAG_HW,
PCILIB_KMEM_FLAG_FORCE = KMEM_FLAG_FORCE,
PCILIB_KMEM_FLAG_TRY = KMEM_FLAG_TRY
} pcilib_kmem_flags_t;
typedef enum {
PCILIB_KMEM_REUSE_ALLOCATED = PCILIB_TRISTATE_NO,
PCILIB_KMEM_REUSE_REUSED = PCILIB_TRISTATE_YES,
PCILIB_KMEM_REUSE_PARTIAL = PCILIB_TRISTATE_PARTIAL,
PCILIB_KMEM_REUSE_PERSISTENT = 0x100,
PCILIB_KMEM_REUSE_HARDWARE = 0x200
} pcilib_kmem_reuse_state_t;
typedef struct {
int handle_id;
pcilib_kmem_reuse_state_t reused;
uintptr_t pa;
// uintptr_t va;
void *ua;
size_t size;
size_t alignment_offset;
size_t mmap_offset;
} pcilib_kmem_addr_t;
/**
* single allocation - we set only addr, n_blocks = 0
* multiple allocation - addr is not set, blocks are set, n_blocks > 0
* sgmap allocation - addr contains ua, but pa's are set in blocks, n_blocks > 0
*/
typedef struct {
pcilib_kmem_addr_t addr;
pcilib_kmem_reuse_state_t reused;
size_t n_blocks;
pcilib_kmem_addr_t blocks[];
} pcilib_kmem_buffer_t;
typedef void pcilib_kmem_handle_t;
struct pcilib_kmem_list_s {
pcilib_kmem_list_t *next, *prev;
pcilib_kmem_buffer_t buf; // variable size, should be last item in struct
};
pcilib_kmem_handle_t *pcilib_alloc_kernel_memory(pcilib_t *ctx, pcilib_kmem_type_t type, size_t nmemb, size_t size, size_t alignment, pcilib_kmem_use_t use, pcilib_kmem_flags_t flags);
void pcilib_free_kernel_memory(pcilib_t *ctx, pcilib_kmem_handle_t *k, pcilib_kmem_flags_t flags);
//int pcilib_kmem_sync(pcilib_t *ctx, pcilib_kmem_handle_t *k, pcilib_kmem_sync_direction_t dir);
int pcilib_kmem_sync_block(pcilib_t *ctx, pcilib_kmem_handle_t *k, pcilib_kmem_sync_direction_t dir, size_t block);
void *pcilib_kmem_get_ua(pcilib_t *ctx, pcilib_kmem_handle_t *k);
uintptr_t pcilib_kmem_get_pa(pcilib_t *ctx, pcilib_kmem_handle_t *k);
uintptr_t pcilib_kmem_get_ba(pcilib_t *ctx, pcilib_kmem_handle_t *k);
void *pcilib_kmem_get_block_ua(pcilib_t *ctx, pcilib_kmem_handle_t *k, size_t block);
uintptr_t pcilib_kmem_get_block_pa(pcilib_t *ctx, pcilib_kmem_handle_t *k, size_t block);
uintptr_t pcilib_kmem_get_block_ba(pcilib_t *ctx, pcilib_kmem_handle_t *k, size_t block);
size_t pcilib_kmem_get_block_size(pcilib_t *ctx, pcilib_kmem_handle_t *k, size_t block);
pcilib_kmem_reuse_state_t pcilib_kmem_is_reused(pcilib_t *ctx, pcilib_kmem_handle_t *k);
int pcilib_clean_kernel_memory(pcilib_t *ctx, pcilib_kmem_use_t use, pcilib_kmem_flags_t flags);
#endif /* _PCILIB_KMEM_H */
|