1 /*
2  * RAW muxers
3  * Copyright (c) 2001 Fabrice Bellard
4  * Copyright (c) 2005 Alex Beregszaszi
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "config_components.h"
24 
25 #include "libavutil/intreadwrite.h"
26 
27 #include "avformat.h"
28 #include "rawenc.h"
29 #include "mux.h"
30 
ff_raw_write_packet(AVFormatContext *s, AVPacket *pkt)31 int ff_raw_write_packet(AVFormatContext *s, AVPacket *pkt)
32 {
33     avio_write(s->pb, pkt->data, pkt->size);
34     return 0;
35 }
36 
force_one_stream(AVFormatContext *s)37 static int force_one_stream(AVFormatContext *s)
38 {
39     if (s->nb_streams != 1) {
40         av_log(s, AV_LOG_ERROR, "%s files have exactly one stream\n",
41                s->oformat->name);
42         return AVERROR(EINVAL);
43     }
44     if (   s->oformat->audio_codec != AV_CODEC_ID_NONE
45         && s->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_AUDIO) {
46         av_log(s, AV_LOG_ERROR, "%s files have exactly one audio stream\n",
47                s->oformat->name);
48         return AVERROR(EINVAL);
49     }
50     if (   s->oformat->video_codec != AV_CODEC_ID_NONE
51         && s->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_VIDEO) {
52         av_log(s, AV_LOG_ERROR, "%s files have exactly one video stream\n",
53                s->oformat->name);
54         return AVERROR(EINVAL);
55     }
56     return 0;
57 }
58 
59 /* Note: Do not forget to add new entries to the Makefile as well. */
60 
61 #if CONFIG_AC3_MUXER
62 const AVOutputFormat ff_ac3_muxer = {
63     .name              = "ac3",
64     .long_name         = NULL_IF_CONFIG_SMALL("raw AC-3"),
65     .mime_type         = "audio/x-ac3",
66     .extensions        = "ac3",
67     .audio_codec       = AV_CODEC_ID_AC3,
68     .video_codec       = AV_CODEC_ID_NONE,
69     .init              = force_one_stream,
70     .write_packet      = ff_raw_write_packet,
71     .flags             = AVFMT_NOTIMESTAMPS,
72 };
73 #endif
74 
75 #if CONFIG_ADX_MUXER
76 
adx_write_trailer(AVFormatContext *s)77 static int adx_write_trailer(AVFormatContext *s)
78 {
79     AVIOContext *pb = s->pb;
80     AVCodecParameters *par = s->streams[0]->codecpar;
81 
82     if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
83         int64_t file_size = avio_tell(pb);
84         uint64_t sample_count = (file_size - 36) / par->ch_layout.nb_channels / 18 * 32;
85         if (sample_count <= UINT32_MAX) {
86             avio_seek(pb, 12, SEEK_SET);
87             avio_wb32(pb, sample_count);
88             avio_seek(pb, file_size, SEEK_SET);
89         }
90     }
91 
92     return 0;
93 }
94 
95 const AVOutputFormat ff_adx_muxer = {
96     .name              = "adx",
97     .long_name         = NULL_IF_CONFIG_SMALL("CRI ADX"),
98     .extensions        = "adx",
99     .audio_codec       = AV_CODEC_ID_ADPCM_ADX,
100     .video_codec       = AV_CODEC_ID_NONE,
101     .init              = force_one_stream,
102     .write_packet      = ff_raw_write_packet,
103     .write_trailer     = adx_write_trailer,
104     .flags             = AVFMT_NOTIMESTAMPS,
105 };
106 #endif
107 
108 #if CONFIG_APTX_MUXER
109 const AVOutputFormat ff_aptx_muxer = {
110     .name              = "aptx",
111     .long_name         = NULL_IF_CONFIG_SMALL("raw aptX (Audio Processing Technology for Bluetooth)"),
112     .extensions        = "aptx",
113     .audio_codec       = AV_CODEC_ID_APTX,
114     .video_codec       = AV_CODEC_ID_NONE,
115     .init              = force_one_stream,
116     .write_packet      = ff_raw_write_packet,
117     .flags             = AVFMT_NOTIMESTAMPS,
118 };
119 #endif
120 
121 #if CONFIG_APTX_HD_MUXER
122 const AVOutputFormat ff_aptx_hd_muxer = {
123     .name              = "aptx_hd",
124     .long_name         = NULL_IF_CONFIG_SMALL("raw aptX HD (Audio Processing Technology for Bluetooth)"),
125     .extensions        = "aptxhd",
126     .audio_codec       = AV_CODEC_ID_APTX_HD,
127     .video_codec       = AV_CODEC_ID_NONE,
128     .init              = force_one_stream,
129     .write_packet      = ff_raw_write_packet,
130     .flags             = AVFMT_NOTIMESTAMPS,
131 };
132 #endif
133 
134 #if CONFIG_AVS2_MUXER
135 const AVOutputFormat ff_avs2_muxer = {
136     .name              = "avs2",
137     .long_name         = NULL_IF_CONFIG_SMALL("raw AVS2-P2/IEEE1857.4 video"),
138     .extensions        = "avs,avs2",
139     .audio_codec       = AV_CODEC_ID_NONE,
140     .video_codec       = AV_CODEC_ID_AVS2,
141     .init              = force_one_stream,
142     .write_packet      = ff_raw_write_packet,
143     .flags             = AVFMT_NOTIMESTAMPS,
144 };
145 #endif
146 
147 #if CONFIG_AVS3_MUXER
148 const AVOutputFormat ff_avs3_muxer = {
149     .name              = "avs3",
150     .long_name         = NULL_IF_CONFIG_SMALL("AVS3-P2/IEEE1857.10"),
151     .extensions        = "avs3",
152     .audio_codec       = AV_CODEC_ID_NONE,
153     .video_codec       = AV_CODEC_ID_AVS3,
154     .init              = force_one_stream,
155     .write_packet      = ff_raw_write_packet,
156     .flags             = AVFMT_NOTIMESTAMPS,
157 };
158 #endif
159 
160 
161 #if CONFIG_CAVSVIDEO_MUXER
162 const AVOutputFormat ff_cavsvideo_muxer = {
163     .name              = "cavsvideo",
164     .long_name         = NULL_IF_CONFIG_SMALL("raw Chinese AVS (Audio Video Standard) video"),
165     .extensions        = "cavs",
166     .audio_codec       = AV_CODEC_ID_NONE,
167     .video_codec       = AV_CODEC_ID_CAVS,
168     .init              = force_one_stream,
169     .write_packet      = ff_raw_write_packet,
170     .flags             = AVFMT_NOTIMESTAMPS,
171 };
172 #endif
173 
174 #if CONFIG_CODEC2RAW_MUXER
175 const AVOutputFormat ff_codec2raw_muxer = {
176     .name              = "codec2raw",
177     .long_name         = NULL_IF_CONFIG_SMALL("raw codec2 muxer"),
178     .audio_codec       = AV_CODEC_ID_CODEC2,
179     .video_codec       = AV_CODEC_ID_NONE,
180     .init              = force_one_stream,
181     .write_packet      = ff_raw_write_packet,
182     .flags             = AVFMT_NOTIMESTAMPS,
183 };
184 #endif
185 
186 
187 #if CONFIG_DATA_MUXER
188 const AVOutputFormat ff_data_muxer = {
189     .name              = "data",
190     .long_name         = NULL_IF_CONFIG_SMALL("raw data"),
191     .init              = force_one_stream,
192     .write_packet      = ff_raw_write_packet,
193     .flags             = AVFMT_NOTIMESTAMPS,
194 };
195 #endif
196 
197 #if CONFIG_DFPWM_MUXER
198 const AVOutputFormat ff_dfpwm_muxer = {
199     .name              = "dfpwm",
200     .long_name         = NULL_IF_CONFIG_SMALL("raw DFPWM1a"),
201     .extensions        = "dfpwm",
202     .audio_codec       = AV_CODEC_ID_DFPWM,
203     .video_codec       = AV_CODEC_ID_NONE,
204     .init              = force_one_stream,
205     .write_packet      = ff_raw_write_packet,
206     .flags             = AVFMT_NOTIMESTAMPS,
207 };
208 #endif
209 
210 #if CONFIG_DIRAC_MUXER
211 const AVOutputFormat ff_dirac_muxer = {
212     .name              = "dirac",
213     .long_name         = NULL_IF_CONFIG_SMALL("raw Dirac"),
214     .extensions        = "drc,vc2",
215     .audio_codec       = AV_CODEC_ID_NONE,
216     .video_codec       = AV_CODEC_ID_DIRAC,
217     .init              = force_one_stream,
218     .write_packet      = ff_raw_write_packet,
219     .flags             = AVFMT_NOTIMESTAMPS,
220 };
221 #endif
222 
223 #if CONFIG_DNXHD_MUXER
224 const AVOutputFormat ff_dnxhd_muxer = {
225     .name              = "dnxhd",
226     .long_name         = NULL_IF_CONFIG_SMALL("raw DNxHD (SMPTE VC-3)"),
227     .extensions        = "dnxhd,dnxhr",
228     .audio_codec       = AV_CODEC_ID_NONE,
229     .video_codec       = AV_CODEC_ID_DNXHD,
230     .init              = force_one_stream,
231     .write_packet      = ff_raw_write_packet,
232     .flags             = AVFMT_NOTIMESTAMPS,
233 };
234 #endif
235 
236 #if CONFIG_DTS_MUXER
237 const AVOutputFormat ff_dts_muxer = {
238     .name              = "dts",
239     .long_name         = NULL_IF_CONFIG_SMALL("raw DTS"),
240     .mime_type         = "audio/x-dca",
241     .extensions        = "dts",
242     .audio_codec       = AV_CODEC_ID_DTS,
243     .video_codec       = AV_CODEC_ID_NONE,
244     .init              = force_one_stream,
245     .write_packet      = ff_raw_write_packet,
246     .flags             = AVFMT_NOTIMESTAMPS,
247 };
248 #endif
249 
250 #if CONFIG_EAC3_MUXER
251 const AVOutputFormat ff_eac3_muxer = {
252     .name              = "eac3",
253     .long_name         = NULL_IF_CONFIG_SMALL("raw E-AC-3"),
254     .mime_type         = "audio/x-eac3",
255     .extensions        = "eac3",
256     .audio_codec       = AV_CODEC_ID_EAC3,
257     .video_codec       = AV_CODEC_ID_NONE,
258     .init              = force_one_stream,
259     .write_packet      = ff_raw_write_packet,
260     .flags             = AVFMT_NOTIMESTAMPS,
261 };
262 #endif
263 
264 #if CONFIG_G722_MUXER
265 const AVOutputFormat ff_g722_muxer = {
266     .name              = "g722",
267     .long_name         = NULL_IF_CONFIG_SMALL("raw G.722"),
268     .mime_type         = "audio/G722",
269     .extensions        = "g722",
270     .audio_codec       = AV_CODEC_ID_ADPCM_G722,
271     .video_codec       = AV_CODEC_ID_NONE,
272     .init              = force_one_stream,
273     .write_packet      = ff_raw_write_packet,
274     .flags             = AVFMT_NOTIMESTAMPS,
275 };
276 #endif
277 
278 #if CONFIG_G723_1_MUXER
279 const AVOutputFormat ff_g723_1_muxer = {
280     .name              = "g723_1",
281     .long_name         = NULL_IF_CONFIG_SMALL("raw G.723.1"),
282     .mime_type         = "audio/g723",
283     .extensions        = "tco,rco",
284     .audio_codec       = AV_CODEC_ID_G723_1,
285     .video_codec       = AV_CODEC_ID_NONE,
286     .init              = force_one_stream,
287     .write_packet      = ff_raw_write_packet,
288     .flags             = AVFMT_NOTIMESTAMPS,
289 };
290 #endif
291 
292 #if CONFIG_G726_MUXER
293 const AVOutputFormat ff_g726_muxer = {
294     .name              = "g726",
295     .long_name         = NULL_IF_CONFIG_SMALL("raw big-endian G.726 (\"left-justified\")"),
296     .audio_codec       = AV_CODEC_ID_ADPCM_G726,
297     .video_codec       = AV_CODEC_ID_NONE,
298     .init              = force_one_stream,
299     .write_packet      = ff_raw_write_packet,
300     .flags             = AVFMT_NOTIMESTAMPS,
301 };
302 #endif
303 
304 #if CONFIG_G726LE_MUXER
305 const AVOutputFormat ff_g726le_muxer = {
306     .name              = "g726le",
307     .long_name         = NULL_IF_CONFIG_SMALL("raw little-endian G.726 (\"right-justified\")"),
308     .audio_codec       = AV_CODEC_ID_ADPCM_G726LE,
309     .video_codec       = AV_CODEC_ID_NONE,
310     .init              = force_one_stream,
311     .write_packet      = ff_raw_write_packet,
312     .flags             = AVFMT_NOTIMESTAMPS,
313 };
314 #endif
315 
316 #if CONFIG_GSM_MUXER
317 const AVOutputFormat ff_gsm_muxer = {
318     .name              = "gsm",
319     .long_name         = NULL_IF_CONFIG_SMALL("raw GSM"),
320     .mime_type         = "audio/x-gsm",
321     .extensions        = "gsm",
322     .audio_codec       = AV_CODEC_ID_GSM,
323     .video_codec       = AV_CODEC_ID_NONE,
324     .init              = force_one_stream,
325     .write_packet      = ff_raw_write_packet,
326     .flags             = AVFMT_NOTIMESTAMPS,
327 };
328 #endif
329 
330 #if CONFIG_H261_MUXER
331 const AVOutputFormat ff_h261_muxer = {
332     .name              = "h261",
333     .long_name         = NULL_IF_CONFIG_SMALL("raw H.261"),
334     .mime_type         = "video/x-h261",
335     .extensions        = "h261",
336     .audio_codec       = AV_CODEC_ID_NONE,
337     .video_codec       = AV_CODEC_ID_H261,
338     .init              = force_one_stream,
339     .write_packet      = ff_raw_write_packet,
340     .flags             = AVFMT_NOTIMESTAMPS,
341 };
342 #endif
343 
344 #if CONFIG_H263_MUXER
345 const AVOutputFormat ff_h263_muxer = {
346     .name              = "h263",
347     .long_name         = NULL_IF_CONFIG_SMALL("raw H.263"),
348     .mime_type         = "video/x-h263",
349     .extensions        = "h263",
350     .audio_codec       = AV_CODEC_ID_NONE,
351     .video_codec       = AV_CODEC_ID_H263,
352     .init              = force_one_stream,
353     .write_packet      = ff_raw_write_packet,
354     .flags             = AVFMT_NOTIMESTAMPS,
355 };
356 #endif
357 
358 #if CONFIG_H264_MUXER
h264_check_bitstream(AVFormatContext *s, AVStream *st, const AVPacket *pkt)359 static int h264_check_bitstream(AVFormatContext *s, AVStream *st,
360                                 const AVPacket *pkt)
361 {
362     if (pkt->size >= 5 && AV_RB32(pkt->data) != 0x0000001 &&
363                           AV_RB24(pkt->data) != 0x000001)
364         return ff_stream_add_bitstream_filter(st, "h264_mp4toannexb", NULL);
365     return 1;
366 }
367 
368 const AVOutputFormat ff_h264_muxer = {
369     .name              = "h264",
370     .long_name         = NULL_IF_CONFIG_SMALL("raw H.264 video"),
371     .extensions        = "h264,264",
372     .audio_codec       = AV_CODEC_ID_NONE,
373     .video_codec       = AV_CODEC_ID_H264,
374     .init              = force_one_stream,
375     .write_packet      = ff_raw_write_packet,
376     .check_bitstream   = h264_check_bitstream,
377     .flags             = AVFMT_NOTIMESTAMPS,
378 };
379 #endif
380 
381 #if CONFIG_HEVC_MUXER
hevc_check_bitstream(AVFormatContext *s, AVStream *st, const AVPacket *pkt)382 static int hevc_check_bitstream(AVFormatContext *s, AVStream *st,
383                                 const AVPacket *pkt)
384 {
385     if (pkt->size >= 5 && AV_RB32(pkt->data) != 0x0000001 &&
386                           AV_RB24(pkt->data) != 0x000001)
387         return ff_stream_add_bitstream_filter(st, "hevc_mp4toannexb", NULL);
388     return 1;
389 }
390 
391 const AVOutputFormat ff_hevc_muxer = {
392     .name              = "hevc",
393     .long_name         = NULL_IF_CONFIG_SMALL("raw HEVC video"),
394     .extensions        = "hevc,h265,265",
395     .audio_codec       = AV_CODEC_ID_NONE,
396     .video_codec       = AV_CODEC_ID_HEVC,
397     .init              = force_one_stream,
398     .write_packet      = ff_raw_write_packet,
399     .check_bitstream   = hevc_check_bitstream,
400     .flags             = AVFMT_NOTIMESTAMPS,
401 };
402 #endif
403 
404 #if CONFIG_M4V_MUXER
405 const AVOutputFormat ff_m4v_muxer = {
406     .name              = "m4v",
407     .long_name         = NULL_IF_CONFIG_SMALL("raw MPEG-4 video"),
408     .extensions        = "m4v",
409     .audio_codec       = AV_CODEC_ID_NONE,
410     .video_codec       = AV_CODEC_ID_MPEG4,
411     .init              = force_one_stream,
412     .write_packet      = ff_raw_write_packet,
413     .flags             = AVFMT_NOTIMESTAMPS,
414 };
415 #endif
416 
417 #if CONFIG_MJPEG_MUXER
418 const AVOutputFormat ff_mjpeg_muxer = {
419     .name              = "mjpeg",
420     .long_name         = NULL_IF_CONFIG_SMALL("raw MJPEG video"),
421     .mime_type         = "video/x-mjpeg",
422     .extensions        = "mjpg,mjpeg",
423     .audio_codec       = AV_CODEC_ID_NONE,
424     .video_codec       = AV_CODEC_ID_MJPEG,
425     .init              = force_one_stream,
426     .write_packet      = ff_raw_write_packet,
427     .flags             = AVFMT_NOTIMESTAMPS,
428 };
429 #endif
430 
431 #if CONFIG_MLP_MUXER
432 const AVOutputFormat ff_mlp_muxer = {
433     .name              = "mlp",
434     .long_name         = NULL_IF_CONFIG_SMALL("raw MLP"),
435     .extensions        = "mlp",
436     .audio_codec       = AV_CODEC_ID_MLP,
437     .video_codec       = AV_CODEC_ID_NONE,
438     .init              = force_one_stream,
439     .write_packet      = ff_raw_write_packet,
440     .flags             = AVFMT_NOTIMESTAMPS,
441 };
442 #endif
443 
444 #if CONFIG_MP2_MUXER
445 const AVOutputFormat ff_mp2_muxer = {
446     .name              = "mp2",
447     .long_name         = NULL_IF_CONFIG_SMALL("MP2 (MPEG audio layer 2)"),
448     .mime_type         = "audio/mpeg",
449     .extensions        = "mp2,m2a,mpa",
450     .audio_codec       = AV_CODEC_ID_MP2,
451     .video_codec       = AV_CODEC_ID_NONE,
452     .init              = force_one_stream,
453     .write_packet      = ff_raw_write_packet,
454     .flags             = AVFMT_NOTIMESTAMPS,
455 };
456 #endif
457 
458 #if CONFIG_MPEG1VIDEO_MUXER
459 const AVOutputFormat ff_mpeg1video_muxer = {
460     .name              = "mpeg1video",
461     .long_name         = NULL_IF_CONFIG_SMALL("raw MPEG-1 video"),
462     .mime_type         = "video/mpeg",
463     .extensions        = "mpg,mpeg,m1v",
464     .audio_codec       = AV_CODEC_ID_NONE,
465     .video_codec       = AV_CODEC_ID_MPEG1VIDEO,
466     .init              = force_one_stream,
467     .write_packet      = ff_raw_write_packet,
468     .flags             = AVFMT_NOTIMESTAMPS,
469 };
470 #endif
471 
472 #if CONFIG_MPEG2VIDEO_MUXER
473 const AVOutputFormat ff_mpeg2video_muxer = {
474     .name              = "mpeg2video",
475     .long_name         = NULL_IF_CONFIG_SMALL("raw MPEG-2 video"),
476     .extensions        = "m2v",
477     .audio_codec       = AV_CODEC_ID_NONE,
478     .video_codec       = AV_CODEC_ID_MPEG2VIDEO,
479     .init              = force_one_stream,
480     .write_packet      = ff_raw_write_packet,
481     .flags             = AVFMT_NOTIMESTAMPS,
482 };
483 #endif
484 
485 #if CONFIG_OBU_MUXER
obu_check_bitstream(AVFormatContext *s, AVStream *st, const AVPacket *pkt)486 static int obu_check_bitstream(AVFormatContext *s, AVStream *st,
487                                const AVPacket *pkt)
488 {
489     return ff_stream_add_bitstream_filter(st, "av1_metadata", "td=insert");
490 }
491 
492 const AVOutputFormat ff_obu_muxer = {
493     .name              = "obu",
494     .long_name         = NULL_IF_CONFIG_SMALL("AV1 low overhead OBU"),
495     .extensions        = "obu",
496     .audio_codec       = AV_CODEC_ID_NONE,
497     .video_codec       = AV_CODEC_ID_AV1,
498     .init              = force_one_stream,
499     .write_packet      = ff_raw_write_packet,
500     .check_bitstream   = obu_check_bitstream,
501     .flags             = AVFMT_NOTIMESTAMPS,
502 };
503 #endif
504 
505 #if CONFIG_RAWVIDEO_MUXER
506 const AVOutputFormat ff_rawvideo_muxer = {
507     .name              = "rawvideo",
508     .long_name         = NULL_IF_CONFIG_SMALL("raw video"),
509     .extensions        = "yuv,rgb",
510     .audio_codec       = AV_CODEC_ID_NONE,
511     .video_codec       = AV_CODEC_ID_RAWVIDEO,
512     .write_packet      = ff_raw_write_packet,
513     .flags             = AVFMT_NOTIMESTAMPS,
514 };
515 #endif
516 
517 #if CONFIG_SBC_MUXER
518 const AVOutputFormat ff_sbc_muxer = {
519     .name              = "sbc",
520     .long_name         = NULL_IF_CONFIG_SMALL("raw SBC"),
521     .mime_type         = "audio/x-sbc",
522     .extensions        = "sbc,msbc",
523     .audio_codec       = AV_CODEC_ID_SBC,
524     .init              = force_one_stream,
525     .write_packet      = ff_raw_write_packet,
526     .flags             = AVFMT_NOTIMESTAMPS,
527 };
528 #endif
529 
530 #if CONFIG_TRUEHD_MUXER
531 const AVOutputFormat ff_truehd_muxer = {
532     .name              = "truehd",
533     .long_name         = NULL_IF_CONFIG_SMALL("raw TrueHD"),
534     .extensions        = "thd",
535     .audio_codec       = AV_CODEC_ID_TRUEHD,
536     .video_codec       = AV_CODEC_ID_NONE,
537     .init              = force_one_stream,
538     .write_packet      = ff_raw_write_packet,
539     .flags             = AVFMT_NOTIMESTAMPS,
540 };
541 #endif
542 
543 #if CONFIG_VC1_MUXER
544 const AVOutputFormat ff_vc1_muxer = {
545     .name              = "vc1",
546     .long_name         = NULL_IF_CONFIG_SMALL("raw VC-1 video"),
547     .extensions        = "vc1",
548     .audio_codec       = AV_CODEC_ID_NONE,
549     .video_codec       = AV_CODEC_ID_VC1,
550     .init              = force_one_stream,
551     .write_packet      = ff_raw_write_packet,
552     .flags             = AVFMT_NOTIMESTAMPS,
553 };
554 #endif
555