simplify and symmetry the implementation in KERNEL/PROTECTED build Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com> Change-Id: Iefdeea5f6ef6348c774b2ca9f7e45fe89c0c22dd