def dump_domain(self, path, max_records=0, min_scan_ids=None):
if min_scan_ids:
cmd = self._make_prune_cmd(min_scan_ids)
else:
cmd = rpc_pb2.Command()
cmd.filepath = path
cmd.max_records = max_records
retv = self._service.DumpDomainToJSON(cmd, 60 * 60)
if retv.status != rpc_pb2.CommandReply.SUCCESS:
raise Exception("ZDB failure: %s" % retv.error)
return retv
def _make_prune_cmd(self, min_scan_ids):
cmd = rpc_pb2.Command()
for (host_port, pretty_protocol, pretty_subprotocol), min_scan_id in \
min_scan_ids.iteritems():
network_port = socket.htons(host_port)
protocol = Protocol.from_pretty_name(pretty_protocol)
subprotocol = Subprotocol.from_pretty_name(pretty_subprotocol)
ak = hoststore_pb2.AnonymousKey(
port=network_port,
protocol=protocol.value,
subprotocol=subprotocol.value,
)
min_id_obj = cmd.min_scan_ids.add()
min_id_obj.key.CopyFrom(ak)
min_id_obj.min_scan_id = min_scan_id
return cmd
def dump_certificates(self,
path,
incremental=False,
max_records=0,
start_prefix=0,
stop_prefix=0,
num_threads=16):
cmd = rpc_pb2.Command(filepath=path,
incremental_dump=False,
max_records=max_records,
start_ip=start_prefix,
stop_ip=stop_prefix,
threads=num_threads)
retv = self._service.DumpCertificatesToJSON(cmd, 8 * 60 * 60)
if retv.status != rpc_pb2.CommandReply.SUCCESS:
raise Exception("ZDB failure: %s" % retv.error)
return retv
def regenerate_certificates(self):
retv = self._service.RegenerateCertificateDeltas(
rpc_pb2.Command(), 10 * 60 * 60)
if retv.status != rpc_pb2.CommandReply.SUCCESS:
raise Exception("ZDB failure: %s" % retv.error)
return retv
def validate_certificates(self, path):
retv = self._service.ValidateCertificates(
rpc_pb2.Command(filepath=path), 600 * 60 * 60)
if retv.status != rpc_pb2.CommandReply.SUCCESS:
raise Exception("ZDB failure: %s" % retv.error)
return retv
def update_ases(self, path):
retv = self._service.UpdateASData(rpc_pb2.Command(filepath=path),
self.TIMEOUT)
if retv.status != rpc_pb2.CommandReply.SUCCESS:
raise Exception("ZDB failure: %s" % retv.error)
return retv
def statistics(self, datastore):
ds = rpc_pb2.Command.Datastore.Value(datastore)
return self._service.Statistics(rpc_pb2.Command(datastore=ds),
self.TIMEOUT)
def status(self):
return self._service.Status(rpc_pb2.Command(), self.TIMEOUT)
def shutdown(self):
return self._service.Shutdown(rpc_pb2.Command(), self.TIMEOUT)