node

SQL注入對於網站及服務器來講都是一個非常危險問題，程序員在這一方面沒處理好的話網站可能隨時給注入了，今天我們來講一篇關於node-mysql中防止SQL注入常用做法，具體如下。

備注： 本文針對mysqljs/mysql。

為了防止SQL注入，可以將SQL中傳入參數進行編碼，而不是直接進行字符串拼接。在node-mysql中，防止SQL注入的常用方法有以下四種：

方法一：使用escape()對傳入參數進行編碼：

參數編碼方法有如下三個：

mysql.escape(param)
connection.escape(param)
pool.escape(param)

例如：

var userId = 1, name = 'test';
var query = connection.query('SELECT * FROM users WHERE id = ' + connection.escape(userId) + ', name = ' + connection.escape(name), function(err, results) {
    // ...
});
console.log(query.sql); // SELECT * FROM users WHERE id = 1, name = 'test'
escape()方法編碼規則如下：
Numbers不進行轉換；
Booleans轉換為true/false；
Date對象轉換為’YYYY-mm-dd HH:ii:ss’字符串；
Buffers轉換為hex字符串，如X’0fa5’；
Strings進行安全轉義；
Arrays轉換為列表，如[‘a’, ‘b’]會轉換為’a’, ‘b’；
多維數組轉換為組列表，如[[‘a’, ‘b’], [‘c’, ‘d’]]會轉換為’a’, ‘b’), (‘c’, ‘d’)；
Objects會轉換為key=value鍵值對的形式。嵌套的對象轉換為字符串；
undefined/null會轉換為NULL；
MySQL不支持NaN/Infinity，並且會觸發MySQL錯誤。

方法二：使用connection.query()的查詢參數占位符：

可使用 ? 做為查詢參數占位符。在使用查詢參數占位符時，在其內部自動調用 connection.escape() 方法對傳入參數進行編碼。

如：

var userId = 1, name = 'test';
var query = connection.query('SELECT * FROM users WHERE id = ?, name = ?', [userId, name], function(err, results) {
    // ...
});
console.log(query.sql); // SELECT * FROM users WHERE id = 1, name = 'test'
上面程序還可以改寫成如下：

var post  = {userId: 1, name: 'test'};
var query = connection.query('SELECT * FROM users WHERE ?', post, function(err, results) {
    // ...
});
console.log(query.sql); // SELECT * FROM users WHERE id = 1, name = 'test'

方法三： 使用escapeId()編碼SQL查詢標識符：

如果你不信任用戶傳入的SQL標識符（數據庫、表、字符名），可以使用escapeId()方法進行編碼。最常用於排序等。escapeId()有如下三個功能相似的方法：

mysql.escapeId(identifier)
connection.escapeId(identifier)
pool.escapeId(identifier)
例如：

var sorter = 'date';
var sql    = 'SELECT * FROM posts ORDER BY ' + connection.escapeId(sorter);
connection.query(sql, function(err, results) {
  // ...
});

方法四： 使用mysql.format()轉義參數：

准備查詢，該函數會選擇合適的轉義方法轉義參數 　　 mysql.format()用於准備查詢語句，該函數會自動的選擇合適的方法轉義參數。

例如：

var userId = 1;
var sql = "SELECT * FROM ?? WHERE ?? = ?";
var inserts = ['users', 'id', userId];
sql = mysql.format(sql, inserts); // SELECT * FROM users WHERE id = 1