J & J - 정성태의 닷넷 이야기

사용자

메뉴

최근 덧글

[정성태] 그냥 RSS Reader 기능과 약간의 UI 편의성 때문에 사용...[이종효] 오래된 소프트웨어는 보안 위협이 되기도 합니다. 혹시 어떤 기능...[정성태] @Keystroke IEEE의 문서를 소개해 주시다니... +_...[손민수 (Keystroke)] 괜히 듀얼채널 구성할 때 한번에 같은 제품 사라고 하는 것이 아...[정성태] 전각(Full-width)/반각(Half-width) 기능을 토...[정성태] Vector에 대한 내용은 없습니다. Vector가 닷넷 BCL...[orion] 글 읽고 찾아보니 디자인 타임에는 InitializeCompon...[orion] 연휴 전에 재현 프로젝트 올리자 생각해 놓고 여의치 않아서 못 ...[정성태] 아래의 글에 정리했으니 참고하세요. C# - Typed D...[정성태] 간단한 재현 프로젝트라도 있을까요? 저런 식으로 설명만 해...

글쓰기

제목

이름

암호

전자우편

HTML

홈페이지

유형

내용

<div style='display: inline'>
<h1 style='font-family: Malgun Gothic, Consolas; font-size: 20pt; color: #006699; text-align: center; font-weight: bold'>C# - .NET 4.0 이하에서 Console.IsInputRedirected 구현</h1>

예전에 netcat 프로그램을 만들면서, 
 
<pre style='margin: 10px 0px 10px 10px; padding: 10px 0px 10px 10px; background-color: #fbedbb; overflow: auto; font-family: Consolas, Verdana;' >
C# - 간단하게 만들어 보는 리눅스의 nc(netcat) 프로그램
; <a target='tab' href='https://www.sysnet.pe.kr/2/0/12311'>https://www.sysnet.pe.kr/2/0/12311</a>
</pre>
 
pipeline으로 연결된 입력을 전달받기 위해 Console.IsInputRedirected 속성을 사용했습니다. 
 
<pre style='margin: 10px 0px 10px 10px; padding: 10px 0px 10px 10px; background-color: #fbedbb; overflow: auto; font-family: Consolas, Verdana;' >
Console.IsInputRedirected Property
; <a target='tab' href='https://learn.microsoft.com/en-us/dotnet/api/system.console.isinputredirected'>https://learn.microsoft.com/en-us/dotnet/api/system.console.isinputredirected</a>
</pre>
 
이 속성은 .NET 4.5부터 제공되므로 그 이하의 버전에서는 사용할 수 없습니다. 대신 .NET 4.5의 IsInputRedirected 구현 코드를 가져와 쓸 수는 있기 때문에 역어셈블을 통해 다음과 같이 알아낼 수 있습니다. 
 
<pre style='margin: 10px 0px 10px 10px; padding: 10px 0px 10px 10px; background-color: #fbedbb; overflow: auto; font-family: Consolas, Verdana;' >
[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 &amp; 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;
 }
}
</pre>
 
끝났군요, ^^ 그럼 위의 코드를 적당히 재구성해서 다음과 같이 만들 수 있습니다. 
 
<pre style='margin: 10px 0px 10px 10px; padding: 10px 0px 10px 10px; background-color: #fbedbb; overflow: auto; font-family: Consolas, Verdana;' >
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 
// <a target='tab' href='https://github.com/stjeong/Utilities/blob/master/netcat/ConsoleHelper.cs'>https://github.com/stjeong/Utilities/blob/master/netcat/ConsoleHelper.cs</a>
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 <a target='tab' href='https://stackoverflow.com/questions/3453220/how-to-detect-if-console-in-stdin-has-been-redirected'>StdHandle</a> { 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 &amp; 2) != 2)
        {
            return true;
        }

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

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

C:\temp&gt; ConsoleApp1.exe
False
False
*/
</pre>
 
<hr style='width: 50%' /> 
 
<pre style='margin: 10px 0px 10px 10px; padding: 10px 0px 10px 10px; background-color: #fbedbb; overflow: auto; font-family: Consolas, Verdana;' >
// Conventional wisdom is retarded, aka What the @#%&* is _O_U16TEXT?
// <a target='tab' href='https://archives.miloush.net/michkap/archive/2008/03/18/8306597.html'>https://archives.miloush.net/michkap/archive/2008/03/18/8306597.html</a>

#include &lt;iostream&gt;
#include &lt;Windows.h&gt;

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

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

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

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

if ((Result == FALSE) &amp;&amp; (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);
}
</pre>

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

</div>

첨부파일

스팸 방지용 인증 번호

2071 (왼쪽의 숫자를 입력해야 합니다.)