src/add-ons/media/media-add-ons/dvb/pes.cpp

   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