cmd: replace mem unit formatting

[aead]

This commit replaces the custom memory unit formatting with a package.

Further, this commit fixes a race condition in the metric command.