tests/modetest/modetest.c

   1 /*
   2  * DRM based mode setting test program
   3  * Copyright 2008 Tungsten Graphics
   4  *   Jakob Bornecrantz <jakob@tungstengraphics.com>
   5  * Copyright 2008 Intel Corporation
   6  *   Jesse Barnes <jesse.barnes@intel.com>
   7  *
   8  * Permission is hereby granted, free of charge, to any person obtaining a
   9  * copy of this software and associated documentation files (the "Software"),
  10  * to deal in the Software without restriction, including without limitation
  11  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  12  * and/or sell copies of the Software, and to permit persons to whom the
  13  * Software is furnished to do so, subject to the following conditions:
  14  *
  15  * The above copyright notice and this permission notice shall be included in
  16  * all copies or substantial portions of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  23  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  24  * IN THE SOFTWARE.
  25  */
  26
  27 /*
  28  * This fairly simple test program dumps output in a similar format to the
  29  * "xrandr" tool everyone knows & loves.  It's necessarily slightly different
  30  * since the kernel separates outputs into encoder and connector structures,
  31  * each with their own unique ID.  The program also allows test testing of the
  32  * memory management and mode setting APIs by allowing the user to specify a
  33  * connector and mode to use for mode setting.  If all works as expected, a
  34  * blue background should be painted on the monitor attached to the specified
  35  * connector after the selected mode is set.
  36  *
  37  * TODO: use cairo to write the mode info on the selected output once
  38  *       the mode has been programmed, along with possible test patterns.
  39  */
  40 #include "config.h"
  41
  42 #include <assert.h>
  43 #include <ctype.h>
  44 #include <stdbool.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <stdint.h>
  48 #include <inttypes.h>
  49 #include <unistd.h>
  50 #include <string.h>
  51 #include <errno.h>
  52 #include <sys/poll.h>
  53 #include <sys/time.h>
  54
  55 #include "xf86drm.h"
  56 #include "xf86drmMode.h"
  57 #include "drm_fourcc.h"
  58 #include "libkms.h"
  59
  60 #include "buffers.h"
  61
  62 struct crtc {
  63         drmModeCrtc *crtc;
  64         drmModeObjectProperties *props;
  65         drmModePropertyRes **props_info;
  66         drmModeModeInfo *mode;
  67 };
  68
  69 struct encoder {
  70         drmModeEncoder *encoder;
  71 };
  72
  73 struct connector {
  74         drmModeConnector *connector;
  75         drmModeObjectProperties *props;
  76         drmModePropertyRes **props_info;
  77 };
  78
  79 struct fb {
  80         drmModeFB *fb;
  81 };
  82
  83 struct plane {
  84         drmModePlane *plane;
  85         drmModeObjectProperties *props;
  86         drmModePropertyRes **props_info;
  87 };
  88
  89 struct resources {
  90         drmModeRes *res;
  91         drmModePlaneRes *plane_res;
  92
  93         struct crtc *crtcs;
  94         struct encoder *encoders;
  95         struct connector *connectors;
  96         struct fb *fbs;
  97         struct plane *planes;
  98 };
  99
 100 struct device {
 101         int fd;
 102
 103         struct resources *resources;
 104         struct kms_driver *kms;
 105
 106         struct {
 107                 unsigned int width;
 108                 unsigned int height;
 109
 110                 unsigned int fb_id;
 111                 struct kms_bo *bo;
 112         } mode;
 113 };
 114
 115 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
 116
 117 struct type_name {
 118         int type;
 119         const char *name;
 120 };
 121
 122 #define type_name_fn(res) \
 123 const char * res##_str(int type) {                      \
 124         unsigned int i;                                 \
 125         for (i = 0; i < ARRAY_SIZE(res##_names); i++) { \
 126                 if (res##_names[i].type == type)        \
 127                         return res##_names[i].name;     \
 128         }                                               \
 129         return "(invalid)";                             \
 130 }
 131
 132 struct type_name encoder_type_names[] = {
 133         { DRM_MODE_ENCODER_NONE, "none" },
 134         { DRM_MODE_ENCODER_DAC, "DAC" },
 135         { DRM_MODE_ENCODER_TMDS, "TMDS" },
 136         { DRM_MODE_ENCODER_LVDS, "LVDS" },
 137         { DRM_MODE_ENCODER_TVDAC, "TVDAC" },
 138 };
 139
 140 static type_name_fn(encoder_type)
 141
 142 struct type_name connector_status_names[] = {
 143         { DRM_MODE_CONNECTED, "connected" },
 144         { DRM_MODE_DISCONNECTED, "disconnected" },
 145         { DRM_MODE_UNKNOWNCONNECTION, "unknown" },
 146 };
 147
 148 static type_name_fn(connector_status)
 149
 150 struct type_name connector_type_names[] = {
 151         { DRM_MODE_CONNECTOR_Unknown, "unknown" },
 152         { DRM_MODE_CONNECTOR_VGA, "VGA" },
 153         { DRM_MODE_CONNECTOR_DVII, "DVI-I" },
 154         { DRM_MODE_CONNECTOR_DVID, "DVI-D" },
 155         { DRM_MODE_CONNECTOR_DVIA, "DVI-A" },
 156         { DRM_MODE_CONNECTOR_Composite, "composite" },
 157         { DRM_MODE_CONNECTOR_SVIDEO, "s-video" },
 158         { DRM_MODE_CONNECTOR_LVDS, "LVDS" },
 159         { DRM_MODE_CONNECTOR_Component, "component" },
 160         { DRM_MODE_CONNECTOR_9PinDIN, "9-pin DIN" },
 161         { DRM_MODE_CONNECTOR_DisplayPort, "DP" },
 162         { DRM_MODE_CONNECTOR_HDMIA, "HDMI-A" },
 163         { DRM_MODE_CONNECTOR_HDMIB, "HDMI-B" },
 164         { DRM_MODE_CONNECTOR_TV, "TV" },
 165         { DRM_MODE_CONNECTOR_eDP, "eDP" },
 166 };
 167
 168 static type_name_fn(connector_type)
 169
 170 #define bit_name_fn(res)                                        \
 171 const char * res##_str(int type) {                              \
 172         unsigned int i;                                         \
 173         const char *sep = "";                                   \
 174         for (i = 0; i < ARRAY_SIZE(res##_names); i++) {         \
 175                 if (type & (1 << i)) {                          \
 176                         printf("%s%s", sep, res##_names[i]);    \
 177                         sep = ", ";                             \
 178                 }                                               \
 179         }                                                       \
 180         return NULL;                                            \
 181 }
 182
 183 static const char *mode_type_names[] = {
 184         "builtin",
 185         "clock_c",
 186         "crtc_c",
 187         "preferred",
 188         "default",
 189         "userdef",
 190         "driver",
 191 };
 192
 193 static bit_name_fn(mode_type)
 194
 195 static const char *mode_flag_names[] = {
 196         "phsync",
 197         "nhsync",
 198         "pvsync",
 199         "nvsync",
 200         "interlace",
 201         "dblscan",
 202         "csync",
 203         "pcsync",
 204         "ncsync",
 205         "hskew",
 206         "bcast",
 207         "pixmux",
 208         "dblclk",
 209         "clkdiv2"
 210 };
 211
 212 static bit_name_fn(mode_flag)
 213
 214 static void dump_encoders(struct device *dev)
 215 {
 216         drmModeEncoder *encoder;
 217         int i;
 218
 219         printf("Encoders:\n");
 220         printf("id\tcrtc\ttype\tpossible crtcs\tpossible clones\t\n");
 221         for (i = 0; i < dev->resources->res->count_encoders; i++) {
 222                 encoder = dev->resources->encoders[i].encoder;
 223                 if (!encoder)
 224                         continue;
 225
 226                 printf("%d\t%d\t%s\t0x%08x\t0x%08x\n",
 227                        encoder->encoder_id,
 228                        encoder->crtc_id,
 229                        encoder_type_str(encoder->encoder_type),
 230                        encoder->possible_crtcs,
 231                        encoder->possible_clones);
 232         }
 233         printf("\n");
 234 }
 235
 236 static void dump_mode(drmModeModeInfo *mode)
 237 {
 238         printf("  %s %d %d %d %d %d %d %d %d %d",
 239                mode->name,
 240                mode->vrefresh,
 241                mode->hdisplay,
 242                mode->hsync_start,
 243                mode->hsync_end,
 244                mode->htotal,
 245                mode->vdisplay,
 246                mode->vsync_start,
 247                mode->vsync_end,
 248                mode->vtotal);
 249
 250         printf(" flags: ");
 251         mode_flag_str(mode->flags);
 252         printf("; type: ");
 253         mode_type_str(mode->type);
 254         printf("\n");
 255 }
 256
 257 static void dump_blob(struct device *dev, uint32_t blob_id)
 258 {
 259         uint32_t i;
 260         unsigned char *blob_data;
 261         drmModePropertyBlobPtr blob;
 262
 263         blob = drmModeGetPropertyBlob(dev->fd, blob_id);
 264         if (!blob)
 265                 return;
 266
 267         blob_data = blob->data;
 268
 269         for (i = 0; i < blob->length; i++) {
 270                 if (i % 16 == 0)
 271                         printf("\n\t\t\t");
 272                 printf("%.2hhx", blob_data[i]);
 273         }
 274         printf("\n");
 275
 276         drmModeFreePropertyBlob(blob);
 277 }
 278
 279 static void dump_prop(struct device *dev, drmModePropertyPtr prop,
 280                       uint32_t prop_id, uint64_t value)
 281 {
 282         int i;
 283         printf("\t%d", prop_id);
 284         if (!prop) {
 285                 printf("\n");
 286                 return;
 287         }
 288
 289         printf(" %s:\n", prop->name);
 290
 291         printf("\t\tflags:");
 292         if (prop->flags & DRM_MODE_PROP_PENDING)
 293                 printf(" pending");
 294         if (prop->flags & DRM_MODE_PROP_RANGE)
 295                 printf(" range");
 296         if (prop->flags & DRM_MODE_PROP_IMMUTABLE)
 297                 printf(" immutable");
 298         if (prop->flags & DRM_MODE_PROP_ENUM)
 299                 printf(" enum");
 300         if (prop->flags & DRM_MODE_PROP_BITMASK)
 301                 printf(" bitmask");
 302         if (prop->flags & DRM_MODE_PROP_BLOB)
 303                 printf(" blob");
 304         printf("\n");
 305
 306         if (prop->flags & DRM_MODE_PROP_RANGE) {
 307                 printf("\t\tvalues:");
 308                 for (i = 0; i < prop->count_values; i++)
 309                         printf(" %"PRIu64, prop->values[i]);
 310                 printf("\n");
 311         }
 312
 313         if (prop->flags & DRM_MODE_PROP_ENUM) {
 314                 printf("\t\tenums:");
 315                 for (i = 0; i < prop->count_enums; i++)
 316                         printf(" %s=%llu", prop->enums[i].name,
 317                                prop->enums[i].value);
 318                 printf("\n");
 319         } else if (prop->flags & DRM_MODE_PROP_BITMASK) {
 320                 printf("\t\tvalues:");
 321                 for (i = 0; i < prop->count_enums; i++)
 322                         printf(" %s=0x%llx", prop->enums[i].name,
 323                                (1LL << prop->enums[i].value));
 324                 printf("\n");
 325         } else {
 326                 assert(prop->count_enums == 0);
 327         }
 328
 329         if (prop->flags & DRM_MODE_PROP_BLOB) {
 330                 printf("\t\tblobs:\n");
 331                 for (i = 0; i < prop->count_blobs; i++)
 332                         dump_blob(dev, prop->blob_ids[i]);
 333                 printf("\n");
 334         } else {
 335                 assert(prop->count_blobs == 0);
 336         }
 337
 338         printf("\t\tvalue:");
 339         if (prop->flags & DRM_MODE_PROP_BLOB)
 340                 dump_blob(dev, value);
 341         else
 342                 printf(" %"PRIu64"\n", value);
 343 }
 344
 345 static void dump_connectors(struct device *dev)
 346 {
 347         int i, j;
 348
 349         printf("Connectors:\n");
 350         printf("id\tencoder\tstatus\t\ttype\tsize (mm)\tmodes\tencoders\n");
 351         for (i = 0; i < dev->resources->res->count_connectors; i++) {
 352                 struct connector *_connector = &dev->resources->connectors[i];
 353                 drmModeConnector *connector = _connector->connector;
 354                 if (!connector)
 355                         continue;
 356
 357                 printf("%d\t%d\t%s\t%s\t%dx%d\t\t%d\t",
 358                        connector->connector_id,
 359                        connector->encoder_id,
 360                        connector_status_str(connector->connection),
 361                        connector_type_str(connector->connector_type),
 362                        connector->mmWidth, connector->mmHeight,
 363                        connector->count_modes);
 364
 365                 for (j = 0; j < connector->count_encoders; j++)
 366                         printf("%s%d", j > 0 ? ", " : "", connector->encoders[j]);
 367                 printf("\n");
 368
 369                 if (connector->count_modes) {
 370                         printf("  modes:\n");
 371                         printf("\tname refresh (Hz) hdisp hss hse htot vdisp "
 372                                "vss vse vtot)\n");
 373                         for (j = 0; j < connector->count_modes; j++)
 374                                 dump_mode(&connector->modes[j]);
 375                 }
 376
 377                 if (_connector->props) {
 378                         printf("  props:\n");
 379                         for (j = 0; j < (int)_connector->props->count_props; j++)
 380                                 dump_prop(dev, _connector->props_info[j],
 381                                           _connector->props->props[j],
 382                                           _connector->props->prop_values[j]);
 383                 }
 384         }
 385         printf("\n");
 386 }
 387
 388 static void dump_crtcs(struct device *dev)
 389 {
 390         int i;
 391         uint32_t j;
 392
 393         printf("CRTCs:\n");
 394         printf("id\tfb\tpos\tsize\n");
 395         for (i = 0; i < dev->resources->res->count_crtcs; i++) {
 396                 struct crtc *_crtc = &dev->resources->crtcs[i];
 397                 drmModeCrtc *crtc = _crtc->crtc;
 398                 if (!crtc)
 399                         continue;
 400
 401                 printf("%d\t%d\t(%d,%d)\t(%dx%d)\n",
 402                        crtc->crtc_id,
 403                        crtc->buffer_id,
 404                        crtc->x, crtc->y,
 405                        crtc->width, crtc->height);
 406                 dump_mode(&crtc->mode);
 407
 408                 if (_crtc->props) {
 409                         printf("  props:\n");
 410                         for (j = 0; j < _crtc->props->count_props; j++)
 411                                 dump_prop(dev, _crtc->props_info[j],
 412                                           _crtc->props->props[j],
 413                                           _crtc->props->prop_values[j]);
 414                 } else {
 415                         printf("  no properties found\n");
 416                 }
 417         }
 418         printf("\n");
 419 }
 420
 421 static void dump_framebuffers(struct device *dev)
 422 {
 423         drmModeFB *fb;
 424         int i;
 425
 426         printf("Frame buffers:\n");
 427         printf("id\tsize\tpitch\n");
 428         for (i = 0; i < dev->resources->res->count_fbs; i++) {
 429                 fb = dev->resources->fbs[i].fb;
 430                 if (!fb)
 431                         continue;
 432
 433                 printf("%u\t(%ux%u)\t%u\n",
 434                        fb->fb_id,
 435                        fb->width, fb->height,
 436                        fb->pitch);
 437         }
 438         printf("\n");
 439 }
 440
 441 static void dump_planes(struct device *dev)
 442 {
 443         unsigned int i, j;
 444
 445         printf("Planes:\n");
 446         printf("id\tcrtc\tfb\tCRTC x,y\tx,y\tgamma size\tpossible crtcs\n");
 447
 448         if (!dev->resources->plane_res)
 449                 return;
 450
 451         for (i = 0; i < dev->resources->plane_res->count_planes; i++) {
 452                 struct plane *plane = &dev->resources->planes[i];
 453                 drmModePlane *ovr = plane->plane;
 454                 if (!ovr)
 455                         continue;
 456
 457                 printf("%d\t%d\t%d\t%d,%d\t\t%d,%d\t%-8d\t0x%08x\n",
 458                        ovr->plane_id, ovr->crtc_id, ovr->fb_id,
 459                        ovr->crtc_x, ovr->crtc_y, ovr->x, ovr->y,
 460                        ovr->gamma_size, ovr->possible_crtcs);
 461
 462                 if (!ovr->count_formats)
 463                         continue;
 464
 465                 printf("  formats:");
 466                 for (j = 0; j < ovr->count_formats; j++)
 467                         printf(" %4.4s", (char *)&ovr->formats[j]);
 468                 printf("\n");
 469
 470                 if (plane->props) {
 471                         printf("  props:\n");
 472                         for (j = 0; j < plane->props->count_props; j++)
 473                                 dump_prop(dev, plane->props_info[j],
 474                                           plane->props->props[j],
 475                                           plane->props->prop_values[j]);
 476                 } else {
 477                         printf("  no properties found\n");
 478                 }
 479         }
 480         printf("\n");
 481
 482         return;
 483 }
 484
 485 static void free_resources(struct resources *res)
 486 {
 487         if (!res)
 488                 return;
 489
 490 #define free_resource(_res, __res, type, Type)                                  \
 491         do {                                                                    \
 492                 int i;                                                          \
 493                 if (!(_res)->type##s)                                           \
 494                         break;                                                  \
 495                 for (i = 0; i < (int)(_res)->__res->count_##type##s; ++i) {     \
 496                         if (!(_res)->type##s[i].type)                           \
 497                                 break;                                          \
 498                         drmModeFree##Type((_res)->type##s[i].type);             \
 499                 }                                                               \
 500                 free((_res)->type##s);                                          \
 501         } while (0)
 502
 503 #define free_properties(_res, __res, type)                                      \
 504         do {                                                                    \
 505                 int i;                                                          \
 506                 for (i = 0; i < (int)(_res)->__res->count_##type##s; ++i) {     \
 507                         drmModeFreeObjectProperties(res->type##s[i].props);     \
 508                         free(res->type##s[i].props_info);                       \
 509                 }                                                               \
 510         } while (0)
 511
 512         if (res->res) {
 513                 free_properties(res, res, crtc);
 514
 515                 free_resource(res, res, crtc, Crtc);
 516                 free_resource(res, res, encoder, Encoder);
 517                 free_resource(res, res, connector, Connector);
 518                 free_resource(res, res, fb, FB);
 519
 520                 drmModeFreeResources(res->res);
 521         }
 522
 523         if (res->plane_res) {
 524                 free_properties(res, plane_res, plane);
 525
 526                 free_resource(res, plane_res, plane, Plane);
 527
 528                 drmModeFreePlaneResources(res->plane_res);
 529         }
 530
 531         free(res);
 532 }
 533
 534 static struct resources *get_resources(struct device *dev)
 535 {
 536         struct resources *res;
 537         int i;
 538
 539         res = malloc(sizeof *res);
 540         if (res == 0)
 541                 return NULL;
 542
 543         memset(res, 0, sizeof *res);
 544
 545         res->res = drmModeGetResources(dev->fd);
 546         if (!res->res) {
 547                 fprintf(stderr, "drmModeGetResources failed: %s\n",
 548                         strerror(errno));
 549                 goto error;
 550         }
 551
 552         res->crtcs = malloc(res->res->count_crtcs * sizeof *res->crtcs);
 553         res->encoders = malloc(res->res->count_encoders * sizeof *res->encoders);
 554         res->connectors = malloc(res->res->count_connectors * sizeof *res->connectors);
 555         res->fbs = malloc(res->res->count_fbs * sizeof *res->fbs);
 556
 557         if (!res->crtcs || !res->encoders || !res->connectors || !res->fbs)
 558                 goto error;
 559
 560         memset(res->crtcs , 0, res->res->count_crtcs * sizeof *res->crtcs);
 561         memset(res->encoders, 0, res->res->count_encoders * sizeof *res->encoders);
 562         memset(res->connectors, 0, res->res->count_connectors * sizeof *res->connectors);
 563         memset(res->fbs, 0, res->res->count_fbs * sizeof *res->fbs);
 564
 565 #define get_resource(_res, __res, type, Type)                                   \
 566         do {                                                                    \
 567                 int i;                                                          \
 568                 for (i = 0; i < (int)(_res)->__res->count_##type##s; ++i) {     \
 569                         (_res)->type##s[i].type =                               \
 570                                 drmModeGet##Type(dev->fd, (_res)->__res->type##s[i]); \
 571                         if (!(_res)->type##s[i].type)                           \
 572                                 fprintf(stderr, "could not get %s %i: %s\n",    \
 573                                         #type, (_res)->__res->type##s[i],       \
 574                                         strerror(errno));                       \
 575                 }                                                               \
 576         } while (0)
 577
 578         get_resource(res, res, crtc, Crtc);
 579         get_resource(res, res, encoder, Encoder);
 580         get_resource(res, res, connector, Connector);
 581         get_resource(res, res, fb, FB);
 582
 583 #define get_properties(_res, __res, type, Type)                                 \
 584         do {                                                                    \
 585                 int i;                                                          \
 586                 for (i = 0; i < (int)(_res)->__res->count_##type##s; ++i) {     \
 587                         struct type *obj = &res->type##s[i];                    \
 588                         unsigned int j;                                         \
 589                         obj->props =                                            \
 590                                 drmModeObjectGetProperties(dev->fd, obj->type->type##_id, \
 591                                                            DRM_MODE_OBJECT_##Type); \
 592                         if (!obj->props) {                                      \
 593                                 fprintf(stderr,                                 \
 594                                         "could not get %s %i properties: %s\n", \
 595                                         #type, obj->type->type##_id,            \
 596                                         strerror(errno));                       \
 597                                 continue;                                       \
 598                         }                                                       \
 599                         obj->props_info = malloc(obj->props->count_props *      \
 600                                                  sizeof *obj->props_info);      \
 601                         if (!obj->props_info)                                   \
 602                                 continue;                                       \
 603                         for (j = 0; j < obj->props->count_props; ++j)           \
 604                                 obj->props_info[j] =                            \
 605                                         drmModeGetProperty(dev->fd, obj->props->props[j]); \
 606                 }                                                               \
 607         } while (0)
 608
 609         get_properties(res, res, crtc, CRTC);
 610         get_properties(res, res, connector, CONNECTOR);
 611
 612         for (i = 0; i < res->res->count_crtcs; ++i)
 613                 res->crtcs[i].mode = &res->crtcs[i].crtc->mode;
 614
 615         res->plane_res = drmModeGetPlaneResources(dev->fd);
 616         if (!res->plane_res) {
 617                 fprintf(stderr, "drmModeGetPlaneResources failed: %s\n",
 618                         strerror(errno));
 619                 return res;
 620         }
 621
 622         res->planes = malloc(res->plane_res->count_planes * sizeof *res->planes);
 623         if (!res->planes)
 624                 goto error;
 625
 626         memset(res->planes, 0, res->plane_res->count_planes * sizeof *res->planes);
 627
 628         get_resource(res, plane_res, plane, Plane);
 629         get_properties(res, plane_res, plane, PLANE);
 630
 631         return res;
 632
 633 error:
 634         free_resources(res);
 635         return NULL;
 636 }
 637
 638 static int get_crtc_index(struct device *dev, uint32_t id)
 639 {
 640         int i;
 641
 642         for (i = 0; i < dev->resources->res->count_crtcs; ++i) {
 643                 drmModeCrtc *crtc = dev->resources->crtcs[i].crtc;
 644                 if (crtc && crtc->crtc_id == id)
 645                         return i;
 646         }
 647
 648         return -1;
 649 }
 650
 651 static drmModeConnector *get_connector_by_id(struct device *dev, uint32_t id)
 652 {
 653         drmModeConnector *connector;
 654         int i;
 655
 656         for (i = 0; i < dev->resources->res->count_connectors; i++) {
 657                 connector = dev->resources->connectors[i].connector;
 658                 if (connector && connector->connector_id == id)
 659                         return connector;
 660         }
 661
 662         return NULL;
 663 }
 664
 665 static drmModeEncoder *get_encoder_by_id(struct device *dev, uint32_t id)
 666 {
 667         drmModeEncoder *encoder;
 668         int i;
 669
 670         for (i = 0; i < dev->resources->res->count_encoders; i++) {
 671                 encoder = dev->resources->encoders[i].encoder;
 672                 if (encoder && encoder->encoder_id == id)
 673                         return encoder;
 674         }
 675
 676         return NULL;
 677 }
 678
 679 /* -----------------------------------------------------------------------------
 680  * Pipes and planes
 681  */
 682
 683 /*
 684  * Mode setting with the kernel interfaces is a bit of a chore.
 685  * First you have to find the connector in question and make sure the
 686  * requested mode is available.
 687  * Then you need to find the encoder attached to that connector so you
 688  * can bind it with a free crtc.
 689  */
 690 struct pipe_arg {
 691         uint32_t *con_ids;
 692         unsigned int num_cons;
 693         uint32_t crtc_id;
 694         char mode_str[64];
 695         char format_str[5];
 696         unsigned int fourcc;
 697         drmModeModeInfo *mode;
 698         struct crtc *crtc;
 699         unsigned int fb_id[2], current_fb_id;
 700         struct timeval start;
 701
 702         int swap_count;
 703 };
 704
 705 struct plane_arg {
 706         uint32_t crtc_id;  /* the id of CRTC to bind to */
 707         bool has_position;
 708         int32_t x, y;
 709         uint32_t w, h;
 710         double scale;
 711         unsigned int fb_id;
 712         char format_str[5]; /* need to leave room for terminating \0 */
 713         unsigned int fourcc;
 714 };
 715
 716 static drmModeModeInfo *
 717 connector_find_mode(struct device *dev, uint32_t con_id, const char *mode_str)
 718 {
 719         drmModeConnector *connector;
 720         drmModeModeInfo *mode;
 721         int i;
 722
 723         connector = get_connector_by_id(dev, con_id);
 724         if (!connector || !connector->count_modes)
 725                 return NULL;
 726
 727         for (i = 0; i < connector->count_modes; i++) {
 728                 mode = &connector->modes[i];
 729                 if (!strcmp(mode->name, mode_str))
 730                         return mode;
 731         }
 732
 733         return NULL;
 734 }
 735
 736 static struct crtc *pipe_find_crtc(struct device *dev, struct pipe_arg *pipe)
 737 {
 738         uint32_t possible_crtcs = ~0;
 739         uint32_t active_crtcs = 0;
 740         unsigned int crtc_idx;
 741         unsigned int i;
 742         int j;
 743
 744         for (i = 0; i < pipe->num_cons; ++i) {
 745                 uint32_t crtcs_for_connector = 0;
 746                 drmModeConnector *connector;
 747                 drmModeEncoder *encoder;
 748                 int idx;
 749
 750                 connector = get_connector_by_id(dev, pipe->con_ids[i]);
 751                 if (!connector)
 752                         return NULL;
 753
 754                 for (j = 0; j < connector->count_encoders; ++j) {
 755                         encoder = get_encoder_by_id(dev, connector->encoders[j]);
 756                         if (!encoder)
 757                                 continue;
 758
 759                         crtcs_for_connector |= encoder->possible_crtcs;
 760
 761                         idx = get_crtc_index(dev, encoder->crtc_id);
 762                         if (idx >= 0)
 763                                 active_crtcs |= 1 << idx;
 764                 }
 765
 766                 possible_crtcs &= crtcs_for_connector;
 767         }
 768
 769         if (!possible_crtcs)
 770                 return NULL;
 771
 772         /* Return the first possible and active CRTC if one exists, or the first
 773          * possible CRTC otherwise.
 774          */
 775         if (possible_crtcs & active_crtcs)
 776                 crtc_idx = ffs(possible_crtcs & active_crtcs);
 777         else
 778                 crtc_idx = ffs(possible_crtcs);
 779
 780         return &dev->resources->crtcs[crtc_idx - 1];
 781 }
 782
 783 static int pipe_find_crtc_and_mode(struct device *dev, struct pipe_arg *pipe)
 784 {
 785         drmModeModeInfo *mode = NULL;
 786         int i;
 787
 788         pipe->mode = NULL;
 789
 790         for (i = 0; i < (int)pipe->num_cons; i++) {
 791                 mode = connector_find_mode(dev, pipe->con_ids[i],
 792                                            pipe->mode_str);
 793                 if (mode == NULL) {
 794                         fprintf(stderr,
 795                                 "failed to find mode \"%s\" for connector %u\n",
 796                                 pipe->mode_str, pipe->con_ids[i]);
 797                         return -EINVAL;
 798                 }
 799         }
 800
 801         /* If the CRTC ID was specified, get the corresponding CRTC. Otherwise
 802          * locate a CRTC that can be attached to all the connectors.
 803          */
 804         if (pipe->crtc_id != (uint32_t)-1) {
 805                 for (i = 0; i < dev->resources->res->count_crtcs; i++) {
 806                         struct crtc *crtc = &dev->resources->crtcs[i];
 807
 808                         if (pipe->crtc_id == crtc->crtc->crtc_id) {
 809                                 pipe->crtc = crtc;
 810                                 break;
 811                         }
 812                 }
 813         } else {
 814                 pipe->crtc = pipe_find_crtc(dev, pipe);
 815         }
 816
 817         if (!pipe->crtc) {
 818                 fprintf(stderr, "failed to find CRTC for pipe\n");
 819                 return -EINVAL;
 820         }
 821
 822         pipe->mode = mode;
 823         pipe->crtc->mode = mode;
 824
 825         return 0;
 826 }
 827
 828 /* -----------------------------------------------------------------------------
 829  * Properties
 830  */
 831
 832 struct property_arg {
 833         uint32_t obj_id;
 834         uint32_t obj_type;
 835         char name[DRM_PROP_NAME_LEN+1];
 836         uint32_t prop_id;
 837         uint64_t value;
 838 };
 839
 840 static void set_property(struct device *dev, struct property_arg *p)
 841 {
 842         drmModeObjectProperties *props = NULL;
 843         drmModePropertyRes **props_info = NULL;
 844         const char *obj_type;
 845         int ret;
 846         int i;
 847
 848         p->obj_type = 0;
 849         p->prop_id = 0;
 850
 851 #define find_object(_res, __res, type, Type)                                    \
 852         do {                                                                    \
 853                 for (i = 0; i < (int)(_res)->__res->count_##type##s; ++i) {     \
 854                         struct type *obj = &(_res)->type##s[i];                 \
 855                         if (obj->type->type##_id != p->obj_id)                  \
 856                                 continue;                                       \
 857                         p->obj_type = DRM_MODE_OBJECT_##Type;                   \
 858                         obj_type = #Type;                                       \
 859                         props = obj->props;                                     \
 860                         props_info = obj->props_info;                           \
 861                 }                                                               \
 862         } while(0)                                                              \
 863
 864         find_object(dev->resources, res, crtc, CRTC);
 865         if (p->obj_type == 0)
 866                 find_object(dev->resources, res, connector, CONNECTOR);
 867         if (p->obj_type == 0)
 868                 find_object(dev->resources, plane_res, plane, PLANE);
 869         if (p->obj_type == 0) {
 870                 fprintf(stderr, "Object %i not found, can't set property\n",
 871                         p->obj_id);
 872                         return;
 873         }
 874
 875         if (!props) {
 876                 fprintf(stderr, "%s %i has no properties\n",
 877                         obj_type, p->obj_id);
 878                 return;
 879         }
 880
 881         for (i = 0; i < (int)props->count_props; ++i) {
 882                 if (!props_info[i])
 883                         continue;
 884                 if (strcmp(props_info[i]->name, p->name) == 0)
 885                         break;
 886         }
 887
 888         if (i == (int)props->count_props) {
 889                 fprintf(stderr, "%s %i has no %s property\n",
 890                         obj_type, p->obj_id, p->name);
 891                 return;
 892         }
 893
 894         p->prop_id = props->props[i];
 895
 896         ret = drmModeObjectSetProperty(dev->fd, p->obj_id, p->obj_type,
 897                                        p->prop_id, p->value);
 898         if (ret < 0)
 899                 fprintf(stderr, "failed to set %s %i property %s to %" PRIu64 ": %s\n",
 900                         obj_type, p->obj_id, p->name, p->value, strerror(errno));
 901 }
 902
 903 /* -------------------------------------------------------------------------- */
 904
 905 static void
 906 page_flip_handler(int fd, unsigned int frame,
 907                   unsigned int sec, unsigned int usec, void *data)
 908 {
 909         struct pipe_arg *pipe;
 910         unsigned int new_fb_id;
 911         struct timeval end;
 912         double t;
 913
 914         pipe = data;
 915         if (pipe->current_fb_id == pipe->fb_id[0])
 916                 new_fb_id = pipe->fb_id[1];
 917         else
 918                 new_fb_id = pipe->fb_id[0];
 919
 920         drmModePageFlip(fd, pipe->crtc->crtc->crtc_id, new_fb_id,
 921                         DRM_MODE_PAGE_FLIP_EVENT, pipe);
 922         pipe->current_fb_id = new_fb_id;
 923         pipe->swap_count++;
 924         if (pipe->swap_count == 60) {
 925                 gettimeofday(&end, NULL);
 926                 t = end.tv_sec + end.tv_usec * 1e-6 -
 927                         (pipe->start.tv_sec + pipe->start.tv_usec * 1e-6);
 928                 fprintf(stderr, "freq: %.02fHz\n", pipe->swap_count / t);
 929                 pipe->swap_count = 0;
 930                 pipe->start = end;
 931         }
 932 }
 933
 934 static int set_plane(struct device *dev, struct plane_arg *p)
 935 {
 936         drmModePlane *ovr;
 937         uint32_t handles[4], pitches[4], offsets[4] = {0}; /* we only use [0] */
 938         uint32_t plane_id = 0;
 939         struct kms_bo *plane_bo;
 940         uint32_t plane_flags = 0;
 941         int crtc_x, crtc_y, crtc_w, crtc_h;
 942         struct crtc *crtc = NULL;
 943         unsigned int pipe;
 944         unsigned int i;
 945
 946         /* Find an unused plane which can be connected to our CRTC. Find the
 947          * CRTC index first, then iterate over available planes.
 948          */
 949         for (i = 0; i < (unsigned int)dev->resources->res->count_crtcs; i++) {
 950                 if (p->crtc_id == dev->resources->res->crtcs[i]) {
 951                         crtc = &dev->resources->crtcs[i];
 952                         pipe = i;
 953                         break;
 954                 }
 955         }
 956
 957         if (!crtc) {
 958                 fprintf(stderr, "CRTC %u not found\n", p->crtc_id);
 959                 return -1;
 960         }
 961
 962         for (i = 0; i < dev->resources->plane_res->count_planes && !plane_id; i++) {
 963                 ovr = dev->resources->planes[i].plane;
 964                 if (!ovr)
 965                         continue;
 966
 967                 if ((ovr->possible_crtcs & (1 << pipe)) && !ovr->crtc_id)
 968                         plane_id = ovr->plane_id;
 969         }
 970
 971         if (!plane_id) {
 972                 fprintf(stderr, "no unused plane available for CRTC %u\n",
 973                         crtc->crtc->crtc_id);
 974                 return -1;
 975         }
 976
 977         fprintf(stderr, "testing %dx%d@%s overlay plane %u\n",
 978                 p->w, p->h, p->format_str, plane_id);
 979
 980         plane_bo = create_test_buffer(dev->kms, p->fourcc, p->w, p->h, handles,
 981                                       pitches, offsets, PATTERN_TILES);
 982         if (plane_bo == NULL)
 983                 return -1;
 984
 985         /* just use single plane format for now.. */
 986         if (drmModeAddFB2(dev->fd, p->w, p->h, p->fourcc,
 987                         handles, pitches, offsets, &p->fb_id, plane_flags)) {
 988                 fprintf(stderr, "failed to add fb: %s\n", strerror(errno));
 989                 return -1;
 990         }
 991
 992         crtc_w = p->w * p->scale;
 993         crtc_h = p->h * p->scale;
 994         if (!p->has_position) {
 995                 /* Default to the middle of the screen */
 996                 crtc_x = (crtc->mode->hdisplay - crtc_w) / 2;
 997                 crtc_y = (crtc->mode->vdisplay - crtc_h) / 2;
 998         } else {
 999                 crtc_x = p->x;
1000                 crtc_y = p->y;
1001         }
1002
1003         /* note src coords (last 4 args) are in Q16 format */
1004         if (drmModeSetPlane(dev->fd, plane_id, crtc->crtc->crtc_id, p->fb_id,
1005                             plane_flags, crtc_x, crtc_y, crtc_w, crtc_h,
1006                             0, 0, p->w << 16, p->h << 16)) {
1007                 fprintf(stderr, "failed to enable plane: %s\n",
1008                         strerror(errno));
1009                 return -1;
1010         }
1011
1012         ovr->crtc_id = crtc->crtc->crtc_id;
1013
1014         return 0;
1015 }
1016
1017 static void set_mode(struct device *dev, struct pipe_arg *pipes, unsigned int count)
1018 {
1019         uint32_t handles[4], pitches[4], offsets[4] = {0}; /* we only use [0] */
1020         unsigned int fb_id;
1021         struct kms_bo *bo;
1022         unsigned int i;
1023         unsigned int j;
1024         int ret, x;
1025
1026         dev->mode.width = 0;
1027         dev->mode.height = 0;
1028
1029         for (i = 0; i < count; i++) {
1030                 struct pipe_arg *pipe = &pipes[i];
1031
1032                 ret = pipe_find_crtc_and_mode(dev, pipe);
1033                 if (ret < 0)
1034                         continue;
1035
1036                 dev->mode.width += pipe->mode->hdisplay;
1037                 if (dev->mode.height < pipe->mode->vdisplay)
1038                         dev->mode.height = pipe->mode->vdisplay;
1039         }
1040
1041         bo = create_test_buffer(dev->kms, pipes[0].fourcc,
1042                                 dev->mode.width, dev->mode.height,
1043                                 handles, pitches, offsets, PATTERN_SMPTE);
1044         if (bo == NULL)
1045                 return;
1046
1047         ret = drmModeAddFB2(dev->fd, dev->mode.width, dev->mode.height,
1048                             pipes[0].fourcc, handles, pitches, offsets, &fb_id, 0);
1049         if (ret) {
1050                 fprintf(stderr, "failed to add fb (%ux%u): %s\n",
1051                         dev->mode.width, dev->mode.height, strerror(errno));
1052                 return;
1053         }
1054
1055         x = 0;
1056         for (i = 0; i < count; i++) {
1057                 struct pipe_arg *pipe = &pipes[i];
1058
1059                 if (pipe->mode == NULL)
1060                         continue;
1061
1062                 printf("setting mode %s@%s on connectors ",
1063                        pipe->mode_str, pipe->format_str);
1064                 for (j = 0; j < pipe->num_cons; ++j)
1065                         printf("%u, ", pipe->con_ids[j]);
1066                 printf("crtc %d\n", pipe->crtc->crtc->crtc_id);
1067
1068                 ret = drmModeSetCrtc(dev->fd, pipe->crtc->crtc->crtc_id, fb_id,
1069                                      x, 0, pipe->con_ids, pipe->num_cons,
1070                                      pipe->mode);
1071
1072                 /* XXX: Actually check if this is needed */
1073                 drmModeDirtyFB(dev->fd, fb_id, NULL, 0);
1074
1075                 x += pipe->mode->hdisplay;
1076
1077                 if (ret) {
1078                         fprintf(stderr, "failed to set mode: %s\n", strerror(errno));
1079                         return;
1080                 }
1081         }
1082
1083         dev->mode.bo = bo;
1084         dev->mode.fb_id = fb_id;
1085 }
1086
1087 static void set_planes(struct device *dev, struct plane_arg *p, unsigned int count)
1088 {
1089         unsigned int i;
1090
1091         /* set up planes/overlays */
1092         for (i = 0; i < count; i++)
1093                 if (set_plane(dev, &p[i]))
1094                         return;
1095 }
1096
1097 static void test_page_flip(struct device *dev, struct pipe_arg *pipes, unsigned int count)
1098 {
1099         uint32_t handles[4], pitches[4], offsets[4] = {0}; /* we only use [0] */
1100         unsigned int other_fb_id;
1101         struct kms_bo *other_bo;
1102         drmEventContext evctx;
1103         unsigned int i;
1104         int ret;
1105
1106         other_bo = create_test_buffer(dev->kms, pipes[0].fourcc,
1107                                       dev->mode.width, dev->mode.height,
1108                                       handles, pitches, offsets, PATTERN_PLAIN);
1109         if (other_bo == NULL)
1110                 return;
1111
1112         ret = drmModeAddFB2(dev->fd, dev->mode.width, dev->mode.height,
1113                             pipes[0].fourcc, handles, pitches, offsets,
1114                             &other_fb_id, 0);
1115         if (ret) {
1116                 fprintf(stderr, "failed to add fb: %s\n", strerror(errno));
1117                 return;
1118         }
1119
1120         for (i = 0; i < count; i++) {
1121                 struct pipe_arg *pipe = &pipes[i];
1122
1123                 if (pipe->mode == NULL)
1124                         continue;
1125
1126                 ret = drmModePageFlip(dev->fd, pipe->crtc->crtc->crtc_id,
1127                                       other_fb_id, DRM_MODE_PAGE_FLIP_EVENT,
1128                                       pipe);
1129                 if (ret) {
1130                         fprintf(stderr, "failed to page flip: %s\n", strerror(errno));
1131                         return;
1132                 }
1133                 gettimeofday(&pipe->start, NULL);
1134                 pipe->swap_count = 0;
1135                 pipe->fb_id[0] = dev->mode.fb_id;
1136                 pipe->fb_id[1] = other_fb_id;
1137                 pipe->current_fb_id = other_fb_id;
1138         }
1139
1140         memset(&evctx, 0, sizeof evctx);
1141         evctx.version = DRM_EVENT_CONTEXT_VERSION;
1142         evctx.vblank_handler = NULL;
1143         evctx.page_flip_handler = page_flip_handler;
1144
1145         while (1) {
1146 #if 0
1147                 struct pollfd pfd[2];
1148
1149                 pfd[0].fd = 0;
1150                 pfd[0].events = POLLIN;
1151                 pfd[1].fd = fd;
1152                 pfd[1].events = POLLIN;
1153
1154                 if (poll(pfd, 2, -1) < 0) {
1155                         fprintf(stderr, "poll error\n");
1156                         break;
1157                 }
1158
1159                 if (pfd[0].revents)
1160                         break;
1161 #else
1162                 struct timeval timeout = { .tv_sec = 3, .tv_usec = 0 };
1163                 fd_set fds;
1164                 int ret;
1165
1166                 FD_ZERO(&fds);
1167                 FD_SET(0, &fds);
1168                 FD_SET(dev->fd, &fds);
1169                 ret = select(dev->fd + 1, &fds, NULL, NULL, &timeout);
1170
1171                 if (ret <= 0) {
1172                         fprintf(stderr, "select timed out or error (ret %d)\n",
1173                                 ret);
1174                         continue;
1175                 } else if (FD_ISSET(0, &fds)) {
1176                         break;
1177                 }
1178 #endif
1179
1180                 drmHandleEvent(dev->fd, &evctx);
1181         }
1182
1183         kms_bo_destroy(&other_bo);
1184 }
1185
1186 #define min(a, b)       ((a) < (b) ? (a) : (b))
1187
1188 static int parse_connector(struct pipe_arg *pipe, const char *arg)
1189 {
1190         unsigned int len;
1191         unsigned int i;
1192         const char *p;
1193         char *endp;
1194
1195         pipe->crtc_id = (uint32_t)-1;
1196         strcpy(pipe->format_str, "XR24");
1197
1198         /* Count the number of connectors and allocate them. */
1199         pipe->num_cons = 1;
1200         for (p = arg; isdigit(*p) || *p == ','; ++p) {
1201                 if (*p == ',')
1202                         pipe->num_cons++;
1203         }
1204
1205         pipe->con_ids = malloc(pipe->num_cons * sizeof *pipe->con_ids);
1206         if (pipe->con_ids == NULL)
1207                 return -1;
1208
1209         /* Parse the connectors. */
1210         for (i = 0, p = arg; i < pipe->num_cons; ++i, p = endp + 1) {
1211                 pipe->con_ids[i] = strtoul(p, &endp, 10);
1212                 if (*endp != ',')
1213                         break;
1214         }
1215
1216         if (i != pipe->num_cons - 1)
1217                 return -1;
1218
1219         /* Parse the remaining parameters. */
1220         if (*endp == '@') {
1221                 arg = endp + 1;
1222                 pipe->crtc_id = strtoul(arg, &endp, 10);
1223         }
1224         if (*endp != ':')
1225                 return -1;
1226
1227         arg = endp + 1;
1228
1229         p = strchrnul(arg, '@');
1230         len = min(sizeof pipe->mode_str - 1, (unsigned int)(p - arg));
1231         strncpy(pipe->mode_str, arg, len);
1232         pipe->mode_str[len] = '\0';
1233
1234         if (*p == '@') {
1235                 strncpy(pipe->format_str, p + 1, 4);
1236                 pipe->format_str[4] = '\0';
1237         }
1238
1239         pipe->fourcc = format_fourcc(pipe->format_str);
1240         if (pipe->fourcc == 0)  {
1241                 fprintf(stderr, "unknown format %s\n", pipe->format_str);
1242                 return -1;
1243         }
1244
1245         return 0;
1246 }
1247
1248 static int parse_plane(struct plane_arg *plane, const char *p)
1249 {
1250         char *end;
1251
1252         memset(plane, 0, sizeof *plane);
1253
1254         plane->crtc_id = strtoul(p, &end, 10);
1255         if (*end != ':')
1256                 return -EINVAL;
1257
1258         p = end + 1;
1259         plane->w = strtoul(p, &end, 10);
1260         if (*end != 'x')
1261                 return -EINVAL;
1262
1263         p = end + 1;
1264         plane->h = strtoul(p, &end, 10);
1265
1266         if (*end == '+' || *end == '-') {
1267                 plane->x = strtol(end, &end, 10);
1268                 if (*end != '+' && *end != '-')
1269                         return -EINVAL;
1270                 plane->y = strtol(end, &end, 10);
1271
1272                 plane->has_position = true;
1273         }
1274
1275         if (*end == '*') {
1276                 p = end + 1;
1277                 plane->scale = strtod(p, &end);
1278                 if (plane->scale <= 0.0)
1279                         return -EINVAL;
1280         } else {
1281                 plane->scale = 1.0;
1282         }
1283
1284         if (*end == '@') {
1285                 p = end + 1;
1286                 if (strlen(p) != 4)
1287                         return -EINVAL;
1288
1289                 strcpy(plane->format_str, p);
1290         } else {
1291                 strcpy(plane->format_str, "XR24");
1292         }
1293
1294         plane->fourcc = format_fourcc(plane->format_str);
1295         if (plane->fourcc == 0) {
1296                 fprintf(stderr, "unknown format %s\n", plane->format_str);
1297                 return -EINVAL;
1298         }
1299
1300         return 0;
1301 }
1302
1303 static int parse_property(struct property_arg *p, const char *arg)
1304 {
1305         if (sscanf(arg, "%d:%32[^:]:%" SCNu64, &p->obj_id, p->name, &p->value) != 3)
1306                 return -1;
1307
1308         p->obj_type = 0;
1309         p->name[DRM_PROP_NAME_LEN] = '\0';
1310
1311         return 0;
1312 }
1313
1314 static void usage(char *name)
1315 {
1316         fprintf(stderr, "usage: %s [-cDdefMPpsvw]\n", name);
1317
1318         fprintf(stderr, "\n Query options:\n\n");
1319         fprintf(stderr, "\t-c\tlist connectors\n");
1320         fprintf(stderr, "\t-e\tlist encoders\n");
1321         fprintf(stderr, "\t-f\tlist framebuffers\n");
1322         fprintf(stderr, "\t-p\tlist CRTCs and planes (pipes)\n");
1323
1324         fprintf(stderr, "\n Test options:\n\n");
1325         fprintf(stderr, "\t-P <crtc_id>:<w>x<h>[+<x>+<y>][*<scale>][@<format>]\tset a plane\n");
1326         fprintf(stderr, "\t-s <connector_id>[,<connector_id>][@<crtc_id>]:<mode>[@<format>]\tset a mode\n");
1327         fprintf(stderr, "\t-v\ttest vsynced page flipping\n");
1328         fprintf(stderr, "\t-w <obj_id>:<prop_name>:<value>\tset property\n");
1329
1330         fprintf(stderr, "\n Generic options:\n\n");
1331         fprintf(stderr, "\t-d\tdrop master after mode set\n");
1332         fprintf(stderr, "\t-M module\tuse the given driver\n");
1333         fprintf(stderr, "\t-D device\tuse the given device\n");
1334
1335         fprintf(stderr, "\n\tDefault is to dump all info.\n");
1336         exit(0);
1337 }
1338
1339 static int page_flipping_supported(void)
1340 {
1341         /*FIXME: generic ioctl needed? */
1342         return 1;
1343 #if 0
1344         int ret, value;
1345         struct drm_i915_getparam gp;
1346
1347         gp.param = I915_PARAM_HAS_PAGEFLIPPING;
1348         gp.value = &value;
1349
1350         ret = drmCommandWriteRead(fd, DRM_I915_GETPARAM, &gp, sizeof(gp));
1351         if (ret) {
1352                 fprintf(stderr, "drm_i915_getparam: %m\n");
1353                 return 0;
1354         }
1355
1356         return *gp.value;
1357 #endif
1358 }
1359
1360 static char optstr[] = "cdD:efM:P:ps:vw:";
1361
1362 int main(int argc, char **argv)
1363 {
1364         struct device dev;
1365
1366         int c;
1367         int encoders = 0, connectors = 0, crtcs = 0, planes = 0, framebuffers = 0;
1368         int drop_master = 0;
1369         int test_vsync = 0;
1370         const char *modules[] = { "i915", "radeon", "nouveau", "vmwgfx", "omapdrm", "exynos", "tilcdc", "msm" };
1371         char *device = NULL;
1372         char *module = NULL;
1373         unsigned int i;
1374         int count = 0, plane_count = 0;
1375         unsigned int prop_count = 0;
1376         struct pipe_arg *pipe_args = NULL;
1377         struct plane_arg *plane_args = NULL;
1378         struct property_arg *prop_args = NULL;
1379         unsigned int args = 0;
1380         int ret;
1381
1382         memset(&dev, 0, sizeof dev);
1383
1384         opterr = 0;
1385         while ((c = getopt(argc, argv, optstr)) != -1) {
1386                 args++;
1387
1388                 switch (c) {
1389                 case 'c':
1390                         connectors = 1;
1391                         break;
1392                 case 'D':
1393                         device = optarg;
1394                         args--;
1395                         break;
1396                 case 'd':
1397                         drop_master = 1;
1398                         break;
1399                 case 'e':
1400                         encoders = 1;
1401                         break;
1402                 case 'f':
1403                         framebuffers = 1;
1404                         break;
1405                 case 'M':
1406                         module = optarg;
1407                         /* Preserve the default behaviour of dumping all information. */
1408                         args--;
1409                         break;
1410                 case 'P':
1411                         plane_args = realloc(plane_args,
1412                                              (plane_count + 1) * sizeof *plane_args);
1413                         if (plane_args == NULL) {
1414                                 fprintf(stderr, "memory allocation failed\n");
1415                                 return 1;
1416                         }
1417
1418                         if (parse_plane(&plane_args[plane_count], optarg) < 0)
1419                                 usage(argv[0]);
1420
1421                         plane_count++;
1422                         break;
1423                 case 'p':
1424                         crtcs = 1;
1425                         planes = 1;
1426                         break;
1427                 case 's':
1428                         pipe_args = realloc(pipe_args,
1429                                             (count + 1) * sizeof *pipe_args);
1430                         if (pipe_args == NULL) {
1431                                 fprintf(stderr, "memory allocation failed\n");
1432                                 return 1;
1433                         }
1434
1435                         if (parse_connector(&pipe_args[count], optarg) < 0)
1436                                 usage(argv[0]);
1437
1438                         count++;
1439                         break;
1440                 case 'v':
1441                         test_vsync = 1;
1442                         break;
1443                 case 'w':
1444                         prop_args = realloc(prop_args,
1445                                            (prop_count + 1) * sizeof *prop_args);
1446                         if (prop_args == NULL) {
1447                                 fprintf(stderr, "memory allocation failed\n");
1448                                 return 1;
1449                         }
1450
1451                         if (parse_property(&prop_args[prop_count], optarg) < 0)
1452                                 usage(argv[0]);
1453
1454                         prop_count++;
1455                         break;
1456                 default:
1457                         usage(argv[0]);
1458                         break;
1459                 }
1460         }
1461
1462         if (!args)
1463                 encoders = connectors = crtcs = planes = framebuffers = 1;
1464
1465         if (module) {
1466                 dev.fd = drmOpen(module, device);
1467                 if (dev.fd < 0) {
1468                         fprintf(stderr, "failed to open device '%s'.\n", module);
1469                         return 1;
1470                 }
1471         } else {
1472                 for (i = 0; i < ARRAY_SIZE(modules); i++) {
1473                         printf("trying to open device '%s'...", modules[i]);
1474                         dev.fd = drmOpen(modules[i], device);
1475                         if (dev.fd < 0) {
1476                                 printf("failed.\n");
1477                         } else {
1478                                 printf("success.\n");
1479                                 break;
1480                         }
1481                 }
1482
1483                 if (dev.fd < 0) {
1484                         fprintf(stderr, "no device found.\n");
1485                         return 1;
1486                 }
1487         }
1488
1489         if (test_vsync && !page_flipping_supported()) {
1490                 fprintf(stderr, "page flipping not supported by drm.\n");
1491                 return -1;
1492         }
1493
1494         if (test_vsync && !count) {
1495                 fprintf(stderr, "page flipping requires at least one -s option.\n");
1496                 return -1;
1497         }
1498
1499         dev.resources = get_resources(&dev);
1500         if (!dev.resources) {
1501                 drmClose(dev.fd);
1502                 return 1;
1503         }
1504
1505 #define dump_resource(dev, res) if (res) dump_##res(dev)
1506
1507         dump_resource(&dev, encoders);
1508         dump_resource(&dev, connectors);
1509         dump_resource(&dev, crtcs);
1510         dump_resource(&dev, planes);
1511         dump_resource(&dev, framebuffers);
1512
1513         for (i = 0; i < prop_count; ++i)
1514                 set_property(&dev, &prop_args[i]);
1515
1516         if (count || plane_count) {
1517                 ret = kms_create(dev.fd, &dev.kms);
1518                 if (ret) {
1519                         fprintf(stderr, "failed to create kms driver: %s\n",
1520                                 strerror(-ret));
1521                         return 1;
1522                 }
1523
1524                 if (count)
1525                         set_mode(&dev, pipe_args, count);
1526
1527                 if (plane_count)
1528                         set_planes(&dev, plane_args, plane_count);
1529
1530                 if (test_vsync)
1531                         test_page_flip(&dev, pipe_args, count);
1532
1533                 if (drop_master)
1534                         drmDropMaster(dev.fd);
1535
1536                 kms_bo_destroy(&dev.mode.bo);
1537                 kms_destroy(&dev.kms);
1538
1539                 getchar();
1540         }
1541
1542         free_resources(dev.resources);
1543
1544         return 0;
1545 }