7 from StringIO import StringIO
12 from Cryptodome.PublicKey import RSA
13 from Cryptodome.Cipher import PKCS1_OAEP
14 from Cryptodome.Cipher import AES
15 from Cryptodome.Random import get_random_bytes
16 from Cryptodome.Hash import HMAC, SHA256
17 from Cryptodome.Util import Padding
18 import xml.etree.ElementTree as ET
20 pp = pprint.PrettyPrinter(indent=4)
22 def base64key_decode(payload):
29 raise ValueError('Invalid base64 string')
30 return base64.urlsafe_b64decode(payload.encode('utf-8'))
34 handshake_performed = False # Is a handshake already performed and the keys loaded
36 last_playback_context = ''
37 #esn = "NFCDCH-LX-CQE0NU6PA5714R25VPLXVU2A193T36"
38 esn = "WWW-BROWSE-D7GW1G4NPXGR1F0X1H3EQGY3V1F5WE"
39 #esn = "NFCDIE-02-DCH84Q2EK3N6VFVQJ0NLRQ27498N0F"
40 current_message_id = 0
41 session = requests.session()
42 rndm = random.SystemRandom()
45 'manifest': 'http://www.netflix.com/api/msl/NFCDCH-LX/cadmium/manifest',
46 'license': 'http://www.netflix.com/api/msl/NFCDCH-LX/cadmium/license'
49 def __init__(self, email, password, kodi_helper):
51 The Constructor checks for already existing crypto Keys.
52 If they exist it will load the existing keys
55 self.password = password
56 self.kodi_helper = kodi_helper
58 os.mkdir(self.kodi_helper.msl_data_path)
62 if self.file_exists(self.kodi_helper.msl_data_path, 'msl_data.json'):
63 self.kodi_helper.log(msg='MSL Data exists. Use old Tokens.')
64 self.__load_msl_data()
65 self.handshake_performed = True
66 elif self.file_exists(self.kodi_helper.msl_data_path, 'rsa_key.bin'):
67 self.kodi_helper.log(msg='RSA Keys do already exist load old ones')
68 self.__load_rsa_keys()
69 self.__perform_key_handshake()
71 self.kodi_helper.log(msg='Create new RSA Keys')
72 # Create new Key Pair and save
73 self.rsa_key = RSA.generate(2048)
74 self.__save_rsa_keys()
75 self.__perform_key_handshake()
77 def load_manifest(self, viewable_id):
79 Loads the manifets for the given viewable_id and returns a mpd-XML-Manifest
80 :param viewable_id: The id of of the viewable
81 :return: MPD XML Manifest or False if no success
83 manifest_request_data = {
85 'lookupType': 'PREPARE',
86 'viewableIds': [viewable_id],
88 "playready-h264bpl30-dash",
89 "playready-h264mpl30-dash",
90 "playready-h264mpl31-dash",
91 "playready-h264mpl40-dash",
92 # "hevc-main-L30-dash-cenc",
93 # "hevc-main-L31-dash-cenc",
94 # "hevc-main-L40-dash-cenc",
95 # "hevc-main-L41-dash-cenc",
96 # "hevc-main-L50-dash-cenc",
97 # "hevc-main-L51-dash-cenc",
98 # "hevc-main10-L30-dash-cenc",
99 # "hevc-main10-L31-dash-cenc",
100 # "hevc-main10-L40-dash-cenc",
101 # "hevc-main10-L41-dash-cenc",
102 # "hevc-main10-L50-dash-cenc",
103 # "hevc-main10-L51-dash-cenc",
104 # "hevc-main10-L30-dash-cenc-prk",
105 # "hevc-main10-L31-dash-cenc-prk",
106 # "hevc-main10-L40-dash-cenc-prk",
107 # "hevc-main10-L41-dash-cenc-prk",
108 # "hevc-main-L30-L31-dash-cenc-tl",
109 # "hevc-main-L31-L40-dash-cenc-tl",
110 # "hevc-main-L40-L41-dash-cenc-tl",
111 # "hevc-main-L50-L51-dash-cenc-tl",
112 # "hevc-main10-L30-L31-dash-cenc-tl",
113 # "hevc-main10-L31-L40-dash-cenc-tl",
114 # "hevc-main10-L40-L41-dash-cenc-tl",
115 # "hevc-main10-L50-L51-dash-cenc-tl",
116 # "hevc-dv-main10-L30-dash-cenc",
117 # "hevc-dv-main10-L31-dash-cenc",
118 # "hevc-dv-main10-L40-dash-cenc",
119 # "hevc-dv-main10-L41-dash-cenc",
120 # "hevc-dv-main10-L50-dash-cenc",
121 # "hevc-dv-main10-L51-dash-cenc",
122 # "hevc-dv5-main10-L30-dash-cenc-prk",
123 # "hevc-dv5-main10-L31-dash-cenc-prk",
124 # "hevc-dv5-main10-L40-dash-cenc-prk",
125 # "hevc-dv5-main10-L41-dash-cenc-prk",
126 # "hevc-dv5-main10-L50-dash-cenc-prk",
127 # "hevc-dv5-main10-L51-dash-cenc-prk",
128 # "hevc-hdr-main10-L30-dash-cenc",
129 # "hevc-hdr-main10-L31-dash-cenc",
130 # "hevc-hdr-main10-L40-dash-cenc",
131 # "hevc-hdr-main10-L41-dash-cenc",
132 # "hevc-hdr-main10-L50-dash-cenc",
133 # "hevc-hdr-main10-L51-dash-cenc",
134 # "hevc-hdr-main10-L30-dash-cenc-prk",
135 # "hevc-hdr-main10-L31-dash-cenc-prk",
136 # "hevc-hdr-main10-L40-dash-cenc-prk",
137 # "hevc-hdr-main10-L41-dash-cenc-prk",
138 # "hevc-hdr-main10-L50-dash-cenc-prk",
139 # "hevc-hdr-main10-L51-dash-cenc-prk"
141 # 'playready-h264mpl30-dash',
142 #'playready-h264mpl31-dash',
143 #'playready-h264mpl40-dash',
144 #'hevc-main10-L41-dash-cenc',
145 #'hevc-main10-L50-dash-cenc',
146 #'hevc-main10-L51-dash-cenc',
160 'drmSystem': 'widevine',
161 'appId': '14673889385265',
163 'pinCapableClient': False,
164 'uiplaycontext': 'null'
166 'sessionId': '14673889385265',
168 'flavor': 'PRE_FETCH',
170 'supportPreviewContent': True,
171 'forceClearStreams': False,
172 'languages': ['de-DE'],
173 'clientVersion': '4.0004.899.011',
176 request_data = self.__generate_msl_request_data(manifest_request_data)
177 resp = self.session.post(self.endpoints['manifest'], request_data)
180 # if the json() does not fail we have an error because the manifest response is a chuncked json response
182 self.kodi_helper.log(msg='Error getting Manifest: '+resp.text)
185 # json() failed so parse the chunked response
186 self.kodi_helper.log(msg='Got chunked Manifest Response: ' + resp.text)
187 resp = self.__parse_chunked_msl_response(resp.text)
188 self.kodi_helper.log(msg='Parsed chunked Response: ' + json.dumps(resp))
189 data = self.__decrypt_payload_chunk(resp['payloads'][0])
190 return self.__tranform_to_dash(data)
192 def get_license(self, challenge, sid):
194 Requests and returns a license for the given challenge and sid
195 :param challenge: The base64 encoded challenge
196 :param sid: The sid paired to the challengew
197 :return: Base64 representation of the license key or False if no success
199 license_request_data = {
201 'licenseType': 'STANDARD',
202 'clientVersion': '4.0004.899.011',
203 'uiVersion': 'akira',
204 'languages': ['de-DE'],
205 'playbackContextId': self.last_playback_context,
206 'drmContextIds': [self.last_drm_context],
208 'dataBase64': challenge,
211 'clientTime': int(time.time()),
212 'xid': int((int(time.time()) + 0.1612) * 1000)
215 request_data = self.__generate_msl_request_data(license_request_data)
217 resp = self.session.post(self.endpoints['license'], request_data)
220 # If is valid json the request for the licnese failed
222 self.kodi_helper.log(msg='Error getting license: '+resp.text)
225 # json() failed so we have a chunked json response
226 resp = self.__parse_chunked_msl_response(resp.text)
227 data = self.__decrypt_payload_chunk(resp['payloads'][0])
228 if data['success'] is True:
229 return data['result']['licenses'][0]['data']
231 self.kodi_helper.log(msg='Error getting license: ' + json.dumps(data))
234 def __decrypt_payload_chunk(self, payloadchunk):
235 payloadchunk = json.JSONDecoder().decode(payloadchunk)
236 encryption_envelope = json.JSONDecoder().decode(base64.standard_b64decode(payloadchunk['payload']))
238 cipher = AES.new(self.encryption_key, AES.MODE_CBC, base64.standard_b64decode(encryption_envelope['iv']))
239 plaintext = cipher.decrypt(base64.standard_b64decode(encryption_envelope['ciphertext']))
240 # unpad the plaintext
241 plaintext = json.JSONDecoder().decode(Padding.unpad(plaintext, 16))
242 data = plaintext['data']
244 # uncompress data if compressed
245 if plaintext['compressionalgo'] == 'GZIP':
246 data = zlib.decompress(base64.standard_b64decode(data), 16 + zlib.MAX_WBITS)
248 data = base64.standard_b64decode(data)
250 data = json.JSONDecoder().decode(data)[1]['payload']['data']
251 data = base64.standard_b64decode(data)
252 return json.JSONDecoder().decode(data)
255 def __tranform_to_dash(self, manifest):
257 self.save_file(self.kodi_helper.msl_data_path, 'manifest.json', json.dumps(manifest))
258 manifest = manifest['result']['viewables'][0]
260 self.last_playback_context = manifest['playbackContextId']
261 self.last_drm_context = manifest['drmContextId']
265 if 'psshb64' in manifest:
266 if len(manifest['psshb64']) >= 1:
267 pssh = manifest['psshb64'][0]
269 seconds = manifest['runtime']/1000
270 duration = "PT"+str(seconds)+".00S"
272 root = ET.Element('MPD')
273 root.attrib['xmlns'] = 'urn:mpeg:dash:schema:mpd:2011'
274 root.attrib['xmlns:cenc'] = 'urn:mpeg:cenc:2013'
275 root.attrib['mediaPresentationDuration'] = duration
277 period = ET.SubElement(root, 'Period', start='PT0S', duration=duration)
279 # One Adaption Set for Video
280 for video_track in manifest['videoTracks']:
281 video_adaption_set = ET.SubElement(period, 'AdaptationSet', mimeType='video/mp4', contentType="video")
283 protection = ET.SubElement(video_adaption_set, 'ContentProtection',
284 schemeIdUri='urn:uuid:EDEF8BA9-79D6-4ACE-A3C8-27DCD51D21ED')
286 ET.SubElement(protection, 'cenc:pssh').text = pssh
288 for downloadable in video_track['downloadables']:
290 hdcp_versions = '0.0'
291 for hdcp in downloadable['hdcpVersions']:
295 rep = ET.SubElement(video_adaption_set, 'Representation',
296 width=str(downloadable['width']),
297 height=str(downloadable['height']),
298 bandwidth=str(downloadable['bitrate']*1024),
300 nflxContentProfile=str(downloadable['contentProfile']),
302 mimeType='video/mp4')
305 ET.SubElement(rep, 'BaseURL').text = self.__get_base_url(downloadable['urls'])
306 # Init an Segment block
307 segment_base = ET.SubElement(rep, 'SegmentBase', indexRange="0-60000", indexRangeExact="true")
308 ET.SubElement(segment_base, 'Initialization', range='0-60000')
312 # Multiple Adaption Set for audio
313 for audio_track in manifest['audioTracks']:
314 audio_adaption_set = ET.SubElement(period, 'AdaptationSet',
315 lang=audio_track['bcp47'],
317 mimeType='audio/mp4')
318 for downloadable in audio_track['downloadables']:
321 if downloadable['contentProfile'] == 'ddplus-2.0-dash' or downloadable['contentProfile'] == 'ddplus-5.1-dash':
323 print "codec is: " + codec
324 rep = ET.SubElement(audio_adaption_set, 'Representation',
326 bandwidth=str(downloadable['bitrate']*1024),
327 mimeType='audio/mp4')
330 ET.SubElement(rep, 'AudioChannelConfiguration',
331 schemeIdUri='urn:mpeg:dash:23003:3:audio_channel_configuration:2011',
332 value=str(audio_track['channelsCount']))
335 ET.SubElement(rep, 'BaseURL').text = self.__get_base_url(downloadable['urls'])
337 segment_base = ET.SubElement(rep, 'SegmentBase', indexRange="0-60000", indexRangeExact="true")
338 ET.SubElement(segment_base, 'Initialization', range='0-60000')
341 xml = ET.tostring(root, encoding='utf-8', method='xml')
342 xml = xml.replace('\n', '').replace('\r', '')
345 def __get_base_url(self, urls):
349 def __parse_chunked_msl_response(self, message):
357 while i < len(message):
358 if message[i] == '{':
359 opencount = opencount + 1
360 if message[i] == '}':
361 closecount = closecount + 1
362 if opencount == closecount:
367 payloads.append(message[old_end:i + 1])
375 def __generate_msl_request_data(self, data):
376 header_encryption_envelope = self.__encrypt(self.__generate_msl_header())
378 'headerdata': base64.standard_b64encode(header_encryption_envelope),
379 'signature': self.__sign(header_encryption_envelope),
380 'mastertoken': self.mastertoken,
383 # Serialize the given Data
384 serialized_data = json.dumps(data)
385 serialized_data = serialized_data.replace('"', '\\"')
386 serialized_data = '[{},{"headers":{},"path":"/cbp/cadmium-13","payload":{"data":"' + serialized_data + '"},"query":""}]\n'
388 compressed_data = self.__compress_data(serialized_data)
390 # Create FIRST Payload Chunks
392 "messageid": self.current_message_id,
393 "data": compressed_data,
394 "compressionalgo": "GZIP",
398 first_payload_encryption_envelope = self.__encrypt(json.dumps(first_payload))
399 first_payload_chunk = {
400 'payload': base64.standard_b64encode(first_payload_encryption_envelope),
401 'signature': self.__sign(first_payload_encryption_envelope),
404 request_data = json.dumps(header) + json.dumps(first_payload_chunk)
409 def __compress_data(self, data):
412 with gzip.GzipFile(fileobj=out, mode="w") as f:
414 return base64.standard_b64encode(out.getvalue())
417 def __generate_msl_header(self, is_handshake=False, is_key_request=False, compressionalgo="GZIP", encrypt=True):
419 Function that generates a MSL header dict
420 :return: The base64 encoded JSON String of the header
422 self.current_message_id = self.rndm.randint(0, pow(2, 52))
426 'handshake': is_handshake,
427 'nonreplayable': False,
429 'languages': ["en-US"],
430 'compressionalgos': []
432 'recipient': 'Netflix',
434 'messageid': self.current_message_id,
435 'timestamp': 1467733923
438 # Add compression algo if not empty
439 if compressionalgo is not "":
440 header_data['capabilities']['compressionalgos'].append(compressionalgo)
442 # If this is a keyrequest act diffrent then other requests
444 public_key = base64.standard_b64encode(self.rsa_key.publickey().exportKey(format='DER'))
445 header_data['keyrequestdata'] = [{
446 'scheme': 'ASYMMETRIC_WRAPPED',
448 'publickey': public_key,
449 'mechanism': 'JWK_RSA',
450 'keypairid': 'superKeyPair'
454 if 'usertoken' in self.tokens:
457 # Auth via email and password
458 header_data['userauthdata'] = {
459 'scheme': 'EMAIL_PASSWORD',
462 'password': self.password
466 return json.dumps(header_data)
470 def __encrypt(self, plaintext):
472 Encrypt the given Plaintext with the encryption key
474 :return: Serialized JSON String of the encryption Envelope
476 iv = get_random_bytes(16)
477 encryption_envelope = {
479 'keyid': self.esn + '_' + str(self.sequence_number),
481 'iv': base64.standard_b64encode(iv)
484 plaintext = Padding.pad(plaintext, 16)
486 cipher = AES.new(self.encryption_key, AES.MODE_CBC, iv)
487 ciphertext = cipher.encrypt(plaintext)
488 encryption_envelope['ciphertext'] = base64.standard_b64encode(ciphertext)
489 return json.dumps(encryption_envelope)
492 def __sign(self, text):
494 Calculates the HMAC signature for the given text with the current sign key and SHA256
496 :return: Base64 encoded signature
498 signature = HMAC.new(self.sign_key, text, SHA256).digest()
499 return base64.standard_b64encode(signature)
502 def __perform_key_handshake(self):
503 header = self.__generate_msl_header(is_key_request=True, is_handshake=True, compressionalgo="", encrypt=False)
511 'headerdata': base64.standard_b64encode(header),
514 self.kodi_helper.log(msg='Key Handshake Request:')
515 self.kodi_helper.log(msg=json.dumps(request))
517 resp = self.session.post(self.endpoints['manifest'], json.dumps(request, sort_keys=True))
518 if resp.status_code == 200:
520 if 'errordata' in resp:
521 self.kodi_helper.log(msg='Key Exchange failed')
522 self.kodi_helper.log(msg=base64.standard_b64decode(resp['errordata']))
524 self.__parse_crypto_keys(json.JSONDecoder().decode(base64.standard_b64decode(resp['headerdata'])))
526 self.kodi_helper.log(msg='Key Exchange failed')
527 self.kodi_helper.log(msg=resp.text)
529 def __parse_crypto_keys(self, headerdata):
530 self.__set_master_token(headerdata['keyresponsedata']['mastertoken'])
532 encrypted_encryption_key = base64.standard_b64decode(headerdata['keyresponsedata']['keydata']['encryptionkey'])
533 encrypted_sign_key = base64.standard_b64decode(headerdata['keyresponsedata']['keydata']['hmackey'])
534 cipher_rsa = PKCS1_OAEP.new(self.rsa_key)
536 # Decrypt encryption key
537 encryption_key_data = json.JSONDecoder().decode(cipher_rsa.decrypt(encrypted_encryption_key))
538 self.encryption_key = base64key_decode(encryption_key_data['k'])
541 sign_key_data = json.JSONDecoder().decode(cipher_rsa.decrypt(encrypted_sign_key))
542 self.sign_key = base64key_decode(sign_key_data['k'])
544 self.__save_msl_data()
545 self.handshake_performed = True
547 def __load_msl_data(self):
548 msl_data = json.JSONDecoder().decode(self.load_file(self.kodi_helper.msl_data_path, 'msl_data.json'))
549 self.__set_master_token(msl_data['tokens']['mastertoken'])
550 self.encryption_key = base64.standard_b64decode(msl_data['encryption_key'])
551 self.sign_key = base64.standard_b64decode(msl_data['sign_key'])
553 def __save_msl_data(self):
555 Saves the keys and tokens in json file
559 "encryption_key": base64.standard_b64encode(self.encryption_key),
560 'sign_key': base64.standard_b64encode(self.sign_key),
562 'mastertoken': self.mastertoken
565 serialized_data = json.JSONEncoder().encode(data)
566 self.save_file(self.kodi_helper.msl_data_path, 'msl_data.json', serialized_data)
568 def __set_master_token(self, master_token):
569 self.mastertoken = master_token
570 self.sequence_number = json.JSONDecoder().decode(base64.standard_b64decode(master_token['tokendata']))['sequencenumber']
572 def __load_rsa_keys(self):
573 loaded_key = self.load_file(self.kodi_helper.msl_data_path, 'rsa_key.bin')
574 self.rsa_key = RSA.importKey(loaded_key)
576 def __save_rsa_keys(self):
577 self.kodi_helper.log(msg='Save RSA Keys')
578 # Get the DER Base64 of the keys
579 encrypted_key = self.rsa_key.exportKey()
580 self.save_file(self.kodi_helper.msl_data_path, 'rsa_key.bin', encrypted_key)
583 def file_exists(msl_data_path, filename):
585 Checks if a given file exists
586 :param filename: The filename
589 return os.path.isfile(msl_data_path + filename)
592 def save_file(msl_data_path, filename, content):
594 Saves the given content under given filename
595 :param filename: The filename
596 :param content: The content of the file
598 with open(msl_data_path + filename, 'w') as file_:
603 def load_file(msl_data_path, filename):
605 Loads the content of a given filename
606 :param filename: The file to load
607 :return: The content of the file
609 with open(msl_data_path + filename) as file_:
610 file_content = file_.read()