Browse Source

tidy up peer count parser code

dynamic-accesslists
erdgeist 14 years ago
parent
commit
9cfab8dbc6
  1. 17
      proxy.c

17
proxy.c

@ -245,9 +245,9 @@ typedef struct {
int64 fd; /* A file handle, if connected, <= 0 is disconnected (0 initially, -1 else) */ int64 fd; /* A file handle, if connected, <= 0 is disconnected (0 initially, -1 else) */
io_batch outdata; /* The iobatch containing our sync data */ io_batch outdata; /* The iobatch containing our sync data */
int packet_tcount; /* Number of unprocessed torrents in packet we currently receive */ size_t packet_tcount; /* Number of unprocessed torrents in packet we currently receive */
char packet_tprefix; /* Prefix byte for all torrents in current packet */ uint8_t packet_tprefix; /* Prefix byte for all torrents in current packet */
char packet_type; /* Type of current packet */ uint8_t packet_type; /* Type of current packet */
uint32_t packet_tid; /* Tracker id for current packet */ uint32_t packet_tid; /* Tracker id for current packet */
} proxy_peer; } proxy_peer;
@ -754,7 +754,7 @@ static void livesync_proxytell( uint8_t prefix, uint8_t *info_hash, uint8_t *pee
g_peerbuffer_pos += sizeof(ot_hash); g_peerbuffer_pos += sizeof(ot_hash);
printf( "%d.%d.%d.%d:%hd (%02X %02X)\n", g_peerbuffer_pos[0], g_peerbuffer_pos[1], g_peerbuffer_pos[2], g_peerbuffer_pos[3], printf( "%hhu.%hhu.%hhu.%hhu:%hu (%02X %02X)\n", g_peerbuffer_pos[0], g_peerbuffer_pos[1], g_peerbuffer_pos[2], g_peerbuffer_pos[3],
g_peerbuffer_pos[4] | ( g_peerbuffer_pos[5] << 8 ), g_peerbuffer_pos[6], g_peerbuffer_pos[7] ); g_peerbuffer_pos[4] | ( g_peerbuffer_pos[5] << 8 ), g_peerbuffer_pos[6], g_peerbuffer_pos[7] );
g_peerbuffer_pos += sizeof(ot_peer); g_peerbuffer_pos += sizeof(ot_peer);
@ -778,7 +778,7 @@ static void process_indata( proxy_peer * peer ) {
peer->packet_tprefix = data[5]; peer->packet_tprefix = data[5];
peer->packet_tcount = data[6] * 256 + data[7]; peer->packet_tcount = data[6] * 256 + data[7];
data += 8; data += 8;
printf( "type: %d, prefix: %02X, torrentcount: %d\n", peer->packet_type, peer->packet_tprefix, peer->packet_tcount ); printf( "type: %hhu, prefix: %02X, torrentcount: %zd\n", peer->packet_type, peer->packet_tprefix, peer->packet_tcount );
} }
next_torrent: next_torrent:
@ -790,13 +790,12 @@ next_torrent:
data += sizeof(ot_hash) - 1; data += sizeof(ot_hash) - 1;
/* Type 0 has peer count encoded before each peers */ /* Type 0 has peer count encoded before each peers */
if( peer->packet_type == 0 ) { peers = peer->packet_type;
if( !peers ) {
int shift = 0; int shift = 0;
peers = 0;
do peers |= ( 0x7f & *data ) << ( 7 * shift ); do peers |= ( 0x7f & *data ) << ( 7 * shift );
while ( *(data++) & 0x80 && shift++ < 6 ); while ( *(data++) & 0x80 && shift++ < 6 );
} else }
peers = peer->packet_type;
/* Ensure enough data being read to hold all peers */ /* Ensure enough data being read to hold all peers */
if( data + 7 * peers > dataend ) break; if( data + 7 * peers > dataend ) break;

Loading…
Cancel
Save