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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
* Copyright (C) 2001-2012 Jacek Sieka, arnetheduck on gmail point com
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#pragma once
#include <deque>
#include "forward.h"
#include "MerkleTree.h"
#include "CriticalSection.h"
#include "NonCopyable.h"
#include "Segment.h"
#include "GetSet.h"
namespace dcpp {
using std::deque;
class Transfer : private NonCopyable {
public:
enum Type {
TYPE_FILE,
TYPE_FULL_LIST,
TYPE_PARTIAL_LIST,
TYPE_TREE,
TYPE_LAST
};
static const string names[TYPE_LAST];
static const string USER_LIST_NAME;
static const string USER_LIST_NAME_BZ;
Transfer(UserConnection& conn, const string& path, const TTHValue& tth);
virtual ~Transfer() { };
int64_t getPos() const { return pos; }
int64_t getStartPos() const { return getSegment().getStart(); }
void resetPos() { pos = 0; }
void addPos(int64_t aBytes, int64_t aActual) { pos += aBytes; actual+= aActual; }
enum { MIN_SAMPLES = 15, MIN_SECS = 15 };
/** Record a sample for average calculation */
void tick();
int64_t getActual() const { return actual; }
int64_t getSize() const { return getSegment().getSize(); }
void setSize(int64_t size) { segment.setSize(size); }
double getAverageSpeed() const;
int64_t getSecondsLeft() const {
double avg = getAverageSpeed();
return (avg > 0) ? (getBytesLeft() / avg) : 0;
}
int64_t getBytesLeft() const {
return getSize() - getPos();
}
virtual void getParams(const UserConnection& aSource, ParamMap& params);
UserPtr getUser();
const UserPtr getUser() const;
HintedUser getHintedUser() const;
const string& getPath() const { return path; }
const TTHValue& getTTH() const { return tth; }
UserConnection& getUserConnection() { return userConnection; }
const UserConnection& getUserConnection() const { return userConnection; }
bool getOverlapped() const { return getSegment().getOverlapped(); }
void setOverlapped(bool overlap) { segment.setOverlapped(overlap); }
GETSET(Segment, segment, Segment);
GETSET(Type, type, Type);
GETSET(uint64_t, start, Start);
private:
typedef std::pair<uint64_t, int64_t> Sample;
typedef deque<Sample> SampleList;
SampleList samples;
mutable CriticalSection cs;
/** The file being transferred */
string path;
/** TTH of the file being transferred */
TTHValue tth;
/** Bytes transferred over socket */
int64_t actual;
/** Bytes transferred to/from file */
int64_t pos;
UserConnection& userConnection;
};
} // namespace dcpp
|
