如何判断存储过程是否被其他存储过程调用？事务内调用校验咨询

Hey there! Let's break down your questions one by one:

1. 能否判断一个存储过程是否由另一个存储过程调用？ #

Absolutely! SQL Server提供了内置工具来追踪执行中存储过程的调用栈。你可以用sys.dm_exec_callstack这类系统动态管理视图(DMV)，或者结合sys.dm_exec_sql_text和会话信息来识别调用者。需要注意的是，部分方法需要特定权限（比如VIEW SERVER STATE），且在不同SQL Server版本中行为可能略有差异。

2. 你当前用@@TRANCOUNT的校验方式是否正确？ #

直白说：这种方法不可靠。原因如下：
你的逻辑假设只要存在活跃事务（@@TRANCOUNT > 0），调用就一定来自父存储过程。但如果有人手动开启事务后直接调用子存储过程，你的校验会错误地允许这种操作——它无法区分事务是父存储过程开启的，还是其他任何主动启动的事务。

3. 更优的限制方法 #

这里有几种可靠的方案，能确保子存储过程仅被指定父存储过程调用：

方案1：检查调用栈（最精准） #

用sys.dm_exec_callstack获取直接调用者的名称，在子存储过程开头加入这段代码：

DECLARE @callerObjectName NVARCHAR(128);

SELECT @callerObjectName = OBJECT_NAME(esp.objectid)
FROM sys.dm_exec_callstack esp
WHERE esp.depth = 1; -- depth=0是当前存储过程，depth=1是直接调用者

-- 替换成你的父存储过程真实名称
IF @callerObjectName <> 'YourParentProcedureName'
BEGIN
    RAISERROR('此存储过程仅允许由指定父存储过程调用！', 16, 1);
    RETURN;
END

注意：这个方法需要VIEW SERVER STATE权限。如果没有该权限，可以用sys.dm_exec_input_buffer作为备选：

DECLARE @sessionId INT = @@SPID;
DECLARE @inputBuffer NVARCHAR(4000);

SELECT @inputBuffer = event_info
FROM sys.dm_exec_input_buffer(@sessionId, NULL);

-- 检查输入缓冲区是否包含父存储过程的调用语句
IF CHARINDEX('EXEC YourParentProcedureName', @inputBuffer) = 0
BEGIN
    RAISERROR('此存储过程仅允许由指定父存储过程调用！', 16, 1);
    RETURN;
END

这个备选方案依赖字符串匹配，精准度稍低，但无需提升权限。

方案2：传递专属"令牌"参数 #

给子存储过程添加一个只有父存储过程知晓的隐藏参数：

• 父存储过程调用方式：

  EXEC ChildProcedure @AuthorizationToken = 'OnlyParentKnowsThisSecret123';
  

• 子存储过程校验逻辑：

  CREATE PROCEDURE ChildProcedure
      @AuthorizationToken NVARCHAR(100) = NULL
  AS
  BEGIN
      IF @AuthorizationToken <> 'OnlyParentKnowsThisSecret123'
      BEGIN
          RAISERROR('此存储过程仅允许由指定父存储过程调用！', 16, 1);
          RETURN;
      END
  
      -- 子存储过程业务逻辑
  END
  

这种方法简单易懂，无需特殊权限，在内部环境中足够安全。唯一的小缺点是如果令牌泄露，可能被伪造调用，但在可控系统中风险极低。

方案3：限制执行权限 #

如果你的SQL Server版本支持，可以配置权限，让只有父存储过程的执行上下文能调用子存储过程。比如：

• 创建证书或非对称密钥，用它给父存储过程签名，然后将子存储过程的EXECUTE权限授予证书对应的用户。这样只有来自签名父存储过程的调用才会被允许。
  这是最安全的方案，但需要对SQL Server安全特性有一定了解，配置步骤也更繁琐。

总结 #

你原本用@@TRANCOUNT的校验逻辑范围太宽，容易被绕过。大多数场景下，调用栈检查（有权限时）或专属令牌参数这两种方案，就能满足你可靠限制调用来源的需求。

内容的提问来源于stack exchange，提问作者Ivan-Mark Debono