source: branches/1.4x/libtransmission/peer-msgs-test.c @ 7455

Last change on this file since 7455 was 7455, checked in by charles, 12 years ago

(1.4x libT) backport handshake, peer, bandwidth, peer-io to 1.4x.

File size: 1.4 KB
Line 
1#include <stdio.h>
2#include "transmission.h"
3#include "net.h"
4#include "peer-msgs.h"
5#include "utils.h"
6
7#define VERBOSE 0
8
9#define check( A ) \
10    { \
11        ++test; \
12        if( A ){ \
13            if( VERBOSE ) \
14                fprintf( stderr, "PASS test #%d (%s, %d)\n", test, __FILE__,\
15                         __LINE__ );\
16        } else { \
17            if( VERBOSE ) \
18                fprintf( stderr, "FAIL test #%d (%s, %d)\n", test, __FILE__,\
19                         __LINE__ );\
20            return test; \
21        } \
22    }
23
24int
25main( void )
26{
27    uint32_t           i;
28    int                test = 0;
29    uint8_t            infohash[SHA_DIGEST_LENGTH];
30    tr_address         addr;
31    tr_piece_index_t   pieceCount = 1313;
32    size_t             numwant;
33    size_t             numgot;
34    tr_piece_index_t pieces[] = { 1059, 431, 808, 1217, 287, 376, 1188, 353, 508 };
35    tr_piece_index_t buf[16];
36
37    for( i = 0; i < SHA_DIGEST_LENGTH; ++i )
38        infohash[i] = 0xaa;
39    tr_netResolve( "80.4.4.200", &addr );
40
41    numwant = 7;
42    numgot = tr_generateAllowedSet( buf, numwant, pieceCount, infohash, &addr );
43    check( numgot == numwant );
44    for( i=0; i<numgot; ++i )
45        check( buf[i] == pieces[i] );
46
47    numwant = 9;
48    numgot = tr_generateAllowedSet( buf, numwant, pieceCount, infohash, &addr );
49    check( numgot == numwant );
50    for( i=0; i<numgot; ++i )
51        check( buf[i] == pieces[i] );
52
53    return 0;
54}
55
Note: See TracBrowser for help on using the repository browser.