Someone else asked me the same question and I wrote a blog-post about a solution: Write your own Arduino millis() in Rust
If you just care about the code, you can find it here: https://github.com/Rahix/avr-hal/blob/master/boards/arduino-uno/examples/uno-millis.rs