linux-imx

/*

 * common code for virtio vsock

 *

 * Copyright (C) 2013-2015 Red Hat, Inc.

 * Author: Asias He <asias@redhat.com>

 *         Stefan Hajnoczi <stefanha@redhat.com>

 *

 * This work is licensed under the terms of the GNU GPL, version 2.

 */

#include <linux/spinlock.h>

#include <linux/module.h>

#include <linux/sched/signal.h>

#include <linux/ctype.h>

#include <linux/list.h>

#include <linux/virtio.h>

#include <linux/virtio_ids.h>

#include <linux/virtio_config.h>

#include <linux/virtio_vsock.h>

#include <uapi/linux/vsockmon.h>

​

#include <net/sock.h>

#include <net/af_vsock.h>

​

#define CREATE_TRACE_POINTS

#include <trace/events/vsock_virtio_transport_common.h>

​

/* How long to wait for graceful shutdown of a connection */

#define VSOCK_CLOSE_TIMEOUT (8 * HZ)

​

static const struct virtio_transport *virtio_transport_get_ops(void)

{

    const struct vsock_transport *t = vsock_core_get_transport();

​

    return container_of(t, struct virtio_transport, transport);

}

​

static struct virtio_vsock_pkt *

virtio_transport_alloc_pkt(struct virtio_vsock_pkt_info *info,

               size_t len,

               u32 src_cid,

               u32 src_port,

               u32 dst_cid,

               u32 dst_port)

{

    struct virtio_vsock_pkt *pkt;

    int err;

​

    pkt = kzalloc(sizeof(*pkt), GFP_KERNEL);

    if (!pkt)

        return NULL;

​

    pkt->hdr.type       = cpu_to_le16(info->type);

    pkt->hdr.op     = cpu_to_le16(info->op);

    pkt->hdr.src_cid    = cpu_to_le64(src_cid);

    pkt->hdr.dst_cid    = cpu_to_le64(dst_cid);

    pkt->hdr.src_port   = cpu_to_le32(src_port);

    pkt->hdr.dst_port   = cpu_to_le32(dst_port);

    pkt->hdr.flags      = cpu_to_le32(info->flags);

    pkt->len        = len;

    pkt->hdr.len        = cpu_to_le32(len);

    pkt->reply      = info->reply;

    pkt->vsk        = info->vsk;

​

    if (info->msg && len > 0) {

        pkt->buf = kmalloc(len, GFP_KERNEL);

        if (!pkt->buf)

            goto out_pkt;

        err = memcpy_from_msg(pkt->buf, info->msg, len);

        if (err)

            goto out;

    }

​

    trace_virtio_transport_alloc_pkt(src_cid, src_port,

                     dst_cid, dst_port,

                     len,

                     info->type,

                     info->op,

                     info->flags);

​

    return pkt;

​

out:

    kfree(pkt->buf);

out_pkt:

    kfree(pkt);

    return NULL;

}

​

/* Packet capture */

static struct sk_buff *virtio_transport_build_skb(void *opaque)

{

    struct virtio_vsock_pkt *pkt = opaque;