This reworks the code a bit and also disallows NULL contexts.

Fixes Coverity CID 1396273, 1396279.
Signed-off-by: Marton Balint <cus@passwd.hu>