V4L/DVB (13271): TS speed check. Logging transport stream speed in Kbits per second

[mchehab@redhat.com: add asm/div64.h and allocate vars only if dvb_demux_speedcheck specified]
Signed-off-by: Abylay Ospan <aospan@netup.ru>
Signed-off-by: Mauro Carvalho Chehab <mchehab@redhat.com>

2 files changed, 44 insertions, 0 deletions

diff --git a/drivers/media/dvb/dvb-core/dvb_demux.c b/drivers/media/dvb/dvb-core/dvb_demux.c
index 91c537bca8ad..b78cfb7d1897 100644
--- a/drivers/media/dvb/dvb-core/dvb_demux.c
+++ b/drivers/media/dvb/dvb-core/dvb_demux.c
@@ -30,6 +30,7 @@
 #include <linux/string.h>
 #include <linux/crc32.h>
 #include <asm/uaccess.h>
+#include <asm/div64.h>
 
 #include "dvb_demux.h"
 
@@ -44,6 +45,11 @@ module_param(dvb_demux_tscheck, int, 0644);
 MODULE_PARM_DESC(dvb_demux_tscheck,
                 "enable transport stream continuity and TEI check");
 
+static int dvb_demux_speedcheck;
+module_param(dvb_demux_speedcheck, int, 0644);
+MODULE_PARM_DESC(dvb_demux_speedcheck,
+                "enable transport stream speed check");
+
 #define dprintk_tscheck(x...) do {                              \
                 if (dvb_demux_tscheck && printk_ratelimit())    \
                         printk(x);                              \
@@ -387,6 +393,39 @@ static void dvb_dmx_swfilter_packet(struct dvb_demux *demux, const u8 *buf)
         u16 pid = ts_pid(buf);
         int dvr_done = 0;
 
+        if (dvb_demux_speedcheck) {
+                struct timespec cur_time, delta_time;
+                u64 speed_bytes, speed_timedelta;
+
+                demux->speed_pkts_cnt++;
+
+                /* show speed every SPEED_PKTS_INTERVAL packets */
+                if (!(demux->speed_pkts_cnt % SPEED_PKTS_INTERVAL)) {
+                        cur_time = current_kernel_time();
+
+                        if (demux->speed_last_time.tv_sec != 0 &&
+                                        demux->speed_last_time.tv_nsec != 0) {
+                                delta_time = timespec_sub(cur_time,
+                                                demux->speed_last_time);
+                                speed_bytes = (u64)demux->speed_pkts_cnt
+                                        * 188 * 8;
+                                /* convert to 1024 basis */
+                                speed_bytes = 1000 * div64_u64(speed_bytes,
+                                                1024);
+                                speed_timedelta =
+                                        (u64)timespec_to_ns(&delta_time);
+                                speed_timedelta = div64_u64(speed_timedelta,
+                                                1000000); /* nsec -> usec */
+                                printk(KERN_INFO "TS speed %llu Kbits/sec \n",
+                                                div64_u64(speed_bytes,
+                                                        speed_timedelta));
+                        };
+
+                        demux->speed_last_time = cur_time;
+                        demux->speed_pkts_cnt = 0;
+                };
+        };
+
         if (dvb_demux_tscheck) {
                 if (!demux->cnt_storage)
                         demux->cnt_storage = vmalloc(MAX_PID + 1);
diff --git a/drivers/media/dvb/dvb-core/dvb_demux.h b/drivers/media/dvb/dvb-core/dvb_demux.h
index 2fe05d03240d..a7d876fd02dd 100644
--- a/drivers/media/dvb/dvb-core/dvb_demux.h
+++ b/drivers/media/dvb/dvb-core/dvb_demux.h
@@ -44,6 +44,8 @@
 
 #define MAX_PID 0x1fff
 
+#define SPEED_PKTS_INTERVAL 50000
+
 struct dvb_demux_filter {
         struct dmx_section_filter filter;
         u8 maskandmode[DMX_MAX_FILTER_SIZE];
@@ -131,6 +133,9 @@ struct dvb_demux {
         spinlock_t lock;
 
         uint8_t *cnt_storage; /* for TS continuity check */
+
+        struct timespec speed_last_time; /* for TS speed check */
+        uint32_t speed_pkts_cnt; /* for TS speed check */
 };
 
 int dvb_dmx_init(struct dvb_demux *dvbdemux);