This is more idiomatic and fixes bugs in the error handling paths.
Signed-off-by: Mark Brown <broonie@linaro.org>