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