Commit | Line | Data |
---|---|---|
c1d7c514 | 1 | // SPDX-License-Identifier: GPL-2.0 |
0f82731f CM |
2 | /* |
3 | * Copyright (C) 2007 Oracle. All rights reserved. | |
0f82731f CM |
4 | */ |
5 | ||
18077bb4 | 6 | #include <asm/unaligned.h> |
d352ac68 | 7 | |
18077bb4 LZ |
8 | #include "ctree.h" |
9 | ||
10 | static inline u8 get_unaligned_le8(const void *p) | |
11 | { | |
12 | return *(u8 *)p; | |
13 | } | |
14 | ||
15 | static inline void put_unaligned_le8(u8 val, void *p) | |
16 | { | |
17 | *(u8 *)p = val; | |
18 | } | |
19 | ||
5e394689 DS |
20 | static bool check_setget_bounds(const struct extent_buffer *eb, |
21 | const void *ptr, unsigned off, int size) | |
22 | { | |
23 | const unsigned long member_offset = (unsigned long)ptr + off; | |
24 | ||
25 | if (member_offset > eb->len) { | |
26 | btrfs_warn(eb->fs_info, | |
27 | "bad eb member start: ptr 0x%lx start %llu member offset %lu size %d", | |
28 | (unsigned long)ptr, eb->start, member_offset, size); | |
29 | return false; | |
30 | } | |
31 | if (member_offset + size > eb->len) { | |
32 | btrfs_warn(eb->fs_info, | |
33 | "bad eb member end: ptr 0x%lx start %llu member offset %lu size %d", | |
34 | (unsigned long)ptr, eb->start, member_offset, size); | |
35 | return false; | |
36 | } | |
37 | ||
38 | return true; | |
39 | } | |
40 | ||
18077bb4 LZ |
41 | /* |
42 | * this is some deeply nasty code. | |
d352ac68 CM |
43 | * |
44 | * The end result is that anyone who #includes ctree.h gets a | |
18077bb4 | 45 | * declaration for the btrfs_set_foo functions and btrfs_foo functions, |
01327610 | 46 | * which are wrappers of btrfs_set_token_#bits functions and |
18077bb4 | 47 | * btrfs_get_token_#bits functions, which are defined in this file. |
d352ac68 CM |
48 | * |
49 | * These setget functions do all the extent_buffer related mapping | |
50 | * required to efficiently read and write specific fields in the extent | |
51 | * buffers. Every pointer to metadata items in btrfs is really just | |
52 | * an unsigned long offset into the extent buffer which has been | |
53 | * cast to a specific type. This gives us all the gcc type checking. | |
54 | * | |
18077bb4 LZ |
55 | * The extent buffer api is used to do the page spanning work required to |
56 | * have a metadata blocksize different from the page size. | |
cb495113 DS |
57 | * |
58 | * There are 2 variants defined, one with a token pointer and one without. | |
d352ac68 CM |
59 | */ |
60 | ||
18077bb4 | 61 | #define DEFINE_BTRFS_SETGET_BITS(bits) \ |
cc4c13d5 DS |
62 | u##bits btrfs_get_token_##bits(struct btrfs_map_token *token, \ |
63 | const void *ptr, unsigned long off) \ | |
0f82731f | 64 | { \ |
8f9da810 DS |
65 | const unsigned long member_offset = (unsigned long)ptr + off; \ |
66 | const unsigned long idx = member_offset >> PAGE_SHIFT; \ | |
67 | const unsigned long oip = offset_in_page(member_offset); \ | |
68 | const int size = sizeof(u##bits); \ | |
ba8a9a05 DS |
69 | u8 lebytes[sizeof(u##bits)]; \ |
70 | const int part = PAGE_SIZE - oip; \ | |
18077bb4 | 71 | \ |
48bc3950 | 72 | ASSERT(token); \ |
870b388d | 73 | ASSERT(token->kaddr); \ |
5e394689 | 74 | ASSERT(check_setget_bounds(token->eb, ptr, off, size)); \ |
8f9da810 DS |
75 | if (token->offset <= member_offset && \ |
76 | member_offset + size <= token->offset + PAGE_SIZE) { \ | |
77 | return get_unaligned_le##bits(token->kaddr + oip); \ | |
18077bb4 | 78 | } \ |
ba8a9a05 DS |
79 | token->kaddr = page_address(token->eb->pages[idx]); \ |
80 | token->offset = idx << PAGE_SHIFT; \ | |
81 | if (oip + size <= PAGE_SIZE) \ | |
8f9da810 | 82 | return get_unaligned_le##bits(token->kaddr + oip); \ |
ba8a9a05 DS |
83 | \ |
84 | memcpy(lebytes, token->kaddr + oip, part); \ | |
8f9da810 DS |
85 | token->kaddr = page_address(token->eb->pages[idx + 1]); \ |
86 | token->offset = (idx + 1) << PAGE_SHIFT; \ | |
ba8a9a05 DS |
87 | memcpy(lebytes + part, token->kaddr, size - part); \ |
88 | return get_unaligned_le##bits(lebytes); \ | |
0f82731f | 89 | } \ |
cb495113 DS |
90 | u##bits btrfs_get_##bits(const struct extent_buffer *eb, \ |
91 | const void *ptr, unsigned long off) \ | |
92 | { \ | |
1441ed9b DS |
93 | const unsigned long member_offset = (unsigned long)ptr + off; \ |
94 | const unsigned long oip = offset_in_page(member_offset); \ | |
84da071f DS |
95 | const unsigned long idx = member_offset >> PAGE_SHIFT; \ |
96 | char *kaddr = page_address(eb->pages[idx]); \ | |
1441ed9b | 97 | const int size = sizeof(u##bits); \ |
84da071f DS |
98 | const int part = PAGE_SIZE - oip; \ |
99 | u8 lebytes[sizeof(u##bits)]; \ | |
cb495113 | 100 | \ |
5e394689 | 101 | ASSERT(check_setget_bounds(eb, ptr, off, size)); \ |
84da071f | 102 | if (oip + size <= PAGE_SIZE) \ |
1441ed9b | 103 | return get_unaligned_le##bits(kaddr + oip); \ |
84da071f DS |
104 | \ |
105 | memcpy(lebytes, kaddr + oip, part); \ | |
106 | kaddr = page_address(eb->pages[idx + 1]); \ | |
107 | memcpy(lebytes + part, kaddr, size - part); \ | |
108 | return get_unaligned_le##bits(lebytes); \ | |
cb495113 | 109 | } \ |
cc4c13d5 | 110 | void btrfs_set_token_##bits(struct btrfs_map_token *token, \ |
1cbb1f45 | 111 | const void *ptr, unsigned long off, \ |
cc4c13d5 | 112 | u##bits val) \ |
0f82731f | 113 | { \ |
ce7afe87 DS |
114 | const unsigned long member_offset = (unsigned long)ptr + off; \ |
115 | const unsigned long idx = member_offset >> PAGE_SHIFT; \ | |
116 | const unsigned long oip = offset_in_page(member_offset); \ | |
117 | const int size = sizeof(u##bits); \ | |
118 | __le##bits leres; \ | |
18077bb4 | 119 | \ |
48bc3950 | 120 | ASSERT(token); \ |
870b388d | 121 | ASSERT(token->kaddr); \ |
5e394689 | 122 | ASSERT(check_setget_bounds(token->eb, ptr, off, size)); \ |
ce7afe87 DS |
123 | if (token->offset <= member_offset && \ |
124 | member_offset + size <= token->offset + PAGE_SIZE) { \ | |
125 | put_unaligned_le##bits(val, token->kaddr + oip); \ | |
18077bb4 LZ |
126 | return; \ |
127 | } \ | |
ce7afe87 DS |
128 | if (oip + size <= PAGE_SIZE) { \ |
129 | token->kaddr = page_address(token->eb->pages[idx]); \ | |
130 | token->offset = idx << PAGE_SHIFT; \ | |
131 | put_unaligned_le##bits(val, token->kaddr + oip); \ | |
18077bb4 LZ |
132 | return; \ |
133 | } \ | |
ce7afe87 DS |
134 | token->kaddr = page_address(token->eb->pages[idx + 1]); \ |
135 | token->offset = (idx + 1) << PAGE_SHIFT; \ | |
136 | leres = cpu_to_le##bits(val); \ | |
137 | write_extent_buffer(token->eb, &leres, member_offset, size); \ | |
cb495113 | 138 | } \ |
2b48966a | 139 | void btrfs_set_##bits(const struct extent_buffer *eb, void *ptr, \ |
cb495113 DS |
140 | unsigned long off, u##bits val) \ |
141 | { \ | |
029e4a42 DS |
142 | const unsigned long member_offset = (unsigned long)ptr + off; \ |
143 | const unsigned long oip = offset_in_page(member_offset); \ | |
144 | const int size = sizeof(u##bits); \ | |
145 | __le##bits leres; \ | |
cb495113 | 146 | \ |
5e394689 | 147 | ASSERT(check_setget_bounds(eb, ptr, off, size)); \ |
029e4a42 DS |
148 | if (oip + size <= PAGE_SIZE) { \ |
149 | const unsigned long idx = member_offset >> PAGE_SHIFT; \ | |
150 | char *kaddr = page_address(eb->pages[idx]); \ | |
151 | put_unaligned_le##bits(val, kaddr + oip); \ | |
cb495113 DS |
152 | return; \ |
153 | } \ | |
029e4a42 DS |
154 | leres = cpu_to_le##bits(val); \ |
155 | write_extent_buffer(eb, &leres, member_offset, size); \ | |
18077bb4 | 156 | } |
0f82731f | 157 | |
18077bb4 LZ |
158 | DEFINE_BTRFS_SETGET_BITS(8) |
159 | DEFINE_BTRFS_SETGET_BITS(16) | |
160 | DEFINE_BTRFS_SETGET_BITS(32) | |
161 | DEFINE_BTRFS_SETGET_BITS(64) | |
0f82731f | 162 | |
1cbb1f45 | 163 | void btrfs_node_key(const struct extent_buffer *eb, |
e644d021 CM |
164 | struct btrfs_disk_key *disk_key, int nr) |
165 | { | |
166 | unsigned long ptr = btrfs_node_key_ptr_offset(nr); | |
e644d021 CM |
167 | read_eb_member(eb, (struct btrfs_key_ptr *)ptr, |
168 | struct btrfs_key_ptr, key, disk_key); | |
169 | } |