def handled_list(ui_, params, stored_cfg):
""" INTERNAL: HACKED"""
if params['FMSREAD'] != 'listall' and params['FMSREAD'] != 'list':
return False
trust_map = None
if params['FMSREAD'] == 'list':
trust_map = stored_cfg.fmsread_trust_map.copy() # paranoid copy
fms_ids = trust_map.keys()
fms_ids.sort()
ui_.status(("Only listing repo USKs from trusted "
+ "FMS IDs:\n %s\n\n") % '\n '.join(fms_ids))
parser = USKNotificationParser(trust_map)
parser.add_default_repos(KNOWN_REPOS)
recv_msgs(get_connection(stored_cfg.defaults['FMS_HOST'],
stored_cfg.defaults['FMS_PORT'],
None),
parser,
stored_cfg.fmsread_groups)
show_table(parser, ui_.status)
return True
def get_uri_from_hash(ui_, dummy, params, stored_cfg):
""" Use FMS to get the URI for a repo hash. """
show_fms_info(ui_, params, stored_cfg)
parser = USKNotificationParser(get_trust_map(ui_, params, stored_cfg))
parser.add_default_repos(KNOWN_REPOS)
ui_.status("Raking through fms messages. This may take a while...\n")
recv_msgs(get_connection(stored_cfg.defaults['FMS_HOST'],
stored_cfg.defaults['FMS_PORT'],
None),
parser,
stored_cfg.fmsread_groups,
None,
True)
target_usk = None
fms_id_map, announce_map, update_map = parser.invert_table()
# Find URI
for usk in announce_map:
if params['FMSREAD_HASH'] == get_usk_hash(usk):
# We don't care who announced. The hash matches.
target_usk = usk
break
if target_usk is None:
raise util.Abort(("No announcement found for [%s]. "
+"Use --uri to set the URI.") %
params['FMSREAD_HASH'])
if params['VERBOSITY'] >= 2:
ui_.status("Found URI announcement:\n%s\n" % target_usk)
trusted_notifiers = stored_cfg.trusted_notifiers(params['FMSREAD_HASH'])
notifiers = {}
for clean_id in update_map[params['FMSREAD_HASH']]:
notifiers[fms_id_map[clean_id]] = (parser.table[clean_id][1]
[params['FMSREAD_HASH']])
fms_ids = notifiers.keys()
fms_ids.sort()
ui_.status("Found Updates:\n")
for fms_id in fms_ids:
if fms_id in trusted_notifiers:
ui_.status(" [trusted]:%i:%s\n" % (notifiers[fms_id], fms_id))
else:
ui_.status(" [untrusted]:%i:%s\n" % (notifiers[fms_id], fms_id))
check_trust_map(ui_, stored_cfg, params['FMSREAD_HASH'],
notifiers, trusted_notifiers)
# Check for updates against the updated trust map.
trusted_notifiers = stored_cfg.trusted_notifiers(params['FMSREAD_HASH'])
for fms_id in fms_ids:
if fms_id in trusted_notifiers:
if (notifiers[fms_id] >
stored_cfg.get_index(params['FMSREAD_HASH'])):
stored_cfg.update_index(params['FMSREAD_HASH'],
notifiers[fms_id])
return target_usk
def execute_fmsread(ui_, params, stored_cfg):
""" Run the fmsread command. """
if handled_trust_cmd(ui_, params, stored_cfg):
return
show_fms_info(ui_, params, stored_cfg)
# Listing announced Repo USKs
if handled_list(ui_, params, stored_cfg):
return
# Updating Repo USK indices for repos which are
# listed in the fmsread_trust_map section of the
# config file.
trust_map = stored_cfg.fmsread_trust_map.copy() # paranoid copy
dump_trust_map(ui_, params, trust_map)
ui_.status("Raking through fms messages. This may take a while...\n")
parser = USKNotificationParser()
recv_msgs(get_connection(stored_cfg.defaults['FMS_HOST'],
stored_cfg.defaults['FMS_PORT'],
None),
parser,
stored_cfg.fmsread_groups)
# IMPORTANT: Must include versions that are in the trust map
# but which we haven't seen before.
full_version_table = stored_cfg.version_table.copy()
for usk_hash in known_hashes(trust_map):
if not usk_hash in full_version_table:
full_version_table[usk_hash] = None # works
changed, untrusted = parser.get_updated(trust_map, full_version_table)
if params['VERBOSITY'] >= 2 and len(untrusted) > 0:
text = 'Skipped untrusted updates:\n'
for usk_hash in untrusted:
text += " %i:%s\n" % (untrusted[usk_hash][0], usk_hash)
text += '\n'
ui_.status(text)
if len(changed) == 0:
ui_.status('No updates found.\n')
return
# Back map to uris ? Can't always do it.
if len(changed) > 0:
text = 'Updates:\n'
for usk_hash in changed:
text += '%s:%i\n' % (usk_hash, changed[usk_hash])
ui_.status(text)
if ((not params['REQUEST_URI'] is None) and
get_usk_hash(params['REQUEST_URI']) in changed):
ui_.status("Current repo has update to index %s.\n" %
changed[get_usk_hash(params['REQUEST_URI'])])
if params['DRYRUN']:
ui_.status('Exiting without saving because --dryrun was set.\n')
return
for usk_hash in changed:
stored_cfg.update_index(usk_hash, changed[usk_hash])
Config.to_file(stored_cfg)
ui_.status('Saved updated indices.\n')
