添加rtthread相关代码
This commit is contained in:
778
riscv/rtthread/components/drivers/virtio/virtio_console.c
Executable file
778
riscv/rtthread/components/drivers/virtio/virtio_console.c
Executable file
@@ -0,0 +1,778 @@
|
||||
/*
|
||||
* Copyright (c) 2006-2023, RT-Thread Development Team
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Change Logs:
|
||||
* Date Author Notes
|
||||
* 2021-11-11 GuEe-GUI the first version
|
||||
*/
|
||||
|
||||
#include <rthw.h>
|
||||
#include <rtthread.h>
|
||||
#include <cpuport.h>
|
||||
|
||||
#ifdef RT_USING_VIRTIO_CONSOLE
|
||||
|
||||
#include <virtio_console.h>
|
||||
|
||||
struct port_device
|
||||
{
|
||||
struct rt_device parent;
|
||||
|
||||
rt_list_t node;
|
||||
rt_uint32_t port_id;
|
||||
rt_bool_t rx_notify;
|
||||
rt_bool_t need_destroy;
|
||||
|
||||
struct virtio_console_device *console;
|
||||
|
||||
struct virtq *queue_rx, *queue_tx;
|
||||
rt_uint32_t queue_rx_index, queue_tx_index;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
struct rt_spinlock spinlock_rx, spinlock_tx;
|
||||
#endif
|
||||
|
||||
struct rt_device_notify rx_notify_helper;
|
||||
|
||||
struct
|
||||
{
|
||||
char rx_char, tx_char;
|
||||
} info[VIRTIO_CONSOLE_QUEUE_SIZE];
|
||||
};
|
||||
|
||||
static void virtio_console_send_ctrl(struct virtio_console_device *virtio_console_dev,
|
||||
struct virtio_console_control *ctrl)
|
||||
{
|
||||
rt_uint16_t id;
|
||||
struct virtio_device *virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
struct virtq *queue_ctrl_tx;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_base_t level = rt_spin_lock_irqsave(&virtio_dev->spinlock);
|
||||
#endif
|
||||
|
||||
queue_ctrl_tx = &virtio_dev->queues[VIRTIO_CONSOLE_QUEUE_CTRL_TX];
|
||||
|
||||
id = queue_ctrl_tx->avail->idx % queue_ctrl_tx->num;
|
||||
|
||||
rt_memcpy(&virtio_console_dev->info[id].tx_ctrl, ctrl, sizeof(struct virtio_console_control));
|
||||
|
||||
virtio_free_desc(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX, id);
|
||||
|
||||
virtio_fill_desc(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX, id,
|
||||
virtio_console_dev->info[id].tx_ctrl_paddr, sizeof(struct virtio_console_control), 0, 0);
|
||||
|
||||
virtio_submit_chain(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX, id);
|
||||
|
||||
virtio_queue_notify(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX);
|
||||
|
||||
virtio_alloc_desc(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX);
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&virtio_dev->spinlock, level);
|
||||
#endif
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_port_init(rt_device_t dev);
|
||||
static rt_err_t virtio_console_port_open(rt_device_t dev, rt_uint16_t oflag);
|
||||
static rt_err_t virtio_console_port_close(rt_device_t dev);
|
||||
static rt_ssize_t virtio_console_port_read(rt_device_t dev, rt_off_t pos, void *buffer, rt_size_t size);
|
||||
static rt_ssize_t virtio_console_port_write(rt_device_t dev, rt_off_t pos, const void *buffer, rt_size_t size);
|
||||
static rt_err_t virtio_console_port_control(rt_device_t dev, int cmd, void *args);
|
||||
|
||||
#ifdef RT_USING_DEVICE_OPS
|
||||
const static struct rt_device_ops virtio_console_port_ops =
|
||||
{
|
||||
virtio_console_port_init,
|
||||
virtio_console_port_open,
|
||||
virtio_console_port_close,
|
||||
virtio_console_port_read,
|
||||
virtio_console_port_write,
|
||||
virtio_console_port_control
|
||||
};
|
||||
#endif
|
||||
|
||||
static rt_err_t virtio_console_port_create(struct virtio_console_device *virtio_console_dev)
|
||||
{
|
||||
rt_uint32_t port_id;
|
||||
char dev_name[RT_NAME_MAX];
|
||||
struct port_device *port_dev, *prev_port_dev = RT_NULL;
|
||||
struct virtio_device *virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
|
||||
if (virtio_console_dev->port_nr > 0 && !virtio_has_feature(virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
return -RT_ENOSYS;
|
||||
}
|
||||
|
||||
if (virtio_console_dev->port_nr >= virtio_console_dev->max_port_nr)
|
||||
{
|
||||
return -RT_EFULL;
|
||||
}
|
||||
|
||||
port_id = 0;
|
||||
|
||||
/* The port device list is always ordered, so just find next number for id */
|
||||
rt_list_for_each_entry(port_dev, &virtio_console_dev->port_head, node)
|
||||
{
|
||||
if (port_dev->port_id != port_id)
|
||||
{
|
||||
break;
|
||||
}
|
||||
++port_id;
|
||||
prev_port_dev = port_dev;
|
||||
}
|
||||
|
||||
port_dev = rt_malloc(sizeof(struct port_device));
|
||||
|
||||
if (port_dev == RT_NULL)
|
||||
{
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
|
||||
port_dev->parent.type = RT_Device_Class_Char;
|
||||
#ifdef RT_USING_DEVICE_OPS
|
||||
port_dev->parent.ops = &virtio_console_port_ops;
|
||||
#else
|
||||
port_dev->parent.init = virtio_console_port_init;
|
||||
port_dev->parent.open = virtio_console_port_open;
|
||||
port_dev->parent.close = virtio_console_port_close;
|
||||
port_dev->parent.read = virtio_console_port_read;
|
||||
port_dev->parent.write = virtio_console_port_write;
|
||||
port_dev->parent.control = virtio_console_port_control;
|
||||
#endif
|
||||
|
||||
port_dev->parent.rx_indicate = RT_NULL;
|
||||
port_dev->parent.tx_complete = RT_NULL;
|
||||
|
||||
rt_list_init(&port_dev->node);
|
||||
port_dev->port_id = port_id;
|
||||
port_dev->need_destroy = RT_FALSE;
|
||||
port_dev->rx_notify = RT_TRUE;
|
||||
port_dev->console = virtio_console_dev;
|
||||
port_dev->queue_rx_index = VIRTIO_CONSOLE_PORT_QUEUE_INDEX(port_dev->port_id, VIRTIO_CONSOLE_QUEUE_DATA_RX);
|
||||
port_dev->queue_tx_index = VIRTIO_CONSOLE_PORT_QUEUE_INDEX(port_dev->port_id, VIRTIO_CONSOLE_QUEUE_DATA_TX);
|
||||
port_dev->queue_rx = &virtio_dev->queues[port_dev->queue_rx_index];
|
||||
port_dev->queue_tx = &virtio_dev->queues[port_dev->queue_tx_index];
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_lock_init(&port_dev->spinlock_rx);
|
||||
rt_spin_lock_init(&port_dev->spinlock_tx);
|
||||
#endif
|
||||
|
||||
rt_snprintf(dev_name, RT_NAME_MAX, "vport%dp%d", virtio_console_dev->console_id, port_id);
|
||||
|
||||
if (rt_device_register((rt_device_t)port_dev, dev_name, RT_DEVICE_FLAG_RDWR | RT_DEVICE_FLAG_INT_RX) != RT_EOK)
|
||||
{
|
||||
rt_free(port_dev);
|
||||
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
if (prev_port_dev != RT_NULL)
|
||||
{
|
||||
rt_list_insert_after(&prev_port_dev->node, &port_dev->node);
|
||||
}
|
||||
else
|
||||
{
|
||||
/* Port0 */
|
||||
rt_list_insert_after(&virtio_console_dev->port_head, &port_dev->node);
|
||||
}
|
||||
|
||||
virtio_console_dev->port_nr++;
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static void virtio_console_port_destroy(struct virtio_console_device *virtio_console_dev,
|
||||
struct port_device *port_dev)
|
||||
{
|
||||
struct virtio_console_control set_ctrl;
|
||||
|
||||
set_ctrl.id = port_dev->port_id;
|
||||
set_ctrl.event = VIRTIO_CONSOLE_PORT_OPEN;
|
||||
set_ctrl.value = 0;
|
||||
|
||||
virtio_console_send_ctrl(virtio_console_dev, &set_ctrl);
|
||||
|
||||
virtio_console_dev->port_nr--;
|
||||
|
||||
rt_list_remove(&port_dev->node);
|
||||
|
||||
rt_device_unregister((rt_device_t)port_dev);
|
||||
|
||||
rt_free(port_dev);
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_port_init(rt_device_t dev)
|
||||
{
|
||||
rt_uint16_t id;
|
||||
rt_uint16_t idx[VIRTIO_CONSOLE_QUEUE_SIZE];
|
||||
rt_uint16_t rx_queue_index, tx_queue_index;
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
struct virtio_console_device *virtio_console_dev = port_dev->console;
|
||||
struct virtio_device *virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
struct virtq *queue_rx, *queue_tx;
|
||||
|
||||
rx_queue_index = VIRTIO_CONSOLE_PORT_QUEUE_INDEX(port_dev->port_id, VIRTIO_CONSOLE_QUEUE_DATA_RX);
|
||||
tx_queue_index = VIRTIO_CONSOLE_PORT_QUEUE_INDEX(port_dev->port_id, VIRTIO_CONSOLE_QUEUE_DATA_TX);
|
||||
|
||||
queue_rx = &virtio_dev->queues[rx_queue_index];
|
||||
queue_tx = &virtio_dev->queues[tx_queue_index];
|
||||
|
||||
virtio_alloc_desc_chain(virtio_dev, rx_queue_index, queue_rx->num, idx);
|
||||
virtio_alloc_desc_chain(virtio_dev, tx_queue_index, queue_tx->num, idx);
|
||||
|
||||
for (id = 0; id < queue_rx->num; ++id)
|
||||
{
|
||||
void *addr = &port_dev->info[id].rx_char;
|
||||
|
||||
virtio_fill_desc(virtio_dev, rx_queue_index, id,
|
||||
VIRTIO_VA2PA(addr), sizeof(char), VIRTQ_DESC_F_WRITE, 0);
|
||||
|
||||
queue_rx->avail->ring[id] = id;
|
||||
}
|
||||
rt_hw_dsb();
|
||||
|
||||
queue_rx->avail->flags = 0;
|
||||
queue_rx->avail->idx = queue_rx->num;
|
||||
|
||||
queue_rx->used_idx = queue_rx->used->idx;
|
||||
|
||||
queue_tx->avail->flags = VIRTQ_AVAIL_F_NO_INTERRUPT;
|
||||
queue_tx->avail->idx = 0;
|
||||
|
||||
virtio_queue_notify(virtio_dev, rx_queue_index);
|
||||
|
||||
if (virtio_has_feature(virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
struct virtio_console_control set_ctrl;
|
||||
|
||||
set_ctrl.id = VIRTIO_CONSOLE_PORT_BAD_ID;
|
||||
set_ctrl.event = VIRTIO_CONSOLE_DEVICE_READY;
|
||||
set_ctrl.value = 1;
|
||||
|
||||
virtio_console_send_ctrl(virtio_console_dev, &set_ctrl);
|
||||
}
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_port_open(rt_device_t dev, rt_uint16_t oflag)
|
||||
{
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
|
||||
if (port_dev->port_id == 0 && virtio_has_feature(&port_dev->console->virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
/* Port0 is reserve in multiport */
|
||||
return -RT_ERROR;
|
||||
}
|
||||
|
||||
port_dev->rx_notify = RT_TRUE;
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_port_close(rt_device_t dev)
|
||||
{
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
|
||||
if (port_dev->need_destroy)
|
||||
{
|
||||
virtio_console_port_destroy(port_dev->console, port_dev);
|
||||
|
||||
/*
|
||||
* We released the device memory in virtio_console_port_destroy,
|
||||
* rt_device_close has not finished yet, make the return value
|
||||
* to empty so that rt_device_close will not access the device memory.
|
||||
*/
|
||||
return -RT_EEMPTY;
|
||||
}
|
||||
|
||||
port_dev->rx_notify = RT_FALSE;
|
||||
|
||||
return RT_EOK;
|
||||
}
|
||||
|
||||
static rt_ssize_t virtio_console_port_read(rt_device_t dev, rt_off_t pos, void *buffer, rt_size_t size)
|
||||
{
|
||||
rt_off_t i = 0;
|
||||
rt_uint16_t id;
|
||||
rt_uint32_t len;
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
struct virtio_device *virtio_dev = &port_dev->console->virtio_dev;
|
||||
rt_uint32_t queue_rx_index = port_dev->queue_rx_index;
|
||||
struct virtq *queue_rx = port_dev->queue_rx;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_base_t level = rt_spin_lock_irqsave(&port_dev->spinlock_rx);
|
||||
#endif
|
||||
|
||||
while (i < size)
|
||||
{
|
||||
if (queue_rx->used_idx == queue_rx->used->idx)
|
||||
{
|
||||
break;
|
||||
}
|
||||
rt_hw_dsb();
|
||||
|
||||
id = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].id;
|
||||
len = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].len;
|
||||
|
||||
if (len > sizeof(char))
|
||||
{
|
||||
rt_kprintf("%s: Receive buffer's size = %u is too big!\n", port_dev->parent.parent.name, len);
|
||||
len = sizeof(char);
|
||||
}
|
||||
|
||||
*((char *)buffer + i) = port_dev->info[id].rx_char;
|
||||
|
||||
queue_rx->used_idx++;
|
||||
|
||||
virtio_submit_chain(virtio_dev, queue_rx_index, id);
|
||||
|
||||
virtio_queue_notify(virtio_dev, queue_rx_index);
|
||||
|
||||
i += len;
|
||||
}
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&port_dev->spinlock_rx, level);
|
||||
#endif
|
||||
|
||||
size = i;
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
static rt_ssize_t virtio_console_port_write(rt_device_t dev, rt_off_t pos, const void *buffer, rt_size_t size)
|
||||
{
|
||||
char ch = 0;
|
||||
rt_off_t i = 0;
|
||||
rt_uint16_t id;
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
struct virtio_device *virtio_dev = &port_dev->console->virtio_dev;
|
||||
rt_uint32_t queue_tx_index = port_dev->queue_tx_index;
|
||||
struct virtq *queue_tx = port_dev->queue_tx;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_base_t level = rt_spin_lock_irqsave(&port_dev->spinlock_tx);
|
||||
#endif
|
||||
|
||||
while (i < size || ch == '\r')
|
||||
{
|
||||
id = queue_tx->avail->idx % queue_tx->num;
|
||||
|
||||
/* Keep the way until 'new line' are unified */
|
||||
if (ch != '\r')
|
||||
{
|
||||
ch = *((const char *)buffer + i);
|
||||
}
|
||||
else
|
||||
{
|
||||
i -= sizeof(char);
|
||||
}
|
||||
|
||||
port_dev->info[id].tx_char = ch;
|
||||
|
||||
ch = (ch == '\n' ? '\r' : 0);
|
||||
|
||||
virtio_free_desc(virtio_dev, queue_tx_index, id);
|
||||
|
||||
virtio_fill_desc(virtio_dev, queue_tx_index, id,
|
||||
VIRTIO_VA2PA(&port_dev->info[id].tx_char), sizeof(char), 0, 0);
|
||||
|
||||
virtio_submit_chain(virtio_dev, queue_tx_index, id);
|
||||
|
||||
virtio_queue_notify(virtio_dev, queue_tx_index);
|
||||
|
||||
virtio_alloc_desc(virtio_dev, queue_tx_index);
|
||||
|
||||
i += sizeof(char);
|
||||
}
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&port_dev->spinlock_tx, level);
|
||||
#endif
|
||||
|
||||
return size;
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_port_control(rt_device_t dev, int cmd, void *args)
|
||||
{
|
||||
rt_err_t status = RT_EOK;
|
||||
struct port_device *port_dev = (struct port_device *)dev;
|
||||
|
||||
switch (cmd)
|
||||
{
|
||||
case RT_DEVICE_CTRL_NOTIFY_SET:
|
||||
if (args == RT_NULL)
|
||||
{
|
||||
status = -RT_ERROR;
|
||||
break;
|
||||
}
|
||||
rt_memcpy(&port_dev->rx_notify_helper, args, sizeof(port_dev->rx_notify_helper));
|
||||
break;
|
||||
case RT_DEVICE_CTRL_CLR_INT:
|
||||
/* Disable RX */
|
||||
port_dev->rx_notify = RT_FALSE;
|
||||
break;
|
||||
case RT_DEVICE_CTRL_SET_INT:
|
||||
/* Enable RX */
|
||||
port_dev->rx_notify = RT_TRUE;
|
||||
break;
|
||||
case VIRTIO_DEVICE_CTRL_CONSOLE_PORT_DESTROY:
|
||||
{
|
||||
port_dev->need_destroy = RT_TRUE;
|
||||
port_dev->rx_notify = RT_FALSE;
|
||||
}
|
||||
break;
|
||||
default:
|
||||
status = -RT_EINVAL;
|
||||
break;
|
||||
}
|
||||
|
||||
return status;
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_init(rt_device_t dev)
|
||||
{
|
||||
struct virtio_console_device *virtio_console_dev = (struct virtio_console_device *)dev;
|
||||
struct virtio_device *virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
|
||||
if (virtio_has_feature(virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
rt_uint16_t id;
|
||||
rt_uint16_t idx[VIRTIO_CONSOLE_QUEUE_SIZE];
|
||||
struct virtq *queue_ctrl_rx, *queue_ctrl_tx;
|
||||
|
||||
queue_ctrl_rx = &virtio_dev->queues[VIRTIO_CONSOLE_QUEUE_CTRL_RX];
|
||||
queue_ctrl_tx = &virtio_dev->queues[VIRTIO_CONSOLE_QUEUE_CTRL_TX];
|
||||
|
||||
virtio_alloc_desc_chain(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_RX, queue_ctrl_rx->num, idx);
|
||||
virtio_alloc_desc_chain(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_TX, queue_ctrl_tx->num, idx);
|
||||
|
||||
for (id = 0; id < queue_ctrl_rx->num; ++id)
|
||||
{
|
||||
void *addr = &virtio_console_dev->info[id].rx_ctrl;
|
||||
|
||||
virtio_fill_desc(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_RX, id,
|
||||
VIRTIO_VA2PA(addr), sizeof(struct virtio_console_control), VIRTQ_DESC_F_WRITE, 0);
|
||||
|
||||
queue_ctrl_rx->avail->ring[id] = id;
|
||||
}
|
||||
rt_hw_dsb();
|
||||
|
||||
for (id = 0; id < queue_ctrl_tx->num; ++id)
|
||||
{
|
||||
virtio_console_dev->info[id].tx_ctrl_paddr = VIRTIO_VA2PA(&virtio_console_dev->info[id].tx_ctrl);
|
||||
}
|
||||
|
||||
queue_ctrl_rx->avail->flags = 0;
|
||||
queue_ctrl_rx->avail->idx = queue_ctrl_rx->num;
|
||||
|
||||
queue_ctrl_rx->used_idx = queue_ctrl_rx->used->idx;
|
||||
|
||||
queue_ctrl_tx->avail->flags = VIRTQ_AVAIL_F_NO_INTERRUPT;
|
||||
queue_ctrl_tx->avail->idx = 0;
|
||||
|
||||
virtio_queue_notify(virtio_dev, VIRTIO_CONSOLE_QUEUE_CTRL_RX);
|
||||
}
|
||||
|
||||
return virtio_console_port_create(virtio_console_dev);
|
||||
}
|
||||
|
||||
static rt_err_t virtio_console_control(rt_device_t dev, int cmd, void *args)
|
||||
{
|
||||
rt_err_t status = RT_EOK;
|
||||
struct virtio_console_device *virtio_console_dev = (struct virtio_console_device *)dev;
|
||||
|
||||
switch (cmd)
|
||||
{
|
||||
case VIRTIO_DEVICE_CTRL_CONSOLE_PORT_CREATE:
|
||||
status = virtio_console_port_create(virtio_console_dev);
|
||||
break;
|
||||
default:
|
||||
status = -RT_EINVAL;
|
||||
break;
|
||||
}
|
||||
|
||||
return status;
|
||||
}
|
||||
|
||||
#ifdef RT_USING_DEVICE_OPS
|
||||
const static struct rt_device_ops virtio_console_ops =
|
||||
{
|
||||
virtio_console_init,
|
||||
RT_NULL,
|
||||
RT_NULL,
|
||||
RT_NULL,
|
||||
RT_NULL,
|
||||
virtio_console_control
|
||||
};
|
||||
#endif
|
||||
|
||||
static void virtio_console_isr(int irqno, void *param)
|
||||
{
|
||||
rt_uint32_t id;
|
||||
rt_uint32_t len;
|
||||
struct port_device *port_dev;
|
||||
struct virtio_console_device *virtio_console_dev = (struct virtio_console_device *)param;
|
||||
struct virtio_device *virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
const char *dev_name = virtio_console_dev->parent.parent.name;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_base_t level = rt_spin_lock_irqsave(&virtio_dev->spinlock);
|
||||
#endif
|
||||
|
||||
virtio_interrupt_ack(virtio_dev);
|
||||
rt_hw_dsb();
|
||||
|
||||
do {
|
||||
struct virtq *queue_rx;
|
||||
struct virtio_console_control *ctrl, set_ctrl;
|
||||
|
||||
if (!virtio_has_feature(virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
break;
|
||||
}
|
||||
|
||||
queue_rx = &virtio_dev->queues[VIRTIO_CONSOLE_QUEUE_CTRL_RX];
|
||||
|
||||
if (queue_rx->used_idx == queue_rx->used->idx)
|
||||
{
|
||||
break;
|
||||
}
|
||||
rt_hw_dsb();
|
||||
|
||||
id = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].id;
|
||||
len = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].len;
|
||||
|
||||
queue_rx->used_idx++;
|
||||
|
||||
if (len != sizeof(struct virtio_console_control))
|
||||
{
|
||||
rt_kprintf("%s: Invalid ctrl!\n", dev_name);
|
||||
break;
|
||||
}
|
||||
|
||||
ctrl = &virtio_console_dev->info[id].rx_ctrl;
|
||||
|
||||
switch (ctrl->event)
|
||||
{
|
||||
case VIRTIO_CONSOLE_PORT_ADD:
|
||||
{
|
||||
set_ctrl.id = ctrl->id;
|
||||
set_ctrl.event = VIRTIO_CONSOLE_PORT_READY;
|
||||
set_ctrl.value = 1;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&virtio_dev->spinlock, level);
|
||||
#endif
|
||||
|
||||
virtio_console_send_ctrl(virtio_console_dev, &set_ctrl);
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
level = rt_spin_lock_irqsave(&virtio_dev->spinlock);
|
||||
#endif
|
||||
}
|
||||
break;
|
||||
case VIRTIO_CONSOLE_PORT_REMOVE:
|
||||
break;
|
||||
case VIRTIO_CONSOLE_RESIZE:
|
||||
break;
|
||||
case VIRTIO_CONSOLE_PORT_OPEN:
|
||||
{
|
||||
set_ctrl.id = ctrl->id;
|
||||
set_ctrl.event = VIRTIO_CONSOLE_PORT_OPEN;
|
||||
set_ctrl.value = 1;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&virtio_dev->spinlock, level);
|
||||
#endif
|
||||
|
||||
virtio_console_send_ctrl(virtio_console_dev, &set_ctrl);
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
level = rt_spin_lock_irqsave(&virtio_dev->spinlock);
|
||||
#endif
|
||||
}
|
||||
break;
|
||||
case VIRTIO_CONSOLE_PORT_NAME:
|
||||
break;
|
||||
default:
|
||||
rt_kprintf("%s: Unsupport ctrl[id: %d, event: %d, value: %d]!\n",
|
||||
dev_name, ctrl->id, ctrl->event, ctrl->value);
|
||||
break;
|
||||
}
|
||||
|
||||
} while (0);
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&virtio_dev->spinlock, level);
|
||||
#endif
|
||||
|
||||
rt_list_for_each_entry(port_dev, &virtio_console_dev->port_head, node)
|
||||
{
|
||||
rt_uint32_t queue_rx_index = port_dev->queue_rx_index;
|
||||
struct virtq *queue_rx = port_dev->queue_rx;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_base_t level = rt_spin_lock_irqsave(&port_dev->spinlock_rx);
|
||||
#endif
|
||||
|
||||
if (queue_rx->used_idx != queue_rx->used->idx)
|
||||
{
|
||||
rt_hw_dsb();
|
||||
|
||||
id = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].id;
|
||||
len = queue_rx->used->ring[queue_rx->used_idx % queue_rx->num].len;
|
||||
|
||||
if (port_dev->rx_notify)
|
||||
{
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&port_dev->spinlock_rx, level);
|
||||
#endif
|
||||
/* Will call virtio_console_port_read to inc used_idx */
|
||||
|
||||
if (port_dev->parent.rx_indicate != RT_NULL)
|
||||
{
|
||||
port_dev->parent.rx_indicate(&port_dev->parent, len);
|
||||
}
|
||||
|
||||
if (port_dev->rx_notify_helper.notify != RT_NULL)
|
||||
{
|
||||
port_dev->rx_notify_helper.notify(port_dev->rx_notify_helper.dev);
|
||||
}
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
level = rt_spin_lock_irqsave(&port_dev->spinlock_rx);
|
||||
#endif
|
||||
}
|
||||
else
|
||||
{
|
||||
queue_rx->used_idx++;
|
||||
|
||||
virtio_submit_chain(virtio_dev, queue_rx_index, id);
|
||||
|
||||
virtio_queue_notify(virtio_dev, queue_rx_index);
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_unlock_irqrestore(&port_dev->spinlock_rx, level);
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
rt_err_t rt_virtio_console_init(rt_ubase_t *mmio_base, rt_uint32_t irq)
|
||||
{
|
||||
int i;
|
||||
rt_size_t queues_num;
|
||||
static int dev_no = 0;
|
||||
char dev_name[RT_NAME_MAX];
|
||||
struct virtio_device *virtio_dev;
|
||||
struct virtio_console_device *virtio_console_dev;
|
||||
|
||||
RT_ASSERT(RT_USING_VIRTIO_CONSOLE_PORT_MAX_NR > 0);
|
||||
|
||||
virtio_console_dev = rt_malloc(sizeof(struct virtio_console_device));
|
||||
|
||||
if (virtio_console_dev == RT_NULL)
|
||||
{
|
||||
goto _alloc_fail;
|
||||
}
|
||||
|
||||
virtio_dev = &virtio_console_dev->virtio_dev;
|
||||
virtio_dev->irq = irq;
|
||||
virtio_dev->mmio_base = mmio_base;
|
||||
|
||||
virtio_console_dev->config = (struct virtio_console_config *)virtio_dev->mmio_config->config;
|
||||
|
||||
#ifdef RT_USING_SMP
|
||||
rt_spin_lock_init(&virtio_dev->spinlock);
|
||||
#endif
|
||||
|
||||
virtio_reset_device(virtio_dev);
|
||||
virtio_status_acknowledge_driver(virtio_dev);
|
||||
|
||||
virtio_dev->mmio_config->driver_features = virtio_dev->mmio_config->device_features & ~(
|
||||
(1 << VIRTIO_F_RING_EVENT_IDX) |
|
||||
(1 << VIRTIO_F_RING_INDIRECT_DESC));
|
||||
|
||||
virtio_status_driver_ok(virtio_dev);
|
||||
|
||||
if (!virtio_has_feature(virtio_dev, VIRTIO_CONSOLE_F_MULTIPORT))
|
||||
{
|
||||
virtio_console_dev->max_port_nr = 1;
|
||||
queues_num = 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (virtio_console_dev->config->max_nr_ports > RT_USING_VIRTIO_CONSOLE_PORT_MAX_NR)
|
||||
{
|
||||
virtio_console_dev->max_port_nr = RT_USING_VIRTIO_CONSOLE_PORT_MAX_NR;
|
||||
virtio_console_dev->config->max_nr_ports = virtio_console_dev->max_port_nr;
|
||||
}
|
||||
else
|
||||
{
|
||||
virtio_console_dev->max_port_nr = virtio_console_dev->config->max_nr_ports;
|
||||
}
|
||||
|
||||
queues_num = VIRTIO_CONSOLE_PORT_QUEUE_INDEX(virtio_console_dev->max_port_nr, VIRTIO_CONSOLE_QUEUE_DATA_RX);
|
||||
}
|
||||
|
||||
if (virtio_queues_alloc(virtio_dev, queues_num) != RT_EOK)
|
||||
{
|
||||
goto _alloc_fail;
|
||||
}
|
||||
|
||||
for (i = 0; i < virtio_dev->queues_num; ++i)
|
||||
{
|
||||
if (virtio_queue_init(virtio_dev, i, VIRTIO_CONSOLE_QUEUE_SIZE) != RT_EOK)
|
||||
{
|
||||
for (; i >= 0; --i)
|
||||
{
|
||||
virtio_queue_destroy(virtio_dev, i);
|
||||
}
|
||||
goto _alloc_fail;
|
||||
}
|
||||
}
|
||||
|
||||
virtio_console_dev->parent.type = RT_Device_Class_Char;
|
||||
#ifdef RT_USING_DEVICE_OPS
|
||||
virtio_console_dev->parent.ops = &virtio_console_ops;
|
||||
#else
|
||||
virtio_console_dev->parent.init = virtio_console_init;
|
||||
virtio_console_dev->parent.open = RT_NULL;
|
||||
virtio_console_dev->parent.close = RT_NULL;
|
||||
virtio_console_dev->parent.read = RT_NULL;
|
||||
virtio_console_dev->parent.write = RT_NULL;
|
||||
virtio_console_dev->parent.control = virtio_console_control;
|
||||
#endif
|
||||
|
||||
virtio_console_dev->parent.rx_indicate = RT_NULL;
|
||||
virtio_console_dev->parent.tx_complete = RT_NULL;
|
||||
|
||||
virtio_console_dev->console_id = dev_no;
|
||||
virtio_console_dev->port_nr = 0;
|
||||
rt_list_init(&virtio_console_dev->port_head);
|
||||
|
||||
rt_snprintf(dev_name, RT_NAME_MAX, "virtio-console%d", dev_no++);
|
||||
|
||||
rt_hw_interrupt_install(irq, virtio_console_isr, virtio_console_dev, dev_name);
|
||||
rt_hw_interrupt_umask(irq);
|
||||
|
||||
return rt_device_register((rt_device_t)virtio_console_dev, dev_name, RT_DEVICE_FLAG_RDWR | RT_DEVICE_FLAG_INT_RX);
|
||||
|
||||
_alloc_fail:
|
||||
|
||||
if (virtio_console_dev != RT_NULL)
|
||||
{
|
||||
virtio_queues_free(virtio_dev);
|
||||
rt_free(virtio_console_dev);
|
||||
}
|
||||
return -RT_ENOMEM;
|
||||
}
|
||||
#endif /* RT_USING_VIRTIO_CONSOLE */
|
||||
Reference in New Issue
Block a user