This is preparation for USB support which will have different diag
commands.

Based on code by Kevin Fang.
Signed-off-by: Kalle Valo <kvalo@qca.qualcomm.com>