1 #!/usr/local/bin/python3
3 # Copyright (c) 2014 The FreeBSD Foundation
4 # Copyright 2014 John-Mark Gurney
6 # Copyright 2019 Enji Cooper
8 # This software was developed by John-Mark Gurney under
9 # the sponsorship from the FreeBSD Foundation.
10 # Redistribution and use in source and binary forms, with or without
11 # modification, are permitted provided that the following conditions
13 # 1. Redistributions of source code must retain the above copyright
14 # notice, this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 # notice, this list of conditions and the following disclaimer in the
17 # documentation and/or other materials provided with the distribution.
19 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
37 from fcntl import ioctl
43 from struct import pack as _pack
49 from cryptodevh import *
51 __all__ = [ 'Crypto', 'MismatchError', ]
53 class FindOp(dpkt.Packet):
60 class SessionOp(dpkt.Packet):
67 ('mackeylen', 'i', 0),
72 class SessionOp2(dpkt.Packet):
79 ('mackeylen', 'i', 0),
89 class CryptOp(dpkt.Packet):
102 class CryptAEAD(dpkt.Packet):
118 # h2py.py can't handle multiarg macros
119 CIOCGSESSION = 3224396645
120 CIOCFSESSION = 2147771238
122 CIOCASYMFEAT = 1074029417
123 CIOCKEY2 = 3230688107
124 CIOCFINDDEV = 3223610220
125 if platform.architecture()[0] == '64bit':
126 CIOCGSESSION2 = 3225445226
127 CIOCCRYPT = 3224396647
128 CIOCCRYPTAEAD = 3225445229
130 CIOCGSESSION2 = 3224396650
131 CIOCCRYPT = 3223085927
132 CIOCCRYPTAEAD = 3223872365
134 _cryptodev = os.open('/dev/crypto', os.O_RDWR)
136 def str_to_ascii(val):
137 if sys.version_info[0] >= 3:
138 if isinstance(val, str):
139 return val.encode("ascii")
142 def _findop(crid, name):
145 fop.name = str_to_ascii(name)
146 s = array.array('B', fop.pack_hdr())
147 ioctl(_cryptodev, CIOCFINDDEV, s, 1)
151 idx = fop.name.index(b'\x00')
152 name = fop.name[:idx]
156 return fop.crid, name
158 def array_tobytes(array_obj):
159 if sys.version_info[:2] >= (3, 2):
160 return array_obj.tobytes()
161 return array_obj.tostring()
164 if sys.version_info[0] >= 3:
171 return _findop(-1, name)[0]
174 def getcridname(crid):
175 return _findop(crid, '')[1]
177 def __init__(self, cipher=0, key=None, mac=0, mackey=None,
178 crid=CRYPTOCAP_F_SOFTWARE | CRYPTOCAP_F_HARDWARE, maclen=None,
181 self._maclen = maclen
187 ses.keylen = len(key)
188 k = array.array('B', key)
189 ses.key = k.buffer_info()[0]
193 if mackey is not None:
194 ses.mackeylen = len(mackey)
195 mk = array.array('B', mackey)
196 ses.mackey = mk.buffer_info()[0]
198 if not cipher and not mac:
199 raise ValueError('one of cipher or mac MUST be specified.')
206 s = array.array('B', ses.pack_hdr())
208 ioctl(_cryptodev, CIOCGSESSION2, s, 1)
214 if self._ses is None:
218 ioctl(_cryptodev, CIOCFSESSION, _pack('I', self._ses))
223 def _doop(self, op, src, iv, mac=None):
230 s = array.array('B', src)
231 cop.src = cop.dst = s.buffer_info()[0]
233 assert len(mac) == self._maclen, \
234 '%d != %d' % (len(tag), self._maclen)
235 if self._maclen is not None:
237 m = array.array('B', [0] * self._maclen)
239 m = array.array('B', mac)
240 cop.mac = m.buffer_info()[0]
241 ivbuf = array.array('B', str_to_ascii(iv))
242 cop.iv = ivbuf.buffer_info()[0]
245 ioctl(_cryptodev, CIOCCRYPT, bytes(cop))
251 if self._maclen is not None:
252 return s, array_tobytes(m)
256 def _doaead(self, op, src, aad, iv, tag=None):
258 caead.ses = self._ses
260 caead.flags = CRD_F_IV_EXPLICIT
263 src = str_to_ascii(src)
265 s = array.array('B', src)
266 caead.src = caead.dst = s.buffer_info()[0]
267 aad = str_to_ascii(aad)
268 caead.aadlen = len(aad)
269 saad = array.array('B', aad)
270 caead.aad = saad.buffer_info()[0]
272 if self._maclen is None:
273 raise ValueError('must have a tag length')
275 tag = str_to_ascii(tag)
277 tag = array.array('B', [0] * self._maclen)
279 assert len(tag) == self._maclen, \
280 '%d != %d' % (len(tag), self._maclen)
281 tag = array.array('B', tag)
283 caead.tag = tag.buffer_info()[0]
285 ivbuf = array.array('B', iv)
286 caead.ivlen = len(iv)
287 caead.iv = ivbuf.buffer_info()[0]
289 ioctl(_cryptodev, CIOCCRYPTAEAD, bytes(caead))
296 return s, array_tobytes(tag)
298 def perftest(self, op, size, timeo=3):
299 inp = array.array('B', (random.randint(0, 255) for x in range(size)))
300 inp = str_to_ascii(inp)
301 out = array.array('B', inp)
309 s = array.array('B', inp)
310 cop.src = s.buffer_info()[0]
311 cop.dst = out.buffer_info()[0]
312 if self._maclen is not None:
313 m = array.array('B', [0] * self._maclen)
314 cop.mac = m.buffer_info()[0]
315 ivbuf = array.array('B', (random.randint(0, 255) for x in range(16)))
316 cop.iv = ivbuf.buffer_info()[0]
319 def alarmhandle(a, b, exit=exit):
322 oldalarm = signal.signal(signal.SIGALRM, alarmhandle)
329 ioctl(_cryptodev, CIOCCRYPT, cop)
334 signal.signal(signal.SIGALRM, oldalarm)
336 print('time:', end - start)
337 print('perf MB/sec:', (reps * size) / (end - start) / 1024 / 1024)
339 def encrypt(self, data, iv, aad=None):
341 return self._doop(COP_ENCRYPT, data, iv)
343 return self._doaead(COP_ENCRYPT, data, aad,
346 def decrypt(self, data, iv, aad=None, tag=None):
348 return self._doop(COP_DECRYPT, data, iv, mac=tag)
350 return self._doaead(COP_DECRYPT, data, aad,
353 class MismatchError(Exception):
357 def __init__(self, fname, fields):
358 self.fields = set(fields)
362 self.field_re = re.compile(r"\[(?P<field>[^]]+)\]")
365 self.fp = open(self.fname)
368 def __exit__(self, exc_type, exc_value, exc_tb):
369 if self.fp is not None:
378 if self._pending is not None:
382 i = self.fp.readline()
386 if not i.startswith('#') and i.strip():
389 matches = self.field_re.match(i)
391 raise ValueError("Unknown line: %r" % (i))
392 yield matches.group("field"), self.fielditer()
396 line = self.fp.readline()
410 line = self.eatblanks()
411 if not line or line[0] == '[':
417 f, v = line.split(' =')
422 print('line:', repr(line))
427 raise ValueError('already present: %r' % repr(f))
429 line = self.fp.readline().strip()
433 # we should have everything
434 remain = self.fields.copy() - set(values.keys())
435 # XXX - special case GCM decrypt
436 if remain and not ('FAIL' in values and 'PT' in remain):
437 raise ValueError('not all fields found: %r' % repr(remain))
441 # The CCM files use a bit of a different syntax that doesn't quite fit
442 # the generic KATParser. In particular, some keys are set globally at
443 # the start of the file, and some are set globally at the start of a
446 def __init__(self, fname):
452 self.fp = open(self.fname)
456 def __exit__(self, exc_type, exc_value, exc_tb):
457 if self.fp is not None:
460 def read_globals(self):
461 self.global_values = {}
463 line = self.fp.readline()
466 if line[0] == '#' or not line.strip():
473 f, v = line.split(' =')
475 print('line:', repr(line))
480 if f in self.global_values:
481 raise ValueError('already present: %r' % repr(f))
482 self.global_values[f] = v
484 def read_section_values(self, kwpairs):
485 self.section_values = self.global_values.copy()
486 for pair in kwpairs.split(', '):
487 f, v = pair.split(' = ')
488 if f in self.section_values:
489 raise ValueError('already present: %r' % repr(f))
490 self.section_values[f] = v
493 line = self.fp.readline()
496 if line[0] == '#' or not line.strip():
503 f, v = line.split(' =')
505 print('line:', repr(line))
514 if f in self.section_values:
515 raise ValueError('already present: %r' % repr(f))
516 self.section_values[f] = v
527 line = self.fp.readline()
531 if (line and line[0] == '#') or not line.strip():
535 section = line[1:].split(']', 1)[0]
536 self.read_section_values(section)
539 values = self.section_values.copy()
543 f, v = line.split(' =')
545 print('line:', repr(line))
550 raise ValueError('already present: %r' % repr(f))
552 line = self.fp.readline().strip()
559 return binascii.hexlify(''.join(s.split()))
561 if sys.version_info[0] < 3:
562 KATCCMParser.next = KATCCMParser.__next__
563 KATParser.next = KATParser.__next__
565 if __name__ == '__main__':
568 crid = Crypto.findcrid('aesni0')
569 print('aesni:', crid)
571 print('aesni0 not found')
575 name = Crypto.getcridname(i)
576 print('%2d: %r' % (i, repr(name)))
580 columns = [ 'COUNT', 'DataUnitLen', 'Key', 'DataUnitSeqNumber', 'PT', 'CT' ]
581 fname = '/usr/home/jmg/aesni.testing/format tweak value input - data unit seq no/XTSGenAES128.rsp'
582 with KATParser(fname, columns) as kp:
588 key = _spdechex('c939cc13397c1d37de6ae0e1cb7c423c')
589 iv = _spdechex('00000000000000000000000000000001')
590 pt = _spdechex('ab3cabed693a32946055524052afe3c9cb49664f09fc8b7da824d924006b7496353b8c1657c5dec564d8f38d7432e1de35aae9d95590e66278d4acce883e51abaf94977fcd3679660109a92bf7b2973ccd547f065ec6cee4cb4a72a5e9f45e615d920d76cb34cba482467b3e21422a7242e7d931330c0fbf465c3a3a46fae943029fd899626dda542750a1eee253df323c6ef1573f1c8c156613e2ea0a6cdbf2ae9701020be2d6a83ecb7f3f9d8e')
591 #pt = _spdechex('00000000000000000000000000000000')
592 ct = _spdechex('f42c33853ecc5ce2949865fdb83de3bff1089e9360c94f830baebfaff72836ab5236f77212f1e7396c8c54ac73d81986375a6e9e299cfeca5ba051ed25e8d1affa5beaf6c1d2b45e90802408f2ced21663497e906de5f29341e5e52ddfea5363d628b3eb7806835e17bae051b3a6da3f8e2941fe44384eac17a9d298d2c331ca8320c775b5d53263a5e905059d891b21dede2d8110fd427c7bd5a9a274ddb47b1945ee79522203b6e297d0e399ef')
594 c = Crypto(CRYPTO_AES_ICM, key)
595 enc = c.encrypt(pt, iv)
597 print('enc:', binascii.hexlify(enc))
598 print(' ct:', binascii.hexlify(ct))
602 dec = c.decrypt(ct, iv)
604 print('dec:', binascii.hexlify(dec))
605 print(' pt:', binascii.hexlify(pt))
609 key = _spdechex('c939cc13397c1d37de6ae0e1cb7c423c')
610 iv = _spdechex('00000000000000000000000000000001')
611 pt = _spdechex('ab3cabed693a32946055524052afe3c9cb49664f09fc8b7da824d924006b7496353b8c1657c5dec564d8f38d7432e1de35aae9d95590e66278d4acce883e51abaf94977fcd3679660109a92bf7b2973ccd547f065ec6cee4cb4a72a5e9f45e615d920d76cb34cba482467b3e21422a7242e7d931330c0fbf465c3a3a46fae943029fd899626dda542750a1eee253df323c6ef1573f1c8c156613e2ea0a6cdbf2ae9701020be2d6a83ecb7f3f9d8e0a3f')
612 #pt = _spdechex('00000000000000000000000000000000')
613 ct = _spdechex('f42c33853ecc5ce2949865fdb83de3bff1089e9360c94f830baebfaff72836ab5236f77212f1e7396c8c54ac73d81986375a6e9e299cfeca5ba051ed25e8d1affa5beaf6c1d2b45e90802408f2ced21663497e906de5f29341e5e52ddfea5363d628b3eb7806835e17bae051b3a6da3f8e2941fe44384eac17a9d298d2c331ca8320c775b5d53263a5e905059d891b21dede2d8110fd427c7bd5a9a274ddb47b1945ee79522203b6e297d0e399ef3768')
615 c = Crypto(CRYPTO_AES_ICM, key)
616 enc = c.encrypt(pt, iv)
618 print('enc:', binascii.hexlify(enc))
619 print(' ct:', binascii.hexlify(ct))
623 dec = c.decrypt(ct, iv)
625 print('dec:', binascii.hexlify(dec))
626 print(' pt:', binascii.hexlify(pt))
630 key = _spdechex('c939cc13397c1d37de6ae0e1cb7c423c')
631 iv = _spdechex('6eba2716ec0bd6fa5cdef5e6d3a795bc')
632 pt = _spdechex('ab3cabed693a32946055524052afe3c9cb49664f09fc8b7da824d924006b7496353b8c1657c5dec564d8f38d7432e1de35aae9d95590e66278d4acce883e51abaf94977fcd3679660109a92bf7b2973ccd547f065ec6cee4cb4a72a5e9f45e615d920d76cb34cba482467b3e21422a7242e7d931330c0fbf465c3a3a46fae943029fd899626dda542750a1eee253df323c6ef1573f1c8c156613e2ea0a6cdbf2ae9701020be2d6a83ecb7f3f9d8e0a3f')
633 ct = _spdechex('f1f81f12e72e992dbdc304032705dc75dc3e4180eff8ee4819906af6aee876d5b00b7c36d282a445ce3620327be481e8e53a8e5a8e5ca9abfeb2281be88d12ffa8f46d958d8224738c1f7eea48bda03edbf9adeb900985f4fa25648b406d13a886c25e70cfdecdde0ad0f2991420eb48a61c64fd797237cf2798c2675b9bb744360b0a3f329ac53bbceb4e3e7456e6514f1a9d2f06c236c31d0f080b79c15dce1096357416602520daa098b17d1af427')
634 c = Crypto(CRYPTO_AES_CBC, key)
636 enc = c.encrypt(pt, iv)
638 print('enc:', binascii.hexlify(enc))
639 print(' ct:', binascii.hexlify(ct))
643 dec = c.decrypt(ct, iv)
645 print('dec:', binascii.hexlify(dec))
646 print(' pt:', binascii.hexlify(pt))
650 key = _spdechex('c939cc13397c1d37de6ae0e1cb7c423c')
651 iv = _spdechex('b3d8cc017cbb89b39e0f67e2')
652 pt = _spdechex('c3b3c41f113a31b73d9a5cd4321030')
653 aad = _spdechex('24825602bd12a984e0092d3e448eda5f')
654 ct = _spdechex('93fe7d9e9bfd10348a5606e5cafa7354')
655 ct = _spdechex('93fe7d9e9bfd10348a5606e5cafa73')
656 tag = _spdechex('0032a1dc85f1c9786925a2e71d8272dd')
657 tag = _spdechex('8d11a0929cb3fbe1fef01a4a38d5f8ea')
659 c = Crypto(CRYPTO_AES_NIST_GCM_16, key)
661 enc, enctag = c.encrypt(pt, iv, aad=aad)
663 print('enc:', binascii.hexlify(enc))
664 print(' ct:', binascii.hexlify(ct))
668 print('etg:', binascii.hexlify(enctag))
669 print('tag:', binascii.hexlify(tag))
672 # Make sure we get EBADMSG
673 #enctag = enctag[:-1] + 'a'
674 dec, dectag = c.decrypt(ct, iv, aad=aad, tag=enctag)
676 print('dec:', binascii.hexlify(dec))
677 print(' pt:', binascii.hexlify(pt))
681 print('dtg:', binascii.hexlify(dectag))
682 print('tag:', binascii.hexlify(tag))
686 key = _spdechex('c939cc13397c1d37de6ae0e1cb7c423c')
687 iv = _spdechex('b3d8cc017cbb89b39e0f67e2')
690 pt = _spdechex('c3b3c41f113a31b73d9a5cd432103069')
691 aad = _spdechex('24825602bd12a984e0092d3e448eda5f')
692 ct = _spdechex('93fe7d9e9bfd10348a5606e5cafa7354')
693 tag = _spdechex('0032a1dc85f1c9786925a2e71d8272dd')
695 c = Crypto(CRYPTO_AES_GCM_16, key)
697 enc, enctag = c.encrypt(pt, iv, aad=aad)
699 print('enc:', binascii.hexlify(enc))
700 print(' ct:', binascii.hexlify(ct))
704 print('etg:', binascii.hexlify(enctag))
705 print('tag:', binascii.hexlify(tag))
708 for i in range(100000):
709 c = Crypto(CRYPTO_AES_XTS, binascii.unhexlify('1bbfeadf539daedcae33ced497343f3ca1f2474ad932b903997d44707db41382'))
710 data = binascii.unhexlify('52a42bca4e9425a25bbc8c8bf6129dec')
711 ct = binascii.unhexlify('517e602becd066b65fa4f4f56ddfe240')
712 iv = _pack('QQ', 71, 0)
714 enc = c.encrypt(data, iv)
717 c = Crypto(CRYPTO_AES_XTS, binascii.unhexlify('1bbfeadf539daedcae33ced497343f3ca1f2474ad932b903997d44707db41382'))
718 data = binascii.unhexlify('52a42bca4e9425a25bbc8c8bf6129dec')
719 ct = binascii.unhexlify('517e602becd066b65fa4f4f56ddfe240')
720 iv = _pack('QQ', 71, 0)
722 enc = c.encrypt(data, iv)
725 dec = c.decrypt(enc, iv)
728 #c.perftest(COP_ENCRYPT, 192*1024, reps=30000)
731 key = binascii.unhexlify('1bbfeadf539daedcae33ced497343f3ca1f2474ad932b903997d44707db41382')
732 print('XTS %d testing:' % (len(key) * 8))
733 c = Crypto(CRYPTO_AES_XTS, key)
734 for i in [ 8192, 192*1024]:
735 print('block size: %d' % i)
736 c.perftest(COP_ENCRYPT, i)
737 c.perftest(COP_DECRYPT, i)