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:
+18
-4
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user