[PATCH 2/3] kdb, kms: Save and restore the LUT on atomic KMS enter/exit