parent
c3e502d844
commit
d68e401b42
@ -0,0 +1,95 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2023 - 2024. KeyWare.Co.Ltd All rights reserved. |
||||||
|
* 项目名称:C++ 信息安全性设计准则 |
||||||
|
* 项目描述:用于检查C++源代码的安全性设计准则的Sonarqube插件 |
||||||
|
* 版权说明:本软件属北京关键科技股份有限公司所有,在未获得北京关键科技股份有限公司正式授权情况下,任何企业和个人,不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。 |
||||||
|
*/ |
||||||
|
package com.keyware.sonar.cxx.rules.checkers; |
||||||
|
|
||||||
|
import com.keyware.sonar.cxx.SubscriptionAstVisitor; |
||||||
|
import com.sonar.cxx.sslr.api.AstNode; |
||||||
|
import com.sonar.cxx.sslr.api.AstNodeType; |
||||||
|
import com.sonar.cxx.sslr.api.Grammar; |
||||||
|
import org.sonar.check.Priority; |
||||||
|
import org.sonar.check.Rule; |
||||||
|
import org.sonar.cxx.parser.CxxGrammarImpl; |
||||||
|
import org.sonar.cxx.squidbridge.annotations.ActivatedByDefault; |
||||||
|
import org.sonar.cxx.squidbridge.annotations.SqaleConstantRemediation; |
||||||
|
import org.sonar.cxx.squidbridge.checks.SquidCheck; |
||||||
|
|
||||||
|
import javax.annotation.Nonnull; |
||||||
|
import java.util.ArrayList; |
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
/** |
||||||
|
* TODO SQLVerifyChecker |
||||||
|
* |
||||||
|
* @author RenFengJiang |
||||||
|
* @date 2024/1/14 |
||||||
|
*/ |
||||||
|
@Rule(key = "SQLVerifyChecker", name = "在使用SQL语句前对输入数据进行验证", description = "在拼接SQL语句前对输入数据进行验证,确保输入数据不包含SQL语句的关键字符;或使用参数化SQL查询语句,并将输入数据作为SQL语句的参数", priority = Priority.INFO, tags = {"28suo"}) |
||||||
|
@ActivatedByDefault |
||||||
|
@SqaleConstantRemediation("5min") |
||||||
|
public class SQLVerifyChecker extends SquidCheck<Grammar> { |
||||||
|
|
||||||
|
@Override |
||||||
|
public void init() { |
||||||
|
// 指定当前访问器需要访问的节点类型,functionBody(函数)主体节点
|
||||||
|
this.subscribeTo( |
||||||
|
CxxGrammarImpl.functionBody |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* 访问AST节点 |
||||||
|
* |
||||||
|
* @param node 要处理的AST节点,该节点类型为通过subscribeTo方法订阅的类型 |
||||||
|
*/ |
||||||
|
@Override |
||||||
|
public void visitNode(@Nonnull AstNode node) { |
||||||
|
IfSQL ifSQL = new IfSQL(this); |
||||||
|
ifSQL.accept(node); |
||||||
|
} |
||||||
|
|
||||||
|
class IfSQL extends SubscriptionAstVisitor{ |
||||||
|
|
||||||
|
List lists = new ArrayList<String>(); |
||||||
|
public IfSQL(SquidCheck<Grammar> checker){ |
||||||
|
super(checker); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public List<AstNodeType> visitNodeTypes() { |
||||||
|
// 指定当前访问器需要访问的节点类型,这里指定了simpleDeclaration(简单声明)节点类型
|
||||||
|
return List.of(CxxGrammarImpl.functionBody); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
public void visitNode(@Nonnull AstNode node) { |
||||||
|
//选择节点语句
|
||||||
|
List<AstNode> nodeDescendants = node.getDescendants(CxxGrammarImpl.selectionStatement); |
||||||
|
for (AstNode astNode:nodeDescendants) { |
||||||
|
//判断节点是不是if节点
|
||||||
|
if("if".equals(astNode.getToken().getValue())){ |
||||||
|
//获取其中的参数
|
||||||
|
List<AstNode> astNodeList = astNode.getDescendants(CxxGrammarImpl.expressionList); |
||||||
|
for (AstNode expr:astNodeList) { |
||||||
|
lists.add(expr.getToken().getValue()); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
//获取表达是数据信息
|
||||||
|
List<AstNode> descendants = node.getDescendants(CxxGrammarImpl.expressionList); |
||||||
|
for (AstNode ast:descendants) { |
||||||
|
String name = ast.getParent().getToken().getValue(); |
||||||
|
//判断是否时使用sanitizeString方法
|
||||||
|
if("sanitizeString".equals(name)){ |
||||||
|
//判断方法参数是否进行校验
|
||||||
|
if(!lists.contains(ast.getToken().getValue())){ |
||||||
|
reportIssue(ast, "在使用SQL语句前对输入数据进行验证"); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,34 @@ |
|||||||
|
/* |
||||||
|
* Copyright (c) 2023 - 2024. KeyWare.Co.Ltd All rights reserved. |
||||||
|
* 项目名称:C++ 信息安全性设计准则 |
||||||
|
* 项目描述:用于检查C++源代码的安全性设计准则的Sonarqube插件 |
||||||
|
* 版权说明:本软件属北京关键科技股份有限公司所有,在未获得北京关键科技股份有限公司正式授权情况下,任何企业和个人,不能获取、阅读、安装、传播本软件涉及的任何受知识产权保护的内容。 |
||||||
|
*/ |
||||||
|
package com.keyware.sonar.cxx.rules.checkers; |
||||||
|
|
||||||
|
import com.keyware.sonar.cxx.CxxFileTesterHelper; |
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
import org.sonar.cxx.CxxAstScanner; |
||||||
|
import org.sonar.cxx.squidbridge.api.SourceFile; |
||||||
|
import org.sonar.cxx.squidbridge.checks.CheckMessagesVerifier; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
|
||||||
|
/** |
||||||
|
* TODO SQLVerifyChecker |
||||||
|
* |
||||||
|
* @author RenFengJiang |
||||||
|
* @date 2024/1/14 |
||||||
|
*/ |
||||||
|
public class SQLVerifyCheckerTest { |
||||||
|
|
||||||
|
@Test |
||||||
|
public void checkTest() throws IOException { |
||||||
|
var checker = new SQLVerifyChecker(); |
||||||
|
var tester = CxxFileTesterHelper.create("SQLVerifyChecker.cc"); |
||||||
|
SourceFile file = CxxAstScanner.scanSingleInputFile(tester.asInputFile(), checker); |
||||||
|
CheckMessagesVerifier.verify(file.getCheckMessages()) |
||||||
|
.next().atLine(7).withMessage("在使用SQL语句前对输入数据进行验证") |
||||||
|
.noMore(); |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,19 @@ |
|||||||
|
int main() { |
||||||
|
|
||||||
|
std::string inputQuery = ""; |
||||||
|
// if(a(inputQuery)){
|
||||||
|
// }
|
||||||
|
// 对输入的SQL语句进行验证和处理
|
||||||
|
std::string sqlQuery = sanitizeString(inputQuery); // error
|
||||||
|
|
||||||
|
// 执行SQL语句
|
||||||
|
stmt = con->createStatement(); |
||||||
|
res = stmt->executeQuery(sqlQuery); |
||||||
|
|
||||||
|
// 处理查询结果
|
||||||
|
while (res->next()) { |
||||||
|
// 从结果集中获取数据并进行处理
|
||||||
|
std::string resultData = res->getString("column_name"); |
||||||
|
std::cout << "查询结果: " << resultData << std::endl; |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue