adb/framebuffer_service.cpp

   1 /*
   2  * Copyright (C) 2007 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <errno.h>
  18 #include <fcntl.h>
  19 #include <linux/fb.h>
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <sys/ioctl.h>
  24 #include <sys/mman.h>
  25 #include <sys/types.h>
  26 #include <sys/wait.h>
  27 #include <unistd.h>
  28
  29 #include "sysdeps.h"
  30
  31 #include "adb.h"
  32 #include "adb_io.h"
  33 #include "fdevent.h"
  34
  35 /* TODO:
  36 ** - sync with vsync to avoid tearing
  37 */
  38 /* This version number defines the format of the fbinfo struct.
  39    It must match versioning in ddms where this data is consumed. */
  40 #define DDMS_RAWIMAGE_VERSION 1
  41 struct fbinfo {
  42     unsigned int version;
  43     unsigned int bpp;
  44     unsigned int size;
  45     unsigned int width;
  46     unsigned int height;
  47     unsigned int red_offset;
  48     unsigned int red_length;
  49     unsigned int blue_offset;
  50     unsigned int blue_length;
  51     unsigned int green_offset;
  52     unsigned int green_length;
  53     unsigned int alpha_offset;
  54     unsigned int alpha_length;
  55 } __attribute__((packed));
  56
  57 void framebuffer_service(int fd, void *cookie)
  58 {
  59     struct fbinfo fbinfo;
  60     unsigned int i, bsize;
  61     char buf[640];
  62     int fd_screencap;
  63     int w, h, f;
  64     int fds[2];
  65     pid_t pid;
  66
  67     if (pipe2(fds, O_CLOEXEC) < 0) goto pipefail;
  68
  69     pid = fork();
  70     if (pid < 0) goto done;
  71
  72     if (pid == 0) {
  73         dup2(fds[1], STDOUT_FILENO);
  74         adb_close(fds[0]);
  75         adb_close(fds[1]);
  76         const char* command = "screencap";
  77         const char *args[2] = {command, NULL};
  78         execvp(command, (char**)args);
  79         exit(1);
  80     }
  81
  82     adb_close(fds[1]);
  83     fd_screencap = fds[0];
  84
  85     /* read w, h & format */
  86     if(!ReadFdExactly(fd_screencap, &w, 4)) goto done;
  87     if(!ReadFdExactly(fd_screencap, &h, 4)) goto done;
  88     if(!ReadFdExactly(fd_screencap, &f, 4)) goto done;
  89
  90     fbinfo.version = DDMS_RAWIMAGE_VERSION;
  91     /* see hardware/hardware.h */
  92     switch (f) {
  93         case 1: /* RGBA_8888 */
  94             fbinfo.bpp = 32;
  95             fbinfo.size = w * h * 4;
  96             fbinfo.width = w;
  97             fbinfo.height = h;
  98             fbinfo.red_offset = 0;
  99             fbinfo.red_length = 8;
 100             fbinfo.green_offset = 8;
 101             fbinfo.green_length = 8;
 102             fbinfo.blue_offset = 16;
 103             fbinfo.blue_length = 8;
 104             fbinfo.alpha_offset = 24;
 105             fbinfo.alpha_length = 8;
 106             break;
 107         case 2: /* RGBX_8888 */
 108             fbinfo.bpp = 32;
 109             fbinfo.size = w * h * 4;
 110             fbinfo.width = w;
 111             fbinfo.height = h;
 112             fbinfo.red_offset = 0;
 113             fbinfo.red_length = 8;
 114             fbinfo.green_offset = 8;
 115             fbinfo.green_length = 8;
 116             fbinfo.blue_offset = 16;
 117             fbinfo.blue_length = 8;
 118             fbinfo.alpha_offset = 24;
 119             fbinfo.alpha_length = 0;
 120             break;
 121         case 3: /* RGB_888 */
 122             fbinfo.bpp = 24;
 123             fbinfo.size = w * h * 3;
 124             fbinfo.width = w;
 125             fbinfo.height = h;
 126             fbinfo.red_offset = 0;
 127             fbinfo.red_length = 8;
 128             fbinfo.green_offset = 8;
 129             fbinfo.green_length = 8;
 130             fbinfo.blue_offset = 16;
 131             fbinfo.blue_length = 8;
 132             fbinfo.alpha_offset = 24;
 133             fbinfo.alpha_length = 0;
 134             break;
 135         case 4: /* RGB_565 */
 136             fbinfo.bpp = 16;
 137             fbinfo.size = w * h * 2;
 138             fbinfo.width = w;
 139             fbinfo.height = h;
 140             fbinfo.red_offset = 11;
 141             fbinfo.red_length = 5;
 142             fbinfo.green_offset = 5;
 143             fbinfo.green_length = 6;
 144             fbinfo.blue_offset = 0;
 145             fbinfo.blue_length = 5;
 146             fbinfo.alpha_offset = 0;
 147             fbinfo.alpha_length = 0;
 148             break;
 149         case 5: /* BGRA_8888 */
 150             fbinfo.bpp = 32;
 151             fbinfo.size = w * h * 4;
 152             fbinfo.width = w;
 153             fbinfo.height = h;
 154             fbinfo.red_offset = 16;
 155             fbinfo.red_length = 8;
 156             fbinfo.green_offset = 8;
 157             fbinfo.green_length = 8;
 158             fbinfo.blue_offset = 0;
 159             fbinfo.blue_length = 8;
 160             fbinfo.alpha_offset = 24;
 161             fbinfo.alpha_length = 8;
 162            break;
 163         default:
 164             goto done;
 165     }
 166
 167     /* write header */
 168     if(!WriteFdExactly(fd, &fbinfo, sizeof(fbinfo))) goto done;
 169
 170     /* write data */
 171     for(i = 0; i < fbinfo.size; i += bsize) {
 172       bsize = sizeof(buf);
 173       if (i + bsize > fbinfo.size)
 174         bsize = fbinfo.size - i;
 175       if(!ReadFdExactly(fd_screencap, buf, bsize)) goto done;
 176       if(!WriteFdExactly(fd, buf, bsize)) goto done;
 177     }
 178
 179 done:
 180     adb_close(fds[0]);
 181
 182     TEMP_FAILURE_RETRY(waitpid(pid, NULL, 0));
 183 pipefail:
 184     adb_close(fd);
 185 }