1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
/*
* This file Copyright (C) 2008-2014 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
* $Id$
*/
#ifndef __TRANSMISSION__
#error only libtransmission should #include this header.
#endif
#ifndef TR_BITFIELD_H
#define TR_BITFIELD_H 1
#include "transmission.h"
/** @brief Implementation of the BitTorrent spec's Bitfield array of bits */
typedef struct tr_bitfield
{
uint8_t * bits;
size_t alloc_count;
size_t bit_count;
size_t true_count;
/* Special cases for when full or empty but we don't know the bitCount.
This occurs when a magnet link's peers send have all / have none */
bool have_all_hint;
bool have_none_hint;
}
tr_bitfield;
/***
****
***/
void tr_bitfieldSetHasAll (tr_bitfield*);
void tr_bitfieldSetHasNone (tr_bitfield*);
void tr_bitfieldAdd (tr_bitfield*, size_t bit);
void tr_bitfieldRem (tr_bitfield*, size_t bit);
void tr_bitfieldAddRange (tr_bitfield*, size_t begin, size_t end);
void tr_bitfieldRemRange (tr_bitfield*, size_t begin, size_t end);
/***
**** life cycle
***/
extern const tr_bitfield TR_BITFIELD_INIT;
void tr_bitfieldConstruct (tr_bitfield*, size_t bit_count);
static inline void
tr_bitfieldDestruct (tr_bitfield * b)
{
tr_bitfieldSetHasNone (b);
}
/***
****
***/
void tr_bitfieldSetFromFlags (tr_bitfield*, const bool * bytes, size_t n);
void tr_bitfieldSetFromBitfield (tr_bitfield*, const tr_bitfield*);
void tr_bitfieldSetRaw (tr_bitfield*, const void * bits, size_t byte_count, bool bounded);
void* tr_bitfieldGetRaw (const tr_bitfield * b, size_t * byte_count);
/***
****
***/
size_t tr_bitfieldCountRange (const tr_bitfield*, size_t begin, size_t end);
size_t tr_bitfieldCountTrueBits (const tr_bitfield * b);
static inline bool
tr_bitfieldHasAll (const tr_bitfield * b)
{
return b->bit_count ? (b->true_count == b->bit_count) : b->have_all_hint;
}
static inline bool
tr_bitfieldHasNone (const tr_bitfield * b)
{
return b->bit_count ? (b->true_count == 0) : b->have_none_hint;
}
bool tr_bitfieldHas (const tr_bitfield * b, size_t n);
#endif
|