Handle the allocation error case first, so that we don't have further
nested if for handling the common case of success.
Signed-off-by: NRoland Dreier <rolandd@cisco.com>