Php中用PDO查询Mysql来避免SQL注入风险的方法

当我们使用传统的 mysql_connect 、mysql_query方法来连接查询数据库时，如果过滤不严，就有sql注入风险，导致网站被攻击，失去控制。虽然可以用mysql_real_escape_string()函数过滤用户提交的值，但是也有缺陷。而使用php的pdo扩展的 prepare 方法，就可以避免sql injection 风险。

pdo(php data object) 是php5新加入的一个重大功能，因为在php 5以前的php4/php3都是一堆的数据库扩展来跟各个数据库的连接和处理，如 php_mysql.dll。 php6中也将默认使用pdo的方式连接，mysql扩展将被作为辅助 。官方：http://php.net/manual/en/book.pdo.php

1、pdo配置
使用pdo扩展之前，先要启用这个扩展，php.ini中，去掉"extension=php_pdo.dll"前面的";"号,若要连接数据库，还需要去掉与pdo相关的数据库扩展前面的";"号（一般用的是php_pdo_mysql.dll），然后重启apache服务器即可。

extension=php_pdo.dll
extension=php_pdo_mysql.dll

2、pdo连接mysql数据库

$dbh = new pdo("mysql:host=localhost;dbname=db_demo","root","password");

默认不是长连接，若要使用数据库长连接，需要在最后加如下参数:

$dbh = new pdo("mysql:host=localhost;dbname=db_demo","root","password","array(pdo::attr_persistent => true)");
$dbh = null; //(释放)

3、pdo设置属性

1) pdo有三种错误处理方式：

• pdo::errmode_silent不显示错误信息，只设置错误码
• pdo::errmode_warning显示警告错
• pdo::errmode_exception抛出异常

可通过以下语句来设置错误处理方式为抛出异常

$db->setattribute(pdo::attr_errmode, pdo::errmode_exception);

当设置为pdo::errmode_silent时可以通过调用errorcode() 或errorinfo()来获得错误信息，当然其他情况下也可以。

2) 因为不同数据库对返回的字段名称大小写处理不同，所以pdo提供了pdo::attr_case设置项（包括pdo::case_lower，pdo::case_natural，pdo::case_upper），来确定返回的字段名称的大小写。

3) 通过设置pdo::attr_oracle_nulls类型（包括pdo::null_natural，pdo::null_empty_string，pdo::null_to_string）来指定数据库返回的null值在php中对应的数值。

4、pdo常用方法及其应用
pdo::query() 主要是用于有记录结果返回的操作，特别是select操作
pdo::exec() 主要是针对没有结果集合返回的操作，如insert、update等操作
pdo::prepare() 主要是预处理操作，需要通过$rs->execute()来执行预处理里面的sql语句，这个方法可以绑定参数，功能比较强大（防止sql注入就靠这个）
pdo::lastinsertid() 返回上次插入操作，主键列类型是自增的最后的自增id
pdostatement::fetch() 是用来获取一条记录
pdostatement::fetchall() 是获取所有记录集到一个集合
pdostatement::fetchcolumn() 是获取结果指定第一条记录的某个字段，缺省是第一个字段
pdostatement::rowcount() :主要是用于pdo::query()和pdo::prepare()进行delete、insert、update操作影响的结果集，对pdo::exec()方法和select操作无效。

5、pdo操作mysql数据库实例

<?php
$pdo = new pdo("mysql:host=localhost;dbname=db_demo","root","");
if($pdo -> exec("insert into db_demo(name,content) values('title','content')")){
echo "插入成功！";
echo $pdo -> lastinsertid();
}
?>

<?php
$pdo = new pdo("mysql:host=localhost;dbname=db_demo","root","");
$rs = $pdo -> query("select * from test");
$rs->setfetchmode(pdo::fetch_assoc); //关联数组形式
//$rs->setfetchmode(pdo::fetch_num); //数字索引数组形式
while($row = $rs -> fetch()){
print_r($row);
}
?>

<?php
foreach( $db->query( "select * from feeds" ) as $row )
{
    print_r( $row );
}
?>

统计有多少行数据

$sql="select count(*) from test";
$num = $dbh->query($sql)->fetchcolumn();

prepare方式

$stmt = $dbh->prepare("select * from test");
if ($stmt->execute()) {
 while ($row = $stmt->fetch()) {
     print_r($row);
 }
}

prepare参数化查询

$stmt = $dbh->prepare("select * from test where name = ?");
if ($stmt->execute(array("david"))) {
 while ($row = $stmt->fetch(pdo::fetch_assoc)) {
     print_r($row);
 }
}

【下面来说说重点了，如何防止 sql注入】

使用pdo访问mysql数据库时，真正的real prepared statements 默认情况下是不使用的。为了解决这个问题，你必须禁用 prepared statements的仿真效果。下面是使用pdo创建链接的例子：

$dbh = new pdo('mysql:dbname=dbtest;host=127.0.0.1;charset=utf8', 'user', 'pass');
$dbh->setattribute(pdo::attr_emulate_prepares, false);

setattribute（）这一行是强制性的，它会告诉 pdo 禁用模拟预处理语句，并使用 real parepared statements 。这可以确保sql语句和相应的值在传递到mysql服务器之前是不会被php解析的（禁止了所有可能的恶意sql注入攻击）。虽然你可以配置文件中设置字符集的属性(charset=utf8)，但是需要格外注意的是，老版本的 php（ < 5.3.6）在dsn中是忽略字符参数的。

我们来看一段完整的代码使用实例：

$dbh = new pdo("mysql:host=localhost; dbname=demo", "user", "pass");
$dbh->setattribute(pdo::attr_emulate_prepares, false); //禁用prepared statements的仿真效果
$dbh->exec("set names 'utf8'");
$sql="select * from test where name = ? and password = ?";
$stmt = $dbh->prepare($sql);
$exeres = $stmt->execute(array($testname, $pass));
if ($exeres) {
 while ($row = $stmt->fetch(pdo::fetch_assoc)) {
     print_r($row);
 }
}
$dbh = null;

上面这段代码就可以防范sql注入。为什么呢？

当调用 prepare() 时，查询语句已经发送给了数据库服务器，此时只有占位符 ? 发送过去，没有用户提交的数据；当调用到 execute()时，用户提交过来的值才会传送给数据库，他们是分开传送的，两者独立的，sql攻击者没有一点机会。

但是我们需要注意的是以下几种情况，pdo并不能帮助你防范sql注入

1、你不能让占位符 ? 代替一组值，如：

select * from blog where userid in ( ? );

2、你不能让占位符代替数据表名或列名，如：

select * from blog order by ?;

3、你不能让占位符 ? 代替任何其他sql语法，如：

select extract( ? from datetime_column) as variable_datetime_element from blog;