11

.NETConsoleクラスとそのデフォルトのTextWriter実装 (Console.Out例 として暗黙的に利用可能Console.WriteLine()) は、アプリケーションがその出力を別のプログラムにパイプしているときにエラーを通知せず、アプリケーションが終了する前に別のプログラムがパイプを終了または閉じます。これは、アプリケーションが必要以上に長く実行され、出力がブラック ホールに書き込まれる可能性があることを意味します。

リダイレクト パイプのもう一方の端が閉じていることを検出するにはどうすればよいですか?

より詳細な説明は次のとおりです。

この問題を示すサンプル プログラムのペアを次に示します。Produce計算の効果をシミュレートするために、かなりゆっくりと多くの整数を出力します。

using System;
class Produce
{
    static void Main()
    {
        for (int i = 0; i < 10000; ++i)
        {
            System.Threading.Thread.Sleep(100); // added for effect
            Console.WriteLine(i);
        }
    }
}

Consume入力の最初の 10 行のみを読み取り、終了します。

using System;
class Consume
{
    static void Main()
    {
        for (int i = 0; i < 10; ++i)
            Console.ReadLine();
    }
}

これら 2 つのプログラムがコンパイルされ、最初の出力が 2 番目のプログラムにパイプされた場合、次のようになります。

Produce | Consume

Produce... が終了した後もずっと実行し続けることが観察できConsumeます。

実際には、私のConsumeプログラムは Unix スタイルheadであり、私のProduceプログラムは計算にコストのかかるデータを出力します。パイプのもう一方の端が接続を閉じたときに出力を終了したいと思います。

.NET でこれを行うにはどうすればよいですか?

(明らかな代替手段は、コマンドライン引数を渡して出力を制限することであることを知っています。実際、それは私が現在行っていることですが、もっと多くのことをできるようにしたいので、これを行う方法を知りたいですいつ読み取りを終了するかについての構成可能な判断 (例:grep前にパイプを通すhead)。

更新: .NETの実装は、エラー 0x6D ( ) および 0xE8 ( )System.IO.__ConsoleStreamを無視するようにハードコードされているようです。これはおそらく、コンソール ストリームを再実装する必要があることを意味します。はぁ...)ERROR_BROKEN_PIPEERROR_NO_DATA

4

2 に答える 2

7

これを解決するには、Win32 ファイル ハンドルを介して独自の基本的なストリーム実装を作成する必要がありました。非同期サポート、バッファリング、またはシークを実装する必要がなかったので、これはそれほど難しくありませんでした。

残念ながら、安全でないコードを使用する必要がありますが、ローカルで完全に信頼して実行されるコンソール アプリケーションでは、通常、これは問題になりません。

コア ストリームは次のとおりです。

class HandleStream : Stream
{
    SafeHandle _handle;
    FileAccess _access;
    bool _eof;

    public HandleStream(SafeHandle handle, FileAccess access)
    {
        _handle = handle;
        _access = access;
    }

    public override bool CanRead
    {
        get { return (_access & FileAccess.Read) != 0; }
    }

    public override bool CanSeek
    {
        get { return false; }
    }

    public override bool CanWrite
    {
        get { return (_access & FileAccess.Write) != 0; }
    }

    public override void Flush()
    {
        // use external buffering if you need it.
    }

    public override long Length
    {
        get { throw new NotSupportedException(); }
    }

    public override long Position
    {
        get { throw new NotSupportedException(); }
        set { throw new NotSupportedException(); }
    }

    static void CheckRange(byte[] buffer, int offset, int count)
    {
        if (offset < 0 || count < 0 || (offset + count) < 0
            || (offset + count) > buffer.Length)
            throw new ArgumentOutOfRangeException();
    }

    public bool EndOfStream
    {
        get { return _eof; }
    }

    public override int Read(byte[] buffer, int offset, int count)
    {
        CheckRange(buffer, offset, count);
        int result = ReadFileNative(_handle, buffer, offset, count);
        _eof |= result == 0;
        return result;
    }

    public override void Write(byte[] buffer, int offset, int count)
    {
        int notUsed;
        Write(buffer, offset, count, out notUsed);
    }

    public void Write(byte[] buffer, int offset, int count, out int written)
    {
        CheckRange(buffer, offset, count);
        int result = WriteFileNative(_handle, buffer, offset, count);
        _eof |= result == 0;
        written = result;
    }

    public override long Seek(long offset, SeekOrigin origin)
    {
        throw new NotSupportedException();
    }

    public override void SetLength(long value)
    {
        throw new NotSupportedException();
    }

    [return: MarshalAs(UnmanagedType.Bool)]
    [DllImport("kernel32", SetLastError=true)]
    static extern unsafe bool ReadFile(
        SafeHandle hFile, byte* lpBuffer, int nNumberOfBytesToRead,
        out int lpNumberOfBytesRead, IntPtr lpOverlapped);

    [return: MarshalAs(UnmanagedType.Bool)]
    [DllImport("kernel32.dll", SetLastError=true)]
    static extern unsafe bool WriteFile(
        SafeHandle hFile, byte* lpBuffer, int nNumberOfBytesToWrite, 
        out int lpNumberOfBytesWritten, IntPtr lpOverlapped);

    unsafe static int WriteFileNative(SafeHandle hFile, byte[] buffer, int offset, int count)
    {
        if (buffer.Length == 0)
            return 0;

        fixed (byte* bufAddr = &buffer[0])
        {
            int result;
            if (!WriteFile(hFile, bufAddr + offset, count, out result, IntPtr.Zero))
            {
                // Using Win32Exception just to get message resource from OS.
                Win32Exception ex = new Win32Exception(Marshal.GetLastWin32Error());
                int hr = ex.NativeErrorCode | unchecked((int) 0x80000000);
                throw new IOException(ex.Message, hr);
            }

            return result;
        }
    }

    unsafe static int ReadFileNative(SafeHandle hFile, byte[] buffer, int offset, int count)
    {
        if (buffer.Length == 0)
            return 0;

        fixed (byte* bufAddr = &buffer[0])
        {
            int result;
            if (!ReadFile(hFile, bufAddr + offset, count, out result, IntPtr.Zero))
            {
                Win32Exception ex = new Win32Exception(Marshal.GetLastWin32Error());
                int hr = ex.NativeErrorCode | unchecked((int) 0x80000000);
                throw new IOException(ex.Message, hr);
            }
            return result;
        }
    }
}

BufferedStream必要に応じてバッファリングのためにラップできますが、コンソール出力の場合、TextWriter文字レベルのバッファリングが行われ、改行のみがフラッシュされます。

ストリームは、自分自身Win32Exceptionを呼び出すのではなく、エラー メッセージを抽出するために悪用しますFormatMessage

このストリームに基づいて、コンソール I/O 用の単純なラッパーを作成できました。

static class ConsoleStreams
{
    enum StdHandle
    {
        Input = -10,
        Output = -11,
        Error = -12,
    }

    [DllImport("kernel32.dll", SetLastError = true)]
    static extern IntPtr GetStdHandle(int nStdHandle);  

    static SafeHandle GetStdHandle(StdHandle h)
    {
        return new SafeFileHandle(GetStdHandle((int) h), true);
    }

    public static HandleStream OpenStandardInput()
    {
        return new HandleStream(GetStdHandle(StdHandle.Input), FileAccess.Read);
    }

    public static HandleStream OpenStandardOutput()
    {
        return new HandleStream(GetStdHandle(StdHandle.Output), FileAccess.Write);
    }

    public static HandleStream OpenStandardError()
    {
        return new HandleStream(GetStdHandle(StdHandle.Error), FileAccess.Write);
    }

    static TextReader _in;
    static StreamWriter _out;
    static StreamWriter _error;

    public static TextWriter Out
    {
        get
        {
            if (_out == null)
            {
                _out = new StreamWriter(OpenStandardOutput());
                _out.AutoFlush = true;
            }
            return _out;
        }
    }

    public static TextWriter Error
    {
        get
        {
            if (_error == null)
            {
                _error = new StreamWriter(OpenStandardError());
                _error.AutoFlush = true;
            }
            return _error;
        }
    }

    public static TextReader In
    {
        get
        {
            if (_in == null)
                _in = new StreamReader(OpenStandardInput());
            return _in;
        }
    }
}

最終的な結果は、パイプのもう一方の端が接続を終了した後にコンソール出力に書き込むと、次のメッセージで素晴らしい例外が発生することです。

パイプが閉じられています

最も外側のレベルで をキャッチして無視することIOExceptionで、うまくいくように見えます。

于 2009-01-23T15:24:40.453 に答える
1

ERROR_BROKEN_PIPE および ERROR_NO_DATA エラーの報告がなければ、__ConsoleStream は役に立たないことに同意します。なぜ彼らがこれを除外することにしたのか、私は興味があります。

フォローしたい人は、次のリンクをチェックして、かなり古いものの関連する__ConsoleStreamのリストを確認してください...

http://www.123aspx.com/Rotor/RotorSrc.aspx?rot=42958

于 2009-01-22T18:43:56.930 に答える