
    #fdk                         S r SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJ	r	J
r
JrJr  SSKJrJr  SSKr " S S5      rg)z&backend.py: interface for ufw backends    N)errorwarndebug	_findpath)UFWErrorUFWRulec                       \ rS rSrSr  S#S jrS rS rS rS$S jr	S	 r
S
 rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS r S  r!S! r"S"r#g)%
UFWBackend   zInterface for backendsNc                 J   S U l         Xl        X l        / U l        / U l        [        [        R                  R                  U5      n[        R                  R                  US5      [        R                  R                  US5      [        R                  R                  US5      S.U l        Ub  U R                  R                  U5        SSSSS	S
.U l        [        R                  R                  U l        U R!                  5         U R#                  5         U R%                  5         [        R&                  R)                  U R                  S   5      U l        [        R                  R                  [        R                  R,                  S5      U l        [        R                  R                  [        R                  R,                  S5      U l        [        R                  R                  [        R                  R,                  S5      U l        [        R                  R                  [        R                  R,                  S5      U l         [        R6                  R9                  U R.                  5      U l        S U l!        g ! [<         a    [?        S5      n[A        U5      ef = f)Nzdefault/ufwzufw/ufw.confzufw/applications.d)defaultsconfappsr   d      i,  i  )offlowmediumhighfullr   iptablesziptables-restore	ip6tableszip6tables-restorez#Couldn't determine iptables version)"r   namedryrunrulesrules6r   ufwcommon
config_dirospathjoinfilesupdate	loglevels	do_checks
_do_checks_get_defaults_read_rulesapplicationsget_profilesprofilesiptables_dirr   iptables_restorer   ip6tables_restoreutilget_iptables_versioniptables_versionOSError_r   caps)selfr   r   extra_filesrootdirdatadirperr_msgs           -/usr/lib/python3/dist-packages/ufw/backend.py__init__UFWBackend.__init__    s   	
cjj++W5"$'',,q-"@ ggll1n= ggll1.BCF
 "JJk*!"!$$'"%"%	( --((55djj6HISZZ%<%<jI "SZZ-D-D-?!Acjj&=&={K!#cjj.E.E.A"C	$$'HH$A$A$--$PD! 	  	$=>G7##	$s   .J !J"c                    U R                   b  g0 U l         0 U R                   S'   SU R                   S   S'   SU R                   S   S'   U R                  (       a  [        R                  " 5       S:X  a  U R                  (       d   [
        R                  R                  U R                  5      nS	U;   a  S
U;   a  SU R                   S   S'   OSU R                   S   S'   U R                  5       (       a\   [
        R                  R                  U R                  5      nS	U;   a  S
U;   a  SU R                   S   S'   gSU R                   S   S'   ggggg! [         a:  nSU-  nU R                  5       (       a  [        U5        [        U5         SnAgSnAff = f! [         a  n[        SU-  5         SnANSnAff = f)z\Initialize the capabilities database. This needs to be called
before accessing the database.NlimitT4F6r   zinitcaps
%sz
recent-setzrecent-update)r5   r&   r    getuidr   r   r0   get_netfilter_capabilitiesr   r3   
is_enabledr   r   use_ipv6r   )r6   nf_capsemsgs       r<   initcapsUFWBackend.initcapsL   s   
 99 			' #'		'3"'		'3 >>biikQ.t{{((==dmmL w&?g+E*.		'"3'*/		'"3' }}.!hhAA$..QG  7*'/I.2DIIg&s+.3DIIg&s+ % 8C.>  $q(??$$#JS	"  ..1,--.s0   )E 5)F 
F0FF
F=%F88F=c                 L    SU R                   ;   a  U R                   S   S:X  a  gg)z!Is firewall configured as enabledenabledyesTF)r   r6   s    r<   rE   UFWBackend.is_enabledx   s%    %==#u,    c                     SU R                   ;   a8  U R                   S   S:X  a%  [        R                  R                  S5      (       a  gg)z"Is firewall configured to use IPv6ipv6rN   z/proc/sys/net/ipv6TF)r   r    r!   existsrO   s    r<   rF   UFWBackend.use_ipv6   s9    T]]"== E)77>>.//rQ   c                 "   SU-   S-   nSnU R                   U   S:X  a  SnOU R                   U   S:X  a  SnOSnU(       a  US:X  a  S	n[        S
5      n[        R                  R	                  SS/5      u  pxUS:w  a  [        U5      eSU;   a  SnU R                  5       (       ab  [        R                  R	                  SS/5      u  pxUS:X  a  SU;   a  Sn[        R                  R	                  SS/5      u  pxUS:X  a  SU;   a  SnU(       d  SnU$ )z.Get default policy for specified primary chaindefault__policy acceptallowrejectdenyforwardFzproblem running sysctlsysctlznet.ipv4.ip_forwardr   1Tz net.ipv6.conf.default.forwardingznet.ipv6.conf.all.forwardingdisabled)r   r4   r   r0   cmdr   rF   )	r6   primarycheck_forwardpolicyrstrrM   r;   rcouts	            r<   _get_default_policyUFWBackend._get_default_policy   s   g%	1== H,D]]6"h.DDW	1G01Gh0E%FGIRQww''cz }}HHLL(4V)WX	7scz"GHHLL(4R)ST	7scz"G!rQ   c                 :
   U R                   (       d  [        S5      n[        U5        g[        R                  " 5       [        R
                  " 5       :w  a  [        S5      n[        U5      e[        R                  " 5       [        R                  " 5       :w  a  [        S5      n[        U5      e[        R                  " 5       nUS:w  a  [        S5      n[        U5      e0 n0 n0 n/ n[        R                  R                  U R                  S   5      (       d'  [        S5      U R                  S   -  n[        U5        O[        R                  " S	5      n[        R                  " U R                  S   5       HW  n	UR                  U	5      (       a  M  UR!                  [        R                  R#                  U R                  S   U	5      5        MY     [%        U R                  R'                  5       5      [        R                  R)                  [*        R,                  S   5      /-   U-    GH6  n
U
R/                  S
5      (       d  [        R0                  " 5       < S
U
< 3n
 [3        SU
-   5        XR                  S   :X  a3  [        R                  R                  U R                  S   5      (       d  M   [        R4                  " U
5      nU[4        R6                     nSnSnSn [<        R>                  " UR@                  5      S   U:X  d'  [<        R>                  " UR@                  5      S   U:X  a  SnUR@                  S:w  aI  U(       dB  X;  a=  [        S5      [E        U5      U
[E        UR@                  5      S.-  n[        U5        SXZ'   U[4        RF                  -  (       a"  X;  a  [        S5      U
-  n[        U5        SX:'   U[4        RH                  -  (       a2  X;  a-  URJ                  S:w  a  [        S5      U
-  n[        U5        SXJ'   U
S
:X  a  GM  U
n[        R                  RM                  U
5      n
U
(       d  [9        [N        RP                  SU-  5      eGM     U R                   He  nUS:w  d  M  [        R                  RS                  U R                  U   5      (       a  M>  [        S5      UU R                  U   S.-  n[        U5      e   g! [8         a    [        S5      U
-  n[        U5      e[:         a    e f = f! [B         a     GNf = f)a  Perform basic security checks:
is setuid or setgid (for non-Linux systems)
checks that script is owned by root
checks that every component in absolute path are owned by root
warn if script is group writable
warn if part of script path is group writable

Doing this at the beginning causes a race condition with later
operations that don't do these checks.  However, if the user running
this script is root, then need to be root to exploit the race
condition (and you are hosed anyway...)
zChecks disabledTz%ERROR: this script should not be SUIDz%ERROR: this script should not be SGIDr   z&You need to be root to run this scriptr   z'%s' does not existz^\./z	Checking zCouldn't stat '%s'clickpkg	snappypkgFz4uid is %(uid)s but '%(path)s' is owned by %(st_uid)s)uidr!   st_uidz%s is world writable!z%s is group writable!zCould not find parent for '%s'z&'%(f)s' file '%(name)s' does not exist)fr   N)*r&   r4   r   r    rC   geteuidr   getgidgetegidr!   isdirr#   recompilelistdirsearchappendr"   listvaluesabspathsysargv
startswithgetcwdr   statST_MODEr3   	Exceptionpwdgetpwuidrp   KeyErrorstrS_IWOTHS_IWGRPst_giddirnameerrnoENOENTisfile)r6   r;   ro   warned_world_writewarned_group_writewarned_ownerr,   warn_msgpatprofiler!   statinfomode
click_user	snap_useris_unpack_user	last_pathrq   s                     r<   r'   UFWBackend._do_checks   s    ~~)*GM 99;"**,&?@G7##99;"**,&?@G7##iik!8@AG7##  ww}}TZZ/00./4::f3EFHN**V$C::djj&89zz'**OOBGGLLF1CW$MN : **,-''//#((1+.01D ??3''"$))+t4kD()::f--!ww}}TZZ-?@@!wwt}H#DLL1D (
'	!&||HOO4Q7:ELL9!<	I)- ??a'0  ".  /:=c(7;9<X__9M3O PH N)-L&$,,&4+I !89TBHN/3&,$,,&4+I 1, !89TBHN/3&,3; 	wwt,!%,,"B#,#./ /e v AF{277>>$**Q-#@#@DE"#TZZ];=w''	 [  , 45>G"7++     s   5)S%AT.T	
TTc                    0 U l         U R                  S   U R                  S   4 H  n [        R                  R	                  U5      n[        R                  " S5      nU H  nUR                  U5      (       d  M  [        R                  " SUR                  5       5      nUS   R                  5       R                  S5      U R                   US   R                  5       '   M     UR                  5         M     / S	QnS
 Ha  nSU-  U R                   ;  a  [        SU-  5      n[        U5      eU R                   SU-     n	X;  d  MH  [        SXS.-  5      n[        U5      e   g! [
         a    [        S5      U-  n[        U5      ef = f)z#Get all settings from defaults filer   r   zCouldn't open '%s' for readingz^\w+="?\w+"?=   "'r   )rZ   dropr\   )inputoutputr^   zdefault_%s_policyzMissing policy for '%s'z+Invalid policy '%(policy)s' for '%(chain)s')re   chainN)r   r#   r   r0   open_file_readr   r4   r   rv   rw   ry   splitstriplowerclose)
r6   rq   origr;   r   linetmppoliciescr:   s
             r<   r(   UFWBackend._get_defaults  s`   **Z($**V*<=A(xx..q1 **_-C::d##((46C47FLLN4H4H4ODMM#a&,,.1 
 JJL > 01A"a(=5;<w''1Q78A I()68 9w'' 2  (<=Cw''(s   E%%$F	c                 |   [         R                  " SU5      (       d  [        S5      n[        U5      e[        R
                  " U[        R                  5      (       d  [        SU-  5      n[        U5      e[        R                  R                  U5      nUS   nSn[         R                  " SU-   S-   5      nUS    Hd  n	UR                  U	5      (       a,  [        R                  R                  XbS-   U-   S	-   5        S
nME  [        R                  R                  Xi5        Mf     U(       d(  [        R                  R                  XbS-   U-   S	-   5         [        R                  R                  U5        UR                  5       R!                  S5      U R"                  UR                  5       '   g! [         a    e f = f)zSets option in defaults filez^[\w_]+$zInvalid optionz'%s' is not writabler   F^r   r   
Tr   N)rv   matchr4   r   r    accessW_OKr   r0   
open_filesrw   ry   write_to_fileclose_filesr   r   r   r   )
r6   fnoptvaluer;   fnsfdfoundr   r   s
             r<   set_defaultUFWBackend.set_default:  s_   xxS))()G7## yyRWW%%."56G7##hh!!"%Zjjc)*KDzz$&&r9u+<t+CD&&r0   HH""2Sy5'84'?@	HH  %
 &+[[]%8%8%?ciik"	  		s   F/ /F;c                    U R                   (       d  US:X  a!  U R                  U R                  S   SS5        OUS:X  a!  U R                  U R                  S   SS5        OgUS:X  a!  U R                  U R                  S   SS5        O@US	:X  a!  U R                  U R                  S   SS
5        O[        S5      U-  n[	        U5      e[        S5      U-  nU$ )z+Sets default application policy of firewallr[   r   DEFAULT_APPLICATION_POLICYz"ACCEPT"r]   z"DROP"r\   z"REJECT"skipz"SKIP"zUnsupported policy '%s'z*Default application policy changed to '%s')r   r   r#   r4   r   )r6   re   r;   rf   s       r<   set_default_application_policy)UFWBackend.set_default_application_policy^  s    {{   J!7(D(46 6!  J!7(D(24 8#  J!7(D(46 6!  J!7(D(24 56&Aw''=>&IrQ   c                 l   / n[        U R                  R                  5       5      nUR                  U;   GaC  UR                  U;   Ga2  [
        R                  R                  U R                  UR                     5      n[
        R                  R                  U R                  UR                     5      nU GH  nUR                  5       nSUl	        UR                  SS5        [
        R                  R                  U5      u  pUR                  U	5        UR                  US5        UR                  Ul	        UR                  UR                  :X  ap  SUl        [
        R                  R                  U5      u  pUR                  U	5        UR                  US5        UR                  Ul        UR                  U5        GM  U H  n
UR                  5       nSUl        [
        R                  R                  U
5      u  pUR                  U	5        UR                  US5        UR                   S:X  a  UR                  UR                   5        UR                  Ul        UR                  U5        M     GM     GOUR                  U;   a  [
        R                  R                  U R                  UR                     5       H  nUR                  5       nSUl        [
        R                  R                  U5      u  pUR                  U	5        UR                  US5        UR                  Ul        UR                  U5        M     OUR                  U;   a  [
        R                  R                  U R                  UR                     5       H  nUR                  5       nSUl	        [
        R                  R                  U5      u  pUR                  U	5        UR                  US5        UR                  Ul	        UR                  U5        M     [#        U5      S:  a  [%        S5      n['        U5      eU$ )z4Return a list of UFWRules based on the template rulerY   anysrcdstr   z&No rules found for application profile)r{   r,   keysdportsportr   r*   	get_portsdup_ruledappset_portr0   parse_port_protoset_protocolsapprz   protocollenr4   r   )r6   templater   profile_namesdportssportsir   portprotojruler:   r;   s                 r<   get_app_rules_from_template&UFWBackend.get_app_rules_from_templatey  s   T]]//12>>]*x~~/N%%//hnn0MNF%%//hnn0MNF'')UE* # 9 9! <  'T5)#==>>X^^3  "CH$'HH$=$=a$@MT$$U+LLu-'}}CHLL%#"||~$&	(+(A(A!(D))%0dE2==E1 --cll;$,MM	T* $+ B ^^},%%//hnn0MN((*	 # 9 9! <!!%(dE*$MM	T" O ^^},%%//hnn0MN((*	 # 9 9! <!!%(dE*$MM	T" O u:>@AG7##rQ   c                    / n/ nSnSnSnU R                   U R                  -    GHP  nUR                  U:X  d  UR                  U:X  a  UR	                  5       nX:X  a  M;  UR                  5       n	U	R                  S5        U	R                  S:w  a  U	R                  U	R                  S5        U	R                  S:w  a  U	R                  U	R                  S5        U R                  U	5      n
U
 HH  nUR                  5         UR                  (       a  UR                  U5        M7  UR                  U5        MJ     UnSnGM  UR                  (       a  UR                  U5        GM?  UR                  U5        GMS     U(       aC  X l         X0l        U[        S5      U-  -  n U R                  S5        U R                  S5        XV4$ XV4$ ! [         a    [        S5      n[        U5      ef = f)	zUpdate rule for profile in place. Returns result string and bool
on whether or not the profile is used in the current ruleset.
rY   Fr   r   r   TzRules updated for profile '%s'z!Couldn't update application rules)r   r   r   r   get_app_tupler   r   r   r   	normalizev6rz   r4   _write_rulesr   r   )r6   r   updated_rulesupdated_rules6
last_tuplerf   updated_profilertuplr   new_app_rulesnew_rr;   s                r<   update_app_ruleUFWBackend.update_app_rule  s    
 dkk)Avv AFFg$5 (%   !zz|H))%0}}* ))(--?}}* ))(--?$($D$D*2%4M "/) 88*11%8)007 "/ "&J&*O44"))!,!((+E *H &J(KA677CCD(!!%(!!$'
 &&&&	  (?@w''(s   "G !G'c                 d   XR                   ;   a  U$ SnSn[        U R                   R                  5       5       H.  nUR                  5       UR                  5       :X  d  M'  UnUS-  nM0     SX14-  n[	        U5        US:X  a  U$ US:  a  [        S5      U-  nO[        S5      U-  n[        U5      e)z2Find the application profile name for profile_namerY   r   r   z'%d' matches for '%s'z>Found multiple matches for '%s'. Please use exact profile namez&Could not find a profile matching '%s')r,   r{   r   r   r   r4   r   )r6   profile_namer   matchesn	debug_msgr;   s          r<   find_application_name UFWBackend.find_application_name  s    ==(dmm((*+AwwyL..001 ,
 ,w.EE	ia<Lq[XY%'G @A%'GwrQ   c                 F   U(       a#  U[        U R                  5      :  a
  [        5       eU(       d#  U[        U R                  5      :  a
  [        5       eUS:  a
  [        5       e/ nU(       a  U R                  nOU R                  n0 nSn[	        U5       HP  u  pgXa:  a    OHSnUR
                  S:w  d  UR                  S:w  d  M0  UR                  5       nX;   a  US-  nML  SXH'   MR     / nU(       aA  U R                  nU R                  US-
  U-      R                  5       n	U	R                  S5        O@U R                  nU R                  US-
  U-      R                  5       n	U	R                  S5        Sn
U H&  n[        R                  " Xy5      S:X  a  U
s  $ U
S-  n
M(     g)a  Return the absolute position in the other list of the rule with the
user position of the given list. For example, find_other_position(4,
True) will return the absolute position of the rule in the ipv4 list
matching the user specified '4' rule in the ipv6 list.
r   r   rY   TF)r   r   
ValueErrorr   	enumerater   r   r   r   set_v6r   r   )r6   positionr   r   	app_rulestuple_offsetr   r   r   
match_rulecounts              r<   find_other_positionUFWBackend.find_other_position  sq    (S--, hTZZ0,a<,KKEJJE 	e$DA}Dvv|qvv|($ A%L&*IO % JJEX\L%@AJJLJe$KKEHqL<$?@IIKJd#A}}Q+q0QJE 
 rQ   c                 N   Sn[        S5      nSU R                  ;  d4  U R                  S   [        U R                  R	                  5       5      ;  a  SnU[        S5      -  nX4$ U R                  U R                  S      nUS:X  a  US-  nX4$ USU R                  S   -  -  nX4$ )z"Gets current log level of firewallr   z	Logging: loglevelunknownr   zon (%s))r4   r   r{   r%   r   )r6   levelrf   s      r<   get_loglevelUFWBackend.get_loglevelO  s    ~T]]*==$D1D1D1F,GGEAiL D } NN4==#<=Ez } 	T]]:%>??}rQ   c                    U[        U R                  R                  5       5      S/-   ;  a  [        S5      U-  n[	        U5      eUnUS:X  a5  SU R
                  ;  d  U R
                  S   S:X  a  SnOU R
                  S   nU R                  U R                  S   SU5        U R                  U5        US:X  a  [        S5      $ [        S	5      $ )
zSets log level of firewallonzInvalid log level '%s'r  r   r   r   LOGLEVELzLogging disabledzLogging enabled)	r{   r%   r   r4   r   r   r   r#   update_logging)r6   r  r;   	new_levels       r<   set_loglevelUFWBackend.set_loglevel_  s    T^^0023tf<<01U;G7##	D=.}}Z(E1!	 MM*5	F+ZCI&'((&''rQ   c                 D    U R                  5       U R                  5       -   $ )zReturn list of all rules)get_rules_ipv4get_rules_ipv6rO   s    r<   	get_rulesUFWBackend.get_rulesu  s    ""$t':':'<<<rQ   c                     U R                   $ )zReturn list of IPv4 rules)r   rO   s    r<   r  UFWBackend.get_rules_ipv4y  s    zzrQ   c                     U R                   $ )zReturn list of IPv6 rules)r   rO   s    r<   r  UFWBackend.get_rules_ipv6}  s    {{rQ   c                    / nU(       a  U R                   nOU R                  nSn0 nU HS  nSnUR                  S:w  d  UR                  S:w  a)  UR	                  5       nXd;   a  [        SU-  5        MJ  SXF'   US-  nMU     U$ )z/Return number of ufw rules (not iptables rules)r   rY   Skipping found tuple '%s'Tr   )r   r   r   r   r   r   )r6   r   r   r   r   r   r   s          r<   get_rules_countUFWBackend.get_rules_count  s    KKEJJE	ADvv|qvv|($5>?&*IOQJE  rQ   c                    U R                  5       nSn0 nU Hf  nSnUR                  S:w  d  UR                  S:w  a)  UR                  5       nXd;   a  [	        SU-  5        MJ  SXF'   U[        U5      :X  a  Us  $ US-  nMh     g)z:Return rule specified by number seen via "status numbered"r   rY   r  TN)r  r   r   r   r   int)r6   numr   r   r   r   r   s          r<   get_rule_by_numberUFWBackend.get_rule_by_number  s     	ADvv|qvv|($5>?&*IOC QJE  rQ   c                     / nSnU R                  5        H2  nUS-  nUR                  U5      nUS:  d  M!  UR                  U5        M4     U$ )zbSee if there is a matching rule in the existing ruleset. Note this
does not group rules by tuples.r   r   )r  fuzzy_dst_matchrz   )r6   r   matchedr   r   rets         r<   get_matchingUFWBackend.get_matching  sR     !AQJE&&q)CQwu%	 " rQ   c                     [        S5      e)z*Set default policy for specified directionz/UFWBackend.set_default_policy: need to overrider   )r6   re   	directions      r<   set_default_policyUFWBackend.set_default_policy  s    HIIrQ   c                     [        S5      e)zGet status of running firewallz,UFWBackend.get_running_raw: need to overrider)  )r6   
rules_types     r<   get_running_rawUFWBackend.get_running_raw  s    EFFrQ   c                     [        S5      e)zGet managed rulesz'UFWBackend.get_status: need to overrider)  )r6   verbose
show_counts      r<   
get_statusUFWBackend.get_status  s    @AArQ   c                     [        S5      e)zUpdate firewall with rulez%UFWBackend.set_rule: need to overrider)  )r6   r   allow_reloads      r<   set_ruleUFWBackend.set_rule  s    >??rQ   c                     [        S5      e)zStart the firewallz+UFWBackend.start_firewall: need to overrider)  rO   s    r<   start_firewallUFWBackend.start_firewall      DEErQ   c                     [        S5      e)zStop the firewallz*UFWBackend.stop_firewall: need to overrider)  rO   s    r<   stop_firewallUFWBackend.stop_firewall  s    CDDrQ   c                     [        S5      e)z%Get a list if rules based on templatez6UFWBackend.get_app_rules_from_system: need to overrider)  )r6   r   r   s      r<   get_app_rules_from_system$UFWBackend.get_app_rules_from_system  s     " # 	#rQ   c                     [        S5      e)z#Update loglevel of running firewallz+UFWBackend.update_logging: need to overrider)  )r6   r  s     r<   r  UFWBackend.update_logging  r=  rQ   c                     [        S5      e)zReset the firewallz"UFWBackend.reset: need to overrider)  rO   s    r<   resetUFWBackend.reset  s    ;<<rQ   )r5   r   r&   r   r#   r   r/   r   r.   r2   r%   r   r,   r   r   )NNN)r   F)$__name__
__module____qualname____firstlineno____doc__r=   rJ   rE   rF   ri   r'   r(   r   r   r   r   r   r   r  r  r  r  r  r  r   r&  r+  r/  r4  r8  r;  r?  rB  r  rG  __static_attributes__ rQ   r<   r
   r
      s     15'+*X*4X#Ln(`(:"@H6BH<'| 0:x (,=0,JGB@FE#
F=rQ   r
   )rM  r   r    r   rv   r   r~   ufw.utilr   r   r   r   r   
ufw.commonr   r   ufw.applicationsr
   rO  rQ   r<   <module>rS     s7    ,"  	 
 	  
  2 2 ( B= B=rQ   