Kevin Pouget
ffaafde16f
* ggml-virtgpu-backend: validate the consistency of the received objects This patch adds consistency checks in the ggml-virtgpu-backend (running on the host side) to ensure that the data received from the guest is consistent (valid pointers, valid sizes and offsets). * ggml-virtgpu-backend: add fallback/skips for optional ggml backend methods ``` 1. bck->iface.synchronize(bck) 2. buft->iface.get_alloc_size(buft, op) 3. buft->iface.get_max_size(buft) ``` these three methods are optional in the GGML interface. `get_max_size` was already properly defaulted, but `backend sychronize` and `butf get_max_size` would have segfaulted the backend if not implemented. * ggml-virtgpu-backend: fix log format missing argument * ggml-virtgpu-backend: improve the abort message * ggml-virtgpu-backend: more safety checks * ggml-virtgpu-backend: new error code * ggml-virtgpu-backend: initialize all the error codes * ggml-virtgpu: add a missing comment generated by the code generator * ggml-virtgpu: add the '[virtgpu]' prefix to the device/buffer names * ggml-virtgpu: apir_device_buffer_from_ptr: improve the error message * ggml-virtgpu: shared: make it match the latest api_remoting.h of Virglrenderer APIR (still unmerged) * ggml-virtgpu: update the code generator to have dispatch_command_name in a host/guest shared file * ggml-virtgpu: REMOTE_CALL: fail if the backend returns an error * docs/backend/VirtGPU.md: indicate that the RAM+VRAM size is limed to 64 GB with libkrun * ggml-virtgpu: turn off clang-format header ordering for some of the files Compilation breaks when ordered alphabetically. * ggml-virtgpu: clang-format * ggml-virtgpu/backend/shared/api_remoting: better comments for the APIR return codes
118 lines
3.1 KiB
C
118 lines
3.1 KiB
C
#pragma once
|
|
|
|
// clang-format off
|
|
#include "virtgpu-utils.h"
|
|
#include "virtgpu-shm.h"
|
|
#include "virtgpu-apir.h"
|
|
|
|
#include "backend/shared/api_remoting.h"
|
|
#include "backend/shared/apir_cs.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/sysmacros.h>
|
|
#include <threads.h>
|
|
#include <xf86drm.h>
|
|
|
|
#include <cstring>
|
|
|
|
#include "ggml-remoting.h"
|
|
|
|
#define VIRGL_RENDERER_UNSTABLE_APIS 1
|
|
#include "apir_hw.h"
|
|
#include <drm/virtgpu_drm.h>
|
|
#include "venus_hw.h"
|
|
// clang-format on
|
|
|
|
#ifndef VIRTGPU_DRM_CAPSET_APIR
|
|
// Will be defined include/drm/virtgpu_drm.h when
|
|
// https://gitlab.freedesktop.org/virgl/virglrenderer/-/merge_requests/1590/diffs
|
|
// is merged
|
|
# define VIRTGPU_DRM_CAPSET_APIR 10
|
|
#endif
|
|
|
|
// Mesa/Virlgrenderer Venus internal. Only necessary during the
|
|
// Venus->APIR transition in Virglrenderer
|
|
#define VENUS_COMMAND_TYPE_LENGTH 331
|
|
|
|
#ifndef VIRTGPU_DRM_CAPSET_VENUS // only available with Linux >= v6.16
|
|
# define VIRTGPU_DRM_CAPSET_VENUS 4
|
|
#endif
|
|
|
|
typedef uint32_t virgl_renderer_capset;
|
|
|
|
/* from src/virtio/vulkan/vn_renderer_virtgpu.c */
|
|
#define VIRTGPU_PCI_VENDOR_ID 0x1af4
|
|
#define VIRTGPU_PCI_DEVICE_ID 0x1050
|
|
#define VIRTGPU_BLOB_MEM_GUEST_VRAM 0x0004
|
|
#define VIRTGPU_PARAM_GUEST_VRAM 9
|
|
|
|
#define SHMEM_DATA_SIZE 0x1830000 // 24MiB
|
|
#define SHMEM_REPLY_SIZE 0x4000
|
|
|
|
#define ARRAY_SIZE(x) (sizeof(x) / sizeof((x)[0]))
|
|
|
|
enum virt_gpu_result_t {
|
|
APIR_SUCCESS = 0,
|
|
APIR_ERROR_INITIALIZATION_FAILED = -1,
|
|
};
|
|
|
|
#define PRINTFLIKE(f, a) __attribute__((format(__printf__, f, a)))
|
|
|
|
struct virtgpu {
|
|
bool use_apir_capset;
|
|
|
|
int fd;
|
|
|
|
struct {
|
|
virgl_renderer_capset id;
|
|
uint32_t version;
|
|
virgl_renderer_capset_apir data;
|
|
} capset;
|
|
|
|
util_sparse_array shmem_array;
|
|
|
|
/* APIR communication pages */
|
|
virtgpu_shmem reply_shmem;
|
|
virtgpu_shmem data_shmem;
|
|
|
|
/* Mutex to protect shared data_shmem buffer from concurrent access */
|
|
mtx_t data_shmem_mutex;
|
|
|
|
/* Cached device information to prevent memory leaks and race conditions */
|
|
struct {
|
|
char * description;
|
|
char * name;
|
|
int32_t device_count;
|
|
uint32_t type;
|
|
size_t memory_free;
|
|
size_t memory_total;
|
|
} cached_device_info;
|
|
|
|
/* Cached buffer type information to prevent memory leaks and race conditions */
|
|
struct {
|
|
apir_buffer_type_host_handle_t host_handle;
|
|
char * name;
|
|
size_t alignment;
|
|
size_t max_size;
|
|
} cached_buffer_type;
|
|
};
|
|
|
|
static inline int virtgpu_ioctl(virtgpu * gpu, unsigned long request, void * args) {
|
|
return drmIoctl(gpu->fd, request, args);
|
|
}
|
|
|
|
virtgpu * create_virtgpu();
|
|
|
|
apir_encoder * remote_call_prepare(virtgpu * gpu, ApirCommandType apir_cmd_type, int32_t cmd_flags);
|
|
|
|
uint32_t remote_call(virtgpu * gpu,
|
|
apir_encoder * enc,
|
|
apir_decoder ** dec,
|
|
float max_wait_ms,
|
|
long long * call_duration_ns);
|
|
|
|
void remote_call_finish(virtgpu * gpu, apir_encoder * enc, apir_decoder * dec);
|