Files
stacks-puppet-node/blockstack/lib/scripts.py
Jude Nelson c6c75e0646 typo
2018-02-12 19:34:18 -05:00

229 lines
6.2 KiB
Python

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Blockstack
~~~~~
copyright: (c) 2014-2015 by Halfmoon Labs, Inc.
copyright: (c) 2016 by Blockstack.org
This file is part of Blockstack
Blockstack is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Blockstack is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with Blockstack. If not, see <http://www.gnu.org/licenses/>.
"""
import virtualchain
log = virtualchain.get_logger("blockstack-server")
from .config import *
from .b40 import *
from .schemas import *
def is_name_valid(fqn):
"""
Is a fully-qualified name acceptable?
Return True if so
Return False if not
"""
if fqn.count( "." ) != 1:
return False
name, namespace_id = fqn.split(".")
if len(name) == 0 or len(namespace_id) == 0:
return False
if not is_b40( name ) or "+" in name or "." in name:
return False
if not is_namespace_valid( namespace_id ):
return False
if len(fqn) > LENGTHS['blockchain_id_name']:
# too long
return False
return True
def is_namespace_valid( namespace_id ):
"""
Is a namespace ID valid?
"""
if not is_b40( namespace_id ) or "+" in namespace_id or namespace_id.count(".") > 0:
return False
if len(namespace_id) == 0 or len(namespace_id) > LENGTHS['blockchain_id_namespace_id']:
return False
return True
def get_namespace_from_name( name ):
"""
Get a fully-qualified name's namespace, if it has one.
It's the sequence of characters after the last "." in the name.
If there is no "." in the name, then it belongs to the null
namespace (i.e. the empty string will be returned)
"""
if "." not in name:
# empty namespace
return ""
return name.split(".")[-1]
def get_name_from_fq_name( name ):
"""
Given a fully-qualified name, get the name part.
It's the sequence of characters before the last "." in the name.
Return None if malformed
"""
if "." not in name:
# malformed
return None
return name.split(".")[0]
def is_address_subdomain(fqa):
"""
Tests whether fqa is a fully-qualified subdomain name
@fqa must be a string
If it isn't, returns False, None, None.
If it is, returns True and a tuple (subdomain_name, domain)
"""
# do these checks early to avoid pathological names that make re.match take forever
if fqa.count(".") != 2:
return False, None, None
grp = re.match(OP_SUBDOMAIN_NAME_PATTERN, fqa)
if grp is None:
return False, None, None
subdomain_name, domain = grp.groups()
return True, subdomain_name, domain
def is_subdomain(fqn):
"""
Short-hand of is_address_subdomain(), but only returns True/False
"""
return is_address_subdomain(fqn)[0]
def price_name( name, namespace, block_height ):
"""
Calculate the price of a name (without its namespace ID), given the
namespace parameters.
The minimum price is NAME_COST_UNIT
"""
base = namespace['base']
coeff = namespace['coeff']
buckets = namespace['buckets']
bucket_exponent = 0
discount = 1.0
if len(name) < len(buckets):
bucket_exponent = buckets[len(name)-1]
else:
bucket_exponent = buckets[-1]
# no vowel discount?
if sum( [name.lower().count(v) for v in ["a", "e", "i", "o", "u", "y"]] ) == 0:
# no vowels!
discount = max( discount, namespace['no_vowel_discount'] )
# non-alpha discount?
if sum( [name.lower().count(v) for v in ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "-", "_"]] ) > 0:
# non-alpha!
discount = max( discount, namespace['nonalpha_discount'] )
price = (float(coeff * (base ** bucket_exponent)) / float(discount)) * NAME_COST_UNIT
if price < NAME_COST_UNIT:
price = NAME_COST_UNIT
price_multiplier = get_epoch_price_multiplier( block_height, namespace['namespace_id'] )
return price * price_multiplier
def price_namespace( namespace_id, block_height ):
"""
Calculate the cost of a namespace.
"""
price_table = get_epoch_namespace_prices( block_height )
if len(namespace_id) >= len(price_table):
return price_table[0]
return price_table[len(namespace_id)]
def find_by_opcode( checked_ops, opcode ):
"""
Given all previously-accepted operations in this block,
find the ones that are of a particular opcode.
@opcode can be one opcode, or a list of opcodes
"""
if type(opcode) != list:
opcode = [opcode]
ret = []
for opdata in checked_ops:
if op_get_opcode_name(opdata['op']) in opcode:
ret.append(opdata)
return ret
def get_public_key_hex_from_tx( inputs, address ):
"""
Given a list of inputs and the address of one of the inputs,
find the public key.
This only works for p2pkh scripts.
We only really need this for NAMESPACE_REVEAL, but we included
it in other transactions' consensus data for legacy reasons that
now have to be supported forever :(
"""
ret = None
for inp in inputs:
input_scriptsig = inp['script']
input_script_code = virtualchain.btc_script_deserialize(input_scriptsig)
if len(input_script_code) == 2:
# signature pubkey
pubkey_candidate = input_script_code[1]
pubkey = None
try:
pubkey = virtualchain.BitcoinPublicKey(pubkey_candidate)
except Exception as e:
traceback.print_exc()
log.warn("Invalid public key {}".format(pubkey_candidate))
continue
if address != pubkey.address():
continue
# success!
return pubkey_candidate
return None