git.sesse.net Git - vlc/blob - modules/control/netsync.c

   1 /*****************************************************************************\r
   2  * netsync.c: synchronisation between several network clients.\r
   3  *****************************************************************************\r
   4  * Copyright (C) 2004 VideoLAN\r
   5  * $Id$\r
   6  *\r
   7  * Authors: Gildas Bazin <gbazin@videolan.org>\r
   8  *\r
   9  * This program is free software; you can redistribute it and/or modify\r
  10  * it under the terms of the GNU General Public License as published by\r
  11  * the Free Software Foundation; either version 2 of the License, or\r
  12  * (at your option) any later version.\r
  13  *\r
  14  * This program is distributed in the hope that it will be useful,\r
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  17  * GNU General Public License for more details.\r
  18  *\r
  19  * You should have received a copy of the GNU General Public License\r
  20  * along with this program; if not, write to the Free Software\r
  21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.\r
  22  *****************************************************************************/\r
  23 \r
  24 /*****************************************************************************\r
  25  * Preamble\r
  26  *****************************************************************************/\r
  27 #include <stdlib.h>\r
  28 #include <vlc/vlc.h>\r
  29 #include <vlc/intf.h>\r
  30 \r
  31 #ifdef HAVE_UNISTD_H\r
  32 #    include <unistd.h>\r
  33 #endif\r
  34 \r
  35 #ifdef HAVE_SYS_TIME_H\r
  36 #    include <sys/time.h>\r
  37 #endif\r
  38 #include <sys/types.h>\r
  39 \r
  40 #ifdef WIN32\r
  41 #   include <winsock2.h>\r
  42 #   include <ws2tcpip.h>\r
  43 #   ifndef IN_MULTICAST\r
  44 #       define IN_MULTICAST(a) IN_CLASSD(a)\r
  45 #   endif\r
  46 #else\r
  47 #   include <sys/socket.h>\r
  48 #   include <netinet/in.h>\r
  49 #   if HAVE_ARPA_INET_H\r
  50 #      include <arpa/inet.h>\r
  51 #   elif defined( SYS_BEOS )\r
  52 #      include <net/netdb.h>\r
  53 #   endif\r
  54 #endif\r
  55 \r
  56 #ifdef UNDER_CE\r
  57 #   define close(a) CloseHandle(a);\r
  58 #elif defined( WIN32 )\r
  59 #   define close(a) closesocket(a);\r
  60 #endif\r
  61 \r
  62 #include "network.h"\r
  63 #include "input_ext-plugins.h"\r
  64 \r
  65 #define NETSYNC_PORT_MASTER 9875\r
  66 #define NETSYNC_PORT_SLAVE  9876\r
  67 \r
  68 /* Needed for Solaris */\r
  69 #ifndef INADDR_NONE\r
  70 #define INADDR_NONE 0xffffffff\r
  71 #endif\r
  72 \r
  73 /*****************************************************************************\r
  74  * Module descriptor\r
  75  *****************************************************************************/\r
  76 static int  Activate( vlc_object_t * );\r
  77 static void Close   ( vlc_object_t * );\r
  78 \r
  79 static mtime_t GetClockRef( intf_thread_t *, mtime_t );\r
  80 \r
  81 #define NETSYNC_TEXT N_( "Act as master for network synchronisation" )\r
  82 #define NETSYNC_LONGTEXT N_( "Allows you to specify if this client should " \\r
  83   "act as the master client for the network synchronisation." )\r
  84 \r
  85 #define MIP_TEXT N_( "Master client ip address" )\r
  86 #define MIP_LONGTEXT N_( "Allows you to specify the ip address of " \\r
  87   "the master client used for the network synchronisation." )\r
  88 \r
  89 vlc_module_begin();\r
  90     set_description( _("Network synchronisation") );\r
  91 \r
  92     add_bool( "netsync-master", 0, NULL,\r
  93               NETSYNC_TEXT, NETSYNC_LONGTEXT, VLC_TRUE );\r
  94     add_string( "netsync-master-ip", NULL, NULL, MIP_TEXT, MIP_LONGTEXT,\r
  95                 VLC_TRUE );\r
  96 \r
  97     set_capability( "interface", 0 );\r
  98     set_callbacks( Activate, Close );\r
  99 vlc_module_end();\r
 100 \r
 101 struct intf_sys_t\r
 102 {\r
 103     input_thread_t *p_input;\r
 104 };\r
 105 \r
 106 /*****************************************************************************\r
 107  * Local prototypes\r
 108  *****************************************************************************/\r
 109 static void Run( intf_thread_t *p_intf );\r
 110 \r
 111 /*****************************************************************************\r
 112  * Activate: initialize and create stuff\r
 113  *****************************************************************************/\r
 114 static int Activate( vlc_object_t *p_this )\r
 115 {\r
 116     intf_thread_t *p_intf = (intf_thread_t*)p_this;\r
 117 \r
 118     msg_Info( p_intf, "Using the netsync interface module..." );\r
 119 \r
 120     p_intf->p_sys = malloc( sizeof( intf_sys_t ) );\r
 121     if( !p_intf->p_sys )\r
 122     {\r
 123         msg_Err( p_intf, "no memory" );\r
 124         return VLC_ENOMEM;\r
 125     }\r
 126 \r
 127     p_intf->p_sys->p_input = NULL;\r
 128 \r
 129     p_intf->pf_run = Run;\r
 130     return VLC_SUCCESS;\r
 131 }\r
 132 \r
 133 /*****************************************************************************\r
 134  * Close: destroy interface\r
 135  *****************************************************************************/\r
 136 void Close( vlc_object_t *p_this )\r
 137 {\r
 138     intf_thread_t *p_intf = (intf_thread_t*)p_this;\r
 139 \r
 140     free( p_intf->p_sys );\r
 141 }\r
 142 \r
 143 /*****************************************************************************\r
 144  * Run: interface thread\r
 145  *****************************************************************************/\r
 146 static void Run( intf_thread_t *p_intf )\r
 147 {\r
 148 #define MAX_MSG_LENGTH (2 * sizeof(int64_t))\r
 149 \r
 150     vlc_bool_t b_master = config_GetInt( p_intf, "netsync-master" );\r
 151     char *psz_master = config_GetPsz( p_intf, "netsync-master-ip" );\r
 152     struct sockaddr_in master_addr;\r
 153     char p_data[MAX_MSG_LENGTH];\r
 154     int i_socket;\r
 155 \r
 156     if( !psz_master || inet_addr( psz_master ) == INADDR_NONE )\r
 157     {\r
 158         if( !b_master )\r
 159         {\r
 160             if( psz_master ) free( psz_master );\r
 161             msg_Err( p_intf, "invalid master address." );\r
 162             return;\r
 163         }\r
 164 \r
 165         if( !psz_master ) psz_master = strdup("");\r
 166     }\r
 167 \r
 168     memset( &master_addr, 0, sizeof( struct sockaddr_in ) );\r
 169     master_addr.sin_family = AF_INET;\r
 170     master_addr.sin_port = htons( (uint16_t)NETSYNC_PORT_MASTER );\r
 171     master_addr.sin_addr.s_addr = inet_addr( psz_master );\r
 172     free( psz_master );\r
 173 \r
 174     i_socket = net_OpenUDP( p_intf, NULL, b_master ? NETSYNC_PORT_MASTER :\r
 175                             NETSYNC_PORT_SLAVE, NULL, 0 );\r
 176     if( i_socket < 0 )\r
 177     {\r
 178         msg_Err( p_intf, "failed opening UDP socket." );\r
 179         return;\r
 180     }\r
 181 \r
 182     /* High priority thread */\r
 183     vlc_thread_set_priority( p_intf, VLC_THREAD_PRIORITY_INPUT );\r
 184 \r
 185     while( !p_intf->b_die )\r
 186     {\r
 187         struct timeval  timeout;\r
 188         fd_set fds_r;\r
 189 \r
 190         /* Update the input */\r
 191         if( p_intf->p_sys->p_input == NULL )\r
 192         {\r
 193             p_intf->p_sys->p_input =\r
 194                 (input_thread_t *)vlc_object_find( p_intf, VLC_OBJECT_INPUT,\r
 195                                                    FIND_ANYWHERE );\r
 196         }\r
 197         else if( p_intf->p_sys->p_input->b_dead )\r
 198         {\r
 199             vlc_object_release( p_intf->p_sys->p_input );\r
 200             p_intf->p_sys->p_input = NULL;\r
 201         }\r
 202 \r
 203         if( p_intf->p_sys->p_input == NULL )\r
 204         {\r
 205             /* Wait a bit */\r
 206             msleep( INTF_IDLE_SLEEP );\r
 207             continue;\r
 208         }\r
 209 \r
 210         /*\r
 211          * We now have an input\r
 212          */\r
 213 \r
 214         /* Initialize file descriptor set and timeout (0.5s) */\r
 215         FD_ZERO( &fds_r );\r
 216         FD_SET( i_socket, &fds_r );\r
 217         timeout.tv_sec = 0;\r
 218         timeout.tv_usec = 500000;\r
 219 \r
 220         if( b_master )\r
 221         {\r
 222             struct sockaddr_in from;\r
 223             mtime_t i_date, i_clockref, i_master_clockref;\r
 224             int i_struct_size, i_read, i_ret;\r
 225 \r
 226             /* Don't block */\r
 227             i_ret = select( i_socket + 1, &fds_r, 0, 0, &timeout );\r
 228             if( i_ret == 0 ) continue;\r
 229             if( i_ret < 0 )\r
 230             {\r
 231                 /* Wait a bit */\r
 232                 msleep( INTF_IDLE_SLEEP );\r
 233                 continue;\r
 234             }\r
 235 \r
 236             /* We received something */\r
 237             i_struct_size = sizeof(struct sockaddr_in);\r
 238             i_read = recvfrom( i_socket, p_data, MAX_MSG_LENGTH, 0,\r
 239                                (struct sockaddr*)&from, &i_struct_size );\r
 240 \r
 241             from.sin_port = htons( (uint16_t)NETSYNC_PORT_SLAVE );\r
 242 \r
 243             i_clockref = ntoh64(*(int64_t *)p_data);\r
 244 \r
 245             i_date = mdate();\r
 246             *(int64_t *)p_data = hton64( i_date );\r
 247 \r
 248             i_master_clockref = GetClockRef( p_intf, i_clockref );\r
 249             *(((int64_t *)p_data)+1) = hton64( i_master_clockref );\r
 250 \r
 251             /* Reply to the sender */\r
 252             sendto( i_socket, p_data, 2 * sizeof(int64_t), 0,\r
 253                     (struct sockaddr *)&from, sizeof(struct sockaddr_in) );\r
 254 \r
 255             msg_Dbg( p_intf, "Master clockref: "I64Fd" -> "I64Fd", from %s "\r
 256                      "(date: "I64Fd")", i_clockref, i_master_clockref, \r
 257                      inet_ntoa(from.sin_addr), i_date );\r
 258         }\r
 259         else\r
 260         {\r
 261             mtime_t i_send_date, i_receive_date, i_master_date, i_diff_date;\r
 262             mtime_t i_master_clockref, i_client_clockref, i_drift;\r
 263             mtime_t i_clockref = 0;\r
 264             int i_sent, i_read, i_ret;\r
 265 \r
 266             /* Send clock request to the master */\r
 267             *(int64_t *)p_data = hton64( i_clockref );\r
 268             i_send_date = mdate();\r
 269 \r
 270             i_sent = sendto( i_socket, p_data, sizeof(int64_t), 0,\r
 271                              (struct sockaddr *)&master_addr,\r
 272                              sizeof(struct sockaddr_in) );\r
 273             if( i_sent <= 0 )\r
 274             {\r
 275                 /* Wait a bit */\r
 276                 msleep( INTF_IDLE_SLEEP );\r
 277                 continue;\r
 278             }\r
 279 \r
 280             /* Don't block */\r
 281             i_ret = select(i_socket + 1, &fds_r, 0, 0, &timeout);\r
 282             if( i_ret == 0 ) continue;\r
 283             if( i_ret < 0 )\r
 284             {\r
 285                 /* Wait a bit */\r
 286                 msleep( INTF_IDLE_SLEEP );\r
 287                 continue;\r
 288             }\r
 289 \r
 290             i_receive_date = mdate();\r
 291 \r
 292             i_read = recv( i_socket, p_data, MAX_MSG_LENGTH, 0 );\r
 293             if( i_read <= 0 )\r
 294             {\r
 295                 /* Wait a bit */\r
 296                 msleep( INTF_IDLE_SLEEP );\r
 297                 continue;\r
 298             }\r
 299 \r
 300             i_master_date = ntoh64(*(int64_t *)p_data);\r
 301             i_master_clockref = ntoh64(*(((int64_t *)p_data)+1));\r
 302 \r
 303             i_diff_date = i_receive_date -\r
 304                           ((i_receive_date - i_send_date) / 2 + i_master_date);\r
 305 \r
 306             i_client_clockref = i_drift = 0;\r
 307             if( p_intf->p_sys->p_input && i_master_clockref )\r
 308             {\r
 309                 i_client_clockref = GetClockRef( p_intf, i_clockref );\r
 310                 i_drift = i_client_clockref - i_master_clockref - i_diff_date;\r
 311 \r
 312                 /* Update our clock to match the master's one */\r
 313                 if( i_client_clockref )\r
 314                     p_intf->p_sys->p_input->i_pts_delay -= i_drift;\r
 315             }\r
 316 \r
 317             msg_Dbg( p_intf, "Slave clockref: "I64Fd" -> "I64Fd" -> "I64Fd", "\r
 318                      "clock diff: "I64Fd" drift: "I64Fd,\r
 319                      i_clockref, i_master_clockref, \r
 320                      i_client_clockref, i_diff_date, i_drift );\r
 321 \r
 322             /* Wait a bit */\r
 323             msleep( INTF_IDLE_SLEEP );\r
 324         }\r
 325     }\r
 326 \r
 327     if( p_intf->p_sys->p_input ) vlc_object_release( p_intf->p_sys->p_input );\r
 328     net_Close( i_socket );\r
 329 }\r
 330 \r
 331 static mtime_t GetClockRef( intf_thread_t *p_intf, mtime_t i_pts )\r
 332 {\r
 333     input_thread_t *p_input = p_intf->p_sys->p_input;\r
 334     pgrm_descriptor_t *p_pgrm;\r
 335 \r
 336     if( !p_input ) return 0;\r
 337 \r
 338     p_pgrm = p_input->stream.p_selected_program;\r
 339     if( p_pgrm ) return input_ClockGetTS( p_input, p_pgrm, i_pts );\r
 340 \r
 341     return 0;\r
 342 }\r