118 lines
11 KiB
Plaintext
118 lines
11 KiB
Plaintext
|
Ë
|
|||
|
|
9ˆØij' ã
|
|||
|
|
ó. — d Z ddlZddlZddlmZmZmZ ej e« Z g d¢Z
|
|||
|
|
g d¢Z de dee
|
|||
|
|
ee f fd„Zde dee
|
|||
|
|
e f fd „Zde dee fd
|
|||
|
|
„Zdde d e dee
|
|||
|
|
ee f fd „Zdde d e dee fd
|
|||
|
|
„Z dde d e dee fd„Z dde d e de
|
|||
|
|
defd„Zy)u
|
|||
|
|
SQL 验�工具集
|
|||
|
|
é N)ÚTupleÚListÚDict)ÚDROPÚDELETEÚUPDATEÚINSERTÚALTERÚTRUNCATEÚCREATEz
|
|||
|
|
DROP DATABASEz
|
|||
|
|
DROP TABLEz
|
|||
|
|
DROP INDEXÚGRANTÚREVOKEÚPURGEÚFLUSHÚKILL)5ÚSELECTÚWITHÚFROMÚWHEREÚJOINz LEFT JOINz
|
|||
|
|
RIGHT JOINz
|
|||
|
|
INNER JOINz
|
|||
|
|
OUTER JOINÚONÚUSINGzGROUP BYÚHAVINGzORDER BYÚLIMITÚOFFSETÚUNIONz UNION ALLÚEXCEPTÚ INTERSECTÚASÚCASEÚWHENÚTHENÚELSEÚENDÚCOUNTÚSUMÚAVGÚMINÚMAXÚDISTINCTÚANDÚORÚNOTÚINÚEXISTSÚBETWEENÚLIKEzIS NULLz IS NOT NULLÚCASTÚCOALESCEÚNULLIFÚIFNULLÚDATEÚTIMEÚ TIMESTAMPÚEXTRACTÚ DATE_FORMATÚ STR_TO_DATEÚ CURRENT_DATEÚCURRENT_TIMESTAMPÚsqlÚreturnc ó — | j « }g }t D ]E }dt j |« z dz }t j ||« sŒ5|j |« ŒG t
|
|||
|
|
|« dk( }|st j d|› �« ||fS )u¡
|
|||
|
|
检查SQL是�包��险�作
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥ï¼ˆå¤§å°�写ä¸�æ•�感）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(是�安全, �险关键�列表)
|
|||
|
|
z\br u 检测到å�±é™©æ“�作: ) ÚupperÚDANGEROUS_KEYWORDSÚreÚescapeÚsearchÚappendÚlenÚloggerÚwarning)r> Ú sql_upperÚ
|
|||
|
|
found_dangersÚkeywordÚpatternÚis_safes úMC:\Users\24019\Desktop\backman-camel\text2sql_agent_camel\utils\validators.pyÚcheck_dangerous_operationsrP s… € ð — ‘ “ €IØ€Mç%ˆàœ"Ÿ)™) GÓ,Ñ,¨uÑ4ˆÜ
|
|||
|
|
�9‰9�W˜iÕ (Ø × Ñ Õ )ð &ô �-Ó AÑ%€Gá Ü�‰Ð0° °Ð@ÔAà �MÐ !Ð!ó c óì — | j « j « }|j « r|j « d nd}|dv rdd|› �fS t | « \ }}|sdddj |« › �fS y) uw
|
|||
|
|
验è¯�SQLä¸�是DML/DDLæ“�作（仅å…�许SELECTç‰æŸ¥è¯¢ï¼‰
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(是�通过, 错误消�)
|
|||
|
|
r Ú )r r r r r
|
|||
|
|
r r Fu ç¦�æ¢çš„æ“�作: u SQL包å�«å�±é™©æ“�作: ú, )TrS )ÚstriprA ÚsplitrP Újoin)r> rJ Ú
|
|||
|
|
first_wordrN Údangerss rO Úvalidate_no_dmlrZ ; s„ € ð — ‘ “ ×!Ñ!Ó#€Ið *3¯ © Ô):�— ‘ Ó" 1Ò%À€JØÐZÑZØÐ)¨*¨ Ð6Ð6Ð6ä1°#Ó6Ñ€GˆWÙ ØÐ/°· ± ¸'Ó0BÐ/CÐDÐDÐDà rQ c óÚ — dddddddœ}g }| j « }|j « D ]< \ }}t j || t j « sŒ,|j |« Œ> |S )u|
|
|||
|
|
检查明显的SQL注入模�
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
�现的注入模�列表
|
|||
|
|
zUNION\s+ALL\s+SELECTzUNION\s+SELECTz(--|\#|/\*).*SELECTz ;\s*(DROP|DELETE|UPDATE|INSERT)z OR\s+['\"]?\s*1\s*['\"]?\s*=\s*1z 1\s*=\s*1)Úunion_all_injectionÚunion_injectionÚcomment_injectionÚsemicolon_injectionÚor_true_conditionÚ always_true)ÚlowerÚitemsrC rE Ú
|
|||
|
|
IGNORECASErF )r> ÚpatternsÚfindingsÚ sql_lowerÚnamerM s rO Ú check_sql_injection_patternsri P sk € ð 7Ø,Ø 3Ø AØ @Ø#ñ
|
|||
|
|
€Hð €HØ— ‘ “ €Ià!Ÿ ™ Ö)‰
|
|||
|
|
ˆˆgÜ
|
|||
|
|
�9‰9�W˜c¤2§=¡=Õ 1Ø �O‰O˜DÕ !ð *ð €OrQ Údialectc ó° ‡— ddl m}mŠ g } || |¬« }t |‰j « sdg fS |j
|
|||
|
|
}|j j d« }|s7t ˆfd„|D « « }|r|j d« t |« dk( |fS |j
|
|||
|
|
}t « } |D ]‰ }
|
|||
|
|
t |
|
|||
|
|
‰j « r | j |
|
|||
|
|
j « Œ5t |
|
|||
|
|
‰j « sŒL|
|
|||
|
|
j } t | ‰j « sŒo| j | j « Œ‹ |D ]ß }
|
|||
|
|
|
|
|||
|
|
j# ‰j$ « rŒ t |
|
|||
|
|
‰j& « rŒ6t |
|
|||
|
|
‰j « r'|
|
|||
|
|
j } | | vsŒ]|j d| › d �« Œst |
|
|||
|
|
‰j( « sŒŠ|
|
|||
|
|
j } t | ‰j « sŒ| j } | | vsŒ¾|j d| › d
|
|||
|
|
|
|
|||
|
|
j* › d �« Œá t |« dk( |fS # t, $ r"}
|
|||
|
|
t. j1 d |
|
|||
|
|
› �« Y d
|
|||
|
|
}
|
|||
|
|
~
|
|||
|
|
Œ6d
|
|||
|
|
}
|
|||
|
|
~
|
|||
|
|
ww xY w)uá
|
|||
|
|
验è¯�è�šå�ˆæŸ¥è¯¢çš„GROUP BYæ£ç¡®æ€§
|
|||
|
|
|
|||
|
|
检查：SELECTä¸çš„é�žè�šå�ˆå—段是å�¦éƒ½åœ¨GROUP BYä¸
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
dialect: SQL方言
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
(是�有效, 错误列表)
|
|||
|
|
r ©Ú parse_oneÚexp©rj TÚgroupc 3 óX •K — | ]! }|j ‰j « d u–— Œ# y w)N)ÚfindÚAggFunc)Ú.0Úexprrn s €rO Ú <genexpr>z/validate_aggregation_groupby.<locals>.<genexpr>Œ s+ øè ø€ ð á (�Dð — ‘ ˜#Ÿ+™+Ó&¨dÔ2Ù (ùs ƒ'*u) 包å�«è�šå�ˆå‡½æ•°ä½†ç¼ºå°‘GROUP BYå�å�¥u å—段 'u" ' 在SELECTä¸ä½†ä¸�在GROUP BYä¸u
|
|||
|
|
' (别å��为'u# ') 在SELECTä¸ä½†ä¸�在GROUP BYä¸u GROUP BY验è¯�异常: N)Úutils.sql_parserrm rn Ú
|
|||
|
|
isinstanceÚSelectÚ expressionsÚargsÚgetÚanyrF rG ÚsetÚColumnÚaddrh ÚOrderedÚthisrr rs ÚLiteralÚAliasÚaliasÚ ExceptionrH Údebug)r> rj rm ÚerrorsÚparsedÚ select_exprsÚgroup_byÚhas_aggÚgroup_by_exprsÚ
|
|||
|
|
group_by_colsru r‚ Úcol_nameÚern s @rO Ú validate_aggregation_groupbyr‘ m s ø€ ÷ 0à
|
|||
|
|
€Fð?3Ù˜3¨Ô0ˆô ˜& #§*¡*Ô-ؘ�8ˆOð ×)Ñ)ˆ ð —;‘;—?‘? 7Ó+ˆÙäó á (óó ˆGñ Ø—
|
|||
|
|
‘
|
|||
|
|
Ð IÔJÜ�v“; !Ñ# VÐ+Ð +à!×-Ñ-ˆô ›ˆ
|
|||
|
|
Û"ˆDܘ$ §
|
|||
|
|
¡
|
|||
|
|
Ô+Ø ×!Ñ! $§)¡)Õ,ܘD #§+¡+Õ.à—y‘y�Ü ˜d C§J¡JÕ/Ø!×%Ñ% d§i¡iÕ0ð #ó !ˆDà�y‰y˜Ÿ™Ô%Øô ˜$ § ¡ Ô,Øô ˜$ §
|
|||
|
|
¡
|
|||
|
|
Ô+Ø Ÿ9™9�Ø =Ò0Ø—M‘MØ" 8 *Ð,NÐOõô ˜D #§)¡)Õ,à—y‘y�Ü ˜d C§J¡JÕ/Ø #Ÿy™y�HØ }Ò4Ø Ÿ
|
|||
|
|
™
|
|||
|
|
Ø & x j°
|
|||
|
|
¸d¿j¹j¸\ÐIlÐ mõñ- !ô: ˆv‹;˜!Ñ ˜VÐ #Ð#øô ò 3Ü� ‰ Ð-¨a¨SÐ1×2Ñ2ûð3úsI �#H* ±A H* ÂA#H* Ã5"H* ÄA;H* Æ,H* Ç"H* Ç$H* Ç5$H* È* IÈ3IÉIc óº — ddl m}m} g } || |¬« }|j |j « D ]w }|j
|
|||
|
|
j
|
|||
|
|
d« }|€0|j
|
|||
|
|
j
|
|||
|
|
d« }|�Œ>|j d« ŒPt ||j « sŒg|j d« Œy |S # t $ r#} t j d | › �« Y d} ~ |S d} ~ ww xY w)
|
|||
|
|
už
|
|||
|
|
检查JOIN�件是�完整
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
sql: SQLè¯å�¥
|
|||
|
|
dialect: SQL方言
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
é—®é¢˜åˆ—è¡¨ï¼ˆç©ºè¡¨ç¤ºæ— é—®é¢˜ï¼‰
|
|||
|
|
r rl ro ÚonNÚusingu JOIN缺少ONæ�¡ä»¶u JOINçš„ONæ�¡ä»¶ä¸ºç©ºu JOINæ�¡ä»¶æ£€æŸ¥å¼‚常: )
|
|||
|
|
rw rm rn Úfind_allÚJoinr{ r| rF rx ÚEmptyr† rH r‡ )
|
|||
|
|
r> rj rm rn Úissuesr‰ rW Ú on_conditionr” r� s
|
|||
|
|
rO Úcheck_join_conditionsrš Â sÄ € ÷ 0à
|
|||
|
|
€Fð5Ù˜3¨Ô0ˆð —O‘O C§H¡HÖ-ˆDà Ÿ9™9Ÿ=™=¨Ó.ˆLØÐ#à Ÿ ™ Ÿ
|
|||
|
|
™
|
|||
|
|
gÓ.�Ø‘=Ø—M‘MÐ"6Õ7ô ˜l¨C¯I©IÕ6Ø—M‘MÐ"9Õ:ñ .ð €Møô ò 5Ü� ‰ Ð/°¨sÐ3×4Ð4à €Mûð5ús$ ŒA"B. Á/(B. ÂB. Â. CÂ7CÃCc ó — ddl m}m} g } || |¬« }|j j d« }|r¢|j
|
|||
|
|
D ]“ }|j } t | |j « sŒ&| j }
|