ȏ4J d Z ddlZddlZddlZddlZddlmZmZ ddlm Z ej
e Z e fde
dedee
fd Zdd e
d e
defd
Zd e
de
de
fdZ G d d ZdS )u
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see
N)IteratorOptional )
MAX_FILE_SIZE
target_dirmax_sizereturnc # | K t j | D ]\ }}}|D ]}t j || } t j | |k r"t j || V i# t
$ r&}t d|| Y d }~d }~ww xY wd S )Nz+Error occurred while processing file %s: %s)oswalkpathjoingetsize Exceptionloggerwarning)r r rootdirsfilesfiler
es S/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/fs_scanner/utils.pydir_traverse_generatorr ! s WZ 0 0 dE D7<<d++D
7??4((H44 ',,tT22222
A4
s AB
B8 B33B8/abs_pathroot_dirc | r"t j | |pd nd}| t j t j fdt
t D S )Nr c ^ g | ])}t j |d *S N)r sepr ).0i
path_partss r
z/all_possible_relative_paths..6 s/ H H HABFKK
122 ' ' H H H )r r
relpathstripr! splitrangelen)r r rel_pathr$ s @r all_possible_relative_pathsr- 0 st >FMrwx S9992H ''--bf55J H H H Hs:1G1G H H HHr& r, c | | st d| d| d | d t | S )Nz
rel_path 'z ' is not a suffix of abs_path '')endswith
ValueErrorr+ )r r, s r get_base_dirr2 9 s] X & &
M M M( M M M
$s8}}n$ %%r& c e Zd Zej ej dZdZ ej d Z
ej d Z de fdZ
ededefd Z dd e d edee fd
ZdS )HashCalculator)md5sha256i s [ -
]+s [ -- -]+ algorithmc b || _ | j | | _ |dk r| j nd | _ d S )Nr5 c | S r )xs r z)HashCalculator.__init__..P s a r& )r7 HASHING_ALGORITHMS_hasher_factory_normalize_data_for_md5 _normalizer)selfr7 s r __init__zHashCalculator.__init__L s> " # 6y A ,5,>,>D ( (KK
r& datar c t |t st d | j d| }| j d| }| S )a
This method normalizes binary data by:
- Removing the special control characters: 0x00-0x08, 0x0E-0x1F, 0x7F-0xFF;
- 0x00-0x08, 0x0E-0x1F are ASCII control characters minus TAB, LF, VT, FF and CR;
- 0x7F-0xFF are unicode control characters (DEL from C0 and C1 set).
- Consolidating the clusters of whitespace characters (0x20, 0x09-0x0D) into single space character (0x20);
- Turns all uppercase ASCII characters to lowercase.
z*Normalization function expects bytes inputr& )
isinstancebytes TypeError_remove_control_characterssub_consolidate_whitespacelower)clsrC s r r? z&HashCalculator._normalize_data_for_md5S sd $&& J HII I-11#t<<*..tT::zz||r& Ffilepathapply_normalizationc t j | sd S t |d 5 }|r' | n| } fdt dt j D D ]}|
| d d d n # 1 swxY w Y | S )Nrbc 3 < K | ]}||j z V d S r ) BUFFER_SIZE)r" r# normalized_datarA s r z+HashCalculator.calc_hash..p sI A(8$8 89 r& r ) r r
isfileopenr@ readr> r* r+ rS update hexdigest)rA rN rO r hasherchunkrT s ` @r calc_hashzHashCalculator.calc_hashd sI w~~h'' F
(D
!
! %T '! ---YY[[
))++F q#o"6"68HII
%
%
e$$$$
% % % % % % % % % % % % % % % % !!!s BCC "C N)F)__name__
__module__ __qualname__hashlibr5 r6 r= rS recompilerK rI strrB classmethodrG r? boolr r] r: r& r r4 r4 C s {. K (bj)< = =!+,M!N!N
#
5 U [ " :?" ""26" # " " " " " "r& r4 )r )__doc__ra loggingr rb typingr r configr getLoggerr^ r rd intr listr- r2 r4 r: r& r rn s< * % % % % % % % % ! ! ! ! ! ! 8 $ $ &3 "
c] I I# I It I I I I&3 &# &# & & & &2" 2" 2" 2" 2" 2" 2" 2" 2" 2"r&