From: Jaroslav Kysela <perex@perex.cz>
Date: Tue, 27 Sep 2016 19:23:51 +0000 (+0200)
Subject: epgdb: improve statistics logs
X-Git-Tag: v4.2.1~299
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=0c79a1db0c73ce97dcc73035d25aec6582a51d4d;p=thirdparty%2Ftvheadend.git

epgdb: improve statistics logs
---

diff --git a/src/epgdb.c b/src/epgdb.c
index cedfbf666..2e5f52612 100644
--- a/src/epgdb.c
+++ b/src/epgdb.c
@@ -344,13 +344,15 @@ void epg_init ( void )
 #if ENABLE_ZLIB
   if (remain > 12 && memcmp(rp, "\xff\xffGZIP00", 8) == 0) {
     uint32_t orig = (rp[8] << 24) | (rp[9] << 16) | (rp[10] << 8) | rp[11];
-    tvhinfo(LS_EPGDB, "gzip format detected, inflating (ratio %.1f%%)",
-           (float)((remain * 100.0) / orig));
+    tvhinfo(LS_EPGDB, "gzip format detected, inflating (ratio %.1f%% deflated size %zd)",
+           (float)((remain * 100.0) / orig), remain);
     rp = zlib_mem = tvh_gzip_inflate(rp + 12, remain - 12, orig);
     remain = rp ? orig : 0;
   }
 #endif
 
+  tvhinfo(LS_EPGDB, "parsing %zd bytes", remain);
+
   /* Process */
   memset(&stats, 0, sizeof(stats));
   while ( remain > 4 ) {
@@ -402,7 +404,7 @@ void epg_init ( void )
   /* Stats */
   tvhinfo(LS_EPGDB, "loaded v%d", ver);
   tvhinfo(LS_EPGDB, "  config     %d", stats.config.total);
-  tvhinfo(LS_EPGDB, "  channels   %d", stats.channels.total);
+  //tvhinfo(LS_EPGDB, "  channels   %d", stats.channels.total);
   tvhinfo(LS_EPGDB, "  brands     %d", stats.brands.total);
   tvhinfo(LS_EPGDB, "  seasons    %d", stats.seasons.total);
   tvhinfo(LS_EPGDB, "  episodes   %d", stats.episodes.total);
@@ -468,7 +470,7 @@ static int _epg_write_sect ( sbuf_t *sb, const char *sect )
 static void epg_save_tsk_callback ( void *p, int dearmed )
 {
   sbuf_t *sb = p;
-  size_t size = sb->sb_ptr;
+  size_t size = sb->sb_ptr, orig;
   int fd, r;
 
   tvhinfo(LS_EPGDB, "save start");
@@ -476,15 +478,15 @@ static void epg_save_tsk_callback ( void *p, int dearmed )
   if (fd >= 0) {
 #if ENABLE_ZLIB
     if (config.epg_compress) {
-      r = tvh_gzip_deflate_fd_header(fd, sb->sb_data, sb->sb_ptr, 3) < 0;
+      r = tvh_gzip_deflate_fd_header(fd, sb->sb_data, size, &orig, 3) < 0;
    } else
 #endif
-      r = tvh_write(fd, sb->sb_data, sb->sb_ptr);
+      r = tvh_write(fd, sb->sb_data, orig = size);
     close(fd);
     if (r)
-      tvherror(LS_EPGDB, "write error (size %zd)", size);
+      tvherror(LS_EPGDB, "write error (size %zd)", orig);
     else
-      tvhinfo(LS_EPGDB, "stored (size %zd)", size);
+      tvhinfo(LS_EPGDB, "stored (size %zd)", orig);
   } else
     tvherror(LS_EPGDB, "unable to open epgdb file");
   sbuf_free(sb);
diff --git a/src/settings.c b/src/settings.c
index 812997097..3718c488f 100644
--- a/src/settings.c
+++ b/src/settings.c
@@ -184,7 +184,7 @@ hts_settings_save(htsmsg_t *record, const char *pathfmt, ...)
     size_t msglen;
     r = htsmsg_binary_serialize(record, &msgdata, &msglen, 2*1024*1024);
     if (!r && msglen >= 4) {
-      r = tvh_gzip_deflate_fd_header(fd, msgdata + 4, msglen - 4, 3);
+      r = tvh_gzip_deflate_fd_header(fd, msgdata + 4, msglen - 4, NULL, 3);
       if (r)
         ok = 0;
     } else {
diff --git a/src/tvheadend.h b/src/tvheadend.h
index 534c017bf..c34e1f213 100644
--- a/src/tvheadend.h
+++ b/src/tvheadend.h
@@ -821,7 +821,7 @@ char *regexp_escape ( const char *str );
 uint8_t *tvh_gzip_inflate ( const uint8_t *data, size_t size, size_t orig );
 uint8_t *tvh_gzip_deflate ( const uint8_t *data, size_t orig, size_t *size );
 int      tvh_gzip_deflate_fd ( int fd, const uint8_t *data, size_t orig, size_t *size, int speed );
-int      tvh_gzip_deflate_fd_header ( int fd, const uint8_t *data, size_t orig, int speed );
+int      tvh_gzip_deflate_fd_header ( int fd, const uint8_t *data, size_t orig, size_t *size, int speed );
 #endif
 
 /* URL decoding */
diff --git a/src/zlib.c b/src/zlib.c
index 3dcc994d5..2df7b09cb 100644
--- a/src/zlib.c
+++ b/src/zlib.c
@@ -153,12 +153,14 @@ int tvh_gzip_deflate_fd ( int fd, const uint8_t *data, size_t orig, size_t *size
   return r;
 }
 
-int tvh_gzip_deflate_fd_header ( int fd, const uint8_t *data, size_t orig, int speed )
+int tvh_gzip_deflate_fd_header ( int fd, const uint8_t *data, size_t orig,
+                                 size_t *deflated_size, int speed )
 {
   uint8_t data2[4];
-  size_t size;
+  size_t size = 0;
   int r;
 
+  if (deflated_size) *deflated_size = 0;
   r = tvh_write(fd, "\xff\xffGZIP00\x00\x00\x00\x00", 12);
   if (r)
     return 1;
@@ -166,10 +168,11 @@ int tvh_gzip_deflate_fd_header ( int fd, const uint8_t *data, size_t orig, int s
     r = tvh_gzip_deflate_fd(fd, data, orig, &size, 3) < 0;
     if (r || size > UINT_MAX)
       return 1;
-    r = lseek(fd, 8, SEEK_SET) != (off_t)8;
-    if (r)
-      return 1;
   }
+  r = lseek(fd, 8, SEEK_SET) != (off_t)8;
+  if (r)
+    return 1;
+  if (deflated_size) *deflated_size = size + 12;
   data2[0] = (orig >> 24) & 0xff;
   data2[1] = (orig >> 16) & 0xff;
   data2[2] = (orig >> 8) & 0xff;