drv.c

   1 /*
   2  * Copyright 2016 The Chromium OS Authors. All rights reserved.
   3  * Use of this source code is governed by a BSD-style license that can be
   4  * found in the LICENSE file.
   5  */
   6 #include <assert.h>
   7 #include <fcntl.h>
   8 #include <pthread.h>
   9 #include <stdint.h>
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <sys/mman.h>
  14 #include <xf86drm.h>
  15
  16 #include "drv_priv.h"
  17 #include "helpers.h"
  18 #include "util.h"
  19
  20 #ifdef DRV_AMDGPU
  21 extern struct backend backend_amdgpu;
  22 #endif
  23 extern struct backend backend_cirrus;
  24 extern struct backend backend_evdi;
  25 #ifdef DRV_EXYNOS
  26 extern struct backend backend_exynos;
  27 #endif
  28 extern struct backend backend_gma500;
  29 #ifdef DRV_I915
  30 extern struct backend backend_i915;
  31 #endif
  32 #ifdef DRV_MARVELL
  33 extern struct backend backend_marvell;
  34 #endif
  35 #ifdef DRV_MEDIATEK
  36 extern struct backend backend_mediatek;
  37 #endif
  38 extern struct backend backend_nouveau;
  39 #ifdef DRV_ROCKCHIP
  40 extern struct backend backend_rockchip;
  41 #endif
  42 #ifdef DRV_TEGRA
  43 extern struct backend backend_tegra;
  44 #endif
  45 extern struct backend backend_udl;
  46 extern struct backend backend_vgem;
  47 extern struct backend backend_virtio_gpu;
  48
  49 static struct backend *drv_get_backend(int fd)
  50 {
  51         drmVersionPtr drm_version;
  52         unsigned int i;
  53
  54         drm_version = drmGetVersion(fd);
  55
  56         if (!drm_version)
  57                 return NULL;
  58
  59         struct backend *backend_list[] = {
  60 #ifdef DRV_AMDGPU
  61                 &backend_amdgpu,
  62 #endif
  63                 &backend_cirrus,
  64                 &backend_evdi,
  65 #ifdef DRV_EXYNOS
  66                 &backend_exynos,
  67 #endif
  68                 &backend_gma500,
  69 #ifdef DRV_I915
  70                 &backend_i915,
  71 #endif
  72 #ifdef DRV_MARVELL
  73                 &backend_marvell,
  74 #endif
  75 #ifdef DRV_MEDIATEK
  76                 &backend_mediatek,
  77 #endif
  78                 &backend_nouveau,
  79 #ifdef DRV_ROCKCHIP
  80                 &backend_rockchip,
  81 #endif
  82 #ifdef DRV_TEGRA
  83                 &backend_tegra,
  84 #endif
  85                 &backend_udl,
  86                 &backend_vgem,
  87                 &backend_virtio_gpu,
  88         };
  89
  90         for(i = 0; i < ARRAY_SIZE(backend_list); i++)
  91                 if (!strcmp(drm_version->name, backend_list[i]->name)) {
  92                         drmFreeVersion(drm_version);
  93                         return backend_list[i];
  94                 }
  95
  96         drmFreeVersion(drm_version);
  97         return NULL;
  98 }
  99
 100 struct driver *drv_create(int fd)
 101 {
 102         struct driver *drv;
 103         int ret;
 104
 105         drv = (struct driver *) calloc(1, sizeof(*drv));
 106
 107         if (!drv)
 108                 return NULL;
 109
 110         drv->fd = fd;
 111         drv->backend = drv_get_backend(fd);
 112
 113         if (!drv->backend)
 114                 goto free_driver;
 115
 116         if (pthread_mutex_init(&drv->driver_lock, NULL))
 117                 goto free_driver;
 118
 119         drv->buffer_table = drmHashCreate();
 120         if (!drv->buffer_table)
 121                 goto free_lock;
 122
 123         drv->map_table = drmHashCreate();
 124         if (!drv->map_table)
 125                 goto free_buffer_table;
 126
 127         LIST_INITHEAD(&drv->backend->combinations);
 128
 129         if (drv->backend->init) {
 130                 ret = drv->backend->init(drv);
 131                 if (ret)
 132                         goto free_map_table;
 133         }
 134
 135         return drv;
 136
 137 free_map_table:
 138         drmHashDestroy(drv->map_table);
 139 free_buffer_table:
 140         drmHashDestroy(drv->buffer_table);
 141 free_lock:
 142         pthread_mutex_destroy(&drv->driver_lock);
 143 free_driver:
 144         free(drv);
 145         return NULL;
 146 }
 147
 148 void drv_destroy(struct driver *drv)
 149 {
 150         pthread_mutex_lock(&drv->driver_lock);
 151
 152         if (drv->backend->close)
 153                 drv->backend->close(drv);
 154
 155         drmHashDestroy(drv->buffer_table);
 156         drmHashDestroy(drv->map_table);
 157
 158         list_for_each_entry_safe(struct combination_list_element, elem,
 159                                  &drv->backend->combinations, link) {
 160                 LIST_DEL(&elem->link);
 161                 free(elem);
 162         }
 163
 164         pthread_mutex_unlock(&drv->driver_lock);
 165         pthread_mutex_destroy(&drv->driver_lock);
 166
 167         free(drv);
 168 }
 169
 170 int drv_get_fd(struct driver *drv)
 171 {
 172         return drv->fd;
 173 }
 174
 175 const char *
 176 drv_get_name(struct driver *drv)
 177 {
 178         return drv->backend->name;
 179 }
 180
 181 int drv_is_combination_supported(struct driver *drv, uint32_t format,
 182                                  uint64_t usage, uint64_t modifier)
 183 {
 184
 185         if (format == DRM_FORMAT_NONE || usage == BO_USE_NONE)
 186                 return 0;
 187
 188         list_for_each_entry(struct combination_list_element, elem,
 189                             &drv->backend->combinations, link) {
 190                 if (format == elem->combination.format &&
 191                     usage == (elem->combination.usage & usage) &&
 192                     modifier == elem->combination.modifier)
 193                         return 1;
 194         }
 195
 196         return 0;
 197 }
 198
 199 struct bo *drv_bo_new(struct driver *drv, uint32_t width, uint32_t height,
 200                       uint32_t format)
 201 {
 202
 203         struct bo *bo;
 204         bo = (struct bo *) calloc(1, sizeof(*bo));
 205
 206         if (!bo)
 207                 return NULL;
 208
 209         bo->drv = drv;
 210         bo->width = width;
 211         bo->height = height;
 212         bo->format = format;
 213         bo->num_planes = drv_num_planes_from_format(format);
 214
 215         if (!bo->num_planes) {
 216                 free(bo);
 217                 return NULL;
 218         }
 219
 220         return bo;
 221 }
 222
 223 struct bo *drv_bo_create(struct driver *drv, uint32_t width, uint32_t height,
 224                          uint32_t format, uint64_t flags)
 225 {
 226         int ret;
 227         size_t plane;
 228         struct bo *bo;
 229
 230         bo = drv_bo_new(drv, width, height, format);
 231
 232         if (!bo)
 233                 return NULL;
 234
 235         ret = drv->backend->bo_create(bo, width, height, format, flags);
 236
 237         if (ret) {
 238                 free(bo);
 239                 return NULL;
 240         }
 241
 242         pthread_mutex_lock(&drv->driver_lock);
 243
 244         for (plane = 0; plane < bo->num_planes; plane++)
 245                 drv_increment_reference_count(drv, bo, plane);
 246
 247         pthread_mutex_unlock(&drv->driver_lock);
 248
 249         return bo;
 250 }
 251
 252 void drv_bo_destroy(struct bo *bo)
 253 {
 254         size_t plane;
 255         uintptr_t total = 0;
 256         struct driver *drv = bo->drv;
 257
 258         pthread_mutex_lock(&drv->driver_lock);
 259
 260         for (plane = 0; plane < bo->num_planes; plane++)
 261                 drv_decrement_reference_count(drv, bo, plane);
 262
 263         for (plane = 0; plane < bo->num_planes; plane++)
 264                 total += drv_get_reference_count(drv, bo, plane);
 265
 266         pthread_mutex_unlock(&drv->driver_lock);
 267
 268         if (total == 0)
 269                 bo->drv->backend->bo_destroy(bo);
 270
 271         free(bo);
 272 }
 273
 274 struct bo *drv_bo_import(struct driver *drv, struct drv_import_fd_data *data)
 275 {
 276         int ret;
 277         size_t plane;
 278         struct bo *bo;
 279         struct drm_prime_handle prime_handle;
 280
 281         bo = drv_bo_new(drv, data->width, data->height, data->format);
 282
 283         if (!bo)
 284                 return NULL;
 285
 286         for (plane = 0; plane < bo->num_planes; plane++) {
 287
 288                 memset(&prime_handle, 0, sizeof(prime_handle));
 289                 prime_handle.fd = data->fds[plane];
 290
 291                 ret = drmIoctl(drv->fd, DRM_IOCTL_PRIME_FD_TO_HANDLE,
 292                                &prime_handle);
 293
 294                 if (ret) {
 295                         fprintf(stderr, "drv: DRM_IOCTL_PRIME_FD_TO_HANDLE failed "
 296                                 "(fd=%u)\n", prime_handle.fd);
 297
 298                         if (plane > 0) {
 299                                 bo->num_planes = plane;
 300                                 drv_bo_destroy(bo);
 301                         } else {
 302                                 free(bo);
 303                         }
 304
 305                         return NULL;
 306                 }
 307
 308                 bo->handles[plane].u32 = prime_handle.handle;
 309                 bo->strides[plane] = data->strides[plane];
 310                 bo->offsets[plane] = data->offsets[plane];
 311                 bo->sizes[plane] = data->sizes[plane];
 312                 bo->format_modifiers[plane] = data->format_modifiers[plane];
 313                 bo->total_size += data->sizes[plane];
 314
 315                 pthread_mutex_lock(&drv->driver_lock);
 316                 drv_increment_reference_count(drv, bo, plane);
 317                 pthread_mutex_unlock(&drv->driver_lock);
 318         }
 319
 320         return bo;
 321 }
 322
 323 void *drv_bo_map(struct bo *bo, uint32_t x, uint32_t y, uint32_t width,
 324                  uint32_t height, uint32_t flags, void **map_data, size_t plane)
 325 {
 326         void *ptr;
 327         uint8_t *addr;
 328         size_t offset;
 329         struct map_info *data;
 330
 331         assert(width > 0);
 332         assert(height > 0);
 333         assert(x + width <= drv_bo_get_width(bo));
 334         assert(y + height <= drv_bo_get_height(bo));
 335
 336         pthread_mutex_lock(&bo->drv->driver_lock);
 337
 338         if (!drmHashLookup(bo->drv->map_table, bo->handles[plane].u32, &ptr)) {
 339                 data = (struct map_info *) ptr;
 340                 data->refcount++;
 341                 goto success;
 342         }
 343
 344         data = calloc(1, sizeof(*data));
 345         addr = bo->drv->backend->bo_map(bo, data, plane);
 346         if (addr == MAP_FAILED) {
 347                 *map_data = NULL;
 348                 free(data);
 349                 pthread_mutex_unlock(&bo->drv->driver_lock);
 350                 return MAP_FAILED;
 351         }
 352
 353         data->refcount = 1;
 354         data->addr = addr;
 355         data->handle = bo->handles[plane].u32;
 356         drmHashInsert(bo->drv->buffer_table, bo->handles[plane].u32,
 357                       (void *) data);
 358
 359 success:
 360         *map_data = (void *) data;
 361         offset = drv_bo_get_plane_stride(bo, plane) * y;
 362         offset += drv_stride_from_format(bo->format, x, plane);
 363         addr = (uint8_t *) data->addr;
 364         addr += drv_bo_get_plane_offset(bo, plane) + offset;
 365         pthread_mutex_unlock(&bo->drv->driver_lock);
 366
 367         return (void *) addr;
 368 }
 369
 370 int drv_bo_unmap(struct bo *bo, void *map_data)
 371 {
 372         struct map_info *data = map_data;
 373         int ret = 0;
 374
 375         assert(data);
 376         assert(data->refcount >= 0);
 377
 378         pthread_mutex_lock(&bo->drv->driver_lock);
 379
 380         if (!--data->refcount) {
 381                 if (bo->drv->backend->bo_unmap)
 382                         ret = bo->drv->backend->bo_unmap(bo, data);
 383                 else
 384                         ret = munmap(data->addr, data->length);
 385                 drmHashDelete(bo->drv->map_table, data->handle);
 386                 free(data);
 387         }
 388
 389         pthread_mutex_unlock(&bo->drv->driver_lock);
 390
 391         return ret;
 392 }
 393
 394 uint32_t drv_bo_get_width(struct bo *bo)
 395 {
 396         return bo->width;
 397 }
 398
 399 uint32_t drv_bo_get_height(struct bo *bo)
 400 {
 401         return bo->height;
 402 }
 403
 404 uint32_t drv_bo_get_stride_or_tiling(struct bo *bo)
 405 {
 406         return bo->tiling ? bo->tiling : drv_bo_get_plane_stride(bo, 0);
 407 }
 408
 409 size_t drv_bo_get_num_planes(struct bo *bo)
 410 {
 411         return bo->num_planes;
 412 }
 413
 414 union bo_handle drv_bo_get_plane_handle(struct bo *bo, size_t plane)
 415 {
 416         return bo->handles[plane];
 417 }
 418
 419 #ifndef DRM_RDWR
 420 #define DRM_RDWR O_RDWR
 421 #endif
 422
 423 int drv_bo_get_plane_fd(struct bo *bo, size_t plane)
 424 {
 425
 426         int ret, fd;
 427         assert(plane < bo->num_planes);
 428
 429         ret = drmPrimeHandleToFD(bo->drv->fd, bo->handles[plane].u32,
 430                                  DRM_CLOEXEC | DRM_RDWR, &fd);
 431
 432         return (ret) ? ret : fd;
 433
 434 }
 435
 436 uint32_t drv_bo_get_plane_offset(struct bo *bo, size_t plane)
 437 {
 438         assert(plane < bo->num_planes);
 439         return bo->offsets[plane];
 440 }
 441
 442 uint32_t drv_bo_get_plane_size(struct bo *bo, size_t plane)
 443 {
 444         assert(plane < bo->num_planes);
 445         return bo->sizes[plane];
 446 }
 447
 448 uint32_t drv_bo_get_plane_stride(struct bo *bo, size_t plane)
 449 {
 450         assert(plane < bo->num_planes);
 451         return bo->strides[plane];
 452 }
 453
 454 uint64_t drv_bo_get_plane_format_modifier(struct bo *bo, size_t plane)
 455 {
 456         assert(plane < bo->num_planes);
 457         return bo->format_modifiers[plane];
 458 }
 459
 460 uint32_t drv_bo_get_format(struct bo *bo)
 461 {
 462         return bo->format;
 463 }
 464
 465 uint32_t drv_resolve_format(struct driver *drv, uint32_t format)
 466 {
 467         if (drv->backend->resolve_format)
 468                 return drv->backend->resolve_format(format);
 469
 470         return format;
 471 }
 472
 473 /*
 474  * This function returns the stride for a given format, width and plane.
 475  */
 476 int drv_stride_from_format(uint32_t format, uint32_t width, size_t plane)
 477 {
 478         int stride = width * DIV_ROUND_UP(drv_bpp_from_format(format, plane),
 479                                           8);
 480
 481         /*
 482          * Only downsample for certain multiplanar formats which have horizontal
 483          * subsampling for chroma planes.  Only formats supported by our drivers
 484          * are listed here -- add more as needed.
 485          */
 486         if (plane != 0) {
 487                 switch (format) {
 488                 case DRM_FORMAT_NV12:
 489                 case DRM_FORMAT_YVU420:
 490                         stride = stride / 2;
 491                         break;
 492                 }
 493         }
 494
 495         return stride;
 496 }
 497
 498 size_t drv_num_planes_from_format(uint32_t format)
 499 {
 500         switch (format) {
 501         case DRM_FORMAT_ABGR1555:
 502         case DRM_FORMAT_ABGR2101010:
 503         case DRM_FORMAT_ABGR4444:
 504         case DRM_FORMAT_ABGR8888:
 505         case DRM_FORMAT_ARGB1555:
 506         case DRM_FORMAT_ARGB2101010:
 507         case DRM_FORMAT_ARGB4444:
 508         case DRM_FORMAT_ARGB8888:
 509         case DRM_FORMAT_AYUV:
 510         case DRM_FORMAT_BGR233:
 511         case DRM_FORMAT_BGR565:
 512         case DRM_FORMAT_BGR888:
 513         case DRM_FORMAT_BGRA1010102:
 514         case DRM_FORMAT_BGRA4444:
 515         case DRM_FORMAT_BGRA5551:
 516         case DRM_FORMAT_BGRA8888:
 517         case DRM_FORMAT_BGRX1010102:
 518         case DRM_FORMAT_BGRX4444:
 519         case DRM_FORMAT_BGRX5551:
 520         case DRM_FORMAT_BGRX8888:
 521         case DRM_FORMAT_C8:
 522         case DRM_FORMAT_GR88:
 523         case DRM_FORMAT_R8:
 524         case DRM_FORMAT_RG88:
 525         case DRM_FORMAT_RGB332:
 526         case DRM_FORMAT_RGB565:
 527         case DRM_FORMAT_RGB888:
 528         case DRM_FORMAT_RGBA1010102:
 529         case DRM_FORMAT_RGBA4444:
 530         case DRM_FORMAT_RGBA5551:
 531         case DRM_FORMAT_RGBA8888:
 532         case DRM_FORMAT_RGBX1010102:
 533         case DRM_FORMAT_RGBX4444:
 534         case DRM_FORMAT_RGBX5551:
 535         case DRM_FORMAT_RGBX8888:
 536         case DRM_FORMAT_UYVY:
 537         case DRM_FORMAT_VYUY:
 538         case DRM_FORMAT_XBGR1555:
 539         case DRM_FORMAT_XBGR2101010:
 540         case DRM_FORMAT_XBGR4444:
 541         case DRM_FORMAT_XBGR8888:
 542         case DRM_FORMAT_XRGB1555:
 543         case DRM_FORMAT_XRGB2101010:
 544         case DRM_FORMAT_XRGB4444:
 545         case DRM_FORMAT_XRGB8888:
 546         case DRM_FORMAT_YUYV:
 547         case DRM_FORMAT_YVYU:
 548                 return 1;
 549         case DRM_FORMAT_NV12:
 550                 return 2;
 551         case DRM_FORMAT_YVU420:
 552                 return 3;
 553         }
 554
 555         fprintf(stderr, "drv: UNKNOWN FORMAT %d\n", format);
 556         return 0;
 557 }
 558
 559 uint32_t drv_size_from_format(uint32_t format, uint32_t stride,
 560                               uint32_t height, size_t plane)
 561 {
 562         assert(plane < drv_num_planes_from_format(format));
 563         uint32_t vertical_subsampling;
 564
 565         switch (format) {
 566         case DRM_FORMAT_NV12:
 567         case DRM_FORMAT_YVU420:
 568                 vertical_subsampling = (plane == 0) ? 1 : 2;
 569                 break;
 570         default:
 571                 vertical_subsampling = 1;
 572         }
 573
 574         return stride * DIV_ROUND_UP(height, vertical_subsampling);
 575 }
 576
 577 uint32_t drv_num_buffers_per_bo(struct bo *bo)
 578 {
 579         uint32_t count = 0;
 580         size_t plane, p;
 581
 582         for (plane = 0; plane < bo->num_planes; plane++) {
 583                 for (p = 0; p < plane; p++)
 584                         if (bo->handles[p].u32 == bo->handles[plane].u32)
 585                                 break;
 586                 if (p == plane)
 587                         count++;
 588         }
 589
 590         return count;
 591 }