a qщqe)>у@sтddlZddlZddlmZddlmZmZmZmZm Z m Z z ddlZddl mZmZmZWn"ey~eаeбаdбYn0ddlmZmZddlmZdd lmZmZmZmZd gZedZ Gdd Дd ГZ!Gd dДdГZ"dS)щN)┌suppress)┌cast┌Iterable┌List┌Mapping┌Optional┌Union)┌ NetworkXError┌NetworkXNoPath┌NodeNotFoundzNetworkX failed to import.щ)┌EdgeAttrIntMax┌EdgeAttrList)┌ PermissionMap)┌AVRule┌ SELinuxPolicy┌ TERuletype┌Type┌InfoFlowAnalysis┌InfoFlowStepc @s┤eZdZUdZeeed<eed<eed<d.e eee eeee fe ee efddЬdd ДZeed ЬddДГZejedd ЬddДГZeed ЬddДГZejeddЬddДГZeeed ЬddДГZeje eeee fddЬddДГZeeeedЬddДZd/eee feee feeedЬddДZeee feee feedЬddДZd0eee feed!d"Ьd#d$ДZe d Ьd%d&ДZeeed'Ьd(d)ДZdd Ьd*d+ДZdd Ьd,d-ДZdS)1rzInformation flow analysis.┌_exclude┌_min_weight┌ _perm_maprN)┌policy┌perm_map┌ min_weight┌exclude┌booleans┌returncCsВtаtб|_||_||_||_||_||_d|_ d|_ ztаб|_ |j аб|_Wn,ty||jаdб|jаdбВYn0dS)aи Parameters: policy The policy to analyze. perm_map The permission map or path to the permission map file. minweight The minimum permission weight to include in the analysis. (default is 1) exclude The types excluded from the information flow analysis. (default is none) booleans If None, all rules will be added to the analysis (default). otherwise it should be set to a dict with keys corresponding to boolean names and values of True/False. Any unspecified booleans will use the policy's default values. TzKNetworkX is not available. This is requried for Information Flow Analysis.z2This is typically in the python3-networkx package.N)┌logging┌ getLogger┌__name__┌logrrrrr┌rebuildgraph┌rebuildsubgraph┌nxZDiGraph┌G┌copy┌subG┌ NameErrorZcritical)┌selfrrrrrйr+·6/usr/lib64/python3.9/site-packages/setools/infoflow.py┌__init__!s zInfoFlowAnalysis.__init__)rcCs|jSйN)rйr*r+r+r,rEszInfoFlowAnalysis.min_weight)┌weightrcCs.d|krdksntdГВ||_d|_dS)Nrщ z4Min information flow weight must be an integer 1-10.T)┌ ValueErrorrr$)r*r0r+r+r,rIs cCs|jSr.)rr/r+r+r,rRszInfoFlowAnalysis.perm_map)rrcCs||_d|_d|_dS)NT)rr#r$)r*rr+r+r,rVscCs|jSr.)rr/r+r+r,r\szInfoFlowAnalysis.exclude)┌typesrcs*|rЗfddД|DГИ_ngИ_dИ_dS)Ncsg|]}Иjа|бСqSr+)r┌lookup_type)┌.0┌tr/r+r,┌ cєz,InfoFlowAnalysis.exclude..T)rr$)r*r3r+r/r,r`s)┌source┌targetrccsВ|jа|б}|jа|б}|jr&|аб|jаdа||ббttt ГП*|а tj|j ||dНбVWdГn1st0YdS)a Generator which yields one shortest path between the source and target types (there may be more). Parameters: source The source type. target The target type. Yield: generator(steps) steps Yield: tuple(source, target, rules) source The source type for this step of the information flow. target The target type for this step of the information flow. rules The list of rules creating this information flow step. z@Generating one shortest information flow path from {0} to {1}...)r9r:N)rr4r$┌_build_subgraphr"┌info┌formatrr r┌!_InfoFlowAnalysis__generate_stepsr%┌ shortest_pathr()r*r9r:┌sr6r+r+r,r?is zInfoFlowAnalysis.shortest_pathщ)r9r:┌maxlenrccsЮ|dkrtdГВ|jа|б}|jа|б}|jr6|аб|jаdа|||ббtt t ГП4tа|j |||бD]}|а|бVqjWdГn1sР0YdS)aи Generator which yields all paths between the source and target up to the specified maximum path length. This algorithm tends to get very expensive above 3-5 steps, depending on the policy complexity. Parameters: source The source type. target The target type. maxlen Maximum length of paths. Yield: generator(steps) steps Yield: tuple(source, target, rules) source The source type for this step of the information flow. target The target type for this step of the information flow. rules The list of rules creating this information flow step. rz%Maximum path length must be positive.zHGenerating all information flow paths from {0} to {1}, max length {2}...N)r2rr4r$r;r"r<r=rr rr%Zall_simple_pathsr(r>)r*r9r:rBr@r6┌pathr+r+r,┌ all_pathsЛs zInfoFlowAnalysis.all_pathsccsК|jа|б}|jа|б}|jr&|аб|jаdа||ббttt ГП2t а|j||бD]}|а |бVqVWdГn1s|0YdS)aю Generator which yields all shortest paths between the source and target types. Parameters: source The source type. target The target type. Yield: generator(steps) steps Yield: tuple(source, target, rules) source The source type for this step of the information flow. target The target type for this step of the information flow. rules The list of rules creating this information flow step. zAGenerating all shortest information flow paths from {0} to {1}...N)rr4r$r;r"r<r=rr rr%┌all_shortest_pathsr(r>)r*r9r:r@r6rCr+r+r,rE┤s z#InfoFlowAnalysis.all_shortest_pathsTr)┌type_┌outrccsЮ|jа|б}|jr|аб|jаdа|r,dnd|ббttГПL|rR|j а |б}n|j а|б}|D]\}}t|j ||ГVqbWdГn1sР0YdS)a( Generator which yields all information flows in/out of a specified source type. Parameters: source The starting type. Keyword Parameters: out If true, information flows out of the type will be returned. If false, information flows in to the type will be returned. Default is true. Yield: generator(steps) steps A generator that returns the tuple of source, target, and rules for each information flow. z(Generating all information flows {0} {1}zout ofZintoN) rr4r$r;r"r<r=rr r(Z out_edgesZin_edgesr)r*rFrGr@Zflowsr9r:r+r+r,┌ infoflows╫s zInfoFlowAnalysis.infoflowscCs.|jr|абdtа|jбЫdtа|jбЫЭS)zQ Get the information flow graph statistics. Return: str z Graph nodes: z Graph edges: )r#┌_build_graphr%┌number_of_nodesr&┌number_of_edgesr/r+r+r,┌ get_stats■s zInfoFlowAnalysis.get_stats)rCrccs4tdt|ГГD] }t|j||d||ГVqdS)aт Generator which returns the source, target, and associated rules for each information flow step. Parameter: path A list of graph node names representing an information flow path. Yield: tuple(source, target, rules) source The source type for this step of the information flow. target The target type for this step of the information flow. rules The list of rules creating this information flow step. rN)┌range┌lenrr()r*rCr@r+r+r,Z__generate_stepssz!InfoFlowAnalysis.__generate_stepscCs*|jабdа|jб|j_|jа|jб|jаdа|jбб|jа бD]в}|j tjkrXqF|jа tt|Гб\}}tа|jаб|jаббD]`\}}||krЖ|r╛t|j||ddН}|jа|б||_|rЖt|j||ddН}|jа|б||_qЖqFd|_d|_|jаdб|jаdаtа|jбtа|jбббdS)NzInformation flow graph for {0}.z+Building information flow graph from {0}...T)┌createFz*Completed building information flow graph.z$Graph stats: nodes: {0}, edges: {1}.)r&┌clearr=r┌namerZ map_policyr"r<ZterulesZruletyperZallowZrule_weightrr┌ itertools┌productr9┌expandr:r┌rules┌appendr0r#r$┌debugr%rJrK)r*┌ruleZrweightZwweightr@r6┌edger+r+r,rI.s2 ■zInfoFlowAnalysis._build_graphc s╛ИjrИабИjаdбИjаdаИjббИjаdаИjббИjаdаИjduббЗfddДИj а бDГ}Иj а|бабИ_ Иjdkr╓g}Иj абD],\}}tИj ||Г}|jИjkrЬ|а|бqЬИj а|бИjduРrДg}Иj абD]Ж\}}tИj ||Г}g}|jD]$}|jfiИjдОРs|а|бРqg}|D]&}||vРr>|jа|б|а|бРq>|jsЁ|а|бqЁИj а|бdИ_Иjаd бИjаd аtаИj бtаИj бббdS)Nz%Building information flow subgraph...zExcluding {0!r}zMin weight {0}z(Exclude disabled conditional policy: {0}csg|]}|Иjvr|СqSr+)r)r5┌nr/r+r,r7\r8z4InfoFlowAnalysis._build_subgraph..rFz-Completed building information flow subgraph.z'Subgraph stats: nodes: {0}, edges: {1}.)r#rIr"r<rWr=rrrr&┌nodesZsubgraphr'r(Zedgesrr0rVZremove_edges_fromrUZenabled┌remover$r%rJrK) r*r[Zdelete_listr@r6rYZ rule_listrXZ deleted_rulesr+r/r,r;QsN ■z InfoFlowAnalysis._build_subgraph)rNN)rA)T)r!┌ __module__┌__qualname__┌__doc__rr┌__annotations__┌intrrrrr┌strr┌boolr-┌propertyr┌setterrr┌InfoFlowPathr?rDrErHrLr>rIr;r+r+r+r,rsF ■■$"" ) # ' #c@sFeZdZdZedГZedГZd eee ddЬddДZ d d ДZddДZdS)raR A graph edge. Also used for returning information flow steps. Parameters: graph The NetworkX graph. source The source type of the edge. target The target type of the edge. Keyword Parameters: create (T/F) create the edge if it does not exist. The default is False. rUZcapacityFN)r9r:rOrcCsP||_||_||_|jа||бsL|rD|jj||ddНd|_d|_ntdГВdS)Nr)r0z$InfoFlowStep does not exist in graph)r&r9r:Zhas_edgeZadd_edgerUr0r2)r*Zgraphr9r:rOr+r+r,r-бszInfoFlowStep.__init__cs4t|tГr&ЗfddДt|аdбОDГSИа|бSdS)Ncsg|]}Иа|бСqSr+)┌_index_to_item)r5┌ir/r+r,r7▓r8z,InfoFlowStep.__getitem__..rA)┌ isinstance┌slicerM┌indicesrg)r*┌keyr+r/r,┌__getitem__оs zInfoFlowStep.__getitem__cCs.|dkr|jS|dkr|jStdа|бГВdS)z'Return source or target based on index.rrz,Invalid index (edges only have 2 items): {0}N)r9r:┌ IndexErrorr=)r*┌indexr+r+r,rg╢s zInfoFlowStep._index_to_item)F) r!r]r^r_rrUr r0rrcr-rmrgr+r+r+r,rЙs )#rRr┌ contextlibr┌typingrrrrrrZnetworkxr%Znetworkx.exceptionr r r┌ImportErrorr r!rWZdescriptorsr rZpermmaprZ policyreprrrr┌__all__rfrrr+r+r+r,┌s" r