* Win32 network support by Boris Dor�s <babal@via.ecp.fr>.

[vlc] / plugins / mpeg / input_ts.h

diff --git a/plugins/mpeg/input_ts.h b/plugins/mpeg/input_ts.h
index 97dc361d3278f31665bcc4e77dc94fc414caedde..20f4183d790873e2cce069b3e697de4245bcc2dc 100644 (file)
--- a/plugins/mpeg/input_ts.h
+++ b/plugins/mpeg/input_ts.h
@@ -2,9 +2,10 @@
  * input_ts.h: structures of the input not exported to other modules
  *****************************************************************************
  * Copyright (C) 1999, 2000 VideoLAN
- * $Id: input_ts.h,v 1.8 2001/06/20 07:43:48 sam Exp $
+ * $Id: input_ts.h,v 1.9 2001/06/21 07:22:03 sam Exp $
  *
  * Authors: Henri Fallon <henri@via.ecp.fr>
+ *          Boris Dorès <babal@via.ecp.fr>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -21,15 +22,158 @@
  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  *****************************************************************************/
 
-
 #define NB_DATA 16384 
 #define NB_PES  8192
 
-/* Will be used whne NetworkOpen is ready */
-typedef struct thread_ts_data_s { 
-    
-    // FILE *                  stream;
+#define BUFFER_SIZE (7 * TS_PACKET_SIZE)
+
+/*****************************************************************************
+ * thread_ts_data_t: private input data
+ *****************************************************************************/
+typedef struct thread_ts_data_s
+{ 
+    /* The file descriptor we select() on */
     fd_set fds;
     
+#if defined( WIN32 )
+    char p_buffer[ BUFFER_SIZE ];      /* temporary buffer for readv_network */
+    int  i_length;                               /* length of the UDP packet */
+    int  i_offset;           /* number of bytes already read from the buffer */
+#endif
+    
 } thread_ts_data_t;
 
+/*****************************************************************************
+ * readv_*: readv() replacements for iovec-impaired C libraries
+ *****************************************************************************/
+
+#if defined( WIN32 )
+static __inline__ int readv_file( int i_fd, struct iovec *p_iovec, int i_count )
+{
+    int i_index, i_len, i_total = 0;
+    u8 *p_base;
+
+    for( i_index = i_count; i_index; i_index-- )
+    {
+        register signed int i_bytes;
+
+        i_len  = p_iovec->iov_len;
+        p_base = p_iovec->iov_base;
+
+        /* Loop is unrolled one time to spare the (i_bytes <= 0) test */
+
+        if( i_len > 0 )
+        {
+            i_bytes = read( i_fd, p_base, i_len );
+
+            if( ( i_total == 0 ) && ( i_bytes < 0 ) )
+            {
+                return -1;
+            }
+
+            if( i_bytes <= 0 )
+            {
+                return i_total;
+            }
+
+            i_len -= i_bytes; i_total += i_bytes; p_base += i_bytes;
+
+            while( i_len > 0 )
+            {
+                i_bytes = read( i_fd, p_base, i_len );
+
+                if( i_bytes <= 0 )
+                {
+                    return i_total;
+                }
+
+                i_len -= i_bytes; i_total += i_bytes; p_base += i_bytes;
+            }
+        }
+
+        p_iovec++;
+    }
+
+    return i_total;
+}
+
+static __inline__ int read_network( int i_fd, char * p_base,
+                                    thread_ts_data_t *p_sys, int i_len )
+{
+    int i_bytes;
+
+    if( p_sys->i_offset >= p_sys->i_length )
+    {
+        p_sys->i_length = recv( i_fd, p_sys->p_buffer, BUFFER_SIZE, 0 );
+        if ( p_sys->i_length == SOCKET_ERROR )
+        {
+            return -1;
+        }
+        p_sys->i_offset = 0;
+    }
+
+    if( i_len <= p_sys->i_length - p_sys->i_offset )
+    {
+         i_bytes = i_len;
+    }
+    else
+    {
+         i_bytes = p_sys->i_length - p_sys->i_offset;
+    }
+
+    memcpy( p_base, p_sys->p_buffer + p_sys->i_offset, i_bytes );
+    p_sys->i_offset += i_bytes;
+
+    return i_bytes;
+}
+
+static __inline__ int readv_network( int i_fd, struct iovec *p_iovec,
+                                     int i_count, thread_ts_data_t *p_sys )
+{
+    int i_index, i_len, i_total = 0;
+    u8 *p_base;
+
+    for( i_index = i_count; i_index; i_index-- )
+    {
+        register signed int i_bytes;
+
+        i_len  = p_iovec->iov_len;
+        p_base = p_iovec->iov_base;
+
+        /* Loop is unrolled one time to spare the (i_bytes <= 0) test */
+        if( i_len > 0 )
+        {
+            i_bytes = read_network( i_fd, p_base, p_sys, i_len );
+
+            if( ( i_total == 0 ) && ( i_bytes < 0 ) )
+            {
+                return -1;
+            }
+
+            if( i_bytes <= 0 )
+            {
+                return i_total;
+            }
+
+            i_len -= i_bytes; i_total += i_bytes; p_base += i_bytes;
+
+            while( i_len > 0 )
+            {
+                i_bytes = read_network( i_fd, p_base, p_sys, i_len );
+
+                if( i_bytes <= 0 )
+                {
+                    return i_total;
+                }
+
+                i_len -= i_bytes; i_total += i_bytes; p_base += i_bytes;
+            }
+        }
+
+        p_iovec++;
+    }
+
+    return i_total;
+}
+#endif
+