xref: /linux/drivers/media/test-drivers/vidtv/vidtv_psi.c (revision 0ea5c948cb64bab5bc7a5516774eb8536f05aa0d)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * This file contains the logic to work with MPEG Program-Specific Information.
4  * These are defined both in ISO/IEC 13818-1 (systems) and ETSI EN 300 468.
5  * PSI is carried in the form of table structures, and although each table might
6  * technically be broken into one or more sections, we do not do this here,
7  * hence 'table' and 'section' are interchangeable for vidtv.
8  *
9  * Copyright (C) 2020 Daniel W. S. Almeida
10  */
11 
12 #define pr_fmt(fmt) KBUILD_MODNAME ":%s, %d: " fmt, __func__, __LINE__
13 
14 #include <linux/bcd.h>
15 #include <linux/crc32.h>
16 #include <linux/kernel.h>
17 #include <linux/ktime.h>
18 #include <linux/printk.h>
19 #include <linux/ratelimit.h>
20 #include <linux/slab.h>
21 #include <linux/string.h>
22 #include <linux/time.h>
23 #include <linux/types.h>
24 
25 #include "vidtv_common.h"
26 #include "vidtv_psi.h"
27 #include "vidtv_ts.h"
28 
29 #define CRC_SIZE_IN_BYTES 4
30 #define MAX_VERSION_NUM 32
31 #define INITIAL_CRC 0xffffffff
32 #define ISO_LANGUAGE_CODE_LEN 3
33 
34 static const u32 CRC_LUT[256] = {
35 	/* from libdvbv5 */
36 	0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b,
37 	0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61,
38 	0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, 0x4c11db70, 0x48d0c6c7,
39 	0x4593e01e, 0x4152fda9, 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75,
40 	0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, 0x791d4014, 0x7ddc5da3,
41 	0x709f7b7a, 0x745e66cd, 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039,
42 	0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, 0xbe2b5b58, 0xbaea46ef,
43 	0xb7a96036, 0xb3687d81, 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d,
44 	0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, 0xc7361b4c, 0xc3f706fb,
45 	0xceb42022, 0xca753d95, 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1,
46 	0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, 0x34867077, 0x30476dc0,
47 	0x3d044b19, 0x39c556ae, 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072,
48 	0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, 0x018aeb13, 0x054bf6a4,
49 	0x0808d07d, 0x0cc9cdca, 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde,
50 	0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, 0x5e9f46bf, 0x5a5e5b08,
51 	0x571d7dd1, 0x53dc6066, 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba,
52 	0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, 0xbfa1b04b, 0xbb60adfc,
53 	0xb6238b25, 0xb2e29692, 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6,
54 	0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, 0xe0b41de7, 0xe4750050,
55 	0xe9362689, 0xedf73b3e, 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2,
56 	0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, 0xd5b88683, 0xd1799b34,
57 	0xdc3abded, 0xd8fba05a, 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637,
58 	0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, 0x4f040d56, 0x4bc510e1,
59 	0x46863638, 0x42472b8f, 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53,
60 	0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, 0x36194d42, 0x32d850f5,
61 	0x3f9b762c, 0x3b5a6b9b, 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff,
62 	0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, 0xf12f560e, 0xf5ee4bb9,
63 	0xf8ad6d60, 0xfc6c70d7, 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b,
64 	0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, 0xc423cd6a, 0xc0e2d0dd,
65 	0xcda1f604, 0xc960ebb3, 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7,
66 	0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, 0x9b3660c6, 0x9ff77d71,
67 	0x92b45ba8, 0x9675461f, 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3,
68 	0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, 0x4e8ee645, 0x4a4ffbf2,
69 	0x470cdd2b, 0x43cdc09c, 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8,
70 	0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, 0x119b4be9, 0x155a565e,
71 	0x18197087, 0x1cd86d30, 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec,
72 	0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, 0x2497d08d, 0x2056cd3a,
73 	0x2d15ebe3, 0x29d4f654, 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0,
74 	0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, 0xe3a1cbc1, 0xe760d676,
75 	0xea23f0af, 0xeee2ed18, 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4,
76 	0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, 0x9abc8bd5, 0x9e7d9662,
77 	0x933eb0bb, 0x97ffad0c, 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668,
78 	0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4
79 };
80 
dvb_crc32(u32 crc,u8 * data,u32 len)81 static u32 dvb_crc32(u32 crc, u8 *data, u32 len)
82 {
83 	/* from libdvbv5 */
84 	while (len--)
85 		crc = (crc << 8) ^ CRC_LUT[((crc >> 24) ^ *data++) & 0xff];
86 	return crc;
87 }
88 
vidtv_psi_update_version_num(struct vidtv_psi_table_header * h)89 static void vidtv_psi_update_version_num(struct vidtv_psi_table_header *h)
90 {
91 	h->version++;
92 }
93 
vidtv_psi_get_sec_len(struct vidtv_psi_table_header * h)94 static u16 vidtv_psi_get_sec_len(struct vidtv_psi_table_header *h)
95 {
96 	u16 mask;
97 
98 	mask = GENMASK(11, 0);
99 
100 	return be16_to_cpu(h->bitfield) & mask;
101 }
102 
vidtv_psi_get_pat_program_pid(struct vidtv_psi_table_pat_program * p)103 u16 vidtv_psi_get_pat_program_pid(struct vidtv_psi_table_pat_program *p)
104 {
105 	u16 mask;
106 
107 	mask = GENMASK(12, 0);
108 
109 	return be16_to_cpu(p->bitfield) & mask;
110 }
111 
vidtv_psi_pmt_stream_get_elem_pid(struct vidtv_psi_table_pmt_stream * s)112 u16 vidtv_psi_pmt_stream_get_elem_pid(struct vidtv_psi_table_pmt_stream *s)
113 {
114 	u16 mask;
115 
116 	mask = GENMASK(12, 0);
117 
118 	return be16_to_cpu(s->bitfield) & mask;
119 }
120 
vidtv_psi_set_desc_loop_len(__be16 * bitfield,u16 new_len,u8 desc_len_nbits)121 static void vidtv_psi_set_desc_loop_len(__be16 *bitfield, u16 new_len,
122 					u8 desc_len_nbits)
123 {
124 	__be16 new;
125 	u16 mask;
126 
127 	mask = GENMASK(15, desc_len_nbits);
128 
129 	new = cpu_to_be16((be16_to_cpu(*bitfield) & mask) | new_len);
130 	*bitfield = new;
131 }
132 
vidtv_psi_set_sec_len(struct vidtv_psi_table_header * h,u16 new_len)133 static void vidtv_psi_set_sec_len(struct vidtv_psi_table_header *h, u16 new_len)
134 {
135 	u16 old_len = vidtv_psi_get_sec_len(h);
136 	__be16 new;
137 	u16 mask;
138 
139 	mask = GENMASK(15, 13);
140 
141 	new = cpu_to_be16((be16_to_cpu(h->bitfield) & mask) | new_len);
142 
143 	if (old_len > MAX_SECTION_LEN)
144 		pr_warn_ratelimited("section length: %d > %d, old len was %d\n",
145 				    new_len,
146 				    MAX_SECTION_LEN,
147 				    old_len);
148 
149 	h->bitfield = new;
150 }
151 
152 /*
153  * Packetize PSI sections into TS packets:
154  * push a TS header (4bytes) every 184 bytes
155  * manage the continuity_counter
156  * add stuffing (i.e. padding bytes) after the CRC
157  */
vidtv_psi_ts_psi_write_into(struct psi_write_args * args)158 static u32 vidtv_psi_ts_psi_write_into(struct psi_write_args *args)
159 {
160 	struct vidtv_mpeg_ts ts_header = {
161 		.sync_byte = TS_SYNC_BYTE,
162 		.bitfield = cpu_to_be16((args->new_psi_section << 14) | args->pid),
163 		.scrambling = 0,
164 		.payload = 1,
165 		.adaptation_field = 0, /* no adaptation field */
166 	};
167 	u32 nbytes_past_boundary = (args->dest_offset % TS_PACKET_LEN);
168 	bool aligned = (nbytes_past_boundary == 0);
169 	u32 remaining_len = args->len;
170 	u32 payload_write_len = 0;
171 	u32 payload_offset = 0;
172 	u32 nbytes = 0;
173 
174 	if (!args->crc && !args->is_crc)
175 		pr_warn_ratelimited("Missing CRC for chunk\n");
176 
177 	if (args->crc)
178 		*args->crc = dvb_crc32(*args->crc, args->from, args->len);
179 
180 	if (args->new_psi_section && !aligned) {
181 		pr_warn_ratelimited("Cannot write a new PSI section in a misaligned buffer\n");
182 
183 		/* forcibly align and hope for the best */
184 		nbytes += vidtv_memset(args->dest_buf,
185 				       args->dest_offset + nbytes,
186 				       args->dest_buf_sz,
187 				       TS_FILL_BYTE,
188 				       TS_PACKET_LEN - nbytes_past_boundary);
189 	}
190 
191 	while (remaining_len) {
192 		nbytes_past_boundary = (args->dest_offset + nbytes) % TS_PACKET_LEN;
193 		aligned = (nbytes_past_boundary == 0);
194 
195 		if (aligned) {
196 			/* if at a packet boundary, write a new TS header */
197 			ts_header.continuity_counter = *args->continuity_counter;
198 
199 			nbytes += vidtv_memcpy(args->dest_buf,
200 					       args->dest_offset + nbytes,
201 					       args->dest_buf_sz,
202 					       &ts_header,
203 					       sizeof(ts_header));
204 			/*
205 			 * This will trigger a discontinuity if the buffer is full,
206 			 * effectively dropping the packet.
207 			 */
208 			vidtv_ts_inc_cc(args->continuity_counter);
209 		}
210 
211 		/* write the pointer_field in the first byte of the payload */
212 		if (args->new_psi_section)
213 			nbytes += vidtv_memset(args->dest_buf,
214 					       args->dest_offset + nbytes,
215 					       args->dest_buf_sz,
216 					       0x0,
217 					       1);
218 
219 		/* write as much of the payload as possible */
220 		nbytes_past_boundary = (args->dest_offset + nbytes) % TS_PACKET_LEN;
221 		payload_write_len = min(TS_PACKET_LEN - nbytes_past_boundary, remaining_len);
222 
223 		nbytes += vidtv_memcpy(args->dest_buf,
224 				       args->dest_offset + nbytes,
225 				       args->dest_buf_sz,
226 				       args->from + payload_offset,
227 				       payload_write_len);
228 
229 		/* 'payload_write_len' written from a total of 'len' requested*/
230 		remaining_len -= payload_write_len;
231 		payload_offset += payload_write_len;
232 	}
233 
234 	/*
235 	 * fill the rest of the packet if there is any remaining space unused
236 	 */
237 
238 	nbytes_past_boundary = (args->dest_offset + nbytes) % TS_PACKET_LEN;
239 
240 	if (args->is_crc)
241 		nbytes += vidtv_memset(args->dest_buf,
242 				       args->dest_offset + nbytes,
243 				       args->dest_buf_sz,
244 				       TS_FILL_BYTE,
245 				       TS_PACKET_LEN - nbytes_past_boundary);
246 
247 	return nbytes;
248 }
249 
table_section_crc32_write_into(struct crc32_write_args * args)250 static u32 table_section_crc32_write_into(struct crc32_write_args *args)
251 {
252 	struct psi_write_args psi_args = {
253 		.dest_buf           = args->dest_buf,
254 		.from               = &args->crc,
255 		.len                = CRC_SIZE_IN_BYTES,
256 		.dest_offset        = args->dest_offset,
257 		.pid                = args->pid,
258 		.new_psi_section    = false,
259 		.continuity_counter = args->continuity_counter,
260 		.is_crc             = true,
261 		.dest_buf_sz        = args->dest_buf_sz,
262 	};
263 
264 	/* the CRC is the last entry in the section */
265 
266 	return vidtv_psi_ts_psi_write_into(&psi_args);
267 }
268 
vidtv_psi_desc_chain(struct vidtv_psi_desc * head,struct vidtv_psi_desc * desc)269 static void vidtv_psi_desc_chain(struct vidtv_psi_desc *head, struct vidtv_psi_desc *desc)
270 {
271 	if (head) {
272 		while (head->next)
273 			head = head->next;
274 
275 		head->next = desc;
276 	}
277 }
278 
vidtv_psi_service_desc_init(struct vidtv_psi_desc * head,enum service_type service_type,char * service_name,char * provider_name)279 struct vidtv_psi_desc_service *vidtv_psi_service_desc_init(struct vidtv_psi_desc *head,
280 							   enum service_type service_type,
281 							   char *service_name,
282 							   char *provider_name)
283 {
284 	struct vidtv_psi_desc_service *desc;
285 	u32 service_name_len = service_name ? strlen(service_name) : 0;
286 	u32 provider_name_len = provider_name ? strlen(provider_name) : 0;
287 
288 	desc = kzalloc(sizeof(*desc), GFP_KERNEL);
289 	if (!desc)
290 		return NULL;
291 
292 	desc->type = SERVICE_DESCRIPTOR;
293 
294 	desc->length = sizeof_field(struct vidtv_psi_desc_service, service_type)
295 		       + sizeof_field(struct vidtv_psi_desc_service, provider_name_len)
296 		       + provider_name_len
297 		       + sizeof_field(struct vidtv_psi_desc_service, service_name_len)
298 		       + service_name_len;
299 
300 	desc->service_type = service_type;
301 
302 	desc->service_name_len = service_name_len;
303 
304 	if (service_name && service_name_len) {
305 		desc->service_name = kstrdup(service_name, GFP_KERNEL);
306 		if (!desc->service_name)
307 			goto free_desc;
308 	}
309 
310 	desc->provider_name_len = provider_name_len;
311 
312 	if (provider_name && provider_name_len) {
313 		desc->provider_name = kstrdup(provider_name, GFP_KERNEL);
314 		if (!desc->provider_name)
315 			goto free_desc_service_name;
316 	}
317 
318 	vidtv_psi_desc_chain(head, (struct vidtv_psi_desc *)desc);
319 	return desc;
320 
321 free_desc_service_name:
322 	if (service_name && service_name_len)
323 		kfree(desc->service_name);
324 free_desc:
325 	kfree(desc);
326 	return NULL;
327 }
328 
329 struct vidtv_psi_desc_registration
vidtv_psi_registration_desc_init(struct vidtv_psi_desc * head,__be32 format_id,u8 * additional_ident_info,u32 additional_info_len)330 *vidtv_psi_registration_desc_init(struct vidtv_psi_desc *head,
331 				  __be32 format_id,
332 				  u8 *additional_ident_info,
333 				  u32 additional_info_len)
334 {
335 	struct vidtv_psi_desc_registration *desc;
336 
337 	desc = kzalloc(sizeof(*desc) + sizeof(format_id) + additional_info_len, GFP_KERNEL);
338 	if (!desc)
339 		return NULL;
340 
341 	desc->type = REGISTRATION_DESCRIPTOR;
342 
343 	desc->length = sizeof_field(struct vidtv_psi_desc_registration, format_id)
344 		       + additional_info_len;
345 
346 	desc->format_id = format_id;
347 
348 	if (additional_ident_info && additional_info_len)
349 		memcpy(desc->additional_identification_info,
350 		       additional_ident_info,
351 		       additional_info_len);
352 
353 	vidtv_psi_desc_chain(head, (struct vidtv_psi_desc *)desc);
354 	return desc;
355 }
356 
357 struct vidtv_psi_desc_network_name
vidtv_psi_network_name_desc_init(struct vidtv_psi_desc * head,char * network_name)358 *vidtv_psi_network_name_desc_init(struct vidtv_psi_desc *head, char *network_name)
359 {
360 	u32 network_name_len = network_name ? strlen(network_name) : 0;
361 	struct vidtv_psi_desc_network_name *desc;
362 
363 	desc = kzalloc(sizeof(*desc), GFP_KERNEL);
364 	if (!desc)
365 		return NULL;
366 
367 	desc->type = NETWORK_NAME_DESCRIPTOR;
368 
369 	desc->length = network_name_len;
370 
371 	if (network_name && network_name_len) {
372 		desc->network_name = kstrdup(network_name, GFP_KERNEL);
373 		if (!desc->network_name) {
374 			kfree(desc);
375 			return NULL;
376 		}
377 	}
378 
379 	vidtv_psi_desc_chain(head, (struct vidtv_psi_desc *)desc);
380 	return desc;
381 }
382 
383 struct vidtv_psi_desc_service_list
vidtv_psi_service_list_desc_init(struct vidtv_psi_desc * head,struct vidtv_psi_desc_service_list_entry * entry)384 *vidtv_psi_service_list_desc_init(struct vidtv_psi_desc *head,
385 				  struct vidtv_psi_desc_service_list_entry *entry)
386 {
387 	struct vidtv_psi_desc_service_list_entry *curr_e = NULL;
388 	struct vidtv_psi_desc_service_list_entry *head_e = NULL;
389 	struct vidtv_psi_desc_service_list_entry *prev_e = NULL;
390 	struct vidtv_psi_desc_service_list *desc;
391 	u16 length = 0;
392 
393 	desc = kzalloc(sizeof(*desc), GFP_KERNEL);
394 	if (!desc)
395 		return NULL;
396 
397 	desc->type = SERVICE_LIST_DESCRIPTOR;
398 
399 	while (entry) {
400 		curr_e = kzalloc(sizeof(*curr_e), GFP_KERNEL);
401 		if (!curr_e) {
402 			while (head_e) {
403 				curr_e = head_e;
404 				head_e = head_e->next;
405 				kfree(curr_e);
406 			}
407 			kfree(desc);
408 			return NULL;
409 		}
410 
411 		curr_e->service_id = entry->service_id;
412 		curr_e->service_type = entry->service_type;
413 
414 		length += sizeof(struct vidtv_psi_desc_service_list_entry) -
415 			  sizeof(struct vidtv_psi_desc_service_list_entry *);
416 
417 		if (!head_e)
418 			head_e = curr_e;
419 		if (prev_e)
420 			prev_e->next = curr_e;
421 
422 		prev_e = curr_e;
423 		entry = entry->next;
424 	}
425 
426 	desc->length = length;
427 	desc->service_list = head_e;
428 
429 	vidtv_psi_desc_chain(head, (struct vidtv_psi_desc *)desc);
430 	return desc;
431 }
432 
433 struct vidtv_psi_desc_short_event
vidtv_psi_short_event_desc_init(struct vidtv_psi_desc * head,char * iso_language_code,char * event_name,char * text)434 *vidtv_psi_short_event_desc_init(struct vidtv_psi_desc *head,
435 				 char *iso_language_code,
436 				 char *event_name,
437 				 char *text)
438 {
439 	u32 iso_len =  iso_language_code ? strlen(iso_language_code) : 0;
440 	u32 event_name_len = event_name ? strlen(event_name) : 0;
441 	struct vidtv_psi_desc_short_event *desc;
442 	u32 text_len =  text ? strlen(text) : 0;
443 
444 	desc = kzalloc(sizeof(*desc), GFP_KERNEL);
445 	if (!desc)
446 		return NULL;
447 
448 	desc->type = SHORT_EVENT_DESCRIPTOR;
449 
450 	desc->length = ISO_LANGUAGE_CODE_LEN +
451 		       sizeof_field(struct vidtv_psi_desc_short_event, event_name_len) +
452 		       event_name_len +
453 		       sizeof_field(struct vidtv_psi_desc_short_event, text_len) +
454 		       text_len;
455 
456 	desc->event_name_len = event_name_len;
457 	desc->text_len = text_len;
458 
459 	if (iso_len != ISO_LANGUAGE_CODE_LEN)
460 		iso_language_code = "eng";
461 
462 	desc->iso_language_code = kstrdup(iso_language_code, GFP_KERNEL);
463 	if (!desc->iso_language_code)
464 		goto free_desc;
465 
466 	if (event_name && event_name_len) {
467 		desc->event_name = kstrdup(event_name, GFP_KERNEL);
468 		if (!desc->event_name)
469 			goto free_desc_language_code;
470 	}
471 
472 	if (text && text_len) {
473 		desc->text = kstrdup(text, GFP_KERNEL);
474 		if (!desc->text)
475 			goto free_desc_event_name;
476 	}
477 
478 	vidtv_psi_desc_chain(head, (struct vidtv_psi_desc *)desc);
479 	return desc;
480 
481 free_desc_event_name:
482 	if (event_name && event_name_len)
483 		kfree(desc->event_name);
484 free_desc_language_code:
485 	kfree(desc->iso_language_code);
486 free_desc:
487 	kfree(desc);
488 	return NULL;
489 }
490 
vidtv_psi_desc_clone(struct vidtv_psi_desc * desc)491 struct vidtv_psi_desc *vidtv_psi_desc_clone(struct vidtv_psi_desc *desc)
492 {
493 	struct vidtv_psi_desc_network_name *desc_network_name;
494 	struct vidtv_psi_desc_service_list *desc_service_list;
495 	struct vidtv_psi_desc_short_event  *desc_short_event;
496 	struct vidtv_psi_desc_service *service;
497 	struct vidtv_psi_desc *head = NULL;
498 	struct vidtv_psi_desc *prev = NULL;
499 	struct vidtv_psi_desc *curr = NULL;
500 
501 	while (desc) {
502 		switch (desc->type) {
503 		case SERVICE_DESCRIPTOR:
504 			service = (struct vidtv_psi_desc_service *)desc;
505 			curr = (struct vidtv_psi_desc *)
506 			       vidtv_psi_service_desc_init(head,
507 							   service->service_type,
508 							   service->service_name,
509 							   service->provider_name);
510 		break;
511 
512 		case NETWORK_NAME_DESCRIPTOR:
513 			desc_network_name = (struct vidtv_psi_desc_network_name *)desc;
514 			curr = (struct vidtv_psi_desc *)
515 			       vidtv_psi_network_name_desc_init(head,
516 								desc_network_name->network_name);
517 		break;
518 
519 		case SERVICE_LIST_DESCRIPTOR:
520 			desc_service_list = (struct vidtv_psi_desc_service_list *)desc;
521 			curr = (struct vidtv_psi_desc *)
522 			       vidtv_psi_service_list_desc_init(head,
523 								desc_service_list->service_list);
524 		break;
525 
526 		case SHORT_EVENT_DESCRIPTOR:
527 			desc_short_event = (struct vidtv_psi_desc_short_event *)desc;
528 			curr = (struct vidtv_psi_desc *)
529 			       vidtv_psi_short_event_desc_init(head,
530 							       desc_short_event->iso_language_code,
531 							       desc_short_event->event_name,
532 							       desc_short_event->text);
533 		break;
534 
535 		case REGISTRATION_DESCRIPTOR:
536 		default:
537 			curr = kmemdup(desc, sizeof(*desc) + desc->length, GFP_KERNEL);
538 			if (!curr)
539 				return NULL;
540 		}
541 
542 		if (!curr)
543 			return NULL;
544 
545 		curr->next = NULL;
546 		if (!head)
547 			head = curr;
548 		if (prev)
549 			prev->next = curr;
550 
551 		prev = curr;
552 		desc = desc->next;
553 	}
554 
555 	return head;
556 }
557 
vidtv_psi_desc_destroy(struct vidtv_psi_desc * desc)558 void vidtv_psi_desc_destroy(struct vidtv_psi_desc *desc)
559 {
560 	struct vidtv_psi_desc_service_list_entry *sl_entry_tmp = NULL;
561 	struct vidtv_psi_desc_service_list_entry *sl_entry = NULL;
562 	struct vidtv_psi_desc *curr = desc;
563 	struct vidtv_psi_desc *tmp  = NULL;
564 
565 	while (curr) {
566 		tmp  = curr;
567 		curr = curr->next;
568 
569 		switch (tmp->type) {
570 		case SERVICE_DESCRIPTOR:
571 			kfree(((struct vidtv_psi_desc_service *)tmp)->provider_name);
572 			kfree(((struct vidtv_psi_desc_service *)tmp)->service_name);
573 
574 			break;
575 		case REGISTRATION_DESCRIPTOR:
576 			/* nothing to do */
577 			break;
578 
579 		case NETWORK_NAME_DESCRIPTOR:
580 			kfree(((struct vidtv_psi_desc_network_name *)tmp)->network_name);
581 			break;
582 
583 		case SERVICE_LIST_DESCRIPTOR:
584 			sl_entry = ((struct vidtv_psi_desc_service_list *)tmp)->service_list;
585 			while (sl_entry) {
586 				sl_entry_tmp = sl_entry;
587 				sl_entry = sl_entry->next;
588 				kfree(sl_entry_tmp);
589 			}
590 			break;
591 
592 		case SHORT_EVENT_DESCRIPTOR:
593 			kfree(((struct vidtv_psi_desc_short_event *)tmp)->iso_language_code);
594 			kfree(((struct vidtv_psi_desc_short_event *)tmp)->event_name);
595 			kfree(((struct vidtv_psi_desc_short_event *)tmp)->text);
596 		break;
597 
598 		default:
599 			pr_warn_ratelimited("Possible leak: not handling descriptor type %d\n",
600 					    tmp->type);
601 			break;
602 		}
603 
604 		kfree(tmp);
605 	}
606 }
607 
608 static u16
vidtv_psi_desc_comp_loop_len(struct vidtv_psi_desc * desc)609 vidtv_psi_desc_comp_loop_len(struct vidtv_psi_desc *desc)
610 {
611 	u32 length = 0;
612 
613 	if (!desc)
614 		return 0;
615 
616 	while (desc) {
617 		length += sizeof_field(struct vidtv_psi_desc, type);
618 		length += sizeof_field(struct vidtv_psi_desc, length);
619 		length += desc->length; /* from 'length' field until the end of the descriptor */
620 		desc    = desc->next;
621 	}
622 
623 	return length;
624 }
625 
vidtv_psi_desc_assign(struct vidtv_psi_desc ** to,struct vidtv_psi_desc * desc)626 void vidtv_psi_desc_assign(struct vidtv_psi_desc **to,
627 			   struct vidtv_psi_desc *desc)
628 {
629 	if (desc == *to)
630 		return;
631 
632 	if (*to)
633 		vidtv_psi_desc_destroy(*to);
634 
635 	*to = desc;
636 }
637 
vidtv_pmt_desc_assign(struct vidtv_psi_table_pmt * pmt,struct vidtv_psi_desc ** to,struct vidtv_psi_desc * desc)638 void vidtv_pmt_desc_assign(struct vidtv_psi_table_pmt *pmt,
639 			   struct vidtv_psi_desc **to,
640 			   struct vidtv_psi_desc *desc)
641 {
642 	vidtv_psi_desc_assign(to, desc);
643 	vidtv_psi_pmt_table_update_sec_len(pmt);
644 
645 	if (vidtv_psi_get_sec_len(&pmt->header) > MAX_SECTION_LEN)
646 		vidtv_psi_desc_assign(to, NULL);
647 
648 	vidtv_psi_update_version_num(&pmt->header);
649 }
650 
vidtv_sdt_desc_assign(struct vidtv_psi_table_sdt * sdt,struct vidtv_psi_desc ** to,struct vidtv_psi_desc * desc)651 void vidtv_sdt_desc_assign(struct vidtv_psi_table_sdt *sdt,
652 			   struct vidtv_psi_desc **to,
653 			   struct vidtv_psi_desc *desc)
654 {
655 	vidtv_psi_desc_assign(to, desc);
656 	vidtv_psi_sdt_table_update_sec_len(sdt);
657 
658 	if (vidtv_psi_get_sec_len(&sdt->header) > MAX_SECTION_LEN)
659 		vidtv_psi_desc_assign(to, NULL);
660 
661 	vidtv_psi_update_version_num(&sdt->header);
662 }
663 
vidtv_psi_desc_write_into(struct desc_write_args * args)664 static u32 vidtv_psi_desc_write_into(struct desc_write_args *args)
665 {
666 	struct psi_write_args psi_args = {
667 		.dest_buf           = args->dest_buf,
668 		.from               = &args->desc->type,
669 		.pid                = args->pid,
670 		.new_psi_section    = false,
671 		.continuity_counter = args->continuity_counter,
672 		.is_crc             = false,
673 		.dest_buf_sz        = args->dest_buf_sz,
674 		.crc                = args->crc,
675 		.len		    = sizeof_field(struct vidtv_psi_desc, type) +
676 				      sizeof_field(struct vidtv_psi_desc, length),
677 	};
678 	struct vidtv_psi_desc_service_list_entry *serv_list_entry = NULL;
679 	u32 nbytes = 0;
680 
681 	psi_args.dest_offset        = args->dest_offset + nbytes;
682 
683 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
684 
685 	switch (args->desc->type) {
686 	case SERVICE_DESCRIPTOR:
687 		psi_args.dest_offset = args->dest_offset + nbytes;
688 		psi_args.len = sizeof_field(struct vidtv_psi_desc_service, service_type) +
689 			       sizeof_field(struct vidtv_psi_desc_service, provider_name_len);
690 		psi_args.from = &((struct vidtv_psi_desc_service *)args->desc)->service_type;
691 
692 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
693 
694 		psi_args.dest_offset = args->dest_offset + nbytes;
695 		psi_args.len = ((struct vidtv_psi_desc_service *)args->desc)->provider_name_len;
696 		psi_args.from = ((struct vidtv_psi_desc_service *)args->desc)->provider_name;
697 
698 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
699 
700 		psi_args.dest_offset = args->dest_offset + nbytes;
701 		psi_args.len = sizeof_field(struct vidtv_psi_desc_service, service_name_len);
702 		psi_args.from = &((struct vidtv_psi_desc_service *)args->desc)->service_name_len;
703 
704 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
705 
706 		psi_args.dest_offset = args->dest_offset + nbytes;
707 		psi_args.len = ((struct vidtv_psi_desc_service *)args->desc)->service_name_len;
708 		psi_args.from = ((struct vidtv_psi_desc_service *)args->desc)->service_name;
709 
710 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
711 		break;
712 
713 	case NETWORK_NAME_DESCRIPTOR:
714 		psi_args.dest_offset = args->dest_offset + nbytes;
715 		psi_args.len = args->desc->length;
716 		psi_args.from = ((struct vidtv_psi_desc_network_name *)args->desc)->network_name;
717 
718 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
719 		break;
720 
721 	case SERVICE_LIST_DESCRIPTOR:
722 		serv_list_entry = ((struct vidtv_psi_desc_service_list *)args->desc)->service_list;
723 		while (serv_list_entry) {
724 			psi_args.dest_offset = args->dest_offset + nbytes;
725 			psi_args.len = sizeof(struct vidtv_psi_desc_service_list_entry) -
726 				       sizeof(struct vidtv_psi_desc_service_list_entry *);
727 			psi_args.from = serv_list_entry;
728 
729 			nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
730 
731 			serv_list_entry = serv_list_entry->next;
732 		}
733 		break;
734 
735 	case SHORT_EVENT_DESCRIPTOR:
736 		psi_args.dest_offset = args->dest_offset + nbytes;
737 		psi_args.len = ISO_LANGUAGE_CODE_LEN;
738 		psi_args.from = ((struct vidtv_psi_desc_short_event *)
739 				  args->desc)->iso_language_code;
740 
741 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
742 
743 		psi_args.dest_offset = args->dest_offset + nbytes;
744 		psi_args.len = sizeof_field(struct vidtv_psi_desc_short_event, event_name_len);
745 		psi_args.from = &((struct vidtv_psi_desc_short_event *)
746 				  args->desc)->event_name_len;
747 
748 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
749 
750 		psi_args.dest_offset = args->dest_offset + nbytes;
751 		psi_args.len = ((struct vidtv_psi_desc_short_event *)args->desc)->event_name_len;
752 		psi_args.from = ((struct vidtv_psi_desc_short_event *)args->desc)->event_name;
753 
754 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
755 
756 		psi_args.dest_offset = args->dest_offset + nbytes;
757 		psi_args.len = sizeof_field(struct vidtv_psi_desc_short_event, text_len);
758 		psi_args.from = &((struct vidtv_psi_desc_short_event *)args->desc)->text_len;
759 
760 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
761 
762 		psi_args.dest_offset = args->dest_offset + nbytes;
763 		psi_args.len = ((struct vidtv_psi_desc_short_event *)args->desc)->text_len;
764 		psi_args.from = ((struct vidtv_psi_desc_short_event *)args->desc)->text;
765 
766 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
767 
768 		break;
769 
770 	case REGISTRATION_DESCRIPTOR:
771 	default:
772 		psi_args.dest_offset = args->dest_offset + nbytes;
773 		psi_args.len = args->desc->length;
774 		psi_args.from = &args->desc->data;
775 
776 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
777 		break;
778 	}
779 
780 	return nbytes;
781 }
782 
783 static u32
vidtv_psi_table_header_write_into(struct header_write_args * args)784 vidtv_psi_table_header_write_into(struct header_write_args *args)
785 {
786 	struct psi_write_args psi_args = {
787 		.dest_buf           = args->dest_buf,
788 		.from               = args->h,
789 		.len                = sizeof(struct vidtv_psi_table_header),
790 		.dest_offset        = args->dest_offset,
791 		.pid                = args->pid,
792 		.new_psi_section    = true,
793 		.continuity_counter = args->continuity_counter,
794 		.is_crc             = false,
795 		.dest_buf_sz        = args->dest_buf_sz,
796 		.crc                = args->crc,
797 	};
798 
799 	return vidtv_psi_ts_psi_write_into(&psi_args);
800 }
801 
802 void
vidtv_psi_pat_table_update_sec_len(struct vidtv_psi_table_pat * pat)803 vidtv_psi_pat_table_update_sec_len(struct vidtv_psi_table_pat *pat)
804 {
805 	u16 length = 0;
806 	u32 i;
807 
808 	/* see ISO/IEC 13818-1 : 2000 p.43 */
809 
810 	/* from immediately after 'section_length' until 'last_section_number'*/
811 	length += PAT_LEN_UNTIL_LAST_SECTION_NUMBER;
812 
813 	/* do not count the pointer */
814 	for (i = 0; i < pat->num_pat; ++i)
815 		length += sizeof(struct vidtv_psi_table_pat_program) -
816 			  sizeof(struct vidtv_psi_table_pat_program *);
817 
818 	length += CRC_SIZE_IN_BYTES;
819 
820 	vidtv_psi_set_sec_len(&pat->header, length);
821 }
822 
vidtv_psi_pmt_table_update_sec_len(struct vidtv_psi_table_pmt * pmt)823 void vidtv_psi_pmt_table_update_sec_len(struct vidtv_psi_table_pmt *pmt)
824 {
825 	struct vidtv_psi_table_pmt_stream *s = pmt->stream;
826 	u16 desc_loop_len;
827 	u16 length = 0;
828 
829 	/* see ISO/IEC 13818-1 : 2000 p.46 */
830 
831 	/* from immediately after 'section_length' until 'program_info_length'*/
832 	length += PMT_LEN_UNTIL_PROGRAM_INFO_LENGTH;
833 
834 	desc_loop_len = vidtv_psi_desc_comp_loop_len(pmt->descriptor);
835 	vidtv_psi_set_desc_loop_len(&pmt->bitfield2, desc_loop_len, 10);
836 
837 	length += desc_loop_len;
838 
839 	while (s) {
840 		/* skip both pointers at the end */
841 		length += sizeof(struct vidtv_psi_table_pmt_stream) -
842 			  sizeof(struct vidtv_psi_desc *) -
843 			  sizeof(struct vidtv_psi_table_pmt_stream *);
844 
845 		desc_loop_len = vidtv_psi_desc_comp_loop_len(s->descriptor);
846 		vidtv_psi_set_desc_loop_len(&s->bitfield2, desc_loop_len, 10);
847 
848 		length += desc_loop_len;
849 
850 		s = s->next;
851 	}
852 
853 	length += CRC_SIZE_IN_BYTES;
854 
855 	vidtv_psi_set_sec_len(&pmt->header, length);
856 }
857 
vidtv_psi_sdt_table_update_sec_len(struct vidtv_psi_table_sdt * sdt)858 void vidtv_psi_sdt_table_update_sec_len(struct vidtv_psi_table_sdt *sdt)
859 {
860 	struct vidtv_psi_table_sdt_service *s = sdt->service;
861 	u16 desc_loop_len;
862 	u16 length = 0;
863 
864 	/* see ETSI EN 300 468 V 1.10.1 p.24 */
865 
866 	/*
867 	 * from immediately after 'section_length' until
868 	 * 'reserved_for_future_use'
869 	 */
870 	length += SDT_LEN_UNTIL_RESERVED_FOR_FUTURE_USE;
871 
872 	while (s) {
873 		/* skip both pointers at the end */
874 		length += sizeof(struct vidtv_psi_table_sdt_service) -
875 			  sizeof(struct vidtv_psi_desc *) -
876 			  sizeof(struct vidtv_psi_table_sdt_service *);
877 
878 		desc_loop_len = vidtv_psi_desc_comp_loop_len(s->descriptor);
879 		vidtv_psi_set_desc_loop_len(&s->bitfield, desc_loop_len, 12);
880 
881 		length += desc_loop_len;
882 
883 		s = s->next;
884 	}
885 
886 	length += CRC_SIZE_IN_BYTES;
887 	vidtv_psi_set_sec_len(&sdt->header, length);
888 }
889 
890 struct vidtv_psi_table_pat_program*
vidtv_psi_pat_program_init(struct vidtv_psi_table_pat_program * head,u16 service_id,u16 program_map_pid)891 vidtv_psi_pat_program_init(struct vidtv_psi_table_pat_program *head,
892 			   u16 service_id,
893 			   u16 program_map_pid)
894 {
895 	struct vidtv_psi_table_pat_program *program;
896 	const u16 RESERVED = 0x07;
897 
898 	program = kzalloc(sizeof(*program), GFP_KERNEL);
899 	if (!program)
900 		return NULL;
901 
902 	program->service_id = cpu_to_be16(service_id);
903 
904 	/* pid for the PMT section in the TS */
905 	program->bitfield = cpu_to_be16((RESERVED << 13) | program_map_pid);
906 	program->next = NULL;
907 
908 	if (head) {
909 		while (head->next)
910 			head = head->next;
911 
912 		head->next = program;
913 	}
914 
915 	return program;
916 }
917 
918 void
vidtv_psi_pat_program_destroy(struct vidtv_psi_table_pat_program * p)919 vidtv_psi_pat_program_destroy(struct vidtv_psi_table_pat_program *p)
920 {
921 	struct vidtv_psi_table_pat_program *tmp  = NULL;
922 	struct vidtv_psi_table_pat_program *curr = p;
923 
924 	while (curr) {
925 		tmp  = curr;
926 		curr = curr->next;
927 		kfree(tmp);
928 	}
929 }
930 
931 /* This function transfers ownership of p to the table */
932 void
vidtv_psi_pat_program_assign(struct vidtv_psi_table_pat * pat,struct vidtv_psi_table_pat_program * p)933 vidtv_psi_pat_program_assign(struct vidtv_psi_table_pat *pat,
934 			     struct vidtv_psi_table_pat_program *p)
935 {
936 	struct vidtv_psi_table_pat_program *program;
937 	u16 program_count;
938 
939 	do {
940 		program_count = 0;
941 		program = p;
942 
943 		if (p == pat->program)
944 			return;
945 
946 		while (program) {
947 			++program_count;
948 			program = program->next;
949 		}
950 
951 		pat->num_pat = program_count;
952 		pat->program  = p;
953 
954 		/* Recompute section length */
955 		vidtv_psi_pat_table_update_sec_len(pat);
956 
957 		p = NULL;
958 	} while (vidtv_psi_get_sec_len(&pat->header) > MAX_SECTION_LEN);
959 
960 	vidtv_psi_update_version_num(&pat->header);
961 }
962 
vidtv_psi_pat_table_init(u16 transport_stream_id)963 struct vidtv_psi_table_pat *vidtv_psi_pat_table_init(u16 transport_stream_id)
964 {
965 	struct vidtv_psi_table_pat *pat;
966 	const u16 SYNTAX = 0x1;
967 	const u16 ZERO = 0x0;
968 	const u16 ONES = 0x03;
969 
970 	pat = kzalloc(sizeof(*pat), GFP_KERNEL);
971 	if (!pat)
972 		return NULL;
973 
974 	pat->header.table_id = 0x0;
975 
976 	pat->header.bitfield = cpu_to_be16((SYNTAX << 15) | (ZERO << 14) | (ONES << 12));
977 	pat->header.id           = cpu_to_be16(transport_stream_id);
978 	pat->header.current_next = 0x1;
979 
980 	pat->header.version = 0x1f;
981 
982 	pat->header.one2         = 0x03;
983 	pat->header.section_id   = 0x0;
984 	pat->header.last_section = 0x0;
985 
986 	vidtv_psi_pat_table_update_sec_len(pat);
987 
988 	return pat;
989 }
990 
vidtv_psi_pat_write_into(struct vidtv_psi_pat_write_args * args)991 u32 vidtv_psi_pat_write_into(struct vidtv_psi_pat_write_args *args)
992 {
993 	struct vidtv_psi_table_pat_program *p = args->pat->program;
994 	struct header_write_args h_args       = {
995 		.dest_buf           = args->buf,
996 		.dest_offset        = args->offset,
997 		.pid                = VIDTV_PAT_PID,
998 		.h                  = &args->pat->header,
999 		.continuity_counter = args->continuity_counter,
1000 		.dest_buf_sz        = args->buf_sz,
1001 	};
1002 	struct psi_write_args psi_args        = {
1003 		.dest_buf           = args->buf,
1004 		.pid                = VIDTV_PAT_PID,
1005 		.new_psi_section    = false,
1006 		.continuity_counter = args->continuity_counter,
1007 		.is_crc             = false,
1008 		.dest_buf_sz        = args->buf_sz,
1009 	};
1010 	struct crc32_write_args c_args        = {
1011 		.dest_buf           = args->buf,
1012 		.pid                = VIDTV_PAT_PID,
1013 		.dest_buf_sz        = args->buf_sz,
1014 	};
1015 	u32 crc = INITIAL_CRC;
1016 	u32 nbytes = 0;
1017 
1018 	vidtv_psi_pat_table_update_sec_len(args->pat);
1019 
1020 	h_args.crc = &crc;
1021 
1022 	nbytes += vidtv_psi_table_header_write_into(&h_args);
1023 
1024 	/* note that the field 'u16 programs' is not really part of the PAT */
1025 
1026 	psi_args.crc = &crc;
1027 
1028 	while (p) {
1029 		/* copy the PAT programs */
1030 		psi_args.from = p;
1031 		/* skip the pointer */
1032 		psi_args.len = sizeof(*p) -
1033 			       sizeof(struct vidtv_psi_table_pat_program *);
1034 		psi_args.dest_offset = args->offset + nbytes;
1035 		psi_args.continuity_counter = args->continuity_counter;
1036 
1037 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1038 
1039 		p = p->next;
1040 	}
1041 
1042 	c_args.dest_offset        = args->offset + nbytes;
1043 	c_args.continuity_counter = args->continuity_counter;
1044 	c_args.crc                = cpu_to_be32(crc);
1045 
1046 	/* Write the CRC32 at the end */
1047 	nbytes += table_section_crc32_write_into(&c_args);
1048 
1049 	return nbytes;
1050 }
1051 
1052 void
vidtv_psi_pat_table_destroy(struct vidtv_psi_table_pat * p)1053 vidtv_psi_pat_table_destroy(struct vidtv_psi_table_pat *p)
1054 {
1055 	vidtv_psi_pat_program_destroy(p->program);
1056 	kfree(p);
1057 }
1058 
1059 struct vidtv_psi_table_pmt_stream*
vidtv_psi_pmt_stream_init(struct vidtv_psi_table_pmt_stream * head,enum vidtv_psi_stream_types stream_type,u16 es_pid)1060 vidtv_psi_pmt_stream_init(struct vidtv_psi_table_pmt_stream *head,
1061 			  enum vidtv_psi_stream_types stream_type,
1062 			  u16 es_pid)
1063 {
1064 	struct vidtv_psi_table_pmt_stream *stream;
1065 	const u16 RESERVED1 = 0x07;
1066 	const u16 RESERVED2 = 0x0f;
1067 	const u16 ZERO = 0x0;
1068 	u16 desc_loop_len;
1069 
1070 	stream = kzalloc(sizeof(*stream), GFP_KERNEL);
1071 	if (!stream)
1072 		return NULL;
1073 
1074 	stream->type = stream_type;
1075 
1076 	stream->bitfield = cpu_to_be16((RESERVED1 << 13) | es_pid);
1077 
1078 	desc_loop_len = vidtv_psi_desc_comp_loop_len(stream->descriptor);
1079 
1080 	stream->bitfield2 = cpu_to_be16((RESERVED2 << 12) |
1081 					(ZERO << 10)      |
1082 					desc_loop_len);
1083 	stream->next = NULL;
1084 
1085 	if (head) {
1086 		while (head->next)
1087 			head = head->next;
1088 
1089 		head->next = stream;
1090 	}
1091 
1092 	return stream;
1093 }
1094 
vidtv_psi_pmt_stream_destroy(struct vidtv_psi_table_pmt_stream * s)1095 void vidtv_psi_pmt_stream_destroy(struct vidtv_psi_table_pmt_stream *s)
1096 {
1097 	struct vidtv_psi_table_pmt_stream *tmp_stream  = NULL;
1098 	struct vidtv_psi_table_pmt_stream *curr_stream = s;
1099 
1100 	while (curr_stream) {
1101 		tmp_stream  = curr_stream;
1102 		curr_stream = curr_stream->next;
1103 		vidtv_psi_desc_destroy(tmp_stream->descriptor);
1104 		kfree(tmp_stream);
1105 	}
1106 }
1107 
vidtv_psi_pmt_stream_assign(struct vidtv_psi_table_pmt * pmt,struct vidtv_psi_table_pmt_stream * s)1108 void vidtv_psi_pmt_stream_assign(struct vidtv_psi_table_pmt *pmt,
1109 				 struct vidtv_psi_table_pmt_stream *s)
1110 {
1111 	do {
1112 		/* This function transfers ownership of s to the table */
1113 		if (s == pmt->stream)
1114 			return;
1115 
1116 		pmt->stream = s;
1117 		vidtv_psi_pmt_table_update_sec_len(pmt);
1118 
1119 		s = NULL;
1120 	} while (vidtv_psi_get_sec_len(&pmt->header) > MAX_SECTION_LEN);
1121 
1122 	vidtv_psi_update_version_num(&pmt->header);
1123 }
1124 
vidtv_psi_pmt_get_pid(struct vidtv_psi_table_pmt * section,struct vidtv_psi_table_pat * pat)1125 u16 vidtv_psi_pmt_get_pid(struct vidtv_psi_table_pmt *section,
1126 			  struct vidtv_psi_table_pat *pat)
1127 {
1128 	struct vidtv_psi_table_pat_program *program = pat->program;
1129 
1130 	/*
1131 	 * service_id is the same as program_number in the
1132 	 * corresponding program_map_section
1133 	 * see ETSI EN 300 468 v1.15.1 p. 24
1134 	 */
1135 	while (program) {
1136 		if (program->service_id == section->header.id)
1137 			return vidtv_psi_get_pat_program_pid(program);
1138 
1139 		program = program->next;
1140 	}
1141 
1142 	return TS_LAST_VALID_PID + 1; /* not found */
1143 }
1144 
vidtv_psi_pmt_table_init(u16 program_number,u16 pcr_pid)1145 struct vidtv_psi_table_pmt *vidtv_psi_pmt_table_init(u16 program_number,
1146 						     u16 pcr_pid)
1147 {
1148 	struct vidtv_psi_table_pmt *pmt;
1149 	const u16 RESERVED1 = 0x07;
1150 	const u16 RESERVED2 = 0x0f;
1151 	const u16 SYNTAX = 0x1;
1152 	const u16 ONES = 0x03;
1153 	const u16 ZERO = 0x0;
1154 	u16 desc_loop_len;
1155 
1156 	pmt = kzalloc(sizeof(*pmt), GFP_KERNEL);
1157 	if (!pmt)
1158 		return NULL;
1159 
1160 	if (!pcr_pid)
1161 		pcr_pid = 0x1fff;
1162 
1163 	pmt->header.table_id = 0x2;
1164 
1165 	pmt->header.bitfield = cpu_to_be16((SYNTAX << 15) | (ZERO << 14) | (ONES << 12));
1166 
1167 	pmt->header.id = cpu_to_be16(program_number);
1168 	pmt->header.current_next = 0x1;
1169 
1170 	pmt->header.version = 0x1f;
1171 
1172 	pmt->header.one2 = ONES;
1173 	pmt->header.section_id   = 0;
1174 	pmt->header.last_section = 0;
1175 
1176 	pmt->bitfield = cpu_to_be16((RESERVED1 << 13) | pcr_pid);
1177 
1178 	desc_loop_len = vidtv_psi_desc_comp_loop_len(pmt->descriptor);
1179 
1180 	pmt->bitfield2 = cpu_to_be16((RESERVED2 << 12) |
1181 				     (ZERO << 10)      |
1182 				     desc_loop_len);
1183 
1184 	vidtv_psi_pmt_table_update_sec_len(pmt);
1185 
1186 	return pmt;
1187 }
1188 
vidtv_psi_pmt_write_into(struct vidtv_psi_pmt_write_args * args)1189 u32 vidtv_psi_pmt_write_into(struct vidtv_psi_pmt_write_args *args)
1190 {
1191 	struct vidtv_psi_desc *table_descriptor   = args->pmt->descriptor;
1192 	struct vidtv_psi_table_pmt_stream *stream = args->pmt->stream;
1193 	struct vidtv_psi_desc *stream_descriptor;
1194 	u32 crc = INITIAL_CRC;
1195 	u32 nbytes = 0;
1196 	struct header_write_args h_args = {
1197 		.dest_buf           = args->buf,
1198 		.dest_offset        = args->offset,
1199 		.h                  = &args->pmt->header,
1200 		.pid                = args->pid,
1201 		.continuity_counter = args->continuity_counter,
1202 		.dest_buf_sz        = args->buf_sz,
1203 	};
1204 	struct psi_write_args psi_args  = {
1205 		.dest_buf = args->buf,
1206 		.from     = &args->pmt->bitfield,
1207 		.len      = sizeof_field(struct vidtv_psi_table_pmt, bitfield) +
1208 			    sizeof_field(struct vidtv_psi_table_pmt, bitfield2),
1209 		.pid                = args->pid,
1210 		.new_psi_section    = false,
1211 		.is_crc             = false,
1212 		.dest_buf_sz        = args->buf_sz,
1213 		.crc                = &crc,
1214 	};
1215 	struct desc_write_args d_args   = {
1216 		.dest_buf           = args->buf,
1217 		.desc               = table_descriptor,
1218 		.pid                = args->pid,
1219 		.dest_buf_sz        = args->buf_sz,
1220 	};
1221 	struct crc32_write_args c_args  = {
1222 		.dest_buf           = args->buf,
1223 		.pid                = args->pid,
1224 		.dest_buf_sz        = args->buf_sz,
1225 	};
1226 
1227 	vidtv_psi_pmt_table_update_sec_len(args->pmt);
1228 
1229 	h_args.crc                = &crc;
1230 
1231 	nbytes += vidtv_psi_table_header_write_into(&h_args);
1232 
1233 	/* write the two bitfields */
1234 	psi_args.dest_offset        = args->offset + nbytes;
1235 	psi_args.continuity_counter = args->continuity_counter;
1236 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1237 
1238 	while (table_descriptor) {
1239 		/* write the descriptors, if any */
1240 		d_args.dest_offset        = args->offset + nbytes;
1241 		d_args.continuity_counter = args->continuity_counter;
1242 		d_args.crc                = &crc;
1243 
1244 		nbytes += vidtv_psi_desc_write_into(&d_args);
1245 
1246 		table_descriptor = table_descriptor->next;
1247 	}
1248 
1249 	psi_args.len += sizeof_field(struct vidtv_psi_table_pmt_stream, type);
1250 	while (stream) {
1251 		/* write the streams, if any */
1252 		psi_args.from = stream;
1253 		psi_args.dest_offset = args->offset + nbytes;
1254 		psi_args.continuity_counter = args->continuity_counter;
1255 
1256 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1257 
1258 		stream_descriptor = stream->descriptor;
1259 
1260 		while (stream_descriptor) {
1261 			/* write the stream descriptors, if any */
1262 			d_args.dest_offset        = args->offset + nbytes;
1263 			d_args.desc               = stream_descriptor;
1264 			d_args.continuity_counter = args->continuity_counter;
1265 			d_args.crc                = &crc;
1266 
1267 			nbytes += vidtv_psi_desc_write_into(&d_args);
1268 
1269 			stream_descriptor = stream_descriptor->next;
1270 		}
1271 
1272 		stream = stream->next;
1273 	}
1274 
1275 	c_args.dest_offset        = args->offset + nbytes;
1276 	c_args.crc                = cpu_to_be32(crc);
1277 	c_args.continuity_counter = args->continuity_counter;
1278 
1279 	/* Write the CRC32 at the end */
1280 	nbytes += table_section_crc32_write_into(&c_args);
1281 
1282 	return nbytes;
1283 }
1284 
vidtv_psi_pmt_table_destroy(struct vidtv_psi_table_pmt * pmt)1285 void vidtv_psi_pmt_table_destroy(struct vidtv_psi_table_pmt *pmt)
1286 {
1287 	vidtv_psi_desc_destroy(pmt->descriptor);
1288 	vidtv_psi_pmt_stream_destroy(pmt->stream);
1289 	kfree(pmt);
1290 }
1291 
vidtv_psi_sdt_table_init(u16 network_id,u16 transport_stream_id)1292 struct vidtv_psi_table_sdt *vidtv_psi_sdt_table_init(u16 network_id,
1293 						     u16 transport_stream_id)
1294 {
1295 	struct vidtv_psi_table_sdt *sdt;
1296 	const u16 RESERVED = 0xff;
1297 	const u16 SYNTAX = 0x1;
1298 	const u16 ONES = 0x03;
1299 	const u16 ONE = 0x1;
1300 
1301 	sdt  = kzalloc(sizeof(*sdt), GFP_KERNEL);
1302 	if (!sdt)
1303 		return NULL;
1304 
1305 	sdt->header.table_id = 0x42;
1306 	sdt->header.bitfield = cpu_to_be16((SYNTAX << 15) | (ONE << 14) | (ONES << 12));
1307 
1308 	/*
1309 	 * This is a 16-bit field which serves as a label for identification
1310 	 * of the TS, about which the SDT informs, from any other multiplex
1311 	 * within the delivery system.
1312 	 */
1313 	sdt->header.id = cpu_to_be16(transport_stream_id);
1314 	sdt->header.current_next = ONE;
1315 
1316 	sdt->header.version = 0x1f;
1317 
1318 	sdt->header.one2  = ONES;
1319 	sdt->header.section_id   = 0;
1320 	sdt->header.last_section = 0;
1321 
1322 	/*
1323 	 * FIXME: The network_id range from 0xFF01 to 0xFFFF is used to
1324 	 * indicate temporary private use. For now, let's use the first
1325 	 * value.
1326 	 * This can be changed to something more useful, when support for
1327 	 * NIT gets added
1328 	 */
1329 	sdt->network_id = cpu_to_be16(network_id);
1330 	sdt->reserved = RESERVED;
1331 
1332 	vidtv_psi_sdt_table_update_sec_len(sdt);
1333 
1334 	return sdt;
1335 }
1336 
vidtv_psi_sdt_write_into(struct vidtv_psi_sdt_write_args * args)1337 u32 vidtv_psi_sdt_write_into(struct vidtv_psi_sdt_write_args *args)
1338 {
1339 	struct header_write_args h_args = {
1340 		.dest_buf           = args->buf,
1341 		.dest_offset        = args->offset,
1342 		.h                  = &args->sdt->header,
1343 		.pid                = VIDTV_SDT_PID,
1344 		.dest_buf_sz        = args->buf_sz,
1345 	};
1346 	struct psi_write_args psi_args  = {
1347 		.dest_buf = args->buf,
1348 		.len = sizeof_field(struct vidtv_psi_table_sdt, network_id) +
1349 		       sizeof_field(struct vidtv_psi_table_sdt, reserved),
1350 		.pid                = VIDTV_SDT_PID,
1351 		.new_psi_section    = false,
1352 		.is_crc             = false,
1353 		.dest_buf_sz        = args->buf_sz,
1354 	};
1355 	struct desc_write_args d_args   = {
1356 		.dest_buf           = args->buf,
1357 		.pid                = VIDTV_SDT_PID,
1358 		.dest_buf_sz        = args->buf_sz,
1359 	};
1360 	struct crc32_write_args c_args  = {
1361 		.dest_buf           = args->buf,
1362 		.pid                = VIDTV_SDT_PID,
1363 		.dest_buf_sz        = args->buf_sz,
1364 	};
1365 	struct vidtv_psi_table_sdt_service *service = args->sdt->service;
1366 	struct vidtv_psi_desc *service_desc;
1367 	u32 nbytes  = 0;
1368 	u32 crc = INITIAL_CRC;
1369 
1370 	/* see ETSI EN 300 468 v1.15.1 p. 11 */
1371 
1372 	vidtv_psi_sdt_table_update_sec_len(args->sdt);
1373 
1374 	h_args.continuity_counter = args->continuity_counter;
1375 	h_args.crc                = &crc;
1376 
1377 	nbytes += vidtv_psi_table_header_write_into(&h_args);
1378 
1379 	psi_args.from               = &args->sdt->network_id;
1380 	psi_args.dest_offset        = args->offset + nbytes;
1381 	psi_args.continuity_counter = args->continuity_counter;
1382 	psi_args.crc                = &crc;
1383 
1384 	/* copy u16 network_id + u8 reserved)*/
1385 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1386 
1387 	/* skip both pointers at the end */
1388 	psi_args.len = sizeof(struct vidtv_psi_table_sdt_service) -
1389 		       sizeof(struct vidtv_psi_desc *) -
1390 		       sizeof(struct vidtv_psi_table_sdt_service *);
1391 
1392 	while (service) {
1393 		/* copy the services, if any */
1394 		psi_args.from = service;
1395 		psi_args.dest_offset = args->offset + nbytes;
1396 		psi_args.continuity_counter = args->continuity_counter;
1397 
1398 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1399 
1400 		service_desc = service->descriptor;
1401 
1402 		while (service_desc) {
1403 			/* copy the service descriptors, if any */
1404 			d_args.dest_offset        = args->offset + nbytes;
1405 			d_args.desc               = service_desc;
1406 			d_args.continuity_counter = args->continuity_counter;
1407 			d_args.crc                = &crc;
1408 
1409 			nbytes += vidtv_psi_desc_write_into(&d_args);
1410 
1411 			service_desc = service_desc->next;
1412 		}
1413 
1414 		service = service->next;
1415 	}
1416 
1417 	c_args.dest_offset        = args->offset + nbytes;
1418 	c_args.crc                = cpu_to_be32(crc);
1419 	c_args.continuity_counter = args->continuity_counter;
1420 
1421 	/* Write the CRC at the end */
1422 	nbytes += table_section_crc32_write_into(&c_args);
1423 
1424 	return nbytes;
1425 }
1426 
vidtv_psi_sdt_table_destroy(struct vidtv_psi_table_sdt * sdt)1427 void vidtv_psi_sdt_table_destroy(struct vidtv_psi_table_sdt *sdt)
1428 {
1429 	vidtv_psi_sdt_service_destroy(sdt->service);
1430 	kfree(sdt);
1431 }
1432 
1433 struct vidtv_psi_table_sdt_service
vidtv_psi_sdt_service_init(struct vidtv_psi_table_sdt_service * head,u16 service_id,bool eit_schedule,bool eit_present_following)1434 *vidtv_psi_sdt_service_init(struct vidtv_psi_table_sdt_service *head,
1435 			    u16 service_id,
1436 			    bool eit_schedule,
1437 			    bool eit_present_following)
1438 {
1439 	struct vidtv_psi_table_sdt_service *service;
1440 
1441 	service = kzalloc(sizeof(*service), GFP_KERNEL);
1442 	if (!service)
1443 		return NULL;
1444 
1445 	/*
1446 	 * ETSI 300 468: this is a 16bit field which serves as a label to
1447 	 * identify this service from any other service within the TS.
1448 	 * The service id is the same as the program number in the
1449 	 * corresponding program_map_section
1450 	 */
1451 	service->service_id            = cpu_to_be16(service_id);
1452 	service->EIT_schedule          = eit_schedule;
1453 	service->EIT_present_following = eit_present_following;
1454 	service->reserved              = 0x3f;
1455 
1456 	service->bitfield = cpu_to_be16(RUNNING << 13);
1457 
1458 	if (head) {
1459 		while (head->next)
1460 			head = head->next;
1461 
1462 		head->next = service;
1463 	}
1464 
1465 	return service;
1466 }
1467 
1468 void
vidtv_psi_sdt_service_destroy(struct vidtv_psi_table_sdt_service * service)1469 vidtv_psi_sdt_service_destroy(struct vidtv_psi_table_sdt_service *service)
1470 {
1471 	struct vidtv_psi_table_sdt_service *curr = service;
1472 	struct vidtv_psi_table_sdt_service *tmp  = NULL;
1473 
1474 	while (curr) {
1475 		tmp  = curr;
1476 		curr = curr->next;
1477 		vidtv_psi_desc_destroy(tmp->descriptor);
1478 		kfree(tmp);
1479 	}
1480 }
1481 
1482 void
vidtv_psi_sdt_service_assign(struct vidtv_psi_table_sdt * sdt,struct vidtv_psi_table_sdt_service * service)1483 vidtv_psi_sdt_service_assign(struct vidtv_psi_table_sdt *sdt,
1484 			     struct vidtv_psi_table_sdt_service *service)
1485 {
1486 	do {
1487 		if (service == sdt->service)
1488 			return;
1489 
1490 		sdt->service = service;
1491 
1492 		/* recompute section length */
1493 		vidtv_psi_sdt_table_update_sec_len(sdt);
1494 
1495 		service = NULL;
1496 	} while (vidtv_psi_get_sec_len(&sdt->header) > MAX_SECTION_LEN);
1497 
1498 	vidtv_psi_update_version_num(&sdt->header);
1499 }
1500 
1501 /*
1502  * PMTs contain information about programs. For each program,
1503  * there is one PMT section. This function will create a section
1504  * for each program found in the PAT
1505  */
1506 struct vidtv_psi_table_pmt**
vidtv_psi_pmt_create_sec_for_each_pat_entry(struct vidtv_psi_table_pat * pat,u16 pcr_pid)1507 vidtv_psi_pmt_create_sec_for_each_pat_entry(struct vidtv_psi_table_pat *pat,
1508 					    u16 pcr_pid)
1509 
1510 {
1511 	struct vidtv_psi_table_pat_program *program;
1512 	struct vidtv_psi_table_pmt **pmt_secs;
1513 	u32 i = 0, num_pmt = 0;
1514 
1515 	/*
1516 	 * The number of PMT entries is the number of PAT entries
1517 	 * that contain service_id. That exclude special tables, like NIT
1518 	 */
1519 	program = pat->program;
1520 	while (program) {
1521 		if (program->service_id)
1522 			num_pmt++;
1523 		program = program->next;
1524 	}
1525 
1526 	pmt_secs = kcalloc(num_pmt,
1527 			   sizeof(struct vidtv_psi_table_pmt *),
1528 			   GFP_KERNEL);
1529 	if (!pmt_secs)
1530 		return NULL;
1531 
1532 	for (program = pat->program; program; program = program->next) {
1533 		if (!program->service_id)
1534 			continue;
1535 		pmt_secs[i] = vidtv_psi_pmt_table_init(be16_to_cpu(program->service_id),
1536 						       pcr_pid);
1537 
1538 		if (!pmt_secs[i]) {
1539 			while (i > 0) {
1540 				i--;
1541 				vidtv_psi_pmt_table_destroy(pmt_secs[i]);
1542 			}
1543 			return NULL;
1544 		}
1545 		i++;
1546 	}
1547 	pat->num_pmt = num_pmt;
1548 
1549 	return pmt_secs;
1550 }
1551 
1552 /* find the PMT section associated with 'program_num' */
1553 struct vidtv_psi_table_pmt
vidtv_psi_find_pmt_sec(struct vidtv_psi_table_pmt ** pmt_sections,u16 nsections,u16 program_num)1554 *vidtv_psi_find_pmt_sec(struct vidtv_psi_table_pmt **pmt_sections,
1555 			u16 nsections,
1556 			u16 program_num)
1557 {
1558 	struct vidtv_psi_table_pmt *sec = NULL;
1559 	u32 i;
1560 
1561 	for (i = 0; i < nsections; ++i) {
1562 		sec = pmt_sections[i];
1563 		if (be16_to_cpu(sec->header.id) == program_num)
1564 			return sec;
1565 	}
1566 
1567 	return NULL; /* not found */
1568 }
1569 
vidtv_psi_nit_table_update_sec_len(struct vidtv_psi_table_nit * nit)1570 static void vidtv_psi_nit_table_update_sec_len(struct vidtv_psi_table_nit *nit)
1571 {
1572 	u16 length = 0;
1573 	struct vidtv_psi_table_transport *t = nit->transport;
1574 	u16 desc_loop_len;
1575 	u16 transport_loop_len = 0;
1576 
1577 	/*
1578 	 * from immediately after 'section_length' until
1579 	 * 'network_descriptor_length'
1580 	 */
1581 	length += NIT_LEN_UNTIL_NETWORK_DESCRIPTOR_LEN;
1582 
1583 	desc_loop_len = vidtv_psi_desc_comp_loop_len(nit->descriptor);
1584 	vidtv_psi_set_desc_loop_len(&nit->bitfield, desc_loop_len, 12);
1585 
1586 	length += desc_loop_len;
1587 
1588 	length += sizeof_field(struct vidtv_psi_table_nit, bitfield2);
1589 
1590 	while (t) {
1591 		/* skip both pointers at the end */
1592 		transport_loop_len += sizeof(struct vidtv_psi_table_transport) -
1593 				      sizeof(struct vidtv_psi_desc *) -
1594 				      sizeof(struct vidtv_psi_table_transport *);
1595 
1596 		length += transport_loop_len;
1597 
1598 		desc_loop_len = vidtv_psi_desc_comp_loop_len(t->descriptor);
1599 		vidtv_psi_set_desc_loop_len(&t->bitfield, desc_loop_len, 12);
1600 
1601 		length += desc_loop_len;
1602 
1603 		t = t->next;
1604 	}
1605 
1606 	// Actually sets the transport stream loop len, maybe rename this function later
1607 	vidtv_psi_set_desc_loop_len(&nit->bitfield2, transport_loop_len, 12);
1608 	length += CRC_SIZE_IN_BYTES;
1609 
1610 	vidtv_psi_set_sec_len(&nit->header, length);
1611 }
1612 
1613 struct vidtv_psi_table_nit
vidtv_psi_nit_table_init(u16 network_id,u16 transport_stream_id,char * network_name,struct vidtv_psi_desc_service_list_entry * service_list)1614 *vidtv_psi_nit_table_init(u16 network_id,
1615 			  u16 transport_stream_id,
1616 			  char *network_name,
1617 			  struct vidtv_psi_desc_service_list_entry *service_list)
1618 {
1619 	struct vidtv_psi_table_transport *transport;
1620 	struct vidtv_psi_table_nit *nit;
1621 	const u16 SYNTAX = 0x1;
1622 	const u16 ONES = 0x03;
1623 	const u16 ONE = 0x1;
1624 
1625 	nit = kzalloc(sizeof(*nit), GFP_KERNEL);
1626 	if (!nit)
1627 		return NULL;
1628 
1629 	transport = kzalloc(sizeof(*transport), GFP_KERNEL);
1630 	if (!transport)
1631 		goto free_nit;
1632 
1633 	nit->header.table_id = 0x40; // ACTUAL_NETWORK
1634 
1635 	nit->header.bitfield = cpu_to_be16((SYNTAX << 15) | (ONE << 14) | (ONES << 12));
1636 
1637 	nit->header.id = cpu_to_be16(network_id);
1638 	nit->header.current_next = ONE;
1639 
1640 	nit->header.version = 0x1f;
1641 
1642 	nit->header.one2  = ONES;
1643 	nit->header.section_id   = 0;
1644 	nit->header.last_section = 0;
1645 
1646 	nit->bitfield = cpu_to_be16(0xf);
1647 	nit->bitfield2 = cpu_to_be16(0xf);
1648 
1649 	nit->descriptor = (struct vidtv_psi_desc *)
1650 			  vidtv_psi_network_name_desc_init(NULL, network_name);
1651 	if (!nit->descriptor)
1652 		goto free_transport;
1653 
1654 	transport->transport_id = cpu_to_be16(transport_stream_id);
1655 	transport->network_id = cpu_to_be16(network_id);
1656 	transport->bitfield = cpu_to_be16(0xf);
1657 	transport->descriptor = (struct vidtv_psi_desc *)
1658 				vidtv_psi_service_list_desc_init(NULL, service_list);
1659 	if (!transport->descriptor)
1660 		goto free_nit_desc;
1661 
1662 	nit->transport = transport;
1663 
1664 	vidtv_psi_nit_table_update_sec_len(nit);
1665 
1666 	return nit;
1667 
1668 free_nit_desc:
1669 	vidtv_psi_desc_destroy((struct vidtv_psi_desc *)nit->descriptor);
1670 
1671 free_transport:
1672 	kfree(transport);
1673 free_nit:
1674 	kfree(nit);
1675 	return NULL;
1676 }
1677 
vidtv_psi_nit_write_into(struct vidtv_psi_nit_write_args * args)1678 u32 vidtv_psi_nit_write_into(struct vidtv_psi_nit_write_args *args)
1679 {
1680 	struct header_write_args h_args = {
1681 		.dest_buf           = args->buf,
1682 		.dest_offset        = args->offset,
1683 		.h                  = &args->nit->header,
1684 		.pid                = VIDTV_NIT_PID,
1685 		.dest_buf_sz        = args->buf_sz,
1686 	};
1687 	struct psi_write_args psi_args  = {
1688 		.dest_buf           = args->buf,
1689 		.from               = &args->nit->bitfield,
1690 		.len                = sizeof_field(struct vidtv_psi_table_nit, bitfield),
1691 		.pid                = VIDTV_NIT_PID,
1692 		.new_psi_section    = false,
1693 		.is_crc             = false,
1694 		.dest_buf_sz        = args->buf_sz,
1695 	};
1696 	struct desc_write_args d_args   = {
1697 		.dest_buf           = args->buf,
1698 		.pid                = VIDTV_NIT_PID,
1699 		.dest_buf_sz        = args->buf_sz,
1700 	};
1701 	struct crc32_write_args c_args  = {
1702 		.dest_buf           = args->buf,
1703 		.pid                = VIDTV_NIT_PID,
1704 		.dest_buf_sz        = args->buf_sz,
1705 	};
1706 	struct vidtv_psi_desc *table_descriptor     = args->nit->descriptor;
1707 	struct vidtv_psi_table_transport *transport = args->nit->transport;
1708 	struct vidtv_psi_desc *transport_descriptor;
1709 	u32 crc = INITIAL_CRC;
1710 	u32 nbytes = 0;
1711 
1712 	vidtv_psi_nit_table_update_sec_len(args->nit);
1713 
1714 	h_args.continuity_counter = args->continuity_counter;
1715 	h_args.crc                = &crc;
1716 
1717 	nbytes += vidtv_psi_table_header_write_into(&h_args);
1718 
1719 	/* write the bitfield */
1720 
1721 	psi_args.dest_offset        = args->offset + nbytes;
1722 	psi_args.continuity_counter = args->continuity_counter;
1723 	psi_args.crc                = &crc;
1724 
1725 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1726 
1727 	while (table_descriptor) {
1728 		/* write the descriptors, if any */
1729 		d_args.dest_offset        = args->offset + nbytes;
1730 		d_args.desc               = table_descriptor;
1731 		d_args.continuity_counter = args->continuity_counter;
1732 		d_args.crc                = &crc;
1733 
1734 		nbytes += vidtv_psi_desc_write_into(&d_args);
1735 
1736 		table_descriptor = table_descriptor->next;
1737 	}
1738 
1739 	/* write the second bitfield */
1740 	psi_args.from = &args->nit->bitfield2;
1741 	psi_args.len = sizeof_field(struct vidtv_psi_table_nit, bitfield2);
1742 	psi_args.dest_offset = args->offset + nbytes;
1743 
1744 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1745 
1746 	psi_args.len  = sizeof_field(struct vidtv_psi_table_transport, transport_id) +
1747 			sizeof_field(struct vidtv_psi_table_transport, network_id)   +
1748 			sizeof_field(struct vidtv_psi_table_transport, bitfield);
1749 	while (transport) {
1750 		/* write the transport sections, if any */
1751 		psi_args.from = transport;
1752 		psi_args.dest_offset = args->offset + nbytes;
1753 
1754 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1755 
1756 		transport_descriptor = transport->descriptor;
1757 
1758 		while (transport_descriptor) {
1759 			/* write the transport descriptors, if any */
1760 			d_args.dest_offset        = args->offset + nbytes;
1761 			d_args.desc               = transport_descriptor;
1762 			d_args.continuity_counter = args->continuity_counter;
1763 			d_args.crc                = &crc;
1764 
1765 			nbytes += vidtv_psi_desc_write_into(&d_args);
1766 
1767 			transport_descriptor = transport_descriptor->next;
1768 		}
1769 
1770 		transport = transport->next;
1771 	}
1772 
1773 	c_args.dest_offset        = args->offset + nbytes;
1774 	c_args.crc                = cpu_to_be32(crc);
1775 	c_args.continuity_counter = args->continuity_counter;
1776 
1777 	/* Write the CRC32 at the end */
1778 	nbytes += table_section_crc32_write_into(&c_args);
1779 
1780 	return nbytes;
1781 }
1782 
vidtv_psi_transport_destroy(struct vidtv_psi_table_transport * t)1783 static void vidtv_psi_transport_destroy(struct vidtv_psi_table_transport *t)
1784 {
1785 	struct vidtv_psi_table_transport *tmp_t  = NULL;
1786 	struct vidtv_psi_table_transport *curr_t = t;
1787 
1788 	while (curr_t) {
1789 		tmp_t  = curr_t;
1790 		curr_t = curr_t->next;
1791 		vidtv_psi_desc_destroy(tmp_t->descriptor);
1792 		kfree(tmp_t);
1793 	}
1794 }
1795 
vidtv_psi_nit_table_destroy(struct vidtv_psi_table_nit * nit)1796 void vidtv_psi_nit_table_destroy(struct vidtv_psi_table_nit *nit)
1797 {
1798 	vidtv_psi_desc_destroy(nit->descriptor);
1799 	vidtv_psi_transport_destroy(nit->transport);
1800 	kfree(nit);
1801 }
1802 
vidtv_psi_eit_table_update_sec_len(struct vidtv_psi_table_eit * eit)1803 void vidtv_psi_eit_table_update_sec_len(struct vidtv_psi_table_eit *eit)
1804 {
1805 	struct vidtv_psi_table_eit_event *e = eit->event;
1806 	u16 desc_loop_len;
1807 	u16 length = 0;
1808 
1809 	/*
1810 	 * from immediately after 'section_length' until
1811 	 * 'last_table_id'
1812 	 */
1813 	length += EIT_LEN_UNTIL_LAST_TABLE_ID;
1814 
1815 	while (e) {
1816 		/* skip both pointers at the end */
1817 		length += sizeof(struct vidtv_psi_table_eit_event) -
1818 			  sizeof(struct vidtv_psi_desc *) -
1819 			  sizeof(struct vidtv_psi_table_eit_event *);
1820 
1821 		desc_loop_len = vidtv_psi_desc_comp_loop_len(e->descriptor);
1822 		vidtv_psi_set_desc_loop_len(&e->bitfield, desc_loop_len, 12);
1823 
1824 		length += desc_loop_len;
1825 
1826 		e = e->next;
1827 	}
1828 
1829 	length += CRC_SIZE_IN_BYTES;
1830 
1831 	vidtv_psi_set_sec_len(&eit->header, length);
1832 }
1833 
vidtv_psi_eit_event_assign(struct vidtv_psi_table_eit * eit,struct vidtv_psi_table_eit_event * e)1834 void vidtv_psi_eit_event_assign(struct vidtv_psi_table_eit *eit,
1835 				struct vidtv_psi_table_eit_event *e)
1836 {
1837 	do {
1838 		if (e == eit->event)
1839 			return;
1840 
1841 		eit->event = e;
1842 		vidtv_psi_eit_table_update_sec_len(eit);
1843 
1844 		e = NULL;
1845 	} while (vidtv_psi_get_sec_len(&eit->header) > EIT_MAX_SECTION_LEN);
1846 
1847 	vidtv_psi_update_version_num(&eit->header);
1848 }
1849 
1850 struct vidtv_psi_table_eit
vidtv_psi_eit_table_init(u16 network_id,u16 transport_stream_id,__be16 service_id)1851 *vidtv_psi_eit_table_init(u16 network_id,
1852 			  u16 transport_stream_id,
1853 			  __be16 service_id)
1854 {
1855 	struct vidtv_psi_table_eit *eit;
1856 	const u16 SYNTAX = 0x1;
1857 	const u16 ONE = 0x1;
1858 	const u16 ONES = 0x03;
1859 
1860 	eit = kzalloc(sizeof(*eit), GFP_KERNEL);
1861 	if (!eit)
1862 		return NULL;
1863 
1864 	eit->header.table_id = 0x4e; //actual_transport_stream: present/following
1865 
1866 	eit->header.bitfield = cpu_to_be16((SYNTAX << 15) | (ONE << 14) | (ONES << 12));
1867 
1868 	eit->header.id = service_id;
1869 	eit->header.current_next = ONE;
1870 
1871 	eit->header.version = 0x1f;
1872 
1873 	eit->header.one2  = ONES;
1874 	eit->header.section_id   = 0;
1875 	eit->header.last_section = 0;
1876 
1877 	eit->transport_id = cpu_to_be16(transport_stream_id);
1878 	eit->network_id = cpu_to_be16(network_id);
1879 
1880 	eit->last_segment = eit->header.last_section; /* not implemented */
1881 	eit->last_table_id = eit->header.table_id; /* not implemented */
1882 
1883 	vidtv_psi_eit_table_update_sec_len(eit);
1884 
1885 	return eit;
1886 }
1887 
vidtv_psi_eit_write_into(struct vidtv_psi_eit_write_args * args)1888 u32 vidtv_psi_eit_write_into(struct vidtv_psi_eit_write_args *args)
1889 {
1890 	struct header_write_args h_args = {
1891 		.dest_buf        = args->buf,
1892 		.dest_offset     = args->offset,
1893 		.h               = &args->eit->header,
1894 		.pid             = VIDTV_EIT_PID,
1895 		.dest_buf_sz     = args->buf_sz,
1896 	};
1897 	struct psi_write_args psi_args  = {
1898 		.dest_buf        = args->buf,
1899 		.len             = sizeof_field(struct vidtv_psi_table_eit, transport_id) +
1900 				   sizeof_field(struct vidtv_psi_table_eit, network_id)   +
1901 				   sizeof_field(struct vidtv_psi_table_eit, last_segment) +
1902 				   sizeof_field(struct vidtv_psi_table_eit, last_table_id),
1903 		.pid             = VIDTV_EIT_PID,
1904 		.new_psi_section = false,
1905 		.is_crc          = false,
1906 		.dest_buf_sz     = args->buf_sz,
1907 	};
1908 	struct desc_write_args d_args   = {
1909 		.dest_buf           = args->buf,
1910 		.pid                = VIDTV_EIT_PID,
1911 		.dest_buf_sz        = args->buf_sz,
1912 	};
1913 	struct crc32_write_args c_args  = {
1914 		.dest_buf           = args->buf,
1915 		.pid                = VIDTV_EIT_PID,
1916 		.dest_buf_sz        = args->buf_sz,
1917 	};
1918 	struct vidtv_psi_table_eit_event *event = args->eit->event;
1919 	struct vidtv_psi_desc *event_descriptor;
1920 	u32 crc = INITIAL_CRC;
1921 	u32 nbytes  = 0;
1922 
1923 	vidtv_psi_eit_table_update_sec_len(args->eit);
1924 
1925 	h_args.continuity_counter = args->continuity_counter;
1926 	h_args.crc                = &crc;
1927 
1928 	nbytes += vidtv_psi_table_header_write_into(&h_args);
1929 
1930 	psi_args.from               = &args->eit->transport_id;
1931 	psi_args.dest_offset        = args->offset + nbytes;
1932 	psi_args.continuity_counter = args->continuity_counter;
1933 	psi_args.crc                = &crc;
1934 
1935 	nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1936 
1937 	/* skip both pointers at the end */
1938 	psi_args.len = sizeof(struct vidtv_psi_table_eit_event) -
1939 		       sizeof(struct vidtv_psi_desc *) -
1940 		       sizeof(struct vidtv_psi_table_eit_event *);
1941 	while (event) {
1942 		/* copy the events, if any */
1943 		psi_args.from = event;
1944 		psi_args.dest_offset = args->offset + nbytes;
1945 
1946 		nbytes += vidtv_psi_ts_psi_write_into(&psi_args);
1947 
1948 		event_descriptor = event->descriptor;
1949 
1950 		while (event_descriptor) {
1951 			/* copy the event descriptors, if any */
1952 			d_args.dest_offset        = args->offset + nbytes;
1953 			d_args.desc               = event_descriptor;
1954 			d_args.continuity_counter = args->continuity_counter;
1955 			d_args.crc                = &crc;
1956 
1957 			nbytes += vidtv_psi_desc_write_into(&d_args);
1958 
1959 			event_descriptor = event_descriptor->next;
1960 		}
1961 
1962 		event = event->next;
1963 	}
1964 
1965 	c_args.dest_offset        = args->offset + nbytes;
1966 	c_args.crc                = cpu_to_be32(crc);
1967 	c_args.continuity_counter = args->continuity_counter;
1968 
1969 	/* Write the CRC at the end */
1970 	nbytes += table_section_crc32_write_into(&c_args);
1971 
1972 	return nbytes;
1973 }
1974 
1975 struct vidtv_psi_table_eit_event
vidtv_psi_eit_event_init(struct vidtv_psi_table_eit_event * head,u16 event_id)1976 *vidtv_psi_eit_event_init(struct vidtv_psi_table_eit_event *head, u16 event_id)
1977 {
1978 	static const u8 DURATION[] = {0x23, 0x59, 0x59}; /* BCD encoded */
1979 	struct vidtv_psi_table_eit_event *e;
1980 	struct timespec64 ts;
1981 	struct tm time;
1982 	int mjd, l;
1983 	__be16 mjd_be;
1984 
1985 	e = kzalloc(sizeof(*e), GFP_KERNEL);
1986 	if (!e)
1987 		return NULL;
1988 
1989 	e->event_id = cpu_to_be16(event_id);
1990 
1991 	ts = ktime_to_timespec64(ktime_get_real());
1992 	time64_to_tm(ts.tv_sec, 0, &time);
1993 
1994 	/* Convert date to Modified Julian Date - per EN 300 468 Annex C */
1995 	if (time.tm_mon < 2)
1996 		l = 1;
1997 	else
1998 		l = 0;
1999 
2000 	mjd = 14956 + time.tm_mday;
2001 	mjd += (time.tm_year - l) * 36525 / 100;
2002 	mjd += (time.tm_mon + 2 + l * 12) * 306001 / 10000;
2003 	mjd_be = cpu_to_be16(mjd);
2004 
2005 	/*
2006 	 * Store MJD and hour/min/sec to the event.
2007 	 *
2008 	 * Let's make the event to start on a full hour
2009 	 */
2010 	memcpy(e->start_time, &mjd_be, sizeof(mjd_be));
2011 	e->start_time[2] = bin2bcd(time.tm_hour);
2012 	e->start_time[3] = 0;
2013 	e->start_time[4] = 0;
2014 
2015 	/*
2016 	 * TODO: for now, the event will last for a day. Should be
2017 	 * enough for testing purposes, but if one runs the driver
2018 	 * for more than that, the current event will become invalid.
2019 	 * So, we need a better code here in order to change the start
2020 	 * time once the event expires.
2021 	 */
2022 	memcpy(e->duration, DURATION, sizeof(e->duration));
2023 
2024 	e->bitfield = cpu_to_be16(RUNNING << 13);
2025 
2026 	if (head) {
2027 		while (head->next)
2028 			head = head->next;
2029 
2030 		head->next = e;
2031 	}
2032 
2033 	return e;
2034 }
2035 
vidtv_psi_eit_event_destroy(struct vidtv_psi_table_eit_event * e)2036 void vidtv_psi_eit_event_destroy(struct vidtv_psi_table_eit_event *e)
2037 {
2038 	struct vidtv_psi_table_eit_event *tmp_e  = NULL;
2039 	struct vidtv_psi_table_eit_event *curr_e = e;
2040 
2041 	while (curr_e) {
2042 		tmp_e  = curr_e;
2043 		curr_e = curr_e->next;
2044 		vidtv_psi_desc_destroy(tmp_e->descriptor);
2045 		kfree(tmp_e);
2046 	}
2047 }
2048 
vidtv_psi_eit_table_destroy(struct vidtv_psi_table_eit * eit)2049 void vidtv_psi_eit_table_destroy(struct vidtv_psi_table_eit *eit)
2050 {
2051 	vidtv_psi_eit_event_destroy(eit->event);
2052 	kfree(eit);
2053 }
2054