Supports rollback when allocate callback function fails

Supports rollback when allocate callback function fails #5837