diff --git a/lib/include/openamp/rpmsg.h b/lib/include/openamp/rpmsg.h index 17eaddec2..c897bfc28 100644 --- a/lib/include/openamp/rpmsg.h +++ b/lib/include/openamp/rpmsg.h @@ -54,6 +54,7 @@ typedef void (*rpmsg_ept_release_cb)(struct rpmsg_endpoint *ept); typedef void (*rpmsg_ns_unbind_cb)(struct rpmsg_endpoint *ept); typedef void (*rpmsg_ns_bind_cb)(struct rpmsg_device *rdev, const char *name, uint32_t dest); +typedef int (*rpmsg_notify_wait_cb)(struct rpmsg_device *rdev, uint32_t id); /** * @brief Structure that binds a local RPMsg address to its user @@ -143,6 +144,9 @@ struct rpmsg_device { /** Callback handler for name service announcement, called when remote ept is destroyed */ rpmsg_ns_bind_cb ns_unbind_cb; + /** callback handler for rpmsg service, called when service can't get tx buffer */ + rpmsg_notify_wait_cb notify_wait_cb; + /** RPMsg device operations */ struct rpmsg_device_ops ops; diff --git a/lib/rpmsg/rpmsg.c b/lib/rpmsg/rpmsg.c index 3be947d3c..165e3a4b7 100644 --- a/lib/rpmsg/rpmsg.c +++ b/lib/rpmsg/rpmsg.c @@ -354,3 +354,11 @@ void rpmsg_destroy_ept(struct rpmsg_endpoint *ept) (void)rpmsg_send_ns_message(ept, RPMSG_NS_DESTROY); rpmsg_unregister_endpoint(ept); } + +int rpmsg_notify_wait(struct rpmsg_device *rdev, uint32_t id) +{ + if (rdev->notify_wait_cb) + return rdev->notify_wait_cb(rdev, id); + + return RPMSG_ERR_NXIO; +} diff --git a/lib/rpmsg/rpmsg_internal.h b/lib/rpmsg/rpmsg_internal.h index f1c61c5f2..9ebf35b8a 100644 --- a/lib/rpmsg/rpmsg_internal.h +++ b/lib/rpmsg/rpmsg_internal.h @@ -111,6 +111,8 @@ rpmsg_get_ept_from_addr(struct rpmsg_device *rdev, uint32_t addr) return rpmsg_get_endpoint(rdev, NULL, addr, RPMSG_ADDR_ANY); } +int rpmsg_notify_wait(struct rpmsg_device *rdev, uint32_t id); + /** * @internal * diff --git a/lib/rpmsg/rpmsg_virtio.c b/lib/rpmsg/rpmsg_virtio.c index 783b76b0d..624fd0474 100644 --- a/lib/rpmsg/rpmsg_virtio.c +++ b/lib/rpmsg/rpmsg_virtio.c @@ -358,6 +358,16 @@ static void rpmsg_virtio_release_rx_buffer(struct rpmsg_device *rdev, metal_mutex_release(&rdev->lock); } +static int rpmsg_virtio_notify_wait(struct rpmsg_virtio_device *rvdev, + struct virtqueue *vq) +{ + struct virtio_vring_info *vring_info; + + vring_info = &rvdev->vdev->vrings_info[vq->vq_queue_index]; + + return rpmsg_notify_wait(&rvdev->rdev, vring_info->notifyid); +} + static void *rpmsg_virtio_get_tx_payload_buffer(struct rpmsg_device *rdev, uint32_t *len, int wait) {