theread.me/_posts/2022-06-16-rust-boxed-str-vs-string.md

363 lines
16 KiB
Markdown
Raw Normal View History

2022-06-17 12:40:07 +00:00
---
layout: post
title: "What is `Box<str>` and how is it different from `String` in Rust?"
subtitle: Using `rust-lldb` to understand rust memory internals
2022-06-17 12:40:07 +00:00
date: 2022-06-16 00:00:00
permalink: rust-box-str-vs-string/
categories: programming
---
2022-07-27 08:48:57 +00:00
Today I and a friend went down a rabbit hole about Rust and how it manages the
heap when we use `Box`, or `String`, or `Vec`, and while we were at it, I found
out there is such a thing as `Box<str>`, which might look a bit _strange_ to an
untrained eye, since most of the time the `str` primitive type is passed around
as `&str`.
2022-06-17 12:40:07 +00:00
-----
TL;DR:
2022-07-27 08:48:57 +00:00
`Box<str>` is a primitive `str` allocated on the heap, whereas `String` is
actually a `Vec<u8>`, also allocated on the heap, which allows for efficient
removals and appends. `Box<str>` (16 bytes) uses less memory than `String` (24
bytes).
2022-06-17 12:40:07 +00:00
------
2022-07-27 08:48:57 +00:00
I will be using `rust-lldb` throughout this post to understand what is going on
in the rust programs we write and run. The source code for this blog post is
available on
2023-12-06 18:15:32 +00:00
[rust-memory-playground](https://git.theread.me/thereadme/rust-memory-playground).
2022-06-17 12:40:07 +00:00
```bash
2023-12-06 18:15:32 +00:00
git clone https://git.theread.me/thereadme/rust-memory-playground
2022-06-17 12:40:07 +00:00
cd rust-memory-playground
```
# The Stack
2022-07-27 08:48:57 +00:00
Most of the primitive data types used throughout a program, and the information
about the program itself are usually allocated on the stack. Consider this
simple program:
2022-06-17 12:40:07 +00:00
```rust
fn add_ten(a: u8) -> u8 {
let b = 10;
a + b
}
fn main() {
println!("{}", add_ten(9));
}
```
2022-07-27 08:48:57 +00:00
Let's examine the stack when we are running `a + b` by setting a breakpoint on
that line:
2022-06-17 12:40:07 +00:00
```
$ cargo build && rust-lldb target/debug/stack-program
(lldb) breakpoint set -f main.rs -l 3
Breakpoint 1: where = stack-program`stack_program::add_ten::h42edbf0bdcb04851 + 24 at main.rs:3:5, address = 0x0000000100001354
(lldb) run
2023-12-06 18:15:32 +00:00
Process 65188 launched: '/Users/workshop/rust-memory-playground/target/debug/stack-program' (arm64)
2022-06-17 12:40:07 +00:00
Process 65188 stopped
* thread #1, queue = 'com.apple.main-thread', stop reason = breakpoint 1.1
frame #0: 0x0000000100001354 stack-program`stack_program::add_ten::h42edbf0bdcb04851(a=5) at main.rs:3:5
1 fn add_ten(a: u8) -> u8 {
2 let b = 10;
-> 3 a + b
4 }
5
6
7 fn main() {
(lldb) frame var -L -f X
0x000000016fdfed7e: (unsigned char) a = 0x09
0x000000016fdfed7f: (unsigned char) b = 0x0A
```
2022-07-27 08:48:57 +00:00
Our program allocates two variables on the stack directly here. Notice that they
2023-09-14 19:22:14 +00:00
are allocated right next to each other, their address only one byte apart. Most
2022-07-27 08:48:57 +00:00
primitive types are allocated on the stack, and are copied when being passed
around because they are small enough, so that copying them around is more
reasonable than allocating them in the heap and passing around a pointer to
them. In this case, `u8` can be allocated in a single byte, it would not make
sense for us to allocate a pointer (which can vary in size, but are usually
larger than 8 bytes). Every time you call a function, a copy of the values
passed to it, along with the values defined in the function itself constitute
the stack of that function.
The stack of a whole program includes more information though, such as the
_backtrace_, which allows the program to know how to navigate: once I am done
with this function, where should I return to? that information is available in
the stack as well. Note the first couple of lines here, indicating that we are
currently in `stack_program::add_then`, and we came here from
`stack_program::main`, and so once we are finished with `add_then`, we will go
back to `main`:
2022-06-17 12:40:07 +00:00
```
(lldb) thread backtrace
* thread #1, queue = 'com.apple.main-thread', stop reason = breakpoint 1.1
* frame #0: 0x0000000100001350 stack-program`stack_program::add_ten::hf7dc9cccae290c37(a='\t') at main.rs:3:5
frame #1: 0x00000001000013a8 stack-program`stack_program::main::he22b9cf577b52c34 at main.rs:8:20
frame #2: 0x00000001000015a4 stack-program`core::ops::function::FnOnce::call_once::hd6bac0cd3fcb8c07((null)=(stack-program`stack_program::main::he22b9cf577b52c34 at main.rs:7), (null)=<unavailable>) at function.rs:227:5
frame #3: 0x00000001000014c4 stack-program`std::sys_common::backtrace::__rust_begin_short_backtrace::hc4df46810f9a7139(f=(stack-program`stack_program::main::he22b9cf577b52c34 at main.rs:7)) at backtrace.rs:122:18
frame #4: 0x0000000100001178 stack-program`std::rt::lang_start::_$u7b$$u7b$closure$u7d$$u7d$::hbec5b809d627978a at rt.rs:145:18
frame #5: 0x000000010001440c stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] core::ops::function::impls::_$LT$impl$u20$core..ops..function..FnOnce$LT$A$GT$$u20$for$u20$$RF$F$GT$::call_once::h485d4c2966ec30a8 at function.rs:259:13 [opt]
frame #6: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panicking::try::do_call::h375a887be0bea938 at panicking.rs:492:40 [opt]
frame #7: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panicking::try::hecad40482ef3be15 at panicking.rs:456:19 [opt]
frame #8: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panic::catch_unwind::haf1f664eb41a88eb at panic.rs:137:14 [opt]
frame #9: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::rt::lang_start_internal::_$u7b$$u7b$closure$u7d$$u7d$::h976eba434e9ff4cf at rt.rs:128:48 [opt]
frame #10: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panicking::try::do_call::h8f2501ab92e340b0 at panicking.rs:492:40 [opt]
frame #11: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panicking::try::hbeb9f8df83454d42 at panicking.rs:456:19 [opt]
frame #12: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e [inlined] std::panic::catch_unwind::h0a9390b2202af6e9 at panic.rs:137:14 [opt]
frame #13: 0x0000000100014400 stack-program`std::rt::lang_start_internal::hc453db0ee48af82e at rt.rs:128:20 [opt]
frame #14: 0x0000000100001140 stack-program`std::rt::lang_start::h69bdd2191bba2dab(main=(stack-program`stack_program::main::he22b9cf577b52c34 at main.rs:7), argc=1, argv=0x000000016fdff168) at rt.rs:144:17
frame #15: 0x0000000100001434 stack-program`main + 32
frame #16: 0x00000001000750f4 dyld`start + 520
```
# Box, String and Vec: Pointers to Heap
2022-07-27 08:48:57 +00:00
There are times when we are working with data types large enough that we would
really like to avoid copying them when we are passing them around. Let's say you
have just copied a file that is 1,000,000 bytes (1Mb) in size. In this case it
is much more memory and compute efficient to have a pointer to this value (8
bytes) rather than copying all the 1,000,000 bytes.
2022-06-17 12:40:07 +00:00
2022-07-27 08:48:57 +00:00
This is where types such as `Box`, `String` and `Vec` come into play: these
types allow you to allocate something on heap, which is a chunk of memory
separate from the stack that you can allocate on, and later reference those
values using a pointer available on the stack.
2022-06-17 12:40:07 +00:00
2022-07-27 08:48:57 +00:00
Let's start with `Box`, the most generic one, which allows you to allocate some
data on the heap, consider this example:
2022-06-17 12:40:07 +00:00
```rust
fn main() {
let a = Box::new(5_u8);
let b = 10_u8;
println!("{}, {}", a, b);
}
```
We again use `lldb` to check out what is happening:
```
$ cargo build && rust-lldb target/debug/stack-and-heap-program
(lldb) breakpoint set -f main.rs -l 4
Breakpoint 1: where = stack-and-heap-program`stack_and_heap_program::main::ha895783273646dc7 + 100 at main.rs:4:5, address = 0x0000000100005264
(lldb) run
2023-12-06 18:15:32 +00:00
Process 67451 launched: '/Users/workshop/rust-memory-playground/target/debug/stack-and-heap-program' (arm64)
2022-06-17 12:40:07 +00:00
Process 67451 stopped
* thread #1, queue = 'com.apple.main-thread', stop reason = breakpoint 1.1
frame #0: 0x0000000100005264 stack-and-heap-program`stack_and_heap_program::main::ha895783273646dc7 at main.rs:4:5
1 fn main() {
2 let a = Box::new(5_u8);
3 let b = 10_u8;
-> 4 println!("{}, {}", a, b);
5 }
(lldb) frame var -L -f X
0x000000016fdfed48: (unsigned char *) a = 0x0000600000008010 "\U00000005"
0x000000016fdfed57: (unsigned char) b = 0x0A
(lldb) memory read -count 1 -f X 0x0000600000008010
0x600000008010: 0x05
```
2022-07-27 08:48:57 +00:00
Note that here, instead of `a` having the value `5`, has the value
`0x0000600000008010`, which is a pointer to a location in memory! `lldb` is
recognises that this is a pointer (note the `*` sign beside the variable type)
and shows us what the memory location contains, but we can also directly read
that memory location, and of course we find `5` there. The address of the
heap-allocated `5` is far from the stack-allocated `10`, since stack and heap
are separate parts of memory.
2022-06-17 12:40:07 +00:00
2022-07-27 08:48:57 +00:00
Using `Box` for an unsigned 8-bit value does not really make sense, the value
itself is smaller than the pointer created by `Box`, however allocating on heap
is useful when we have data that we need be able to pass around the program
without copying it.
2022-06-17 12:40:07 +00:00
2022-07-27 08:48:57 +00:00
Turns out, `String` and `Vec` cover two of the most common cases where we may
want to allocate something on heap! Let's look at what goes on behind allocating
a variable of type `String`:
2022-06-17 12:40:07 +00:00
```rust
fn main() {
2022-06-28 14:47:13 +00:00
let s = String::from("hello");
2022-06-17 12:40:07 +00:00
println!("{}", s);
}
```
And here we go again:
```
(lldb) breakpoint set -f main.rs -l 3
Breakpoint 1: where = string-program`string_program::main::h64ca96ee87b0ceaf + 44 at main.rs:3:5, address = 0x000000010000476c
(lldb) run
2023-12-06 18:15:32 +00:00
Process 68317 launched: '/Users/workshop/rust-memory-playground/target/debug/string-program' (arm64)
2022-06-17 12:40:07 +00:00
Process 68317 stopped
* thread #1, queue = 'com.apple.main-thread', stop reason = breakpoint 1.1
frame #0: 0x000000010000476c string-program`string_program::main::h64ca96ee87b0ceaf at main.rs:3:5
1 fn main() {
2022-06-28 14:47:13 +00:00
2 let s = String::from("hello");
2022-06-17 12:40:07 +00:00
-> 3 println!("{}", s);
4 }
(lldb) frame var -L -T
2022-06-28 14:47:13 +00:00
0x000000016fdfed78: (alloc::string::String) s = "hello" {
0x000000016fdfed78: (alloc::vec::Vec<unsigned char, alloc::alloc::Global>) vec = size=5 {
2022-06-17 12:40:07 +00:00
0x0000600000004010: (unsigned char) [0] = 'h'
0x0000600000004011: (unsigned char) [1] = 'e'
0x0000600000004012: (unsigned char) [2] = 'l'
0x0000600000004013: (unsigned char) [3] = 'l'
0x0000600000004014: (unsigned char) [4] = 'o'
}
}
```
2022-07-27 08:48:57 +00:00
This is a formatted output from `lldb`, and here you can see that the `String`
type is basically a `Vec<unsigned char, alloc::Global>` (note that `unsigned
char` is represented using `u8` in Rust, so in Rust terminology the type is
`Vec<u8>`), let's now look at the same command but this time raw and unformatted
(`-R`):
2022-06-17 12:40:07 +00:00
```
(lldb) frame var -L -T -R
0x000000016fdfed78: (alloc::string::String) s = {
0x000000016fdfed78: (alloc::vec::Vec<unsigned char, alloc::alloc::Global>) vec = {
0x000000016fdfed78: (alloc::raw_vec::RawVec<unsigned char, alloc::alloc::Global>) buf = {
0x000000016fdfed78: (core::ptr::unique::Unique<unsigned char>) ptr = {
0x000000016fdfed78: (unsigned char *) pointer = 0x0000600000004010
0x000000016fdfed78: (core::marker::PhantomData<unsigned char>) _marker = {}
}
2022-10-15 17:25:26 +00:00
0x000000016fdfed80: (unsigned long) cap = 5
2022-06-17 12:40:07 +00:00
0x000000016fdfed78: (alloc::alloc::Global) alloc = {}
}
2022-10-15 17:25:26 +00:00
0x000000016fdfed88: (unsigned long) len = 5
2022-06-17 12:40:07 +00:00
}
}
```
2022-07-27 08:48:57 +00:00
Ah! I see the `ptr` field of `RawVec` with a value of `0x0000600000004010`, that
is the memory address of the beginning of our string (namely the `h` of our
`hello`)! There is also `cap` and `len`, which respectively stand for capacity
2022-10-15 17:25:26 +00:00
and length, with the value 5, indicating that our string is of capacity and
length 5; the difference between the two being that [you can have a `Vec` with a
2022-07-27 08:48:57 +00:00
capacity of 10 while it has zero
items](https://doc.rust-lang.org/nightly/std/vec/struct.Vec.html#capacity-and-reallocation),
this would allow you to append 10 items to the `Vec` without having a new
allocation for each append, making the process more efficient, and also a [Vec
is not automatically shrunk
down](https://doc.rust-lang.org/nightly/std/vec/struct.Vec.html#guarantees) in
size when items are removed from it to avoid unnecessary deallocations, hence
the length might be smaller than the capacity. So in a nutshell, our String is
basically something like this (inspired by
[std::vec::Vec](https://doc.rust-lang.org/nightly/std/vec/struct.Vec.html#guarantees)):
2022-06-17 12:40:07 +00:00
```
Stack:
--------------------------------
| String |
| \-> Vec |
| \-> (ptr, cap, len) |
| | |
-----------------|--------------
Heap: v
-----------------------------
| ('h', 'e', 'l', 'l', 'o') |
-----------------------------
```
2022-07-27 08:48:57 +00:00
Okay, so far so good. We have `String`, which uses a `Vec` under the hood, which
is represented by a pointer, capacity and length triplet.
2022-06-17 12:40:07 +00:00
2022-07-27 08:48:57 +00:00
If `String` is already heap-allocated, why would anyone want `Box<str>`!? Let's
look at how `Box<str>` would be represented in memory:
2022-06-17 12:40:07 +00:00
```rust
fn main() {
let boxed_str: Box<str> = "hello".into();
println!("boxed_str: {}", boxed_str);
}
```
And `lldb` tells us:
```
0x000000016fdfed80: (alloc::boxed::Box<>) boxed_str = {
0x000000016fdfed80: data_ptr = 0x0000600000004010 "hello"
0x000000016fdfed88: length = 5
}
```
2022-07-27 08:48:57 +00:00
Okay, so a `Box<str>` is much simpler than a `String`: there is no `Vec`, and no
`capacity`, and the underlying data is a primitive `str` that does not allow
efficient appending or removing. It is a smaller representation as well, due to
the missing `capacity` field, comparing their memory size on stack using
[std::mem::size_of_val](https://doc.rust-lang.org/std/mem/fn.size_of_val.html):
2022-06-17 12:40:07 +00:00
```rust
let boxed_str: Box<str> = "hello".into();
println!("size of boxed_str on stack: {}", std::mem::size_of_val(&boxed_str));
2022-06-28 14:47:13 +00:00
let s = String::from("hello");
2022-06-17 12:40:07 +00:00
println!("size of string on stack: {}", std::mem::size_of_val(&s));
```
Results in:
```
size of boxed_str on stack: 16
size of string on stack: 24
```
2022-07-27 08:48:57 +00:00
Note that their size on heap is the same, because they are both storing the
bytes for `hello` on the heap (the measurements below show all of the heap
allocations of the program, and not only the string. What matters here is that
these two programs have exact same heap size in total):
2022-06-17 12:40:07 +00:00
```
$ cargo run --bin string-dhat
Finished dev [unoptimized + debuginfo] target(s) in 0.01s
Running `target/debug/string-dhat`
hello
dhat: Total: 1,029 bytes in 2 blocks
dhat: At t-gmax: 1,029 bytes in 2 blocks
dhat: At t-end: 1,024 bytes in 1 blocks
dhat: The data has been saved to dhat-heap.json, and is viewable with dhat/dh_view.html
$ cargo run --bin box-str-dhat
Finished dev [unoptimized + debuginfo] target(s) in 0.01s
Running `target/debug/box-str-dhat`
boxed_str: hello
dhat: Total: 1,029 bytes in 2 blocks
dhat: At t-gmax: 1,029 bytes in 2 blocks
dhat: At t-end: 1,024 bytes in 1 blocks
dhat: The data has been saved to dhat-heap.json, and is viewable with dhat/dh_view.html
```
2022-06-17 13:50:31 +00:00
There is also `Box<[T]>` which is the fixed size counterpart to `Vec<T>`.
2022-06-17 13:50:31 +00:00
# Should I use `Box<str>` or `String`?
2022-07-27 08:48:57 +00:00
The only use case for `Box<str>` over `String` that I can think of, is
optimising for memory usage when the string is fixed and you do not intend to
append or remove from it. I looked for examples of `Box<str>` being used, and I
found a few examples:
- Hyper uses it in a part to reduce memory usage, since the string they have is
read-only: [hyper#2727](https://github.com/hyperium/hyper/pull/2727)
- Rust-analyzer uses it to store some strings in their snippets data structre:
[rust-lang/rust-analyzer/crates/ide-completion/src/snippet.rs](https://github.com/rust-lang/rust-analyzer/blob/5c88d9344c5b32988bfbfc090f50aba5de1db062/crates/ide-completion/src/snippet.rs#L123)
- It is also used in some parts in the compiler itself, probably with the same
aim of optimising memory usage:
[rust-lang/rust/src/libsyntax/symbol.rs](https://github.com/rust-lang/rust/blob/7846610470392abc3ab1470853bbe7b408fe4254/src/libsyntax/symbol.rs#L82-L85)