Changeset 68836 in vbox for trunk/src/VBox/Runtime/r3

Timestamp:

Sep 22, 2017 4:37:46 PM (7 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

118090

Message:

IPRT: Added RTStrmIsTerminal and RTStrmQueryTerminalWidth.

File:

• trunk/src/VBox/Runtime/r3/stream.cpp (modified) (2 diffs)

trunk/src/VBox/Runtime/r3/stream.cpp

@@ -60 +60 @@
 #ifdef RT_OS_WINDOWS
 # include <iprt/win/windows.h>
-#endif
-#ifndef RT_OS_WINDOWS
+#else
 # include <termios.h>
 # include <unistd.h>
@@ -553 +552 @@
     return rc;
 }
+
+
+RTR3DECL(bool) RTStrmIsTerminal(PRTSTREAM pStream)
+{
+    AssertPtrReturn(pStream, false);
+    AssertReturn(pStream->u32Magic == RTSTREAM_MAGIC, false);
+
+    if (pStream->pFile)
+    {
+        int fh = fileno(pStream->pFile);
+        if (isatty(fh))
+        {
+#ifdef RT_OS_WINDOWS
+            DWORD  dwMode;
+            HANDLE hCon = (HANDLE)_get_osfhandle(fh);
+            if (GetConsoleMode(hCon, &dwMode))
+                return true;
+#else
+            return true;
+#endif
+        }
+    }
+    return false;
+}
+
+
+RTR3DECL(int) RTStrmQueryTerminalWidth(PRTSTREAM pStream, uint32_t *pcchWidth)
+{
+    AssertPtrReturn(pcchWidth, VERR_INVALID_HANDLE);
+    *pcchWidth = 80;
+
+    AssertPtrReturn(pStream, VERR_INVALID_HANDLE);
+    AssertReturn(pStream->u32Magic == RTSTREAM_MAGIC, VERR_INVALID_HANDLE);
+
+    if (pStream->pFile)
+    {
+        int fh = fileno(pStream->pFile);
+        if (isatty(fh))
+        {
+#ifdef RT_OS_WINDOWS
+            CONSOLE_SCREEN_BUFFER_INFO Info;
+            HANDLE hCon = (HANDLE)_get_osfhandle(fh);
+            RT_ZERO(Info);
+            if (GetConsoleScreenBufferInfo(hCon, &Info))
+            {
+                *pcchWidth = Info.dwSize.X ? Info.dwSize.X : 80;
+                return VINF_SUCCESS;
+            }
+            return RTErrConvertFromWin32(GetLastError());
+#else
+            struct winsize Info;
+            RT_ZERO(Info);
+            int rc = ioctl(fh, TIOCGWINSZ, &Info);
+            if (rc >= 0)
+            {
+                *pcchWidth = Info.ws_row ? Info.ws_row : 80;
+                return VINF_SUCCESS;
+            }
+            return RTErrConvertFromErrno(errno);
+#endif
+        }
+    }
+    return VERR_INVALID_FUNCTION;
+}
+