1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <linux/fb.h>
#include <zlib.h>
#include <libpng/png.h>
#include "private/android_filesystem_config.h"
#define LOG_TAG "screenshot"
#include <utils/Log.h>
void take_screenshot(FILE *fb_in, FILE *fb_out) {
int fb;
char imgbuf[0x10000];
struct fb_var_screeninfo vinfo;
png_structp png;
png_infop info;
unsigned int r,c,rowlen;
unsigned int bytespp,offset;
fb = fileno(fb_in);
if(fb < 0) {
LOGE("failed to open framebuffer\n");
return;
}
fb_in = fdopen(fb, "r");
if(ioctl(fb, FBIOGET_VSCREENINFO, &vinfo) < 0) {
LOGE("failed to get framebuffer info\n");
return;
}
fcntl(fb, F_SETFD, FD_CLOEXEC);
png = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
if (png == NULL) {
LOGE("failed png_create_write_struct\n");
fclose(fb_in);
return;
}
png_init_io(png, fb_out);
info = png_create_info_struct(png);
if (info == NULL) {
LOGE("failed png_create_info_struct\n");
png_destroy_write_struct(&png, NULL);
fclose(fb_in);
return;
}
if (setjmp(png_jmpbuf(png))) {
LOGE("failed png setjmp\n");
png_destroy_write_struct(&png, NULL);
fclose(fb_in);
return;
}
bytespp = vinfo.bits_per_pixel / 8;
png_set_IHDR(png, info,
vinfo.xres, vinfo.yres, vinfo.bits_per_pixel / 4,
PNG_COLOR_TYPE_RGB_ALPHA, PNG_INTERLACE_NONE,
PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
png_write_info(png, info);
rowlen=vinfo.xres * bytespp;
if (rowlen > sizeof(imgbuf)) {
LOGE("crazy rowlen: %d\n", rowlen);
png_destroy_write_struct(&png, NULL);
fclose(fb_in);
return;
}
offset = vinfo.xoffset * bytespp + vinfo.xres * vinfo.yoffset * bytespp;
fseek(fb_in, offset, SEEK_SET);
for(r=0; r<vinfo.yres; r++) {
int len = fread(imgbuf, 1, rowlen, fb_in);
if (len <= 0) break;
png_write_row(png, (png_bytep)imgbuf);
}
png_write_end(png, info);
fclose(fb_in);
png_destroy_write_struct(&png, NULL);
}
int main(int argc, char**argv) {
FILE *png = NULL;
FILE *fb_in = NULL;
if (argc < 2) {
fprintf(stderr, "usage: screenshot filename.png\n");
exit(1);
}
fb_in = fopen("/dev/graphics/fb0", "r");
if (!fb_in) {
fprintf(stderr, "error: could not read framebuffer\n");
exit(1);
}
/* switch to non-root user and group */
gid_t groups[] = { AID_LOG, AID_SDCARD_RW };
setgroups(sizeof(groups)/sizeof(groups[0]), groups);
setuid(AID_SHELL);
png = fopen(argv[1], "w");
if (!png) {
fprintf(stderr, "error: writing file %s: %s\n", argv[1], strerror(errno));
exit(1);
}
take_screenshot(fb_in, png);
exit(0);
}
|