From c6f3b52cad3b0768699f5a1a6a0df456dfc2cfd7 Mon Sep 17 00:00:00 2001 From: Christophe Massiot Date: Mon, 10 Oct 2011 12:45:27 +0300 Subject: [PATCH] mpeg/psi: Move section functions to mpeg/psi/psi.h --- mpeg/psi.h | 491 +------------------------------------------- mpeg/psi/psi.h | 538 +++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 539 insertions(+), 490 deletions(-) create mode 100644 mpeg/psi/psi.h diff --git a/mpeg/psi.h b/mpeg/psi.h index dcd345d..b3bf069 100644 --- a/mpeg/psi.h +++ b/mpeg/psi.h @@ -37,6 +37,7 @@ #include #include #include +#include #ifdef __cplusplus extern "C" @@ -189,496 +190,6 @@ static inline void desc0a_print(uint8_t *p_desc, f_print pf_print, } } -/***************************************************************************** - * p_psi_crc_table - ***************************************************************************** - * This table is used to compute a PSI CRC byte per byte instead of bit per - * bit. It's been generated by 'gen_crc' in the 'misc' directory: - * - * uint32_t table[256]; - * uint32_t i, j, k; - * - * for(i = 0; i < 256; i++) - * { - * k = 0; - * for (j = (i << 24) | 0x800000; j != 0x80000000; j <<= 1) - * k = (k << 1) ^ (((k ^ j) & 0x80000000) ? 0x04c11db7 : 0); - * table[i] = k; - * } - * - * A CRC is computed like this: - * - * initialization - * -------------- - * uint32_t i_crc = 0xffffffff; - * - * for each data byte do - * --------------------- - * i_crc = (i_crc << 8) ^ s_crc32_table[(i_crc >> 24) ^ (data_byte)]; - *****************************************************************************/ -static const uint32_t p_psi_crc_table[256] = { - 0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, - 0x130476dc, 0x17c56b6b, 0x1a864db2, 0x1e475005, - 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61, - 0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, - 0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9, - 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75, - 0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, - 0x791d4014, 0x7ddc5da3, 0x709f7b7a, 0x745e66cd, - 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039, - 0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, - 0xbe2b5b58, 0xbaea46ef, 0xb7a96036, 0xb3687d81, - 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d, - 0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, - 0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95, - 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1, - 0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, - 0x34867077, 0x30476dc0, 0x3d044b19, 0x39c556ae, - 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072, - 0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, - 0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca, - 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde, - 0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, - 0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1, 0x53dc6066, - 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba, - 0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, - 0xbfa1b04b, 0xbb60adfc, 0xb6238b25, 0xb2e29692, - 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6, - 0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, - 0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e, - 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2, - 0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, - 0xd5b88683, 0xd1799b34, 0xdc3abded, 0xd8fba05a, - 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637, - 0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, - 0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f, - 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53, - 0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, - 0x36194d42, 0x32d850f5, 0x3f9b762c, 0x3b5a6b9b, - 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff, - 0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, - 0xf12f560e, 0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7, - 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b, - 0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, - 0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3, - 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7, - 0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, - 0x9b3660c6, 0x9ff77d71, 0x92b45ba8, 0x9675461f, - 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3, - 0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, - 0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c, - 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8, - 0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, - 0x119b4be9, 0x155a565e, 0x18197087, 0x1cd86d30, - 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec, - 0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, - 0x2497d08d, 0x2056cd3a, 0x2d15ebe3, 0x29d4f654, - 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0, - 0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, - 0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18, - 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4, - 0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, - 0x9abc8bd5, 0x9e7d9662, 0x933eb0bb, 0x97ffad0c, - 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668, - 0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4 -}; - -/***************************************************************************** - * PSI section - *****************************************************************************/ -#define PSI_HEADER_SIZE 3 -#define PSI_HEADER_SIZE_SYNTAX1 8 -#define PSI_CRC_SIZE 4 -#define PSI_MAX_SIZE 1021 -#define PSI_PRIVATE_MAX_SIZE 4093 - -#define PSI_DECLARE(p_table) \ - uint8_t p_table[PSI_MAX_SIZE + PSI_HEADER_SIZE] -#define PSI_PRIVATE_DECLARE(p_table) \ - uint8_t p_table[PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE] - -static inline uint8_t *psi_allocate(void) -{ - return malloc((PSI_MAX_SIZE + PSI_HEADER_SIZE) * sizeof(uint8_t)); -} - -static inline uint8_t *psi_private_allocate(void) -{ - return malloc((PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE) * sizeof(uint8_t)); -} - -static inline void psi_set_tableid(uint8_t *p_section, uint8_t i_table_id) -{ - p_section[0] = i_table_id; -} - -static inline uint8_t psi_get_tableid(const uint8_t *p_section) -{ - return p_section[0]; -} - -static inline void psi_set_syntax(uint8_t *p_section) -{ - p_section[1] |= 0x80; -} - -static inline bool psi_get_syntax(const uint8_t *p_section) -{ - return !!(p_section[1] & 0x80); -} - -static inline void psi_init(uint8_t *p_section, bool b_syntax) -{ - /* set reserved bits */ - p_section[1] = 0x70; - if (b_syntax) { - psi_set_syntax(p_section); - p_section[5] = 0xc0; - } -} - -static inline void psi_set_length(uint8_t *p_section, uint16_t i_length) -{ - p_section[1] &= ~0xf; - p_section[1] |= (i_length >> 8) & 0xf; - p_section[2] = i_length & 0xff; -} - -static inline uint16_t psi_get_length(const uint8_t *p_section) -{ - return ((p_section[1] & 0xf) << 8) | p_section[2]; -} - -static inline void psi_set_tableidext(uint8_t *p_section, - uint16_t i_table_id_ext) -{ - p_section[3] = i_table_id_ext >> 8; - p_section[4] = i_table_id_ext & 0xff; -} - -static inline uint16_t psi_get_tableidext(const uint8_t *p_section) -{ - return (p_section[3] << 8) | p_section[4]; -} - -static inline void psi_set_version(uint8_t *p_section, uint8_t i_version) -{ - p_section[5] = i_version << 1; -} - -static inline uint8_t psi_get_version(const uint8_t *p_section) -{ - return (p_section[5] & 0x1e) >> 1; -} - -static inline void psi_set_current(uint8_t *p_section) -{ - p_section[5] |= 0x1; -} - -static inline bool psi_get_current(const uint8_t *p_section) -{ - return !!(p_section[5] & 0x1); -} - -static inline void psi_set_section(uint8_t *p_section, uint8_t i_section) -{ - p_section[6] = i_section; -} - -static inline uint8_t psi_get_section(const uint8_t *p_section) -{ - return p_section[6]; -} - -static inline void psi_set_lastsection(uint8_t *p_section, - uint8_t i_last_section) -{ - p_section[7] = i_last_section; -} - -static inline uint8_t psi_get_lastsection(const uint8_t *p_section) -{ - return p_section[7]; -} - -/***************************************************************************** - * PSI validation - *****************************************************************************/ -static inline void psi_set_crc(uint8_t *p_section) -{ - uint32_t i_crc = 0xffffffff; - uint16_t i_end = (((p_section[1] & 0xf) << 8) | p_section[2]) - + PSI_HEADER_SIZE - PSI_CRC_SIZE; - uint16_t i; - - for (i = 0; i < i_end; i++) - i_crc = (i_crc << 8) ^ p_psi_crc_table[(i_crc >> 24) ^ (p_section[i])]; - - p_section[i_end] = i_crc >> 24; - p_section[i_end + 1] = (i_crc >> 16) & 0xff; - p_section[i_end + 2] = (i_crc >> 8) & 0xff; - p_section[i_end + 3] = i_crc & 0xff; -} - -static inline bool psi_check_crc(const uint8_t *p_section) -{ - uint32_t i_crc = 0xffffffff; - uint16_t i_end = (((p_section[1] & 0xf) << 8) | p_section[2]) - + PSI_HEADER_SIZE - PSI_CRC_SIZE; - uint16_t i; - - for (i = 0; i < i_end; i++) - i_crc = (i_crc << 8) ^ p_psi_crc_table[(i_crc >> 24) ^ (p_section[i])]; - - return p_section[i_end] == (i_crc >> 24) - && p_section[i_end + 1] == ((i_crc >> 16) & 0xff) - && p_section[i_end + 2] == ((i_crc >> 8) & 0xff) - && p_section[i_end + 3] == (i_crc & 0xff); -} - -static inline bool psi_validate(const uint8_t *p_section) -{ - if (psi_get_syntax(p_section) - && (psi_get_length(p_section) < PSI_HEADER_SIZE_SYNTAX1 - - PSI_HEADER_SIZE + PSI_CRC_SIZE)) - return false; - - /* only do the CRC check when it is strictly necessary */ - - return true; -} - -static inline bool psi_compare(const uint8_t *p_section1, - const uint8_t *p_section2) -{ - return psi_get_version(p_section1) == psi_get_version(p_section2) - && psi_get_length(p_section1) == psi_get_length(p_section2) - && !memcmp(p_section1, p_section2, - psi_get_length(p_section1) + PSI_HEADER_SIZE); -} - -/***************************************************************************** - * PSI section gathering - *****************************************************************************/ -static inline void psi_assemble_init(uint8_t **pp_psi_buffer, - uint16_t *pi_psi_buffer_used) -{ - *pp_psi_buffer = NULL; - *pi_psi_buffer_used = 0; -} - -static inline void psi_assemble_reset(uint8_t **pp_psi_buffer, - uint16_t *pi_psi_buffer_used) -{ - free(*pp_psi_buffer); - psi_assemble_init(pp_psi_buffer, pi_psi_buffer_used); -} - -static inline bool psi_assemble_empty(uint8_t **pp_psi_buffer, - uint16_t *pi_psi_buffer_used) -{ - return *pp_psi_buffer == NULL; -} - -static inline uint8_t *psi_assemble_payload(uint8_t **pp_psi_buffer, - uint16_t *pi_psi_buffer_used, - const uint8_t **pp_payload, - uint8_t *pi_length) -{ - uint16_t i_remaining_size = PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE - - *pi_psi_buffer_used; - uint16_t i_copy_size = *pi_length < i_remaining_size ? *pi_length : - i_remaining_size; - uint8_t *p_section = NULL; - - if (*pp_psi_buffer == NULL) { - if (**pp_payload == 0xff) { - /* padding table to the end of buffer */ - *pi_length = 0; - return NULL; - } - *pp_psi_buffer = psi_private_allocate(); - } - - memcpy(*pp_psi_buffer + *pi_psi_buffer_used, *pp_payload, i_copy_size); - *pi_psi_buffer_used += i_copy_size; - - if (*pi_psi_buffer_used >= PSI_HEADER_SIZE) { - uint16_t i_section_size = psi_get_length(*pp_psi_buffer) - + PSI_HEADER_SIZE; - - if (i_section_size > PSI_PRIVATE_MAX_SIZE) { - /* invalid section */ - psi_assemble_reset(pp_psi_buffer, pi_psi_buffer_used); - *pi_length = 0; - return NULL; - } - if (i_section_size <= *pi_psi_buffer_used) { - p_section = *pp_psi_buffer; - i_copy_size -= (*pi_psi_buffer_used - i_section_size); - *pp_psi_buffer = NULL; - *pi_psi_buffer_used = 0; - } - } - - *pp_payload += i_copy_size; - *pi_length -= i_copy_size; - return p_section; -} - -/***************************************************************************** - * PSI section splitting - *****************************************************************************/ -static inline void psi_split_end(uint8_t *p_ts, uint8_t *pi_ts_offset) -{ - if (*pi_ts_offset != TS_SIZE) { - memset(p_ts + *pi_ts_offset, 0xff, TS_SIZE - *pi_ts_offset); - *pi_ts_offset = TS_SIZE; - } -} - -static inline void psi_split_section(uint8_t *p_ts, uint8_t *pi_ts_offset, - const uint8_t *p_section, - uint16_t *pi_section_offset) -{ - uint16_t i_section_length = psi_get_length(p_section) + PSI_HEADER_SIZE - - *pi_section_offset; - uint8_t i_ts_length, i_copy; - - if (!*pi_ts_offset) { - ts_init(p_ts); - ts_set_payload(p_ts); - *pi_ts_offset = ts_payload(p_ts) - p_ts; - } - - if (!*pi_section_offset) { - if (TS_SIZE - *pi_ts_offset < 2) { - psi_split_end(p_ts, pi_ts_offset); - return; - } - if (!ts_get_unitstart(p_ts)) { - uint8_t *p_payload = ts_payload(p_ts); - uint8_t i_payload_length = *pi_ts_offset - (p_payload - p_ts); - if (i_payload_length) - memmove(p_payload + 1, p_payload, i_payload_length); - (*pi_ts_offset)++; - *p_payload = i_payload_length; /* pointer_field */ - ts_set_unitstart(p_ts); - } - } - i_ts_length = TS_SIZE - *pi_ts_offset; - - i_copy = i_ts_length < i_section_length ? - i_ts_length : i_section_length; - memcpy(p_ts + *pi_ts_offset, p_section + *pi_section_offset, i_copy); - *pi_ts_offset += i_copy; - *pi_section_offset += i_copy; -} - -/***************************************************************************** - * PSI table gathering - *****************************************************************************/ -#define PSI_TABLE_MAX_SECTIONS 256 - -#define PSI_TABLE_DECLARE(pp_table) \ - uint8_t *pp_table[PSI_TABLE_MAX_SECTIONS] - -static inline uint8_t **psi_table_allocate(void) -{ - return malloc(PSI_TABLE_MAX_SECTIONS * sizeof(uint8_t *)); -} - -static inline void psi_table_init(uint8_t **pp_sections) -{ - int i; - for (i = 0; i < PSI_TABLE_MAX_SECTIONS; i++) - pp_sections[i] = NULL; -} - -static inline void psi_table_free(uint8_t **pp_sections) -{ - int i; - for (i = 0; i < PSI_TABLE_MAX_SECTIONS; i++) - free(pp_sections[i]); -} - -static inline bool psi_table_validate(uint8_t * const *pp_sections) -{ - return pp_sections[0] != NULL; -} - -static inline void psi_table_copy(uint8_t **pp_dest, uint8_t **pp_src) -{ - memcpy(pp_dest, pp_src, PSI_TABLE_MAX_SECTIONS * sizeof(uint8_t *)); -} - -#define psi_table_get_tableid(pp_sections) \ - psi_get_tableid(pp_sections[0]) -#define psi_table_get_version(pp_sections) \ - psi_get_version(pp_sections[0]) -#define psi_table_get_current(pp_sections) \ - psi_get_current(pp_sections[0]) -#define psi_table_get_lastsection(pp_sections) \ - psi_get_lastsection(pp_sections[0]) -#define psi_table_get_tableidext(pp_sections) \ - psi_get_tableidext(pp_sections[0]) - -static inline bool psi_table_section(uint8_t **pp_sections, uint8_t *p_section) -{ - uint8_t i_section = psi_get_section( p_section ); - uint8_t i_last_section = psi_get_lastsection( p_section ); - uint8_t i_version = psi_get_version( p_section ); - uint16_t i_tableidext = psi_get_tableidext( p_section ); - int i; - - free(pp_sections[i_section]); - pp_sections[i_section] = p_section; - - for (i = 0; i <= i_last_section; i++) { - uint8_t *p = pp_sections[i]; - if (p == NULL) - return false; - if (psi_get_lastsection(p) != i_last_section - || psi_get_version(p) != i_version - || psi_get_tableidext(p) != i_tableidext) - return false; - } - - /* free spurious, invalid sections */ - for (; i < PSI_TABLE_MAX_SECTIONS; i++) { - free(pp_sections[i]); - pp_sections[i] = NULL; - } - - /* a new, full table is available */ - return true; -} - -static inline uint8_t *psi_table_get_section(uint8_t **pp_sections, uint8_t n) -{ - return pp_sections[n]; -} - -static inline bool psi_table_compare(uint8_t **pp_sections1, - uint8_t **pp_sections2) -{ - uint8_t i_last_section = psi_table_get_lastsection(pp_sections1); - uint8_t i; - - if (i_last_section != psi_table_get_lastsection(pp_sections2)) - return false; - - for (i = 0; i <= i_last_section; i++) { - const uint8_t *p_section1 = psi_table_get_section(pp_sections1, i); - const uint8_t *p_section2 = psi_table_get_section(pp_sections2, i); - if (!psi_compare(p_section1, p_section2)) - return false; - } - - return true; -} - /***************************************************************************** * Program Association Table *****************************************************************************/ diff --git a/mpeg/psi/psi.h b/mpeg/psi/psi.h new file mode 100644 index 0000000..67916e8 --- /dev/null +++ b/mpeg/psi/psi.h @@ -0,0 +1,538 @@ +/***************************************************************************** + * psi.h: ISO/IEC 13818-1 Program Specific Information (sections) + ***************************************************************************** + * Copyright (C) 2009-2010 VideoLAN + * + * Authors: Christophe Massiot + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * "Software"), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject + * to the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + *****************************************************************************/ + +/* + * Normative references: + * - ISO/IEC 13818-1:2007(E) (MPEG-2 Systems) + */ + +#ifndef __BITSTREAM_MPEG_SECTIONS_H__ +#define __BITSTREAM_MPEG_SECTIONS_H__ + +#include +#include + +#ifdef __cplusplus +extern "C" +{ +#endif + +/***************************************************************************** + * p_psi_crc_table + ***************************************************************************** + * This table is used to compute a PSI CRC byte per byte instead of bit per + * bit. It's been generated by 'gen_crc' in the 'misc' directory: + * + * uint32_t table[256]; + * uint32_t i, j, k; + * + * for(i = 0; i < 256; i++) + * { + * k = 0; + * for (j = (i << 24) | 0x800000; j != 0x80000000; j <<= 1) + * k = (k << 1) ^ (((k ^ j) & 0x80000000) ? 0x04c11db7 : 0); + * table[i] = k; + * } + * + * A CRC is computed like this: + * + * initialization + * -------------- + * uint32_t i_crc = 0xffffffff; + * + * for each data byte do + * --------------------- + * i_crc = (i_crc << 8) ^ s_crc32_table[(i_crc >> 24) ^ (data_byte)]; + *****************************************************************************/ +static const uint32_t p_psi_crc_table[256] = { + 0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, + 0x130476dc, 0x17c56b6b, 0x1a864db2, 0x1e475005, + 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61, + 0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, + 0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9, + 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75, + 0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, + 0x791d4014, 0x7ddc5da3, 0x709f7b7a, 0x745e66cd, + 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039, + 0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, + 0xbe2b5b58, 0xbaea46ef, 0xb7a96036, 0xb3687d81, + 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d, + 0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, + 0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95, + 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1, + 0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, + 0x34867077, 0x30476dc0, 0x3d044b19, 0x39c556ae, + 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072, + 0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, + 0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca, + 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde, + 0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, + 0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1, 0x53dc6066, + 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba, + 0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, + 0xbfa1b04b, 0xbb60adfc, 0xb6238b25, 0xb2e29692, + 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6, + 0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, + 0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e, + 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2, + 0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, + 0xd5b88683, 0xd1799b34, 0xdc3abded, 0xd8fba05a, + 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637, + 0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, + 0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f, + 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53, + 0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, + 0x36194d42, 0x32d850f5, 0x3f9b762c, 0x3b5a6b9b, + 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff, + 0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, + 0xf12f560e, 0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7, + 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b, + 0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, + 0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3, + 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7, + 0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, + 0x9b3660c6, 0x9ff77d71, 0x92b45ba8, 0x9675461f, + 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3, + 0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, + 0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c, + 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8, + 0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, + 0x119b4be9, 0x155a565e, 0x18197087, 0x1cd86d30, + 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec, + 0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, + 0x2497d08d, 0x2056cd3a, 0x2d15ebe3, 0x29d4f654, + 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0, + 0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, + 0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18, + 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4, + 0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, + 0x9abc8bd5, 0x9e7d9662, 0x933eb0bb, 0x97ffad0c, + 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668, + 0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4 +}; + +/***************************************************************************** + * PSI section + *****************************************************************************/ +#define PSI_HEADER_SIZE 3 +#define PSI_HEADER_SIZE_SYNTAX1 8 +#define PSI_CRC_SIZE 4 +#define PSI_MAX_SIZE 1021 +#define PSI_PRIVATE_MAX_SIZE 4093 + +#define PSI_DECLARE(p_table) \ + uint8_t p_table[PSI_MAX_SIZE + PSI_HEADER_SIZE] +#define PSI_PRIVATE_DECLARE(p_table) \ + uint8_t p_table[PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE] + +static inline uint8_t *psi_allocate(void) +{ + return malloc((PSI_MAX_SIZE + PSI_HEADER_SIZE) * sizeof(uint8_t)); +} + +static inline uint8_t *psi_private_allocate(void) +{ + return malloc((PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE) * sizeof(uint8_t)); +} + +static inline void psi_set_tableid(uint8_t *p_section, uint8_t i_table_id) +{ + p_section[0] = i_table_id; +} + +static inline uint8_t psi_get_tableid(const uint8_t *p_section) +{ + return p_section[0]; +} + +static inline void psi_set_syntax(uint8_t *p_section) +{ + p_section[1] |= 0x80; +} + +static inline bool psi_get_syntax(const uint8_t *p_section) +{ + return !!(p_section[1] & 0x80); +} + +static inline void psi_init(uint8_t *p_section, bool b_syntax) +{ + /* set reserved bits */ + p_section[1] = 0x70; + if (b_syntax) { + psi_set_syntax(p_section); + p_section[5] = 0xc0; + } +} + +static inline void psi_set_length(uint8_t *p_section, uint16_t i_length) +{ + p_section[1] &= ~0xf; + p_section[1] |= (i_length >> 8) & 0xf; + p_section[2] = i_length & 0xff; +} + +static inline uint16_t psi_get_length(const uint8_t *p_section) +{ + return ((p_section[1] & 0xf) << 8) | p_section[2]; +} + +static inline void psi_set_tableidext(uint8_t *p_section, + uint16_t i_table_id_ext) +{ + p_section[3] = i_table_id_ext >> 8; + p_section[4] = i_table_id_ext & 0xff; +} + +static inline uint16_t psi_get_tableidext(const uint8_t *p_section) +{ + return (p_section[3] << 8) | p_section[4]; +} + +static inline void psi_set_version(uint8_t *p_section, uint8_t i_version) +{ + p_section[5] = i_version << 1; +} + +static inline uint8_t psi_get_version(const uint8_t *p_section) +{ + return (p_section[5] & 0x1e) >> 1; +} + +static inline void psi_set_current(uint8_t *p_section) +{ + p_section[5] |= 0x1; +} + +static inline bool psi_get_current(const uint8_t *p_section) +{ + return !!(p_section[5] & 0x1); +} + +static inline void psi_set_section(uint8_t *p_section, uint8_t i_section) +{ + p_section[6] = i_section; +} + +static inline uint8_t psi_get_section(const uint8_t *p_section) +{ + return p_section[6]; +} + +static inline void psi_set_lastsection(uint8_t *p_section, + uint8_t i_last_section) +{ + p_section[7] = i_last_section; +} + +static inline uint8_t psi_get_lastsection(const uint8_t *p_section) +{ + return p_section[7]; +} + +/***************************************************************************** + * PSI validation + *****************************************************************************/ +static inline void psi_set_crc(uint8_t *p_section) +{ + uint32_t i_crc = 0xffffffff; + uint16_t i_end = (((p_section[1] & 0xf) << 8) | p_section[2]) + + PSI_HEADER_SIZE - PSI_CRC_SIZE; + uint16_t i; + + for (i = 0; i < i_end; i++) + i_crc = (i_crc << 8) ^ p_psi_crc_table[(i_crc >> 24) ^ (p_section[i])]; + + p_section[i_end] = i_crc >> 24; + p_section[i_end + 1] = (i_crc >> 16) & 0xff; + p_section[i_end + 2] = (i_crc >> 8) & 0xff; + p_section[i_end + 3] = i_crc & 0xff; +} + +static inline bool psi_check_crc(const uint8_t *p_section) +{ + uint32_t i_crc = 0xffffffff; + uint16_t i_end = (((p_section[1] & 0xf) << 8) | p_section[2]) + + PSI_HEADER_SIZE - PSI_CRC_SIZE; + uint16_t i; + + for (i = 0; i < i_end; i++) + i_crc = (i_crc << 8) ^ p_psi_crc_table[(i_crc >> 24) ^ (p_section[i])]; + + return p_section[i_end] == (i_crc >> 24) + && p_section[i_end + 1] == ((i_crc >> 16) & 0xff) + && p_section[i_end + 2] == ((i_crc >> 8) & 0xff) + && p_section[i_end + 3] == (i_crc & 0xff); +} + +static inline bool psi_validate(const uint8_t *p_section) +{ + if (psi_get_syntax(p_section) + && (psi_get_length(p_section) < PSI_HEADER_SIZE_SYNTAX1 + - PSI_HEADER_SIZE + PSI_CRC_SIZE)) + return false; + + /* only do the CRC check when it is strictly necessary */ + + return true; +} + +static inline bool psi_compare(const uint8_t *p_section1, + const uint8_t *p_section2) +{ + return psi_get_version(p_section1) == psi_get_version(p_section2) + && psi_get_length(p_section1) == psi_get_length(p_section2) + && !memcmp(p_section1, p_section2, + psi_get_length(p_section1) + PSI_HEADER_SIZE); +} + +/***************************************************************************** + * PSI section gathering + *****************************************************************************/ +static inline void psi_assemble_init(uint8_t **pp_psi_buffer, + uint16_t *pi_psi_buffer_used) +{ + *pp_psi_buffer = NULL; + *pi_psi_buffer_used = 0; +} + +static inline void psi_assemble_reset(uint8_t **pp_psi_buffer, + uint16_t *pi_psi_buffer_used) +{ + free(*pp_psi_buffer); + psi_assemble_init(pp_psi_buffer, pi_psi_buffer_used); +} + +static inline bool psi_assemble_empty(uint8_t **pp_psi_buffer, + uint16_t *pi_psi_buffer_used) +{ + return *pp_psi_buffer == NULL; +} + +static inline uint8_t *psi_assemble_payload(uint8_t **pp_psi_buffer, + uint16_t *pi_psi_buffer_used, + const uint8_t **pp_payload, + uint8_t *pi_length) +{ + uint16_t i_remaining_size = PSI_PRIVATE_MAX_SIZE + PSI_HEADER_SIZE + - *pi_psi_buffer_used; + uint16_t i_copy_size = *pi_length < i_remaining_size ? *pi_length : + i_remaining_size; + uint8_t *p_section = NULL; + + if (*pp_psi_buffer == NULL) { + if (**pp_payload == 0xff) { + /* padding table to the end of buffer */ + *pi_length = 0; + return NULL; + } + *pp_psi_buffer = psi_private_allocate(); + } + + memcpy(*pp_psi_buffer + *pi_psi_buffer_used, *pp_payload, i_copy_size); + *pi_psi_buffer_used += i_copy_size; + + if (*pi_psi_buffer_used >= PSI_HEADER_SIZE) { + uint16_t i_section_size = psi_get_length(*pp_psi_buffer) + + PSI_HEADER_SIZE; + + if (i_section_size > PSI_PRIVATE_MAX_SIZE) { + /* invalid section */ + psi_assemble_reset(pp_psi_buffer, pi_psi_buffer_used); + *pi_length = 0; + return NULL; + } + if (i_section_size <= *pi_psi_buffer_used) { + p_section = *pp_psi_buffer; + i_copy_size -= (*pi_psi_buffer_used - i_section_size); + *pp_psi_buffer = NULL; + *pi_psi_buffer_used = 0; + } + } + + *pp_payload += i_copy_size; + *pi_length -= i_copy_size; + return p_section; +} + +/***************************************************************************** + * PSI section splitting + *****************************************************************************/ +static inline void psi_split_end(uint8_t *p_ts, uint8_t *pi_ts_offset) +{ + if (*pi_ts_offset != TS_SIZE) { + memset(p_ts + *pi_ts_offset, 0xff, TS_SIZE - *pi_ts_offset); + *pi_ts_offset = TS_SIZE; + } +} + +static inline void psi_split_section(uint8_t *p_ts, uint8_t *pi_ts_offset, + const uint8_t *p_section, + uint16_t *pi_section_offset) +{ + uint16_t i_section_length = psi_get_length(p_section) + PSI_HEADER_SIZE + - *pi_section_offset; + uint8_t i_ts_length, i_copy; + + if (!*pi_ts_offset) { + ts_init(p_ts); + ts_set_payload(p_ts); + *pi_ts_offset = ts_payload(p_ts) - p_ts; + } + + if (!*pi_section_offset) { + if (TS_SIZE - *pi_ts_offset < 2) { + psi_split_end(p_ts, pi_ts_offset); + return; + } + if (!ts_get_unitstart(p_ts)) { + uint8_t *p_payload = ts_payload(p_ts); + uint8_t i_payload_length = *pi_ts_offset - (p_payload - p_ts); + if (i_payload_length) + memmove(p_payload + 1, p_payload, i_payload_length); + (*pi_ts_offset)++; + *p_payload = i_payload_length; /* pointer_field */ + ts_set_unitstart(p_ts); + } + } + i_ts_length = TS_SIZE - *pi_ts_offset; + + i_copy = i_ts_length < i_section_length ? + i_ts_length : i_section_length; + memcpy(p_ts + *pi_ts_offset, p_section + *pi_section_offset, i_copy); + *pi_ts_offset += i_copy; + *pi_section_offset += i_copy; +} + +/***************************************************************************** + * PSI table gathering + *****************************************************************************/ +#define PSI_TABLE_MAX_SECTIONS 256 + +#define PSI_TABLE_DECLARE(pp_table) \ + uint8_t *pp_table[PSI_TABLE_MAX_SECTIONS] + +static inline uint8_t **psi_table_allocate(void) +{ + return malloc(PSI_TABLE_MAX_SECTIONS * sizeof(uint8_t *)); +} + +static inline void psi_table_init(uint8_t **pp_sections) +{ + int i; + for (i = 0; i < PSI_TABLE_MAX_SECTIONS; i++) + pp_sections[i] = NULL; +} + +static inline void psi_table_free(uint8_t **pp_sections) +{ + int i; + for (i = 0; i < PSI_TABLE_MAX_SECTIONS; i++) + free(pp_sections[i]); +} + +static inline bool psi_table_validate(uint8_t * const *pp_sections) +{ + return pp_sections[0] != NULL; +} + +static inline void psi_table_copy(uint8_t **pp_dest, uint8_t **pp_src) +{ + memcpy(pp_dest, pp_src, PSI_TABLE_MAX_SECTIONS * sizeof(uint8_t *)); +} + +#define psi_table_get_tableid(pp_sections) \ + psi_get_tableid(pp_sections[0]) +#define psi_table_get_version(pp_sections) \ + psi_get_version(pp_sections[0]) +#define psi_table_get_current(pp_sections) \ + psi_get_current(pp_sections[0]) +#define psi_table_get_lastsection(pp_sections) \ + psi_get_lastsection(pp_sections[0]) +#define psi_table_get_tableidext(pp_sections) \ + psi_get_tableidext(pp_sections[0]) + +static inline bool psi_table_section(uint8_t **pp_sections, uint8_t *p_section) +{ + uint8_t i_section = psi_get_section( p_section ); + uint8_t i_last_section = psi_get_lastsection( p_section ); + uint8_t i_version = psi_get_version( p_section ); + uint16_t i_tableidext = psi_get_tableidext( p_section ); + int i; + + free(pp_sections[i_section]); + pp_sections[i_section] = p_section; + + for (i = 0; i <= i_last_section; i++) { + uint8_t *p = pp_sections[i]; + if (p == NULL) + return false; + if (psi_get_lastsection(p) != i_last_section + || psi_get_version(p) != i_version + || psi_get_tableidext(p) != i_tableidext) + return false; + } + + /* free spurious, invalid sections */ + for (; i < PSI_TABLE_MAX_SECTIONS; i++) { + free(pp_sections[i]); + pp_sections[i] = NULL; + } + + /* a new, full table is available */ + return true; +} + +static inline uint8_t *psi_table_get_section(uint8_t **pp_sections, uint8_t n) +{ + return pp_sections[n]; +} + +static inline bool psi_table_compare(uint8_t **pp_sections1, + uint8_t **pp_sections2) +{ + uint8_t i_last_section = psi_table_get_lastsection(pp_sections1); + uint8_t i; + + if (i_last_section != psi_table_get_lastsection(pp_sections2)) + return false; + + for (i = 0; i <= i_last_section; i++) { + const uint8_t *p_section1 = psi_table_get_section(pp_sections1, i); + const uint8_t *p_section2 = psi_table_get_section(pp_sections2, i); + if (!psi_compare(p_section1, p_section2)) + return false; + } + + return true; +} + +#ifdef __cplusplus +} +#endif + +#endif