Applied Pokology
Back to blog...
_____
---' __\_______
______) Integral structs in GNU poke
__)
__)
---._______)
Jose E. Marchesi
July 7, 2020
This weekend I finally implemented support for the so-called "integral
structs" in poke. This expands the expressivity power of Poke structs
to cover cases where data is stored in composited integral containers,
i.e. where data is structured within stored integers.
Basically, when we structure data using Poke structs, arrays and the
like, we often use the same structure than a C programmer would use.
For example, to model ELF RELA structures, which are defined in C like:
,----
| type struct
| {
| Elf64_Addr r_offset; /* Address */
| Elf64_Xword r_info; /* Relocation type and symbol index */
| Elf64_Sxword r_addend; /* Addend */
| } Elf64_Rela;
`----
we could use something like this in Poke:
,----
| type Elf64_Rela =
| struct
| {
| Elf64_Addr r_offset;
| Elf64_Xword r_info;
| Elf64_Sxword r_addend;
| };
`----
Here the Poke struct type is pretty equivalent to the C incarnation. In
both cases the fields are always stored in the given order, regardless
of endianness or any other consideration.
However, there are situations where stored integral values are to be
interpreted as composite data. This is the case of the `r_info' field
above, which is a 64-bit unsigned integer (`Elf64_Xword') which is
itself composed by several fields, depicted here:
,----
| 63 0
| +----------------------+----------------------+
| | r_sym | r_type |
| +----------------------+----------------------+
| MSB LSB
`----
In order to support this kind of composition of integers, C programmers
usually resort to either bit masking (most often) or to the often
obscure and undefined behaviour-prone C bit fields. In the case of ELF,
the GNU implementations define a few macros to access these
"sub-fields":
,----
| #define ELF64_R_SYM(i) ((i) >> 32)
| #define ELF64_R_TYPE(i) ((i) & 0xffffffff)
| #define ELF64_R_INFO(sym,type) ((((Elf64_Xword) (sym)) << 32) + (type))
`----
Where `ELF64_R_SYM' and `ELF64_R_TYPE' are used to extract the fields
from an `r_info', and `ELF64_R_INFO' is used to compose it. This is
typical of C data structures.
We could of course mimic the C implementation in Poke:
,----
| fun Elf64_R_Sym = (Elf64_Xword i) uint<32>:
| { return i .>> 32; }
| fun Elf64_R_Type = (Elf64_Xword i) uint<32>:
| { return i & 0xffff_ffff; }
| fun Elf64_R_Info = (uint<32> sym, uint<32> type) Elf64_Xword:
| { return sym as Elf64_Xword <<. 32 + type; }
`----
However, this approach has a huge disadvantage: since we are not able to
encode the logic of these "sub-fields" in proper Poke fields, they
become second class citizens, with all that implies: no constraints on
their own, can't be auto-completed, can't be assigned individually, etc
etc.
But starting today we can use "integral structs"! These are structs
that are defined exactly like your garden variety Poke structs, with a
small addition:
,----
| type Elf64_RelInfo =
| struct uint<64>
| {
| uint<32> r_sym;
| uint<32> r_type;
| };
`----
Note the `uint<64>' addition after `struct'. This can be any
integer type (signed or unsigned). The fields of an integral struct
should be integral themselves (this includes both integers and offsets)
and the total size occupied by the fields should be the same size than
the one declared in the struct's integer type. This is checked and
enforced by the compiler.
The Elf64 RELA in Poke can then be encoded like:
,----
| type Elf64_Rela =
| struct
| {
| Elf64_Addr r_offset;
| struct Elf64_Xword
| {
| uint<32> r_sym;
| uint<32> r_type;
| } r_info;
| Elf64_Sxword r_addend;
| };
`----
When an integral struct is mapped from some IO space, the total number
of bytes occupied by the struct is read as a single integer value, and
then the values of the fields are extracted from it. A similar process
is using when writing. That is what makes it different with respect a
normal Poke struct.
Consider for example we have the following sequence of bytes in our IO
space (like a file):
,----
| 0x10 0x20 0x30 0x40 0x50 0x60 0x70 0x80
`----
Let's see what happens when we map the integral struct above, in both
big and little endian:
,----
| Elf64_RelInfo {
| r_sym=0x10203040U,
| r_type=0x50607080U
| }
| (poke) .set endian little
| (poke) Elf64_RelInfo @ 0#B
| Elf64_RelInfo {
| r_sym=0x80706050U,
| r_type=0x40302010U
| }
`----
For comparison, this is what happens when we do the same with an
"equivalent" (not really) non-integral struct operating on the same
data:
,----
| type Elf64_RelInfoBogus =
| struct
| {
| uint<32> r_sym;
| uint<32> r_type;
| };
`----
We would get:
,----
| (poke) .set endian big
| (poke) Elf64_RelInfoBogus @ 0#B
| Elf64_RelInfoBogus {
| r_sym=0x10203040U,
| r_type=0x50607080U
| }
| (poke) .set endian little
| (poke) Elf64_RelInfoBogus @ 0#B
| Elf64_RelInfoBogus {
| r_sym=0x40302010U,
| r_type=0x80706050U
| }
`----
In this case, and unlike with integral structs, the endianness impacts
the bytes of the individual fields, not of the whole struct.
As you can see, integral structs can be used to denote a lot of commonly
found idioms in data structures and this includes a lot of what is
sometimes denoted in C bit field. However, one should be cautious when
"translating" C structures to Poke, especially when the C programmer has
not been careful and incurres in sometimes obscure
implementation-defined behavior. An integral struct is not always the
right abstraction to use when we see a C bit field!
As an example of the above, consider the following C struct:
,----
| struct regs
| {
| __u8 dst_reg:4;
| __u8 src_reg:4;
| };
`----
Certain virtual architecture uses that data layout to store registers in
instructions (no comment.) Thing is, in bit fields like the above with
sub-byte field sizes, the ordering of the fields is not clearly defined,
and ultimately what order to use is up to the compiler, i.e. to lore and
tradition. As it happens, GCC encodes `src_reg' in the most significant
nibble of the byte and `dst_reg' in the least significant nibble of the
byte when compiling for a little-endian target, and the other way around
when compiling for a little-endian target. (I may have had that wrong,
this always confuses me.)
How could we encode the C struct regs in Poke? Let's see.
A normal Poke struct clearly won't do it:
,----
| type RegsBogus1 =
| struct
| {
| uint<4> src;
| uint<4> dst;
| };
`----
The reason being, the ordering of src and dst does not change when you
switch endianness (since this is Poke, we can in fact talk about real
ordering of bits)... remember, poke is WYPIWIG (what you poke is what
you get) ;)
What about an integral struct?
,----
| type RegsBogus2 =
| struct uint<8>
| {
| uint<4> src;
| uint<4> dst;
| };
`----
This won't work either. In fact, the net effect of the normal decoding
of the normal struct type RegsBogus1 and the
map-an-integer-and-extract-fields decoding of the integral struct
RegsBogus2 is in this case totally equivalent.
A solution is to use a normal struct, and field labels:
,----
| type RegsBogus =
| struct
| {
| var little_p = (get_endian == ENDIAN_LITTLE);
|
| uint<8> src @ !little_p * 4#b;
| uint<8> dst @ little_p * 4#b;
| };
`----
At this point, you may be wondering: is there anything particular in a
field defined in an integral struct? The answer is: no, not at all.
These are regular, first-class fields. Likewise, integral structs are
perfectly regular structs. And of course, since this is poke, you can
have integral structs of say, 11 bits, or 3 bits, map them at offsets
not aligned to bytes, and all the typical poke-atrocities that we enjoy
so much.
However, there exist a few restrictions, some of them fundamental, the
others to be lifted eventually:
- There are no integral unions. This is a fundamental limitation and
will most likely stay like that.
- Integral structs can only have integral fields. This includes
offsets.
- No labels are allowed in the fields of integral structs. This is not
a fundamental limitation, and may be supported at some point.
- No integral structs are supported inside other integral structs. This
is purely because of lazyness on my part. This will be eventually
supported.
- No optional fields are supported in integral structs. Support for
this is actually partially implemented (the mapper supports them but
not the writer) and most probably will be completed one of these days.
Happy poking friends!
BONUS
=====
My intention is to add support for casts from/to integral structs
to/from integers, so we can do things like:
,----
| rel.r_info as uint<64>
`----
And also automatic promotions in arithmetic operators, like:
,----
| rel.r_info + 20 * rel.r_info.r_type
`----
UPDATE: this is implemented now :)