1 files changed, 15 insertions, 10 deletions

diff --git a/proxy.c b/proxy.c
index 4977b5c..d5cbab6 100644
--- a/proxy.c
+++ b/proxy.c
@@ -59,7 +59,7 @@ int      g_self_pipe[2];
 /* So after each bucket wait 1 / OT_BUCKET_COUNT intervals */
 #define OT_SYNC_SLEEP ( ( ( OT_SYNC_INTERVAL_MINUTES ) * 60 * 1000000 ) / ( OT_BUCKET_COUNT ) )
 
-enum { OT_SYNC_PEER };
+enum { OT_SYNC_PEER = 0xbeef };
 enum { FLAG_SERVERSOCKET = 1 };
 
 /* For incoming packets */
@@ -195,8 +195,8 @@ void free_peerlist( ot_peerlist *peer_list ) {
 static void livesync_handle_peersync( ssize_t datalen ) {
   int off = sizeof( g_tracker_id ) + sizeof( uint32_t );
 
-  /* Now basic sanity checks have been done on the live sync packet
-     We might add more testing and logging. */
+  fprintf( stderr, "." );
+
   while( off + (ssize_t)sizeof( ot_hash ) + (ssize_t)sizeof( ot_peer ) <= datalen ) {
     ot_peer *peer = (ot_peer*)(g_inbuffer + off + sizeof(ot_hash));
     ot_hash *hash = (ot_hash*)(g_inbuffer + off);
@@ -227,6 +227,7 @@ enum {
 };
 
 #define PROXYPEER_NEEDSCONNECT(flag)      ((flag)==FLAG_OUTGOING)
+#define PROXYPEER_ISCONNECTED(flag)       (((flag)&FLAG_MASK)==FLAG_CONNECTED)
 #define PROXYPEER_SETDISCONNECTED(flag)   (flag)=(((flag)&FLAG_OUTGOING)|FLAG_DISCONNECTED)
 #define PROXYPEER_SETCONNECTING(flag)     (flag)=(((flag)&FLAG_OUTGOING)|FLAG_CONNECTING)
 #define PROXYPEER_SETWAITTRACKERID(flag)  (flag)=(((flag)&FLAG_OUTGOING)|FLAG_WAITTRACKERID)
@@ -439,16 +440,13 @@ static void handle_write( int64 peersocket ) {
   case FLAG_CONNECTED:
     switch( iob_send( peersocket, &peer->outdata ) ) {
     case 0: /* all data sent */
-      fprintf( stderr, "EMPTY\n" );
       io_dontwantwrite( peersocket );
       break;
     case -3: /* an error occured */
-      fprintf( stderr, "ERROR\n" );
       io_close( peersocket );
       reset_info_block( peer );
       break;
     default: /* Normal operation or eagain */
-      fprintf( stderr, "EGAIN\n" );
       break;
     }
     break;
@@ -594,7 +592,9 @@ static void * streamsync_worker( void * args ) {
       ot_vector *torrents_list = mutex_bucket_lock( bucket );
       size_t tor_offset, count_def = 0, count_one = 0, count_two = 0, count_peers = 0;
       size_t mem, mem_a = 0, mem_b = 0;
-      uint8_t *ptr, *ptr_a, *ptr_b, *ptr_c;
+      uint8_t *ptr = 0, *ptr_a, *ptr_b, *ptr_c;
+
+      if( !torrents_list->size ) goto unlock_continue;
 
       /* For each torrent in this bucket.. */
       for( tor_offset=0; tor_offset<torrents_list->size; ++tor_offset ) {
@@ -614,6 +614,8 @@ static void * streamsync_worker( void * args ) {
       mem = 3 * ( 4 + 1 + 1 + 2 ) + ( count_one + count_two ) * 19 + count_def * 20 +
             ( count_one + 2 * count_two + count_peers ) * 7;
 
+     fprintf( stderr, "Mem: %d\n", mem );
+
       ptr = ptr_a = ptr_b = ptr_c = malloc( mem );
       if( !ptr ) goto unlock_continue;
 
@@ -697,11 +699,14 @@ unlock_continue:
         mem = ptr_c - ptr;
 
         for( i=0; i<g_connection_count; ++i ) {
-          if( g_connections[i].fd != -1 ) {
+          if( PROXYPEER_ISCONNECTED(g_connections[i].state) ) {
             void *tmp = malloc( mem );
-            if( tmp )
+            if( tmp ) {
+              memcpy( tmp, ptr, mem );
               if( !iob_addbuf_free( &g_connections[i].outdata, tmp, mem ) )
                 free( tmp );
+              io_wantwrite( g_connections[i].fd );
+            }
           }
         }
 
@@ -793,7 +798,7 @@ static void * livesync_worker( void * args ) {
       livesync_handle_peersync( datalen );
       break;
     default:
-      fprintf( stderr, "Received an unknown live sync packet type %u.\n", uint32_read_big( sizeof( g_tracker_id ) + (char*)g_inbuffer ) );
+      // fprintf( stderr, "Received an unknown live sync packet type %u.\n", uint32_read_big( sizeof( g_tracker_id ) + (char*)g_inbuffer ) );
       break;
     }
   }