/*
 * Copyright 1998-2008 Sun Microsystems, Inc.  All Rights Reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 */

/*
 */

import java.io.*;
import java.net.*;
import java.rmi.*;
import java.rmi.server.*;
import java.util.zip.*;


public class MultiSocketFactory  {

    private static RMISocketFactory def =
        RMISocketFactory.getDefaultSocketFactory();


    public static class ServerFactory
        implements RMIServerSocketFactory, Serializable
    {

        private String protocol;
        private byte[] data;

        public ServerFactory(String protocol, byte[] data) {
            this.protocol = protocol;
            this.data = data;
        }

        public ServerSocket createServerSocket(int port) throws IOException
        {
            if (protocol.equals("compress")) {
                return new CompressServerSocket(port);

            } else if (protocol.equals("xor")) {
                if (data == null || data.length != 1)
                    throw new IOException("invalid argument for XOR protocol");
                return new XorServerSocket(port, data[0]);

            }

            return def.createServerSocket(port);
        }
    }

    public static class ClientFactory
        implements RMIClientSocketFactory, Serializable
    {

        private String protocol;
        private byte[] data;

        public ClientFactory(String protocol, byte[] data) {
            this.protocol = protocol;
            this.data = data;
        }

        public Socket createSocket(String host, int port)
            throws IOException
        {
            if (protocol.equals("compress")) {
                return new CompressSocket(host, port);

            } else if (protocol.equals("xor")) {
                if (data == null || data.length != 1)
                    throw new IOException("invalid argument for XOR protocol");
                return new XorSocket(host, port, data[0]);

            }

            return def.createSocket(host, port);
        }
    }

    static class CompressSocket extends Socket {
        private InputStream in;
        private OutputStream out;
        public CompressSocket() { super(); }
        public CompressSocket(String host, int port) throws IOException {
            super(host, port);
        }
        public InputStream getInputStream() throws IOException {
            if (in == null) {
                in = new CompressInputStream(super.getInputStream());
            }
            return in;
        }
        public OutputStream getOutputStream() throws IOException {
            if (out == null) {
                out = new CompressOutputStream(super.getOutputStream());
            }
            return out;
        }
    }

    static class CompressServerSocket extends ServerSocket {
        public CompressServerSocket(int port) throws IOException {
            super(port);
        }
        public Socket accept() throws IOException {
            Socket s = new CompressSocket();
            implAccept(s);
            return s;
        }
    }

    static class XorSocket extends Socket {
        private byte pattern;
        private InputStream in;
        private OutputStream out;
        public XorSocket(byte pattern) { super(); this.pattern = pattern; }
        public XorSocket(String host, int port, byte pattern)
            throws IOException
        {
            super(host, port);
            this.pattern = pattern;
        }
        public InputStream getInputStream() throws IOException {
            if (in == null) {
                in = new XorInputStream(super.getInputStream(), pattern);
            }
            return in;
        }
        public OutputStream getOutputStream() throws IOException {
            if (out == null) {
                out = new XorOutputStream(super.getOutputStream(), pattern);
            }
            return out;
        }
    }

    static class XorServerSocket extends ServerSocket {
        private byte pattern;
        public XorServerSocket(int port, byte pattern) throws IOException {
            super(port);
            this.pattern = pattern;
        }
        public Socket accept() throws IOException {
            Socket s = new XorSocket(pattern);
            implAccept(s);
            return s;
        }
    }

    static class XorOutputStream extends FilterOutputStream {
        private byte pattern;
        public XorOutputStream(OutputStream out, byte pattern) {
            super(out);
            this.pattern = pattern;
        }
        public void write(int b) throws IOException {
            out.write(b ^ pattern);
            out.flush();
        }
        public void write(byte b[], int off, int len) throws IOException {
            for (int i = 0; i < len; i++)
                write(b[off + i]);
        }
    }

    static class XorInputStream extends FilterInputStream {
        private byte pattern;
        public XorInputStream(InputStream in, byte pattern) {
            super(in);
            this.pattern = pattern;
        }
        public int read() throws IOException {
            int b = in.read();
//          System.out.print("BEFORE: " + Integer.toHexString(b));
            if (b != -1)
                b = (b ^ pattern) & 0xFF;
//          System.out.println("\tAFTER: " + Integer.toHexString(b));
            return b;
        }
        public int read(byte b[], int off, int len) throws IOException {
            if (len <= 0) {
                return 0;
            }

            int c = read();
            if (c == -1) {
                return -1;
            }
            b[off] = (byte)c;

            int i = 1;
/*****
            try {
                for (; i < len ; i++) {
                    c = read();
                    if (c == -1) {
                        break;
                    }
                    if (b != null) {
                        b[off + i] = (byte)c;
                    }
                }
            } catch (IOException ee) {
            }
*****/
            return i;
        }
    }
}