# do not import any allmydata modules at this level. Do that from inside
# individual functions instead.
-import struct, time, os
+import struct, time, os, sys
from twisted.python import usage, failure
from twisted.internet import defer
+from twisted.scripts import trial as twisted_trial
+from foolscap.logging import cli as foolscap_cli
+from allmydata.scripts.common import BaseOptions
-class DumpOptions(usage.Options):
+
+class DumpOptions(BaseOptions):
def getSynopsis(self):
- return "Usage: tahoe debug dump-share SHARE_FILENAME"
+ return "Usage: tahoe [global-opts] debug dump-share SHARE_FILENAME"
optFlags = [
- ["offsets", None, "Display a table of section offsets"],
+ ["offsets", None, "Display a table of section offsets."],
+ ["leases-only", None, "Dump leases but not CHK contents."],
]
def getUsage(self, width=None):
- t = usage.Options.getUsage(self, width)
+ t = BaseOptions.getUsage(self, width)
t += """
Print lots of information about the given share, by parsing the share's
contents. This includes share type, lease information, encoding parameters,
return t
def parseArgs(self, filename):
- self['filename'] = filename
+ from allmydata.util.encodingutil import argv_to_abspath
+ self['filename'] = argv_to_abspath(filename)
def dump_share(options):
- from allmydata import storage
+ from allmydata.storage.mutable import MutableShareFile
+ from allmydata.util.encodingutil import quote_output
out = options.stdout
# check the version, to see if we have a mutable or immutable share
- print >>out, "share filename: %s" % options['filename']
+ print >>out, "share filename: %s" % quote_output(options['filename'])
f = open(options['filename'], "rb")
prefix = f.read(32)
f.close()
- if prefix == storage.MutableShareFile.MAGIC:
+ if prefix == MutableShareFile.MAGIC:
return dump_mutable_share(options)
# otherwise assume it's immutable
return dump_immutable_share(options)
def dump_immutable_share(options):
- from allmydata import uri, storage
+ from allmydata.storage.immutable import ShareFile
+
+ out = options.stdout
+ f = ShareFile(options['filename'])
+ if not options["leases-only"]:
+ dump_immutable_chk_share(f, out, options)
+ dump_immutable_lease_info(f, out)
+ print >>out
+ return 0
+
+def dump_immutable_chk_share(f, out, options):
+ from allmydata import uri
from allmydata.util import base32
from allmydata.immutable.layout import ReadBucketProxy
+ from allmydata.util.encodingutil import quote_output, to_str
- out = options.stdout
- f = storage.ShareFile(options['filename'])
# use a ReadBucketProxy to parse the bucket and find the uri extension
- bp = ReadBucketProxy(None)
+ bp = ReadBucketProxy(None, None, '')
offsets = bp._parse_offsets(f.read_share_data(0, 0x44))
print >>out, "%20s: %d" % ("version", bp._version)
seek = offsets['uri_extension']
# the storage index isn't stored in the share itself, so we depend upon
# knowing the parent directory name to get it
pieces = options['filename'].split(os.sep)
- if len(pieces) >= 2 and base32.could_be_base32_encoded(pieces[-2]):
- storage_index = base32.a2b(pieces[-2])
- uri_extension_hash = base32.a2b(unpacked["UEB_hash"])
- u = uri.CHKFileVerifierURI(storage_index, uri_extension_hash,
- unpacked["needed_shares"],
- unpacked["total_shares"], unpacked["size"])
- verify_cap = u.to_string()
- print >>out, "%20s: %s" % ("verify-cap", verify_cap)
+ if len(pieces) >= 2:
+ piece = to_str(pieces[-2])
+ if base32.could_be_base32_encoded(piece):
+ storage_index = base32.a2b(piece)
+ uri_extension_hash = base32.a2b(unpacked["UEB_hash"])
+ u = uri.CHKFileVerifierURI(storage_index, uri_extension_hash,
+ unpacked["needed_shares"],
+ unpacked["total_shares"], unpacked["size"])
+ verify_cap = u.to_string()
+ print >>out, "%20s: %s" % ("verify-cap", quote_output(verify_cap, quotemarks=False))
sizes = {}
- sizes['data'] = bp._data_size
+ sizes['data'] = (offsets['plaintext_hash_tree'] -
+ offsets['data'])
sizes['validation'] = (offsets['uri_extension'] -
offsets['plaintext_hash_tree'])
sizes['uri-extension'] = len(UEB_data)
print >>out, " %20s: %s (0x%x)" % (name, offset, offset)
print >>out, "%20s: %s" % ("leases", f._lease_offset)
-
+def dump_immutable_lease_info(f, out):
# display lease information too
print >>out
- leases = list(f.iter_leases())
+ leases = list(f.get_leases())
if leases:
for i,lease in enumerate(leases):
when = format_expiration_time(lease.expiration_time)
else:
print >>out, " No leases."
- print >>out
- return 0
-
def format_expiration_time(expiration_time):
now = time.time()
remains = expiration_time - now
def dump_mutable_share(options):
- from allmydata import storage
+ from allmydata.storage.mutable import MutableShareFile
from allmydata.util import base32, idlib
out = options.stdout
- m = storage.MutableShareFile(options['filename'])
+ m = MutableShareFile(options['filename'])
f = open(options['filename'], "rb")
WE, nodeid = m._read_write_enabler_and_nodeid(f)
num_extra_leases = m._read_num_extra_leases(f)
share_type = "unknown"
f.seek(m.DATA_OFFSET)
- if f.read(1) == "\x00":
+ version = f.read(1)
+ if version == "\x00":
# this slot contains an SMDF share
share_type = "SDMF"
+ elif version == "\x01":
+ share_type = "MDMF"
f.close()
print >>out
if share_type == "SDMF":
dump_SDMF_share(m, data_length, options)
+ elif share_type == "MDMF":
+ dump_MDMF_share(m, data_length, options)
return 0
from allmydata.mutable.common import NeedMoreDataError
from allmydata.util import base32, hashutil
from allmydata.uri import SSKVerifierURI
+ from allmydata.util.encodingutil import quote_output, to_str
offset = m.DATA_OFFSET
# the storage index isn't stored in the share itself, so we depend upon
# knowing the parent directory name to get it
pieces = options['filename'].split(os.sep)
- if len(pieces) >= 2 and base32.could_be_base32_encoded(pieces[-2]):
- storage_index = base32.a2b(pieces[-2])
- fingerprint = hashutil.ssk_pubkey_fingerprint_hash(pubkey)
- u = SSKVerifierURI(storage_index, fingerprint)
- verify_cap = u.to_string()
- print >>out, " verify-cap:", verify_cap
+ if len(pieces) >= 2:
+ piece = to_str(pieces[-2])
+ if base32.could_be_base32_encoded(piece):
+ storage_index = base32.a2b(piece)
+ fingerprint = hashutil.ssk_pubkey_fingerprint_hash(pubkey)
+ u = SSKVerifierURI(storage_index, fingerprint)
+ verify_cap = u.to_string()
+ print >>out, " verify-cap:", quote_output(verify_cap, quotemarks=False)
if options['offsets']:
# NOTE: this offset-calculation code is fragile, and needs to be
print >>out
+def dump_MDMF_share(m, length, options):
+ from allmydata.mutable.layout import MDMFSlotReadProxy
+ from allmydata.util import base32, hashutil
+ from allmydata.uri import MDMFVerifierURI
+ from allmydata.util.encodingutil import quote_output, to_str
+ offset = m.DATA_OFFSET
+ out = options.stdout
+
+ f = open(options['filename'], "rb")
+ storage_index = None; shnum = 0
+
+ class ShareDumper(MDMFSlotReadProxy):
+ def _read(self, readvs, force_remote=False, queue=False):
+ data = []
+ for (where,length) in readvs:
+ f.seek(offset+where)
+ data.append(f.read(length))
+ return defer.succeed({shnum: data})
+
+ p = ShareDumper(None, storage_index, shnum)
+ def extract(func):
+ stash = []
+ # these methods return Deferreds, but we happen to know that they run
+ # synchronously when not actually talking to a remote server
+ d = func()
+ d.addCallback(stash.append)
+ return stash[0]
+
+ verinfo = extract(p.get_verinfo)
+ encprivkey = extract(p.get_encprivkey)
+ signature = extract(p.get_signature)
+ pubkey = extract(p.get_verification_key)
+ block_hash_tree = extract(p.get_blockhashes)
+ share_hash_chain = extract(p.get_sharehashes)
+ f.close()
+
+ (seqnum, root_hash, salt_to_use, segsize, datalen, k, N, prefix,
+ offsets) = verinfo
+
+ print >>out, " MDMF contents:"
+ print >>out, " seqnum: %d" % seqnum
+ print >>out, " root_hash: %s" % base32.b2a(root_hash)
+ #print >>out, " IV: %s" % base32.b2a(IV)
+ print >>out, " required_shares: %d" % k
+ print >>out, " total_shares: %d" % N
+ print >>out, " segsize: %d" % segsize
+ print >>out, " datalen: %d" % datalen
+ print >>out, " enc_privkey: %d bytes" % len(encprivkey)
+ print >>out, " pubkey: %d bytes" % len(pubkey)
+ print >>out, " signature: %d bytes" % len(signature)
+ share_hash_ids = ",".join([str(hid)
+ for hid in sorted(share_hash_chain.keys())])
+ print >>out, " share_hash_chain: %s" % share_hash_ids
+ print >>out, " block_hash_tree: %d nodes" % len(block_hash_tree)
+
+ # the storage index isn't stored in the share itself, so we depend upon
+ # knowing the parent directory name to get it
+ pieces = options['filename'].split(os.sep)
+ if len(pieces) >= 2:
+ piece = to_str(pieces[-2])
+ if base32.could_be_base32_encoded(piece):
+ storage_index = base32.a2b(piece)
+ fingerprint = hashutil.ssk_pubkey_fingerprint_hash(pubkey)
+ u = MDMFVerifierURI(storage_index, fingerprint)
+ verify_cap = u.to_string()
+ print >>out, " verify-cap:", quote_output(verify_cap, quotemarks=False)
-class DumpCapOptions(usage.Options):
+ if options['offsets']:
+ # NOTE: this offset-calculation code is fragile, and needs to be
+ # merged with MutableShareFile's internals.
+
+ print >>out
+ print >>out, " Section Offsets:"
+ def printoffset(name, value, shift=0):
+ print >>out, "%s%.20s: %s (0x%x)" % (" "*shift, name, value, value)
+ printoffset("first lease", m.HEADER_SIZE, 2)
+ printoffset("share data", m.DATA_OFFSET, 2)
+ o_seqnum = m.DATA_OFFSET + struct.calcsize(">B")
+ printoffset("seqnum", o_seqnum, 4)
+ o_root_hash = m.DATA_OFFSET + struct.calcsize(">BQ")
+ printoffset("root_hash", o_root_hash, 4)
+ for k in ["enc_privkey", "share_hash_chain", "signature",
+ "verification_key", "verification_key_end",
+ "share_data", "block_hash_tree", "EOF"]:
+ name = {"share_data": "block data",
+ "verification_key": "pubkey",
+ "verification_key_end": "end of pubkey",
+ "EOF": "end of share data"}.get(k,k)
+ offset = m.DATA_OFFSET + offsets[k]
+ printoffset(name, offset, 4)
+ f = open(options['filename'], "rb")
+ printoffset("extra leases", m._read_extra_lease_offset(f) + 4, 2)
+ f.close()
+
+ print >>out
+
+
+
+class DumpCapOptions(BaseOptions):
def getSynopsis(self):
- return "Usage: tahoe debug dump-cap [options] FILECAP"
+ return "Usage: tahoe [global-opts] debug dump-cap [options] FILECAP"
optParameters = [
["nodeid", "n",
- None, "storage server nodeid (ascii), to construct WE and secrets."],
+ None, "Specify the storage server nodeid (ASCII), to construct WE and secrets."],
["client-secret", "c", None,
- "client's base secret (ascii), to construct secrets"],
+ "Specify the client's base secret (ASCII), to construct secrets."],
["client-dir", "d", None,
- "client's base directory, from which a -c secret will be read"],
+ "Specify the client's base directory, from which a -c secret will be read."],
]
def parseArgs(self, cap):
self.cap = cap
def getUsage(self, width=None):
- t = usage.Options.getUsage(self, width)
+ t = BaseOptions.getUsage(self, width)
t += """
Print information about the given cap-string (aka: URI, file-cap, dir-cap,
read-cap, write-cap). The URI string is parsed and unpacked. This prints the
print >>out, " lease cancel secret:", base32.b2a(cancel)
def dump_uri_instance(u, nodeid, secret, out, show_header=True):
- from allmydata import storage, uri
+ from allmydata import uri
+ from allmydata.storage.server import si_b2a
from allmydata.util import base32, hashutil
+ from allmydata.util.encodingutil import quote_output
if isinstance(u, uri.CHKFileURI):
if show_header:
print >>out, " UEB hash:", base32.b2a(u.uri_extension_hash)
print >>out, " size:", u.size
print >>out, " k/N: %d/%d" % (u.needed_shares, u.total_shares)
- print >>out, " storage index:", storage.si_b2a(u.storage_index)
- _dump_secrets(u.storage_index, secret, nodeid, out)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
+ _dump_secrets(u.get_storage_index(), secret, nodeid, out)
elif isinstance(u, uri.CHKFileVerifierURI):
if show_header:
print >>out, "CHK Verifier URI:"
print >>out, " UEB hash:", base32.b2a(u.uri_extension_hash)
print >>out, " size:", u.size
print >>out, " k/N: %d/%d" % (u.needed_shares, u.total_shares)
- print >>out, " storage index:", storage.si_b2a(u.storage_index)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
elif isinstance(u, uri.LiteralFileURI):
if show_header:
print >>out, "Literal File URI:"
- print >>out, " data:", u.data
+ print >>out, " data:", quote_output(u.data)
- elif isinstance(u, uri.WriteableSSKFileURI):
+ elif isinstance(u, uri.WriteableSSKFileURI): # SDMF
if show_header:
- print >>out, "SSK Writeable URI:"
+ print >>out, "SDMF Writeable URI:"
print >>out, " writekey:", base32.b2a(u.writekey)
print >>out, " readkey:", base32.b2a(u.readkey)
- print >>out, " storage index:", storage.si_b2a(u.storage_index)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
print >>out, " fingerprint:", base32.b2a(u.fingerprint)
print >>out
if nodeid:
we = hashutil.ssk_write_enabler_hash(u.writekey, nodeid)
print >>out, " write_enabler:", base32.b2a(we)
print >>out
- _dump_secrets(u.storage_index, secret, nodeid, out)
-
+ _dump_secrets(u.get_storage_index(), secret, nodeid, out)
elif isinstance(u, uri.ReadonlySSKFileURI):
if show_header:
- print >>out, "SSK Read-only URI:"
+ print >>out, "SDMF Read-only URI:"
print >>out, " readkey:", base32.b2a(u.readkey)
- print >>out, " storage index:", storage.si_b2a(u.storage_index)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
print >>out, " fingerprint:", base32.b2a(u.fingerprint)
elif isinstance(u, uri.SSKVerifierURI):
if show_header:
- print >>out, "SSK Verifier URI:"
- print >>out, " storage index:", storage.si_b2a(u.storage_index)
+ print >>out, "SDMF Verifier URI:"
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
+ print >>out, " fingerprint:", base32.b2a(u.fingerprint)
+
+ elif isinstance(u, uri.WriteableMDMFFileURI): # MDMF
+ if show_header:
+ print >>out, "MDMF Writeable URI:"
+ print >>out, " writekey:", base32.b2a(u.writekey)
+ print >>out, " readkey:", base32.b2a(u.readkey)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
+ print >>out, " fingerprint:", base32.b2a(u.fingerprint)
+ print >>out
+ if nodeid:
+ we = hashutil.ssk_write_enabler_hash(u.writekey, nodeid)
+ print >>out, " write_enabler:", base32.b2a(we)
+ print >>out
+ _dump_secrets(u.get_storage_index(), secret, nodeid, out)
+ elif isinstance(u, uri.ReadonlyMDMFFileURI):
+ if show_header:
+ print >>out, "MDMF Read-only URI:"
+ print >>out, " readkey:", base32.b2a(u.readkey)
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
+ print >>out, " fingerprint:", base32.b2a(u.fingerprint)
+ elif isinstance(u, uri.MDMFVerifierURI):
+ if show_header:
+ print >>out, "MDMF Verifier URI:"
+ print >>out, " storage index:", si_b2a(u.get_storage_index())
print >>out, " fingerprint:", base32.b2a(u.fingerprint)
- elif isinstance(u, uri.NewDirectoryURI):
+
+ elif isinstance(u, uri.ImmutableDirectoryURI): # CHK-based directory
+ if show_header:
+ print >>out, "CHK Directory URI:"
+ dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+ elif isinstance(u, uri.ImmutableDirectoryURIVerifier):
+ if show_header:
+ print >>out, "CHK Directory Verifier URI:"
+ dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+
+ elif isinstance(u, uri.DirectoryURI): # SDMF-based directory
+ if show_header:
+ print >>out, "Directory Writeable URI:"
+ dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+ elif isinstance(u, uri.ReadonlyDirectoryURI):
+ if show_header:
+ print >>out, "Directory Read-only URI:"
+ dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+ elif isinstance(u, uri.DirectoryURIVerifier):
+ if show_header:
+ print >>out, "Directory Verifier URI:"
+ dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+
+ elif isinstance(u, uri.MDMFDirectoryURI): # MDMF-based directory
if show_header:
print >>out, "Directory Writeable URI:"
dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
- elif isinstance(u, uri.ReadonlyNewDirectoryURI):
+ elif isinstance(u, uri.ReadonlyMDMFDirectoryURI):
if show_header:
print >>out, "Directory Read-only URI:"
dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
- elif isinstance(u, uri.NewDirectoryURIVerifier):
+ elif isinstance(u, uri.MDMFDirectoryURIVerifier):
if show_header:
print >>out, "Directory Verifier URI:"
dump_uri_instance(u._filenode_uri, nodeid, secret, out, False)
+
else:
print >>out, "unknown cap type"
-class FindSharesOptions(usage.Options):
+class FindSharesOptions(BaseOptions):
def getSynopsis(self):
- return "Usage: tahoe debug find-shares STORAGE_INDEX NODEDIRS.."
+ return "Usage: tahoe [global-opts] debug find-shares STORAGE_INDEX NODEDIRS.."
+
def parseArgs(self, storage_index_s, *nodedirs):
+ from allmydata.util.encodingutil import argv_to_abspath
self.si_s = storage_index_s
- self.nodedirs = nodedirs
+ self.nodedirs = map(argv_to_abspath, nodedirs)
+
def getUsage(self, width=None):
- t = usage.Options.getUsage(self, width)
+ t = BaseOptions.getUsage(self, width)
t += """
Locate all shares for the given storage index. This command looks through one
or more node directories to find the shares. It returns a list of filenames,
/home/warner/testnet/node-1/storage/shares/44k/44kai1tui348689nrw8fjegc8c/9
/home/warner/testnet/node-2/storage/shares/44k/44kai1tui348689nrw8fjegc8c/2
"""
- from allmydata import storage
+ from allmydata.storage.server import si_a2b, storage_index_to_dir
+ from allmydata.util.encodingutil import listdir_unicode
out = options.stdout
- sharedir = storage.storage_index_to_dir(storage.si_a2b(options.si_s))
+ sharedir = storage_index_to_dir(si_a2b(options.si_s))
for d in options.nodedirs:
- d = os.path.join(os.path.expanduser(d), "storage/shares", sharedir)
+ d = os.path.join(d, "storage/shares", sharedir)
if os.path.exists(d):
- for shnum in os.listdir(d):
+ for shnum in listdir_unicode(d):
print >>out, os.path.join(d, shnum)
return 0
-class CatalogSharesOptions(usage.Options):
+class CatalogSharesOptions(BaseOptions):
"""
"""
def parseArgs(self, *nodedirs):
- self.nodedirs = nodedirs
+ from allmydata.util.encodingutil import argv_to_abspath
+ self.nodedirs = map(argv_to_abspath, nodedirs)
if not nodedirs:
raise usage.UsageError("must specify at least one node directory")
def getSynopsis(self):
- return "Usage: tahoe debug catalog-shares NODEDIRS.."
+ return "Usage: tahoe [global-opts] debug catalog-shares NODEDIRS.."
def getUsage(self, width=None):
- t = usage.Options.getUsage(self, width)
+ t = BaseOptions.getUsage(self, width)
t += """
Locate all shares in the given node directories, and emit a one-line summary
of each share. Run it like this:
# take advantage of the fact that ImmediateReadBucketProxy returns
# Deferreds that are already fired
results = []
+ failures = []
d = defer.maybeDeferred(c, *args, **kwargs)
- d.addCallback(results.append)
+ d.addCallbacks(results.append, failures.append)
+ if failures:
+ failures[0].raiseException()
return results[0]
def describe_share(abs_sharefile, si_s, shnum_s, now, out):
- from allmydata import uri, storage
+ from allmydata import uri
+ from allmydata.storage.mutable import MutableShareFile
+ from allmydata.storage.immutable import ShareFile
from allmydata.mutable.layout import unpack_share
from allmydata.mutable.common import NeedMoreDataError
from allmydata.immutable.layout import ReadBucketProxy
from allmydata.util import base32
+ from allmydata.util.encodingutil import quote_output
import struct
f = open(abs_sharefile, "rb")
prefix = f.read(32)
- if prefix == storage.MutableShareFile.MAGIC:
+ if prefix == MutableShareFile.MAGIC:
# mutable share
- m = storage.MutableShareFile(abs_sharefile)
+ m = MutableShareFile(abs_sharefile)
WE, nodeid = m._read_write_enabler_and_nodeid(f)
- num_extra_leases = m._read_num_extra_leases(f)
data_length = m._read_data_length(f)
- extra_lease_offset = m._read_extra_lease_offset(f)
- container_size = extra_lease_offset - m.DATA_OFFSET
- leases = list(m._enumerate_leases(f))
- expiration_time = min( [lease[1].expiration_time
- for lease in leases] )
+ expiration_time = min( [lease.expiration_time
+ for (i,lease) in m._enumerate_leases(f)] )
expiration = max(0, expiration_time - now)
share_type = "unknown"
f.seek(m.DATA_OFFSET)
- if f.read(1) == "\x00":
+ version = f.read(1)
+ if version == "\x00":
# this slot contains an SMDF share
share_type = "SDMF"
+ elif version == "\x01":
+ share_type = "MDMF"
if share_type == "SDMF":
f.seek(m.DATA_OFFSET)
print >>out, "SDMF %s %d/%d %d #%d:%s %d %s" % \
(si_s, k, N, datalen,
seqnum, base32.b2a(root_hash),
- expiration, abs_sharefile)
+ expiration, quote_output(abs_sharefile))
+ elif share_type == "MDMF":
+ from allmydata.mutable.layout import MDMFSlotReadProxy
+ fake_shnum = 0
+ # TODO: factor this out with dump_MDMF_share()
+ class ShareDumper(MDMFSlotReadProxy):
+ def _read(self, readvs, force_remote=False, queue=False):
+ data = []
+ for (where,length) in readvs:
+ f.seek(m.DATA_OFFSET+where)
+ data.append(f.read(length))
+ return defer.succeed({fake_shnum: data})
+
+ p = ShareDumper(None, "fake-si", fake_shnum)
+ def extract(func):
+ stash = []
+ # these methods return Deferreds, but we happen to know that
+ # they run synchronously when not actually talking to a
+ # remote server
+ d = func()
+ d.addCallback(stash.append)
+ return stash[0]
+
+ verinfo = extract(p.get_verinfo)
+ (seqnum, root_hash, salt_to_use, segsize, datalen, k, N, prefix,
+ offsets) = verinfo
+ print >>out, "MDMF %s %d/%d %d #%d:%s %d %s" % \
+ (si_s, k, N, datalen,
+ seqnum, base32.b2a(root_hash),
+ expiration, quote_output(abs_sharefile))
else:
- print >>out, "UNKNOWN mutable %s" % (abs_sharefile,)
+ print >>out, "UNKNOWN mutable %s" % quote_output(abs_sharefile)
elif struct.unpack(">L", prefix[:4]) == (1,):
# immutable
-
class ImmediateReadBucketProxy(ReadBucketProxy):
def __init__(self, sf):
self.sf = sf
+ ReadBucketProxy.__init__(self, None, None, "")
def __repr__(self):
return "<ImmediateReadBucketProxy>"
def _read(self, offset, size):
return defer.succeed(sf.read_share_data(offset, size))
# use a ReadBucketProxy to parse the bucket and find the uri extension
- sf = storage.ShareFile(abs_sharefile)
+ sf = ShareFile(abs_sharefile)
bp = ImmediateReadBucketProxy(sf)
- bp.start()
expiration_time = min( [lease.expiration_time
- for lease in sf.iter_leases()] )
+ for lease in sf.get_leases()] )
expiration = max(0, expiration_time - now)
UEB_data = call(bp.get_uri_extension)
print >>out, "CHK %s %d/%d %d %s %d %s" % (si_s, k, N, filesize,
ueb_hash, expiration,
- abs_sharefile)
+ quote_output(abs_sharefile))
else:
- print >>out, "UNKNOWN really-unknown %s" % (abs_sharefile,)
+ print >>out, "UNKNOWN really-unknown %s" % quote_output(abs_sharefile)
f.close()
-
def catalog_shares(options):
+ from allmydata.util.encodingutil import listdir_unicode, quote_output
+
out = options.stdout
err = options.stderr
now = time.time()
for d in options.nodedirs:
- d = os.path.join(os.path.expanduser(d), "storage/shares")
+ d = os.path.join(d, "storage/shares")
try:
- abbrevs = os.listdir(d)
+ abbrevs = listdir_unicode(d)
except EnvironmentError:
# ignore nodes that have storage turned off altogether
pass
else:
- for abbrevdir in abbrevs:
+ for abbrevdir in sorted(abbrevs):
if abbrevdir == "incoming":
continue
abbrevdir = os.path.join(d, abbrevdir)
- for si_s in os.listdir(abbrevdir):
- si_dir = os.path.join(abbrevdir, si_s)
- for shnum_s in os.listdir(si_dir):
- abs_sharefile = os.path.join(si_dir, shnum_s)
- abs_sharefile = os.path.abspath(abs_sharefile)
- assert os.path.isfile(abs_sharefile)
- try:
- describe_share(abs_sharefile, si_s, shnum_s, now,
- out)
- except:
- print >>err, "Error processing %s" % abs_sharefile
- failure.Failure().printTraceback(err)
+ # this tool may get run against bad disks, so we can't assume
+ # that listdir_unicode will always succeed. Try to catalog as much
+ # as possible.
+ try:
+ sharedirs = listdir_unicode(abbrevdir)
+ for si_s in sorted(sharedirs):
+ si_dir = os.path.join(abbrevdir, si_s)
+ catalog_shares_one_abbrevdir(si_s, si_dir, now, out,err)
+ except:
+ print >>err, "Error processing %s" % quote_output(abbrevdir)
+ failure.Failure().printTraceback(err)
+
return 0
-class CorruptShareOptions(usage.Options):
+def _as_number(s):
+ try:
+ return int(s)
+ except ValueError:
+ return "not int"
+
+def catalog_shares_one_abbrevdir(si_s, si_dir, now, out, err):
+ from allmydata.util.encodingutil import listdir_unicode, quote_output
+
+ try:
+ for shnum_s in sorted(listdir_unicode(si_dir), key=_as_number):
+ abs_sharefile = os.path.join(si_dir, shnum_s)
+ assert os.path.isfile(abs_sharefile)
+ try:
+ describe_share(abs_sharefile, si_s, shnum_s, now,
+ out)
+ except:
+ print >>err, "Error processing %s" % quote_output(abs_sharefile)
+ failure.Failure().printTraceback(err)
+ except:
+ print >>err, "Error processing %s" % quote_output(si_dir)
+ failure.Failure().printTraceback(err)
+
+class CorruptShareOptions(BaseOptions):
def getSynopsis(self):
- return "Usage: tahoe debug corrupt-share SHARE_FILENAME"
+ return "Usage: tahoe [global-opts] debug corrupt-share SHARE_FILENAME"
optParameters = [
- ["offset", "o", "block-random", "Which bit to flip."],
+ ["offset", "o", "block-random", "Specify which bit to flip."],
]
def getUsage(self, width=None):
- t = usage.Options.getUsage(self, width)
+ t = BaseOptions.getUsage(self, width)
t += """
Corrupt the given share by flipping a bit. This will cause a
verifying/downloading client to log an integrity-check failure incident, and
def corrupt_share(options):
import random
- from allmydata import storage
+ from allmydata.storage.mutable import MutableShareFile
+ from allmydata.storage.immutable import ShareFile
from allmydata.mutable.layout import unpack_header
from allmydata.immutable.layout import ReadBucketProxy
out = options.stdout
f = open(fn, "rb")
prefix = f.read(32)
f.close()
- if prefix == storage.MutableShareFile.MAGIC:
+ if prefix == MutableShareFile.MAGIC:
# mutable
- m = storage.MutableShareFile(fn)
+ m = MutableShareFile(fn)
f = open(fn, "rb")
f.seek(m.DATA_OFFSET)
data = f.read(2000)
flip_bit(start, end)
else:
# otherwise assume it's immutable
- f = storage.ShareFile(fn)
- bp = ReadBucketProxy(None)
+ f = ShareFile(fn)
+ bp = ReadBucketProxy(None, None, '')
offsets = bp._parse_offsets(f.read_share_data(0, 0x24))
start = f._data_offset + offsets["data"]
end = f._data_offset + offsets["plaintext_hash_tree"]
-class ReplOptions(usage.Options):
- pass
+class ReplOptions(BaseOptions):
+ def getSynopsis(self):
+ return "Usage: tahoe [global-opts] debug repl"
def repl(options):
import code
return code.interact()
-class DebugCommand(usage.Options):
+DEFAULT_TESTSUITE = 'allmydata'
+
+class TrialOptions(twisted_trial.Options):
+ def getSynopsis(self):
+ return "Usage: tahoe [global-opts] debug trial [options] [[file|package|module|TestCase|testmethod]...]"
+
+ def parseOptions(self, all_subargs, *a, **kw):
+ self.trial_args = list(all_subargs)
+
+ # any output from the option parsing will be printed twice, but that's harmless
+ return twisted_trial.Options.parseOptions(self, all_subargs, *a, **kw)
+
+ def parseArgs(self, *nonoption_args):
+ if not nonoption_args:
+ self.trial_args.append(DEFAULT_TESTSUITE)
+
+ def getUsage(self, width=None):
+ t = twisted_trial.Options.getUsage(self, width)
+ t += """
+The 'tahoe debug trial' command uses the correct imports for this instance of
+Tahoe-LAFS. The default test suite is '%s'.
+""" % (DEFAULT_TESTSUITE,)
+ return t
+
+def trial(config):
+ sys.argv = ['trial'] + config.trial_args
+
+ # This does not return.
+ twisted_trial.run()
+
+
+def fixOptionsClass( (subcmd, shortcut, OptionsClass, desc) ):
+ class FixedOptionsClass(OptionsClass):
+ def getSynopsis(self):
+ t = OptionsClass.getSynopsis(self)
+ i = t.find("Usage: flogtool ")
+ if i >= 0:
+ return "Usage: tahoe [global-opts] debug flogtool " + t[i+len("Usage: flogtool "):]
+ else:
+ return "Usage: tahoe [global-opts] debug flogtool %s [options]" % (subcmd,)
+ return (subcmd, shortcut, FixedOptionsClass, desc)
+
+class FlogtoolOptions(foolscap_cli.Options):
+ def __init__(self):
+ super(FlogtoolOptions, self).__init__()
+ self.subCommands = map(fixOptionsClass, self.subCommands)
+
+ def getSynopsis(self):
+ return "Usage: tahoe [global-opts] debug flogtool (%s) [command options]" % ("|".join([x[0] for x in self.subCommands]))
+
+ def parseOptions(self, all_subargs, *a, **kw):
+ self.flogtool_args = list(all_subargs)
+ return super(FlogtoolOptions, self).parseOptions(self.flogtool_args, *a, **kw)
+
+ def getUsage(self, width=None):
+ t = super(FlogtoolOptions, self).getUsage(width)
+ t += """
+The 'tahoe debug flogtool' command uses the correct imports for this instance
+of Tahoe-LAFS.
+
+Please run 'tahoe debug flogtool SUBCOMMAND --help' for more details on each
+subcommand.
+"""
+ return t
+
+ def opt_help(self):
+ print str(self)
+ sys.exit(0)
+
+def flogtool(config):
+ sys.argv = ['flogtool'] + config.flogtool_args
+ return foolscap_cli.run_flogtool()
+
+
+class DebugCommand(BaseOptions):
subCommands = [
["dump-share", None, DumpOptions,
"Unpack and display the contents of a share (uri_extension and leases)."],
- ["dump-cap", None, DumpCapOptions, "Unpack a read-cap or write-cap"],
- ["find-shares", None, FindSharesOptions, "Locate sharefiles in node dirs"],
- ["catalog-shares", None, CatalogSharesOptions, "Describe shares in node dirs"],
- ["corrupt-share", None, CorruptShareOptions, "Corrupt a share"],
- ["repl", None, ReplOptions, "Open a python interpreter"],
+ ["dump-cap", None, DumpCapOptions, "Unpack a read-cap or write-cap."],
+ ["find-shares", None, FindSharesOptions, "Locate sharefiles in node dirs."],
+ ["catalog-shares", None, CatalogSharesOptions, "Describe all shares in node dirs."],
+ ["corrupt-share", None, CorruptShareOptions, "Corrupt a share by flipping a bit."],
+ ["repl", None, ReplOptions, "Open a Python interpreter."],
+ ["trial", None, TrialOptions, "Run tests using Twisted Trial with the right imports."],
+ ["flogtool", None, FlogtoolOptions, "Utilities to access log files."],
]
def postOptions(self):
if not hasattr(self, 'subOptions'):
raise usage.UsageError("must specify a subcommand")
def getSynopsis(self):
- return "Usage: tahoe debug SUBCOMMAND"
+ return ""
def getUsage(self, width=None):
- #t = usage.Options.getUsage(self, width)
- t = """
+ #t = BaseOptions.getUsage(self, width)
+ t = """Usage: tahoe debug SUBCOMMAND
Subcommands:
- tahoe debug dump-share Unpack and display the contents of a share
- tahoe debug dump-cap Unpack a read-cap or write-cap
- tahoe debug find-shares Locate sharefiles in node directories
- tahoe debug catalog-shares Describe all shares in node dirs
+ tahoe debug dump-share Unpack and display the contents of a share.
+ tahoe debug dump-cap Unpack a read-cap or write-cap.
+ tahoe debug find-shares Locate sharefiles in node directories.
+ tahoe debug catalog-shares Describe all shares in node dirs.
tahoe debug corrupt-share Corrupt a share by flipping a bit.
+ tahoe debug repl Open a Python interpreter.
+ tahoe debug trial Run tests using Twisted Trial with the right imports.
+ tahoe debug flogtool Utilities to access log files.
Please run e.g. 'tahoe debug dump-share --help' for more details on each
subcommand.
"""
+ # See ticket #1441 for why we print different information when
+ # run via /usr/bin/tahoe. Note that argv[0] is the full path.
+ if sys.argv[0] == '/usr/bin/tahoe':
+ t += """
+To get branch coverage for the Tahoe test suite (on the installed copy of
+Tahoe), install the 'python-coverage' package and then use:
+
+ python-coverage run --branch /usr/bin/tahoe debug trial
+"""
+ else:
+ t += """
+Another debugging feature is that bin%stahoe allows executing an arbitrary
+"runner" command (typically an installed Python script, such as 'coverage'),
+with the Tahoe libraries on the PYTHONPATH. The runner command name is
+prefixed with '@', and any occurrences of '@tahoe' in its arguments are
+replaced by the full path to the tahoe script.
+
+For example, if 'coverage' is installed and on the PATH, you can use:
+
+ bin%stahoe @coverage run --branch @tahoe debug trial
+
+to get branch coverage for the Tahoe test suite. Or, to run python with
+the -3 option that warns about Python 3 incompatibilities:
+
+ bin%stahoe @python -3 @tahoe command [options]
+""" % (os.sep, os.sep, os.sep)
return t
subDispatch = {
"catalog-shares": catalog_shares,
"corrupt-share": corrupt_share,
"repl": repl,
+ "trial": trial,
+ "flogtool": flogtool,
}
subCommands = [
- ["debug", None, DebugCommand, "debug subcommands: use 'tahoe debug' for a list"],
+ ["debug", None, DebugCommand, "debug subcommands: use 'tahoe debug' for a list."],
]
dispatch = {