"Adding a 32-bit key ID field to struct bio" for Inline Encryption

Adding a 32bit key ID field to struct bio, will be helpful to associate I/O requests to the encryption keys programmed in the key slot. But the challenging part is the key slots are limited, let us say 32 key programming slots in the case of UFS Host Controller Inline Encryption engine. This demands providing a reference to the crypto context in the bio structure; reason is when the key slots are exhausted, there can be multiple crypto context to the same hardware key slot and key reprogramming is required if the key programmed in the slot doesn't match with that of crypto context.
I am not sure providing a reference in the bio to the crypto context has security risks associated to same.