3 * Copyright (c) 2009, Microsoft Corporation.
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms and conditions of the GNU General Public License,
7 * version 2, as published by the Free Software Foundation.
9 * This program is distributed in the hope it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
14 * You should have received a copy of the GNU General Public License along with
15 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
16 * Place - Suite 330, Boston, MA 02111-1307 USA.
19 * Haiyang Zhang <haiyangz@microsoft.com>
20 * Hank Janssen <hjanssen@microsoft.com>
23 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
25 #include <linux/kernel.h>
26 #include <linux/sched.h>
27 #include <linux/wait.h>
29 #include <linux/slab.h>
30 #include <linux/vmalloc.h>
33 #include "hyperv_vmbus.h"
36 struct vmbus_connection vmbus_connection
= {
37 .conn_state
= DISCONNECTED
,
38 .next_gpadl_handle
= ATOMIC_INIT(0xE1E10),
42 * vmbus_connect - Sends a connect request on the partition service connection
44 int vmbus_connect(void)
48 struct vmbus_channel_msginfo
*msginfo
= NULL
;
49 struct vmbus_channel_initiate_contact
*msg
;
52 /* Make sure we are not connecting or connected */
53 if (vmbus_connection
.conn_state
!= DISCONNECTED
)
56 /* Initialize the vmbus connection */
57 vmbus_connection
.conn_state
= CONNECTING
;
58 vmbus_connection
.work_queue
= create_workqueue("hv_vmbus_con");
59 if (!vmbus_connection
.work_queue
) {
64 INIT_LIST_HEAD(&vmbus_connection
.chn_msg_list
);
65 spin_lock_init(&vmbus_connection
.channelmsg_lock
);
67 INIT_LIST_HEAD(&vmbus_connection
.chn_list
);
68 spin_lock_init(&vmbus_connection
.channel_lock
);
71 * Setup the vmbus event connection for channel interrupt
74 vmbus_connection
.int_page
=
75 (void *)__get_free_pages(GFP_KERNEL
|__GFP_ZERO
, 0);
76 if (vmbus_connection
.int_page
== NULL
) {
81 vmbus_connection
.recv_int_page
= vmbus_connection
.int_page
;
82 vmbus_connection
.send_int_page
=
83 (void *)((unsigned long)vmbus_connection
.int_page
+
87 * Setup the monitor notification facility. The 1st page for
88 * parent->child and the 2nd page for child->parent
90 vmbus_connection
.monitor_pages
=
91 (void *)__get_free_pages((GFP_KERNEL
|__GFP_ZERO
), 1);
92 if (vmbus_connection
.monitor_pages
== NULL
) {
97 msginfo
= kzalloc(sizeof(*msginfo
) +
98 sizeof(struct vmbus_channel_initiate_contact
),
100 if (msginfo
== NULL
) {
105 init_completion(&msginfo
->waitevent
);
107 msg
= (struct vmbus_channel_initiate_contact
*)msginfo
->msg
;
109 msg
->header
.msgtype
= CHANNELMSG_INITIATE_CONTACT
;
110 msg
->vmbus_version_requested
= VMBUS_REVISION_NUMBER
;
111 msg
->interrupt_page
= virt_to_phys(vmbus_connection
.int_page
);
112 msg
->monitor_page1
= virt_to_phys(vmbus_connection
.monitor_pages
);
113 msg
->monitor_page2
= virt_to_phys(
114 (void *)((unsigned long)vmbus_connection
.monitor_pages
+
118 * Add to list before we send the request since we may
119 * receive the response before returning from this routine
121 spin_lock_irqsave(&vmbus_connection
.channelmsg_lock
, flags
);
122 list_add_tail(&msginfo
->msglistentry
,
123 &vmbus_connection
.chn_msg_list
);
125 spin_unlock_irqrestore(&vmbus_connection
.channelmsg_lock
, flags
);
127 ret
= vmbus_post_msg(msg
,
128 sizeof(struct vmbus_channel_initiate_contact
));
130 spin_lock_irqsave(&vmbus_connection
.channelmsg_lock
, flags
);
131 list_del(&msginfo
->msglistentry
);
132 spin_unlock_irqrestore(&vmbus_connection
.channelmsg_lock
,
137 /* Wait for the connection response */
138 t
= wait_for_completion_timeout(&msginfo
->waitevent
, HZ
);
140 spin_lock_irqsave(&vmbus_connection
.channelmsg_lock
,
142 list_del(&msginfo
->msglistentry
);
143 spin_unlock_irqrestore(&vmbus_connection
.channelmsg_lock
,
149 spin_lock_irqsave(&vmbus_connection
.channelmsg_lock
, flags
);
150 list_del(&msginfo
->msglistentry
);
151 spin_unlock_irqrestore(&vmbus_connection
.channelmsg_lock
, flags
);
153 /* Check if successful */
154 if (msginfo
->response
.version_response
.version_supported
) {
155 vmbus_connection
.conn_state
= CONNECTED
;
157 pr_err("Unable to connect, "
158 "Version %d not supported by Hyper-V\n",
159 VMBUS_REVISION_NUMBER
);
168 vmbus_connection
.conn_state
= DISCONNECTED
;
170 if (vmbus_connection
.work_queue
)
171 destroy_workqueue(vmbus_connection
.work_queue
);
173 if (vmbus_connection
.int_page
) {
174 free_pages((unsigned long)vmbus_connection
.int_page
, 0);
175 vmbus_connection
.int_page
= NULL
;
178 if (vmbus_connection
.monitor_pages
) {
179 free_pages((unsigned long)vmbus_connection
.monitor_pages
, 1);
180 vmbus_connection
.monitor_pages
= NULL
;
190 * Sends a disconnect request on the partition service connection
192 int vmbus_disconnect(void)
195 struct vmbus_channel_message_header
*msg
;
197 /* Make sure we are connected */
198 if (vmbus_connection
.conn_state
!= CONNECTED
)
201 msg
= kzalloc(sizeof(struct vmbus_channel_message_header
), GFP_KERNEL
);
205 msg
->msgtype
= CHANNELMSG_UNLOAD
;
207 ret
= vmbus_post_msg(msg
,
208 sizeof(struct vmbus_channel_message_header
));
212 free_pages((unsigned long)vmbus_connection
.int_page
, 0);
213 free_pages((unsigned long)vmbus_connection
.monitor_pages
, 1);
215 /* TODO: iterate thru the msg list and free up */
216 destroy_workqueue(vmbus_connection
.work_queue
);
218 vmbus_connection
.conn_state
= DISCONNECTED
;
220 pr_info("hv_vmbus disconnected\n");
228 * relid2channel - Get the channel object given its
229 * child relative id (ie channel id)
231 struct vmbus_channel
*relid2channel(u32 relid
)
233 struct vmbus_channel
*channel
;
234 struct vmbus_channel
*found_channel
= NULL
;
237 spin_lock_irqsave(&vmbus_connection
.channel_lock
, flags
);
238 list_for_each_entry(channel
, &vmbus_connection
.chn_list
, listentry
) {
239 if (channel
->offermsg
.child_relid
== relid
) {
240 found_channel
= channel
;
244 spin_unlock_irqrestore(&vmbus_connection
.channel_lock
, flags
);
246 return found_channel
;
250 * process_chn_event - Process a channel event notification
252 static void process_chn_event(u32 relid
)
254 struct vmbus_channel
*channel
;
256 /* ASSERT(relId > 0); */
259 * Find the channel based on this relid and invokes the
260 * channel callback to process the event
262 channel
= relid2channel(relid
);
265 vmbus_onchannel_event(channel
);
267 pr_err("channel not found for relid - %u\n", relid
);
272 * vmbus_on_event - Handler for events
274 void vmbus_on_event(unsigned long data
)
277 u32 maxdword
= MAX_NUM_CHANNELS_SUPPORTED
>> 5;
280 u32
*recv_int_page
= vmbus_connection
.recv_int_page
;
285 for (dword
= 0; dword
< maxdword
; dword
++) {
286 if (!recv_int_page
[dword
])
288 for (bit
= 0; bit
< 32; bit
++) {
289 if (sync_test_and_clear_bit(bit
, (unsigned long *)&recv_int_page
[dword
])) {
290 relid
= (dword
<< 5) + bit
;
294 * Special case - vmbus
295 * channel protocol msg
299 process_chn_event(relid
);
306 * vmbus_post_msg - Send a msg on the vmbus's message connection
308 int vmbus_post_msg(void *buffer
, size_t buflen
)
310 union hv_connection_id conn_id
;
313 conn_id
.u
.id
= VMBUS_MESSAGE_CONNECTION_ID
;
314 return hv_post_message(conn_id
, 1, buffer
, buflen
);
318 * vmbus_set_event - Send an event notification to the parent
320 int vmbus_set_event(u32 child_relid
)
322 /* Each u32 represents 32 channels */
323 sync_set_bit(child_relid
& 31,
324 (unsigned long *)vmbus_connection
.send_int_page
+
327 return hv_signal_event();