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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ports-jrandtest Show documentation
Fork from https://github.com/cryptopony/jrandtest
The newest version!
/*
 * 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;

/**
 * Simple java secured random as a random stream
 * 
 * 
 * @author Zur Aougav
 */
public class JavaSecuredRandom extends FileAlgoRandomStream {

    java.security.SecureRandom algo = null;

    byte[] outAlgoBuffer = new byte[256];

    int outAlgoBufferIx = outAlgoBuffer.length;

    public JavaSecuredRandom() {
        super();
    }

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

    
    @Override
    public void setupKeys() {
        super.setupKeys();
    }

    @Override
    public void setup() {
        algo = new java.security.SecureRandom();
    }

    @Override
    public boolean openInputStream() {
        if (filename != null) {
            super.openInputStream();
        }

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

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

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

        if (outAlgoBufferIx == outAlgoBuffer.length) {
            algo.nextBytes(outAlgoBuffer);
            outAlgoBufferIx = 0;
        }

        byte prng = outAlgoBuffer[outAlgoBufferIx++];

        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 = algo.nextInt();

        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 = algo.nextLong();

        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) {
            //JavaSecureRandom algo = new JavaSecureRandom(args[0]);
            JavaSecuredRandom algo = new JavaSecuredRandom();
            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);
            }
        }
    }
}