Microsoft MVP성태의 닷넷 이야기
.NET Framework: 1033. C# - .NET 4.0 이하에서 Console.IsInputRedirected 구현 [링크 복사], [링크+제목 복사],
조회: 18659
글쓴 사람
정성태 (techsharer at outlook.com)
홈페이지
첨부 파일
 
(연관된 글이 1개 있습니다.)

C# - .NET 4.0 이하에서 Console.IsInputRedirected 구현

예전에 netcat 프로그램을 만들면서,

C# - 간단하게 만들어 보는 리눅스의 nc(netcat) 프로그램
; https://www.sysnet.pe.kr/2/0/12311

pipeline으로 연결된 입력을 전달받기 위해 Console.IsInputRedirected 속성을 사용했습니다.

Console.IsInputRedirected Property
; https://learn.microsoft.com/en-us/dotnet/api/system.console.isinputredirected

이 속성은 .NET 4.5부터 제공되므로 그 이하의 버전에서는 사용할 수 없습니다. 대신 .NET 4.5의 IsInputRedirected 구현 코드를 가져와 쓸 수는 있기 때문에 역어셈블을 통해 다음과 같이 알아낼 수 있습니다.

[DllImport("kernel32.dll")]
internal static extern int GetFileType(SafeFileHandle handle);

[DllImport("kernel32.dll", SetLastError = true)]
internal static extern bool GetConsoleMode(IntPtr hConsoleHandle, out int mode);

[SecuritySafeCritical]
private static bool IsHandleRedirected(IntPtr ioHandle)
{
    SafeFileHandle handle = new SafeFileHandle(ioHandle, ownsHandle: false);
    int fileType = Win32Native.GetFileType(handle);
    if ((fileType & 2) != 2)
    {
        return true;
    }

    int mode;
    bool consoleMode = Win32Native.GetConsoleMode(ioHandle, out mode);
    return !consoleMode;
}

public static bool IsInputRedirected
{
    [SecuritySafeCritical]
    get
    {
        // ...[생략]...
        return IsHandleRedirected(ConsoleInputHandle);
    }
}

private static IntPtr ConsoleInputHandle
{
    [SecurityCritical]
    get
    {
        if (_consoleInputHandle == IntPtr.Zero)
        {
            _consoleInputHandle = Win32Native.GetStdHandle(-10);
        }

        return _consoleInputHandle;
    }
}

끝났군요, ^^ 그럼 위의 코드를 적당히 재구성해서 다음과 같이 만들 수 있습니다.

using Microsoft.Win32.SafeHandles;
using System;
using System.Runtime.InteropServices;
using System.Security;

class Program
{
    static void Main(string[] args)
    {
        // .NET 4.5 or later
        Console.WriteLine(Console.IsInputRedirected);

        // .NET 4.0 or below
        Console.WriteLine(ConsoleHelper.IsInputHandleRedirected());
    }
}

// Utilities/netcat/ConsoleHelper.cs 
// https://github.com/stjeong/Utilities/blob/master/netcat/ConsoleHelper.cs
class ConsoleHelper
{
    [DllImport("kernel32.dll")]
    internal static extern int GetFileType(SafeFileHandle handle);

    [DllImport("kernel32.dll")]
    private static extern IntPtr GetStdHandle(StdHandle std);

    [DllImport("kernel32.dll", SetLastError = true)]
    internal static extern bool GetConsoleMode(IntPtr hConsoleHandle, out int mode);

    private enum StdHandle { Stdin = -10, Stdout = -11, Stderr = -12 };

    [SecuritySafeCritical]
    public static bool IsInputHandleRedirected()
    {
        IntPtr ioHandle = GetStdHandle(StdHandle.Stdin);

        SafeFileHandle handle = new SafeFileHandle(ioHandle, ownsHandle: false);
        int fileType = GetFileType(handle);
        if ((fileType & 2) != 2)
        {
            return true;
        }

        bool consoleMode = GetConsoleMode(ioHandle, out _);
        return !consoleMode;
    }
}

/* 출력 결과
C:\temp> dir . | ConsoleApp1.exe
True
True

C:\temp> ConsoleApp1.exe
False
False
*/




// Conventional wisdom is retarded, aka What the @#%&* is _O_U16TEXT?
// https://archives.miloush.net/michkap/archive/2008/03/18/8306597.html

#include <iostream>
#include <Windows.h>

BOOL IsRedirected(HANDLE handle)
{
    auto FileType = GetFileType(handle);

    if ((FileType == FILE_TYPE_UNKNOWN) && (GetLastError() != ERROR_SUCCESS)) 
    {
        return TRUE;
    }

    BOOL ConsoleOutput;
    DWORD ConsoleMode;
    FileType &= ~(FILE_TYPE_REMOTE);

    if (FileType == FILE_TYPE_CHAR) 
    {
        auto Result = GetConsoleMode(handle, &ConsoleMode);

        if ((Result == FALSE) && (GetLastError() == ERROR_INVALID_HANDLE)) 
        {
            return TRUE;
        }
        else 
        {
            return FALSE;
        }
    }

    return TRUE;
}

int main()
{
    HANDLE stdOut = GetStdHandle(STD_OUTPUT_HANDLE);

    printf("is redirected == %d\n", IsRedirected(stdOut));

    (void)setvbuf(stdout, NULL, _IONBF, 0);
    // 또는, 
    // fflush(stdout); 

    CloseHandle(stdOut);
}




[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[연관 글]






[최초 등록일: ]
[최종 수정일: 10/28/2024]

Creative Commons License
이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.
by SeongTae Jeong, mailto:techsharer at outlook.com

비밀번호

댓글 작성자
 



2023-01-26 02시12분
Input/Output Redirection in the Shell
; https://thoughtbot.com/blog/input-output-redirection-in-the-shell

------------------------------

debugging with visual studio using redirected standard input
; https://stackoverflow.com/questions/14043986/debugging-with-visual-studio-using-redirected-standard-input
정성태

... 151  152  153  154  155  [156]  157  158  159  160  161  162  163  164  165  ...
NoWriterDateCnt.TitleFile(s)
1186정성태12/1/201134269.NET Framework: 279. WPF - 그리기 성능 및 Blurring 문제파일 다운로드1
1185정성태11/29/201124425.NET Framework: 278. WPF - Content의 Changed 이벤트에 해당하는게 뭔가요?파일 다운로드1
1184정성태11/29/201127666.NET Framework: 277. F#과 WPF가 어울리지 못하는 근본적인 이유 [2]
1183정성태11/26/201122887오류 유형: 140. Visual Studio 2010 - Floating된 에디트 윈도우가 사라지지 않는 경우 [2]
1182정성태11/25/201158897.NET Framework: 276. 중복 없는 숫자를 랜덤으로 배열하는 방법 [5]파일 다운로드1
1181정성태11/24/201129179디버깅 기술: 44. windbg의 mscordacwks DLL 로드 문제
1180정성태11/23/201139037.NET Framework: 275. 레지스트리 등록 및 Interop DLL 없이 COM 개체 사용하는 방법 [2]파일 다운로드1
1179정성태11/22/201129557.NET Framework: 274. ReaderWriterLockSlim은 언제 쓰는 걸까요? [4]파일 다운로드1
1178정성태11/19/201126132.NET Framework: 273. 설치된 .NET 버전에 민감한 코드를 포함하는 경우, 다중으로 어셈블리를 만들어야 할까요?파일 다운로드1
1177정성태11/18/201131434.NET Framework: 272. 소켓 연결 시간 제한 - 두 번째 이야기 [1]파일 다운로드1
1176정성태11/17/201130825.NET Framework: 271. C#에서 확인해 보는 관리 힙의 인스턴스 구조 [3]파일 다운로드1
1175정성태11/16/201128795.NET Framework: 270. .NET 참조 개체 인스턴스의 Object Header를 확인하는 방법 [1]파일 다운로드1
1174정성태11/15/201128142.NET Framework: 269. 일반 참조형의 기본 메모리 소비는 얼마나 될까요? [4]
1173정성태11/14/201124264.NET Framework: 268. .NET Array는 왜 12bytes의 기본 메모리를 점유할까? [1]
1172정성태11/13/201121206.NET Framework: 267. windbg - GC Heap에서 .NET 타입에 대한 배열을 찾는 방법
1171정성태11/12/201137946.NET Framework: 266. StringBuilder에서의 OutOfMemoryException 오류 원인 분석 [4]파일 다운로드1
1170정성태11/10/201127336.NET Framework: 265. Named 동기화 개체 생성 시 System.UnauthorizedAccessException 예외 발생하는 경우
1169정성태11/10/201130829.NET Framework: 264. 다중 LAN 카드 환경에서 Dns.GetHostAddresses(local)가 반환해 주는 IP의 우선순위는 어떻게 될까요? [4]
1168정성태11/6/201126747오류 유형: 139. TlbImp : error TI0000 : A single valid machine type compatible with the input type library must be specified
1167정성태11/5/201138540개발 환경 구성: 133. Registry 등록 과정 없이 COM 개체 사용 - 두 번째 이야기 [5]파일 다운로드4
1166정성태11/5/201124572.NET Framework: 263. byte[] pData = new byte[100000]로 인한 성능 차이? [1]파일 다운로드1
1165정성태11/3/201129637개발 환경 구성: 132. "Visual Studio Command Prompt (2010)" 명령행에서 2.0 버전의 MSBuild를 구동하는 방법 [2]파일 다운로드1
1164정성태11/1/201127734.NET Framework: 262. .NET 스레드 콜 스택 덤프 (4) - .NET 4.0을 지원하지 않는 MSE 응용 프로그램 원인 분석
1163정성태10/31/201127343.NET Framework: 261. .NET 스레드 콜 스택 덤프 (3) - MSE 소스 코드 개선파일 다운로드1
1162정성태10/30/201127407.NET Framework: 260. .NET 스레드 콜 스택 덤프 (2) - Managed Stack Explorer 소스 코드를 이용한 스택 덤프 구하는 방법파일 다운로드1
1161정성태10/29/201124130.NET Framework: 259. Type.GetMethod - System.Reflection.AmbiguousMatchException파일 다운로드1
... 151  152  153  154  155  [156]  157  158  159  160  161  162  163  164  165  ...