/*
 * Copyright (c) 2006-2011 Christian Plattner. All rights reserved.
 * Please refer to the LICENSE.txt for licensing details.
 */
package ch.ethz.ssh2.channel;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

/**
 * A StreamForwarder forwards data between two given streams.
 * If two StreamForwarder threads are used (one for each direction)
 * then one can be configured to shutdown the underlying channel/socket
 * if both threads have finished forwarding (EOF).
 *
 * @author Christian Plattner
 * @version 2.50, 03/15/10
 */
public class StreamForwarder extends Thread {
    OutputStream os;
    InputStream is;
    byte[] buffer = new byte[Channel.CHANNEL_BUFFER_SIZE];
    Channel c;
    StreamForwarder sibling;
    Socket s;
    String mode;

    StreamForwarder(Channel c, StreamForwarder sibling, Socket s, InputStream is, OutputStream os, String mode)
    throws IOException {
        this.is = is;
        this.os = os;
        this.mode = mode;
        this.c = c;
        this.sibling = sibling;
        this.s = s;
    }

    @Override
    public void run() {
        try {
            while (true) {
                int len = is.read(buffer);

                if (len <= 0)
                    break;

                os.write(buffer, 0, len);
                os.flush();
            }
        }
        catch (IOException e) {
            try {
                c.cm.closeChannel(c, e, true);
            }
            catch (IOException ignored) {
            }
        }
        finally {
            try {
                os.close();
            }
            catch (IOException ignored) {
            }

            try {
                is.close();
            }
            catch (IOException ignored) {
            }

            if (sibling != null) {
                while (sibling.isAlive()) {
                    try {
                        sibling.join();
                    }
                    catch (InterruptedException ignored) {
                    }
                }

                try {
                    c.cm.closeChannel(c, "StreamForwarder (" + mode + ") is cleaning up the connection", true);
                }
                catch (IOException ignored) {
                }

                try {
                    if (s != null)
                        s.close();
                }
                catch (IOException ignored) {
                }
            }
        }
    }
}