Statistics
| Branch: | Revision:

ffmpeg / libavcodec / vmnc.c @ 7a886063

History | View | Annotate | Download (15.1 KB)

1 eb57c889 Kostya Shishkov
/*
2
 * VMware Screen Codec (VMnc) decoder
3
 * Copyright (c) 2006 Konstantin Shishkov
4
 *
5 b78e7197 Diego Biurrun
 * This file is part of FFmpeg.
6
 *
7
 * FFmpeg is free software; you can redistribute it and/or
8 eb57c889 Kostya Shishkov
 * modify it under the terms of the GNU Lesser General Public
9
 * License as published by the Free Software Foundation; either
10 b78e7197 Diego Biurrun
 * version 2.1 of the License, or (at your option) any later version.
11 eb57c889 Kostya Shishkov
 *
12 b78e7197 Diego Biurrun
 * FFmpeg is distributed in the hope that it will be useful,
13 eb57c889 Kostya Shishkov
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
 * Lesser General Public License for more details.
16
 *
17
 * You should have received a copy of the GNU Lesser General Public
18 b78e7197 Diego Biurrun
 * License along with FFmpeg; if not, write to the Free Software
19 eb57c889 Kostya Shishkov
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 *
21
 */
22
23
/**
24
 * @file vmnc.c
25
 * VMware Screen Codec (VMnc) decoder
26
 * As Alex Beregszaszi discovered, this is effectively RFB data dump
27
 */
28
29
#include <stdio.h>
30
#include <stdlib.h>
31
32
#include "common.h"
33
#include "avcodec.h"
34
35 805934b3 Kostya Shishkov
enum EncTypes {
36
    MAGIC_WMVd = 0x574D5664,
37
    MAGIC_WMVe,
38
    MAGIC_WMVf,
39
    MAGIC_WMVg,
40
    MAGIC_WMVh,
41
    MAGIC_WMVi,
42
    MAGIC_WMVj
43
};
44 eb57c889 Kostya Shishkov
45
enum HexTile_Flags {
46
    HT_RAW =  1, // tile is raw
47
    HT_BKG =  2, // background color is present
48
    HT_FG  =  4, // foreground color is present
49
    HT_SUB =  8, // subrects are present
50
    HT_CLR = 16  // each subrect has own color
51
};
52
53
/*
54
 * Decoder context
55
 */
56
typedef struct VmncContext {
57
    AVCodecContext *avctx;
58
    AVFrame pic;
59
60
    int bpp;
61
    int bpp2;
62
    int bigendian;
63
    uint8_t pal[768];
64
    int width, height;
65 54b55c8d Kostya Shishkov
66
    /* cursor data */
67
    int cur_w, cur_h;
68
    int cur_x, cur_y;
69
    int cur_hx, cur_hy;
70
    uint8_t* curbits, *curmask;
71
    uint8_t* screendta;
72 eb57c889 Kostya Shishkov
} VmncContext;
73
74
/* read pixel value from stream */
75 849f1035 Måns Rullgård
static av_always_inline int vmnc_get_pixel(uint8_t* buf, int bpp, int be) {
76 eb57c889 Kostya Shishkov
    switch(bpp * 2 + be) {
77
    case 2:
78
    case 3: return *buf;
79
    case 4: return LE_16(buf);
80
    case 5: return BE_16(buf);
81
    case 8: return LE_32(buf);
82
    case 9: return BE_32(buf);
83
    default: return 0;
84
    }
85
}
86
87 54b55c8d Kostya Shishkov
static void load_cursor(VmncContext *c, uint8_t *src)
88
{
89
    int i, j, p;
90
    const int bpp = c->bpp2;
91
    uint8_t  *dst8  = c->curbits;
92
    uint16_t *dst16 = (uint16_t*)c->curbits;
93
    uint32_t *dst32 = (uint32_t*)c->curbits;
94
95
    for(j = 0; j < c->cur_h; j++) {
96
        for(i = 0; i < c->cur_w; i++) {
97
            p = vmnc_get_pixel(src, bpp, c->bigendian);
98
            src += bpp;
99
            if(bpp == 1) *dst8++ = p;
100
            if(bpp == 2) *dst16++ = p;
101
            if(bpp == 4) *dst32++ = p;
102
        }
103
    }
104
    dst8 = c->curmask;
105
    dst16 = (uint16_t*)c->curmask;
106
    dst32 = (uint32_t*)c->curmask;
107
    for(j = 0; j < c->cur_h; j++) {
108
        for(i = 0; i < c->cur_w; i++) {
109
            p = vmnc_get_pixel(src, bpp, c->bigendian);
110
            src += bpp;
111
            if(bpp == 1) *dst8++ = p;
112
            if(bpp == 2) *dst16++ = p;
113
            if(bpp == 4) *dst32++ = p;
114
        }
115
    }
116
}
117
118
static void put_cursor(uint8_t *dst, int stride, VmncContext *c, int dx, int dy)
119
{
120 36c32bdd Diego Biurrun
    int i, j;
121 54b55c8d Kostya Shishkov
    int w, h, x, y;
122
    w = c->cur_w;
123
    if(c->width < c->cur_x + c->cur_w) w = c->width - c->cur_x;
124
    h = c->cur_h;
125
    if(c->height < c->cur_y + c->cur_h) h = c->height - c->cur_y;
126
    x = c->cur_x;
127
    y = c->cur_y;
128
    if(x < 0) {
129
        w += x;
130
        x = 0;
131
    }
132
    if(y < 0) {
133
        h += y;
134
        y = 0;
135
    }
136
137
    if((w < 1) || (h < 1)) return;
138
    dst += x * c->bpp2 + y * stride;
139
140
    if(c->bpp2 == 1) {
141
        uint8_t* cd = c->curbits, *msk = c->curmask;
142
        for(j = 0; j < h; j++) {
143
            for(i = 0; i < w; i++)
144
                dst[i] = (dst[i] & cd[i]) ^ msk[i];
145
            msk += c->cur_w;
146
            cd += c->cur_w;
147
            dst += stride;
148
        }
149
    } else if(c->bpp2 == 2) {
150
        uint16_t* cd = (uint16_t*)c->curbits, *msk = (uint16_t*)c->curmask;
151
        uint16_t* dst2;
152
        for(j = 0; j < h; j++) {
153
            dst2 = (uint16_t*)dst;
154
            for(i = 0; i < w; i++)
155
                dst2[i] = (dst2[i] & cd[i]) ^ msk[i];
156
            msk += c->cur_w;
157
            cd += c->cur_w;
158
            dst += stride;
159
        }
160
    } else if(c->bpp2 == 4) {
161
        uint32_t* cd = (uint32_t*)c->curbits, *msk = (uint32_t*)c->curmask;
162
        uint32_t* dst2;
163
        for(j = 0; j < h; j++) {
164
            dst2 = (uint32_t*)dst;
165
            for(i = 0; i < w; i++)
166
                dst2[i] = (dst2[i] & cd[i]) ^ msk[i];
167
            msk += c->cur_w;
168
            cd += c->cur_w;
169
            dst += stride;
170
        }
171
    }
172
}
173
174 eb57c889 Kostya Shishkov
/* fill rectangle with given colour */
175 849f1035 Måns Rullgård
static av_always_inline void paint_rect(uint8_t *dst, int dx, int dy, int w, int h, int color, int bpp, int stride)
176 eb57c889 Kostya Shishkov
{
177
    int i, j;
178
    dst += dx * bpp + dy * stride;
179
    if(bpp == 1){
180
        for(j = 0; j < h; j++) {
181
            memset(dst, color, w);
182
            dst += stride;
183
        }
184
    }else if(bpp == 2){
185
        uint16_t* dst2;
186
        for(j = 0; j < h; j++) {
187
            dst2 = (uint16_t*)dst;
188
            for(i = 0; i < w; i++) {
189
                *dst2++ = color;
190
            }
191
            dst += stride;
192
        }
193
    }else if(bpp == 4){
194
        uint32_t* dst2;
195
        for(j = 0; j < h; j++) {
196
            dst2 = (uint32_t*)dst;
197
            for(i = 0; i < w; i++) {
198
                dst2[i] = color;
199
            }
200
            dst += stride;
201
        }
202
    }
203
}
204
205 849f1035 Måns Rullgård
static av_always_inline void paint_raw(uint8_t *dst, int w, int h, uint8_t* src, int bpp, int be, int stride)
206 eb57c889 Kostya Shishkov
{
207
    int i, j, p;
208
    for(j = 0; j < h; j++) {
209
        for(i = 0; i < w; i++) {
210
            p = vmnc_get_pixel(src, bpp, be);
211
            src += bpp;
212 c0d6fc1f Kostya Shishkov
            switch(bpp){
213
            case 1:
214
                dst[i] = p;
215
                break;
216
            case 2:
217
                ((uint16_t*)dst)[i] = p;
218
                break;
219
            case 4:
220
                ((uint32_t*)dst)[i] = p;
221
                break;
222
            }
223 eb57c889 Kostya Shishkov
        }
224
        dst += stride;
225
    }
226
}
227
228 096bc417 Kostya Shishkov
static int decode_hextile(VmncContext *c, uint8_t* dst, uint8_t* src, int ssize, int w, int h, int stride)
229 eb57c889 Kostya Shishkov
{
230
    int i, j, k;
231
    int bg = 0, fg = 0, rects, color, flags, xy, wh;
232
    const int bpp = c->bpp2;
233
    uint8_t *dst2;
234
    int bw = 16, bh = 16;
235
    uint8_t *ssrc=src;
236
237
    for(j = 0; j < h; j += 16) {
238
        dst2 = dst;
239
        bw = 16;
240
        if(j + 16 > h) bh = h - j;
241
        for(i = 0; i < w; i += 16, dst2 += 16 * bpp) {
242 096bc417 Kostya Shishkov
            if(src - ssrc >= ssize) {
243
                av_log(c->avctx, AV_LOG_ERROR, "Premature end of data!\n");
244
                return -1;
245
            }
246 eb57c889 Kostya Shishkov
            if(i + 16 > w) bw = w - i;
247
            flags = *src++;
248
            if(flags & HT_RAW) {
249 096bc417 Kostya Shishkov
                if(src - ssrc > ssize - bw * bh * bpp) {
250
                    av_log(c->avctx, AV_LOG_ERROR, "Premature end of data!\n");
251
                    return -1;
252
                }
253 eb57c889 Kostya Shishkov
                paint_raw(dst2, bw, bh, src, bpp, c->bigendian, stride);
254 c0d6fc1f Kostya Shishkov
                src += bw * bh * bpp;
255 eb57c889 Kostya Shishkov
            } else {
256
                if(flags & HT_BKG) {
257
                    bg = vmnc_get_pixel(src, bpp, c->bigendian); src += bpp;
258
                }
259
                if(flags & HT_FG) {
260
                    fg = vmnc_get_pixel(src, bpp, c->bigendian); src += bpp;
261
                }
262
                rects = 0;
263
                if(flags & HT_SUB)
264
                    rects = *src++;
265 096bc417 Kostya Shishkov
                color = !!(flags & HT_CLR);
266 eb57c889 Kostya Shishkov
267
                paint_rect(dst2, 0, 0, bw, bh, bg, bpp, stride);
268
269 096bc417 Kostya Shishkov
                if(src - ssrc > ssize - rects * (color * bpp + 2)) {
270
                    av_log(c->avctx, AV_LOG_ERROR, "Premature end of data!\n");
271
                    return -1;
272
                }
273 eb57c889 Kostya Shishkov
                for(k = 0; k < rects; k++) {
274
                    if(color) {
275
                        fg = vmnc_get_pixel(src, bpp, c->bigendian); src += bpp;
276
                    }
277
                    xy = *src++;
278
                    wh = *src++;
279
                    paint_rect(dst2, xy >> 4, xy & 0xF, (wh>>4)+1, (wh & 0xF)+1, fg, bpp, stride);
280
                }
281
            }
282
        }
283
        dst += stride * 16;
284
    }
285
    return src - ssrc;
286
}
287
288
static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size)
289
{
290
    VmncContext * const c = (VmncContext *)avctx->priv_data;
291
    uint8_t *outptr;
292
    uint8_t *src = buf;
293 096bc417 Kostya Shishkov
    int dx, dy, w, h, depth, enc, chunks, res, size_left;
294 eb57c889 Kostya Shishkov
295
    c->pic.reference = 1;
296
    c->pic.buffer_hints = FF_BUFFER_HINTS_VALID | FF_BUFFER_HINTS_PRESERVE | FF_BUFFER_HINTS_REUSABLE;
297
    if(avctx->reget_buffer(avctx, &c->pic) < 0){
298
        av_log(avctx, AV_LOG_ERROR, "reget_buffer() failed\n");
299
        return -1;
300
    }
301
302 805934b3 Kostya Shishkov
    c->pic.key_frame = 0;
303
    c->pic.pict_type = FF_P_TYPE;
304 eb57c889 Kostya Shishkov
305 54b55c8d Kostya Shishkov
    //restore screen after cursor
306
    if(c->screendta) {
307
        int i;
308
        w = c->cur_w;
309
        if(c->width < c->cur_x + w) w = c->width - c->cur_x;
310
        h = c->cur_h;
311
        if(c->height < c->cur_y + h) h = c->height - c->cur_y;
312
        dx = c->cur_x;
313
        if(dx < 0) {
314
            w += dx;
315
            dx = 0;
316
        }
317
        dy = c->cur_y;
318
        if(dy < 0) {
319
            h += dy;
320
            dy = 0;
321
        }
322
        if((w > 0) && (h > 0)) {
323
            outptr = c->pic.data[0] + dx * c->bpp2 + dy * c->pic.linesize[0];
324
            for(i = 0; i < h; i++) {
325
                memcpy(outptr, c->screendta + i * c->cur_w * c->bpp2, w * c->bpp2);
326
                outptr += c->pic.linesize[0];
327
            }
328
        }
329
    }
330 805934b3 Kostya Shishkov
    src += 2;
331
    chunks = BE_16(src); src += 2;
332 eb57c889 Kostya Shishkov
    while(chunks--) {
333
        dx = BE_16(src); src += 2;
334
        dy = BE_16(src); src += 2;
335
        w  = BE_16(src); src += 2;
336
        h  = BE_16(src); src += 2;
337
        enc = BE_32(src); src += 4;
338 54b55c8d Kostya Shishkov
        outptr = c->pic.data[0] + dx * c->bpp2 + dy * c->pic.linesize[0];
339 096bc417 Kostya Shishkov
        size_left = buf_size - (src - buf);
340 805934b3 Kostya Shishkov
        switch(enc) {
341 54b55c8d Kostya Shishkov
        case MAGIC_WMVd: // cursor
342 096bc417 Kostya Shishkov
            if(size_left < 2 + w * h * c->bpp2 * 2) {
343
                av_log(avctx, AV_LOG_ERROR, "Premature end of data! (need %i got %i)\n", 2 + w * h * c->bpp2 * 2, size_left);
344
                return -1;
345
            }
346 805934b3 Kostya Shishkov
            src += 2;
347 54b55c8d Kostya Shishkov
            c->cur_w = w;
348
            c->cur_h = h;
349
            c->cur_hx = dx;
350
            c->cur_hy = dy;
351
            if((c->cur_hx > c->cur_w) || (c->cur_hy > c->cur_h)) {
352
                av_log(avctx, AV_LOG_ERROR, "Cursor hot spot is not in image: %ix%i of %ix%i cursor size\n", c->cur_hx, c->cur_hy, c->cur_w, c->cur_h);
353
                c->cur_hx = c->cur_hy = 0;
354
            }
355
            c->curbits = av_realloc(c->curbits, c->cur_w * c->cur_h * c->bpp2);
356
            c->curmask = av_realloc(c->curmask, c->cur_w * c->cur_h * c->bpp2);
357
            c->screendta = av_realloc(c->screendta, c->cur_w * c->cur_h * c->bpp2);
358
            load_cursor(c, src);
359
            src += w * h * c->bpp2 * 2;
360 805934b3 Kostya Shishkov
            break;
361
        case MAGIC_WMVe: // unknown
362
            src += 2;
363
            break;
364 54b55c8d Kostya Shishkov
        case MAGIC_WMVf: // update cursor position
365
            c->cur_x = dx - c->cur_hx;
366
            c->cur_y = dy - c->cur_hy;
367 805934b3 Kostya Shishkov
            break;
368 7ff0b84f Kostya Shishkov
        case MAGIC_WMVg: // unknown
369
            src += 10;
370
            break;
371
        case MAGIC_WMVh: // unknown
372
            src += 4;
373
            break;
374 805934b3 Kostya Shishkov
        case MAGIC_WMVi: // ServerInitialization struct
375
            c->pic.key_frame = 1;
376
            c->pic.pict_type = FF_I_TYPE;
377
            depth = *src++;
378
            if(depth != c->bpp) {
379
                av_log(avctx, AV_LOG_INFO, "Depth mismatch. Container %i bpp, Frame data: %i bpp\n", c->bpp, depth);
380
            }
381
            src++;
382
            c->bigendian = *src++;
383
            if(c->bigendian & (~1)) {
384
                av_log(avctx, AV_LOG_INFO, "Invalid header: bigendian flag = %i\n", c->bigendian);
385
                return -1;
386
            }
387
            //skip the rest of pixel format data
388
            src += 13;
389
            break;
390 7ff0b84f Kostya Shishkov
        case MAGIC_WMVj: // unknown
391
            src += 2;
392
            break;
393 805934b3 Kostya Shishkov
        case 0x00000000: // raw rectangle data
394
            if((dx + w > c->width) || (dy + h > c->height)) {
395
                av_log(avctx, AV_LOG_ERROR, "Incorrect frame size: %ix%i+%ix%i of %ix%i\n", w, h, dx, dy, c->width, c->height);
396
                return -1;
397
            }
398 096bc417 Kostya Shishkov
            if(size_left < w * h * c->bpp2) {
399
                av_log(avctx, AV_LOG_ERROR, "Premature end of data! (need %i got %i)\n", w * h * c->bpp2, size_left);
400
                return -1;
401
            }
402 805934b3 Kostya Shishkov
            paint_raw(outptr, w, h, src, c->bpp2, c->bigendian, c->pic.linesize[0]);
403
            src += w * h * c->bpp2;
404
            break;
405
        case 0x00000005: // HexTile encoded rectangle
406
            if((dx + w > c->width) || (dy + h > c->height)) {
407
                av_log(avctx, AV_LOG_ERROR, "Incorrect frame size: %ix%i+%ix%i of %ix%i\n", w, h, dx, dy, c->width, c->height);
408
                return -1;
409 eb57c889 Kostya Shishkov
            }
410 096bc417 Kostya Shishkov
            res = decode_hextile(c, outptr, src, size_left, w, h, c->pic.linesize[0]);
411 805934b3 Kostya Shishkov
            if(res < 0)
412
                return -1;
413
            src += res;
414
            break;
415
        default:
416
            av_log(avctx, AV_LOG_ERROR, "Unsupported block type 0x%08X\n", enc);
417
            chunks = 0; // leave chunks decoding loop
418 eb57c889 Kostya Shishkov
        }
419
    }
420 54b55c8d Kostya Shishkov
    if(c->screendta){
421
        int i;
422
        //save screen data before painting cursor
423
        w = c->cur_w;
424
        if(c->width < c->cur_x + w) w = c->width - c->cur_x;
425
        h = c->cur_h;
426
        if(c->height < c->cur_y + h) h = c->height - c->cur_y;
427
        dx = c->cur_x;
428
        if(dx < 0) {
429
            w += dx;
430
            dx = 0;
431
        }
432
        dy = c->cur_y;
433
        if(dy < 0) {
434
            h += dy;
435
            dy = 0;
436
        }
437
        if((w > 0) && (h > 0)) {
438
            outptr = c->pic.data[0] + dx * c->bpp2 + dy * c->pic.linesize[0];
439
            for(i = 0; i < h; i++) {
440
                memcpy(c->screendta + i * c->cur_w * c->bpp2, outptr, w * c->bpp2);
441
                outptr += c->pic.linesize[0];
442
            }
443
            outptr = c->pic.data[0];
444
            put_cursor(outptr, c->pic.linesize[0], c, c->cur_x, c->cur_y);
445
        }
446
    }
447 eb57c889 Kostya Shishkov
    *data_size = sizeof(AVFrame);
448
    *(AVFrame*)data = c->pic;
449
450
    /* always report that the buffer was completely consumed */
451
    return buf_size;
452
}
453
454
455
456
/*
457
 *
458
 * Init VMnc decoder
459
 *
460
 */
461
static int decode_init(AVCodecContext *avctx)
462
{
463
    VmncContext * const c = (VmncContext *)avctx->priv_data;
464
465
    c->avctx = avctx;
466
    avctx->has_b_frames = 0;
467
468
    c->pic.data[0] = NULL;
469
    c->width = avctx->width;
470
    c->height = avctx->height;
471
472 429914a3 Kostya Shishkov
    if (avcodec_check_dimensions(avctx, avctx->width, avctx->height) < 0) {
473 eb57c889 Kostya Shishkov
        return 1;
474
    }
475
    c->bpp = avctx->bits_per_sample;
476
    c->bpp2 = c->bpp/8;
477
478
    switch(c->bpp){
479
    case 8:
480
        avctx->pix_fmt = PIX_FMT_PAL8;
481
        break;
482
    case 16:
483
        avctx->pix_fmt = PIX_FMT_RGB555;
484
        break;
485
    case 32:
486
        avctx->pix_fmt = PIX_FMT_RGB32;
487
        break;
488
    default:
489
        av_log(avctx, AV_LOG_ERROR, "Unsupported bitdepth %i\n", c->bpp);
490
    }
491
492
    return 0;
493
}
494
495
496
497
/*
498
 *
499
 * Uninit VMnc decoder
500
 *
501
 */
502
static int decode_end(AVCodecContext *avctx)
503
{
504
    VmncContext * const c = (VmncContext *)avctx->priv_data;
505
506
    if (c->pic.data[0])
507
        avctx->release_buffer(avctx, &c->pic);
508
509 54b55c8d Kostya Shishkov
    av_free(c->curbits);
510
    av_free(c->curmask);
511
    av_free(c->screendta);
512 eb57c889 Kostya Shishkov
    return 0;
513
}
514
515
AVCodec vmnc_decoder = {
516
    "VMware video",
517
    CODEC_TYPE_VIDEO,
518
    CODEC_ID_VMNC,
519
    sizeof(VmncContext),
520
    decode_init,
521
    NULL,
522
    decode_end,
523
    decode_frame
524
};