kvp_respond_to_host(NULL, HV_E_FAIL);
 }
 
-static void poll_channel(struct vmbus_channel *channel)
-{
-       if (channel->target_cpu != smp_processor_id())
-               smp_call_function_single(channel->target_cpu,
-                                        hv_kvp_onchannelcallback,
-                                        channel, true);
-       else
-               hv_kvp_onchannelcallback(channel);
-}
-
-
 static int kvp_handle_handshake(struct hv_kvp_msg *msg)
 {
        int ret = 1;
                pr_info("KVP: user-mode registering done.\n");
                kvp_register(dm_reg_value);
                kvp_transaction.active = false;
-               if (kvp_transaction.kvp_context)
-                       poll_channel(kvp_transaction.kvp_context);
+               hv_poll_channel(kvp_transaction.kvp_context,
+                               hv_kvp_onchannelcallback);
        }
        return ret;
 }
 
        vmbus_sendpacket(channel, recv_buffer, buf_len, req_id,
                                VM_PKT_DATA_INBAND, 0);
-       poll_channel(channel);
+       hv_poll_channel(channel, hv_kvp_onchannelcallback);
 }
 
 /*
 
 void hv_fcopy_deinit(void);
 void hv_fcopy_onchannelcallback(void *);
 
+static inline void hv_poll_channel(struct vmbus_channel *channel,
+                                  void (*cb)(void *))
+{
+       if (!channel)
+               return;
+
+       if (channel->target_cpu != smp_processor_id())
+               smp_call_function_single(channel->target_cpu,
+                                        cb, channel, true);
+       else
+               cb(channel);
+}
 
 #endif /* _HYPERV_VMBUS_H */