[mirror_ubuntu-zesty-kernel.git] / drivers / hv / hv_fcopy.c

/*
 * An implementation of file copy service.
 *
 * Copyright (C) 2014, Microsoft, Inc.
 *
 * Author : K. Y. Srinivasan <ksrinivasan@novell.com>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 as published
 * by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or
 * NON INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 */

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

#include <linux/nls.h>
#include <linux/workqueue.h>
#include <linux/hyperv.h>
#include <linux/sched.h>

#include "hyperv_vmbus.h"
#include "hv_utils_transport.h"

#define WIN8_SRV_MAJOR		1
#define WIN8_SRV_MINOR		1
#define WIN8_SRV_VERSION	(WIN8_SRV_MAJOR << 16 | WIN8_SRV_MINOR)

#define FCOPY_VER_COUNT 1
static const int fcopy_versions[] = {
	WIN8_SRV_VERSION
};

#define FW_VER_COUNT 1
static const int fw_versions[] = {
	UTIL_FW_VERSION
};

/*
 * Global state maintained for transaction that is being processed.
 * For a class of integration services, including the "file copy service",
 * the specified protocol is a "request/response" protocol which means that
 * there can only be single outstanding transaction from the host at any
 * given point in time. We use this to simplify memory management in this
 * driver - we cache and process only one message at a time.
 *
 * While the request/response protocol is guaranteed by the host, we further
 * ensure this by serializing packet processing in this driver - we do not
 * read additional packets from the VMBUs until the current packet is fully
 * handled.
 */

static struct {
	int state;   /* hvutil_device_state */
	int recv_len; /* number of bytes received. */
	struct hv_fcopy_hdr  *fcopy_msg; /* current message */
	struct vmbus_channel *recv_channel; /* chn we got the request */
	u64 recv_req_id; /* request ID. */
} fcopy_transaction;

static void fcopy_respond_to_host(int error);
static void fcopy_send_data(struct work_struct *dummy);
static void fcopy_timeout_func(struct work_struct *dummy);
static DECLARE_DELAYED_WORK(fcopy_timeout_work, fcopy_timeout_func);
static DECLARE_WORK(fcopy_send_work, fcopy_send_data);
static const char fcopy_devname[] = "vmbus/hv_fcopy";
static u8 *recv_buffer;
static struct hvutil_transport *hvt;
static struct completion release_event;
/*
 * This state maintains the version number registered by the daemon.
 */
static int dm_reg_value;

static void fcopy_poll_wrapper(void *channel)
{
	/* Transaction is finished, reset the state here to avoid races. */
	fcopy_transaction.state = HVUTIL_READY;
	hv_fcopy_onchannelcallback(channel);
}

static void fcopy_timeout_func(struct work_struct *dummy)
{
	/*
	 * If the timer fires, the user-mode component has not responded;
	 * process the pending transaction.
	 */
	fcopy_respond_to_host(HV_E_FAIL);
	hv_poll_channel(fcopy_transaction.recv_channel, fcopy_poll_wrapper);
}

static void fcopy_register_done(void)
{
	pr_debug("FCP: userspace daemon registered\n");
	hv_poll_channel(fcopy_transaction.recv_channel, fcopy_poll_wrapper);
}

static int fcopy_handle_handshake(u32 version)
{
	u32 our_ver = FCOPY_CURRENT_VERSION;

	switch (version) {
	case FCOPY_VERSION_0:
		/* Daemon doesn't expect us to reply */
		dm_reg_value = version;
		break;
	case FCOPY_VERSION_1:
		/* Daemon expects us to reply with our own version */
		if (hvutil_transport_send(hvt, &our_ver, sizeof(our_ver),
		    fcopy_register_done))
			return -EFAULT;
		dm_reg_value = version;
		break;
	default:
		/*
		 * For now we will fail the registration.
		 * If and when we have multiple versions to
		 * deal with, we will be backward compatible.
		 * We will add this code when needed.
		 */
		return -EINVAL;
	}
	pr_debug("FCP: userspace daemon ver. %d connected\n", version);
	return 0;
}

static void fcopy_send_data(struct work_struct *dummy)
{
	struct hv_start_fcopy *smsg_out = NULL;
	int operation = fcopy_transaction.fcopy_msg->operation;
	struct hv_start_fcopy *smsg_in;
	void *out_src;
	int rc, out_len;

	/*
	 * The  strings sent from the host are encoded in
	 * in utf16; convert it to utf8 strings.
	 * The host assures us that the utf16 strings will not exceed
	 * the max lengths specified. We will however, reserve room
	 * for the string terminating character - in the utf16s_utf8s()
	 * function we limit the size of the buffer where the converted
	 * string is placed to W_MAX_PATH -1 to guarantee
	 * that the strings can be properly terminated!
	 */

	switch (operation) {
	case START_FILE_COPY:
		out_len = sizeof(struct hv_start_fcopy);
		smsg_out = kzalloc(sizeof(*smsg_out), GFP_KERNEL);
		if (!smsg_out)
			return;

		smsg_out->hdr.operation = operation;
		smsg_in = (struct hv_start_fcopy *)fcopy_transaction.fcopy_msg;

		utf16s_to_utf8s((wchar_t *)smsg_in->file_name, W_MAX_PATH,
				UTF16_LITTLE_ENDIAN,
				(__u8 *)&smsg_out->file_name, W_MAX_PATH - 1);

		utf16s_to_utf8s((wchar_t *)smsg_in->path_name, W_MAX_PATH,
				UTF16_LITTLE_ENDIAN,
				(__u8 *)&smsg_out->path_name, W_MAX_PATH - 1);

		smsg_out->copy_flags = smsg_in->copy_flags;
		smsg_out->file_size = smsg_in->file_size;
		out_src = smsg_out;
		break;

	default:
		out_src = fcopy_transaction.fcopy_msg;
		out_len = fcopy_transaction.recv_len;
		break;
	}

	fcopy_transaction.state = HVUTIL_USERSPACE_REQ;
	rc = hvutil_transport_send(hvt, out_src, out_len, NULL);
	if (rc) {
		pr_debug("FCP: failed to communicate to the daemon: %d\n", rc);
		if (cancel_delayed_work_sync(&fcopy_timeout_work)) {
			fcopy_respond_to_host(HV_E_FAIL);
			fcopy_transaction.state = HVUTIL_READY;
		}
	}
	kfree(smsg_out);

	return;
}

/*
 * Send a response back to the host.
 */

static void
fcopy_respond_to_host(int error)
{
	struct icmsg_hdr *icmsghdr;
	u32 buf_len;
	struct vmbus_channel *channel;
	u64 req_id;

	/*
	 * Copy the global state for completing the transaction. Note that
	 * only one transaction can be active at a time. This is guaranteed
	 * by the file copy protocol implemented by the host. Furthermore,
	 * the "transaction active" state we maintain ensures that there can
	 * only be one active transaction at a time.
	 */

	buf_len = fcopy_transaction.recv_len;
	channel = fcopy_transaction.recv_channel;
	req_id = fcopy_transaction.recv_req_id;

	icmsghdr = (struct icmsg_hdr *)
			&recv_buffer[sizeof(struct vmbuspipe_hdr)];

	if (channel->onchannel_callback == NULL)
		/*
		 * We have raced with util driver being unloaded;
		 * silently return.
		 */
		return;

	icmsghdr->status = error;
	icmsghdr->icflags = ICMSGHDRFLAG_TRANSACTION | ICMSGHDRFLAG_RESPONSE;
	vmbus_sendpacket(channel, recv_buffer, buf_len, req_id,
				VM_PKT_DATA_INBAND, 0);
}

void hv_fcopy_onchannelcallback(void *context)
{
	struct vmbus_channel *channel = context;
	u32 recvlen;
	u64 requestid;
	struct hv_fcopy_hdr *fcopy_msg;
	struct icmsg_hdr *icmsghdr;
	int fcopy_srv_version;

	if (fcopy_transaction.state > HVUTIL_READY)
		return;

	vmbus_recvpacket(channel, recv_buffer, PAGE_SIZE * 2, &recvlen,
			 &requestid);
	if (recvlen <= 0)
		return;

	icmsghdr = (struct icmsg_hdr *)&recv_buffer[
			sizeof(struct vmbuspipe_hdr)];
	if (icmsghdr->icmsgtype == ICMSGTYPE_NEGOTIATE) {
		vmbus_prep_negotiate_resp(icmsghdr, recv_buffer,
				fw_versions, FW_VER_COUNT,
				fcopy_versions, FCOPY_VER_COUNT,
				NULL, &fcopy_srv_version);
	} else {
		fcopy_msg = (struct hv_fcopy_hdr *)&recv_buffer[
				sizeof(struct vmbuspipe_hdr) +
				sizeof(struct icmsg_hdr)];

		/*
		 * Stash away this global state for completing the
		 * transaction; note transactions are serialized.
		 */

		fcopy_transaction.recv_len = recvlen;
		fcopy_transaction.recv_req_id = requestid;
		fcopy_transaction.fcopy_msg = fcopy_msg;

		if (fcopy_transaction.state < HVUTIL_READY) {
			/* Userspace is not registered yet */
			fcopy_respond_to_host(HV_E_FAIL);
			return;
		}
		fcopy_transaction.state = HVUTIL_HOSTMSG_RECEIVED;

		/*
		 * Send the information to the user-level daemon.
		 */
		schedule_work(&fcopy_send_work);
		schedule_delayed_work(&fcopy_timeout_work,
				      HV_UTIL_TIMEOUT * HZ);
		return;
	}
	icmsghdr->icflags = ICMSGHDRFLAG_TRANSACTION | ICMSGHDRFLAG_RESPONSE;
	vmbus_sendpacket(channel, recv_buffer, recvlen, requestid,
			VM_PKT_DATA_INBAND, 0);
}

/* Callback when data is received from userspace */
static int fcopy_on_msg(void *msg, int len)
{
	int *val = (int *)msg;

	if (len != sizeof(int))
		return -EINVAL;

	if (fcopy_transaction.state == HVUTIL_DEVICE_INIT)
		return fcopy_handle_handshake(*val);

	if (fcopy_transaction.state != HVUTIL_USERSPACE_REQ)
		return -EINVAL;

	/*
	 * Complete the transaction by forwarding the result
	 * to the host. But first, cancel the timeout.
	 */
	if (cancel_delayed_work_sync(&fcopy_timeout_work)) {
		fcopy_transaction.state = HVUTIL_USERSPACE_RECV;
		fcopy_respond_to_host(*val);
		hv_poll_channel(fcopy_transaction.recv_channel,
				fcopy_poll_wrapper);
	}

	return 0;
}

static void fcopy_on_reset(void)
{
	/*
	 * The daemon has exited; reset the state.
	 */
	fcopy_transaction.state = HVUTIL_DEVICE_INIT;

	if (cancel_delayed_work_sync(&fcopy_timeout_work))
		fcopy_respond_to_host(HV_E_FAIL);
	complete(&release_event);
}

int hv_fcopy_init(struct hv_util_service *srv)
{
	recv_buffer = srv->recv_buffer;
	fcopy_transaction.recv_channel = srv->channel;

	init_completion(&release_event);
	/*
	 * When this driver loads, the user level daemon that
	 * processes the host requests may not yet be running.
	 * Defer processing channel callbacks until the daemon
	 * has registered.
	 */
	fcopy_transaction.state = HVUTIL_DEVICE_INIT;

	hvt = hvutil_transport_init(fcopy_devname, 0, 0,
				    fcopy_on_msg, fcopy_on_reset);
	if (!hvt)
		return -EFAULT;

	return 0;
}

void hv_fcopy_deinit(void)
{
	fcopy_transaction.state = HVUTIL_DEVICE_DYING;
	cancel_delayed_work_sync(&fcopy_timeout_work);
	hvutil_transport_destroy(hvt);
	wait_for_completion(&release_event);
}
Commit	Line	Data
01325476 S	1	/*
	2	* An implementation of file copy service.
	3	*
	4	* Copyright (C) 2014, Microsoft, Inc.
	5	*
	6	* Author : K. Y. Srinivasan <ksrinivasan@novell.com>
	7	*
	8	* This program is free software; you can redistribute it and/or modify it
	9	* under the terms of the GNU General Public License version 2 as published
	10	* by the Free Software Foundation.
	11	*
	12	* This program is distributed in the hope that it will be useful, but
	13	* WITHOUT ANY WARRANTY; without even the implied warranty of
	14	* MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or
	15	* NON INFRINGEMENT. See the GNU General Public License for more
	16	* details.
	17	*
	18	*/
	19
	20	#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
	21
01325476 S	22	#include <linux/nls.h>
01325476 S	23	#include <linux/workqueue.h>
01325476 S	24	#include <linux/hyperv.h>
01325476 S	25	#include <linux/sched.h>
01325476 S	26
01325476 S	27	#include "hyperv_vmbus.h"
c7e490fc	28	#include "hv_utils_transport.h"
01325476 S	29
	30	#define WIN8_SRV_MAJOR 1
	31	#define WIN8_SRV_MINOR 1
	32	#define WIN8_SRV_VERSION (WIN8_SRV_MAJOR << 16 \| WIN8_SRV_MINOR)
	33
06b734ab AN	34	#define FCOPY_VER_COUNT 1
	35	static const int fcopy_versions[] = {
	36	WIN8_SRV_VERSION
	37	};
	38
	39	#define FW_VER_COUNT 1
	40	static const int fw_versions[] = {
	41	UTIL_FW_VERSION
	42	};
	43
01325476 S	44	/*
	45	* Global state maintained for transaction that is being processed.
	46	* For a class of integration services, including the "file copy service",
	47	* the specified protocol is a "request/response" protocol which means that
	48	* there can only be single outstanding transaction from the host at any
	49	* given point in time. We use this to simplify memory management in this
	50	* driver - we cache and process only one message at a time.
	51	*
	52	* While the request/response protocol is guaranteed by the host, we further
	53	* ensure this by serializing packet processing in this driver - we do not
	54	* read additional packets from the VMBUs until the current packet is fully
	55	* handled.
01325476 S	56	*/
	57
	58	static struct {
4c93cccc	59	int state; /* hvutil_device_state */
01325476 S	60	int recv_len; /* number of bytes received. */
01325476 S	61	struct hv_fcopy_hdr fcopy_msg; / current message */
01325476 S	62	struct vmbus_channel recv_channel; / chn we got the request */
01325476 S	63	u64 recv_req_id; /* request ID. */
01325476 S	64	} fcopy_transaction;
01325476 S	65
01325476	66	static void fcopy_respond_to_host(int error);
c7e490fc	67	static void fcopy_send_data(struct work_struct *dummy);
1d072339 VK	68	static void fcopy_timeout_func(struct work_struct *dummy);
1d072339 VK	69	static DECLARE_DELAYED_WORK(fcopy_timeout_work, fcopy_timeout_func);
c7e490fc VK	70	static DECLARE_WORK(fcopy_send_work, fcopy_send_data);
c7e490fc VK	71	static const char fcopy_devname[] = "vmbus/hv_fcopy";
01325476	72	static u8 *recv_buffer;
c7e490fc	73	static struct hvutil_transport *hvt;
1d4a207c	74	static struct completion release_event;
a4d1ee5b VK	75	/*
	76	* This state maintains the version number registered by the daemon.
	77	*/
	78	static int dm_reg_value;
01325476	79
3cace4a6 OH	80	static void fcopy_poll_wrapper(void *channel)
	81	{
	82	/* Transaction is finished, reset the state here to avoid races. */
	83	fcopy_transaction.state = HVUTIL_READY;
	84	hv_fcopy_onchannelcallback(channel);
	85	}
	86
1d072339	87	static void fcopy_timeout_func(struct work_struct *dummy)
01325476 S	88	{
	89	/*
	90	* If the timer fires, the user-mode component has not responded;
	91	* process the pending transaction.
	92	*/
	93	fcopy_respond_to_host(HV_E_FAIL);
3cace4a6	94	hv_poll_channel(fcopy_transaction.recv_channel, fcopy_poll_wrapper);
01325476 S	95	}
01325476 S	96
e0fa3e5e VK	97	static void fcopy_register_done(void)
	98	{
	99	pr_debug("FCP: userspace daemon registered\n");
	100	hv_poll_channel(fcopy_transaction.recv_channel, fcopy_poll_wrapper);
	101	}
	102
01325476 S	103	static int fcopy_handle_handshake(u32 version)
01325476 S	104	{
a4d1ee5b VK	105	u32 our_ver = FCOPY_CURRENT_VERSION;
a4d1ee5b VK	106
01325476	107	switch (version) {
a4d1ee5b VK	108	case FCOPY_VERSION_0:
	109	/* Daemon doesn't expect us to reply */
	110	dm_reg_value = version;
	111	break;
	112	case FCOPY_VERSION_1:
	113	/* Daemon expects us to reply with our own version */
e0fa3e5e VK	114	if (hvutil_transport_send(hvt, &our_ver, sizeof(our_ver),
e0fa3e5e VK	115	fcopy_register_done))
a4d1ee5b VK	116	return -EFAULT;
a4d1ee5b VK	117	dm_reg_value = version;
01325476 S	118	break;
	119	default:
	120	/*
	121	* For now we will fail the registration.
	122	* If and when we have multiple versions to
	123	* deal with, we will be backward compatible.
	124	* We will add this code when needed.
	125	*/
	126	return -EINVAL;
	127	}
e0fa3e5e	128	pr_debug("FCP: userspace daemon ver. %d connected\n", version);
01325476 S	129	return 0;
	130	}
	131
c7e490fc	132	static void fcopy_send_data(struct work_struct *dummy)
01325476	133	{
25ef06fe	134	struct hv_start_fcopy *smsg_out = NULL;
01325476 S	135	int operation = fcopy_transaction.fcopy_msg->operation;
01325476 S	136	struct hv_start_fcopy *smsg_in;
c7e490fc VK	137	void *out_src;
c7e490fc VK	138	int rc, out_len;
01325476 S	139
	140	/*
	141	* The strings sent from the host are encoded in
	142	* in utf16; convert it to utf8 strings.
	143	* The host assures us that the utf16 strings will not exceed
	144	* the max lengths specified. We will however, reserve room
	145	* for the string terminating character - in the utf16s_utf8s()
	146	* function we limit the size of the buffer where the converted
	147	* string is placed to W_MAX_PATH -1 to guarantee
	148	* that the strings can be properly terminated!
	149	*/
	150
	151	switch (operation) {
	152	case START_FILE_COPY:
c7e490fc	153	out_len = sizeof(struct hv_start_fcopy);
25ef06fe VK	154	smsg_out = kzalloc(sizeof(*smsg_out), GFP_KERNEL);
	155	if (!smsg_out)
	156	return;
	157
	158	smsg_out->hdr.operation = operation;
01325476 S	159	smsg_in = (struct hv_start_fcopy *)fcopy_transaction.fcopy_msg;
	160
	161	utf16s_to_utf8s((wchar_t *)smsg_in->file_name, W_MAX_PATH,
	162	UTF16_LITTLE_ENDIAN,
25ef06fe	163	(__u8 *)&smsg_out->file_name, W_MAX_PATH - 1);
01325476 S	164
	165	utf16s_to_utf8s((wchar_t *)smsg_in->path_name, W_MAX_PATH,
	166	UTF16_LITTLE_ENDIAN,
25ef06fe	167	(__u8 *)&smsg_out->path_name, W_MAX_PATH - 1);
01325476	168
25ef06fe VK	169	smsg_out->copy_flags = smsg_in->copy_flags;
	170	smsg_out->file_size = smsg_in->file_size;
	171	out_src = smsg_out;
01325476 S	172	break;
	173
	174	default:
c7e490fc VK	175	out_src = fcopy_transaction.fcopy_msg;
c7e490fc VK	176	out_len = fcopy_transaction.recv_len;
01325476 S	177	break;
01325476 S	178	}
c7e490fc VK	179
c7e490fc VK	180	fcopy_transaction.state = HVUTIL_USERSPACE_REQ;
e0fa3e5e	181	rc = hvutil_transport_send(hvt, out_src, out_len, NULL);
c7e490fc VK	182	if (rc) {
	183	pr_debug("FCP: failed to communicate to the daemon: %d\n", rc);
	184	if (cancel_delayed_work_sync(&fcopy_timeout_work)) {
	185	fcopy_respond_to_host(HV_E_FAIL);
	186	fcopy_transaction.state = HVUTIL_READY;
	187	}
	188	}
25ef06fe VK	189	kfree(smsg_out);
25ef06fe VK	190
01325476 S	191	return;
	192	}
	193
	194	/*
	195	* Send a response back to the host.
	196	*/
	197
	198	static void
	199	fcopy_respond_to_host(int error)
	200	{
	201	struct icmsg_hdr *icmsghdr;
	202	u32 buf_len;
	203	struct vmbus_channel *channel;
	204	u64 req_id;
	205
	206	/*
	207	* Copy the global state for completing the transaction. Note that
	208	* only one transaction can be active at a time. This is guaranteed
	209	* by the file copy protocol implemented by the host. Furthermore,
	210	* the "transaction active" state we maintain ensures that there can
	211	* only be one active transaction at a time.
	212	*/
	213
	214	buf_len = fcopy_transaction.recv_len;
	215	channel = fcopy_transaction.recv_channel;
	216	req_id = fcopy_transaction.recv_req_id;
	217
01325476 S	218	icmsghdr = (struct icmsg_hdr *)
	219	&recv_buffer[sizeof(struct vmbuspipe_hdr)];
	220
	221	if (channel->onchannel_callback == NULL)
	222	/*
	223	* We have raced with util driver being unloaded;
	224	* silently return.
	225	*/
	226	return;
	227
	228	icmsghdr->status = error;
	229	icmsghdr->icflags = ICMSGHDRFLAG_TRANSACTION \| ICMSGHDRFLAG_RESPONSE;
	230	vmbus_sendpacket(channel, recv_buffer, buf_len, req_id,
	231	VM_PKT_DATA_INBAND, 0);
	232	}
	233
	234	void hv_fcopy_onchannelcallback(void *context)
	235	{
	236	struct vmbus_channel *channel = context;
	237	u32 recvlen;
	238	u64 requestid;
	239	struct hv_fcopy_hdr *fcopy_msg;
	240	struct icmsg_hdr *icmsghdr;
01325476 S	241	int fcopy_srv_version;
01325476 S	242
3cace4a6	243	if (fcopy_transaction.state > HVUTIL_READY)
01325476	244	return;
01325476 S	245
	246	vmbus_recvpacket(channel, recv_buffer, PAGE_SIZE * 2, &recvlen,
	247	&requestid);
	248	if (recvlen <= 0)
	249	return;
	250
	251	icmsghdr = (struct icmsg_hdr *)&recv_buffer[
	252	sizeof(struct vmbuspipe_hdr)];
	253	if (icmsghdr->icmsgtype == ICMSGTYPE_NEGOTIATE) {
06b734ab AN	254	vmbus_prep_negotiate_resp(icmsghdr, recv_buffer,
	255	fw_versions, FW_VER_COUNT,
	256	fcopy_versions, FCOPY_VER_COUNT,
	257	NULL, &fcopy_srv_version);
01325476 S	258	} else {
	259	fcopy_msg = (struct hv_fcopy_hdr *)&recv_buffer[
	260	sizeof(struct vmbuspipe_hdr) +
	261	sizeof(struct icmsg_hdr)];
	262
	263	/*
	264	* Stash away this global state for completing the
	265	* transaction; note transactions are serialized.
	266	*/
	267
01325476	268	fcopy_transaction.recv_len = recvlen;
01325476 S	269	fcopy_transaction.recv_req_id = requestid;
	270	fcopy_transaction.fcopy_msg = fcopy_msg;
	271
4c93cccc VK	272	if (fcopy_transaction.state < HVUTIL_READY) {
	273	/* Userspace is not registered yet */
	274	fcopy_respond_to_host(HV_E_FAIL);
	275	return;
	276	}
	277	fcopy_transaction.state = HVUTIL_HOSTMSG_RECEIVED;
	278
01325476 S	279	/*
	280	* Send the information to the user-level daemon.
	281	*/
c7e490fc	282	schedule_work(&fcopy_send_work);
c0b200cf S	283	schedule_delayed_work(&fcopy_timeout_work,
c0b200cf S	284	HV_UTIL_TIMEOUT * HZ);
01325476 S	285	return;
	286	}
	287	icmsghdr->icflags = ICMSGHDRFLAG_TRANSACTION \| ICMSGHDRFLAG_RESPONSE;
	288	vmbus_sendpacket(channel, recv_buffer, recvlen, requestid,
	289	VM_PKT_DATA_INBAND, 0);
	290	}
	291
c7e490fc VK	292	/* Callback when data is received from userspace */
c7e490fc VK	293	static int fcopy_on_msg(void *msg, int len)
01325476	294	{
c7e490fc	295	int val = (int )msg;
01325476	296
c7e490fc	297	if (len != sizeof(int))
01325476 S	298	return -EINVAL;
01325476 S	299
c7e490fc VK	300	if (fcopy_transaction.state == HVUTIL_DEVICE_INIT)
c7e490fc VK	301	return fcopy_handle_handshake(*val);
01325476	302
4c93cccc VK	303	if (fcopy_transaction.state != HVUTIL_USERSPACE_REQ)
	304	return -EINVAL;
	305
01325476 S	306	/*
	307	* Complete the transaction by forwarding the result
	308	* to the host. But first, cancel the timeout.
	309	*/
4c93cccc VK	310	if (cancel_delayed_work_sync(&fcopy_timeout_work)) {
4c93cccc VK	311	fcopy_transaction.state = HVUTIL_USERSPACE_RECV;
c7e490fc	312	fcopy_respond_to_host(*val);
3cace4a6 OH	313	hv_poll_channel(fcopy_transaction.recv_channel,
3cace4a6 OH	314	fcopy_poll_wrapper);
242f3122	315	}
01325476	316
01325476 S	317	return 0;
	318	}
	319
c7e490fc	320	static void fcopy_on_reset(void)
01325476 S	321	{
	322	/*
	323	* The daemon has exited; reset the state.
	324	*/
4c93cccc	325	fcopy_transaction.state = HVUTIL_DEVICE_INIT;
d9b16529	326
c7e490fc	327	if (cancel_delayed_work_sync(&fcopy_timeout_work))
d9b16529	328	fcopy_respond_to_host(HV_E_FAIL);
1d4a207c	329	complete(&release_event);
01325476 S	330	}
	331
	332	int hv_fcopy_init(struct hv_util_service *srv)
	333	{
	334	recv_buffer = srv->recv_buffer;
b9830d12	335	fcopy_transaction.recv_channel = srv->channel;
01325476	336
1d4a207c	337	init_completion(&release_event);
01325476 S	338	/*
	339	* When this driver loads, the user level daemon that
	340	* processes the host requests may not yet be running.
	341	* Defer processing channel callbacks until the daemon
	342	* has registered.
	343	*/
4c93cccc	344	fcopy_transaction.state = HVUTIL_DEVICE_INIT;
01325476	345
c7e490fc VK	346	hvt = hvutil_transport_init(fcopy_devname, 0, 0,
	347	fcopy_on_msg, fcopy_on_reset);
	348	if (!hvt)
	349	return -EFAULT;
	350
	351	return 0;
01325476 S	352	}
	353
	354	void hv_fcopy_deinit(void)
	355	{
4c93cccc	356	fcopy_transaction.state = HVUTIL_DEVICE_DYING;
1d072339	357	cancel_delayed_work_sync(&fcopy_timeout_work);
c7e490fc	358	hvutil_transport_destroy(hvt);
1d4a207c	359	wait_for_completion(&release_event);
01325476	360	}