From: Brian Warner Date: Wed, 11 Jul 2007 01:41:52 +0000 (-0700) Subject: runner.py: start to refactor into separate files X-Git-Url: https://git.rkrishnan.org/simplejson/encoder.py.html?a=commitdiff_plain;h=4a16d20d854cdf91e47eabf2695fe2f9a925c912;p=tahoe-lafs%2Ftahoe-lafs.git runner.py: start to refactor into separate files --- diff --git a/src/allmydata/scripts/common.py b/src/allmydata/scripts/common.py new file mode 100644 index 00000000..114022be --- /dev/null +++ b/src/allmydata/scripts/common.py @@ -0,0 +1,48 @@ + +import os +from twisted.python import usage + + +class BasedirMixin: + optFlags = [ + ["multiple", "m", "allow multiple basedirs to be specified at once"], + ] + + def postOptions(self): + if not self.basedirs: + raise usage.UsageError(" parameter is required") + if self['basedir']: + del self['basedir'] + self['basedirs'] = [os.path.abspath(os.path.expanduser(b)) + for b in self.basedirs] + + def parseArgs(self, *args): + self.basedirs = [] + if self['basedir']: + self.basedirs.append(self['basedir']) + if self['multiple']: + self.basedirs.extend(args) + else: + if len(args) == 0 and not self.basedirs: + self.basedirs.append(".") + if len(args) > 0: + self.basedirs.append(args[0]) + if len(args) > 1: + raise usage.UsageError("I wasn't expecting so many arguments") + +class NoDefaultBasedirMixin(BasedirMixin): + def parseArgs(self, *args): + # create-client won't default to --basedir=. + self.basedirs = [] + if self['basedir']: + self.basedirs.append(self['basedir']) + if self['multiple']: + self.basedirs.extend(args) + else: + if len(args) > 0: + self.basedirs.append(args[0]) + if len(args) > 1: + raise usage.UsageError("I wasn't expecting so many arguments") + if not self.basedirs: + raise usage.UsageError("--basedir must be provided") + diff --git a/src/allmydata/scripts/debug.py b/src/allmydata/scripts/debug.py new file mode 100644 index 00000000..23e91a85 --- /dev/null +++ b/src/allmydata/scripts/debug.py @@ -0,0 +1,161 @@ + +import os, sys +from twisted.python import usage +from allmydata.scripts.common import BasedirMixin + +class DumpOptions(usage.Options): + optParameters = [ + ["filename", "f", None, "which file to dump"], + ] + + def parseArgs(self, filename=None): + if filename: + self['filename'] = filename + + def postOptions(self): + if not self['filename']: + raise usage.UsageError(" parameter is required") + +class DumpRootDirnodeOptions(BasedirMixin, usage.Options): + optParameters = [ + ["basedir", "C", None, "the vdrive-server's base directory"], + ] + +class DumpDirnodeOptions(BasedirMixin, usage.Options): + optParameters = [ + ["uri", "u", None, "the URI of the dirnode to dump."], + ["basedir", "C", None, "which directory to create the introducer in"], + ] + optFlags = [ + ["verbose", "v", "be extra noisy (show encrypted data)"], + ] + def parseArgs(self, *args): + if len(args) == 1: + self['uri'] = args[-1] + args = args[:-1] + BasedirMixin.parseArgs(self, *args) + + def postOptions(self): + BasedirMixin.postOptions(self) + if not self['uri']: + raise usage.UsageError(" parameter is required") + +def dump_uri_extension(config, out=sys.stdout, err=sys.stderr): + from allmydata import uri + + filename = config['filename'] + unpacked = uri.unpack_extension_readable(open(filename,"rb").read()) + keys1 = ("size", "num_segments", "segment_size", + "needed_shares", "total_shares") + keys2 = ("codec_name", "codec_params", "tail_codec_params") + keys3 = ("plaintext_hash", "plaintext_root_hash", + "crypttext_hash", "crypttext_root_hash", + "share_root_hash") + for k in keys1: + if k in unpacked: + print >>out, "%19s: %s" % (k, unpacked[k]) + print >>out + for k in keys2: + if k in unpacked: + print >>out, "%19s: %s" % (k, unpacked[k]) + print >>out + for k in keys3: + if k in unpacked: + print >>out, "%19s: %s" % (k, unpacked[k]) + + leftover = set(unpacked.keys()) - set(keys1 + keys2 + keys3) + if leftover: + print >>out + for k in sorted(leftover): + print >>out, "%s: %s" % (k, unpacked[k]) + + print >>out + return 0 + +def dump_root_dirnode(config, out=sys.stdout, err=sys.stderr): + from allmydata import uri + + basedir = config['basedirs'][0] + root_dirnode_file = os.path.join(basedir, "vdrive", "root") + try: + f = open(root_dirnode_file, "rb") + key = f.read() + rooturi = uri.pack_dirnode_uri("fakeFURL", key) + print >>out, rooturi + return 0 + except EnvironmentError: + print >>out, "unable to read root dirnode file from %s" % \ + root_dirnode_file + return 1 + +def dump_directory_node(config, out=sys.stdout, err=sys.stderr): + from allmydata import uri, dirnode + from allmydata.util import hashutil, idlib + basedir = config['basedirs'][0] + dir_uri = config['uri'] + verbose = config['verbose'] + + furl, key = uri.unpack_dirnode_uri(dir_uri) + if uri.is_mutable_dirnode_uri(dir_uri): + wk, we, rk, index = hashutil.generate_dirnode_keys_from_writekey(key) + else: + wk, we, rk, index = hashutil.generate_dirnode_keys_from_readkey(key) + + filename = os.path.join(basedir, "vdrive", idlib.b2a(index)) + + print >>out + print >>out, "dirnode uri: %s" % dir_uri + print >>out, "filename : %s" % filename + print >>out, "index : %s" % idlib.b2a(index) + if wk: + print >>out, "writekey : %s" % idlib.b2a(wk) + print >>out, "write_enabler: %s" % idlib.b2a(we) + else: + print >>out, "writekey : None" + print >>out, "write_enabler: None" + print >>out, "readkey : %s" % idlib.b2a(rk) + + print >>out + + vds = dirnode.VirtualDriveServer(os.path.join(basedir, "vdrive"), False) + data = vds._read_from_file(index) + if we: + if we != data[0]: + print >>out, "ERROR: write_enabler does not match" + + for (H_key, E_key, E_write, E_read) in data[1]: + if verbose: + print >>out, " H_key %s" % idlib.b2a(H_key) + print >>out, " E_key %s" % idlib.b2a(E_key) + print >>out, " E_write %s" % idlib.b2a(E_write) + print >>out, " E_read %s" % idlib.b2a(E_read) + key = dirnode.decrypt(rk, E_key) + print >>out, " key %s" % key + if hashutil.dir_name_hash(rk, key) != H_key: + print >>out, " ERROR: H_key does not match" + if wk and E_write: + if len(E_write) < 14: + print >>out, " ERROR: write data is short:", idlib.b2a(E_write) + write = dirnode.decrypt(wk, E_write) + print >>out, " write: %s" % write + read = dirnode.decrypt(rk, E_read) + print >>out, " read: %s" % read + print >>out + + return 0 + + +subCommands = [ + ["dump-uri-extension", None, DumpOptions, + "Unpack and display the contents of a uri_extension file."], + ["dump-root-dirnode", None, DumpRootDirnodeOptions, + "Compute most of the URI for the vdrive server's root dirnode."], + ["dump-dirnode", None, DumpDirnodeOptions, + "Unpack and display the contents of a vdrive DirectoryNode."], + ] + +dispatch = { + "dump-uri-extension": dump_uri_extension, + "dump-root-dirnode": dump_root_dirnode, + "dump-dirnode": dump_directory_node, + } diff --git a/src/allmydata/scripts/runner.py b/src/allmydata/scripts/runner.py index 8b57b17c..0349fa24 100644 --- a/src/allmydata/scripts/runner.py +++ b/src/allmydata/scripts/runner.py @@ -1,10 +1,11 @@ -#! /usr/bin/env python import os, subprocess, sys, signal, time from cStringIO import StringIO from twisted.python import usage from twisted.python.procutils import which +from allmydata.scripts import debug +from allmydata.scripts.common import BasedirMixin, NoDefaultBasedirMixin def testtwistd(loc): try: @@ -43,49 +44,6 @@ if not twistd: print "Can't find twistd (it comes with Twisted). Aborting." sys.exit(1) -class BasedirMixin: - optFlags = [ - ["multiple", "m", "allow multiple basedirs to be specified at once"], - ] - - def postOptions(self): - if not self.basedirs: - raise usage.UsageError(" parameter is required") - if self['basedir']: - del self['basedir'] - self['basedirs'] = [os.path.abspath(os.path.expanduser(b)) - for b in self.basedirs] - - def parseArgs(self, *args): - self.basedirs = [] - if self['basedir']: - self.basedirs.append(self['basedir']) - if self['multiple']: - self.basedirs.extend(args) - else: - if len(args) == 0 and not self.basedirs: - self.basedirs.append(".") - if len(args) > 0: - self.basedirs.append(args[0]) - if len(args) > 1: - raise usage.UsageError("I wasn't expecting so many arguments") - -class NoDefaultBasedirMixin(BasedirMixin): - def parseArgs(self, *args): - # create-client won't default to --basedir=. - self.basedirs = [] - if self['basedir']: - self.basedirs.append(self['basedir']) - if self['multiple']: - self.basedirs.extend(args) - else: - if len(args) > 0: - self.basedirs.append(args[0]) - if len(args) > 1: - raise usage.UsageError("I wasn't expecting so many arguments") - if not self.basedirs: - raise usage.UsageError("--basedir must be provided") - class StartOptions(BasedirMixin, usage.Options): optParameters = [ ["basedir", "C", None, "which directory to start the node in"], @@ -116,43 +74,6 @@ class CreateIntroducerOptions(NoDefaultBasedirMixin, usage.Options): ["basedir", "C", None, "which directory to create the introducer in"], ] -class DumpOptions(usage.Options): - optParameters = [ - ["filename", "f", None, "which file to dump"], - ] - - def parseArgs(self, filename=None): - if filename: - self['filename'] = filename - - def postOptions(self): - if not self['filename']: - raise usage.UsageError(" parameter is required") - -class DumpRootDirnodeOptions(BasedirMixin, usage.Options): - optParameters = [ - ["basedir", "C", None, "the vdrive-server's base directory"], - ] - -class DumpDirnodeOptions(BasedirMixin, usage.Options): - optParameters = [ - ["uri", "u", None, "the URI of the dirnode to dump."], - ["basedir", "C", None, "which directory to create the introducer in"], - ] - optFlags = [ - ["verbose", "v", "be extra noisy (show encrypted data)"], - ] - def parseArgs(self, *args): - if len(args) == 1: - self['uri'] = args[-1] - args = args[:-1] - BasedirMixin.parseArgs(self, *args) - - def postOptions(self): - BasedirMixin.postOptions(self) - if not self['uri']: - raise usage.UsageError(" parameter is required") - client_tac = """ # -*- python -*- @@ -190,13 +111,7 @@ class Options(usage.Options): ["start", None, StartOptions, "Start a node (of any type)."], ["stop", None, StopOptions, "Stop a node."], ["restart", None, RestartOptions, "Restart a node."], - ["dump-uri-extension", None, DumpOptions, - "Unpack and display the contents of a uri_extension file."], - ["dump-root-dirnode", None, DumpRootDirnodeOptions, - "Compute most of the URI for the vdrive server's root dirnode."], - ["dump-dirnode", None, DumpDirnodeOptions, - "Unpack and display the contents of a vdrive DirectoryNode."], - ] + ] + debug.subCommands def postOptions(self): if not hasattr(self, 'subOptions'): @@ -245,12 +160,9 @@ def runner(argv, run_by_human=True, stdout=sys.stdout, stderr=sys.stderr): return rc for basedir in so.basedirs: rc = start(basedir, so, stdout, stderr) or rc - elif command == "dump-uri-extension": - rc = dump_uri_extension(so, stdout, stderr) - elif command == "dump-root-dirnode": - rc = dump_root_dirnode(so.basedirs[0], so, stdout, stderr) - elif command == "dump-dirnode": - rc = dump_directory_node(so.basedirs[0], so, stdout, stderr) + elif command in debug.dispatch: + rc = debug.dispatch[command](so, stdout, stderr) + return rc def run(): @@ -332,105 +244,3 @@ def stop(basedir, config, out=sys.stdout, err=sys.stderr): time.sleep(1) print >>err, "never saw process go away" return 1 - -def dump_uri_extension(config, out=sys.stdout, err=sys.stderr): - from allmydata import uri - - filename = config['filename'] - unpacked = uri.unpack_extension_readable(open(filename,"rb").read()) - keys1 = ("size", "num_segments", "segment_size", - "needed_shares", "total_shares") - keys2 = ("codec_name", "codec_params", "tail_codec_params") - keys3 = ("plaintext_hash", "plaintext_root_hash", - "crypttext_hash", "crypttext_root_hash", - "share_root_hash") - for k in keys1: - if k in unpacked: - print >>out, "%19s: %s" % (k, unpacked[k]) - print >>out - for k in keys2: - if k in unpacked: - print >>out, "%19s: %s" % (k, unpacked[k]) - print >>out - for k in keys3: - if k in unpacked: - print >>out, "%19s: %s" % (k, unpacked[k]) - - leftover = set(unpacked.keys()) - set(keys1 + keys2 + keys3) - if leftover: - print >>out - for k in sorted(leftover): - print >>out, "%s: %s" % (k, unpacked[k]) - - print >>out - return 0 - -def dump_root_dirnode(basedir, config, out=sys.stdout, err=sys.stderr): - from allmydata import uri - - root_dirnode_file = os.path.join(basedir, "vdrive", "root") - try: - f = open(root_dirnode_file, "rb") - key = f.read() - rooturi = uri.pack_dirnode_uri("fakeFURL", key) - print >>out, rooturi - return 0 - except EnvironmentError: - print >>out, "unable to read root dirnode file from %s" % \ - root_dirnode_file - return 1 - -def dump_directory_node(basedir, config, out=sys.stdout, err=sys.stderr): - from allmydata import uri, dirnode - from allmydata.util import hashutil, idlib - dir_uri = config['uri'] - verbose = config['verbose'] - - furl, key = uri.unpack_dirnode_uri(dir_uri) - if uri.is_mutable_dirnode_uri(dir_uri): - wk, we, rk, index = hashutil.generate_dirnode_keys_from_writekey(key) - else: - wk, we, rk, index = hashutil.generate_dirnode_keys_from_readkey(key) - - filename = os.path.join(basedir, "vdrive", idlib.b2a(index)) - - print >>out - print >>out, "dirnode uri: %s" % dir_uri - print >>out, "filename : %s" % filename - print >>out, "index : %s" % idlib.b2a(index) - if wk: - print >>out, "writekey : %s" % idlib.b2a(wk) - print >>out, "write_enabler: %s" % idlib.b2a(we) - else: - print >>out, "writekey : None" - print >>out, "write_enabler: None" - print >>out, "readkey : %s" % idlib.b2a(rk) - - print >>out - - vds = dirnode.VirtualDriveServer(os.path.join(basedir, "vdrive"), False) - data = vds._read_from_file(index) - if we: - if we != data[0]: - print >>out, "ERROR: write_enabler does not match" - - for (H_key, E_key, E_write, E_read) in data[1]: - if verbose: - print >>out, " H_key %s" % idlib.b2a(H_key) - print >>out, " E_key %s" % idlib.b2a(E_key) - print >>out, " E_write %s" % idlib.b2a(E_write) - print >>out, " E_read %s" % idlib.b2a(E_read) - key = dirnode.decrypt(rk, E_key) - print >>out, " key %s" % key - if hashutil.dir_name_hash(rk, key) != H_key: - print >>out, " ERROR: H_key does not match" - if wk and E_write: - if len(E_write) < 14: - print >>out, " ERROR: write data is short:", idlib.b2a(E_write) - write = dirnode.decrypt(wk, E_write) - print >>out, " write: %s" % write - read = dirnode.decrypt(rk, E_read) - print >>out, " read: %s" % read - print >>out - - return 0 diff --git a/src/allmydata/test/test_runner.py b/src/allmydata/test/test_runner.py index d42b1e22..c7a8aedb 100644 --- a/src/allmydata/test/test_runner.py +++ b/src/allmydata/test/test_runner.py @@ -4,7 +4,7 @@ from twisted.trial import unittest from cStringIO import StringIO from twisted.python import usage import os.path -from allmydata.scripts import runner +from allmydata.scripts import runner, debug from allmydata.util import fileutil class CreateNode(unittest.TestCase): @@ -92,7 +92,8 @@ class CreateNode(unittest.TestCase): class Diagnostics(unittest.TestCase): def test_dump_root_dirnode_failure(self): s = StringIO() - rc = runner.dump_root_dirnode("missing_basedir", {}, s) + config = {'basedirs': ["missing_basedir"]} + rc = debug.dump_root_dirnode(config, s) output = s.getvalue() self.failUnless("unable to read root dirnode file from" in output) self.failIfEqual(rc, 0)