Sqlserver
 sql >> Database >  >> RDS >> Sqlserver

Consentire a un utente di passare il nome della tabella e il nome della colonna impedendo l'iniezione di SQL

Questo è ciò che QUOTENAME() è stato creato per risolvere. Passi i nomi delle colonne e delle tabelle come parametri in QUOTENAME() e quindi ne usi l'output per rappresentare gli oggetti nel tuo database in una query sql dinamica.

//The evil name tries to expliot code like:
//  set @sql = N'CREATE TABLE [' + @tablename + N'] (Foo int)'
var evilName = "someName] (Foo int); Drop table students --";

var query = @"
declare @sql as nvarchar(max)
set @sql = N'CREATE TABLE ' + QUOTENAME(@tablename) + N' (Foo int)'
exec sp_executesql @sql
";
using(var connection = new SqlConnection(ConnectionString))
using(var command = new SqlCommand(query, connection))
{
    command.Parameters.Add("@tablename", SqlDbType.NVarChar, 128).Value = evilName ;
    connection.Open();
    command.ExecuteNonQuery();
}

La query che verrà eseguita sul server sarà

CREATE TABLE [someName]] (Foo int); Drop table students --] (Foo int)

che crea una tabella con un nome di tabella valido e non elimina l'altra mia tabella.