Projet

Général

Profil

Paste
Télécharger au format
Statistiques
| Branche: | Révision:

root / plugins / rtorrent / rtom_spdd @ 46e2de55

Historique | Voir | Annoter | Télécharger (5,03 ko)

1
#!/usr/bin/perl -w
2
#
3
# xmlrpc based munin plugin for monitoring rtorrent's upload/download speed
4
# prerequisites:
5
#  - rtorrent 0.7.5 or newer compiled with --with-xmlrpc-c
6
# check http://libtorrent.rakshasa.no/wiki/RTorrentXMLRPCGuide for further information
7
#
8
# written by Gabor Hudiczius
9
# web: http://projects.cyla.homeip.net/rtwi/wiki/rTorrentOMeter
10
# email: ghudiczius@gmail.com
11
#
12
# 0.0.0 - 071218
13
#  initial release
14
#
15
# 0.0.1 - 071220
16
#  minor textbugs fixed
17
#
18
# 0.1.0d - 080519
19
#  full rewrite in perl
20
#  support for scgi_port and scgi_local
21
#  configurable via munin env variables
22
#  different ul/dl scale can be set for asymmetric connections
23
#  using get_(up|down)_total, and derive
24
#
25
# 0.2.0 - 080619
26
#  upload and download limit displayed on the graph
27
#
28
#
29
# Parameters:
30
#
31
#	config		required
32
#
33
#
34
# Configurable variables
35
#
36
#	src		"socket" when using scgi_socket, or anything else when using scgi_port
37
#	socket		rTorrent's rpc socket (scgi_local)	- using scgi_local	- needed, when "src" is set to "socket"
38
#	ip		rTorrent's ip address			- using scgi_port	- needed, when "src" is NOT set to "socket"
39
#	port		rTorrent's scgi port (scgi_port)	- using scgi_port	- needed, when "src" is NOT set to "socket"
40
#	category        Change graph category
41
#	diff		"yes" for using bps for upload and Bps for download, or anything else for using Bps for both
42
#
43
#
44
# Configuration example
45
#
46
#	[rtom_spdd]
47
#	user username
48
#	env.src socket
49
#	env.socket /home/user/torrent/.socket/rpc.socket
50
#	env.category Sometext
51
#
52
#	[rtom_spdd]
53
#	env.ip 127.0.0.1
54
#	env.port 5000
55
#
56
#
57
#%# family=auto
58

    
59

    
60
if ( $ARGV[0] and $ARGV[0] eq "config" ) {
61
	my $diff = $ENV{"diff"} || "";
62
	my $category = $ENV{"category"} || "";
63
	print "graph_order down up\n";
64
	print "graph_title rTorrent speeds\n";
65
	print "graph_args --base 1024\n";
66
	print "graph_vlabel Bytes per \${graph_period}\n";
67
	print "graph_category filetransfer".${category}."\n";
68
	print "down.label Download B/s\n";
69
	print "down.info Download speed in Bytes per seconds\n";
70
	print "down.type DERIVE\n";
71
	print "down.min 0\n";
72
	print "down.draw AREA\n";
73
	if ( ( defined $diff ) && ( $diff eq "yes" ) ) {
74
		print "up.label Upload b/s\n";
75
		print "up.info Upload speed in bits per seconds\n";
76
		print "up.cdef up,8,*\n";
77
	} else {
78
		print "up.label Upload B/s\n";
79
		print "up.info Upload speed in Bytes per seconds\n";
80
	}
81
	print "up.type DERIVE\n";
82
	print "up.min 0\n";
83
	print "up.draw LINE2\n";
84
	print "downrate.label Download limit B/s\n";
85
	print "downrate.info Download limit in Bytes per seconds\n";
86
	print "downrate.draw LINE2\n";
87
	if ( ( defined $diff ) && ( $diff eq "yes" ) ) {
88
		print "uprate.label Upload limit b/s\n";
89
		print "uprate.info Upload limit in bits per seconds\n";
90
		print "uprate.cdef uprate,8,*\n";
91
	} else {
92
		print "uprate.label Upload limit B/s\n";
93
		print "uprate.info Upload limit in Bytes per seconds\n";
94
	}
95
	print "uprate.draw LINE2\n";
96
	exit 0;
97
}
98

    
99
use IO::Socket;
100

    
101
my $src 	= $ENV{"src"} || "";
102
my $ip		= $ENV{"ip"} || "127.0.0.1";
103
my $port	= $ENV{"port"} || "5000";
104
my $socket	= $ENV{"socket"} || "";
105

    
106
my $pattern	= qr/<value><(int|i4|i8|ex\.i8)>([-]{0,1}\d+)<\/(int|i4|i8|ex\.i8)><\/value>/;
107

    
108
my $line	= "<?xml version=\"1.0\" encoding=\"utf-8\"?><methodCall><methodName>system.multicall</methodName><params><param><value><array><data><value><struct><member><name>methodName</name><value><string>get_up_total</string></value></member><member><name>params</name><value><array><data/></array></value></member></struct></value><value><struct><member><name>methodName</name><value><string>get_down_total</string></value></member><member><name>params</name><value><array><data/></array></value></member></struct></value><value><struct><member><name>methodName</name><value><string>get_upload_rate</string></value></member><member><name>params</name><value><array><data/></array></value></member></struct></value><value><struct><member><name>methodName</name><value><string>get_download_rate</string></value></member><member><name>params</name><value><array><data/></array></value></member></struct></value></data></array></value></param></params></methodCall>";
109
my $llen	= length $line;
110
my $header	= "CONTENT_LENGTH\000${llen}\000SCGI\001\000";
111
my $hlen	= length $header;
112
$line		= "${hlen}:${header},${line}";
113

    
114
if ( ( defined $src ) && ( $src eq "socket" ) ) {
115
	socket( SOCK, PF_UNIX, SOCK_STREAM, 0 );
116
	connect( SOCK, sockaddr_un( $socket ) );
117
} else {
118
	socket( SOCK, PF_INET, SOCK_STREAM, getprotobyname( "tcp" ) );
119
	connect( SOCK, sockaddr_in( $port, inet_aton( $ip ) ) );
120
}
121

    
122
print SOCK $line;
123
flush SOCK;
124

    
125
my $up = -1;
126
my $down = -1;
127
my $uprate = -1;
128
my $downrate = -1;
129
while ( ( $up == -1 ) && ( $line = <SOCK> ) ) {
130
	if ( $line =~ /$pattern/ ) {
131
		$up = $2;
132
	}
133
}
134
while ( ( $down == -1 ) && ( $line = <SOCK> ) ) {
135
	if ( $line =~ /$pattern/ ) {
136
		$down = $2;
137
	}
138
}
139
while ( ( $uprate == -1 ) && ( $line = <SOCK> ) ) {
140
	if ( $line =~ /$pattern/ ) {
141
		$uprate = $2;
142
	}
143
}
144
while ( ( $downrate == -1 ) && ( $line = <SOCK> ) ) {
145
	if ( $line =~ /$pattern/ ) {
146
		$downrate = $2;
147
	}
148
}
149
close (SOCK);
150

    
151
print "up.value ${up}\ndown.value ${down}\nuprate.value ${uprate}\ndownrate.value ${downrate}\n";
152

    
153
exit;