xref: /haiku/src/add-ons/media/media-add-ons/dvb/pes.cpp (revision 820dca4df6c7bf955c46e8f6521b9408f50b2900)
1 /*
2  * Copyright (c) 2004-2007 Marcus Overhagen <marcus@overhagen.de>
3  *
4  * Permission is hereby granted, free of charge, to any person
5  * obtaining a copy of this software and associated documentation
6  * files (the "Software"), to deal in the Software without restriction,
7  * including without limitation the rights to use, copy, modify,
8  * merge, publish, distribute, sublicense, and/or sell copies of
9  * the Software, and to permit persons to whom the Software is
10  * furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be
13  * included in all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
17  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
19  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
20  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
22  * OTHER DEALINGS IN THE SOFTWARE.
23  */
24 
25 #include <stdio.h>
26 #include "pes.h"
27 
28 
29 status_t
30 pes_extract(const uint8 *pes_data, size_t pes_size, const uint8 **data, size_t *size)
31 {
32 	if (pes_size <= 9)
33 		return B_ERROR;
34 
35 	if (pes_data[0] || pes_data[1] || pes_data[2] != 1)
36 		return B_ERROR;
37 
38 	size_t header_size = 9 + pes_data[8];
39 
40 	if (pes_size <= header_size)
41 		return B_ERROR;
42 
43 //	printf("header size %d\n", header_size);
44 
45 	pes_data += header_size;
46 	pes_size -= header_size;
47 
48 	*data = pes_data;
49 	*size = pes_size;
50 
51 	return B_OK;
52 }
53 
54 
55 status_t
56 pes_stream_id(const uint8 *pes_data, size_t pes_size, int *stream_id)
57 {
58 	if (pes_size <= 9)
59 		return B_ERROR;
60 
61 	if (pes_data[0] || pes_data[1] || pes_data[2] != 1)
62 		return B_ERROR;
63 
64 	*stream_id = pes_data[3];
65 
66 	return B_OK;
67 }
68 
69