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

... 106  107  [108]  109  110  111  112  113  114  115  116  117  118  119  120  ...
NoWriterDateCnt.TitleFile(s)
11258정성태8/1/201720320.NET Framework: 668. 지연 서명된 DLL과 서명된 DLL의 차이점파일 다운로드1
11257정성태7/31/201719905.NET Framework: 667. bypassTrustedAppStrongNames 옵션 설명파일 다운로드1
11256정성태7/25/201721837디버깅 기술: 90. windbg의 lm 명령으로 보이지 않는 .NET 4.0 ClassLibrary를 명시적으로 로드하는 방법 [1]
11255정성태7/18/201724354디버깅 기술: 89. Win32 Debug CRT Heap Internals의 0xBAADF00D 표시 재현 [1]파일 다운로드3
11254정성태7/17/201720774개발 환경 구성: 322. "Visual Studio Emulator for Android" 에뮬레이터를 "Android Studio"와 함께 쓰는 방법
11253정성태7/17/201721417Math: 21. "Coding the Matrix" 문제 2.5.1 풀이 [1]파일 다운로드1
11252정성태7/13/201719156오류 유형: 411. RTVS 또는 PTVS 실행 시 Could not load type 'Microsoft.VisualStudio.InteractiveWindow.Shell.IVsInteractiveWindowFactory2'
11251정성태7/13/201718632디버깅 기술: 88. windbg 분석 - webengine4.dll의 MgdExplicitFlush에서 발생한 System.AccessViolationException의 crash 문제 (2)
11250정성태7/13/201722235디버깅 기술: 87. windbg 분석 - webengine4.dll의 MgdExplicitFlush에서 발생한 System.AccessViolationException의 crash 문제 [1]
11249정성태7/12/201719988오류 유형: 410. LoadLibrary("[...].dll") failed - The specified procedure could not be found.
11248정성태7/12/201726536오류 유형: 409. pip install pefile - 'cp949' codec can't decode byte 0xe2 in position 208687: illegal multibyte sequence
11247정성태7/12/201720876오류 유형: 408. SqlConnection 객체 생성 시 무한 대기 문제파일 다운로드1
11246정성태7/11/201718893VS.NET IDE: 118. Visual Studio - 다중 폴더에 포함된 파일들에 대한 "Copy to Output Directory"를 한 번에 설정하는 방법
11245정성태7/10/201724633개발 환경 구성: 321. Visual Studio Emulator for Android 소개 [2]
11244정성태7/10/201724817오류 유형: 407. Visual Studio에서 ASP.NET Core 실행할 때 dotnet.exe 프로세스의 -532462766 오류 발생 [1]
11243정성태7/10/201721593.NET Framework: 666. dotnet.exe - 윈도우 운영체제에서의 .NET Core 버전 찾기 규칙
11242정성태7/8/201721133제니퍼 .NET: 27. 제니퍼 닷넷 적용 사례 (7) - 노후된 스토리지 장비로 인한 웹 서비스 Hang (멈춤) 현상
11241정성태7/8/201719780오류 유형: 406. Xamarin 빌드 에러 XA5209, APT0000
11240정성태7/7/201723604.NET Framework: 665. ClickOnce를 웹 브라우저를 이용하지 않고 쿼리 문자열을 전달하면서 실행하는 방법 [3]파일 다운로드1
11239정성태7/6/201724266.NET Framework: 664. Protocol Handler - 웹 브라우저에서 데스크톱 응용 프로그램을 실행하는 방법 [5]파일 다운로드1
11238정성태7/6/201721787오류 유형: 405. NT 서비스 시작 시 "Error 1067: The process terminated unexpectedly." 오류 발생 [2]
11237정성태7/5/201723443.NET Framework: 663. C# - PDB 파일 경로를 PE 파일로부터 얻는 방법파일 다운로드1
11236정성태7/4/201727176.NET Framework: 662. C# - VHD/VHDX 가상 디스크를 마운트하지 않고 파일을 복사하는 방법파일 다운로드1
11235정성태6/29/201721432Math: 20. Matlab/Octave로 Gram-Schmidt 정규 직교 집합 구하는 방법
11234정성태6/29/201718921오류 유형: 404. SharePoint 2013 설치 과정에서 "The username is invalid The account must be a valid domain account" 오류 발생
11233정성태6/28/201718798오류 유형: 403. SharePoint Server 2013을 Windows Server 2016에 설치할 때 .NET 4.5 설치 오류 발생
... 106  107  [108]  109  110  111  112  113  114  115  116  117  118  119  120  ...