/* SPDX-License-Identifier: GPL-2.0-or-later
 * Copyright Red Hat
 * Author: Laurent Vivier <lvivier@redhat.com>
 *
 * vhost-user common UDP and TCP functions
 */

#ifndef VU_COMMON_H
#define VU_COMMON_H
#include <linux/virtio_net.h>

static inline void *vu_eth(void *base)
{
	return ((char *)base + sizeof(struct virtio_net_hdr_mrg_rxbuf));
}

static inline void *vu_ip(void *base)
{
	return (struct ethhdr *)vu_eth(base) + 1;
}

static inline void *vu_payloadv4(void *base)
{
	return (struct iphdr *)vu_ip(base) + 1;
}

static inline void *vu_payloadv6(void *base)
{
	return (struct ipv6hdr *)vu_ip(base) + 1;
}

void vu_init_elem(struct vu_virtq_element *elem, struct iovec *iov,
		  int elem_cnt);
int vu_collect_one_frame(struct vu_dev *vdev, struct vu_virtq *vq,
			 struct vu_virtq_element *elem, int max_elem,
			 size_t size, size_t hdrlen);
int vu_collect(struct vu_dev *vdev, struct vu_virtq *vq,
	       struct vu_virtq_element *elem, int max_elem,
	       size_t max_frame_size, size_t hdrlen, size_t size);
void vu_set_vnethdr(const struct vu_dev *vdev, struct iovec *iov,
                    int num_buffers, size_t hdrlen);
void vu_flush(const struct vu_dev *vdev, struct vu_virtq *vq,
	      struct vu_virtq_element *elem, int elem_cnt);
void vu_kick_cb(struct vu_dev *vdev, union epoll_ref ref,
		const struct timespec *now);
int vu_send_single(const struct ctx *c, const void *buf, size_t size);
#endif /* VU_COMMON_H */