246 lines
8.3 KiB
Python
246 lines
8.3 KiB
Python
import time
|
|
import json
|
|
import os
|
|
import sys
|
|
import re
|
|
import socket
|
|
|
|
from subprocess import call
|
|
from bitcoinrpc.authproxy import AuthServiceProxy
|
|
|
|
|
|
def publish():
|
|
print "* Signing and Publishing..."
|
|
call(" ".join(command_sign_publish), shell=True)
|
|
|
|
|
|
def processNameOp(domain, value, test=False):
|
|
if not value.strip().startswith("{"):
|
|
return False
|
|
try:
|
|
data = json.loads(value)
|
|
except Exception, err:
|
|
print "Json load error: %s" % err
|
|
return False
|
|
if "zeronet" not in data and "map" not in data:
|
|
# Namecoin standard use {"map": { "blog": {"zeronet": "1D..."} }}
|
|
print "No zeronet and no map in ", data.keys()
|
|
return False
|
|
if "map" in data:
|
|
# If subdomains using the Namecoin standard is present, just re-write in the Zeronet way
|
|
# and call the function again
|
|
data_map = data["map"]
|
|
new_value = {}
|
|
for subdomain in data_map:
|
|
if "zeronet" in data_map[subdomain]:
|
|
new_value[subdomain] = data_map[subdomain]["zeronet"]
|
|
if "zeronet" in data and isinstance(data["zeronet"], basestring):
|
|
# {
|
|
# "zeronet":"19rXKeKptSdQ9qt7omwN82smehzTuuq6S9",
|
|
# ....
|
|
# }
|
|
new_value[""] = data["zeronet"]
|
|
if len(new_value) > 0:
|
|
return processNameOp(domain, json.dumps({"zeronet": new_value}), test)
|
|
else:
|
|
return False
|
|
if "zeronet" in data and isinstance(data["zeronet"], basestring):
|
|
# {
|
|
# "zeronet":"19rXKeKptSdQ9qt7omwN82smehzTuuq6S9"
|
|
# } is valid
|
|
return processNameOp(domain, json.dumps({"zeronet": { "": data["zeronet"]}}), test)
|
|
if not isinstance(data["zeronet"], dict):
|
|
print "Not dict: ", data["zeronet"]
|
|
return False
|
|
if not re.match("^[a-z0-9]([a-z0-9-]{0,62}[a-z0-9])?$", domain):
|
|
print "Invalid domain: ", domain
|
|
return False
|
|
|
|
if test:
|
|
return True
|
|
|
|
if "slave" in sys.argv:
|
|
print "Waiting for master update arrive"
|
|
time.sleep(30) # Wait 30 sec to allow master updater
|
|
|
|
# Note: Requires the file data/names.json to exist and contain "{}" to work
|
|
names_raw = open(names_path, "rb").read()
|
|
names = json.loads(names_raw)
|
|
for subdomain, address in data["zeronet"].items():
|
|
subdomain = subdomain.lower()
|
|
address = re.sub("[^A-Za-z0-9]", "", address)
|
|
print subdomain, domain, "->", address
|
|
if subdomain:
|
|
if re.match("^[a-z0-9]([a-z0-9-]{0,62}[a-z0-9])?$", subdomain):
|
|
names["%s.%s.bit" % (subdomain, domain)] = address
|
|
else:
|
|
print "Invalid subdomain:", domain, subdomain
|
|
else:
|
|
names["%s.bit" % domain] = address
|
|
|
|
new_names_raw = json.dumps(names, indent=2, sort_keys=True)
|
|
if new_names_raw != names_raw:
|
|
open(names_path, "wb").write(new_names_raw)
|
|
print "-", domain, "Changed"
|
|
return True
|
|
else:
|
|
print "-", domain, "Not changed"
|
|
return False
|
|
|
|
|
|
def processBlock(block_id, test=False):
|
|
print "Processing block #%s..." % block_id
|
|
s = time.time()
|
|
block_hash = rpc.getblockhash(block_id)
|
|
block = rpc.getblock(block_hash)
|
|
|
|
print "Checking %s tx" % len(block["tx"])
|
|
updated = 0
|
|
for tx in block["tx"]:
|
|
try:
|
|
transaction = rpc.getrawtransaction(tx, 1)
|
|
for vout in transaction.get("vout", []):
|
|
if "scriptPubKey" in vout and "nameOp" in vout["scriptPubKey"] and "name" in vout["scriptPubKey"]["nameOp"]:
|
|
name_op = vout["scriptPubKey"]["nameOp"]
|
|
updated += processNameOp(name_op["name"].replace("d/", ""), name_op["value"], test)
|
|
except Exception, err:
|
|
print "Error processing tx #%s %s" % (tx, err)
|
|
print "Done in %.3fs (updated %s)." % (time.time() - s, updated)
|
|
return updated
|
|
|
|
# Connecting to RPC
|
|
def initRpc(config):
|
|
"""Initialize Namecoin RPC"""
|
|
rpc_data = {
|
|
'connect': '127.0.0.1',
|
|
'port': '8336',
|
|
'user': 'PLACEHOLDER',
|
|
'password': 'PLACEHOLDER',
|
|
'clienttimeout': '900'
|
|
}
|
|
try:
|
|
fptr = open(config, 'r')
|
|
lines = fptr.readlines()
|
|
fptr.close()
|
|
except:
|
|
return None # Or take some other appropriate action
|
|
|
|
for line in lines:
|
|
if not line.startswith('rpc'):
|
|
continue
|
|
key_val = line.split(None, 1)[0]
|
|
(key, val) = key_val.split('=', 1)
|
|
if not key or not val:
|
|
continue
|
|
rpc_data[key[3:]] = val
|
|
|
|
url = 'http://%(user)s:%(password)s@%(connect)s:%(port)s' % rpc_data
|
|
|
|
return url, int(rpc_data['clienttimeout'])
|
|
|
|
# Loading config...
|
|
|
|
# Check whether platform is on windows or linux
|
|
# On linux namecoin is installed under ~/.namecoin, while on on windows it is in %appdata%/Namecoin
|
|
|
|
if sys.platform == "win32":
|
|
namecoin_location = os.getenv('APPDATA') + "/Namecoin/"
|
|
else:
|
|
namecoin_location = os.path.expanduser("~/.namecoin/")
|
|
|
|
config_path = namecoin_location + 'zeroname_config.json'
|
|
if not os.path.isfile(config_path): # Create sample config
|
|
open(config_path, "w").write(
|
|
json.dumps({'site': 'site', 'zeronet_path': '/home/zeronet', 'privatekey': '', 'lastprocessed': 223910}, indent=2)
|
|
)
|
|
print "* Example config written to %s" % config_path
|
|
sys.exit(0)
|
|
|
|
config = json.load(open(config_path))
|
|
names_path = "%s/data/%s/data/names.json" % (config["zeronet_path"], config["site"])
|
|
os.chdir(config["zeronet_path"]) # Change working dir - tells script where Zeronet install is.
|
|
|
|
# Parameters to sign and publish
|
|
command_sign_publish = [sys.executable, "zeronet.py", "siteSign", config["site"], config["privatekey"], "--publish"]
|
|
if sys.platform == 'win32':
|
|
command_sign_publish = ['"%s"' % param for param in command_sign_publish]
|
|
|
|
# Initialize rpc connection
|
|
rpc_auth, rpc_timeout = initRpc(namecoin_location + "namecoin.conf")
|
|
rpc = AuthServiceProxy(rpc_auth, timeout=rpc_timeout)
|
|
|
|
node_version = rpc.getnetworkinfo()['version']
|
|
|
|
while 1:
|
|
try:
|
|
time.sleep(1)
|
|
if node_version < 160000 :
|
|
last_block = int(rpc.getinfo()["blocks"])
|
|
else:
|
|
last_block = int(rpc.getblockchaininfo()["blocks"])
|
|
break # Connection succeeded
|
|
except socket.timeout: # Timeout
|
|
print ".",
|
|
sys.stdout.flush()
|
|
except Exception, err:
|
|
print "Exception", err.__class__, err
|
|
time.sleep(5)
|
|
rpc = AuthServiceProxy(rpc_auth, timeout=rpc_timeout)
|
|
|
|
if not config["lastprocessed"]: # First startup: Start processing from last block
|
|
config["lastprocessed"] = last_block
|
|
|
|
|
|
print "- Testing domain parsing..."
|
|
assert processBlock(223911, test=True) # Testing zeronetwork.bit
|
|
assert processBlock(227052, test=True) # Testing brainwallets.bit
|
|
assert not processBlock(236824, test=True) # Utf8 domain name (invalid should skip)
|
|
assert not processBlock(236752, test=True) # Uppercase domain (invalid should skip)
|
|
assert processBlock(236870, test=True) # Encoded domain (should pass)
|
|
assert processBlock(438317, test=True) # Testing namecoin standard artifaxradio.bit (should pass)
|
|
# sys.exit(0)
|
|
|
|
print "- Parsing skipped blocks..."
|
|
should_publish = False
|
|
for block_id in range(config["lastprocessed"], last_block + 1):
|
|
if processBlock(block_id):
|
|
should_publish = True
|
|
config["lastprocessed"] = last_block
|
|
|
|
if should_publish:
|
|
publish()
|
|
|
|
while 1:
|
|
print "- Waiting for new block"
|
|
sys.stdout.flush()
|
|
while 1:
|
|
try:
|
|
time.sleep(1)
|
|
if node_version < 160000 :
|
|
rpc.waitforblock()
|
|
else:
|
|
rpc.waitfornewblock()
|
|
print "Found"
|
|
break # Block found
|
|
except socket.timeout: # Timeout
|
|
print ".",
|
|
sys.stdout.flush()
|
|
except Exception, err:
|
|
print "Exception", err.__class__, err
|
|
time.sleep(5)
|
|
rpc = AuthServiceProxy(rpc_auth, timeout=rpc_timeout)
|
|
|
|
if node_version < 160000 :
|
|
last_block = int(rpc.getinfo()["blocks"])
|
|
else:
|
|
last_block = int(rpc.getblockchaininfo()["blocks"])
|
|
should_publish = False
|
|
for block_id in range(config["lastprocessed"] + 1, last_block + 1):
|
|
if processBlock(block_id):
|
|
should_publish = True
|
|
|
|
config["lastprocessed"] = last_block
|
|
open(config_path, "w").write(json.dumps(config, indent=2))
|
|
|
|
if should_publish:
|
|
publish()
|