1 /*
2 * PAT descriptor generator
3 * Copyright (C) 2010-2011 Unix Solutions Ltd.
4 *
5 * Released under MIT license.
6 * See LICENSE-MIT.txt for license terms.
7 */
8 #include <stdio.h>
9 #include <unistd.h>
10 #include <netdb.h>
11 #include <stdlib.h>
12 #include <string.h>
13
14 #include "tsfuncs.h"
15
ts_pat_regenerate_packet_data(struct ts_pat * pat)16 static void ts_pat_regenerate_packet_data(struct ts_pat *pat) {
17 uint8_t *ts_packets;
18 int num_packets;
19 ts_pat_generate(pat, &ts_packets, &num_packets);
20 memcpy(pat->section_header->packet_data, ts_packets, num_packets * TS_PACKET_SIZE);
21 pat->section_header->num_packets = num_packets;
22 free(ts_packets);
23 }
24
ts_pat_init(struct ts_pat * pat,uint16_t transport_stream_id)25 struct ts_pat *ts_pat_init(struct ts_pat *pat, uint16_t transport_stream_id) {
26 pat->ts_header.pid = 0x00;
27 pat->ts_header.pusi = 1;
28 pat->ts_header.payload_field = 1;
29 pat->ts_header.payload_offset = 4;
30
31 pat->section_header->table_id = 0x00;
32 pat->section_header->version_number = 1;
33 pat->section_header->current_next_indicator = 1;
34 pat->section_header->section_syntax_indicator = 1;
35 pat->section_header->private_indicator = 0;
36 pat->section_header->section_length = 9; // Empty section (9)
37 pat->section_header->reserved1 = 3;
38 pat->section_header->reserved2 = 3;
39
40 pat->section_header->ts_id_number = transport_stream_id;
41
42 pat->programs_num = 0;
43
44 pat->initialized = 1;
45
46 ts_pat_regenerate_packet_data(pat);
47
48 return pat;
49 }
50
ts_pat_alloc_init(uint16_t transport_stream_id)51 struct ts_pat *ts_pat_alloc_init(uint16_t transport_stream_id) {
52 struct ts_pat *pat = ts_pat_alloc();
53 if (!pat)
54 return NULL;
55 return ts_pat_init(pat, transport_stream_id);
56 }
57
ts_pat_add_program(struct ts_pat * pat,uint16_t program,uint16_t pat_pid)58 int ts_pat_add_program(struct ts_pat *pat, uint16_t program, uint16_t pat_pid) {
59 int i;
60 if (pat->programs_max == pat->programs_num)
61 return 0;
62
63 for (i=0;i<pat->programs_num;i++) {
64 struct ts_pat_program *prg = pat->programs[i];
65 if (program == prg->program) {
66 ts_LOGf("!!! Program 0x%04x (%d) already exists in PAT!\n", program, program);
67 return 0;
68 }
69 }
70
71 pat->section_header->version_number++;
72 pat->section_header->section_length += 4;
73
74 struct ts_pat_program *pinfo = calloc(1, sizeof(struct ts_pat_program));
75 pinfo->program = program;
76 pinfo->reserved = 7; // All three bits are up
77 pinfo->pid = pat_pid;
78
79 pat->programs[pat->programs_num] = pinfo;
80 pat->programs_num++;
81
82 ts_pat_regenerate_packet_data(pat);
83
84 return 1;
85 }
86
ts_pat_del_program(struct ts_pat * pat,uint16_t program)87 int ts_pat_del_program(struct ts_pat *pat, uint16_t program) {
88 int i, ok=1, del_pos=0;
89
90 if (!pat->programs_num)
91 return 0;
92
93 for (i=0;i<pat->programs_num;i++) {
94 struct ts_pat_program *prg = pat->programs[i];
95 if (program == prg->program) {
96 ts_LOGf("!!! Found program 0x%04x (%d) for deleting.\n", program, program);
97 del_pos = i;
98 ok = 1;
99 break;
100 }
101 }
102
103 if (!ok)
104 return 0;
105
106 for (i=0;i<pat->programs_num;i++) {
107 if (i < del_pos)
108 continue;
109 if (i == del_pos) {
110 FREE(pat->programs[i]);
111 }
112 if (i >= del_pos && i+1 < pat->programs_num) {
113 struct ts_pat_program *next = pat->programs[i+1];
114 pat->programs[i] = next;
115 }
116 }
117
118 pat->section_header->version_number++;
119 pat->section_header->section_length -= 4;
120 pat->programs_num--;
121
122 ts_pat_regenerate_packet_data(pat);
123
124 return 1;
125 }
126