867 lines
74 KiB
HTML
867 lines
74 KiB
HTML
<?xml version="1.0" encoding="UTF-8"?>
|
|
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
|
|
<html xmlns="http://www.w3.org/1999/xhtml" xml:lang="en_US" lang="en_US">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
|
|
<title>Qt 4.8: peerwireclient.cpp Example File (network/torrent/peerwireclient.cpp)</title>
|
|
<link rel="stylesheet" type="text/css" href="style/style.css" />
|
|
<script src="scripts/jquery.js" type="text/javascript"></script>
|
|
<script src="scripts/functions.js" type="text/javascript"></script>
|
|
<link rel="stylesheet" type="text/css" href="style/superfish.css" />
|
|
<link rel="stylesheet" type="text/css" href="style/narrow.css" />
|
|
<!--[if IE]>
|
|
<meta name="MSSmartTagsPreventParsing" content="true">
|
|
<meta http-equiv="imagetoolbar" content="no">
|
|
<![endif]-->
|
|
<!--[if lt IE 7]>
|
|
<link rel="stylesheet" type="text/css" href="style/style_ie6.css">
|
|
<![endif]-->
|
|
<!--[if IE 7]>
|
|
<link rel="stylesheet" type="text/css" href="style/style_ie7.css">
|
|
<![endif]-->
|
|
<!--[if IE 8]>
|
|
<link rel="stylesheet" type="text/css" href="style/style_ie8.css">
|
|
<![endif]-->
|
|
|
|
<script src="scripts/superfish.js" type="text/javascript"></script>
|
|
<script src="scripts/narrow.js" type="text/javascript"></script>
|
|
|
|
</head>
|
|
<body class="" onload="CheckEmptyAndLoadList();">
|
|
<div class="header" id="qtdocheader">
|
|
<div class="content">
|
|
<div id="nav-logo">
|
|
<a href="index.html">Home</a></div>
|
|
<a href="index.html" class="qtref"><span>Qt Reference Documentation</span></a>
|
|
<div id="narrowsearch"></div>
|
|
<div id="nav-topright">
|
|
<ul>
|
|
<li class="nav-topright-home"><a href="http://qt.digia.com/">Qt HOME</a></li>
|
|
<li class="nav-topright-dev"><a href="http://qt-project.org/">DEV</a></li>
|
|
<li class="nav-topright-doc nav-topright-doc-active"><a href="http://qt-project.org/doc/">
|
|
DOC</a></li>
|
|
<li class="nav-topright-blog"><a href="http://blog.qt.digia.com/">BLOG</a></li>
|
|
</ul>
|
|
</div>
|
|
<div id="shortCut">
|
|
<ul>
|
|
<li class="shortCut-topleft-inactive"><span><a href="index.html">Qt 4.8</a></span></li>
|
|
<li class="shortCut-topleft-active"><a href="http://qt-project.org/doc/">ALL VERSIONS </a></li>
|
|
</ul>
|
|
</div>
|
|
<ul class="sf-menu" id="narrowmenu">
|
|
<li><a href="#">API Lookup</a>
|
|
<ul>
|
|
<li><a href="classes.html">Class index</a></li>
|
|
<li><a href="functions.html">Function index</a></li>
|
|
<li><a href="modules.html">Modules</a></li>
|
|
<li><a href="namespaces.html">Namespaces</a></li>
|
|
<li><a href="qtglobal.html">Global Declarations</a></li>
|
|
<li><a href="qdeclarativeelements.html">QML elements</a></li>
|
|
</ul>
|
|
</li>
|
|
<li><a href="#">Qt Topics</a>
|
|
<ul>
|
|
<li><a href="qt-basic-concepts.html">Programming with Qt</a></li>
|
|
<li><a href="qtquick.html">Device UIs & Qt Quick</a></li>
|
|
<li><a href="qt-gui-concepts.html">UI Design with Qt</a></li>
|
|
<li><a href="supported-platforms.html">Supported Platforms</a></li>
|
|
<li><a href="technology-apis.html">Qt and Key Technologies</a></li>
|
|
<li><a href="best-practices.html">How-To's and Best Practices</a></li>
|
|
</ul>
|
|
</li>
|
|
<li><a href="#">Examples</a>
|
|
<ul>
|
|
<li><a href="all-examples.html">Examples</a></li>
|
|
<li><a href="tutorials.html">Tutorials</a></li>
|
|
<li><a href="demos.html">Demos</a></li>
|
|
<li><a href="qdeclarativeexamples.html">QML Examples</a></li>
|
|
</ul>
|
|
</li>
|
|
</ul>
|
|
</div>
|
|
</div>
|
|
<div class="wrapper">
|
|
<div class="hd">
|
|
<span></span>
|
|
</div>
|
|
<div class="bd group">
|
|
<div class="sidebar">
|
|
<div class="searchlabel">
|
|
Search index:</div>
|
|
<div class="search" id="sidebarsearch">
|
|
<form id="qtdocsearch" action="" onsubmit="return false;">
|
|
<fieldset>
|
|
<input type="text" name="searchstring" id="pageType" value="" />
|
|
<div id="resultdialog">
|
|
<a href="#" id="resultclose">Close</a>
|
|
<p id="resultlinks" class="all"><a href="#" id="showallresults">All</a> | <a href="#" id="showapiresults">API</a> | <a href="#" id="showarticleresults">Articles</a> | <a href="#" id="showexampleresults">Examples</a></p>
|
|
<p id="searchcount" class="all"><span id="resultcount"></span><span id="apicount"></span><span id="articlecount"></span><span id="examplecount"></span> results:</p>
|
|
<ul id="resultlist" class="all">
|
|
</ul>
|
|
</div>
|
|
</fieldset>
|
|
</form>
|
|
</div>
|
|
<div class="box first bottombar" id="lookup">
|
|
<h2 title="API Lookup"><span></span>
|
|
API Lookup</h2>
|
|
<div id="list001" class="list">
|
|
<ul id="ul001" >
|
|
<li class="defaultLink"><a href="classes.html">Class index</a></li>
|
|
<li class="defaultLink"><a href="functions.html">Function index</a></li>
|
|
<li class="defaultLink"><a href="modules.html">Modules</a></li>
|
|
<li class="defaultLink"><a href="namespaces.html">Namespaces</a></li>
|
|
<li class="defaultLink"><a href="qtglobal.html">Global Declarations</a></li>
|
|
<li class="defaultLink"><a href="qdeclarativeelements.html">QML elements</a></li>
|
|
</ul>
|
|
</div>
|
|
</div>
|
|
<div class="box bottombar" id="topics">
|
|
<h2 title="Qt Topics"><span></span>
|
|
Qt Topics</h2>
|
|
<div id="list002" class="list">
|
|
<ul id="ul002" >
|
|
<li class="defaultLink"><a href="qt-basic-concepts.html">Programming with Qt</a></li>
|
|
<li class="defaultLink"><a href="qtquick.html">Device UIs & Qt Quick</a></li>
|
|
<li class="defaultLink"><a href="qt-gui-concepts.html">UI Design with Qt</a></li>
|
|
<li class="defaultLink"><a href="supported-platforms.html">Supported Platforms</a></li>
|
|
<li class="defaultLink"><a href="technology-apis.html">Qt and Key Technologies</a></li>
|
|
<li class="defaultLink"><a href="best-practices.html">How-To's and Best Practices</a></li>
|
|
</ul>
|
|
</div>
|
|
</div>
|
|
<div class="box" id="examples">
|
|
<h2 title="Examples"><span></span>
|
|
Examples</h2>
|
|
<div id="list003" class="list">
|
|
<ul id="ul003">
|
|
<li class="defaultLink"><a href="all-examples.html">Examples</a></li>
|
|
<li class="defaultLink"><a href="tutorials.html">Tutorials</a></li>
|
|
<li class="defaultLink"><a href="demos.html">Demos</a></li>
|
|
<li class="defaultLink"><a href="qdeclarativeexamples.html">QML Examples</a></li>
|
|
</ul>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div class="wrap">
|
|
<div class="toolbar">
|
|
<div class="breadcrumb toolblock">
|
|
<ul>
|
|
<li class="first"><a href="index.html">Home</a></li>
|
|
<!-- Breadcrumbs go here -->
|
|
</ul>
|
|
</div>
|
|
<div class="toolbuttons toolblock">
|
|
<ul>
|
|
<li id="smallA" class="t_button">A</li>
|
|
<li id="medA" class="t_button active">A</li>
|
|
<li id="bigA" class="t_button">A</li>
|
|
<li id="print" class="t_button"><a href="javascript:this.print();">
|
|
<span>Print</span></a></li>
|
|
</ul>
|
|
</div>
|
|
</div>
|
|
<div class="content mainContent">
|
|
<h1 class="title">peerwireclient.cpp Example File</h1>
|
|
<span class="small-subtitle">network/torrent/peerwireclient.cpp</span>
|
|
<!-- $$$network/torrent/peerwireclient.cpp-description -->
|
|
<div class="descr"> <a name="details"></a>
|
|
<pre class="cpp"> <span class="comment">/****************************************************************************
|
|
**
|
|
** Copyright (C) 2015 The Qt Company Ltd.
|
|
** Contact: http://www.qt.io/licensing/
|
|
**
|
|
** This file is part of the examples of the Qt Toolkit.
|
|
**
|
|
** $QT_BEGIN_LICENSE:BSD$
|
|
** You may use this file under the terms of the BSD license as follows:
|
|
**
|
|
** "Redistribution and use in source and binary forms, with or without
|
|
** modification, are permitted provided that the following conditions are
|
|
** met:
|
|
** * Redistributions of source code must retain the above copyright
|
|
** notice, this list of conditions and the following disclaimer.
|
|
** * Redistributions in binary form must reproduce the above copyright
|
|
** notice, this list of conditions and the following disclaimer in
|
|
** the documentation and/or other materials provided with the
|
|
** distribution.
|
|
** * Neither the name of The Qt Company Ltd nor the names of its
|
|
** contributors may be used to endorse or promote products derived
|
|
** from this software without specific prior written permission.
|
|
**
|
|
**
|
|
** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE."
|
|
**
|
|
** $QT_END_LICENSE$
|
|
**
|
|
****************************************************************************/</span>
|
|
|
|
<span class="preprocessor">#include "peerwireclient.h"</span>
|
|
|
|
<span class="preprocessor">#include <QHostAddress></span>
|
|
<span class="preprocessor">#include <QTimerEvent></span>
|
|
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> PendingRequestTimeout <span class="operator">=</span> <span class="number">60</span> <span class="operator">*</span> <span class="number">1000</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> ClientTimeout <span class="operator">=</span> <span class="number">120</span> <span class="operator">*</span> <span class="number">1000</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> ConnectTimeout <span class="operator">=</span> <span class="number">60</span> <span class="operator">*</span> <span class="number">1000</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> KeepAliveInterval <span class="operator">=</span> <span class="number">30</span> <span class="operator">*</span> <span class="number">1000</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> RateControlTimerDelay <span class="operator">=</span> <span class="number">2000</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> MinimalHeaderSize <span class="operator">=</span> <span class="number">48</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">int</span> FullHeaderSize <span class="operator">=</span> <span class="number">68</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">char</span> ProtocolId<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> <span class="string">"BitTorrent protocol"</span>;
|
|
<span class="keyword">static</span> <span class="keyword">const</span> <span class="type">char</span> ProtocolIdSize <span class="operator">=</span> <span class="number">19</span>;
|
|
|
|
<span class="comment">// Reads a 32bit unsigned int from data in network order.</span>
|
|
<span class="keyword">static</span> <span class="keyword">inline</span> <span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> fromNetworkData(<span class="keyword">const</span> <span class="type">char</span> <span class="operator">*</span>data)
|
|
{
|
|
<span class="keyword">const</span> <span class="type">unsigned</span> <span class="type">char</span> <span class="operator">*</span>udata <span class="operator">=</span> (<span class="keyword">const</span> <span class="type">unsigned</span> <span class="type">char</span> <span class="operator">*</span>)data;
|
|
<span class="keyword">return</span> (<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(udata<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>) <span class="operator"><</span><span class="operator"><</span> <span class="number">24</span>)
|
|
<span class="operator">|</span> (<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(udata<span class="operator">[</span><span class="number">1</span><span class="operator">]</span>) <span class="operator"><</span><span class="operator"><</span> <span class="number">16</span>)
|
|
<span class="operator">|</span> (<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(udata<span class="operator">[</span><span class="number">2</span><span class="operator">]</span>) <span class="operator"><</span><span class="operator"><</span> <span class="number">8</span>)
|
|
<span class="operator">|</span> (<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(udata<span class="operator">[</span><span class="number">3</span><span class="operator">]</span>));
|
|
}
|
|
|
|
<span class="comment">// Writes a 32bit unsigned int from num to data in network order.</span>
|
|
<span class="keyword">static</span> <span class="keyword">inline</span> <span class="type">void</span> toNetworkData(<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> num<span class="operator">,</span> <span class="type">char</span> <span class="operator">*</span>data)
|
|
{
|
|
<span class="type">unsigned</span> <span class="type">char</span> <span class="operator">*</span>udata <span class="operator">=</span> (<span class="type">unsigned</span> <span class="type">char</span> <span class="operator">*</span>)data;
|
|
udata<span class="operator">[</span><span class="number">3</span><span class="operator">]</span> <span class="operator">=</span> (num <span class="operator">&</span> <span class="number">0xff</span>);
|
|
udata<span class="operator">[</span><span class="number">2</span><span class="operator">]</span> <span class="operator">=</span> (num <span class="operator">&</span> <span class="number">0xff00</span>) <span class="operator">></span><span class="operator">></span> <span class="number">8</span>;
|
|
udata<span class="operator">[</span><span class="number">1</span><span class="operator">]</span> <span class="operator">=</span> (num <span class="operator">&</span> <span class="number">0xff0000</span>) <span class="operator">></span><span class="operator">></span> <span class="number">16</span>;
|
|
udata<span class="operator">[</span><span class="number">0</span><span class="operator">]</span> <span class="operator">=</span> (num <span class="operator">&</span> <span class="number">0xff000000</span>) <span class="operator">></span><span class="operator">></span> <span class="number">24</span>;
|
|
}
|
|
|
|
<span class="comment">// Constructs an unconnected PeerWire client and starts the connect timer.</span>
|
|
PeerWireClient<span class="operator">::</span>PeerWireClient(<span class="keyword">const</span> <span class="type"><a href="qbytearray.html">QByteArray</a></span> <span class="operator">&</span>peerId<span class="operator">,</span> <span class="type"><a href="qobject.html">QObject</a></span> <span class="operator">*</span>parent)
|
|
: <span class="type"><a href="qtcpsocket.html">QTcpSocket</a></span>(parent)<span class="operator">,</span> pendingBlockSizes(<span class="number">0</span>)<span class="operator">,</span>
|
|
pwState(ChokingPeer <span class="operator">|</span> ChokedByPeer)<span class="operator">,</span> receivedHandShake(<span class="keyword">false</span>)<span class="operator">,</span> gotPeerId(<span class="keyword">false</span>)<span class="operator">,</span>
|
|
sentHandShake(<span class="keyword">false</span>)<span class="operator">,</span> nextPacketLength(<span class="operator">-</span><span class="number">1</span>)<span class="operator">,</span> pendingRequestTimer(<span class="number">0</span>)<span class="operator">,</span> invalidateTimeout(<span class="keyword">false</span>)<span class="operator">,</span>
|
|
keepAliveTimer(<span class="number">0</span>)<span class="operator">,</span> torrentPeer(<span class="number">0</span>)
|
|
{
|
|
memset(uploadSpeedData<span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="keyword">sizeof</span>(uploadSpeedData));
|
|
memset(downloadSpeedData<span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="keyword">sizeof</span>(downloadSpeedData));
|
|
|
|
transferSpeedTimer <span class="operator">=</span> startTimer(RateControlTimerDelay);
|
|
timeoutTimer <span class="operator">=</span> startTimer(ConnectTimeout);
|
|
peerIdString <span class="operator">=</span> peerId;
|
|
|
|
connect(<span class="keyword">this</span><span class="operator">,</span> SIGNAL(readyRead())<span class="operator">,</span> <span class="keyword">this</span><span class="operator">,</span> SIGNAL(readyToTransfer()));
|
|
connect(<span class="keyword">this</span><span class="operator">,</span> SIGNAL(connected())<span class="operator">,</span> <span class="keyword">this</span><span class="operator">,</span> SIGNAL(readyToTransfer()));
|
|
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(connected())<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SIGNAL(connected()));
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(readyRead())<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SIGNAL(readyRead()));
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(disconnected())<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SIGNAL(disconnected()));
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(error(<span class="type"><a href="qabstractsocket.html">QAbstractSocket</a></span><span class="operator">::</span>SocketError))<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SIGNAL(error(<span class="type"><a href="qabstractsocket.html">QAbstractSocket</a></span><span class="operator">::</span>SocketError)));
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(bytesWritten(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span>))<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SIGNAL(bytesWritten(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span>)));
|
|
connect(<span class="operator">&</span>socket<span class="operator">,</span> SIGNAL(stateChanged(<span class="type"><a href="qabstractsocket.html">QAbstractSocket</a></span><span class="operator">::</span>SocketState))<span class="operator">,</span>
|
|
<span class="keyword">this</span><span class="operator">,</span> SLOT(socketStateChanged(<span class="type"><a href="qabstractsocket.html">QAbstractSocket</a></span><span class="operator">::</span>SocketState)));
|
|
|
|
}
|
|
|
|
<span class="comment">// Registers the peer ID and SHA1 sum of the torrent, and initiates</span>
|
|
<span class="comment">// the handshake.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>initialize(<span class="keyword">const</span> <span class="type"><a href="qbytearray.html">QByteArray</a></span> <span class="operator">&</span>infoHash<span class="operator">,</span> <span class="type">int</span> pieceCount)
|
|
{
|
|
<span class="keyword">this</span><span class="operator">-</span><span class="operator">></span>infoHash <span class="operator">=</span> infoHash;
|
|
peerPieces<span class="operator">.</span>resize(pieceCount);
|
|
<span class="keyword">if</span> (<span class="operator">!</span>sentHandShake)
|
|
sendHandShake();
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>setPeer(TorrentPeer <span class="operator">*</span>peer)
|
|
{
|
|
torrentPeer <span class="operator">=</span> peer;
|
|
}
|
|
|
|
TorrentPeer <span class="operator">*</span>PeerWireClient<span class="operator">::</span>peer() <span class="keyword">const</span>
|
|
{
|
|
<span class="keyword">return</span> torrentPeer;
|
|
}
|
|
|
|
<span class="type"><a href="qbitarray.html">QBitArray</a></span> PeerWireClient<span class="operator">::</span>availablePieces() <span class="keyword">const</span>
|
|
{
|
|
<span class="keyword">return</span> peerPieces;
|
|
}
|
|
|
|
<span class="type"><a href="qlist.html">QList</a></span><span class="operator"><</span>TorrentBlock<span class="operator">></span> PeerWireClient<span class="operator">::</span>incomingBlocks() <span class="keyword">const</span>
|
|
{
|
|
<span class="keyword">return</span> incoming;
|
|
}
|
|
|
|
<span class="comment">// Sends a "choke" message, asking the peer to stop requesting blocks.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>chokePeer()
|
|
{
|
|
<span class="keyword">const</span> <span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">0</span>};
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
pwState <span class="operator">|</span><span class="operator">=</span> ChokingPeer;
|
|
|
|
<span class="comment">// After receiving a choke message, the peer will assume all</span>
|
|
<span class="comment">// pending requests are lost.</span>
|
|
pendingBlocks<span class="operator">.</span>clear();
|
|
pendingBlockSizes <span class="operator">=</span> <span class="number">0</span>;
|
|
}
|
|
|
|
<span class="comment">// Sends an "unchoke" message, allowing the peer to start/resume</span>
|
|
<span class="comment">// requesting blocks.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>unchokePeer()
|
|
{
|
|
<span class="keyword">const</span> <span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">1</span>};
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
pwState <span class="operator">&</span><span class="operator">=</span> <span class="operator">~</span>ChokingPeer;
|
|
|
|
<span class="keyword">if</span> (pendingRequestTimer)
|
|
killTimer(pendingRequestTimer);
|
|
}
|
|
|
|
<span class="comment">// Sends a "keep-alive" message to prevent the peer from closing</span>
|
|
<span class="comment">// the connection when there's no activity</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendKeepAlive()
|
|
{
|
|
<span class="keyword">const</span> <span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span>};
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
}
|
|
|
|
<span class="comment">// Sends an "interested" message, informing the peer that it has got</span>
|
|
<span class="comment">// pieces that we'd like to download.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendInterested()
|
|
{
|
|
<span class="keyword">const</span> <span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">2</span>};
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
pwState <span class="operator">|</span><span class="operator">=</span> InterestedInPeer;
|
|
|
|
<span class="comment">// After telling the peer that we're interested, we expect to get</span>
|
|
<span class="comment">// unchoked within a certain timeframe; otherwise we'll drop the</span>
|
|
<span class="comment">// connection.</span>
|
|
<span class="keyword">if</span> (pendingRequestTimer)
|
|
killTimer(pendingRequestTimer);
|
|
pendingRequestTimer <span class="operator">=</span> startTimer(PendingRequestTimeout);
|
|
}
|
|
|
|
<span class="comment">// Sends a "not interested" message, informing the peer that it does</span>
|
|
<span class="comment">// not have any pieces that we'd like to download.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendNotInterested()
|
|
{
|
|
<span class="keyword">const</span> <span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">3</span>};
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
pwState <span class="operator">&</span><span class="operator">=</span> <span class="operator">~</span>InterestedInPeer;
|
|
}
|
|
|
|
<span class="comment">// Sends a piece notification / a "have" message, informing the peer</span>
|
|
<span class="comment">// that we have just downloaded a new piece.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendPieceNotification(<span class="type">int</span> piece)
|
|
{
|
|
<span class="keyword">if</span> (<span class="operator">!</span>sentHandShake)
|
|
sendHandShake();
|
|
|
|
<span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">5</span><span class="operator">,</span> <span class="number">4</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span>};
|
|
toNetworkData(piece<span class="operator">,</span> <span class="operator">&</span>message<span class="operator">[</span><span class="number">5</span><span class="operator">]</span>);
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
}
|
|
|
|
<span class="comment">// Sends the complete list of pieces that we have downloaded.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendPieceList(<span class="keyword">const</span> <span class="type"><a href="qbitarray.html">QBitArray</a></span> <span class="operator">&</span>bitField)
|
|
{
|
|
<span class="comment">// The bitfield message may only be sent immediately after the</span>
|
|
<span class="comment">// handshaking sequence is completed, and before any other</span>
|
|
<span class="comment">// messages are sent.</span>
|
|
<span class="keyword">if</span> (<span class="operator">!</span>sentHandShake)
|
|
sendHandShake();
|
|
|
|
<span class="comment">// Don't send the bitfield if it's all zeros.</span>
|
|
<span class="keyword">if</span> (bitField<span class="operator">.</span>count(<span class="keyword">true</span>) <span class="operator">=</span><span class="operator">=</span> <span class="number">0</span>)
|
|
<span class="keyword">return</span>;
|
|
|
|
<span class="type">int</span> bitFieldSize <span class="operator">=</span> bitField<span class="operator">.</span>size();
|
|
<span class="type">int</span> size <span class="operator">=</span> (bitFieldSize <span class="operator">+</span> <span class="number">7</span>) <span class="operator">/</span> <span class="number">8</span>;
|
|
<span class="type"><a href="qbytearray.html">QByteArray</a></span> bits(size<span class="operator">,</span> <span class="char">'\0'</span>);
|
|
<span class="keyword">for</span> (<span class="type">int</span> i <span class="operator">=</span> <span class="number">0</span>; i <span class="operator"><</span> bitFieldSize; <span class="operator">+</span><span class="operator">+</span>i) {
|
|
<span class="keyword">if</span> (bitField<span class="operator">.</span>testBit(i)) {
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> byte <span class="operator">=</span> <span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(i) <span class="operator">/</span> <span class="number">8</span>;
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> bit <span class="operator">=</span> <span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(i) <span class="operator">%</span> <span class="number">8</span>;
|
|
bits<span class="operator">[</span>byte<span class="operator">]</span> <span class="operator">=</span> <span class="type"><a href="qtglobal.html#uchar-typedef">uchar</a></span>(bits<span class="operator">.</span>at(byte)) <span class="operator">|</span> (<span class="number">1</span> <span class="operator"><</span><span class="operator"><</span> (<span class="number">7</span> <span class="operator">-</span> bit));
|
|
}
|
|
}
|
|
|
|
<span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">5</span>};
|
|
toNetworkData(bits<span class="operator">.</span>size() <span class="operator">+</span> <span class="number">1</span><span class="operator">,</span> <span class="operator">&</span>message<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
write(bits);
|
|
}
|
|
|
|
<span class="comment">// Sends a request for a block.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>requestBlock(<span class="type">int</span> piece<span class="operator">,</span> <span class="type">int</span> offset<span class="operator">,</span> <span class="type">int</span> length)
|
|
{
|
|
<span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">6</span>};
|
|
toNetworkData(<span class="number">13</span><span class="operator">,</span> <span class="operator">&</span>message<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
|
|
<span class="type">char</span> numbers<span class="operator">[</span><span class="number">4</span> <span class="operator">*</span> <span class="number">3</span><span class="operator">]</span>;
|
|
toNetworkData(piece<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
toNetworkData(offset<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">4</span><span class="operator">]</span>);
|
|
toNetworkData(length<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">8</span><span class="operator">]</span>);
|
|
write(numbers<span class="operator">,</span> <span class="keyword">sizeof</span>(numbers));
|
|
|
|
incoming <span class="operator"><</span><span class="operator"><</span> TorrentBlock(piece<span class="operator">,</span> offset<span class="operator">,</span> length);
|
|
|
|
<span class="comment">// After requesting a block, we expect the block to be sent by the</span>
|
|
<span class="comment">// other peer within a certain number of seconds. Otherwise, we</span>
|
|
<span class="comment">// drop the connection.</span>
|
|
<span class="keyword">if</span> (pendingRequestTimer)
|
|
killTimer(pendingRequestTimer);
|
|
pendingRequestTimer <span class="operator">=</span> startTimer(PendingRequestTimeout);
|
|
}
|
|
|
|
<span class="comment">// Cancels a request for a block.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>cancelRequest(<span class="type">int</span> piece<span class="operator">,</span> <span class="type">int</span> offset<span class="operator">,</span> <span class="type">int</span> length)
|
|
{
|
|
<span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">8</span>};
|
|
toNetworkData(<span class="number">13</span><span class="operator">,</span> <span class="operator">&</span>message<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
write(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
|
|
<span class="type">char</span> numbers<span class="operator">[</span><span class="number">4</span> <span class="operator">*</span> <span class="number">3</span><span class="operator">]</span>;
|
|
toNetworkData(piece<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
toNetworkData(offset<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">4</span><span class="operator">]</span>);
|
|
toNetworkData(length<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">8</span><span class="operator">]</span>);
|
|
write(numbers<span class="operator">,</span> <span class="keyword">sizeof</span>(numbers));
|
|
|
|
incoming<span class="operator">.</span>removeAll(TorrentBlock(piece<span class="operator">,</span> offset<span class="operator">,</span> length));
|
|
}
|
|
|
|
<span class="comment">// Sends a block to the peer.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendBlock(<span class="type">int</span> piece<span class="operator">,</span> <span class="type">int</span> offset<span class="operator">,</span> <span class="keyword">const</span> <span class="type"><a href="qbytearray.html">QByteArray</a></span> <span class="operator">&</span>data)
|
|
{
|
|
<span class="type"><a href="qbytearray.html">QByteArray</a></span> block;
|
|
|
|
<span class="type">char</span> message<span class="operator">[</span><span class="operator">]</span> <span class="operator">=</span> {<span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">0</span><span class="operator">,</span> <span class="number">1</span><span class="operator">,</span> <span class="number">7</span>};
|
|
toNetworkData(<span class="number">9</span> <span class="operator">+</span> data<span class="operator">.</span>size()<span class="operator">,</span> <span class="operator">&</span>message<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
block <span class="operator">+</span><span class="operator">=</span> <span class="type"><a href="qbytearray.html">QByteArray</a></span>(message<span class="operator">,</span> <span class="keyword">sizeof</span>(message));
|
|
|
|
<span class="type">char</span> numbers<span class="operator">[</span><span class="number">4</span> <span class="operator">*</span> <span class="number">2</span><span class="operator">]</span>;
|
|
toNetworkData(piece<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">0</span><span class="operator">]</span>);
|
|
toNetworkData(offset<span class="operator">,</span> <span class="operator">&</span>numbers<span class="operator">[</span><span class="number">4</span><span class="operator">]</span>);
|
|
block <span class="operator">+</span><span class="operator">=</span> <span class="type"><a href="qbytearray.html">QByteArray</a></span>(numbers<span class="operator">,</span> <span class="keyword">sizeof</span>(numbers));
|
|
block <span class="operator">+</span><span class="operator">=</span> data;
|
|
|
|
BlockInfo blockInfo;
|
|
blockInfo<span class="operator">.</span>pieceIndex <span class="operator">=</span> piece;
|
|
blockInfo<span class="operator">.</span>offset <span class="operator">=</span> offset;
|
|
blockInfo<span class="operator">.</span>length <span class="operator">=</span> data<span class="operator">.</span>size();
|
|
blockInfo<span class="operator">.</span>block <span class="operator">=</span> block;
|
|
|
|
pendingBlocks <span class="operator"><</span><span class="operator"><</span> blockInfo;
|
|
pendingBlockSizes <span class="operator">+</span><span class="operator">=</span> block<span class="operator">.</span>size();
|
|
|
|
<span class="keyword">if</span> (pendingBlockSizes <span class="operator">></span> <span class="number">32</span> <span class="operator">*</span> <span class="number">16384</span>) {
|
|
chokePeer();
|
|
unchokePeer();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
<span class="keyword">emit</span> readyToTransfer();
|
|
}
|
|
|
|
<span class="comment">// Attempts to write 'bytes' bytes to the socket from the buffer.</span>
|
|
<span class="comment">// This is used by RateController, which precisely controls how much</span>
|
|
<span class="comment">// each client can write.</span>
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>writeToSocket(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> bytes)
|
|
{
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> totalWritten <span class="operator">=</span> <span class="number">0</span>;
|
|
<span class="keyword">do</span> {
|
|
<span class="keyword">if</span> (outgoingBuffer<span class="operator">.</span>isEmpty() <span class="operator">&</span><span class="operator">&</span> <span class="operator">!</span>pendingBlocks<span class="operator">.</span>isEmpty()) {
|
|
BlockInfo block <span class="operator">=</span> pendingBlocks<span class="operator">.</span>takeFirst();
|
|
pendingBlockSizes <span class="operator">-</span><span class="operator">=</span> block<span class="operator">.</span>length;
|
|
outgoingBuffer <span class="operator">+</span><span class="operator">=</span> block<span class="operator">.</span>block;
|
|
}
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> written <span class="operator">=</span> socket<span class="operator">.</span>write(outgoingBuffer<span class="operator">.</span>constData()<span class="operator">,</span>
|
|
<a href="qtglobal.html#qMin">qMin</a><span class="operator"><</span><span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span><span class="operator">></span>(bytes <span class="operator">-</span> totalWritten<span class="operator">,</span> outgoingBuffer<span class="operator">.</span>size()));
|
|
<span class="keyword">if</span> (written <span class="operator"><</span><span class="operator">=</span> <span class="number">0</span>)
|
|
<span class="keyword">return</span> totalWritten <span class="operator">?</span> totalWritten : written;
|
|
|
|
totalWritten <span class="operator">+</span><span class="operator">=</span> written;
|
|
uploadSpeedData<span class="operator">[</span><span class="number">0</span><span class="operator">]</span> <span class="operator">+</span><span class="operator">=</span> written;
|
|
outgoingBuffer<span class="operator">.</span>remove(<span class="number">0</span><span class="operator">,</span> written);
|
|
} <span class="keyword">while</span> (totalWritten <span class="operator"><</span> bytes <span class="operator">&</span><span class="operator">&</span> (<span class="operator">!</span>outgoingBuffer<span class="operator">.</span>isEmpty() <span class="operator">|</span><span class="operator">|</span> <span class="operator">!</span>pendingBlocks<span class="operator">.</span>isEmpty()));
|
|
|
|
<span class="keyword">return</span> totalWritten;
|
|
}
|
|
|
|
<span class="comment">// Attempts to read at most 'bytes' bytes from the socket.</span>
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>readFromSocket(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> bytes)
|
|
{
|
|
<span class="type">char</span> buffer<span class="operator">[</span><span class="number">1024</span><span class="operator">]</span>;
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> totalRead <span class="operator">=</span> <span class="number">0</span>;
|
|
<span class="keyword">do</span> {
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> bytesRead <span class="operator">=</span> socket<span class="operator">.</span>read(buffer<span class="operator">,</span> <a href="qtglobal.html#qMin">qMin</a><span class="operator"><</span><span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span><span class="operator">></span>(<span class="keyword">sizeof</span>(buffer)<span class="operator">,</span> bytes <span class="operator">-</span> totalRead));
|
|
<span class="keyword">if</span> (bytesRead <span class="operator"><</span><span class="operator">=</span> <span class="number">0</span>)
|
|
<span class="keyword">break</span>;
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> oldSize <span class="operator">=</span> incomingBuffer<span class="operator">.</span>size();
|
|
incomingBuffer<span class="operator">.</span>resize(oldSize <span class="operator">+</span> bytesRead);
|
|
memcpy(incomingBuffer<span class="operator">.</span>data() <span class="operator">+</span> oldSize<span class="operator">,</span> buffer<span class="operator">,</span> bytesRead);
|
|
|
|
totalRead <span class="operator">+</span><span class="operator">=</span> bytesRead;
|
|
} <span class="keyword">while</span> (totalRead <span class="operator"><</span> bytes);
|
|
|
|
<span class="keyword">if</span> (totalRead <span class="operator">></span> <span class="number">0</span>) {
|
|
downloadSpeedData<span class="operator">[</span><span class="number">0</span><span class="operator">]</span> <span class="operator">+</span><span class="operator">=</span> totalRead;
|
|
<span class="keyword">emit</span> bytesReceived(totalRead);
|
|
processIncomingData();
|
|
}
|
|
<span class="keyword">return</span> totalRead;
|
|
}
|
|
|
|
<span class="comment">// Returns the average number of bytes per second this client is</span>
|
|
<span class="comment">// downloading.</span>
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>downloadSpeed() <span class="keyword">const</span>
|
|
{
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> sum <span class="operator">=</span> <span class="number">0</span>;
|
|
<span class="keyword">for</span> (<span class="type">unsigned</span> <span class="type">int</span> i <span class="operator">=</span> <span class="number">0</span>; i <span class="operator"><</span> <span class="keyword">sizeof</span>(downloadSpeedData) <span class="operator">/</span> <span class="keyword">sizeof</span>(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span>); <span class="operator">+</span><span class="operator">+</span>i)
|
|
sum <span class="operator">+</span><span class="operator">=</span> downloadSpeedData<span class="operator">[</span>i<span class="operator">]</span>;
|
|
<span class="keyword">return</span> sum <span class="operator">/</span> (<span class="number">8</span> <span class="operator">*</span> <span class="number">2</span>);
|
|
}
|
|
|
|
<span class="comment">// Returns the average number of bytes per second this client is</span>
|
|
<span class="comment">// uploading.</span>
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>uploadSpeed() <span class="keyword">const</span>
|
|
{
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> sum <span class="operator">=</span> <span class="number">0</span>;
|
|
<span class="keyword">for</span> (<span class="type">unsigned</span> <span class="type">int</span> i <span class="operator">=</span> <span class="number">0</span>; i <span class="operator"><</span> <span class="keyword">sizeof</span>(uploadSpeedData) <span class="operator">/</span> <span class="keyword">sizeof</span>(<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span>); <span class="operator">+</span><span class="operator">+</span>i)
|
|
sum <span class="operator">+</span><span class="operator">=</span> uploadSpeedData<span class="operator">[</span>i<span class="operator">]</span>;
|
|
<span class="keyword">return</span> sum <span class="operator">/</span> (<span class="number">8</span> <span class="operator">*</span> <span class="number">2</span>);
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>setReadBufferSize(<span class="type">int</span> size)
|
|
{
|
|
socket<span class="operator">.</span>setReadBufferSize(size);
|
|
}
|
|
|
|
<span class="type">bool</span> PeerWireClient<span class="operator">::</span>canTransferMore() <span class="keyword">const</span>
|
|
{
|
|
<span class="keyword">return</span> bytesAvailable() <span class="operator">></span> <span class="number">0</span> <span class="operator">|</span><span class="operator">|</span> socket<span class="operator">.</span>bytesAvailable() <span class="operator">></span> <span class="number">0</span>
|
|
<span class="operator">|</span><span class="operator">|</span> <span class="operator">!</span>outgoingBuffer<span class="operator">.</span>isEmpty() <span class="operator">|</span><span class="operator">|</span> <span class="operator">!</span>pendingBlocks<span class="operator">.</span>isEmpty();
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>connectToHostImplementation(<span class="keyword">const</span> <span class="type"><a href="qstring.html">QString</a></span> <span class="operator">&</span>hostName<span class="operator">,</span>
|
|
<span class="type"><a href="qtglobal.html#quint16-typedef">quint16</a></span> port<span class="operator">,</span> OpenMode openMode)
|
|
|
|
{
|
|
setOpenMode(openMode);
|
|
socket<span class="operator">.</span>connectToHost(hostName<span class="operator">,</span> port<span class="operator">,</span> openMode);
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>diconnectFromHostImplementation()
|
|
{
|
|
socket<span class="operator">.</span>disconnectFromHost();
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>timerEvent(<span class="type"><a href="qtimerevent.html">QTimerEvent</a></span> <span class="operator">*</span>event)
|
|
{
|
|
<span class="keyword">if</span> (event<span class="operator">-</span><span class="operator">></span>timerId() <span class="operator">=</span><span class="operator">=</span> transferSpeedTimer) {
|
|
<span class="comment">// Rotate the upload / download records.</span>
|
|
<span class="keyword">for</span> (<span class="type">int</span> i <span class="operator">=</span> <span class="number">6</span>; i <span class="operator">></span><span class="operator">=</span> <span class="number">0</span>; <span class="operator">-</span><span class="operator">-</span>i) {
|
|
uploadSpeedData<span class="operator">[</span>i <span class="operator">+</span> <span class="number">1</span><span class="operator">]</span> <span class="operator">=</span> uploadSpeedData<span class="operator">[</span>i<span class="operator">]</span>;
|
|
downloadSpeedData<span class="operator">[</span>i <span class="operator">+</span> <span class="number">1</span><span class="operator">]</span> <span class="operator">=</span> downloadSpeedData<span class="operator">[</span>i<span class="operator">]</span>;
|
|
}
|
|
uploadSpeedData<span class="operator">[</span><span class="number">0</span><span class="operator">]</span> <span class="operator">=</span> <span class="number">0</span>;
|
|
downloadSpeedData<span class="operator">[</span><span class="number">0</span><span class="operator">]</span> <span class="operator">=</span> <span class="number">0</span>;
|
|
} <span class="keyword">else</span> <span class="keyword">if</span> (event<span class="operator">-</span><span class="operator">></span>timerId() <span class="operator">=</span><span class="operator">=</span> timeoutTimer) {
|
|
<span class="comment">// Disconnect if we timed out; otherwise the timeout is</span>
|
|
<span class="comment">// restarted.</span>
|
|
<span class="keyword">if</span> (invalidateTimeout) {
|
|
invalidateTimeout <span class="operator">=</span> <span class="keyword">false</span>;
|
|
} <span class="keyword">else</span> {
|
|
abort();
|
|
<span class="keyword">emit</span> infoHashReceived(<span class="type"><a href="qbytearray.html">QByteArray</a></span>());
|
|
}
|
|
} <span class="keyword">else</span> <span class="keyword">if</span> (event<span class="operator">-</span><span class="operator">></span>timerId() <span class="operator">=</span><span class="operator">=</span> pendingRequestTimer) {
|
|
abort();
|
|
} <span class="keyword">else</span> <span class="keyword">if</span> (event<span class="operator">-</span><span class="operator">></span>timerId() <span class="operator">=</span><span class="operator">=</span> keepAliveTimer) {
|
|
sendKeepAlive();
|
|
}
|
|
<span class="type"><a href="qtcpsocket.html">QTcpSocket</a></span><span class="operator">::</span>timerEvent(event);
|
|
}
|
|
|
|
<span class="comment">// Sends the handshake to the peer.</span>
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>sendHandShake()
|
|
{
|
|
sentHandShake <span class="operator">=</span> <span class="keyword">true</span>;
|
|
|
|
<span class="comment">// Restart the timeout</span>
|
|
<span class="keyword">if</span> (timeoutTimer)
|
|
killTimer(timeoutTimer);
|
|
timeoutTimer <span class="operator">=</span> startTimer(ClientTimeout);
|
|
|
|
<span class="comment">// Write the 68 byte PeerWire handshake.</span>
|
|
write(<span class="operator">&</span>ProtocolIdSize<span class="operator">,</span> <span class="number">1</span>);
|
|
write(ProtocolId<span class="operator">,</span> ProtocolIdSize);
|
|
write(<span class="type"><a href="qbytearray.html">QByteArray</a></span>(<span class="number">8</span><span class="operator">,</span> <span class="char">'\0'</span>));
|
|
write(infoHash);
|
|
write(peerIdString);
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>processIncomingData()
|
|
{
|
|
invalidateTimeout <span class="operator">=</span> <span class="keyword">true</span>;
|
|
<span class="keyword">if</span> (<span class="operator">!</span>receivedHandShake) {
|
|
<span class="comment">// Check that we received enough data</span>
|
|
<span class="keyword">if</span> (bytesAvailable() <span class="operator"><</span> MinimalHeaderSize)
|
|
<span class="keyword">return</span>;
|
|
|
|
<span class="comment">// Sanity check the protocol ID</span>
|
|
<span class="type"><a href="qbytearray.html">QByteArray</a></span> id <span class="operator">=</span> read(ProtocolIdSize <span class="operator">+</span> <span class="number">1</span>);
|
|
<span class="keyword">if</span> (id<span class="operator">.</span>at(<span class="number">0</span>) <span class="operator">!</span><span class="operator">=</span> ProtocolIdSize <span class="operator">|</span><span class="operator">|</span> <span class="operator">!</span>id<span class="operator">.</span>mid(<span class="number">1</span>)<span class="operator">.</span>startsWith(ProtocolId)) {
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
|
|
<span class="comment">// Discard 8 reserved bytes, then read the info hash and peer ID</span>
|
|
(<span class="type">void</span>) read(<span class="number">8</span>);
|
|
|
|
<span class="comment">// Read infoHash</span>
|
|
<span class="type"><a href="qbytearray.html">QByteArray</a></span> peerInfoHash <span class="operator">=</span> read(<span class="number">20</span>);
|
|
<span class="keyword">if</span> (<span class="operator">!</span>infoHash<span class="operator">.</span>isEmpty() <span class="operator">&</span><span class="operator">&</span> peerInfoHash <span class="operator">!</span><span class="operator">=</span> infoHash) {
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
|
|
<span class="keyword">emit</span> infoHashReceived(peerInfoHash);
|
|
<span class="keyword">if</span> (infoHash<span class="operator">.</span>isEmpty()) {
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
|
|
<span class="comment">// Send handshake</span>
|
|
<span class="keyword">if</span> (<span class="operator">!</span>sentHandShake)
|
|
sendHandShake();
|
|
receivedHandShake <span class="operator">=</span> <span class="keyword">true</span>;
|
|
}
|
|
|
|
<span class="comment">// Handle delayed peer id arrival</span>
|
|
<span class="keyword">if</span> (<span class="operator">!</span>gotPeerId) {
|
|
<span class="keyword">if</span> (bytesAvailable() <span class="operator"><</span> <span class="number">20</span>)
|
|
<span class="keyword">return</span>;
|
|
gotPeerId <span class="operator">=</span> <span class="keyword">true</span>;
|
|
<span class="keyword">if</span> (read(<span class="number">20</span>) <span class="operator">=</span><span class="operator">=</span> peerIdString) {
|
|
<span class="comment">// We connected to ourself</span>
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
}
|
|
|
|
<span class="comment">// Initialize keep-alive timer</span>
|
|
<span class="keyword">if</span> (<span class="operator">!</span>keepAliveTimer)
|
|
keepAliveTimer <span class="operator">=</span> startTimer(KeepAliveInterval);
|
|
|
|
<span class="keyword">do</span> {
|
|
<span class="comment">// Find the packet length</span>
|
|
<span class="keyword">if</span> (nextPacketLength <span class="operator">=</span><span class="operator">=</span> <span class="operator">-</span><span class="number">1</span>) {
|
|
<span class="keyword">if</span> (bytesAvailable() <span class="operator"><</span> <span class="number">4</span>)
|
|
<span class="keyword">return</span>;
|
|
|
|
<span class="type">char</span> tmp<span class="operator">[</span><span class="number">4</span><span class="operator">]</span>;
|
|
read(tmp<span class="operator">,</span> <span class="keyword">sizeof</span>(tmp));
|
|
nextPacketLength <span class="operator">=</span> fromNetworkData(tmp);
|
|
|
|
<span class="keyword">if</span> (nextPacketLength <span class="operator"><</span> <span class="number">0</span> <span class="operator">|</span><span class="operator">|</span> nextPacketLength <span class="operator">></span> <span class="number">200000</span>) {
|
|
<span class="comment">// Prevent DoS</span>
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
}
|
|
|
|
<span class="comment">// KeepAlive</span>
|
|
<span class="keyword">if</span> (nextPacketLength <span class="operator">=</span><span class="operator">=</span> <span class="number">0</span>) {
|
|
nextPacketLength <span class="operator">=</span> <span class="operator">-</span><span class="number">1</span>;
|
|
<span class="keyword">continue</span>;
|
|
}
|
|
|
|
<span class="comment">// Wait with parsing until the whole packet has been received</span>
|
|
<span class="keyword">if</span> (bytesAvailable() <span class="operator"><</span> nextPacketLength)
|
|
<span class="keyword">return</span>;
|
|
|
|
<span class="comment">// Read the packet</span>
|
|
<span class="type"><a href="qbytearray.html">QByteArray</a></span> packet <span class="operator">=</span> read(nextPacketLength);
|
|
<span class="keyword">if</span> (packet<span class="operator">.</span>size() <span class="operator">!</span><span class="operator">=</span> nextPacketLength) {
|
|
abort();
|
|
<span class="keyword">return</span>;
|
|
}
|
|
|
|
<span class="keyword">switch</span> (packet<span class="operator">.</span>at(<span class="number">0</span>)) {
|
|
<span class="keyword">case</span> ChokePacket:
|
|
<span class="comment">// We have been choked.</span>
|
|
pwState <span class="operator">|</span><span class="operator">=</span> ChokedByPeer;
|
|
incoming<span class="operator">.</span>clear();
|
|
<span class="keyword">if</span> (pendingRequestTimer)
|
|
killTimer(pendingRequestTimer);
|
|
<span class="keyword">emit</span> choked();
|
|
<span class="keyword">break</span>;
|
|
<span class="keyword">case</span> UnchokePacket:
|
|
<span class="comment">// We have been unchoked.</span>
|
|
pwState <span class="operator">&</span><span class="operator">=</span> <span class="operator">~</span>ChokedByPeer;
|
|
<span class="keyword">emit</span> unchoked();
|
|
<span class="keyword">break</span>;
|
|
<span class="keyword">case</span> InterestedPacket:
|
|
<span class="comment">// The peer is interested in downloading.</span>
|
|
pwState <span class="operator">|</span><span class="operator">=</span> PeerIsInterested;
|
|
<span class="keyword">emit</span> interested();
|
|
<span class="keyword">break</span>;
|
|
<span class="keyword">case</span> NotInterestedPacket:
|
|
<span class="comment">// The peer is not interested in downloading.</span>
|
|
pwState <span class="operator">&</span><span class="operator">=</span> <span class="operator">~</span>PeerIsInterested;
|
|
<span class="keyword">emit</span> notInterested();
|
|
<span class="keyword">break</span>;
|
|
<span class="keyword">case</span> HavePacket: {
|
|
<span class="comment">// The peer has a new piece available.</span>
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> index <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">1</span><span class="operator">]</span>);
|
|
<span class="keyword">if</span> (index <span class="operator"><</span> <span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span>(peerPieces<span class="operator">.</span>size())) {
|
|
<span class="comment">// Only accept indexes within the valid range.</span>
|
|
peerPieces<span class="operator">.</span>setBit(<span class="type">int</span>(index));
|
|
}
|
|
<span class="keyword">emit</span> piecesAvailable(availablePieces());
|
|
<span class="keyword">break</span>;
|
|
}
|
|
<span class="keyword">case</span> BitFieldPacket:
|
|
<span class="comment">// The peer has the following pieces available.</span>
|
|
<span class="keyword">for</span> (<span class="type">int</span> i <span class="operator">=</span> <span class="number">1</span>; i <span class="operator"><</span> packet<span class="operator">.</span>size(); <span class="operator">+</span><span class="operator">+</span>i) {
|
|
<span class="keyword">for</span> (<span class="type">int</span> bit <span class="operator">=</span> <span class="number">0</span>; bit <span class="operator"><</span> <span class="number">8</span>; <span class="operator">+</span><span class="operator">+</span>bit) {
|
|
<span class="keyword">if</span> (packet<span class="operator">.</span>at(i) <span class="operator">&</span> (<span class="number">1</span> <span class="operator"><</span><span class="operator"><</span> (<span class="number">7</span> <span class="operator">-</span> bit))) {
|
|
<span class="type">int</span> bitIndex <span class="operator">=</span> <span class="type">int</span>(((i <span class="operator">-</span> <span class="number">1</span>) <span class="operator">*</span> <span class="number">8</span>) <span class="operator">+</span> bit);
|
|
<span class="keyword">if</span> (bitIndex <span class="operator">></span><span class="operator">=</span> <span class="number">0</span> <span class="operator">&</span><span class="operator">&</span> bitIndex <span class="operator"><</span> peerPieces<span class="operator">.</span>size()) {
|
|
<span class="comment">// Occasionally, broken clients claim to have</span>
|
|
<span class="comment">// pieces whose index is outside the valid range.</span>
|
|
<span class="comment">// The most common mistake is the index == size</span>
|
|
<span class="comment">// case.</span>
|
|
peerPieces<span class="operator">.</span>setBit(bitIndex);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
<span class="keyword">emit</span> piecesAvailable(availablePieces());
|
|
<span class="keyword">break</span>;
|
|
<span class="keyword">case</span> RequestPacket: {
|
|
<span class="comment">// The peer requests a block.</span>
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> index <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">1</span><span class="operator">]</span>);
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> begin <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">5</span><span class="operator">]</span>);
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> length <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">9</span><span class="operator">]</span>);
|
|
<span class="keyword">emit</span> blockRequested(<span class="type">int</span>(index)<span class="operator">,</span> <span class="type">int</span>(begin)<span class="operator">,</span> <span class="type">int</span>(length));
|
|
<span class="keyword">break</span>;
|
|
}
|
|
<span class="keyword">case</span> PiecePacket: {
|
|
<span class="type">int</span> index <span class="operator">=</span> <span class="type">int</span>(fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">1</span><span class="operator">]</span>));
|
|
<span class="type">int</span> begin <span class="operator">=</span> <span class="type">int</span>(fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">5</span><span class="operator">]</span>));
|
|
|
|
incoming<span class="operator">.</span>removeAll(TorrentBlock(index<span class="operator">,</span> begin<span class="operator">,</span> packet<span class="operator">.</span>size() <span class="operator">-</span> <span class="number">9</span>));
|
|
|
|
<span class="comment">// The peer sends a block.</span>
|
|
<span class="keyword">emit</span> blockReceived(index<span class="operator">,</span> begin<span class="operator">,</span> packet<span class="operator">.</span>mid(<span class="number">9</span>));
|
|
|
|
<span class="comment">// Kill the pending block timer.</span>
|
|
<span class="keyword">if</span> (pendingRequestTimer) {
|
|
killTimer(pendingRequestTimer);
|
|
pendingRequestTimer <span class="operator">=</span> <span class="number">0</span>;
|
|
}
|
|
<span class="keyword">break</span>;
|
|
}
|
|
<span class="keyword">case</span> CancelPacket: {
|
|
<span class="comment">// The peer cancels a block request.</span>
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> index <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">1</span><span class="operator">]</span>);
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> begin <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">5</span><span class="operator">]</span>);
|
|
<span class="type"><a href="qtglobal.html#quint32-typedef">quint32</a></span> length <span class="operator">=</span> fromNetworkData(<span class="operator">&</span>packet<span class="operator">.</span>data()<span class="operator">[</span><span class="number">9</span><span class="operator">]</span>);
|
|
<span class="keyword">for</span> (<span class="type">int</span> i <span class="operator">=</span> <span class="number">0</span>; i <span class="operator"><</span> pendingBlocks<span class="operator">.</span>size(); <span class="operator">+</span><span class="operator">+</span>i) {
|
|
<span class="keyword">const</span> BlockInfo <span class="operator">&</span>blockInfo <span class="operator">=</span> pendingBlocks<span class="operator">.</span>at(i);
|
|
<span class="keyword">if</span> (blockInfo<span class="operator">.</span>pieceIndex <span class="operator">=</span><span class="operator">=</span> <span class="type">int</span>(index)
|
|
<span class="operator">&</span><span class="operator">&</span> blockInfo<span class="operator">.</span>offset <span class="operator">=</span><span class="operator">=</span> <span class="type">int</span>(begin)
|
|
<span class="operator">&</span><span class="operator">&</span> blockInfo<span class="operator">.</span>length <span class="operator">=</span><span class="operator">=</span> <span class="type">int</span>(length)) {
|
|
pendingBlocks<span class="operator">.</span>removeAt(i);
|
|
<span class="keyword">break</span>;
|
|
}
|
|
}
|
|
<span class="keyword">break</span>;
|
|
}
|
|
<span class="keyword">default</span>:
|
|
<span class="comment">// Unsupported packet type; just ignore it.</span>
|
|
<span class="keyword">break</span>;
|
|
}
|
|
nextPacketLength <span class="operator">=</span> <span class="operator">-</span><span class="number">1</span>;
|
|
} <span class="keyword">while</span> (bytesAvailable() <span class="operator">></span> <span class="number">0</span>);
|
|
}
|
|
|
|
<span class="type">void</span> PeerWireClient<span class="operator">::</span>socketStateChanged(<span class="type"><a href="qabstractsocket.html">QAbstractSocket</a></span><span class="operator">::</span>SocketState state)
|
|
{
|
|
setLocalAddress(socket<span class="operator">.</span>localAddress());
|
|
setLocalPort(socket<span class="operator">.</span>localPort());
|
|
setPeerName(socket<span class="operator">.</span>peerName());
|
|
setPeerAddress(socket<span class="operator">.</span>peerAddress());
|
|
setPeerPort(socket<span class="operator">.</span>peerPort());
|
|
setSocketState(state);
|
|
}
|
|
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>readData(<span class="type">char</span> <span class="operator">*</span>data<span class="operator">,</span> <span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> size)
|
|
{
|
|
<span class="type">int</span> n <span class="operator">=</span> <a href="qtglobal.html#qMin">qMin</a><span class="operator"><</span><span class="type">int</span><span class="operator">></span>(size<span class="operator">,</span> incomingBuffer<span class="operator">.</span>size());
|
|
memcpy(data<span class="operator">,</span> incomingBuffer<span class="operator">.</span>constData()<span class="operator">,</span> n);
|
|
incomingBuffer<span class="operator">.</span>remove(<span class="number">0</span><span class="operator">,</span> n);
|
|
<span class="keyword">return</span> n;
|
|
}
|
|
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>readLineData(<span class="type">char</span> <span class="operator">*</span>data<span class="operator">,</span> <span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> maxlen)
|
|
{
|
|
<span class="keyword">return</span> <span class="type"><a href="qiodevice.html">QIODevice</a></span><span class="operator">::</span>readLineData(data<span class="operator">,</span> maxlen);
|
|
}
|
|
|
|
<span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> PeerWireClient<span class="operator">::</span>writeData(<span class="keyword">const</span> <span class="type">char</span> <span class="operator">*</span>data<span class="operator">,</span> <span class="type"><a href="qtglobal.html#qint64-typedef">qint64</a></span> size)
|
|
{
|
|
<span class="type">int</span> oldSize <span class="operator">=</span> outgoingBuffer<span class="operator">.</span>size();
|
|
outgoingBuffer<span class="operator">.</span>resize(oldSize <span class="operator">+</span> size);
|
|
memcpy(outgoingBuffer<span class="operator">.</span>data() <span class="operator">+</span> oldSize<span class="operator">,</span> data<span class="operator">,</span> size);
|
|
<span class="keyword">emit</span> readyToTransfer();
|
|
<span class="keyword">return</span> size;
|
|
}</pre>
|
|
</div>
|
|
<!-- @@@network/torrent/peerwireclient.cpp -->
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div class="ft">
|
|
<span></span>
|
|
</div>
|
|
</div>
|
|
<div class="footer">
|
|
<p>
|
|
<acronym title="Copyright">©</acronym> 2015 The Qt Company Ltd.
|
|
Documentation contributions included herein are the copyrights of
|
|
their respective owners.</p>
|
|
<br />
|
|
<p>
|
|
The documentation provided herein is licensed under the terms of the
|
|
<a href="http://www.gnu.org/licenses/fdl.html">GNU Free Documentation
|
|
License version 1.3</a> as published by the Free Software Foundation.</p>
|
|
<p>
|
|
Documentation sources may be obtained from <a href="http://www.qt-project.org">
|
|
www.qt-project.org</a>.</p>
|
|
<br />
|
|
<p>
|
|
Qt and respective logos are trademarks of The Qt Company Ltd
|
|
in Finland and/or other countries worldwide. All other trademarks are property
|
|
of their respective owners. <a title="Privacy Policy"
|
|
href="http://en.gitorious.org/privacy_policy/">Privacy Policy</a></p>
|
|
</div>
|
|
|
|
<script src="scripts/functions.js" type="text/javascript"></script>
|
|
</body>
|
|
</html>
|