1 files changed, 17 insertions, 0 deletions

diff --git a/drivers/hv/hyperv_vmbus.h b/drivers/hv/hyperv_vmbus.h
index c8e27e0fdc99..f40a5a935ab6 100644
--- a/drivers/hv/hyperv_vmbus.h
+++ b/drivers/hv/hyperv_vmbus.h
@@ -685,6 +685,23 @@ struct vmbus_msginfo {
 
 extern struct vmbus_connection vmbus_connection;
 
+enum vmbus_message_handler_type {
+        /* The related handler can sleep. */
+        VMHT_BLOCKING = 0,
+
+        /* The related handler must NOT sleep. */
+        VMHT_NON_BLOCKING = 1,
+};
+
+struct vmbus_channel_message_table_entry {
+        enum vmbus_channel_message_type message_type;
+        enum vmbus_message_handler_type handler_type;
+        void (*message_handler)(struct vmbus_channel_message_header *msg);
+};
+
+extern struct vmbus_channel_message_table_entry
+        channel_message_table[CHANNELMSG_COUNT];
+
 /* General vmbus interface */
 
 struct hv_device *vmbus_device_create(const uuid_le *type,