/*-
 * Copyright (c) 2016 Microsoft Corp.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice unmodified, this list of conditions, and the following
 *    disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * $FreeBSD: head/sys/dev/hyperv/utilities/vmbus_icreg.h 305281 2016-09-02 06:23:28Z sephe $
 */

#ifndef _DEV_PV_HYPERVIC_H_
#define _DEV_PV_HYPERVIC_H_

#define VMBUS_IC_BUFRINGSIZE		(4 * PAGE_SIZE)

#define VMBUS_ICMSG_TYPE_NEGOTIATE	0
#define VMBUS_ICMSG_TYPE_HEARTBEAT	1
#define VMBUS_ICMSG_TYPE_KVP		2
#define VMBUS_ICMSG_TYPE_SHUTDOWN	3
#define VMBUS_ICMSG_TYPE_TIMESYNC	4
#define VMBUS_ICMSG_TYPE_VSS		5

#define VMBUS_ICMSG_STATUS_OK		0x00000000
#define VMBUS_ICMSG_STATUS_FAIL		0x80004005

#define VMBUS_ICMSG_FLAG_TRANSACTION	1
#define VMBUS_ICMSG_FLAG_REQUEST	2
#define VMBUS_ICMSG_FLAG_RESPONSE	4

#define VMBUS_IC_VERSION(major, minor)	((major) | (((uint32_t)(minor)) << 16))
#define VMBUS_ICVER_MAJOR(ver)		((ver) & 0xffff)
#define VMBUS_ICVER_MINOR(ver)		(((ver) & 0xffff0000) >> 16)

struct vmbus_pipe_hdr {
	uint32_t		ph_flags;
	uint32_t		ph_msgsz;
} __packed;

struct vmbus_icmsg_hdr {
	struct vmbus_pipe_hdr	ic_pipe;
	uint32_t		ic_fwver;	/* framework version */
	uint16_t		ic_type;
	uint32_t		ic_msgver;	/* message version */
	uint16_t		ic_dsize;	/* data size */
	uint32_t		ic_status;	/* VMBUS_ICMSG_STATUS_ */
	uint8_t			ic_tid;
	uint8_t			ic_flags;	/* VMBUS_ICMSG_FLAG_ */
	uint8_t			ic_rsvd[2];
} __packed;

/* VMBUS_ICMSG_TYPE_NEGOTIATE */
struct vmbus_icmsg_negotiate {
	struct vmbus_icmsg_hdr	ic_hdr;
	uint16_t		ic_fwver_cnt;
	uint16_t		ic_msgver_cnt;
	uint32_t		ic_rsvd;
	/*
	 * This version array contains two set of supported
	 * versions:
	 * - The first set consists of #ic_fwver_cnt supported framework
	 *   versions.
	 * - The second set consists of #ic_msgver_cnt supported message
	 *   versions.
	 */
	uint32_t		ic_ver[0];
} __packed;

/* VMBUS_ICMSG_TYPE_HEARTBEAT */
struct vmbus_icmsg_heartbeat {
	struct vmbus_icmsg_hdr	ic_hdr;
	uint64_t		ic_seq;
	uint32_t		ic_rsvd[8];
} __packed;

/* VMBUS_ICMSG_TYPE_SHUTDOWN */
struct vmbus_icmsg_shutdown {
	struct vmbus_icmsg_hdr	ic_hdr;
	uint32_t		ic_code;
	uint32_t		ic_timeo;
	uint32_t 		ic_haltflags;
	uint8_t			ic_msg[2048];
} __packed;

/* VMBUS_ICMSG_TYPE_TIMESYNC */
struct vmbus_icmsg_timesync {
	struct vmbus_icmsg_hdr	ic_hdr;
	uint64_t		ic_hvtime;
	uint64_t		ic_vmtime;
	uint64_t		ic_rtt;
	uint8_t			ic_tsflags;	/* VMBUS_ICMSG_TS_FLAG_ */
} __packed;

#define VMBUS_ICMSG_TS_FLAG_SYNC	0x01
#define VMBUS_ICMSG_TS_FLAG_SAMPLE	0x02

/* Registry value types */
#define HV_KVP_REG_SZ			1
#define HV_KVP_REG_U32			4
#define HV_KVP_REG_U64			8

/* Hyper-V status codes */
#define HV_KVP_S_OK			0x00000000
#define HV_KVP_E_FAIL			0x80004005
#define HV_KVP_S_CONT			0x80070103

#define HV_KVP_MAX_VAL_SIZE		2048
#define HV_KVP_MAX_KEY_SIZE		512

enum hv_kvp_op {
	HV_KVP_OP_GET = 0,
	HV_KVP_OP_SET,
	HV_KVP_OP_DELETE,
	HV_KVP_OP_ENUMERATE,
	HV_KVP_OP_GET_IP_INFO,
	HV_KVP_OP_SET_IP_INFO,
	HV_KVP_OP_COUNT
};

enum hv_kvp_pool {
	HV_KVP_POOL_EXTERNAL = 0,
	HV_KVP_POOL_GUEST,
	HV_KVP_POOL_AUTO,
	HV_KVP_POOL_AUTO_EXTERNAL,
	HV_KVP_POOL_COUNT
};

union hv_kvp_hdr {
	struct {
		uint8_t		kvu_op;
		uint8_t		kvu_pool;
		uint16_t	kvu_pad;
	} req;
	struct {
		uint32_t	kvu_err;
	} rsp;
#define kvh_op			req.kvu_op
#define kvh_pool		req.kvu_pool
#define kvh_err			rsp.kvu_err
} __packed;

struct hv_kvp_msg_val {
	uint32_t		kvm_valtype;
	uint32_t		kvm_keylen;
	uint32_t		kvm_vallen;
	uint8_t			kvm_key[HV_KVP_MAX_KEY_SIZE];
	uint8_t			kvm_val[HV_KVP_MAX_VAL_SIZE];
} __packed;

struct hv_kvp_msg_enum {
	uint32_t		kvm_index;
	uint32_t		kvm_valtype;
	uint32_t		kvm_keylen;
	uint32_t		kvm_vallen;
	uint8_t			kvm_key[HV_KVP_MAX_KEY_SIZE];
	uint8_t			kvm_val[HV_KVP_MAX_VAL_SIZE];
} __packed;

struct hv_kvp_msg_del {
	uint32_t		kvm_keylen;
	uint8_t			kvm_key[HV_KVP_MAX_KEY_SIZE];
} __packed;

#define ADDR_FAMILY_NONE	0x00
#define ADDR_FAMILY_IPV4	0x01
#define ADDR_FAMILY_IPV6	0x02

#define MAX_MAC_ADDR_SIZE	256
#define MAX_IP_ADDR_SIZE	2048
#define MAX_GATEWAY_SIZE	1024

struct hv_kvp_msg_addr {
	uint8_t			kvm_mac[MAX_MAC_ADDR_SIZE];
	uint8_t			kvm_family;
	uint8_t			kvm_dhcp;
	uint8_t			kvm_addr[MAX_IP_ADDR_SIZE];
	uint8_t			kvm_netmask[MAX_IP_ADDR_SIZE];
	uint8_t			kvm_gateway[MAX_GATEWAY_SIZE];
	uint8_t			kvm_dns[MAX_IP_ADDR_SIZE];
} __packed;

union hv_kvp_msg {
	struct hv_kvp_msg_val	kvm_val;
	struct hv_kvp_msg_enum	kvm_enum;
	struct hv_kvp_msg_del	kvm_del;
};

struct vmbus_icmsg_kvp {
	struct vmbus_icmsg_hdr	ic_hdr;
	union hv_kvp_hdr	ic_kvh;
	union hv_kvp_msg	ic_kvm;
} __packed;

struct vmbus_icmsg_kvp_addr {
	struct vmbus_icmsg_hdr	ic_hdr;
	struct {
		struct {
			uint8_t	kvu_op;
			uint8_t	kvu_pool;
		} req;
	}			ic_kvh;
	struct hv_kvp_msg_addr	ic_kvm;
} __packed;

#endif	/* _DEV_PV_HYPERVIC_H_ */