Mirror-Tools
/
shaka-packager
mirror of https://github.com/shaka-project/shaka-packager.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add decryption verification tests 

Change-Id: I735dc14c618537bc4a69fcfbe2bc634bfe2a2166
This commit is contained in:
Kongqun Yang 2016-04-21 16:27:05 -07:00 committed by KongQun Yang
parent 855d1b028f
commit 976bf69294
6 changed files with 45 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
packager/app/test/packager_test.py
View File

@ -66,8 +66,8 @@ class PackagerAppTest(unittest.TestCase):
' sampling_frequency: 44100\n' ' sampling_frequency: 44100\n'
' language: und\n') ' language: und\n')
self.assertIn(expected_stream_info, stream_info, self.assertIn(expected_stream_info, stream_info,
'\nExpecting: \n %s\n\nBut seeing: \n%s' % ( '\nExpecting: \n %s\n\nBut seeing: \n%s' %
expected_stream_info, stream_info)) (expected_stream_info, stream_info))
def testPackageFirstStream(self): def testPackageFirstStream(self):
self.packager.Package(self._GetStreams(['0']), self._GetFlags()) self.packager.Package(self._GetStreams(['0']), self._GetFlags())
@ -76,8 +76,7 @@ class PackagerAppTest(unittest.TestCase):
def testPackageText(self): def testPackageText(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['text'], self._GetStreams(['text'], test_files=['subtitle-english.vtt']),
test_files=['subtitle-english.vtt']),
self._GetFlags()) self._GetFlags())
self._DiffGold(self.output[0], 'subtitle-english-golden.vtt') self._DiffGold(self.output[0], 'subtitle-english-golden.vtt')
self._DiffGold(self.mpd_output, 'subtitle-english-vtt-golden.mpd') self._DiffGold(self.mpd_output, 'subtitle-english-vtt-golden.mpd')
@ -135,6 +134,8 @@ class PackagerAppTest(unittest.TestCase):
self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4')
self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4')
self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4')
self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4')
def testPackageWebmWithEncryption(self): def testPackageWebmWithEncryption(self):
self.packager.Package( self.packager.Package(
@ -144,14 +145,16 @@ class PackagerAppTest(unittest.TestCase):
self._GetFlags(encryption=True)) self._GetFlags(encryption=True))
self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.webm') self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.webm')
self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-webm-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-webm-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.webm')
def testPackageHevcWithEncryption(self): def testPackageHevcWithEncryption(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['video'], self._GetStreams(['video'],
test_files=['bear-640x360-hevc.mp4']), test_files=['bear-640x360-hevc.mp4']),
self._GetFlags(encryption=True)) self._GetFlags(encryption=True))
self._DiffGold(self.output[0], 'bear-640x360-hevc-v-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-640x360-hevc-cenc-golden.mp4')
self._DiffGold(self.mpd_output, 'bear-640x360-hevc-v-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-hevc-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-hevc-golden.mp4')
def testPackageVp8Mp4WithEncryption(self): def testPackageVp8Mp4WithEncryption(self):
self.packager.Package( self.packager.Package(
@ -161,6 +164,7 @@ class PackagerAppTest(unittest.TestCase):
self._GetFlags(encryption=True)) self._GetFlags(encryption=True))
self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-640x360-vp8-cenc-golden.mp4')
self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-vp8-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-vp8-golden.mp4')
def testPackageVp9Mp4WithEncryption(self): def testPackageVp9Mp4WithEncryption(self):
self.packager.Package( self.packager.Package(
@ -170,12 +174,12 @@ class PackagerAppTest(unittest.TestCase):
self._GetFlags(encryption=True)) self._GetFlags(encryption=True))
self._DiffGold(self.output[0], 'bear-320x240-vp9-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-320x240-vp9-cenc-golden.mp4')
self._DiffGold(self.mpd_output, 'bear-320x240-vp9-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-320x240-vp9-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-vp9-golden.mp4')
def testPackageWithEncryptionAndRandomIv(self): def testPackageWithEncryptionAndRandomIv(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['audio', 'video']), self._GetStreams(['audio', 'video']),
self._GetFlags(encryption=True, self._GetFlags(encryption=True, random_iv=True))
random_iv=True))
self._AssertStreamInfo(self.output[0], 'is_encrypted: true') self._AssertStreamInfo(self.output[0], 'is_encrypted: true')
self._AssertStreamInfo(self.output[1], 'is_encrypted: true') self._AssertStreamInfo(self.output[1], 'is_encrypted: true')
# The outputs are encrypted with random iv, so they are not the same as # The outputs are encrypted with random iv, so they are not the same as
@ -185,12 +189,13 @@ class PackagerAppTest(unittest.TestCase):
self.assertFalse(self._CompareWithGold(self.output[1], self.assertFalse(self._CompareWithGold(self.output[1],
'bear-640x360-v-cenc-golden.mp4')) 'bear-640x360-v-cenc-golden.mp4'))
self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4')
self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4')
def testPackageWithEncryptionAndRealClock(self): def testPackageWithEncryptionAndRealClock(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['audio', 'video']), self._GetStreams(['audio', 'video']),
self._GetFlags(encryption=True, self._GetFlags(encryption=True, use_fake_clock=False))
use_fake_clock=False))
self._AssertStreamInfo(self.output[0], 'is_encrypted: true') self._AssertStreamInfo(self.output[0], 'is_encrypted: true')
self._AssertStreamInfo(self.output[1], 'is_encrypted: true') self._AssertStreamInfo(self.output[1], 'is_encrypted: true')
# The outputs are generated with real clock, so they are not the same as # The outputs are generated with real clock, so they are not the same as
@ -200,12 +205,13 @@ class PackagerAppTest(unittest.TestCase):
self.assertFalse(self._CompareWithGold(self.output[1], self.assertFalse(self._CompareWithGold(self.output[1],
'bear-640x360-v-cenc-golden.mp4')) 'bear-640x360-v-cenc-golden.mp4'))
self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-golden.mpd')
self._VerifyDecryption(self.output[0], 'bear-640x360-a-golden.mp4')
self._VerifyDecryption(self.output[1], 'bear-640x360-v-golden.mp4')
def testPackageWithEncryptionAndDashIfIop(self): def testPackageWithEncryptionAndDashIfIop(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['audio', 'video']), self._GetStreams(['audio', 'video']),
self._GetFlags(encryption=True, self._GetFlags(encryption=True, dash_if_iop=True))
dash_if_iop=True))
self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4')
self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4')
self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-iop-golden.mpd') self._DiffGold(self.mpd_output, 'bear-640x360-av-cenc-iop-golden.mpd')
@ -213,8 +219,7 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithEncryptionAndOutputMediaInfo(self): def testPackageWithEncryptionAndOutputMediaInfo(self):
self.packager.Package( self.packager.Package(
self._GetStreams(['audio', 'video']), self._GetStreams(['audio', 'video']),
self._GetFlags(encryption=True, self._GetFlags(encryption=True, output_media_info=True))
output_media_info=True))
self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') self._DiffGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4')
self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4') self._DiffGold(self.output[1], 'bear-640x360-v-cenc-golden.mp4')
self._DiffMediaInfoGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4') self._DiffMediaInfoGold(self.output[0], 'bear-640x360-a-cenc-golden.mp4')
@ -223,8 +228,7 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithLiveProfile(self): def testPackageWithLiveProfile(self):
self.packager.Package( self.packager.Package(
self._GetStreams( self._GetStreams(
['audio', 'video'], ['audio', 'video'], live=True),
live=True),
self._GetFlags(live=True)) self._GetFlags(live=True))
self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden') self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-golden')
self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden') self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-golden')
@ -233,10 +237,8 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithLiveProfileAndEncryption(self): def testPackageWithLiveProfileAndEncryption(self):
self.packager.Package( self.packager.Package(
self._GetStreams( self._GetStreams(
['audio', 'video'], ['audio', 'video'], live=True),
live=True), self._GetFlags(encryption=True, live=True))
self._GetFlags(encryption=True,
live=True))
self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden')
self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden')
self._DiffLiveMpdGold(self.mpd_output, self._DiffLiveMpdGold(self.mpd_output,
@ -245,11 +247,9 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithLiveProfileAndEncryptionAndDashIfIop(self): def testPackageWithLiveProfileAndEncryptionAndDashIfIop(self):
self.packager.Package( self.packager.Package(
self._GetStreams( self._GetStreams(
['audio', 'video'], ['audio', 'video'], live=True),
live=True),
self._GetFlags(encryption=True, self._GetFlags(encryption=True,
live=True, live=True, dash_if_iop=True))
dash_if_iop=True))
self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden') self._DiffLiveGold(self.output[0], 'bear-640x360-a-live-cenc-golden')
self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden') self._DiffLiveGold(self.output[1], 'bear-640x360-v-live-cenc-golden')
self._DiffLiveMpdGold(self.mpd_output, self._DiffLiveMpdGold(self.mpd_output,
@ -263,8 +263,7 @@ class PackagerAppTest(unittest.TestCase):
test_files=['bear-1280x720.mp4', 'bear-640x360.mp4', test_files=['bear-1280x720.mp4', 'bear-640x360.mp4',
'bear-320x180.mp4']), 'bear-320x180.mp4']),
self._GetFlags(encryption=True, self._GetFlags(encryption=True,
live=True, live=True, dash_if_iop=True))
dash_if_iop=True))
self._DiffLiveGold(self.output[2], 'bear-640x360-a-live-cenc-golden') self._DiffLiveGold(self.output[2], 'bear-640x360-a-live-cenc-golden')
self._DiffLiveGold(self.output[3], 'bear-640x360-v-live-cenc-golden') self._DiffLiveGold(self.output[3], 'bear-640x360-v-live-cenc-golden')
# Mpd cannot be validated right now since we don't generate determinstic # Mpd cannot be validated right now since we don't generate determinstic
@ -274,8 +273,7 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithLiveProfileAndKeyRotation(self): def testPackageWithLiveProfileAndKeyRotation(self):
self.packager.Package( self.packager.Package(
self._GetStreams( self._GetStreams(
['audio', 'video'], ['audio', 'video'], live=True),
live=True),
self._GetFlags(encryption=True, self._GetFlags(encryption=True,
key_rotation=True, key_rotation=True,
live=True)) live=True))
@ -289,8 +287,7 @@ class PackagerAppTest(unittest.TestCase):
def testPackageWithLiveProfileAndKeyRotationAndDashIfIop(self): def testPackageWithLiveProfileAndKeyRotationAndDashIfIop(self):
self.packager.Package( self.packager.Package(
self._GetStreams( self._GetStreams(
['audio', 'video'], ['audio', 'video'], live=True),
live=True),
self._GetFlags(encryption=True, self._GetFlags(encryption=True,
key_rotation=True, key_rotation=True,
live=True, live=True,
@ -371,6 +368,7 @@ class PackagerAppTest(unittest.TestCase):
def _GetFlags(self, def _GetFlags(self,
encryption=False, encryption=False,
decryption=False,
random_iv=False, random_iv=False,
widevine_encryption=False, widevine_encryption=False,
key_rotation=False, key_rotation=False,
@ -392,10 +390,16 @@ class PackagerAppTest(unittest.TestCase):
'31323334353637383930313233343536') # Data '31323334353637383930313233343536') # Data
flags += ['--enable_fixed_key_encryption', flags += ['--enable_fixed_key_encryption',
'--key_id=31323334353637383930313233343536', '--key_id=31323334353637383930313233343536',
'--key=32333435363738393021323334353637', '--key=32333435363738393021323334353637', '--pssh=' + pssh_box,
'--pssh=' + pssh_box, '--clear_lead=1'] '--clear_lead=1']
if not random_iv: if not random_iv:
flags.append('--iv=3334353637383930') flags.append('--iv=3334353637383930')
if decryption:
flags += ['--enable_fixed_key_decryption',
'--key_id=31323334353637383930313233343536',
'--key=32333435363738393021323334353637']
if key_rotation: if key_rotation:
flags.append('--crypto_period_duration=1') flags.append('--crypto_period_duration=1')
@ -489,6 +493,15 @@ class PackagerAppTest(unittest.TestCase):
self.assertIn('Found 1 stream(s).', stream_info) self.assertIn('Found 1 stream(s).', stream_info)
self.assertIn(info, stream_info) self.assertIn(info, stream_info)
def _VerifyDecryption(self, test_encrypted_file, golden_clear_file):
output_extension = os.path.splitext(golden_clear_file)[1][1:]
self.packager.Package(
self._GetStreams(['0'],
output_format=output_extension,
test_files=[test_encrypted_file]),
self._GetFlags(decryption=True))
self._DiffGold(self.output[-1], golden_clear_file)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

0
packager/app/test/testdata/bear-640x360-hevc-v-cenc-golden.mp4 → packager/app/test/testdata/bear-640x360-hevc-cenc-golden.mp4 vendored
View File

0
packager/app/test/testdata/bear-640x360-hevc-v-cenc-golden.mpd → packager/app/test/testdata/bear-640x360-hevc-cenc-golden.mpd vendored
View File

BIN
packager/app/test/testdata/bear-640x360-hevc-golden.mp4 vendored Normal file
View File

Binary file not shown.

BIN
packager/app/test/testdata/bear-640x360-vp8-golden.mp4 vendored Normal file
View File

Binary file not shown.

BIN
packager/app/test/testdata/bear-640x360-vp9-golden.mp4 vendored Normal file
View File

Binary file not shown.