tstools: check for wrap arounds in length calculation
[openblackhole/openblackhole-enigma2.git] / lib / dvb / tstools.cpp
index 02a0b8a..bec17a5 100644 (file)
@@ -45,6 +45,7 @@ void eDVBTSTools::setSearchRange(int maxrange)
        m_maxrange = maxrange;
 }
 
+       /* getPTS extracts a pts value from any PID at a given offset. */
 int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
 {
        if (m_fd < 0)
@@ -136,7 +137,14 @@ void eDVBTSTools::calcEnd()
        if (m_fd < 0)   
                return;
        
-       m_offset_end = lseek(m_fd, 0, SEEK_END);
+       off_t end = lseek(m_fd, 0, SEEK_END);
+       
+       if (abs(end - m_offset_end) > 1*1024*1024)
+       {
+               m_offset_end = end;
+               m_end_valid = 0;
+               eDebug("file size changed, recalc length");
+       }
        
        int maxiter = 10;
        
@@ -161,6 +169,9 @@ int eDVBTSTools::calcLen(pts_t &len)
        if (!(m_begin_valid && m_end_valid))
                return -1;
        len = m_pts_end - m_pts_begin;
+               /* wrap around? */
+       if (len < 0)
+               len += 0x200000000LL;
        return 0;
 }
 
@@ -171,6 +182,10 @@ int eDVBTSTools::calcBitrate()
                return -1;
 
        pts_t len_in_pts = m_pts_end - m_pts_begin;
+
+               /* wrap around? */
+       if (len_in_pts < 0)
+               len_in_pts += 0x200000000LL;
        off_t len_in_bytes = m_offset_end - m_offset_begin;
        
        if (!len_in_pts)