Switch dkimsign.py to use dkim.DKIM directly rather than via dkim.dkim

Coerce argparse arguements to bytes so python3 works too
This commit is contained in:
Scott Kitterman
2017-06-05 00:20:32 -04:00
parent 60a6fc356e
commit 7f3033d30c
+18 -4
View File
@@ -37,24 +37,38 @@ parser.add_argument('selector', action="store")
parser.add_argument('domain', action="store") parser.add_argument('domain', action="store")
parser.add_argument('privatekeyfile', action="store") parser.add_argument('privatekeyfile', action="store")
parser.add_argument('--hcanon', choices=['simple', 'relaxed'], parser.add_argument('--hcanon', choices=['simple', 'relaxed'],
default='relaxed', type=bytes, default='relaxed',
help='Header canonicalization algorithm: default=relaxed') help='Header canonicalization algorithm: default=relaxed')
parser.add_argument('--bcanon', choices=['simple', 'relaxed'], parser.add_argument('--bcanon', choices=['simple', 'relaxed'],
default='simple', type=bytes, default='simple',
help='Body canonicalization algorithm: default=simple') help='Body canonicalization algorithm: default=simple')
parser.add_argument('--identity', help='Optional value for i= tag.') parser.add_argument('--identity', help='Optional value for i= tag.')
args=parser.parse_args(arguments) args=parser.parse_args(arguments)
signature_algorithm=b'rsa-sha256'
include_headers = None
length = None
logger = None
if sys.version_info[0] >= 3: if sys.version_info[0] >= 3:
args.selector = bytes(args.selector, encoding='UTF-8')
args.domain = bytes(args.domain, encoding='UTF-8')
if args.identity is not None:
args.identity = bytes(args.identity, encoding='UTF-8')
args.hcanon = bytes(args.hcanon, encoding='UTF-8')
args.bcanon = bytes(args.bcanon, encoding='UTF-8')
# Make sys.stdin and stdout binary streams. # Make sys.stdin and stdout binary streams.
sys.stdin = sys.stdin.detach() sys.stdin = sys.stdin.detach()
sys.stdout = sys.stdout.detach() sys.stdout = sys.stdout.detach()
canonicalize = (args.hcanon, args.bcanon)
message = sys.stdin.read() message = sys.stdin.read()
try: try:
sig = dkim.sign(message, args.selector, args.domain, open( d = dkim.DKIM(message,logger=logger,
signature_algorithm=signature_algorithm)
sig = d.sign(args.selector, args.domain, open(
args.privatekeyfile, "rb").read(), identity = args.identity, args.privatekeyfile, "rb").read(), identity = args.identity,
canonicalize=(args.hcanon, args.bcanon)) canonicalize=canonicalize, include_headers=include_headers,
length=length)
sys.stdout.write(sig) sys.stdout.write(sig)
sys.stdout.write(message) sys.stdout.write(message)
except Exception as e: except Exception as e: