d9553040e851b8ed74c5bb116d51d9e00d0332d6
[ffmpeg.git] / libavformat / mpjpegdec.c
1 /*
2  * Multipart JPEG format
3  * Copyright (c) 2015 Luca Barbato
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 #include "libavutil/avstring.h"
23
24 #include "avformat.h"
25 #include "internal.h"
26
27 static int get_line(AVIOContext *pb, char *line, int line_size)
28 {
29     int i = ff_get_line(pb, line, line_size);
30
31     if (i > 1 && line[i - 2] == '\r')
32         line[i - 2] = '\0';
33
34     if (pb->error)
35         return pb->error;
36
37     if (pb->eof_reached)
38         return AVERROR_EOF;
39
40     return 0;
41 }
42
43
44 static void trim_right(char* p)
45 {
46     char *end;
47     if (!p || !*p)
48         return;
49     end = p + strlen(p) - 1;
50     while (end != p && av_isspace(*end)) {
51         *end = '\0';
52         end--;
53     }
54 }
55
56 static int split_tag_value(char **tag, char **value, char *line)
57 {
58     char *p = line;
59
60     while (*p != '\0' && *p != ':')
61         p++;
62     if (*p != ':')
63         return AVERROR_INVALIDDATA;
64
65     *p   = '\0';
66     *tag = line;
67     trim_right(*tag);
68
69     p++;
70
71     while (av_isspace(*p))
72         p++;
73
74     *value = p;
75     trim_right(*value);
76
77     return 0;
78 }
79
80 static int check_content_type(char *line)
81 {
82     char *tag, *value;
83     int ret = split_tag_value(&tag, &value, line);
84
85     if (ret < 0)
86         return ret;
87
88     if (av_strcasecmp(tag, "Content-type") ||
89         av_strcasecmp(value, "image/jpeg"))
90         return AVERROR_INVALIDDATA;
91
92     return 0;
93 }
94
95 static int parse_multipart_header(AVIOContext *pb, void *log_ctx);
96
97 static int mpjpeg_read_probe(AVProbeData *p)
98 {
99     AVIOContext *pb;
100     char line[128] = { 0 };
101     int ret = 0;
102
103     if (p->buf_size < 2 || p->buf[0] != '-' || p->buf[1] != '-')
104         return 0;
105
106     pb = avio_alloc_context(p->buf, p->buf_size, 0, NULL, NULL, NULL, NULL);
107     if (!pb)
108         return AVERROR(ENOMEM);
109
110     ret = (parse_multipart_header(pb, NULL)>0)?AVPROBE_SCORE_MAX:0;
111
112     av_free(pb);
113
114     return ret;
115 }
116
117 static int mpjpeg_read_header(AVFormatContext *s)
118 {
119     AVStream *st;
120     char boundary[70 + 2 + 1];
121     int64_t pos = avio_tell(s->pb);
122     int ret;
123
124
125     ret = get_line(s->pb, boundary, sizeof(boundary));
126     if (ret < 0)
127         return ret;
128
129     if (strncmp(boundary, "--", 2))
130         return AVERROR_INVALIDDATA;
131
132     st = avformat_new_stream(s, NULL);
133     if (!st)
134         return AVERROR(ENOMEM);
135
136     st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
137     st->codec->codec_id   = AV_CODEC_ID_MJPEG;
138
139     avpriv_set_pts_info(st, 60, 1, 25);
140
141     avio_seek(s->pb, pos, SEEK_SET);
142
143     return 0;
144 }
145
146 static int parse_content_length(const char *value)
147 {
148     long int val = strtol(value, NULL, 10);
149
150     if (val == LONG_MIN || val == LONG_MAX)
151         return AVERROR(errno);
152     if (val > INT_MAX)
153         return AVERROR(ERANGE);
154     return val;
155 }
156
157 static int parse_multipart_header(AVIOContext *pb, void *log_ctx)
158 {
159     char line[128];
160     int found_content_type = 0;
161     int ret, size = -1;
162
163     ret = get_line(pb, line, sizeof(line));
164     if (ret < 0)
165         return ret;
166
167     if (strncmp(line, "--", 2))
168         return AVERROR_INVALIDDATA;
169
170     while (!pb->eof_reached) {
171         char *tag, *value;
172
173         ret = get_line(pb, line, sizeof(line));
174         if (ret < 0) {
175             if (ret == AVERROR_EOF)
176                 break;
177             return ret;
178         }
179
180         if (line[0] == '\0')
181             break;
182
183         ret = split_tag_value(&tag, &value, line);
184         if (ret < 0)
185             return ret;
186
187         if (!av_strcasecmp(tag, "Content-type")) {
188             if (av_strcasecmp(value, "image/jpeg")) {
189                 if (log_ctx) {
190                     av_log(log_ctx, AV_LOG_ERROR,
191                            "Unexpected %s : %s\n",
192                            tag, value);
193                 }
194
195                 return AVERROR_INVALIDDATA;
196             } else
197                 found_content_type = 1;
198         } else if (!av_strcasecmp(tag, "Content-Length")) {
199             size = parse_content_length(value);
200             if (size < 0)
201                 return size;
202         }
203     }
204
205     if (!found_content_type || size < 0) {
206         return AVERROR_INVALIDDATA;
207     }
208
209     return size;
210 }
211
212 static int mpjpeg_read_packet(AVFormatContext *s, AVPacket *pkt)
213 {
214     int ret;
215     int size = parse_multipart_header(s->pb, s);
216
217     if (size < 0)
218         return size;
219
220     ret = av_get_packet(s->pb, pkt, size);
221     if (ret < 0)
222         return ret;
223
224     // trailing empty line
225     avio_skip(s->pb, 2);
226
227     return 0;
228 }
229
230 AVInputFormat ff_mpjpeg_demuxer = {
231     .name              = "mpjpeg",
232     .long_name         = NULL_IF_CONFIG_SMALL("MIME multipart JPEG"),
233     .mime_type         = "multipart/x-mixed-replace",
234     .extensions        = "mjpg",
235     .read_probe        = mpjpeg_read_probe,
236     .read_header       = mpjpeg_read_header,
237     .read_packet       = mpjpeg_read_packet,
238 };