OLEDB 参数化查询

发表于 | 分类于 | | 本文总阅读量 次

一般情况下，SQL查询是相对固定的，一条语句变化的可能只是条件值，比如之前要求查询二年级学生信息，而后面需要查询三年级的信息，这样的查询一般查询的列不变，后面的条件只有值在变化，针对这种查询可以使用参数化查询的方式来提高效率，也可以时SQL操作更加安全，从根本上杜绝SQL注入的问题。

参数化查询的优势：

  1. 提高效率：之前说过，数据库在执行SQL的过程中，每次都会经过SQL的解析，编译，调用对应的数据库组件，这样如果执行多次同样类型的SQL语句，解析，编译的过程明显是在浪费资源，而参数化查询就是使用编译好的过程（也就是提前告诉数据库要调用哪些数据库组件），这样就跳过了对SQL语句的解析，编译过程，提高了效率（这个过程我觉得有点类似于C/C++语言的编译执行与脚本语言的解释执行）。
  2. 更加安全：从安全编程的角度来说，对于防范SQL注入方面，它比关键字过滤更有效，实现起来也更加方便。

科普SQL注入和安全编程

  • 什么是SQL注入：

    所谓SQL注入，就是通过把SQL命令插入到Web表单提交或输入域名或页面请求的查询字符串，最终达到欺骗服务器执行恶意的SQL命令。举个例子来说在用户登录时会输入用户名密码，这个时候在后台就可以执行这样的SQL语句

    1
    select count(*) from user where username = 'haha' and password = '123456'

    只有输入对了用户名和密码才能登录，但是如果没有对用户输入进行校验，当用户输入一些SQL中的语句，而后台直接将用户输入进行拼接并执行，就会发生注入，比如此时用户输入 ‘haha’ or 1 = 1 – ,此时再后台执行的sql语句就变成了这样：

    1
    select count(*) from user where username = 'haha' or 1 = 1 -- and password = ''

    这样用户就可以不用密码，直接使用用户名就登录了。而防范这类攻击，一般采用的是关键字过滤的方式，但是关键字过滤并不能杜绝这类工具，当一时疏忽忘记了过滤某个关键字仍然会产生这类问题。而且关键字过滤一般采用正则表达式，而正则表达式并不是一般人可以驾驭的。而防范SQL注入最简单也是最一劳永逸的方式就是参数化查询。

  • 为什么参数化查询能够从根本上解决SQL注入

    发生SQL注入一般的原因是程序将用户输入当做SQL语句的一部分进行执行，但是参数化查询它只是将用户输入当做参数，当做查询的条件，从数据库的层面上来说，它不对应于具体的数据库组件，它只是一组数据，而不会执行。这里可以简单的将传统的SQL拼接方式理解为C语言中的宏，宏也可以有参数，但是它不对参数进行校验，只是简单的进行替换，那么我可以使用一些指令作为参数传入，但是函数就不一样，函数的参数就是具体类型的变量或者常量。所以参数化查询从根本上解决的SQL注入的问题。

参数化查询的使用

前面说了这么多参数化查询的好处，那么到底怎么使用它呢？
在Java等语言中内置了数据库操作，而对于C/C++来说，它并没有提供这方方面的标准。不同的平台有自己独特的一套机制，但是从总体来说，思想是共通的，只是语法上的不同，这里主要是说明OLEDB中的使用方式。

  1. 使用“?”符将SQL语句中的条件值常量进行替换，组成一个新的SQL语句，比如上面登录的查询语句可以写成

    1
    select count(*) from user where username = ? and password = ?

  2. 调用ICommandText的SetCommandText设置sql语句。

  3. 调用ICommandParpare的Prepare方法对含有”?”的语句进行预处理
  4. 调用ICommandWithParameters方法的GetParameterInfo方法获取参数详细信息的DBPARAMINFO结构（类似于DBCOLUMNINFO）
  5. 分配对应大小的DBBINDING缓冲用来保存每个参数的绑定信息
  6. 调用IAccessor的CreateAccessor方法创建对应的访问器
  7. 为参数分配缓冲，设置合适的参数后准备DBPARAMS结构
  8. 调用ICommandText的Execute方法并将DBPARAMS结构的指针作为参数传入。
  9. 操作返回的结果集对象
1
2
3
4
5
typedef struct tagDBPROPIDSET {
   DBPROPID *   rgPropertyIDs;
   ULONG        cPropertyIDs;
   GUID         guidPropertySet;
} DBPROPIDSET;

DBPARAMS结构的定义如下:

1
2
3
4
5
6
typedef struct tagDBPARAMS
{
    void *pData;
    DB_UPARAMS cParamSets;
    HACCESSOR hAccessor;
} 	DBPARAMS;

  • pData是保存参数信息的缓冲；
  • cParamSets: 表示又多少个参数
  • hAccessor: 之前获取到的绑定结构的访问器句柄

下面是一个使用的例子:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
BOOL QueryData(LPOLESTR pQueryStr, IOpenRowset* pIOpenRowset, IRowset* &pIRowset)
{
	IAccessor *pParamAccessor = NULL; //与参数化查询相关的访问器接口

	LPOLESTR pSql = _T("Select * From aa26 Where Left(aac031,2) = ?"); //参数化查询语句
	BOOL bRet = FALSE;
	DB_UPARAMS uParams = 0;
	DBPARAMINFO* rgParamInfo = NULL;
	LPOLESTR pParamBuffer = NULL;
	DWORD dwOffset = 0;
	DBBINDING *rgParamBinding = NULL;
	HACCESSOR hAccessor = NULL;
	DBPARAMS dbParams = {0};
	DBBINDSTATUS *pdbBindStatus = NULL;
	//设置SQL
	hRes = pICommandText->SetCommandText(DBGUID_DEFAULT, pSql);

	//预处理SQL命令
	pICommandPrepare->Prepare(0);
	hRes = pICommandText->QueryInterface(IID_ICommandPrepare, (void**)&pICommandPrepare);

	//获取参数信息
	hRes = pICommandText->QueryInterface(IID_ICommandWithParameters, (void**)&pICommandWithParameters);
	COM_SUCCESS(hRes, _T("查询接口ICommandWithParameters失败，错误码为:%08x\n"), hRes);
	hRes = pICommandWithParameters->GetParameterInfo(&uParams, &rgParamInfo, &pParamBuffer);
	COM_SUCCESS(hRes, _T("获取参数信息失败，错误码为:%08x\n"), hRes);

	rgParamBinding = (DBBINDING*)MALLOC(sizeof(DBBINDING) * uParams);
	ZeroMemory(rgParamBinding, sizeof(DBBINDING) * uParams);

	//绑定参数信息
	for (int i = 0; i < uParams; i++)
	{
		rgParamBinding[i].bPrecision = rgParamInfo[i].bPrecision;
		rgParamBinding[i].bScale = rgParamInfo[i].bScale;
		rgParamBinding[i].cbMaxLen = 7 * sizeof(WCHAR); //行政区编号最大长度为6
		rgParamBinding[i].dwMemOwner = DBMEMOWNER_CLIENTOWNED;
		rgParamBinding[i].dwPart = DBPART_LENGTH | DBPART_VALUE;
		rgParamBinding[i].eParamIO = DBPARAMIO_INPUT;
		rgParamBinding[i].iOrdinal = rgParamInfo[i].iOrdinal;
		rgParamBinding[i].obLength = dwOffset;
		rgParamBinding[i].obStatus = 0;
		rgParamBinding[i].obValue = dwOffset + sizeof(ULONG);
		rgParamBinding[i].wType = DBTYPE_WSTR;

		dwOffset = dwOffset + sizeof(ULONG) + rgParamBinding[i].cbMaxLen;
		dwOffset = UPROUND(dwOffset);
	}

	//获取访问器
	pdbBindStatus = (DBBINDSTATUS*)MALLOC(uParams * sizeof(DBBINDSTATUS));
	ZeroMemory(pdbBindStatus, uParams * sizeof(DBBINDSTATUS))
	pParamAccessor->CreateAccessor(DBACCESSOR_PARAMETERDATA, uParams, rgParamBinding, dwOffset, &hAccessor, pdbBindStatus);
	COM_SUCCESS(hRes, _T("获取参数访问器失败，错误码为:%08x\n"), hRes);

	//准备参数
	dbParams.pData = MALLOC(dwOffset);

	ZeroMemory(dbParams.pData, dwOffset);
	dbParams.cParamSets = uParams;
	dbParams.hAccessor = hAccessor;
	for (int i = 0; i < uParams; i++)
	{
		*(ULONG*)((BYTE*)dbParams.pData + rgParamBinding[i].obLength) = _tcslen(pQueryStr) * sizeof(WCHAR);
		StringCchCopy((LPTSTR)((BYTE*)dbParams.pData + rgParamBinding[i].obValue), _tcslen(pQueryStr) + 1, pQueryStr);
	}

	//执行SQL
	hRes = pICommandText->Execute(NULL, IID_IRowset, &dbParams, NULL, (IUnknown**)&pIRowset);
	return bRet;
}

完整代码