1 files changed, 97 insertions, 0 deletions
diff --git a/BitTorrent/Uploader.py b/BitTorrent/Uploader.py
new file mode 100644
index 0000000..d02bfa8
--- /dev/null
+++ b/BitTorrent/Uploader.py
@@ -0,0 +1,97 @@
+# The contents of this file are subject to the BitTorrent Open Source License
+# Version 1.1 (the License).  You may not copy or use this file, in either
+# source code or executable form, except in compliance with the License.  You
+# may obtain a copy of the License at http://www.bittorrent.com/license/.
+#
+# Software distributed under the License is distributed on an AS IS basis,
+# WITHOUT WARRANTY OF ANY KIND, either express or implied.  See the License
+# for the specific language governing rights and limitations under the
+# License.
+
+# Written by Bram Cohen
+
+from BitTorrent.CurrentRateMeasure import Measure
+
+
+class Upload(object):
+
+    def __init__(self, connection, ratelimiter, totalup, totalup2, choker,
+                 storage, max_slice_length, max_rate_period):
+        self.connection = connection
+        self.ratelimiter = ratelimiter
+        self.totalup = totalup
+        self.totalup2 = totalup2
+        self.choker = choker
+        self.storage = storage
+        self.max_slice_length = max_slice_length
+        self.max_rate_period = max_rate_period
+        self.choked = True
+        self.unchoke_time = None
+        self.interested = False
+        self.buffer = []
+        self.measure = Measure(max_rate_period)
+        if storage.do_I_have_anything():
+            connection.send_bitfield(storage.get_have_list())
+
+    def got_not_interested(self):
+        if self.interested:
+            self.interested = False
+            del self.buffer[:]
+            self.choker.not_interested(self.connection)
+
+    def got_interested(self):
+        if not self.interested:
+            self.interested = True
+            self.choker.interested(self.connection)
+
+    def get_upload_chunk(self):
+        if not self.buffer:
+            return None
+        index, begin, length = self.buffer.pop(0)
+        piece = self.storage.get_piece(index, begin, length)
+        if piece is None:
+            self.connection.close()
+            return None
+        return (index, begin, piece)
+
+    def update_rate(self, bytes):
+        self.measure.update_rate(bytes)
+        self.totalup.update_rate(bytes)
+        self.totalup2.update_rate(bytes)
+        
+    def got_request(self, index, begin, length):
+        if not self.interested or length > self.max_slice_length:
+            self.connection.close()
+            return
+        if not self.connection.choke_sent:
+            self.buffer.append((index, begin, length))
+            if self.connection.next_upload is None and \
+                   self.connection.connection.is_flushed():
+                self.ratelimiter.queue(self.connection, self.connection.encoder.context.rlgroup)
+
+    def got_cancel(self, index, begin, length):
+        try:
+            self.buffer.remove((index, begin, length))
+        except ValueError:
+            pass
+
+    def choke(self):
+        if not self.choked:
+            self.choked = True
+            self.connection.send_choke()
+
+    def sent_choke(self):
+        assert self.choked
+        del self.buffer[:]
+
+    def unchoke(self, time):
+        if self.choked:
+            self.choked = False
+            self.unchoke_time = time
+            self.connection.send_unchoke()
+
+    def has_queries(self):
+        return len(self.buffer) > 0
+
+    def get_rate(self):
+        return self.measure.get_rate()