#!/usr/bin/env python # bitcoind_ Munin plugin for Bitcoin Server Variables # # by Mike Koss # Feb 14, 2012, MIT License # # You need to be able to authenticate to the bitcoind server to issue rpc's. # This plugin supporst 2 ways to do that: # # 1) In /etc/munin/plugin-conf.d/bitcoin.conf place: # # [bitcoind_*] # user your-username # # Then be sure your $HOME/.bitcoin/bitcoin.conf has the correct authentication info: # rpcconnect, rpcport, rpcuser, rpcpassword # # 2) Place your bitcoind authentication directly in /etc/munin/plugin-conf.d/bitcoin.conf # # [bitcoind_*] # env.rpcport 8332 # env.rpcconnect 127.0.0.1 # env.rpcuser your-username-here # env.rpcpassword your-password-here # # To install all available graphs: # # sudo munin-node-configure --libdir=. --suggest --shell | sudo bash # # Leave out the "| bash" to get a list of commands you can select from to install # individual graphs. # # Munin plugin tags: # #%# family=auto #%# capabilities=autoconf suggest import os import sys import time import re import urllib2 import json DEBUG = False def main(): # getinfo variable is read from command name - probably the sym-link name. request_var = sys.argv[0].split('_', 1)[1] or 'balance' command = sys.argv[1] if len(sys.argv) > 1 else None request_labels = {'balance': ('Wallet Balance', 'BTC'), 'connections': ('Peer Connections', 'Connections'), 'fees': ("Tip Offered", "BTC"), 'transactions': ("Transactions", "Transactions", ('confirmed', 'waiting')), 'block_age': ("Last Block Age", "Seconds"), 'difficulty': ("Difficulty", ""), } labels = request_labels[request_var] if len(labels) < 3: line_labels = [request_var] else: line_labels = labels[2] if command == 'suggest': for var_name in request_labels.keys(): print var_name return if command == 'config': print 'graph_category bitcoin' print 'graph_title Bitcoin %s' % labels[0] print 'graph_vlabel %s' % labels[1] for label in line_labels: print '%s.label %s' % (label, label) return # Munin should send connection options via environment vars bitcoin_options = get_env_options('rpcconnect', 'rpcport', 'rpcuser', 'rpcpassword') bitcoin_options.rpcconnect = bitcoin_options.get('rpcconnect', '127.0.0.1') bitcoin_options.rpcport = bitcoin_options.get('rpcport', '8332') if bitcoin_options.get('rpcuser') is None: conf_file = os.path.join(os.path.expanduser('~/.bitcoin'), 'bitcoin.conf') bitcoin_options = parse_conf(conf_file) bitcoin_options.require('rpcuser', 'rpcpassword') bitcoin = ServiceProxy('http://%s:%s' % (bitcoin_options.rpcconnect, bitcoin_options.rpcport), username=bitcoin_options.rpcuser, password=bitcoin_options.rpcpassword) (info, error) = bitcoin.getinfo() if error: if command == 'autoconf': print 'no' return else: # TODO: Better way to report errors to Munin-node. raise ValueError("Could not connect to Bitcoin server.") if request_var in ('transactions', 'block_age'): (info, error) = bitcoin.getblockhash(info['blocks']) (info, error) = bitcoin.getblock(info) info['block_age'] = int(time.time()) - info['time'] info['confirmed'] = len(info['tx']) if request_var in ('fees', 'transactions'): (memory_pool, error) = bitcoin.getrawmempool() if memory_pool: info['waiting'] = len(memory_pool) if command == 'autoconf': print 'yes' return for label in line_labels: print "%s.value %s" % (label, info[label]) def parse_conf(filename): """ Bitcoin config file parser. """ options = Options() re_line = re.compile(r'^\s*([^#]*)\s*(#.*)?$') re_setting = re.compile(r'^(.*)\s*=\s*(.*)$') try: with open(filename) as file: for line in file.readlines(): line = re_line.match(line).group(1).strip() m = re_setting.match(line) if m is None: continue (var, value) = (m.group(1), m.group(2).strip()) options[var] = value except: pass return options def get_env_options(*vars): options = Options() for var in vars: options[var] = os.getenv(var) return options class Options(dict): """A dict that allows for object-like property access syntax.""" def __getattr__(self, name): try: return self[name] except KeyError: raise AttributeError(name) def require(self, *names): missing = [] for name in names: if self.get(name) is None: missing.append(name) if len(missing) > 0: raise ValueError("Missing required setting%s: %s." % ('s' if len(missing) > 1 else '', ', '.join(missing))) class ServiceProxy(object): """ Proxy for a JSON-RPC web service. Calls to a function attribute generates a JSON-RPC call to the host service. If a callback keyword arg is included, the call is processed as an asynchronous request. Each call returns (result, error) tuple. """ def __init__(self, url, username=None, password=None): self.url = url self.id = 0 self.username = username self.password = password def __getattr__(self, method): self.id += 1 return Proxy(self, method, id=self.id) class Proxy(object): def __init__(self, service, method, id=None): self.service = service self.method = method self.id = id def __call__(self, *args): if DEBUG: arg_strings = [json.dumps(arg) for arg in args] print "Calling %s(%s) @ %s" % (self.method, ', '.join(arg_strings), self.service.url) data = { 'method': self.method, 'params': args, 'id': self.id, } request = urllib2.Request(self.service.url, json.dumps(data)) if self.service.username: # Strip the newline from the b64 encoding! b64 = ('%s:%s' % (self.service.username, self.service.password)).encode('base64')[:-1] request.add_header('Authorization', 'Basic %s' % b64) try: body = urllib2.urlopen(request).read() except Exception, e: return (None, e) if DEBUG: print 'RPC Response (%s): %s' % (self.method, json.dumps(body, indent=4)) try: data = json.loads(body) except ValueError, e: return (None, e.message) # TODO: Check that id matches? return (data['result'], data['error']) def get_json_url(url): request = urllib2.Request(url) body = urllib2.urlopen(request).read() data = json.loads(body) return data if __name__ == "__main__": main()