class Setting

{

private String Command = String.Empty;

SqlConnection Conn = null;

public Setting(SqlConnection Connection)

{

Conn = Connection;

}

/// <summary>

/// 判断文件是否存在

/// </summary>

/// <param name="path">文件路径</param>

/// <param name="value">返回值匹配</param>

/// <returns></returns>

public bool File_Exists(String path, int value)

{

Command = String.Format(@"

SELECT @r as n", path);

if (int.Parse(Batch.RemoteExec(Conn, Command, false)) == value)

return true;

return false;

}

/// <summary>

/// 设置 configuration

/// </summary>

/// <param name="option">查询内容选项</param>

/// <param name="value">返回值匹配</param>

/// <returns></returns>

public bool Set_configuration(String option, int value)

{

Command = String.Format("exec master.dbo.sp_configure '{0}',{1}; RECONFIGURE;", option, value);

Batch.RemoteExec(Conn, Command, false);

return Check_configuration(option, value);

}

/// <summary>

/// 设置 permission_set

/// </summary>

/// <returns></returns>

public void Set_permission_set()

{

Command = String.Format("ALTER DATABASE master SET TRUSTWORTHY ON;");

Console.WriteLine("[+] ALTER DATABASE master SET TRUSTWORTHY ON");

Batch.CLRExec(Conn, Command);

}

/// <summary>

/// CREATE ASSEMBLY

/// </summary>

/// <returns></returns>

public void CREATE_ASSEMBLY()

{

Command = String.Format(@"CREATE ASSEMBLY [CLR_module]

Console.WriteLine("[+] Import the assembly");

Batch.CLRExec(Conn, Command);

}

/// <summary>

/// CREATE PROCEDURE

/// </summary>

/// <returns></returns>

public void CREATE_PROCEDURE()

{

Command = String.Format(@"CREATE PROCEDURE [dbo].[ClrExec]

Console.WriteLine("[+] Link the assembly to a stored procedure");

Batch.CLRExec(Conn, Command);

}

/// <summary>

/// 安装clr

/// </summary>

/// <returns>true/false</returns>

public bool install_clr()

{

try

{

Set_permission_set();

CREATE_ASSEMBLY();

CREATE_PROCEDURE();

}catch

{

return false;

}

Console.WriteLine("[+] Install clr successful!");

return true;

}

/// <summary>

/// drop_clr

/// </summary>

/// <returns></returns>

public void drop_clr()

{

Command = String.Format(@"drop PROCEDURE dbo.ClrExec

drop assembly CLR_module");

Console.WriteLine("[+] drop PROCEDURE and assembly");

Batch.CLRExec(Conn, Command);

}

/// <summary>

/// 检查 configuration 的配置

/// </summary>

/// <param name="option">查询内容选项</param>

/// <param name="value">返回值匹配</param>

/// <returns></returns>

public bool Check_configuration(String option, int value)

{

Command = String.Format("SELECT cast(value as INT) as v FROM sys.configurations where name = '{0}';", option);

if (int.Parse(Batch.RemoteExec(Conn, Command, false)) == value)

return true;

return false;

}

#region 启用/关闭 OLE Automation Procedures 配置

/// <summary>

/// 开启 OLA

/// </summary>

/// <returns>true/false</returns>

public bool Enable_ola()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("Ole Automation Procedures", 1))

{

Console.WriteLine("[!] cannot enable 'Ole Automation Procedures'");

return false;

}

return true;

}

/// <summary>

/// 关闭 OLA

/// </summary>

/// <returns>true/false</returns>

public bool Disable_ole()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("Ole Automation Procedures", 0))

{

Console.WriteLine("[!] cannot disable 'Ole Automation Procedures'");

return false;

}

if (!Set_configuration("show advanced options", 0))

{

Console.WriteLine("[!] cannot disable 'show advanced options'");

return false;

}

return true;

}

#endregion

#region 启用/关闭 xp_cmdshell

/// <summary>

/// 开启 xp_cmdshell

/// </summary>

/// <returns>true/false</returns>

public bool Enable_xp_cmdshell()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("xp_cmdshell", 1))

{

Console.WriteLine("[!] cannot enable 'xp_cmdshell'");

return false;

}

return true;

}

/// <summary>

/// 关闭 xp_cmdshell

/// </summary>

/// <returns>true/false</returns>

public bool Disable_xp_cmdshell()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("xp_cmdshell", 0))

{

Console.WriteLine("[!] cannot disable 'xp_cmdshell'");

return false;

}

if (!Set_configuration("show advanced options", 0))

{

Console.WriteLine("[!] cannot disable 'show advanced options'");

return false;

}

return true;

}

#endregion

#region 启用/关闭 clr

/// <summary>

/// 开启 clr enabled

/// </summary>

/// <returns>true/false</returns>

public bool Enable_clr()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("clr enabled", 1))

{

Console.WriteLine("[!] cannot enable 'clr enabled'");

return false;

}

return true;

}

/// <summary>

/// 关闭 clr enabled

/// </summary>

/// <returns>true/false</returns>

public bool Disable_clr()

{

if (!Set_configuration("show advanced options", 1))

{

Console.WriteLine("[!] cannot enable 'show advanced options'");

return false;

}

if (!Set_configuration("clr enabled", 0))

{

Console.WriteLine("[!] cannot disable 'clr enabled'");

return false;

}

if (!Set_configuration("show advanced options", 0))

{

Console.WriteLine("[!] cannot disable 'show advanced options'");

return false;

}

return true;

}

#endregion

}