com.fasteasytrade.jrandtest.algo.ARC4 Maven / Gradle / Ivy

Go to download
/*
 * Created on 17/02/2005
 *
 * JRandTest package
 *
 * Copyright (c) 2005, Zur Aougav, [email protected]
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification, 
 * are permitted provided that the following conditions are met:
 * 
 * Redistributions of source code must retain the above copyright notice, this list 
 * of conditions and the following disclaimer. 
 * 
 * Redistributions in binary form must reproduce the above copyright notice, this 
 * list of conditions and the following disclaimer in the documentation and/or 
 * other materials provided with the distribution. 
 * 
 * Neither the name of the JRandTest nor the names of its contributors may be 
 * used to endorse or promote products derived from this software without specific 
 * prior written permission. 
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR 
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.fasteasytrade.jrandtest.algo;

import com.fasteasytrade.jrandtest.io.FileAlgoRandomStream;

/** 
 * Alleged RC4 algorithm as a random stream
 * 
 * 
 * @author Zur Aougav
 */
public class ARC4 extends FileAlgoRandomStream {

    RC4Key algo = null;

    byte[] defaultPublicKey = { 0x01, 0x02, 0x03, 0x04 };

    public ARC4() {
        super();
    }

    public ARC4(String keyFileName) {
        super(keyFileName);
    }

    @Override
    public void setupKeys() {
        publicKeyLength = 256;
        privateKeyLength = 256;
        super.setupKeys();
    }

    @Override
    public void setup() {
        if (publicKey == null) {
            algo = new RC4Key(defaultPublicKey, 0, null);
        } else {
            algo = new RC4Key(publicKey, 0, null);
        }
    }

    @Override
    public boolean openInputStream() {

        if (filename != null) {
            super.openInputStream();
        }

        setup(); // make sure algorithm and keys/states are reset
        count = 0;
        countLastRead = SIZE;
        actualSize = SIZE;
        open = true;
        return open;
    }

    @Override
    public byte readByte() {
        if (!isOpen()) {
            return -1;
        }

        if (filename == null && count > maxCount) {
            open = false;
            return -1;
        }

        byte prng = algo.next();

        if (filename == null) {
            count++;
            return prng;
        }

        /*
         * we have a real filename to encrypt
         */
        byte data = super.readByte();
        return (byte)(prng ^ data);
    }

    @Override
    public int readInt() {
        if (!isOpen()) {
            return -1;
        }

        if (filename == null && count > maxCount) {
            open = false;
            return -1;
        }

        int prng = 0xff & algo.next();
        prng = (prng << 8) | (0xff & algo.next());
        prng = (prng << 8) | (0xff & algo.next());
        prng = (prng << 8) | (0xff & algo.next());

        if (filename == null) {
            count += 4;
            return prng;
        }

        /*
         * we have a real filename to encrypt
         */
        int data = super.readInt();
        return data ^ prng;
    }

    @Override
    public long readLong() {
        if (!isOpen()) {
            return -1;
        }

        if (filename == null && count > maxCount) {
            open = false;
            return -1;
        }

        long prng = 0xff & algo.next();
        for (int i = 0; i < 7; i++) {
            prng = (prng << 8) | (0xff & algo.next());
        }

        if (filename == null) {
            count += 8;
            return prng;
        }

        /*
         * we have a real filename to encrypt
         */
        long data = super.readLong();
        return data ^ prng;
    }

    public static void main(String[] args) {
        if (args != null && args.length > 0 && args[0] != null) {
            //ARC4 algo = new ARC4(args[0]);
            ARC4 algo = new ARC4();
            algo.setup();
            try {
                algo.openInputStream();
                for (int i = 0; i < 100; i++) {
                    System.out.print(algo.readByte());
                    System.out.print(",");
                }
                System.out.println();
            } catch (Exception e) {
                System.out.println("" + e);
            }
        }
    }
}