Use master Go if object_ID(‘[sp_EncryptObject]‘) is not null Drop Procedure [sp_EncryptObject] Go create procedure sp_EncryptObject ( @Object sysname=‘All‘ ) as /* 当@Object=All的时候,对所有的函数,存储过程,视图和触发器进行加密 调用方法: 1. Execute sp_EncryptObject ‘All‘ 2. Execute sp_EncryptObject ‘ObjectName‘ */ begin set nocount on if @Object <>‘All‘ begin if not exists(select 1 from sys.objects a where a.object_id=object_id(@Object) And a.type in(‘P‘,‘V‘,‘TR‘,‘FN‘,‘IF‘,‘TF‘)) begin --SQL Server 2008 raiserror 50001 N‘无效的加密对象!加密对象必须是函数,存储过程,视图或触发器。‘ --SQL Server 2012 --throw 50001, N‘无效的加密对象!加密对象必须是函数,存储过程,视图或触发器。‘,1 return end if exists(select 1 from sys.sql_modules a where a.object_id=object_id(@Object) and a.definition is null) begin --SQL Server 2008 raiserror 50001 N‘对象已经加密!‘ --SQL Server 2012 --throw 50001, N‘对象已经加密!‘,1 return end end declare @sql nvarchar(max),@C1 nchar(1),@C2 nchar(1),@type nvarchar(50),@Replace nvarchar(50) set @C1=nchar(13) set @C2=nchar(10) declare cur_Object cursor for select object_name(a.object_id) As ObjectName,a.definition from sys.sql_modules a inner join sys.objects b on b.object_id=a.object_id and b.is_ms_shipped=0 and not exists(select 1 from sys.extended_properties x where x.major_id=b.object_id and x.minor_id=0 and x.class=1 and x.name=‘microsoft_database_tools_support‘ ) where b.type in(‘P‘,‘V‘,‘TR‘,‘FN‘,‘IF‘,‘TF‘) and (b.name=@Object or @Object=‘All‘) and b.name <>‘sp_EncryptObject‘ and a.definition is not null order by Case when b.type =‘V‘ then 1 when b.type =‘TR‘ then 2 when b.type in(‘FN‘,‘IF‘,‘TF‘) then 3 else 4 end,b.create_date,b.object_id open cur_Object fetch next from cur_Object into @Object,@sql while @@fetch_status=0 begin Begin Try if objectproperty(object_id(@Object),‘ExecIsAfterTrigger‘)=0 set @Replace=‘As‘ ; else set @Replace=‘For ‘; if (patindex(‘%‘+@C1+@C2+@Replace+@C1+@C2+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C1+@C2+@Replace+@C1+@C2,@C1+@C2+‘With Encryption‘+@C1+@C2+@Replace+@C1+@C2) end else if(patindex(‘%‘+@C1+@Replace+@C1+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C1+@Replace+@C1,@C1+‘With Encryption‘+@C1+@Replace+@C1) end else if(patindex(‘%‘+@C2+@Replace+@C2+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C2+@Replace+@C2,@C2+‘With Encryption‘+@C2+@Replace+@C2) end else if(patindex(‘%‘+@C2+@Replace+@C1+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C2+@Replace+@C1,@C1+‘With Encryption‘+@C2+@Replace+@C1) end else if(patindex(‘%‘+@C1+@C2+@Replace+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C1+@C2+@Replace,@C1+@C2+‘With Encryption‘+@C1+@C2+@Replace) end else if(patindex(‘%‘+@C1+@Replace+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C1+@Replace,@C1+‘With Encryption‘+@C1+@Replace) end else if(patindex(‘%‘+@C2+@Replace+‘%‘,@sql)>0) begin set @sql=Replace(@sql,@C2+@Replace,@C2+‘With Encryption‘+@C2+@Replace) end set @type = case when object_id(@Object,‘P‘)>0 then ‘Proc‘ when object_id(@Object,‘V‘)>0 then ‘View‘ when object_id(@Object,‘TR‘)>0 then ‘Trigger‘ when object_id(@Object,‘FN‘)>0 or object_id(@Object,‘IF‘)>0 or object_id(@Object,‘TF‘)>0 then ‘Function‘ end set @sql=Replace(@sql,‘Create ‘+@type,‘Alter ‘+@type) Begin Transaction exec(@sql) print N‘已完成加密对象(‘+@type+‘):‘+@Object Commit Transaction End Try Begin Catch Declare @Error nvarchar(2047) Set @Error=‘Object: ‘+@Object+@C1+@C2+‘Error: ‘+Error_message() Rollback Transaction print @Error print @sql End Catch fetch next from cur_Object into @Object,@sql end close cur_Object deallocate cur_Object end Go exec sp_ms_marksystemobject ‘sp_EncryptObject‘ --标识为系统对象 go
Use master Go if object_ID(‘[sp_DecryptObject]‘) is not null Drop Procedure [sp_DecryptObject] Go create procedure sp_DecryptObject ( @Object sysname, --要解密的对象名:函数,存储过程,视图或触发器 @MaxLength int=4000 --评估内容的长度 ) as set nocount on /* 1. 解密 */ if not exists(select 1 from sys.objects a where a.object_id=object_id(@Object) And a.type in(‘P‘,‘V‘,‘TR‘,‘FN‘,‘IF‘,‘TF‘)) begin --SQL Server 2008 raiserror 50001 N‘无效的对象!要解密的对象必须是函数,存储过程,视图或触发器。‘ --SQL Server 2012 --throw 50001, N‘无效的对象!要解密的对象必须是函数,存储过程,视图或触发器。‘,1 return end if exists(select 1 from sys.sql_modules a where a.object_id=object_id(@Object) and a.definition is not null) begin --SQL Server 2008 raiserror 50001 N‘对象没有加密!‘ --SQL Server 2012 --throw 50001, N‘无效的对象!要解密的对象必须是函数,存储过程,视图或触发器。‘,1 return end declare @sql nvarchar(max) --解密出来的SQL语句 ,@imageval nvarchar(max) --加密字符串 ,@tmpStr nvarchar(max) --临时SQL语句 ,@tmpStr_imageval nvarchar(max) --临时SQL语句(加密后) ,@type char(2) --对象类型(‘P‘,‘V‘,‘TR‘,‘FN‘,‘IF‘,‘TF‘) ,@objectID int --对象ID ,@i int --While循环使用 ,@Oject1 nvarchar(1000) set @objectID=object_id(@Object) set @type=(select a.type from sys.objects a where a.object_id=@objectID) declare @Space4000 nchar(4000) set @Space4000=replicate(‘-‘,4000) /* @tmpStr 会构造下面的SQL语句 ------------------------------------------------------------------------------- alter trigger Tr_Name on Table_Name with encryption for update as return /**/ alter proc Proc_Name with encryption as select 1 as col /**/ alter view View_Name with encryption as select 1 as col /**/ alter function Fn_Name() returns int with encryption as begin return(0) end/**/ */ set @Oject1=quotename(object_schema_name(@objectID))+‘.‘+quotename(@Object) set @tmpStr= case when @type =‘P ‘ then N‘Alter Procedure ‘+@Oject1+‘ with encryption as select 1 as column1 ‘ when @type =‘V ‘ then N‘Alter View ‘+@Oject1+‘ with encryption as select 1 as column1 ‘ when @type =‘FN‘ then N‘Alter Function ‘+@Oject1+‘() returns int with encryption as begin return(0) end ‘ when @type =‘IF‘ then N‘Alter Function ‘+@Oject1+‘() returns table with encryption as return(Select a.name from sys.types a) ‘ when @type =‘TF‘ then N‘Alter Function ‘+@Oject1+‘() returns @t table(name nvarchar(50)) with encryption as begin return end ‘ else ‘Alter Trigger ‘+@Oject1+‘on ‘+quotename(object_schema_name(@objectID))+‘.‘+(select Top(1) quotename(object_name(parent_id)) from sys.triggers a where a.object_id=@objectID)+‘ with encryption for update as return ‘ end set @tmpStr=@tmpStr+‘/*‘+@Space4000 set @i=0 while @i < (ceiling(@MaxLength*1.0/4000)-1) begin set @tmpStr=@tmpStr+ @Space4000 Set @i=@i+1 end set @tmpStr=@tmpStr+‘*/‘ ------------ set @imageval =(select top(1) a.imageval from sys.sysobjvalues a where a.objid=@objectID and a.valclass=1) begin tran exec(@tmpStr) set @tmpStr_imageval =(select top(1) a.imageval from sys.sysobjvalues a where a.objid=@objectID and a.valclass=1) rollback tran ------------- set @tmpStr=stuff(@tmpStr,1,5,‘create‘) set @sql=‘‘ set @i=1 while @i<= (datalength(@imageval)/2) begin set @sql=@sql+isnull(nchar(unicode(substring(@tmpStr,@i,1)) ^ unicode(substring(@tmpStr_imageval,@i,1))^unicode(substring(@imageval,@i,1)) ),‘‘) Set @i+=1 end /* 2. 列印 */ declare @patindex int while @sql>‘‘ begin set @patindex=patindex(‘%‘+char(13)+char(10)+‘%‘,@sql) if @patindex >0 begin print substring(@sql,1,@patindex-1) set @sql=stuff(@sql,1,@patindex+1,‘‘) end else begin set @patindex=patindex(‘%‘+char(13)+‘%‘,@sql) if @patindex >0 begin print substring(@sql,1,@patindex-1) set @sql=stuff(@sql,1,@patindex,‘‘) end else begin set @patindex=patindex(‘%‘+char(10)+‘%‘,@sql) if @patindex >0 begin print substring(@sql,1,@patindex-1) set @sql=stuff(@sql,1,@patindex,‘‘) end else begin print @sql set @sql=‘‘ end end end end Go exec sp_ms_marksystemobject ‘sp_DecryptObject‘ --标识为系统对象 go
时间: 2024-10-10 17:34:28