Quelle  vidtv_common.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * The Virtual DVB test driver serves as a reference DVB driver and helps
 * validate the existing APIs in the media subsystem. It can also aid
 * developers working on userspace applications.
 *
 * Copyright (C) 2020 Daniel W. S. Almeida
 */
#define pr_fmt(fmt) KBUILD_MODNAME ":%s, %d: " fmt, __func__, __LINE__

#include <linux/printk.h>
#include <linux/ratelimit.h>
#include <linux/string.h>
#include <linux/types.h>

#include "vidtv_common.h"

/**
 * vidtv_memcpy() - wrapper routine to be used by MPEG-TS
 * generator, in order to avoid going past the
 * output buffer.
 * @to: Starting element to where a MPEG-TS packet will
 * be copied.
 * @to_offset: Starting position of the @to buffer to be filled.
 * @to_size: Size of the @to buffer.
 * @from: Starting element of the buffer to be copied.
 * @len: Number of elements to be copy from @from buffer
 * into @to+ @to_offset buffer.
 *
 * Note:
 * Real digital TV demod drivers should not have memcpy
 * wrappers. We use it here because emulating MPEG-TS
 * generation at kernelspace requires some extra care.
 *
 * Return:
 * Returns the number of bytes written
 */
u32 vidtv_memcpy(void *to,
   size_t to_offset,
   size_t to_size,
   const void *from,
   size_t len)
{
 if (unlikely(to_offset + len > to_size)) {
  pr_err_ratelimited("overflow detected, skipping. Try increasing the buffer size. Needed %zu, had %zu\n",
       to_offset + len,
       to_size);
  return 0;
 }

 memcpy(to + to_offset, from, len);
 return len;
}

/**
 * vidtv_memset() - wrapper routine to be used by MPEG-TS
 * generator, in order to avoid going past the
 * output buffer.
 * @to: Starting element to set
 * @to_offset: Starting position of the @to buffer to be filled.
 * @to_size: Size of the @to buffer.
 * @c: The value to set the memory to.
 * @len: Number of elements to be copy from @from buffer
 * into @to+ @to_offset buffer.
 *
 * Note:
 * Real digital TV demod drivers should not have memset
 * wrappers. We use it here because emulating MPEG-TS
 * generation at kernelspace requires some extra care.
 *
 * Return:
 * Returns the number of bytes written
 */
u32 vidtv_memset(void *to,
   size_t to_offset,
   size_t to_size,
   const int c,
   size_t len)
{
 if (unlikely(to_offset + len > to_size)) {
  pr_err_ratelimited("overflow detected, skipping. Try increasing the buffer size. Needed %zu, had %zu\n",
       to_offset + len,
       to_size);
  return 0;
 }

 memset(to + to_offset, c, len);
 return len;
}