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
117
118
119
120
121
122
|
#!/usr/local/bin/python
# $Id: conn.py,v 1.3 2008-03-09 14:33:00 slander Exp $
# Copyright (c) 2007 Niall O'Higgins <niallo@unworkable.org>
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
import getopt
import socket
import sys
import threading
import time
HOST = 'localhost'
PORT = 6099
def usage():
print >> sys.stderr, "%s: [-h host] [-p port]" %(sys.argv[0])
os._exit(1)
def status(ctl):
print "peers: %s pieces: %s/%s downloaded: %s" %(len(ctl.peers), len(ctl.pieces), ctl.num_pieces, ctl.torrent_bytes + ctl.bytes)
class CTLconnection(threading.Thread):
''' Connection to the control server, runs in its own thread '''
def __init__(self, host, port):
self.host = host
self.port = port
self.num_peers = 0
self.num_pieces = 0
self.torrent_size = 0
self.torrent_bytes = 0
self.pieces = []
self.peers = []
self.bytes = 0
self.done = False
self._socket = None
self._f = None
threading.Thread.__init__(self)
def stop(self):
self._f.close()
self._done = True
def run(self):
try:
self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._socket.connect_ex((self.host, self.port))
self._f = self._socket.makefile()
except socket.error, e:
#ignore all socket errors
pass
try:
for l in self._f:
try:
d = l.strip().split(':', 1)
except:
# ignore malformed line
continue
if d[0] == 'num_peers':
if not isinstance(d[1], int):
continue
self.num_peers = int(d[1])
elif d[0] == 'num_pieces':
self.num_pieces = int(d[1])
elif d[0] == 'torrent_size':
self.torrent_size = int(d[1])
elif d[0] == 'torrent_bytes':
self.torrent_bytes = int(d[1])
elif d[0] == 'pieces':
try:
new_pieces = d[1].split(',')
new_pieces.sort()
self.pieces = new_pieces
except:
# no pieces yet
continue
elif d[0] == 'bytes':
self.bytes = int(d[1])
elif d[0] == 'peers':
try:
new_peers = d[1].split(',')
new_peers.sort()
self.peers = new_peers
except:
# no peers yet
continue
else:
print "unkown message: %s" %(l)
except socket.error, e:
#ignore all socket errors sleep for 5 seconds and run again
time.sleep(5)
self.run()
self._f.close()
self.done = True
try:
opts, args = getopt.getopt(sys.argv[1:], "h:p:")
except getopt.GetoptError:
usage()
for o, a in opts:
if o == "-h":
HOST = a
if o == "-p":
PORT = int(a)
ctl = CTLconnection(HOST, PORT)
ctl.start()
while not ctl.done:
status(ctl)
time.sleep(5)
|
