Did you mean to add read, write, and erase access features in UFI?
Well, this may not be possible for emmc and ufs toolbox,
because direct access to the RPMB area is only valid if calculated by the SoC.
But for Android ToolBox, especially Mediatek platform will be very possible with brom exploit.
And we have prepared it, hopefully it can be released in the next update.
I give you sneak peek of the accessing UFS as well as for eMMC through Android ToolBox which pretty much like eMMC or UFS Tool Box direct access, but with special task to read, write, and erase RPMB area below.