def _hsdir_set_changed(self, is_first_desc):
"""
Return True if the HSDir has changed between the last upload of this
descriptor and the current state of things
"""
from onionbalance.hs_v3.onionbalance import my_onionbalance
# Derive blinding parameter
_, time_period_number = hashring.get_srv_and_time_period(is_first_desc)
blinded_param = my_onionbalance.consensus.get_blinding_param(self._get_identity_pubkey_bytes(),
time_period_number)
# Get blinded key
# TODO: hoho! this is dirty we are poking into internal stem API. We
# should ask atagar to make it public for us! :)
blinded_key = stem.descriptor.hidden_service._blinded_pubkey(self._get_identity_pubkey_bytes(), blinded_param)
# Calculate current responsible HSDirs
try:
responsible_hsdirs = hashring.get_responsible_hsdirs(blinded_key, is_first_desc)
except hashring.EmptyHashRing:
return False
if is_first_desc:
previous_responsible_hsdirs = self.first_descriptor.responsible_hsdirs
else:
previous_responsible_hsdirs = self.second_descriptor.responsible_hsdirs
if set(responsible_hsdirs) != set(previous_responsible_hsdirs):
logger.info("\t HSDir set changed (%s vs %s)",
set(responsible_hsdirs), set(previous_responsible_hsdirs))
return True
else:
logger.info("\t HSDir set remained the same")
return False
def _publish_descriptor(self, is_first_desc):
"""
Attempt to publish descriptor if needed.
If 'is_first_desc' is set then attempt to upload the first descriptor
of the service, otherwise the second.
"""
from onionbalance.hs_v3.onionbalance import my_onionbalance
if not self._should_publish_descriptor_now(is_first_desc):
logger.info("No reason to publish %s descriptor for %s",
"first" if is_first_desc else "second",
self.onion_address)
return
try:
intro_points = self._get_intros_for_desc()
except NotEnoughIntros:
return
# Derive blinding parameter
_, time_period_number = hashring.get_srv_and_time_period(is_first_desc)
blinding_param = my_onionbalance.consensus.get_blinding_param(
self._get_identity_pubkey_bytes(), time_period_number)
try:
desc = descriptor.OBDescriptor(self.onion_address,
self.identity_priv_key,
blinding_param, intro_points,
is_first_desc)
except descriptor.BadDescriptor:
return
logger.info(
"Service %s created %s descriptor (%s intro points) (blinding param: %s) (size: %s bytes). About to publish:",
self.onion_address, "first" if is_first_desc else "second",
len(desc.intro_set), blinding_param.hex(), len(str(desc.v3_desc)))
# When we do a v3 HSPOST on the control port, Tor decodes the
# descriptor and extracts the blinded pubkey to be used when uploading
# the descriptor. So let's do the same to compute the responsible
# HSDirs:
blinded_key = desc.get_blinded_key()
# Calculate responsible HSDirs for our service
try:
responsible_hsdirs = hashring.get_responsible_hsdirs(
blinded_key, is_first_desc)
except hashring.EmptyHashRing:
logger.warning("Can't publish desc with no hash ring. Delaying...")
return
logger.info("Uploading %s descriptor for %s to %s",
"first" if is_first_desc else "second", self.onion_address,
responsible_hsdirs)
# Upload descriptor
self._upload_descriptor(my_onionbalance.controller.controller, desc,
responsible_hsdirs)
# It would be better to set last_upload_ts when an upload succeeds and
# not when an upload is just attempted. Unfortunately the HS_DESC #
# UPLOADED event does not provide information about the service and
# so it can't be used to determine when descriptor upload succeeds
desc.set_last_upload_ts(datetime.datetime.utcnow())
desc.set_responsible_hsdirs(responsible_hsdirs)
# Set the descriptor
if is_first_desc:
self.first_descriptor = desc
else:
self.second_descriptor = desc