diff --git a/lib/include/openamp/virtio.h b/lib/include/openamp/virtio.h index c3b03952..b9f6328d 100644 --- a/lib/include/openamp/virtio.h +++ b/lib/include/openamp/virtio.h @@ -449,6 +449,9 @@ static inline int virtio_get_features(struct virtio_device *vdev, return -ENXIO; *features = vdev->func->get_features(vdev); + if (VIRTIO_ROLE_IS_DEVICE(vdev)) + vdev->features = *features; + return 0; } @@ -565,6 +568,28 @@ static inline int virtio_free_buf(struct virtio_device *vdev, void *buf) return 0; } +/** + * @brief Check if the virtio device support a specific feature. + * + * @param vdev Pointer to device structure. + * @param feature_bit Feature bit to check. + * + * @return true if the feature is supported, otherwise false. + */ +static inline bool virtio_has_feature(struct virtio_device *vdev, + unsigned int feature_bit) +{ + uint32_t features; + + if (!vdev) + return false; + + if (!vdev->features) + virtio_get_features(vdev, &features); + + return (vdev->features & (1UL << feature_bit)) != 0; +} + #if defined __cplusplus } #endif