libtsfuncs is a library for mpeg PSI parsing and generation.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

tsfuncs.h 13KB

  1. #ifndef LIBTS_TSFUNCS_H
  2. #define LIBTS_TSFUNCS_H
  3. #include <time.h>
  4. #include <netdb.h>
  5. #include "tsdata.h"
  6. #include "log.h"
  7. // Usage bit_on(0xff, 0x02)
  8. #define bit_on(__bit, __mask) ((__bit & __mask) ? 1 : 0)
  9. #define bit_1 (0x01)
  10. #define bit_2 (0x02)
  11. #define bit_3 (0x04)
  12. #define bit_4 (0x08)
  13. #define bit_5 (0x10)
  14. #define bit_6 (0x20)
  15. #define bit_7 (0x40)
  16. #define bit_8 (0x80)
  17. #define NO_PCR (-1ull)
  18. #define NO_PCR_BASE (-1ull)
  19. #define NO_PCR_EXT (0xffff)
  20. #define NO_PCR (-1ull)
  21. #define NO_PTS (-1ull)
  22. #define NO_DTS (-1ull)
  23. enum ts_scrambled_type {
  24. not_scrambled = 0x00,
  25. scrambled_reserved = 0x01,
  26. scrambled_with_odd_key = 0x02,
  27. scrambled_with_even_key = 0x03
  28. };
  29. // Packet manipulation
  30. void ts_packet_init_null (uint8_t *ts_packet);
  31. int ts_packet_is_pusi (uint8_t *ts_packet);
  32. uint16_t ts_packet_get_pid (uint8_t *ts_packet);
  33. void ts_packet_set_pid (uint8_t *ts_packet, uint16_t new_pid);
  34. uint8_t ts_packet_get_cont (uint8_t *ts_packet);
  35. void ts_packet_set_cont (uint8_t *ts_packet, uint8_t value);
  36. void ts_packet_inc_cont (uint8_t *ts_packet, uint8_t increment);
  37. uint8_t ts_packet_get_payload_offset(uint8_t *ts_packet);
  38. int ts_packet_is_scrambled(uint8_t *ts_packet);
  39. int ts_packet_get_scrambled(uint8_t *ts_packet);
  40. void ts_packet_set_not_scrambled(uint8_t *ts_packet);
  41. void ts_packet_set_scrambled(uint8_t *ts_packet, enum ts_scrambled_type stype);
  42. int ts_packet_has_pcr (uint8_t *ts_packet);
  43. uint64_t ts_packet_get_pcr_ex (uint8_t *ts_packet, uint64_t *pcr_base, uint16_t *pcr_ext);
  44. uint64_t ts_packet_get_pcr (uint8_t *ts_packet);
  45. void ts_packet_set_pcr_ex (uint8_t *ts_packet, uint64_t pcr_base, uint16_t pcr_ext);
  46. void ts_packet_set_pcr (uint8_t *ts_packet, uint64_t pcr);
  47. void ts_encode_pts_dts (uint8_t *data, int guard_bits, uint64_t value);
  48. int ts_decode_pts_dts (uint8_t *data, int required_guard, uint64_t *value);
  49. int ts_packet_has_pes (uint8_t *ts_packet, uint16_t *pes_packet_len);
  50. int ts_packet_has_pts_dts (uint8_t *ts_packet, uint64_t *pts, uint64_t *dts);
  51. void ts_packet_change_pts (uint8_t *ts_packet, uint64_t pts);
  52. void ts_packet_change_pts_dts (uint8_t *ts_packet, uint64_t pts, uint64_t dts);
  53. // TS packet headers
  54. uint8_t * ts_packet_header_parse (uint8_t *ts_packet, struct ts_header *ts_header);
  55. void ts_packet_header_generate (uint8_t *ts_packet, struct ts_header *ts_header);
  56. void ts_packet_header_dump (struct ts_header *ts_header);
  57. // Sections
  58. uint8_t * ts_section_header_parse (uint8_t *ts_packet, struct ts_header *ts_header, struct ts_section_header *ts_section_header);
  59. void ts_section_header_generate (uint8_t *ts_packet, struct ts_section_header *ts_section_header, uint8_t start);
  60. void ts_section_header_dump (struct ts_section_header *t);
  61. void ts_section_dump (struct ts_section_header *sec);
  62. void ts_section_header_set_private_vars (struct ts_section_header *ts_section_header);
  63. int ts_section_is_same(struct ts_section_header *s1, struct ts_section_header *s2);
  64. uint8_t * ts_section_data_alloc_section ();
  65. uint8_t * ts_section_data_alloc_packet ();
  66. struct ts_section_header * ts_section_data_alloc ();
  67. void ts_section_data_clear (struct ts_section_header *sec);
  68. void ts_section_data_free (struct ts_section_header **ts_section_header);
  69. void ts_section_data_copy (struct ts_section_header *src, struct ts_section_header *dst);
  70. void ts_section_add_packet (struct ts_section_header *sec, struct ts_header *ts_header, uint8_t *ts_packet);
  71. uint32_t ts_section_data_calculate_crc (uint8_t *section_data, int section_data_size);
  72. void ts_section_data_gen_ts_packets (struct ts_header *ts_header, uint8_t *section_data, int section_data_sz, uint8_t pointer_field, uint8_t **packets, int *num_packets);
  73. // PAT
  74. struct ts_pat * ts_pat_alloc ();
  75. struct ts_pat * ts_pat_alloc_init (uint16_t transport_stream_id);
  76. struct ts_pat * ts_pat_push_packet (struct ts_pat *pat, uint8_t *ts_packet);
  77. void ts_pat_clear (struct ts_pat *pat);
  78. void ts_pat_free (struct ts_pat **pat);
  79. int ts_pat_parse (struct ts_pat *pat);
  80. void ts_pat_dump (struct ts_pat *pat);
  81. void ts_pat_generate (struct ts_pat *pat, uint8_t **ts_packets, int *num_packets);
  82. struct ts_pat * ts_pat_copy (struct ts_pat *pat);
  83. void ts_pat_regenerate_packets (struct ts_pat *pat);
  84. int ts_pat_add_program (struct ts_pat *pat, uint16_t program, uint16_t pat_pid);
  85. int ts_pat_del_program (struct ts_pat *pat, uint16_t program);
  86. int ts_pat_is_same (struct ts_pat *pat1, struct ts_pat *pat2);
  87. // CAT
  88. struct ts_cat * ts_cat_alloc ();
  89. struct ts_cat * ts_cat_push_packet (struct ts_cat *cat, uint8_t *ts_packet);
  90. void ts_cat_clear (struct ts_cat *cat);
  91. void ts_cat_free (struct ts_cat **cat);
  92. int ts_cat_parse (struct ts_cat *cat);
  93. void ts_cat_dump (struct ts_cat *cat);
  94. struct ts_cat * ts_cat_copy (struct ts_cat *cat);
  95. int ts_cat_is_same (struct ts_cat *cat1, struct ts_cat *cat2);
  96. enum CA_system ts_get_CA_sys (uint16_t CA_id);
  97. char * ts_get_CA_sys_txt (enum CA_system CA_sys);
  98. int ts_get_emm_info (struct ts_cat *cat, enum CA_system CA_sys, uint16_t *CA_id, uint16_t *CA_pid);
  99. int ts_get_ecm_info (struct ts_pmt *pmt, enum CA_system CA_sys, uint16_t *CA_id, uint16_t *CA_pid);
  100. // PMT
  101. struct ts_pmt * ts_pmt_alloc ();
  102. struct ts_pmt * ts_pmt_alloc_init (uint16_t org_network_id, uint16_t transport_stream_id);
  103. struct ts_pmt * ts_pmt_push_packet (struct ts_pmt *pmt, uint8_t *ts_packet);
  104. void ts_pmt_clear (struct ts_pmt *pmt);
  105. void ts_pmt_free (struct ts_pmt **pmt);
  106. int ts_pmt_parse (struct ts_pmt *pmt);
  107. void ts_pmt_dump (struct ts_pmt *pmt);
  108. void ts_pmt_generate (struct ts_pmt *pmt, uint8_t **ts_packets, int *num_packets);
  109. struct ts_pmt * ts_pmt_copy (struct ts_pmt *pmt);
  110. void ts_pmt_regenerate_packets (struct ts_pmt *pmt);
  111. int ts_pmt_is_same (struct ts_pmt *pmt1, struct ts_pmt *pmt2);
  112. // NIT
  113. struct ts_nit * ts_nit_alloc ();
  114. struct ts_nit * ts_nit_alloc_init (uint16_t network_id);
  115. struct ts_nit * ts_nit_push_packet (struct ts_nit *nit, uint8_t *ts_packet);
  116. void ts_nit_clear (struct ts_nit *nit);
  117. void ts_nit_free (struct ts_nit **nit);
  118. int ts_nit_parse (struct ts_nit *nit);
  119. void ts_nit_dump (struct ts_nit *nit);
  120. void ts_nit_generate (struct ts_nit *nit, uint8_t **ts_packets, int *num_packets);
  121. int ts_nit_add_network_name_descriptor (struct ts_nit *nit, char *network_name);
  122. int ts_nit_add_frequency_list_descriptor_cable (struct ts_nit *nit, uint16_t ts_id, uint16_t org_net_id, uint32_t *freqs, uint8_t num_freqs);
  123. int ts_nit_add_cable_delivery_descriptor (struct ts_nit *nit, uint16_t ts_id, uint16_t org_net_id, uint32_t freq, uint8_t modulation, uint32_t symbol_rate);
  124. int ts_nit_add_service_list_descriptor (struct ts_nit *nit, uint16_t ts_id, uint16_t org_net_id, uint32_t *services, uint8_t num_services);
  125. // SDT
  126. struct ts_sdt * ts_sdt_alloc ();
  127. struct ts_sdt * ts_sdt_alloc_init (uint16_t org_network_id, uint16_t transport_stream_id);
  128. struct ts_sdt * ts_sdt_push_packet (struct ts_sdt *sdt, uint8_t *ts_packet);
  129. void ts_sdt_clear (struct ts_sdt *sdt);
  130. void ts_sdt_free (struct ts_sdt **sdt);
  131. int ts_sdt_parse (struct ts_sdt *sdt);
  132. void ts_sdt_dump (struct ts_sdt *sdt);
  133. void ts_sdt_generate (struct ts_sdt *sdt, uint8_t **ts_packets, int *num_packets);
  134. int ts_sdt_add_service_descriptor(struct ts_sdt *sdt, uint16_t service_id, uint8_t video, char *provider_name, char *service_name);
  135. // EIT
  136. struct ts_eit * ts_eit_alloc ();
  137. struct ts_eit * ts_eit_alloc_init (uint16_t service_id, uint16_t transport_stream_id, uint16_t org_network_id, uint8_t table_id, uint8_t sec_number, uint8_t last_sec_number);
  138. struct ts_eit * ts_eit_alloc_init_pf (uint16_t service_id, uint16_t transport_stream_id, uint16_t org_network_id, uint8_t sec_number, uint8_t last_sec_number); // Shortcut using table_id 0x4e
  139. struct ts_eit * ts_eit_alloc_init_schedule (uint16_t service_id, uint16_t transport_stream_id, uint16_t org_network_id, uint8_t sec_number, uint8_t last_sec_number); // Shortcut using table_id 0x50
  140. struct ts_eit * ts_eit_push_packet (struct ts_eit *eit, uint8_t *ts_packet);
  141. void ts_eit_clear (struct ts_eit *eit);
  142. void ts_eit_free (struct ts_eit **eit);
  143. int ts_eit_parse (struct ts_eit *eit);
  144. void ts_eit_dump (struct ts_eit *eit);
  145. void ts_eit_generate (struct ts_eit *eit, uint8_t **ts_packets, int *num_packets);
  146. struct ts_eit * ts_eit_copy (struct ts_eit *eit);
  147. void ts_eit_regenerate_packets (struct ts_eit *eit);
  148. int ts_eit_add_short_event_descriptor (struct ts_eit *eit, uint16_t event_id, uint8_t running, time_t start_time, int duration_sec, char *event_name, char *event_short_descr);
  149. int ts_eit_add_extended_event_descriptor(struct ts_eit *eit, uint16_t event_id, uint8_t running, time_t start_time, int duration_sec, char *text);
  150. // TDT
  151. struct ts_tdt * ts_tdt_alloc();
  152. struct ts_tdt * ts_tdt_alloc_init (time_t ts);
  153. struct ts_tdt * ts_tot_alloc_init (time_t ts);
  154. void ts_tdt_clear (struct ts_tdt *tdt);
  155. void ts_tdt_free (struct ts_tdt **tdt);
  156. int ts_tdt_parse (struct ts_tdt *tdt);
  157. struct ts_tdt * ts_tdt_push_packet (struct ts_tdt *tdt, uint8_t *ts_packet);
  158. void ts_tdt_generate (struct ts_tdt *tdt, uint8_t **ts_packets, int *num_packets);
  159. void ts_tdt_dump (struct ts_tdt *tdt);
  160. void ts_tdt_set_time (struct ts_tdt *tdt, time_t ts);
  161. void ts_tot_set_localtime_offset (struct ts_tdt *tdt, time_t now, time_t change_time, uint8_t polarity, uint16_t ofs, uint16_t ofs_next);
  162. void ts_tot_set_localtime_offset_sofia (struct ts_tdt *tdt, time_t now);
  163. // Private section
  164. struct ts_privsec * ts_privsec_alloc();
  165. void ts_privsec_clear (struct ts_privsec *pprivsec);
  166. void ts_privsec_free (struct ts_privsec **pprivsec);
  167. struct ts_privsec * ts_privsec_push_packet (struct ts_privsec *privsec, uint8_t *ts_packet);
  168. int ts_privsec_is_same (struct ts_privsec *p1, struct ts_privsec *p2);
  169. void ts_privsec_dump (struct ts_privsec *privsec);
  170. void ts_privsec_copy (struct ts_privsec *src, struct ts_privsec *dst);
  171. // Time
  172. uint32_t ts_time_encode_bcd (int duration_sec);
  173. void ts_time_decode_bcd (int duration_bcd, int *duration_sec, int *hour, int *min, int *sec);
  174. void ts_time_encode_mjd (uint16_t *mjd, uint32_t *bcd, time_t *ts, struct tm *tm);
  175. time_t ts_time_decode_mjd (uint16_t mjd, uint32_t bcd, struct tm *tm);
  176. // Descriptors
  177. void ts_descriptor_dump (uint8_t *desc_data, int desc_data_len);
  178. int ts_is_stream_type_video (uint8_t stream_type);
  179. int ts_is_stream_type_ac3 (uint8_t stream_type);
  180. int ts_is_stream_type_audio (uint8_t stream_type);
  181. char * h222_stream_type_desc (uint8_t stream_type);
  182. char * h222_stream_id_desc (uint8_t stream_id);
  183. // PES
  184. struct ts_pes * ts_pes_alloc ();
  185. void ts_pes_clear (struct ts_pes *pes);
  186. void ts_pes_free (struct ts_pes **pes);
  187. void ts_pes_fill_type (struct ts_pes *pes, struct ts_pmt *pmt, uint16_t pid);
  188. int ts_pes_is_finished (struct ts_pes *pes, uint8_t *ts_packet);
  189. struct ts_pes * ts_pes_push_packet (struct ts_pes *pes, uint8_t *ts_packet, struct ts_pmt *pmt, uint16_t pid);
  190. int ts_pes_parse (struct ts_pes *pes);
  191. void ts_pes_dump (struct ts_pes *pes);
  192. struct pes_array * pes_array_alloc ();
  193. void pes_array_dump (struct pes_array *pa);
  194. void pes_array_free (struct pes_array **ppa);
  195. struct pes_entry * pes_array_push_packet (struct pes_array *pa, uint16_t pid, struct ts_pat *pat, struct ts_pmt *pmt, uint8_t *ts_packet);
  196. // ES functions
  197. int ts_pes_es_mpeg_audio_header_parse (struct mpeg_audio_header *mpghdr, uint8_t *data, int datasz);
  198. void ts_pes_es_mpeg_audio_header_dump (struct mpeg_audio_header *mpghdr);
  199. void ts_pes_es_parse (struct ts_pes *pes);
  200. void ts_pes_es_dump (struct ts_pes *pes);
  201. // CRC
  202. uint32_t ts_crc32 (uint8_t *data, int data_size);
  203. uint32_t ts_crc32_section (struct ts_section_header *section_header);
  204. int ts_crc32_section_check (struct ts_section_header *section_header, char *table);
  205. // Misc
  206. int dec2bcd (int dec);
  207. int bcd2dec (int bcd);
  208. void ts_compare_data (char *prefix, uint8_t *a, uint8_t *b, int size);
  209. void ts_hex_dump_buf (char *buf, int bufsz, uint8_t *d, int size, int col);
  210. char * ts_hex_dump (uint8_t *d, int size, int col);
  211. void ts_print_bytes (char *prefix, uint8_t *d, int size);
  212. char * init_dvb_string_utf8 (char *text);
  213. char * init_dvb_string_iso_8859_5 (char *text);
  214. int ts_is_psi_pid (uint16_t pid, struct ts_pat *pat);
  215. // Shortcuts
  216. int parse_tdt (uint8_t *ts_packet, int dump);
  217. #endif