mmu: Move code around

Move code that will be needed by unified page walker. No changes to
generated code.

Signed-off-by: Gleb Natapov <gleb@cloudius-systems.com>
Signed-off-by: Avi Kivity <avi@cloudius-systems.com>