1/*
2 * QuickDraw (qdrw) codec
3 * Copyright (c) 2004 Konstantin Shishkov
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * 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 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22/**
23 * @file
24 * Apple QuickDraw codec.
25 */
26
27#include "libavutil/common.h"
28#include "libavutil/intreadwrite.h"
29#include "avcodec.h"
30#include "internal.h"
31
32static int decode_frame(AVCodecContext *avctx,
33                        void *data, int *got_frame,
34                        AVPacket *avpkt)
35{
36    const uint8_t *buf     = avpkt->data;
37    const uint8_t *buf_end = avpkt->data + avpkt->size;
38    int buf_size           = avpkt->size;
39    AVFrame * const p      = data;
40    uint8_t* outdata;
41    int colors;
42    int i, ret;
43    uint32_t *pal;
44    int r, g, b;
45
46    if ((ret = ff_get_buffer(avctx, p, 0)) < 0)
47        return ret;
48    p->pict_type = AV_PICTURE_TYPE_I;
49    p->key_frame = 1;
50
51    outdata = p->data[0];
52
53    if (buf_end - buf < 0x68 + 4)
54        return AVERROR_INVALIDDATA;
55    buf   += 0x68; /* jump to palette */
56    colors = AV_RB32(buf);
57    buf   += 4;
58
59    if (colors < 0 || colors > 256) {
60        av_log(avctx, AV_LOG_ERROR, "Error color count - %i(0x%X)\n", colors, colors);
61        return AVERROR_INVALIDDATA;
62    }
63    if (buf_end - buf < (colors + 1) * 8)
64        return AVERROR_INVALIDDATA;
65
66    pal = (uint32_t*)p->data[1];
67    for (i = 0; i <= colors; i++) {
68        unsigned int idx;
69        idx = AV_RB16(buf); /* color index */
70        buf += 2;
71
72        if (idx > 255) {
73            av_log(avctx, AV_LOG_ERROR, "Palette index out of range: %u\n", idx);
74            buf += 6;
75            continue;
76        }
77        r = *buf++;
78        buf++;
79        g = *buf++;
80        buf++;
81        b = *buf++;
82        buf++;
83        pal[idx] = 0xFFU << 24 | r << 16 | g << 8 | b;
84    }
85    p->palette_has_changed = 1;
86
87    if (buf_end - buf < 18)
88        return AVERROR_INVALIDDATA;
89    buf += 18; /* skip unneeded data */
90    for (i = 0; i < avctx->height; i++) {
91        int size, left, code, pix;
92        const uint8_t *next;
93        uint8_t *out;
94        int tsize = 0;
95
96        /* decode line */
97        out  = outdata;
98        size = AV_RB16(buf); /* size of packed line */
99        buf += 2;
100        if (buf_end - buf < size)
101            return AVERROR_INVALIDDATA;
102
103        left = size;
104        next = buf + size;
105        while (left > 0) {
106            code = *buf++;
107            if (code & 0x80 ) { /* run */
108                pix = *buf++;
109                if ((out + (257 - code)) > (outdata +  p->linesize[0]))
110                    break;
111                memset(out, pix, 257 - code);
112                out   += 257 - code;
113                tsize += 257 - code;
114                left  -= 2;
115            } else { /* copy */
116                if ((out + code) > (outdata +  p->linesize[0]))
117                    break;
118                if (buf_end - buf < code + 1)
119                    return AVERROR_INVALIDDATA;
120                memcpy(out, buf, code + 1);
121                out   += code + 1;
122                buf   += code + 1;
123                left  -= 2 + code;
124                tsize += code + 1;
125            }
126        }
127        buf = next;
128        outdata += p->linesize[0];
129    }
130
131    *got_frame      = 1;
132
133    return buf_size;
134}
135
136static av_cold int decode_init(AVCodecContext *avctx)
137{
138    avctx->pix_fmt= AV_PIX_FMT_PAL8;
139
140    return 0;
141}
142
143AVCodec ff_qdraw_decoder = {
144    .name           = "qdraw",
145    .long_name      = NULL_IF_CONFIG_SMALL("Apple QuickDraw"),
146    .type           = AVMEDIA_TYPE_VIDEO,
147    .id             = AV_CODEC_ID_QDRAW,
148    .init           = decode_init,
149    .decode         = decode_frame,
150    .capabilities   = CODEC_CAP_DR1,
151};
152