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 
framebuffer_service(int fd,void * cookie)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 }
186