assign_in_data_buf is considered "assigned" only if the callback's return code is 0

Change-Id: I8243ec7dddc7953155b83c9978b949b05e22613e
Signed-off-by: Eyal Salomon <esalomon@mellanox.com>