2 * RTP JPEG-compressed video Packetizer, RFC 2435
3 * Copyright (c) 2012 Samuel Pitoiset
5 * This file is part of FFmpeg.
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.
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.
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
22 #include "libavcodec/bytestream.h"
23 #include "libavcodec/mjpeg.h"
24 #include "libavcodec/jpegtables.h"
25 #include "libavutil/intreadwrite.h"
28 void ff_rtp_send_jpeg(AVFormatContext *s1, const uint8_t *buf, int size)
30 RTPMuxContext *s = s1->priv_data;
31 const uint8_t *qtables = NULL;
36 int off = 0; /* fragment offset of the current JPEG frame */
41 s->timestamp = s->cur_timestamp;
43 /* convert video pixel dimensions from pixels to blocks */
44 w = FF_CEIL_RSHIFT(s1->streams[0]->codec->width, 3);
45 h = FF_CEIL_RSHIFT(s1->streams[0]->codec->height, 3);
47 /* get the pixel format type or fail */
48 if (s1->streams[0]->codec->pix_fmt == AV_PIX_FMT_YUVJ422P ||
49 (s1->streams[0]->codec->color_range == AVCOL_RANGE_JPEG &&
50 s1->streams[0]->codec->pix_fmt == AV_PIX_FMT_YUV422P)) {
52 } else if (s1->streams[0]->codec->pix_fmt == AV_PIX_FMT_YUVJ420P ||
53 (s1->streams[0]->codec->color_range == AVCOL_RANGE_JPEG &&
54 s1->streams[0]->codec->pix_fmt == AV_PIX_FMT_YUV420P)) {
57 av_log(s1, AV_LOG_ERROR, "Unsupported pixel format\n");
61 /* preparse the header for getting some infos */
62 for (i = 0; i < size; i++) {
66 if (buf[i + 1] == DQT) {
67 if (buf[i + 4] & 0xF0)
68 av_log(s1, AV_LOG_WARNING,
69 "Only 8-bit precision is supported.\n");
71 /* a quantization table is 64 bytes long */
72 nb_qtables = AV_RB16(&buf[i + 2]) / 65;
73 if (i + 4 + nb_qtables * 65 > size) {
74 av_log(s1, AV_LOG_ERROR, "Too short JPEG header. Aborted!\n");
78 qtables = &buf[i + 4];
79 } else if (buf[i + 1] == SOF0) {
80 if (buf[i + 14] != 17 || buf[i + 17] != 17) {
81 av_log(s1, AV_LOG_ERROR,
82 "Only 1x1 chroma blocks are supported. Aborted!\n");
85 } else if (buf[i + 1] == DHT) {
86 if ( AV_RB16(&buf[i + 2]) < 418
89 || buf[i + 33] != 0x01
90 || buf[i + 62] != 0x10
91 || buf[i + 241] != 0x11
92 || memcmp(buf + i + 5, avpriv_mjpeg_bits_dc_luminance + 1, 16)
93 || memcmp(buf + i + 21, avpriv_mjpeg_val_dc, 12)
94 || memcmp(buf + i + 34, avpriv_mjpeg_bits_dc_chrominance + 1, 16)
95 || memcmp(buf + i + 50, avpriv_mjpeg_val_dc, 12)
96 || memcmp(buf + i + 63, avpriv_mjpeg_bits_ac_luminance + 1, 16)
97 || memcmp(buf + i + 79, avpriv_mjpeg_val_ac_luminance, 162)
98 || memcmp(buf + i + 242, avpriv_mjpeg_bits_ac_chrominance + 1, 16)
99 || memcmp(buf + i + 258, avpriv_mjpeg_val_ac_chrominance, 162)) {
100 av_log(s1, AV_LOG_ERROR,
101 "RFC 2435 requires standard Huffman tables for jpeg\n");
104 } else if (buf[i + 1] == SOS) {
105 /* SOS is last marker in the header */
106 i += AV_RB16(&buf[i + 2]) + 2;
108 av_log(s1, AV_LOG_ERROR,
109 "Insufficient data. Aborted!\n");
116 /* skip JPEG header */
120 for (i = size - 2; i >= 0; i--) {
121 if (buf[i] == 0xff && buf[i + 1] == EOI) {
122 /* Remove the EOI marker */
132 if (off == 0 && nb_qtables)
133 hdr_size += 4 + 64 * nb_qtables;
135 /* payload max in one packet */
136 len = FFMIN(size, s->max_payload_size - hdr_size);
138 /* set main header */
139 bytestream_put_byte(&p, 0);
140 bytestream_put_be24(&p, off);
141 bytestream_put_byte(&p, type);
142 bytestream_put_byte(&p, 255);
143 bytestream_put_byte(&p, w);
144 bytestream_put_byte(&p, h);
146 if (off == 0 && nb_qtables) {
147 /* set quantization tables header */
148 bytestream_put_byte(&p, 0);
149 bytestream_put_byte(&p, 0);
150 bytestream_put_be16(&p, 64 * nb_qtables);
152 for (i = 0; i < nb_qtables; i++)
153 bytestream_put_buffer(&p, &qtables[65 * i + 1], 64);
156 /* copy payload data */
159 /* marker bit is last packet in frame */
160 ff_rtp_send_data(s1, s->buf, len + hdr_size, size == len);