trace: rework alloc tracing