如何判断是在 Linux 控制台中运行还是在 ssh 会话中运行？

Question

我有一个应用程序，如果直接从 Linux 控制台运行，则需要有不同的行为。因此，如果用户使用 SSH 连接来运行 FooBar，或者用户走到控制台并直接登录来运行 FooBar，我希望它做一些不同的事情。

我需要调用什么 C API 来区分这两种情况？我想我必须查看“tty/pts”信息（例如我运行“ps axf”时看到的信息），但我不确定这是否是最好的解决方案，也不确定要调用什么 API获取该信息。

提示表示赞赏。 :)

Answer 1

根据您对被欺骗的担心程度，一个简单的检查是检查是否存在 SSH_CLIENT 和 SSH_CONNECTION 环境变量，在这种情况下您需要使用 getenv 函数。

Answer 2

根据您的标准输入检查 ttyname(3) 的返回值应该会给出为进程提供输入的终端的名称。

如果程序在控制台上运行（并且没有对其输入进行重定向），则它将是 /dev/console。您还可以检查 stdout 以查看它是否连接到 /dev/console - 看看哪个更适合您的使用场景。

Answer 3

ttyname 将告诉您连接到给定文件描述符的终端的名称；例如，ttyname(0) 会告诉您标准输入的终端。

如果输入或输出被重定向，这当然会失败。

除此之外，您可以检查各种环境变量（SSH_CONNECTION、SSH_CLIENT、REMOTEHOST、DISPLAY、SESSIONNAME ）。 Wireshark 具有检测其是否远程运行的逻辑，以便它不会捕获其生成的网络流量；您可能对其 get_conn_cfilter 函数使用的逻辑感兴趣实施这一点。

Answer 4

我将环境变量视为正在发生的事情的合理标志。我不确定您需要什么 C API，但我确信存在一个。

例如，无论使用什么 SSH 客户端，我的计算机上都会设置 SSH_CLIENT 或 SSH_CONNECTION 环境变量。

可能值得检查一下这些基于计算机上运行的 SSH 服务器的通用性。

Answer 5

丑陋，但有效：

// don't force scripts to pause!
if (_isatty(_fileno(stdout)))
{
    _tprintf(_T("\nPress enter to continue..."));
    _gettc(stdin);
}