如何使用PDO查询Mysql来避免SQL注入风险？ThinkPHP 3.1中的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。 PHP5.6中也将默认使用PDO的方式连接，mysql扩展将被作为辅助 。官方：http://php.net/manual/zh/book.pdo.php

1、PDO配置

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

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);
}
?>
view source?
<?php
foreach ($db->query("SELECT * FROM feeds") as $row)
{
    print_r($row);
}
?>

统计有多少行数据

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

prepare方式

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

prepare参数化查询

<?php
$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创建链接的例子：

<?php
$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中是忽略字符参数的。

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

<?php
$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;