diag: Fix incorrect checks in callback mode

Diag driver is currently checking the length of the received packet
incorrectly in callback mode. This patch fixes this issue.

Change-Id: I0544e1e0a665d9f764a88a19285abdccd7477714
Signed-off-by: Ravi Aravamudhan <aravamud@codeaurora.org>