Convert the code to use k_timeout_t when applicable. Signed-off-by: Jukka Rissanen <jukka.rissanen@linux.intel.com>