Instead of using comments in an attempt at getting the locking right,
use proper assertions that actively warn you if you got it wrong.

Also add extra braces in a few sites to comply with coding-style.
Signed-off-by: NPeter Zijlstra <peterz@infradead.org>
Signed-off-by: NPekka Enberg <penberg@kernel.org>