3 b`<@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl m Z mZddlmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lm Z GdddZ!dS)N) DecodeError) text_format)get_random_bytes)random) PKCS1_OAEPAES)CMACSHA256HMACSHA1)RSA)pss)Padding) wv_proto2_pb2)Session)Keyc@sNeZdZddZdddZddZd d Zd d Zd dZddZ ddZ dS)CdmcCstjt|_i|_dS)N)logging getLogger__name__loggersessions)selfr:C:\Users\Broad\Desktop\BeinConnetFix\pywidevine\cdm\cdm.py__init__s z Cdm.__init__NFc Cs|jjdj|||jjd|jdkrbdjddtdD}d}d }|||}|jd }n(|jd krztd} | }n|jj d d S|rt |t t fr|} d|_ n|j|} d|_ | rt|| ||} n|jj dd S| |j|<|jjd|S)Nz(open_session(init_data_b64={}, device={}zopening new cdm sessionZandroidcss|]}tjdVqdS)ZABCDEF0123456789N)rchoice).0_rrr sz#Cdm.open_session..Z01Z00000000000000asciiZchromezdevice type is unusableTFzunable to parse init dataz0session opened and init data parsed successfully)rdebugformatinfoZsession_id_typejoinrangeencodererror isinstancebytes bytearrayraw_pssh_parse_init_datarr) r init_data_b64ZdeviceZ raw_init_dataofflineZ rand_asciiZcounterrest session_idZ rand_bytes init_dataZ new_sessionrrr open_sessions2          zCdm.open_sessioncCstj}y(|jjd|jtj|ddWn^tk r|jjdy|jtj|dd}Wn tk r|jjddSXYnX|jjdx"t j |j D]}|jj|qW|S)Nz"trying to parse init_data directly z:unable to parse as-is, trying with removed pssh box headerz-unable to parse, unsupported init data formatz init_data:) wv_proto2ZWidevineCencHeaderrr$ParseFromStringbase64 b64decoderr*rMessageToString splitlines)rr0Zparsed_init_dataZid_byteslinerrrr/=s     zCdm._parse_init_datacCs^|jjdj||jjd||jkrD|jj||jjddS|jjdj|dSdS)Nzclose_session(session_id={})zclosing cdm sessionzcdm session closedrzsession {} not foundr#)rr$r%r&rpop)rr3rrr close_sessionNs    zCdm.close_sessioncCsV|jjdj|||jjd||jkr:|jjddS|j|}tj}y|jt j |Wn t k r|jjdYnXtj }|j r|jjdy|j|jWn t k r|jjddSXnD|jjdy|jt j |Wn"t k r|jjddSX|jjd x$tj|jD]}|jj|q0W||_d |_d S) Nz/set_service_certificate(session_id={}, cert={})zsetting service certificatezsession id doesn't existr#z%failed to parse cert as SignedMessagez&service cert provided as signedmessagez#failed to parse service certificatez0service cert provided as signeddevicecertificatezservice certificate:Tr)rr$r%r&rr*r7Z SignedMessager8r9r:rZSignedDeviceCertificateTypeMsgrr;r<service_certificate privacy_mode)rr3Zcert_b64sessionmessagerBr=rrrset_service_certificateYs>         zCdm.set_service_certificatec"Cs|jjdj||jjd||jkr8|jjddS|j|}|jrRtj}ntj }tj }t j j |jjs|jjddSt|jjd:}y|j|j}Wn tk r|jjddSXWdQRX|jjd|js tj jjd |_|jjjjj|jntjjjd |_|j|jjj_|jr>tjjd }n tjjd }||jjj_||jjj_tj j!jd |j_t"t#j#|j_$tj%jd |j_%|jj&rt'j(dd|j_)|j*r|jj+r<|jjd|jjdtj,}t|jj-d<}y|j|j} Wn"tk r$|jjddSXWdQRX|j.j||jjd|jjdx$t/j0|j1D]} |jj| qdWt2d} t2d} t3j4| t3j5| } | j6t7j8|j9d}t:j;|j}t?j4|}|j6| }tj@}|js"