sql >> Base de Datos >  >> RDS >> Sqlserver

Permitir que un usuario pase el nombre de la tabla y el nombre de la columna mientras se evita la inyección de SQL

Esto es lo que QUOTENAME() fue creado para resolver. Pasas los nombres de tus columnas y tablas como parámetros en QUOTENAME() y luego usa el resultado para representar objetos en su base de datos en una consulta sql dinámica.

//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 consulta que se ejecutará en el servidor será

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

que crea una tabla con un nombre de tabla válido y no elimina mi otra tabla.