Add Haivision SRT protocol
[libav.git] / libavformat / iv8.c
CommitLineData
00b6fa40
MN
1/*
2 * Copyright (c) 2009 Michael Niedermayer
3 *
2912e87a 4 * This file is part of Libav.
00b6fa40 5 *
2912e87a 6 * Libav is free software; you can redistribute it and/or
00b6fa40
MN
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
2912e87a 11 * Libav is distributed in the hope that it will be useful,
00b6fa40
MN
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
2912e87a 17 * License along with Libav; if not, write to the Free Software
00b6fa40
MN
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21#include "avformat.h"
c3f9ebf7 22#include "internal.h"
00b6fa40
MN
23
24
25static int probe(AVProbeData *p)
26{
da9cea77 27 // the single file I have starts with that, I do not know if others do, too
00b6fa40
MN
28 if( p->buf[0] == 1
29 && p->buf[1] == 1
30 && p->buf[2] == 3
31 && p->buf[3] == 0xB8
32 && p->buf[4] == 0x80
33 && p->buf[5] == 0x60
34 )
35 return AVPROBE_SCORE_MAX-2;
36
37 return 0;
38}
39
6e9651d1 40static int read_header(AVFormatContext *s)
00b6fa40
MN
41{
42 AVStream *st;
43
3b3bbdd3 44 st = avformat_new_stream(s, NULL);
00b6fa40
MN
45 if (!st)
46 return AVERROR(ENOMEM);
47
9200514a
AK
48 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
49 st->codecpar->codec_id = AV_CODEC_ID_MPEG4;
00b6fa40 50 st->need_parsing = AVSTREAM_PARSE_FULL;
c3f9ebf7 51 avpriv_set_pts_info(st, 64, 1, 90000);
00b6fa40
MN
52
53 return 0;
54
55}
56
57static int read_packet(AVFormatContext *s, AVPacket *pkt)
58{
cf543c65
JG
59 int ret, size, pts, type, flags;
60 int first_pkt = 0;
61 int frame_complete = 0;
62
63 while (!frame_complete) {
64
65 type = avio_rb16(s->pb); // 257 or 258
66 size = avio_rb16(s->pb);
67 flags = avio_rb16(s->pb); //some flags, 0x80 indicates end of frame
68 avio_rb16(s->pb); //packet number
69 pts = avio_rb32(s->pb);
70 avio_rb32(s->pb); //6A 13 E3 88
71
72 frame_complete = flags & 0x80;
73
74 size -= 12;
75 if (size < 1)
76 return -1;
77
78 if (type == 258) {
79 avio_skip(s->pb, size);
80 frame_complete = 0;
81 continue;
82 }
83
84 if (!first_pkt) {
85 ret = av_get_packet(s->pb, pkt, size);
86 if (ret < 0)
87 return ret;
88 first_pkt = 1;
89 pkt->pts = pts;
90 pkt->pos -= 16;
91 } else {
92 ret = av_append_packet(s->pb, pkt, size);
93 if (ret < 0) {
94 av_log(s, AV_LOG_ERROR, "failed to grow packet\n");
ce70f28a 95 av_packet_unref(pkt);
cf543c65
JG
96 return ret;
97 }
98 }
99 if (ret < size) {
100 av_log(s, AV_LOG_ERROR, "Truncated packet! Read %d of %d bytes\n",
101 ret, size);
102 pkt->flags |= AV_PKT_FLAG_CORRUPT;
103 break;
104 }
f2d56513 105 }
00b6fa40
MN
106 pkt->stream_index = 0;
107
cf543c65 108 return 0;
00b6fa40
MN
109}
110
c6610a21 111AVInputFormat ff_iv8_demuxer = {
dfc2c4d9 112 .name = "iv8",
0177b7d2 113 .long_name = NULL_IF_CONFIG_SMALL("IndigoVision 8000 video"),
dfc2c4d9
AK
114 .read_probe = probe,
115 .read_header = read_header,
116 .read_packet = read_packet,
20234a4b 117 .flags = AVFMT_GENERIC_INDEX,
00b6fa40 118};