u%j)vddlZddlmZddlZddlmZddlmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZej,dZGd d eZGd d eZy) N)bytecode)unpack) ARSCHeaderARSCResTablePackageARSCResTypeSpec ARSCResTypeARSCResTableEntryARSCResTableConfig) StringBlock)complexToFloat)escapezpyaxmlparser.arscparserceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd#d Zd#dZd#dZdZd#dZd#dZd#dZd#dZd#dZd#dZGddeZd$dZdZd%dZd#dZdZdZ d$d Z!e"d!Z#d$d"Z$y)& ARSCParserz( Parser for resource.arsc files c d|_d|_tj||_t |j|_td|jjdd|_ i|_ i|_ tjtj|_tjd|_tjd|_d|_|jj%|j j&|j j(z|j j&|j j*z}|jj-|t j.z krt |j}|j&|j*z|kDry|j0t2j4k(r)|j"st7|j||_n|j0t2j8k(rt;|j|jksJdt=|j|}|j?}|j&|j*z}g|j|<|jj%|j j&|j@zt |j}|j0t2j4k(sJd|j0zt7|j|}|jj%|j j&|jBzt |j} | j0t2j4k(sJd| j0zt7|j| } |j|jE||j|jE||j|jE| tG||j"|| } |jj%|j&|j(z|j*z| j*z|jj-|t j.z kr4t |j} tHjKd jM| | j&| j*z|kDrn|j|jE| | j0t2jNk(r4|j|jEtQ|j| n | j0t2jRk(rtU|j| } |j|jE| |j|| jW| jXtHjKd jM| jXg}t[d| j\D]\}|j^d z|z|_/|jEtd |jjdd|j^f^|j|jE||D]\}}|jjarn|d k7s(tc|j|| }|j|jE||jesn|jj%|j&n4| j0t2jfk(rtHjidn |jj%| j&| j*z|jj-|t j.z kr4|jj%|j&|j*z|jj-|t j.z kryy)NFzz%ARSCParser.__init__..0s @W@WX[@\rcHtjtjSr)rrrrrrz%ARSCParser.__init__..2sK++K,C,CDrzGot more packages than expectedz#Expected String Pool header, got %xzFound a header: {}z Config: {}lz  *66{7N7NO + 7 78\ ](44 DF# $++++dkk.E.EEF;;$$t{{'7'77ii!X %??#DII.J*//1H<%"<"<">A\A\"\]!+DII!6%**e.H.HHP9N">A[A["[\ *499 5 $))U-G-GGO9Mr@get_language_and_region setdefaultr&rCr(get_type get_value get_indexr8get_resource_string is_complexget_resource_idget_resource_boolget_resource_integerget_resource_colorget_resource_dimen) rHrMnbr!rUlocalec_valuerVnb_irXrYrZs r_analysezARSCParser._analysese ==   MM; L(*DKK %Bs4==677|4R8fj1{{e&?&??%)]]<%@a%H !+!2!2!J!J!L"&++l";"F"FvPXZ\~"^"&-- "=b1f"E -4)*ME6${&*mmL&A"q&4-&PVY 4 4SZZ @ARAR Sknkuku 2 2< @ATATAV WX[XeXeXg h#&==?b#8$+H$5$<$<)3)<)<)> ),)5%6$.#6#6#8#GEGGJ,?,?,A$B#-#6#6#8H#D$+H$5$<$<(,(@(@(E%G&0%8%8%:d%B+.>>+;(/ (<(<,0,@,@,E)G&0%8%8%:f%D+.>>+;(/(>(>,0,B,B3,G)I&0%8%8%:i%G$+I$6$=$=(,(A(A#(F%H&0%8%8%:g%E$+G$4$;$;(,(?(?(D%F&0%8%8%:g%E$+G$4$;$;(,(?(?(D%F!% S)*Ta$hl*aos4==677 ; rcB|j|jgSr)rk get_key_datarHrZs rrmzARSCParser.get_resource_strings !1!1!344rc|jg}|jjdk(r|jd|S|jjdk(r|jd|S)Nrfalser^truerkkeyget_datar8rHrZxs rrozARSCParser.get_resource_ids` ]]_  77    " HHW WW   1 $ HHV rc|jg}|jjdk(r|jd|S|jjdk(r|jd|S)Nrr}rr~rrs rrpzARSCParser.get_resource_bools` ]]_  77    " HHW WW   2 % HHV rcV|j|jjgSr)rkrrr{s rrqzARSCParser.get_resource_integers  !1!1!344rc|jj}|jd|dz dz|dz dz|dz dz|dzfzgS)Nz#%02x%02x%02x%02x)rrrk)rHrZ entry_datas rrrzARSCParser.get_resource_colorsbWW%%' MMO "d*"d*/T)d" #% %  rc" |jt|jjtj |jjtj zgS#t$rtjdt|jjd|jjtj z|j|jjgcYSwxYw)Nz&Out of range dimension unit index for z: ) rkr rrr1DIMENSION_UNITSCOMPLEX_UNIT_MASK IndexErrorr:r;r{s rrszARSCParser.get_resource_dimens 9 "377#3#3#56))#''*:*:* ?MMOSWW%5%5%78 8  9sA0A33BD Dc ddgS)Nrr{s rget_resource_stylezARSCParser.get_resource_styles BxrcHt|jjS)zp Retrieve a list of all package names, which are available in the given resources.arsc. )listr$keysrHs rget_packages_nameszARSCParser.get_packages_namess DMM&&())rcn|jt|j|jS)z Retrieve a list of all available locales in a given packagename. :param package_name: the package name to get locales of rxrr%rrHrMs r get_localeszARSCParser.get_localess* DKK -22455rct|jt|j||jS)z Retrieve a list of all types which are available in the given package and locale. :param package_name: the package name to get types of :param locale: the locale to get types of (default: '') r)rHrMrus r get_typeszARSCParser.get_types&s/ DKK -f5::<==rc|jd}|dz } |j||dD]}|d|d|d|dfzz } |dz }|jd S#t$rY!wxYw) a# Get the XML (as string) of all resources of type 'public'. The public resources table contains the IDs for each item. :param package_name: the package name to get the resources for :param locale: the locale to get the resources for (default: '') ' r_z+ rr^r` utf-8rxr%KeyErrorencoderHrMrur rWs rget_public_resourceszARSCParser.get_public_resources1s 9  [[.v6x@ &FaD!A$!J&&& &   {{7##    s.A A*)A*c0|jd}|dz } |j||dD]A}tt|djdr d|dz}n|d}|d|dd |d z }C |d z }|j d S#t $rY!wxYw) a\ Get the XML (as string) of all resources of type 'string'. Read more about string resources: https://developer.android.com/guide/topics/resources/string-resource.html :param package_name: the package name to get the resources for :param locale: the locale to get the resources for (default: '') rrrar^z<&>z rr)rxr%anymap __contains__rr)rHrMrur rWvalues rget_string_resourceszARSCParser.get_string_resourcesKs 9  [[.v6x@ Js1Q4,,e45,qt3EaDEQqT5II  J   {{7##    sAB BBc |jd}|dz }|jD]|}|d|zz }|j|D]Y}|dt|zz }|dz } |j||dD]}|d|dd t |d d z } |d z }|d z }[|dz }~|dz }|jdS#t $rY4wxYw)z Get the XML (as string) of all resources of type 'string'. This is a combined variant, which has all locales and all package names stored. rz z z rrarrrr^rrz z z r)rxrrreprr%r rr)rHr rMrurWs rget_strings_resourcesz ARSCParser.get_strings_resourceshs 9  335 #L - < xHYQqT6RSTURV< XXY (( % & N "D! #$ {{7## s5B== C C c |jd}|dz } |j||dD]7}t|dk(r |d|dzz }|d|ddt|dd z }9 |d z }|j d S#t$rY!wxYw) aV Get the XML (as string) of all resources of type 'id'. Read more about ID resources: https://developer.android.com/guide/topics/resources/more-resources.html#Id :param package_name: the package name to get the resources for :param locale: the locale to get the resources for (default: '') rrrbr^z rz rrrrs rget_bool_resourceszARSCParser.get_bool_resourcess 9  [[.v6v> E1qtDD E   {{7##    ,A A('A(c|jd}|dz } |j||dD]}|d|dd|ddz } |d z }|jd S#t$rY!wxYw) ae Get the XML (as string) of all resources of type 'integer'. Read more about integer resources: https://developer.android.com/guide/topics/resources/more-resources.html#Integer :param package_name: the package name to get the resources for :param locale: the locale to get the resources for (default: '') rrrdz rrrrs rget_color_resourceszARSCParser.get_color_resources 9  [[.v6w? G1Q41FF G   {{7##    rc|jd}|dz } |j||dD]}|d|dd|ddz } |d z }|jd S#t$rY!wxYw) ag Get the XML (as string) of all resources of type 'dimen'. Read more about Dimension resources: https://developer.android.com/guide/topics/resources/more-resources.html#Dimension :param package_name: the package name to get the resources for :param locale: the locale to get the resources for (default: '') rrrfz "?@rr) __name__ __module__ __qualname____doc__r[rrrrrrrResourceResolverrs"  (   <  M ArrNcPtj||}|j|Sr)rrr)rHrr@resolvers rget_resolved_res_configsz#ARSCParser.get_resolved_res_configsHs%..tV<$$rc|j|jr |jSi}|jD]}i||<i}|j|D]}|}|dk(rd}i|||< |j||dD]$}|ddk(s d||||d<|d||d<& |j||dD]}|d|vs |d|||||d< ||_|S#t$rYOwxYw#t$rYwxYw)NDEFAULTr_rrar`r^)rxrrr%r)rHrrMkruv_localerWs rget_resolved_stringszARSCParser.get_resolved_stringsLsg   ! !)) )  335 L AlOA++l3 !z)(H,.,)![[6v>xH+Q48+>BAlOH5ad;&'dAadG+![[6v>xHFQ419AB1AlOH5a!g>F!   4"#  s0) C* C*'C9C9* C65C69 DDc|j|s tdt|ts td||jvrt j dgS|j|}t|dkDrh|rf||vr|||fgS|rV|tjk(r?t jdt|j|jdgSgSt|jS)a_ Return the resources found with the ID `rid` and select the right one based on the configuration, or return all if no configuration was set. But we try to be generous here and at least try to resolve something: This method uses a fallback to return at least one resource (the first one in the list) if more than one items are found and the default config is used and no default entry could be found. This is usually a bad sign (i.e. the developer did not follow the android documentation: https://developer.android.com/guide/topics/resources/localization.html#failing2) In practise an app might just be designed to run on a single locale and thus only has those locales set. You can disable this fallback behaviour, to just return exactly the given result. :param rid: resource id as int :param config: a config to resolve from, or None to get all results :param fallback: Enable the fallback for resolving default configuration (default: True) :return: a list of ARSCResTableConfig: ARSCResTableEntry z'rid' should be setz'rid' must be an intz6The requested rid could not be found in the resources.r^zLNo default resource config could be found for the given rid, using fallback!r) rx ValueErrorrgintr&r:infor4r default_configrGrr)rHrr@fallback res_optionss rrzARSCParser.get_res_configsos( 23 3#s#34 4 d** * HHM NI**3/ { a F$V!4566f(:(I(I(KK jkT11#6<<>?BCC  ))+, ,rc|j |j||dD]}|d|k(s |cSy#t$rYywxYw)Nrarr)rHrMnamerurWs r get_stringzARSCParser.get_stringsV  [[.v6x@ Q44<H   s 999 AAcL |j|||S#t$rYywxYwr)r(r)rHrM resource_typers rget_res_id_by_keyzARSCParser.get_res_id_by_keys4 %%l3MB3G G  s  ##c@|j|j|Sr)rxr$rs r get_itemszARSCParser.get_itemss }}\**rcV||jd}tjt}t|j|j D]Q\}}|j |k(s||j|k(s0||jj|S|S)Nr) rrrrr'rget_package_namerjextend)rHrM type_namerres_typers rget_type_configszARSCParser.get_type_configss  224Q7L((.!%d&;&;L&I&O&O&Q!R < Hg((*l:%-1B1B1D 1Qx((*+227; <  rcZ|jdstdj||dd}d}d|vr|jdd\}}n|}t |dk7rtdj| t |d|fS#t$rtd j|wxYw) a Resolves an id from a binary XML file in the form "@[package:]DEADBEEF" and returns a tuple of package name and resource id. If no package name was given, i.e. the ID has the form "@DEADBEEF", the package name is set to None. Raises a ValueError if the id is malformed. :param name: the string of the resource, as in the binary XML file :return: a tuple of (resource_id, package_name). @z0Not a valid resource ID, must start with @: '{}'r^N:rz+Numerical ID is not 8 characters long: '{}'rzID is not a hex ID: '{}') startswithrr<splitr4r)rpackagerYs rparse_idzARSCParser.parse_idss#OVVW[\] ]ABx $;"jja0OGVF v;! JQQRXYZ Z Hvr?G+ + H7>>vFG G Hs 8 B$B*c|r+|j||\}}}|sydj||S|jD]"}|j||\}}}|s|}|}|}n|sydj|S)a Returns the XML name for a resource, including the package name if package is None. A full name might look like `@com.example:string/foobar` Otherwise the name is only looked up in the specified package and is returned without the package name. The same example from about without the package name will read as `@string/foobar`. If the ID could not be found, `None` is returned. A description of the XML name can be found here: https://developer.android.com/guide/topics/resources/providing-resources#ResourcesFromXml :param r_id: numerical ID if the resource :param package: package name :return: XML name identifier Nz@{}/{}z @{}:{}/{})rr<r) rHr_idrresourceri_idprns rget_resource_xml_namez ARSCParser.get_resource_xml_names" #';;w#= HdD??8T2 2,,. ![[D1 1dG HD "))'8TBBr)rr)NT)%rrrrr[rxrmrorprqrrrsrrrrrrrrrrrrrobjectrrrrrrrr staticmethodrr rrrrr s{BzAF55   9*6 >$4$:$B$:$2$2$2$2 ((A6(AT%!F)-V + HHB"Crrc$eZdZdZdZdZdZy)r9c<||_||_||_||_yr)r)rPrRrL)rHrLr)rPrRs rr[zPackageContext.__init__s".*&.rc.|jjSrrLrCrs r get_mResIdzPackageContext.get_mResId s##***rc&||j_yrr)rHrCs r set_mResIdzPackageContext.set_mResId s&,#rc6|jjSr)rLr5rs rrzPackageContext.get_package_names##,,..rN)rrrr[rrrrrrr9r9s/+-/rr9)logging pyaxmlparserrrstructrpyaxmlparser.arscutilrrrrr r pyaxmlparser.stringblockr pyaxmlparser.constants constantsr1pyaxmlparser.utilsr xml.sax.saxutilsr getLoggerr:rrr9rrrr"s[$!HH0&-#g12^ C^ CB/V/r