[Linux-kernel-mentees] [RFC, WIP, v4 08/11] media: vidtv: implement a PSI generator

Mauro Carvalho Chehab mchehab+huawei at kernel.org
Sun May 3 07:51:05 UTC 2020


Em Sat,  2 May 2020 00:22:13 -0300
"Daniel W. S. Almeida" <dwlsalmeida at gmail.com> escreveu:

> From: "Daniel W. S. Almeida" <dwlsalmeida at gmail.com>
> 
> PSI packets contain general information about a MPEG Transport Stream.
> A PSI generator is needed so userspace apps can retrieve information
> about the Transport Stream and eventually tune into a (dummy) channel.
> 
> Because the generator is implemented in a separate file, it can be
> reused elsewhere in the media subsystem.
> 
> Currently this commit adds support for working with 3 PSI tables:
> PAT, PMT and SDT.
> 
> Signed-off-by: Daniel W. S. Almeida <dwlsalmeida at gmail.com>
> ---
>  drivers/media/test-drivers/vidtv/Makefile    |    2 +-
>  drivers/media/test-drivers/vidtv/vidtv_psi.c | 1137 ++++++++++++++++++
>  drivers/media/test-drivers/vidtv/vidtv_psi.h |  357 ++++++
>  3 files changed, 1495 insertions(+), 1 deletion(-)
>  create mode 100644 drivers/media/test-drivers/vidtv/vidtv_psi.c
>  create mode 100644 drivers/media/test-drivers/vidtv/vidtv_psi.h
> 
> diff --git a/drivers/media/test-drivers/vidtv/Makefile b/drivers/media/test-drivers/vidtv/Makefile
> index 92001bc348615..e4f744aa53136 100644
> --- a/drivers/media/test-drivers/vidtv/Makefile
> +++ b/drivers/media/test-drivers/vidtv/Makefile
> @@ -1,6 +1,6 @@
>  # SPDX-License-Identifier: GPL-2.0
>  
>  vidtv_demod-objs := vidtv_common.o
> -vidtv_bridge-objs := vidtv_common.o vidtv_ts.o
> +vidtv_bridge-objs := vidtv_common.o vidtv_ts.o vidtv_psi.o
>  
>  obj-$(CONFIG_DVB_VIDTV)	+= vidtv_tuner.o vidtv_demod.o vidtv_bridge.o
> diff --git a/drivers/media/test-drivers/vidtv/vidtv_psi.c b/drivers/media/test-drivers/vidtv/vidtv_psi.c
> new file mode 100644
> index 0000000000000..191d37a248923
> --- /dev/null
> +++ b/drivers/media/test-drivers/vidtv/vidtv_psi.c
> @@ -0,0 +1,1137 @@
> +// SPDX-License-Identifier: GPL-2.0
> +/*
> + * This file contains the logic to work with MPEG Program-Specific Information.
> + * These are defined both in ISO/IEC 13818-1 (systems) and ETSI EN 300 468.
> + * PSI is carried in the form of table structures, and although each table might
> + * technically be broken into one or more sections, we do not do this here,
> + * hence 'table' and 'section' are interchangeable for us.
> + *
> + * This code currently supports three tables: PAT, PMT and SDT. These are the
> + * bare minimum to get userspace to recognize our MPEG transport stream. It can
> + * be extended to support more PSI tables in the future.
> + *
> + * A note on endianness: MPEG layout is big-endian, therefore:
> + * - All fields spanning more than a byte must undergo 'cpu_to_beXX()' before
> + * serialization. These convertions are done in the *_write_into() functions.
> + *
> + * - All byte sized bitfields must have their ordering reversed if
> + *  __LITTLE_ENDIAN_BITFIELD is defined.
> + *
> + * Written by: Daniel W. S. Almeida <dwlsalmeida at gmail.com>
> + */
> +
> +#include <linux/kernel.h>
> +#include <linux/types.h>
> +#include <linux/slab.h>
> +#include <linux/crc32.h>
> +#include <linux/string.h>
> +#include <linux/printk.h>
> +
> +#include "vidtv_psi.h"
> +#include "vidtv_common.h"
> +#include "vidtv_ts.h"
> +
> +#define CRC_SIZE_IN_BYTES 4
> +
> +static u32 vidtv_psi_ts_psi_write_stuffing(void *to,
> +					   u32 len,
> +					   u32 offset,
> +					   u32 buf_sz)
> +{
> +	return vidtv_memset(to, 0xff, len, offset, buf_sz);
> +}

Instead, add thid define:

	#define TS_FILL_BYTE 0xff

And just use vidtv_memset() when needed.



> +
> +static u32
> +vidtv_psi_ts_psi_write_into(struct psi_write_args args)
> +{
> +	/*
> +	 * Packetize PSI sections into TS packets:
> +	 * push a TS header (4bytes) every 184 bytes
> +	 * manage the continuity_counter
> +	 * add stuffing after the CRC
> +	 */
> +
> +	u32  nbytes_past_boundary = (args.dest_offset % TS_PACKET_LEN);
> +	bool aligned              = nbytes_past_boundary == 0;
> +
> +	/*
> +	 * whether we need to fragment the data into multiple ts packets
> +	 * if we are aligned we need to spare one byte for the pointer_field
> +	 */
> +	bool split = (aligned) ?
> +		     args.len > TS_PAYLOAD_LEN - 1 :
> +		     nbytes_past_boundary + args.len > TS_PACKET_LEN;
> +
> +	/* how much we can write in this packet */
> +	u32 payload_write_len = (split) ?
> +				(aligned)     ? TS_PAYLOAD_LEN       :
> +				TS_PACKET_LEN - nbytes_past_boundary :
> +				args.len;
> +
> +	struct psi_write_args new_args = {0};
> +	struct vidtv_mpeg_ts ts_header = {0};
> +
> +	u32 nbytes = 0;  /* number of bytes written by this function */
> +	u32 temp   = 0;
> +
> +	/* Just a sanity check, should not really happen because we stuff
> +	 * the packet when we finish a section, i.e. when we write the crc at
> +	 * the end. But if this happens then we have messed up the logic
> +	 * somewhere.
> +	 */
> +	WARN_ON(args.new_psi_section && !aligned);

Please use a ratelimited printk instead (here and on all similar cases
at the TS generator).

Also, I think that, on such case, the driver should be filling the
remaining frame with pad bytes. E. g.:

	/* 
	 * Assuming that vidtv_memset() args from patch 06/11 were changed 
	 * according with this prototype: 
	 */
	size_t vidtv_memset(void *to, size_t to_offset, size_t to_size,
			    u8 c, size_t len);


	#define TS_FILL_BYTE 0xff

	if (args.new_psi_section && !aligned) {
		pr_warn_ratelimit("Warning: PSI not aligned. Re-aligning it\n");

		vidtv_memset(args.dest_buf,
			     args.dest_offset + nbytes_past_boundary,
			     args.dest_buf_sz,
			     TS_FILL_BYTE,	
			     TS_PACKET_LEN - nbytes_past_boundary);
		args.dest_offset += TS_PACKET_LEN - nbytes_past_boundary;
		aligned = 1;
		nbytes_past_boundary = 0;
	}

> +
> +	if (aligned) {
> +		/* if at a packet boundary, write a new TS header */
> +		ts_header.sync_byte          = TS_SYNC_BYTE;
> +		ts_header.tei                = 0;
> +		ts_header.payload_start      = 1;
> +		ts_header.pid                = args.pid;
> +		ts_header.priority           = 0;
> +		ts_header.scrambling         = 0;
> +		ts_header.continuity_counter = *args.continuity_counter;
> +		ts_header.payload            = 1;
> +		/* no adaptation field */
> +		ts_header.adaptation_field = 0;
> +
> +		cpu_to_be16s(&ts_header.bitfield);
> +
> +		/* copy the header */
> +		nbytes += vidtv_memcpy(args.dest_buf +
> +				       args.dest_offset +
> +				       nbytes,
> +				       &ts_header,
> +				       sizeof(ts_header),
> +				       args.dest_offset + nbytes,
> +				       args.dest_buf_sz);
> +
> +		be16_to_cpus(&ts_header.bitfield);
> +
> +		/*
> +		 * increment the countinuity counter since we have started
> +		 * a new packet
> +		 */
> +		vidtv_ts_inc_cc(args.continuity_counter);
> +	}
> +
> +	if (args.new_psi_section) {
> +		/* write the pointer_field in the first byte of the payload
> */
> +		temp = vidtv_memset(args.dest_buf + args.dest_offset +
> nbytes,
> +				    0x0,
> +				    1,
> +				    args.dest_offset + nbytes,
> +				    args.dest_buf_sz);
> +		/* one byte was used by the pointer field*/
> +		nbytes += temp;
> +		if (payload_write_len == TS_PAYLOAD_LEN)
> +			payload_write_len -= temp;
> +	}
> +
> +	/* write as much of the payload as we possibly can */
> +	nbytes += vidtv_memcpy(args.dest_buf + args.dest_offset + nbytes,
> +			       args.from,
> +			       payload_write_len,
> +			       args.dest_offset + nbytes,
> +			       args.dest_buf_sz);
> +
> +	if (split) {
> +		/* 'payload_write_len' written from a total of 'len' requested*/
> +		args.len -= payload_write_len;
> +		/*
> +		 * recursively write the rest of the data until we do not
> +		 * need to split it anymore
> +		 */
> +		memcpy(&new_args, &args, sizeof(struct psi_write_args));
> +		new_args.from            = args.from + payload_write_len;
> +		new_args.dest_offset     = args.dest_offset + nbytes;
> +		new_args.new_psi_section = false;
> +
> +		nbytes += vidtv_psi_ts_psi_write_into(new_args);
> +	}
> +
> +	/*
> +	 * as the CRC is last in the section, stuff the rest of the
> +	 * packet if there is any remaining space in there
> +	 */
> +	if (args.is_crc)
> +		nbytes += vidtv_psi_ts_psi_write_stuffing(args.dest_buf +
> +							  args.dest_offset +
> +							  nbytes,
> +							  TS_PACKET_LEN -
> +							  payload_write_len,
> +							  args.dest_offset +
> +							  nbytes,
> +							  args.dest_buf_sz);
> +
> +	return nbytes;
> +}
> +
> +static u32 table_section_crc32_write_into(struct crc32_write_args args)
> +{
> +	/* the CRC is the last entry in the section */
> +	u32 nbytes = 0;
> +	u32 crc;
> +	struct psi_write_args psi_args = {0};
> +
> +	crc = crc32_be(0, args.dest_buf, args.dest_offset);
> +
> +	psi_args.dest_buf           = args.dest_buf;
> +	psi_args.from               = &crc;
> +	psi_args.len                = CRC_SIZE_IN_BYTES;
> +	psi_args.dest_offset        = args.dest_offset;
> +	psi_args.pid                = args.pid;
> +	psi_args.new_psi_section    = false;
> +	psi_args.continuity_counter = args.continuity_counter;
> +	psi_args.is_crc             = true;
> +	psi_args.dest_buf_sz        = args.dest_buf_sz;
> +
> +	cpu_to_be32s(&crc);
> +	nbytes += vidtv_psi_ts_psi_write_into(psi_args);
> +	be32_to_cpus(&crc);
> +
> +	return nbytes;
> +}
> +
> +struct vidtv_psi_desc *vidtv_psi_desc_init(struct vidtv_psi_desc *head,
> +					   u8 type,
> +					   u8 length)
> +{
> +	struct vidtv_psi_desc *desc;
> +
> +	/* alloc enough memory for the flexible array too */
> +	desc = kzalloc(sizeof(*desc) + length, GFP_KERNEL);
> +
> +	desc->type   = type;
> +	desc->length = length;
> +
> +	if (head) {
> +		while (head->next)
> +			head = head->next;
> +
> +		head->next = desc;
> +	}
> +
> +	return desc;
> +}
> +
> +void vidtv_psi_desc_destroy(struct vidtv_psi_desc *desc)
> +{
> +	struct vidtv_psi_desc *curr = desc;
> +	struct vidtv_psi_desc *tmp  = NULL;
> +
> +	while (curr) {
> +		tmp  = curr;
> +		curr = curr->next;
> +		kfree(tmp);
> +	}
> +}
> +
> +static u32
> +vidtv_psi_desc_comp_len(struct vidtv_psi_desc *desc)
> +{
> +	u32 length = 0;
> +
> +	if (!desc)
> +		return 0;
> +
> +	while (desc) {
> +		length += desc->length;
> +		desc    = desc->next;
> +	}
> +
> +	return length;
> +}
> +
> +void vidtv_psi_desc_assign(struct vidtv_psi_desc **to,
> +			   struct vidtv_psi_desc *desc)
> +{
> +	/*
> +	 * Caller must recompute the section length afterwards at some point
> +	 * This function transfers ownedship of desc.
> +	 * Start by cleaning the old data
> +	 */
> +	if (*to)
> +		vidtv_psi_desc_destroy(*to);
> +
> +	*to = desc;  /* reassign pointer */
> +}
> +
> +static void vidtv_psi_desc_to_be(struct vidtv_psi_desc *desc)
> +{
> +	/*
> +	 * Convert descriptor endianness to big-endian on a field-by-field
> basis
> +	 * where applicable
> +	 */
> +
> +	switch (desc->type) {
> +	/* nothing do do */
> +	case SERVICE_DESCRIPTOR:
> +		break;
> +	case REGISTRATION_DESCRIPTOR:
> +		cpu_to_be32s(&((struct vidtv_psi_desc_registration *)
> +			     desc)->format_identifier);
> +		pr_alert("%s: descriptor type %d found\n",
> +			 __func__,
> +			 desc->type);
> +		pr_alert("%s: change 'additional_info' endianness before
> calling\n",
> +			 __func__);

The above pr_alert() calls sound weird. Why are you unconditionally
calling it (and still doing the BE conversion) for all
REGISTRATION_DESCRIPTOR types?

> +		break;
> +	default:
> +		pr_err("%s: descriptor type %d not implemented, skipping\n",
> +		       __func__,
> +		       desc->type);

please use pr_*_ratelimited() everywhere, as otherwise the driver
could cause Kernel to be too slow if something bad happens.

> +		break;
> +	}
> +}
> +
> +static void vidtv_psi_desc_to_cpu(struct vidtv_psi_desc *desc)
> +{
> +	/*
> +	 * Convert descriptor endianness to native on a field-by-field basis
> +	 * where applicable
> +	 */
> +
> +	switch (desc->type) {
> +	/* nothing do do */
> +	case SERVICE_DESCRIPTOR:
> +		break;
> +	case REGISTRATION_DESCRIPTOR:
> +		be32_to_cpus(&((struct vidtv_psi_desc_registration *)
> +			     desc)->format_identifier);
> +		pr_alert("%s: descriptor type %d found\n",
> +			 __func__,
> +			 desc->type);
> +		pr_alert("%s: change 'additional_info' endianness before
> calling\n",
> +			 __func__);

The above pr_alert() calls sound weird.

> +		break;
> +	default:
> +		pr_err("%s: descriptor type %d not implemented, skipping\n",
> +		       __func__,
> +		       desc->type);
> +		break;
> +	}
> +}
> +
> +static u32 vidtv_psi_desc_write_into(struct desc_write_args args)
> +{
> +	/* the number of bytes written by this function */
> +	u32    nbytes                  = 0;
> +	struct psi_write_args psi_args = {0};
> +
> +	psi_args.dest_buf = args.dest_buf;
> +	psi_args.from     = args.desc;
> +
> +	psi_args.len      = sizeof_field(struct vidtv_psi_desc, type) +
> +			    sizeof_field(struct vidtv_psi_desc, length);
> +
> +	psi_args.dest_offset        = args.dest_offset;
> +	psi_args.pid                = args.pid;
> +	psi_args.new_psi_section    = false;
> +	psi_args.continuity_counter = args.continuity_counter;
> +	psi_args.is_crc             = false;
> +	psi_args.dest_buf_sz        = args.dest_buf_sz;
> +
> +	vidtv_psi_desc_to_be(args.desc);
> +
> +	nbytes += vidtv_psi_ts_psi_write_into(psi_args);
> +
> +	/* move 'from' pointer to point to u8 data[] */
> +	psi_args.from = args.desc +
> +			sizeof_field(struct vidtv_psi_desc, type) +
> +			sizeof_field(struct vidtv_psi_desc, length) +
> +			sizeof(struct vidtv_psi_desc *);
> +
> +	psi_args.len         = args.desc->length;
> +	psi_args.dest_offset = args.dest_offset + nbytes;
> +
> +	nbytes += vidtv_psi_ts_psi_write_into(psi_args);
> +
> +	vidtv_psi_desc_to_cpu(args.desc);
> +
> +	return nbytes;
> +}
> +
> +static u32
> +vidtv_psi_table_header_write_into(struct header_write_args args)
> +{
> +	/* the number of bytes written by this function */
> +	u32    nbytes                  = 0;
> +	struct psi_write_args psi_args = {0};
> +
> +	psi_args.dest_buf           = args.dest_buf;
> +	psi_args.from               = args.h;
> +	psi_args.len                = sizeof(struct vidtv_psi_table_header);
> +	psi_args.dest_offset        = args.dest_offset;
> +	psi_args.pid                = args.pid;
> +	psi_args.new_psi_section    = true;
> +	psi_args.continuity_counter = args.continuity_counter;
> +	psi_args.is_crc             = false;
> +	psi_args.dest_buf_sz        = args.dest_buf_sz;
> +
> +	cpu_to_be16s(&args.h->bitfield);
> +	cpu_to_be16s(&args.h->id);
> +
> +	nbytes += vidtv_psi_ts_psi_write_into(psi_args);
> +
> +	be16_to_cpus(&args.h->bitfield);
> +	be16_to_cpus(&args.h->id);
> +
> +	return nbytes;
> +}
> +
> +void
> +vidtv_psi_pat_table_comp_sec_len(struct vidtv_psi_table_pat *pat)
> +{
> +	/* see ISO/IEC 13818-1 : 2000 p.43 */
> +	u16 length = 0;
> +	u32 i;
> +
> +	/* from immediately after 'section_length' until
> 'last_section_number'*/
> +	length += PAT_LEN_UNTIL_LAST_SECTION_NUMBER;
> +
> +	/* do not count the pointer */
> +	for (i = 0; i < pat->programs; ++i)
> +		length += sizeof(struct vidtv_psi_table_pat_program) -
> +			  sizeof(struct vidtv_psi_table_pat_program *);
> +
> +	length += CRC_SIZE_IN_BYTES;
> +
> +	WARN_ON(length > MAX_SECTION_LEN);

Please get rid of all WARN_ON, in favor of pr_*_ratelimited() calls.

> +	pat->header.section_length = length;
> +}
> +
> +void vidtv_psi_pmt_table_comp_sec_len(struct vidtv_psi_table_pmt *pmt)
> +{
> +	/* see ISO/IEC 13818-1 : 2000 p.46 */
> +	u16    length                        = 0;
> +	struct vidtv_psi_table_pmt_stream *s = pmt->stream;
> +
> +	/* from immediately after 'section_length' until
> 'program_info_length'*/
> +	length += PMT_LEN_UNTIL_PROGRAM_INFO_LENGTH;
> +
> +	pmt->desc_length  = vidtv_psi_desc_comp_len(pmt->descriptor);
> +	length           += pmt->desc_length;
> +
> +	while (s) {
> +		/* skip both pointers at the end */
> +		length += sizeof(struct vidtv_psi_table_pmt_stream) -
> +			  sizeof(struct vidtv_psi_desc *) -
> +			  sizeof(struct vidtv_psi_table_pmt_stream *);
> +
> +		s->desc_length  = vidtv_psi_desc_comp_len(s->descriptor);
> +		length         += s->desc_length;
> +
> +		s = s->next;
> +	}
> +
> +	length += CRC_SIZE_IN_BYTES;
> +
> +	WARN_ON(length > MAX_SECTION_LEN);
> +	pmt->header.section_length = length;
> +}
> +
> +void vidtv_psi_sdt_table_comp_sec_len(struct vidtv_psi_table_sdt *sdt)
> +{
> +	/* see ETSI EN 300 468 V 1.10.1 p.24 */
> +	u16    length                         = 0;
> +	struct vidtv_psi_table_sdt_service *s = sdt->service;
> +
> +	/*
> +	 * from immediately after 'section_length' until
> +	 * 'reserved_for_future_use'
> +	 */
> +	length += SDT_LEN_UNTIL_RESERVED_FOR_FUTURE_USE;
> +
> +	while (s) {
> +		/* skip both pointers at the end */
> +		length += sizeof(struct vidtv_psi_table_sdt_service) -
> +			  sizeof(struct vidtv_psi_desc *) -
> +			  sizeof(struct vidtv_psi_table_sdt_service *);
> +
> +		s->desc_length  = vidtv_psi_desc_comp_len(s->descriptor);
> +		length         += s->desc_length;
> +
> +		s = s->next;
> +	}
> +
> +	length += CRC_SIZE_IN_BYTES;
> +
> +	WARN_ON(length > MAX_SECTION_LEN);
> +	sdt->header.section_length = length;
> +}
> +
> +struct vidtv_psi_table_pat_program*
> +vidtv_psi_pat_program_init(struct vidtv_psi_table_pat_program *head,
> +			   u16 service_id,
> +			   u16 pid)
> +{
> +	/*
> +	 * if 'head' is attached to a table, caller should recompute
> +	 * the section length afterwards at some point
> +	 */
> +	struct vidtv_psi_table_pat_program *program;
> +
> +	program = kzalloc(sizeof(*program), GFP_KERNEL);
> +
> +	program->service_id = service_id;
> +	/* pid for the PMT section in the TS */
> +	program->pid        = pid;
> +	program->next       = NULL;
> +	program->reserved   = 0x7;
> +
> +	if (head) {
> +		while (head->next)
> +			head = head->next;
> +
> +		head->next = program;
> +	}
> +
> +	return program;
> +}
> +
> +void
> +vidtv_psi_pat_program_destroy(struct vidtv_psi_table_pat_program *p)
> +{
> +	struct vidtv_psi_table_pat_program *curr = p;
> +	struct vidtv_psi_table_pat_program *tmp  = NULL;
> +
> +	while (curr) {
> +		tmp  = curr;
> +		curr = curr->next;
> +		kfree(tmp);
> +	}
> +}
> +
> +void
> +vidtv_psi_pat_program_assign(struct vidtv_psi_table_pat *pat,
> +			     struct vidtv_psi_table_pat_program *p)
> +{
> +	/* This function transfers ownership of p to the table */
> +
> +	u16    program_count                        = 0;
> +	struct vidtv_psi_table_pat_program *program = p;
> +	struct vidtv_psi_table_pat_program *temp    = pat->program;
> +
> +	while (program) {
> +		++program_count;
> +		program = program->next;
> +	}
> +
> +	pat->programs = program_count;
> +	pat->program  = p;
> +
> +	/* Recompute section length */
> +	vidtv_psi_pat_table_comp_sec_len(pat);
> +
> +	/* reassign if the new size is too big */
> +	if (pat->header.section_length > MAX_SECTION_LEN)
> +		vidtv_psi_pat_program_assign(pat, temp);
> +	else
> +		vidtv_psi_pat_program_destroy(temp);
> +}
> +
> +void vidtv_psi_pat_table_init(struct vidtv_psi_table_pat *pat,
> +			      bool update_version_num,
> +			      u16 transport_stream_id)
> +{
> +	static u8 pat_version;
> +
> +	pat->header.table_id = 0x0;
> +	pat->header.syntax   = 0x1;
> +	pat->header.zero     = 0x0;
> +	pat->header.one      = 0x03;
> +
> +	/* transport stream ID, at will */
> +	pat->header.id           = transport_stream_id;
> +	pat->header.current_next = 0x1;
> +
> +	/* ETSI 300 468: indicates changes in the TS described by this
> table*/
> +	if (update_version_num)
> +		++pat_version;
> +
> +	pat->header.version = pat_version;
> +
> +	pat->header.one2         = 0x03;
> +	pat->header.section_id   = 0x0;
> +	pat->header.last_section = 0x0;
> +
> +	pat->programs = 0;
> +
> +	vidtv_psi_pat_table_comp_sec_len(pat);
> +}
> +
> +u32 vidtv_psi_pat_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_pat *pat,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter)
> +{
> +	/* the number of bytes written by this function */
> +	u32   nbytes      = 0;
> +	const u16 pat_pid = VIDTV_PAT_PID;
> +
> +	struct vidtv_psi_table_pat_program *p = pat->program;
> +	struct header_write_args h_args       = {0};
> +	struct psi_write_args args            = {0};
> +	struct crc32_write_args c_args        = {0};
> +
> +	vidtv_psi_pat_table_comp_sec_len(pat);
> +
> +	h_args.dest_buf           = buf;
> +	h_args.dest_offset        = offset;
> +	h_args.h                  = &pat->header;
> +	h_args.pid                = pat_pid;
> +	h_args.continuity_counter = continuity_counter;
> +	h_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += vidtv_psi_table_header_write_into(h_args);
> +
> +	/* note that the field 'u16 programs' is not really part of the PAT
> */ +
> +	args.dest_buf           = buf;
> +	args.pid                = pat_pid;
> +	args.new_psi_section    = false;
> +	args.continuity_counter = continuity_counter;
> +	args.is_crc             = false;
> +	args.dest_buf_sz        = buf_sz;
> +
> +	while (p) {
> +		/* copy the PAT programs */
> +		cpu_to_be16s(&p->service_id);
> +		cpu_to_be16s(&p->bitfield);
> +
> +		args.from = p;
> +		/* skip the pointer */
> +		args.len = sizeof(*p) -
> +			   sizeof(struct vidtv_psi_table_pat_program *);
> +		args.dest_offset = offset + nbytes;
> +
> +		nbytes += vidtv_psi_ts_psi_write_into(args);
> +
> +		be16_to_cpus(&p->service_id);
> +		be16_to_cpus(&p->bitfield);
> +
> +		p = p->next;
> +	}
> +
> +	c_args.dest_buf           = buf;
> +	c_args.dest_offset        = offset + nbytes;
> +	c_args.pid                = pat_pid;
> +	c_args.continuity_counter = continuity_counter;
> +	c_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += table_section_crc32_write_into(c_args);
> +
> +	return nbytes;
> +}
> +
> +void
> +vidtv_psi_pat_table_destroy(struct vidtv_psi_table_pat *p)
> +{
> +	vidtv_psi_pat_program_destroy(p->program);
> +}
> +
> +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)
> +{
> +	struct vidtv_psi_table_pmt_stream *stream;
> +
> +	stream = kzalloc(sizeof(*stream), GFP_KERNEL);
> +
> +	stream->type           = stream_type;
> +	stream->elementary_pid = es_pid;
> +	stream->reserved       = 0x07;
> +
> +	stream->desc_length = vidtv_psi_desc_comp_len(stream->descriptor);
> +
> +	stream->zero      = 0x0;
> +	stream->reserved2 = 0x0f;
> +
> +	if (head) {
> +		while (head->next)
> +			head = head->next;
> +
> +		head->next = stream;
> +	}
> +
> +	return stream;
> +}
> +
> +void vidtv_psi_pmt_stream_destroy(struct vidtv_psi_table_pmt_stream *s)
> +{
> +	struct vidtv_psi_table_pmt_stream *curr_stream = s;
> +	struct vidtv_psi_table_pmt_stream *tmp_stream  = NULL;
> +
> +	while (curr_stream) {
> +		tmp_stream  = curr_stream;
> +		curr_stream = curr_stream->next;
> +		kfree(tmp_stream);
> +	}
> +}
> +
> +void vidtv_psi_pmt_stream_assign(struct vidtv_psi_table_pmt *pmt,
> +				 struct vidtv_psi_table_pmt_stream *s)
> +{
> +	/* This function transfers ownership of s to the table */
> +	struct vidtv_psi_table_pmt_stream *stream = s;
> +	struct vidtv_psi_desc *desc               = s->descriptor;
> +	struct vidtv_psi_table_pmt_stream *temp   = pmt->stream;
> +
> +	while (stream)
> +		stream = stream->next;
> +
> +	while (desc)
> +		desc = desc->next;
> +
> +	pmt->stream = s;
> +	/* Recompute section length */
> +	vidtv_psi_pmt_table_comp_sec_len(pmt);
> +
> +	/* reassign if the new size is too big */
> +	if (pmt->header.section_length > MAX_SECTION_LEN)
> +		vidtv_psi_pmt_stream_assign(pmt, temp);
> +	else
> +		vidtv_psi_pmt_stream_destroy(temp);
> +}
> +
> +u16 vidtv_psi_pmt_get_pid(struct vidtv_psi_table_pmt *section,
> +			  struct vidtv_psi_table_pat *pat)
> +{
> +	struct vidtv_psi_table_pat_program *program = pat->program;
> +
> +	/*
> +	 * service_id is the same as program_number in the
> +	 * corresponding program_map_section
> +	 * see ETSI EN 300 468 v1.15.1 p. 24
> +	 */
> +	while (program) {
> +		if (program->service_id == section->header.id)
> +			return pat->program->pid;
> +
> +		program = program->next;
> +	}
> +
> +	return TS_LAST_VALID_PID + 1; /* not found */
> +}
> +
> +void vidtv_psi_pmt_table_init(struct vidtv_psi_table_pmt *pmt,
> +			      bool update_version_num,
> +			      u16 program_number,
> +			      u16 pcr_pid)
> +{
> +	static u8 pmt_version;
> +
> +	pmt->header.table_id = 0x2;
> +	pmt->header.syntax   = 0x1;
> +	pmt->header.zero     = 0x0;
> +	pmt->header.one      = 0x3;
> +
> +	pmt->header.id           = program_number;
> +	pmt->header.current_next = 0x1;
> +
> +	/* ETSI 300 468: indicates changes in the TS described by this
> table*/
> +	if (update_version_num)
> +		++pmt_version;
> +
> +	pmt->header.version = pmt_version;
> +
> +	pmt->header.one2         = 0x3;
> +	pmt->header.section_id   = 0;
> +	pmt->header.last_section = 0;
> +
> +	pmt->pcr_pid   = (pcr_pid) ? pcr_pid : 0x1fff;
> +	pmt->reserved2 = 0x03;
> +
> +	pmt->reserved3 = 0x0f;
> +	pmt->zero3     = 0x0;
> +
> +	pmt->desc_length = vidtv_psi_desc_comp_len(pmt->descriptor);
> +
> +	vidtv_psi_pmt_table_comp_sec_len(pmt);
> +}
> +
> +u32 vidtv_psi_pmt_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_pmt *pmt,
> +			     u16 pid,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter)
> +{
> +	/* the number of bytes written by this function */
> +	u32    nbytes                             = 0;
> +	struct vidtv_psi_desc *table_descriptor   = pmt->descriptor;
> +	struct vidtv_psi_table_pmt_stream *stream = pmt->stream;
> +	struct vidtv_psi_desc *stream_descriptor  = (stream) ?
> +						    pmt->stream->descriptor :
> +						    NULL;
> +
> +	struct header_write_args h_args = {0};
> +	struct psi_write_args args      = {0};
> +	struct desc_write_args d_args   = {0};
> +	struct crc32_write_args c_args  = {0};
> +
> +	vidtv_psi_pmt_table_comp_sec_len(pmt);
> +
> +	h_args.dest_buf           = buf;
> +	h_args.dest_offset        = offset;
> +	h_args.h                  = &pmt->header;
> +	h_args.pid                = pid;
> +	h_args.continuity_counter = continuity_counter;
> +	h_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += vidtv_psi_table_header_write_into(h_args);
> +
> +	/* write the two bitfields */
> +	cpu_to_be16s(&pmt->bitfield);
> +	cpu_to_be16s(&pmt->bitfield2);
> +
> +	args.dest_buf = buf;
> +	args.from     = pmt + sizeof(struct vidtv_psi_table_header);
> +	args.len      = sizeof_field(struct vidtv_psi_table_pmt, bitfield) +
> +			sizeof_field(struct vidtv_psi_table_pmt, bitfield2);
> +	args.dest_offset        = offset + nbytes;
> +	args.pid                = pid;
> +	args.new_psi_section    = false;
> +	args.continuity_counter = continuity_counter;
> +	args.is_crc             = false;
> +	args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += vidtv_psi_ts_psi_write_into(args);
> +
> +	be16_to_cpus(&pmt->bitfield);
> +	be16_to_cpus(&pmt->bitfield2);
> +
> +	while (table_descriptor) {
> +		/* write the descriptors, if any */
> +		d_args.dest_buf           = buf;
> +		d_args.dest_offset        = offset + nbytes;
> +		d_args.desc               = table_descriptor;
> +		d_args.pid                = pid;
> +		d_args.continuity_counter = continuity_counter;
> +		d_args.dest_buf_sz        = buf_sz;
> +
> +		vidtv_psi_desc_to_be(d_args.desc);
> +		nbytes += vidtv_psi_desc_write_into(d_args);
> +		vidtv_psi_desc_to_cpu(d_args.desc);
> +
> +		table_descriptor = table_descriptor->next;
> +	}
> +
> +	while (stream) {
> +		/* write the streams, if any */
> +		args.from = stream;
> +		args.len  = sizeof_field(struct vidtv_psi_table_pmt_stream,
> +					 type) +
> +			    sizeof_field(struct vidtv_psi_table_pmt_stream,
> +					 bitfield) +
> +			    sizeof_field(struct vidtv_psi_table_pmt_stream,
> +					 bitfield2);
> +		args.dest_offset = offset + nbytes;
> +
> +		cpu_to_be16s(&stream->bitfield);
> +		cpu_to_be16s(&stream->bitfield2);
> +
> +		nbytes += vidtv_psi_ts_psi_write_into(args);
> +
> +		be16_to_cpus(&stream->bitfield);
> +		be16_to_cpus(&stream->bitfield2);
> +
> +		while (stream_descriptor) {
> +			/* write the stream descriptors, if any */
> +			d_args.desc        = stream_descriptor;
> +			d_args.dest_offset = offset + nbytes;
> +
> +			vidtv_psi_desc_to_be(d_args.desc);
> +
> +			nbytes += vidtv_psi_desc_write_into(d_args);
> +
> +			vidtv_psi_desc_to_cpu(d_args.desc);
> +
> +			stream_descriptor = stream_descriptor->next;
> +		}
> +
> +		stream = stream->next;
> +	}
> +
> +	c_args.dest_buf           = buf;
> +	c_args.dest_offset        = offset + nbytes;
> +	c_args.pid                = pid;
> +	c_args.continuity_counter = continuity_counter;
> +	c_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += table_section_crc32_write_into(c_args);
> +
> +	return nbytes;
> +}
> +
> +void vidtv_psi_pmt_table_destroy(struct vidtv_psi_table_pmt *pmt)
> +{
> +	struct vidtv_psi_desc *curr_desc = pmt->descriptor;
> +	struct vidtv_psi_desc *tmp_desc  = NULL;
> +
> +	while (curr_desc) {
> +		tmp_desc  = curr_desc;
> +		curr_desc = curr_desc->next;
> +		vidtv_psi_desc_destroy(tmp_desc);
> +		kfree(tmp_desc);
> +	}
> +
> +	vidtv_psi_pmt_stream_destroy(pmt->stream);
> +}
> +
> +void vidtv_psi_sdt_table_init(struct vidtv_psi_table_sdt *sdt,
> +			      bool update_version_num,
> +			      u16 transport_stream_id)
> +{
> +	static u8 sdt_version;
> +
> +	sdt->header.table_id = 0x42;
> +
> +	sdt->header.one  = 0x3;
> +	sdt->header.zero = 0x1;
> +	/*
> +	 * The PAT, PMT, and CAT all set this to 0.
> +	 * Other tables set this to 1.
> +	 */
> +	sdt->header.syntax = 0x1;
> +
> +	/*
> +	 * This is a 16-bit field which serves as a label for identification
> +	 * of the TS, about which the SDT informs, from any other multiplex
> +	 * within the delivery system.
> +	 */
> +	sdt->header.id           = transport_stream_id;
> +	sdt->header.current_next = 0x1;
> +
> +	/* ETSI 300 468: indicates changes in the TS described by this
> table*/
> +	if (update_version_num)
> +		++sdt_version;
> +
> +	sdt->header.version = sdt_version;
> +
> +	sdt->header.one2         = 0x3;
> +	sdt->header.section_id   = 0;
> +	sdt->header.last_section = 0;
> +
> +	sdt->network_id = transport_stream_id;
> +	sdt->reserved   = 0xff;
> +
> +	vidtv_psi_sdt_table_comp_sec_len(sdt);
> +}
> +
> +u32 vidtv_psi_sdt_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_sdt *sdt,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter)
> +{
> +	u32 nbytes  = 0;              /* the number of bytes written */
> +	u16 sdt_pid = VIDTV_SDT_PID;  /* see ETSI EN 300 468 v1.15.1 p. 11 */
> +
> +	struct vidtv_psi_table_sdt_service *service = sdt->service;
> +	struct vidtv_psi_desc *service_desc = (sdt->service) ?
> +					      sdt->service->descriptor :
> +					      NULL;
> +
> +	struct header_write_args h_args = {0};
> +	struct psi_write_args args      = {0};
> +	struct desc_write_args d_args   = {0};
> +	struct crc32_write_args c_args  = {0};
> +
> +	vidtv_psi_sdt_table_comp_sec_len(sdt);
> +
> +	h_args.dest_buf           = buf;
> +	h_args.dest_offset        = offset;
> +	h_args.h                  = &sdt->header;
> +	h_args.pid                = sdt_pid;
> +	h_args.continuity_counter = continuity_counter;
> +	h_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += vidtv_psi_table_header_write_into(h_args);
> +
> +	args.dest_buf = buf;
> +	args.from     = sdt + sizeof(struct vidtv_psi_table_header);
> +
> +	args.len      = sizeof_field(struct vidtv_psi_table_sdt, network_id)
> +
> +			sizeof_field(struct vidtv_psi_table_sdt, reserved);
> +
> +	args.dest_offset        = offset + nbytes;
> +	args.pid                = sdt_pid;
> +	args.new_psi_section    = false;
> +	args.continuity_counter = continuity_counter;
> +	args.is_crc             = false;
> +	args.dest_buf_sz        = buf_sz;
> +
> +	/* copy u16 network_id + u8 reserved)*/
> +	cpu_to_be16s(&sdt->network_id);
> +
> +	nbytes += vidtv_psi_ts_psi_write_into(args);
> +
> +	be16_to_cpus(&sdt->network_id);
> +
> +	while (service) {
> +		/* copy the services, if any */
> +		args.from = service;
> +		/* skip both pointers at the end */
> +		args.len = sizeof(struct vidtv_psi_table_sdt_service) -
> +			   sizeof(struct vidtv_psi_desc *) -
> +			   sizeof(struct vidtv_psi_table_sdt_service *);
> +		args.dest_offset = offset + nbytes;
> +
> +		cpu_to_be16s(&service->service_id);
> +		cpu_to_be16s(&service->bitfield);
> +
> +		nbytes += vidtv_psi_ts_psi_write_into(args);
> +
> +		be16_to_cpus(&service->service_id);
> +		be16_to_cpus(&service->bitfield);
> +
> +		while (service_desc) {
> +			/* copy the service descriptors, if any */
> +			d_args.dest_buf           = buf;
> +			d_args.dest_offset        = offset + nbytes;
> +			d_args.desc               = service_desc;
> +			d_args.pid                = sdt_pid;
> +			d_args.continuity_counter = continuity_counter;
> +			d_args.dest_buf_sz        = buf_sz;
> +
> +			vidtv_psi_desc_to_be(d_args.desc);
> +
> +			nbytes += vidtv_psi_desc_write_into(d_args);
> +
> +			vidtv_psi_desc_to_cpu(d_args.desc);
> +
> +			service_desc = service_desc->next;
> +		}
> +
> +		service = service->next;
> +	}
> +
> +	c_args.dest_buf           = buf;
> +	c_args.dest_offset        = offset + nbytes;
> +	c_args.pid                = sdt_pid;
> +	c_args.continuity_counter = continuity_counter;
> +	c_args.dest_buf_sz        = buf_sz;
> +
> +	nbytes += table_section_crc32_write_into(c_args);
> +
> +	return nbytes;
> +}
> +
> +void vidtv_psi_sdt_table_destroy(struct vidtv_psi_table_sdt *sdt)
> +{
> +	struct vidtv_psi_table_sdt_service *curr_service = sdt->service;
> +	struct vidtv_psi_table_sdt_service *tmp_service  = NULL;
> +	struct vidtv_psi_desc *curr_desc = (sdt->service) ?
> +					   sdt->service->descriptor : NULL;
> +	struct vidtv_psi_desc *tmp_desc = NULL;
> +
> +	while (curr_service) {
> +		curr_desc = curr_service->descriptor;
> +
> +		while (curr_desc) {
> +			/* clear all descriptors for the service */
> +			tmp_desc  = curr_desc;
> +			curr_desc = curr_desc->next;
> +			vidtv_psi_desc_destroy(tmp_desc);
> +			kfree(tmp_desc);
> +		}
> +
> +		/* then clear the current service */
> +		tmp_service  = curr_service;
> +		curr_service = curr_service->next;
> +		kfree(tmp_service);
> +	}
> +}
> +
> +struct vidtv_psi_table_sdt_service
> +*vidtv_psi_sdt_service_init(struct vidtv_psi_table_sdt_service *head,
> +			    u16 service_id)
> +{
> +	/*
> +	 * if 'head' is attached to a table, caller should recompute
> +	 * the section length afterwards at some point
> +	 */
> +	struct vidtv_psi_table_sdt_service *service;
> +
> +	service = kzalloc(sizeof(*service), GFP_KERNEL);
> +
> +	/*
> +	 * ETSI 300 468: this is a 16bit field which serves as a label to
> +	 * identify this service from any other service within the TS.
> +	 * The service id is the same as the program number in the
> +	 * corresponding program_map_section
> +	 */
> +	service->service_id            = service_id;
> +	service->EIT_schedule          = 0x0;         /* TODO */
> +	service->EIT_present_following = 0x0;         /* TODO */
> +	service->reserved              = 0x3f;        /* all bits on */
> +	service->free_CA_mode          = 0x0;         /* not scrambled */
> +	service->running_status        = RUNNING;
> +
> +	if (head) {
> +		while (head->next)
> +			head = head->next;
> +
> +		head->next = service;
> +	}
> +
> +	return service;
> +}
> +
> +void
> +vidtv_psi_sdt_service_destroy(struct vidtv_psi_table_sdt_service *service)
> +{
> +	struct vidtv_psi_table_sdt_service *curr = service;
> +	struct vidtv_psi_table_sdt_service *tmp  = NULL;
> +
> +	while (curr) {
> +		tmp  = curr;
> +		curr = curr->next;
> +		kfree(tmp);
> +	}
> +}
> +
> +void
> +vidtv_psi_sdt_service_assign(struct vidtv_psi_table_sdt *sdt,
> +			     struct vidtv_psi_table_sdt_service *service)
> +{
> +	struct vidtv_psi_table_sdt_service *temp = sdt->service;
> +
> +	sdt->service = service;
> +
> +	/* recompute section length */
> +	vidtv_psi_sdt_table_comp_sec_len(sdt);
> +
> +	/* reassign if the new size is too big */
> +	if (sdt->header.section_length > MAX_SECTION_LEN)
> +		vidtv_psi_sdt_service_assign(sdt, temp);
> +	else
> +		vidtv_psi_sdt_service_destroy(temp);
> +}
> +
> +void
> +vidtv_psi_pmt_create_sec_for_each_pat_entry(struct vidtv_psi_table_pat *pat,
> +					    struct vidtv_psi_table_pmt sec[])
> +
> +{
> +	/*
> +	 * PMTs contain information about programs. For each program,
> +	 * there is one PMT section. This function will create a section
> +	 * for each program found in the PAT
> +	 */
> +	struct vidtv_psi_table_pat_program *program = pat->program;
> +	u32    i                                    = 0;
> +
> +	while (program) {
> +		vidtv_psi_pmt_table_init(&sec[i],
> +					 false,
> +					 sec[i].header.id,
> +					 0);
> +
> +		++i;
> +		program = program->next;
> +	}
> +}
> diff --git a/drivers/media/test-drivers/vidtv/vidtv_psi.h
> b/drivers/media/test-drivers/vidtv/vidtv_psi.h new file mode 100644
> index 0000000000000..c5c8c143f0e4a
> --- /dev/null
> +++ b/drivers/media/test-drivers/vidtv/vidtv_psi.h
> @@ -0,0 +1,357 @@
> +/* SPDX-License-Identifier: GPL-2.0 */
> +/*
> + * This file contains the logic to work with MPEG Program-Specific
> Information.
> + * These are defined both in ISO/IEC 13818-1 (systems) and ETSI EN 300 468.
> + * PSI is carried in the form of table structures, and although each table
> might
> + * technically be broken into one or more sections, we do not do this here,
> + * hence 'table' and 'section' are interchangeable for us.
> + *
> + * This code currently supports three tables: PAT, PMT and SDT. These are the
> + * bare minimum to get userspace to recognize our MPEG transport stream. It
> can
> + * be extended to support more PSI tables in the future.
> + *
> + * A note on endianness: MPEG layout is big-endian, therefore:
> + * - All fields spanning more than a byte must undergo 'cpu_to_beXX()' before
> + * serialization. These convertions are done in the *_write_into() functions.
> + *
> + * - All byte sized bitfields must have their ordering reversed if
> + *  __LITTLE_ENDIAN_BITFIELD is defined.
> + *
> + * Written by: Daniel W. S. Almeida <dwlsalmeida at gmail.com>
> + */
> +
> +#ifndef VIDTV_PSI_H
> +#define VIDTV_PSI_H
> +
> +#include <linux/types.h>
> +#include <asm/byteorder.h>
> +
> +/*
> + * all section lengths start immediately after the 'section_length' field
> + * see ISO/IEC 13818-1 : 2000 and ETSI EN 300 468 V 1.10.1 for
> + * reference
> + */
> +#define PAT_LEN_UNTIL_LAST_SECTION_NUMBER 5
> +#define PMT_LEN_UNTIL_PROGRAM_INFO_LENGTH 9
> +#define SDT_LEN_UNTIL_RESERVED_FOR_FUTURE_USE 8
> +#define MAX_SECTION_LEN 1021
> +#define VIDTV_PAT_PID 0
> +#define VIDTV_SDT_PID 0x0011
> +
> +enum vidtv_psi_descriptors {
> +	REGISTRATION_DESCRIPTOR	= 0x05,
> +	SERVICE_DESCRIPTOR = 0x48,
> +};
> +
> +enum vidtv_psi_stream_types {
> +	/* see ISO/IEC 13818-1 2000 p. 48 */
> +	STREAM_PRIVATE_DATA = 0x06,
> +};
> +
> +struct vidtv_psi_desc {
> +	u8 type;
> +	u8 length;
> +	struct vidtv_psi_desc *next;
> +	u8 data[];
> +} __packed;
> +
> +struct vidtv_psi_desc_service {
> +	u8 type;
> +	u8 length;
> +	struct vidtv_psi_desc *next;
> +
> +	u8 service_type;
> +	char *name;
> +	char *name_emph;
> +	char *provider;
> +	char *provider_emph;
> +} __packed;
> +
> +struct vidtv_psi_desc_registration {
> +	u8 type;
> +	u8 length;
> +	struct vidtv_psi_desc *next;
> +
> +	/*
> +	 * The format_identifier is a 32-bit value obtained from a
> Registration
> +	 * Authority as designated by ISO/IEC JTC 1/SC 29.
> +	 */
> +	u32 format_identifier;
> +	/*
> +	 * The meaning of additional_identification_info bytes, if any, are
> +	 * defined by the assignee of that format_identifier, and once
> defined
> +	 * they shall not change.
> +	 */
> +	u8 additional_identification_info[];
> +} __packed;
> +
> +struct vidtv_psi_table_header {
> +	u8  table_id;
> +	union {
> +		u16 bitfield;
> +		struct {
> +			u8  syntax:1;
> +			u8  zero:1;
> +			u8  one:2;
> +			u16 section_length:12;
> +		} __packed;
> +	} __packed;
> +
> +	u16 id;			/* TS ID */
> +#if defined(__LITTLE_ENDIAN_BITFIELD)
> +	u8  current_next:1;
> +	u8  version:5;
> +	u8  one2:2;
> +#elif defined(__BIG_ENDIAN_BITFIELD)
> +	u8  one2:2;
> +	u8  version:5;
> +	u8  current_next:1;
> +#else
> +#error  "Please fix <asm/byteorder.h>"

I would change the message. I mean, if something has changed there,
and, for example, one of the defines was renamed, then the vidtv code
is the one that would require changes, not asm/byteorder.

> +#endif
> +	u8  section_id;		/* section_number */
> +	u8  last_section;		/* last_section_number */
> +} __packed;
> +
> +struct vidtv_psi_table_pat_program {
> +	u16 service_id;
> +	union {
> +		u16 bitfield;
> +		struct {
> +			u8  reserved:3;
> +			u16 pid:13;
> +		} __packed;
> +	} __packed;
> +	struct vidtv_psi_table_pat_program *next;
> +} __packed;
> +
> +struct vidtv_psi_table_pat {
> +	struct vidtv_psi_table_header header;
> +	u16 programs;
> +	struct vidtv_psi_table_pat_program *program;
> +} __packed;
> +
> +struct vidtv_psi_table_sdt_service {
> +	u16 service_id;
> +#if defined(__LITTLE_ENDIAN_BITFIELD)
> +	u8 EIT_present_following:1;
> +	u8 EIT_schedule:1;
> +	u8 reserved:6;
> +#elif defined(__BIG_ENDIAN_BITFIELD)
> +	u8 reserved:6;
> +	u8 EIT_schedule:1;
> +	u8 EIT_present_following:1;
> +#else
> +#error  "Please fix <asm/byteorder.h>"
> +#endif
> +	union {
> +		u16 bitfield;
> +		struct {
> +			u16 running_status:3;
> +			u16 free_CA_mode:1;
> +			u16 desc_length:12;
> +		} __packed;
> +	} __packed;
> +	struct vidtv_psi_desc *descriptor;
> +	struct vidtv_psi_table_sdt_service *next;
> +} __packed;
> +
> +struct vidtv_psi_table_sdt {
> +	struct vidtv_psi_table_header header;
> +	u16 network_id; /* original_network_id */
> +	u8  reserved;
> +	struct vidtv_psi_table_sdt_service *service;
> +} __packed;
> +
> +enum service_running_status {
> +	RUNNING,
> +};
> +
> +enum service_type {
> +	/* see ETSI EN 300 468 v1.15.1 p. 77 */
> +	DIGITAL_TELEVISION_SERVICE = 0x1,
> +};
> +
> +struct vidtv_psi_table_pmt_stream {
> +	u8 type;
> +	union {
> +		u16 bitfield;
> +		struct {
> +			u16 reserved:3;
> +			u16 elementary_pid:13;
> +		} __packed;
> +	} __packed;
> +	union {
> +		u16 bitfield2;
> +		struct {
> +			u16 reserved2:4;
> +			u16 zero:2;
> +			u16 desc_length:10;
> +		} __packed;
> +	} __packed;
> +	struct vidtv_psi_desc *descriptor;
> +	struct vidtv_psi_table_pmt_stream *next;
> +} __packed;
> +
> +struct vidtv_psi_table_pmt {
> +	struct vidtv_psi_table_header header;
> +	union {
> +		u16 bitfield;
> +		struct {
> +			u16 reserved2:3;
> +			u16 pcr_pid:13;
> +		} __packed;
> +	} __packed;
> +
> +	union {
> +		u16 bitfield2;
> +		struct {
> +			u16 reserved3:4;
> +			u16 zero3:2;
> +			u16 desc_length:10; /* program_info_length */
> +		} __packed;
> +	} __packed;
> +	struct vidtv_psi_desc *descriptor;
> +	struct vidtv_psi_table_pmt_stream *stream;
> +} __packed;
> +
> +struct psi_write_args {
> +	void *dest_buf; /* the buffer to write into */
> +	void *from;
> +	size_t len; /* how much to write */
> +	u32 dest_offset; /* where to start writing in the buffer */
> +	u16 pid; /* TS packet ID */
> +	bool new_psi_section; /* set when starting a table section */
> +	u8 *continuity_counter; /* TS: incremented on every new packet */
> +	bool is_crc; /* set when writing the CRC at the end */
> +	u32 dest_buf_sz; /* protect against overflow if not zero */
> +};
> +
> +struct desc_write_args {
> +	void *dest_buf;
> +	u32 dest_offset;
> +	struct vidtv_psi_desc *desc;
> +	u16 pid;
> +	u8 *continuity_counter;
> +	u32 dest_buf_sz;
> +};
> +
> +struct crc32_write_args {
> +	void *dest_buf;
> +	u32 dest_offset;
> +	u16 pid;
> +	u8 *continuity_counter;
> +	u32 dest_buf_sz;
> +};
> +
> +struct header_write_args {
> +	void *dest_buf;
> +	u32 dest_offset;
> +	struct vidtv_psi_table_header *h;
> +	u16 pid;
> +	u8 *continuity_counter;
> +	u32 dest_buf_sz;
> +};
> +
> +struct vidtv_psi_desc *vidtv_psi_desc_init(struct vidtv_psi_desc *head,
> +					   u8 type,
> +					   u8 length);
> +
> +void vidtv_psi_pat_table_init(struct vidtv_psi_table_pat *pat,
> +			      bool update_version_num,
> +			      u16 transport_stream_id);
> +
> +struct vidtv_psi_table_pat_program*
> +vidtv_psi_pat_program_init(struct vidtv_psi_table_pat_program *head,
> +			   u16 service_id,
> +			   u16 pid);
> +
> +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);
> +
> +void vidtv_psi_pmt_table_init(struct vidtv_psi_table_pmt *pmt,
> +			      bool update_version_num,
> +			      u16 program_number,
> +			      u16 pcr_pid);
> +
> +void
> +vidtv_psi_sdt_table_init(struct vidtv_psi_table_sdt *sdt,
> +			 bool update_version_num,
> +			 u16 transport_stream_id);
> +
> +struct vidtv_psi_table_sdt_service*
> +vidtv_psi_sdt_service_init(struct vidtv_psi_table_sdt_service *head,
> +			   u16 service_id);
> +
> +void
> +vidtv_psi_desc_destroy(struct vidtv_psi_desc *desc);
> +
> +void
> +vidtv_psi_pat_program_destroy(struct vidtv_psi_table_pat_program *p);
> +
> +void
> +vidtv_psi_pat_table_destroy(struct vidtv_psi_table_pat *p);
> +
> +void
> +vidtv_psi_pmt_stream_destroy(struct vidtv_psi_table_pmt_stream *s);
> +
> +void
> +vidtv_psi_pmt_table_destroy(struct vidtv_psi_table_pmt *pmt);
> +
> +void
> +vidtv_psi_sdt_table_destroy(struct vidtv_psi_table_sdt *sdt);
> +
> +void
> +vidtv_psi_sdt_service_destroy(struct vidtv_psi_table_sdt_service *service);
> +
> +void
> +vidtv_psi_desc_destroy(struct vidtv_psi_desc *desc);
> +
> +void
> +vidtv_psi_pat_program_destroy(struct vidtv_psi_table_pat_program *p);
> +
> +void
> +vidtv_psi_sdt_service_assign(struct vidtv_psi_table_sdt *sdt,
> +			     struct vidtv_psi_table_sdt_service *service);
> +
> +void vidtv_psi_desc_assign(struct vidtv_psi_desc **to,
> +			   struct vidtv_psi_desc *desc);
> +
> +void vidtv_psi_pat_program_assign(struct vidtv_psi_table_pat *pat,
> +				  struct vidtv_psi_table_pat_program *p);
> +
> +void vidtv_psi_pmt_stream_assign(struct vidtv_psi_table_pmt *pmt,
> +				 struct vidtv_psi_table_pmt_stream *s);
> +void
> +vidtv_psi_pmt_create_sec_for_each_pat_entry(struct vidtv_psi_table_pat *pat,
> +					    struct vidtv_psi_table_pmt *sec);
> +
> +u16 vidtv_psi_pmt_get_pid(struct vidtv_psi_table_pmt *section,
> +			  struct vidtv_psi_table_pat *pat);
> +
> +void vidtv_psi_pat_table_comp_sec_len(struct vidtv_psi_table_pat *pat);
> +void vidtv_psi_pmt_table_comp_sec_len(struct vidtv_psi_table_pmt *pmt);
> +void vidtv_psi_sdt_table_comp_sec_len(struct vidtv_psi_table_sdt *sdt);
> +
> +u32 vidtv_psi_pat_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_pat *pat,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter);
> +
> +u32 vidtv_psi_sdt_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_sdt *sdt,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter);
> +
> +u32 vidtv_psi_pmt_write_into(char *buf,
> +			     u32 offset,
> +			     struct vidtv_psi_table_pmt *pmt,
> +			     u16 pid,
> +			     u32 buf_sz,
> +			     u8 *continuity_counter);
> +
> +#endif // VIDTV_PSI_H



Thanks,
Mauro



More information about the Linux-kernel-mentees mailing list