Microsoft MVP성태의 닷넷 이야기
.NET Framework: 1033. C# - .NET 4.0 이하에서 Console.IsInputRedirected 구현 [링크 복사], [링크+제목 복사],
조회: 9358
글쓴 사람
정성태 (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
*/




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

[연관 글]






[최초 등록일: ]
[최종 수정일: 12/22/2022]

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

비밀번호

댓글 작성자
 



2023-01-26 02시12분
정성태

... [61]  62  63  64  65  66  67  68  69  70  71  72  73  74  75  ...
NoWriterDateCnt.TitleFile(s)
12122정성태1/20/20209166.NET Framework: 879. C/C++의 UNREFERENCED_PARAMETER 매크로를 C#에서 우회하는 방법(IDE0060 - Remove unused parameter '...')파일 다운로드1
12121정성태1/20/20209750VS.NET IDE: 139. Visual Studio - Error List: "Could not find schema information for the ..."파일 다운로드1
12120정성태1/19/202011176.NET Framework: 878. C# DLL에서 Win32 C/C++처럼 dllexport 함수를 제공하는 방법 - 네 번째 이야기(IL 코드로 직접 구현)파일 다운로드1
12119정성태1/17/202011210디버깅 기술: 160. Windbg 확장 DLL 만들기 (3) - C#으로 만드는 방법
12118정성태1/17/202011870개발 환경 구성: 466. C# DLL에서 Win32 C/C++처럼 dllexport 함수를 제공하는 방법 - 세 번째 이야기 [1]
12117정성태1/15/202010872디버깅 기술: 159. C# - 디버깅 중인 프로세스를 강제로 다른 디버거에서 연결하는 방법파일 다운로드1
12116정성태1/15/202011358디버깅 기술: 158. Visual Studio로 디버깅 시 sos.dll 확장 명령어를 (비롯한 windbg의 다양한 기능을) 수행하는 방법
12115정성태1/14/202011095디버깅 기술: 157. C# - PEB.ProcessHeap을 이용해 디버깅 중인지 확인하는 방법파일 다운로드1
12114정성태1/13/202012975디버깅 기술: 156. C# - PDB 파일로부터 심벌(Symbol) 및 타입(Type) 정보 열거 [1]파일 다운로드3
12113정성태1/12/202013584오류 유형: 590. Visual C++ 빌드 오류 - fatal error LNK1104: cannot open file 'atls.lib' [1]
12112정성태1/12/202010123오류 유형: 589. PowerShell - 원격 Invoke-Command 실행 시 "WinRM cannot complete the operation" 오류 발생
12111정성태1/12/202013436디버깅 기술: 155. C# - KernelMemoryIO 드라이버를 이용해 실행 프로그램을 숨기는 방법(DKOM: Direct Kernel Object Modification) [16]파일 다운로드1
12110정성태1/11/202012013디버깅 기술: 154. Patch Guard로 인해 블루 스크린(BSOD)가 발생하는 사례 [5]파일 다운로드1
12109정성태1/10/20209918오류 유형: 588. Driver 프로젝트 빌드 오류 - Inf2Cat error -2: "Inf2Cat, signability test failed."
12108정성태1/10/20209977오류 유형: 587. Kernel Driver 시작 시 127(The specified procedure could not be found.) 오류 메시지 발생
12107정성태1/10/202010908.NET Framework: 877. C# - 프로세스의 모든 핸들을 열람 - 두 번째 이야기
12106정성태1/8/202012395VC++: 136. C++ - OSR Driver Loader와 같은 Legacy 커널 드라이버 설치 프로그램 제작 [1]
12105정성태1/8/202011010디버깅 기술: 153. C# - PEB를 조작해 로드된 DLL을 숨기는 방법
12104정성태1/7/202011678DDK: 9. 커널 메모리를 읽고 쓰는 NT Legacy driver와 C# 클라이언트 프로그램 [4]
12103정성태1/7/202014448DDK: 8. Visual Studio 2019 + WDK Legacy Driver 제작- Hello World 예제 [1]파일 다운로드2
12102정성태1/6/202011991디버깅 기술: 152. User 권한(Ring 3)의 프로그램에서 _ETHREAD 주소(및 커널 메모리를 읽을 수 있다면 _EPROCESS 주소) 구하는 방법
12101정성태1/5/202011336.NET Framework: 876. C# - PEB(Process Environment Block)를 통해 로드된 모듈 목록 열람
12100정성태1/3/20209358.NET Framework: 875. .NET 3.5 이하에서 IntPtr.Add 사용
12099정성태1/3/202011683디버깅 기술: 151. Windows 10 - Process Explorer로 확인한 Handle 정보를 windbg에서 조회 [1]
12098정성태1/2/202011274.NET Framework: 874. C# - 커널 구조체의 Offset 값을 하드 코딩하지 않고 사용하는 방법 [3]
12097정성태1/2/20209822디버깅 기술: 150. windbg - Wow64, x86, x64에서의 커널 구조체(예: TEB) 구조체 확인
... [61]  62  63  64  65  66  67  68  69  70  71  72  73  74  75  ...