i!ddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z eZdd eed ed ed eed ededefdZdd eed ed ed eed edef dZy)N)Console)Cdm)Device) DeviceTypes) RemoteCdm)PSSH)create_client_curl pssh_list license_urlcdm_device_pathcdm_remote_apiheaderskey kid_to_labelc.|rj|jd}t|dk(rJ|djddjd|djddjgSy||tj dyt ||||||S) a Extract Widevine CONTENT keys (KID/KEY) from a license. Args: - pssh_list (list[dict]): List of dicts {'pssh': ..., 'kid': ..., 'type': ...} - license_url (str): Widevine license URL. - cdm_device_path (str): Path to local CDM file (device.wvd). Optional if using remote. - cdm_remote_api (list[str]): Remote CDM API config. Optional if using local device. - headers (dict): Optional HTTP headers for the license request (from fetch). - key (str): Optional raw license data to bypass HTTP request. - kid_to_label (dict): Mapping from KID (hex) to track label. Returns: list: List of strings "KID:KEY" (only CONTENT keys) or None if error. :r-NzB[red]Error: Must provide either cdm_device_path or cdm_remote_api.)splitlenreplacestripconsoleprint_get_widevine_keys)r r r r rrrk_splits Y/home/nidran/Documenti/script/streaming-community/StreamingCommunity/core/drm/widevine.pyget_widevine_keysr s" ))C. w<1 qz))#r288:;1WQZ=O=OPSUW=X=^=^=`#9 Z[ io~W^`l mmc  d}d}t}|D]c} t| jddjddj j } | sM| dk7sS|j | e|Atjd tj|}tj|}nutjd |d d k(rtj|d <n8|d d k(rtj |d <ntjd |d yt#d:i|}|j%} g} t} t'|D]\}} | d}t| jddjddj j }| jdd}tjd|d|ddd|d |j)| t+|}|r|j-ni}d|vrd|d<|tjd t/|j1||}|j2dk7r:tjd |j2d!|j4ddd"|d#|?|j6}|j8jdd}d$|vrJ |j;}d%|vrt=j>|d%}ntjd&|d' |stjd) |jA| | |jC| D]}|jDd+k7r|jFjHj j } tKd,| DrX| |vrtjd-| d.| |vrz| d/|jLjI}|| vs| jO||j |  |j d1k(rtQ|tQ|k\rn#tQ|d2k(stQ| d2k\sn| ryt'| D]j\}}|jSd/\} }|dd3d4z}|r|j| j nd}|rd5|nd}tjd6| d7||lntjd8| r| nd |jU| S#t$r"} tjd| Yd} ~ yd} ~ wwxYw#t$r"} tjd | Yd} ~ yd} ~ wwxYw#t$r)} tjd|ddd| Yd} ~ 7d} ~ wwxYw#t$r#} tjd| Yd} ~ fd} ~ wwxYw#t$r#} tjd(| Yd} ~ d} ~ wwxYw#t$r#} tjd*| Yd} ~ d} ~ wwxYw#t$r#} tjd0| Yd} ~ d} ~ wwxYw#t$rYSwxYw#t$rC} tjd9| Yd} ~ |jU| y#t$rYywxYwd} ~ wwxYw# |jU| w#t$rYwwxYwxYw);z7Extract Widevine keys using local or remote CDM device.Nkidrrzn/az[cyan]Using local CDM.z%[red]Error loading local CDM device: z[cyan]Using remote CDM. device_typeANDROIDCHROMEz)[red]Unsupported remote CDM device type: z$[red]Error initializing remote CDM: psshzN/Atypeunknownz[red]z [cyan](PSSH: [yellow]z...[cyan] KID: [red])z'[red]Error creating challenge for PSSH z...: z Content-Typezapplication/octet-streamz [red]License URL is None.)r)dataz[red]License request error: z[red]License error: z Response: z Url: z Headers: zapplication/jsonlicensez1[red]'license' field not found in JSON response: .z![red]Error parsing JSON license: zed]License data is empty.z[red]Error parsing license: CONTENTc3&K|] }|dk( yw)0N).0cs r z%_get_widevine_keys..s1181sz$[yellow]Warning: Extracted KID [red]z( [yellow]is not in the expected KID listrz[red]Error extracting keys: allr*z [cyan]| [red]z - [red]z[white]:[green]z[yellow]No keys extractedz-[red]Unexpected error during key extraction: r3)+setstrgetrlowerraddrrrloadr from_device Exceptionrr%r&ropen enumerateget_license_challengercopyr post status_codetextcontentrjsonbase64 b64decode parse_licenseget_keysr(r#hexr7rappendrrclose) r r r r rrdevicecdm expected_kidsitemr#e session_idall_content_keysextracted_kidsir'kid_info type_info challenge req_headersresponse license_bytes content_typer,key_obj formatted_keykkey_val masked_keylabel label_strs rrr6s F CEM#$((5"%&..sB7==?EEG 3%<   c "# " ./ [[1F//&)C  /0 m, 90;0C0C}- .(:0;0B0B}- I.YfJgIhij-n-C JUNy +a GAt,?$@M$,<<(// >&**3/-0: E)c..ASEW.W9~"s+;'<'ACa H !"23 Y1 wws| W$Sb\C/ 9E ((548=nUG42  C5 |I;WX  Y MM5 6#3=  IIj !]  MMA!E F   MM@D E &  GSb {RWXYWZ[\    rqs!$)* D )ndn#nPSnlpqtlunHLnZ]ntxn>c$t*c3cQTcfjknfocz~cVZcr!