org.bitcoinj.coinjoin.CoinJoinQueue Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dashj-core Show documentation
Dash Java Library
There is a newer version: 21.1.2
/*
 * Copyright 2022 Dash Core Group
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.bitcoinj.coinjoin;

import com.google.common.annotations.VisibleForTesting;
import org.bitcoinj.core.MasternodeSignature;
import org.bitcoinj.core.Message;
import org.bitcoinj.core.NetworkParameters;
import org.bitcoinj.core.ProtocolException;
import org.bitcoinj.core.Sha256Hash;
import org.bitcoinj.core.UnsafeByteArrayOutputStream;
import org.bitcoinj.core.Utils;
import org.bitcoinj.crypto.BLSPublicKey;
import org.bitcoinj.crypto.BLSScheme;
import org.bitcoinj.crypto.BLSSecretKey;
import org.bitcoinj.crypto.BLSSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Objects;

/**
 * A currently in progress mixing merge and denomination information
 */
// dsq
public class CoinJoinQueue extends Message {
    private static final Logger log = LoggerFactory.getLogger(CoinJoinQueue.class);

    private int denomination;
    private Sha256Hash proTxHash;
    private long time;
    private boolean ready;  // Ready to submit
    private MasternodeSignature signature;
    // memory only
    private boolean tried;

    public CoinJoinQueue(NetworkParameters params, byte[] payload, int protocolVersion) {
        super(params, payload, 0, protocolVersion);
    }

    public CoinJoinQueue(
            NetworkParameters params,
            int denomination,
            Sha256Hash proTxHash,
            long time,
            boolean ready) {

        super(params);
        this.denomination = denomination;
        this.proTxHash = proTxHash;
        this.time = time;
        this.ready = ready;
        this.signature = null;
        this.protocolVersion = params.getProtocolVersionNum(NetworkParameters.ProtocolVersion.CURRENT);
    }

    public CoinJoinQueue(
            NetworkParameters params,
            int denomination,
            Sha256Hash proTxHash,
            long time,
            boolean ready,
            MasternodeSignature signature) {

        this(params, denomination, proTxHash, time, ready);
        this.signature = signature;
    }

    @Override
    protected void parse() throws ProtocolException {
        denomination = (int)readUint32();
        proTxHash = readHash();
        time = readInt64();
        ready = readBytes(1)[0] == 1;
        signature = new MasternodeSignature(params, payload, cursor);
        cursor += signature.getMessageSize();

        length = cursor - offset;
    }

    @Override
    protected void bitcoinSerializeToStream(OutputStream stream) throws IOException {
        Utils.uint32ToByteStreamLE(denomination, stream);
        stream.write(proTxHash.getReversedBytes());
        Utils.int64ToByteStreamLE(time, stream);
        stream.write(ready ? 1 : 0);
        signature.bitcoinSerialize(stream);
    }

    public Sha256Hash getSignatureHash() {
        try {
            ByteArrayOutputStream bos = new UnsafeByteArrayOutputStream();
            Utils.uint32ToByteStreamLE(denomination, bos);
            bos.write(proTxHash.getReversedBytes());
            Utils.int64ToByteStreamLE(time, bos);
            bos.write(ready ? 1 : 0);
            return Sha256Hash.twiceOf(bos.toByteArray());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public boolean checkSignature(BLSPublicKey pubKey) {
        Sha256Hash hash = getSignatureHash();

        // use the currently active scheme
        BLSSignature sig = new BLSSignature(signature.getBytes(), BLSScheme.isLegacyDefault());

        if (!sig.verifyInsecure(pubKey, hash)) {
            log.info("CoinJoinQueue-CheckSignature: verifyInsecure failed\n");
            return false;
        }

        return true;
    }

    @VisibleForTesting
    public boolean sign(BLSSecretKey blsKeyOperator) {
        Sha256Hash hash = getSignatureHash();
        BLSSignature sig = blsKeyOperator.sign(hash);
        if (!sig.isValid()) {
            return false;
        }
        signature = new MasternodeSignature(sig.bitcoinSerialize());

        return true;
    }

    @Override
    public String toString() {
        return String.format(
                "CoinJoinQueue(denom=%s[%d], t=%d[exp=%s], ready=%s, proTxHash=%s)",
                CoinJoin.denominationToString(denomination),
                denomination,
                time,
                isTimeOutOfBounds(),
                ready,
                proTxHash.toString().substring(0, 16)
        );
    }

    public int getDenomination() {
        return denomination;
    }

    public long getTime() {
        return time;
    }

    public boolean isReady() {
        return ready;
    }

    public boolean isTried() {
        return tried;
    }

    public void setTried(boolean tried) {
        this.tried = tried;
    }

    public MasternodeSignature getSignature() {
        return signature;
    }

    public boolean isTimeOutOfBounds() {
        return isTimeOutOfBounds(Utils.currentTimeSeconds());
    }
    public boolean isTimeOutOfBounds(long currentTime) {
        return currentTime - time > CoinJoinConstants.COINJOIN_QUEUE_TIMEOUT ||
                time - currentTime > CoinJoinConstants.COINJOIN_QUEUE_TIMEOUT;
    }

    public Sha256Hash getProTxHash() {
        return proTxHash;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        CoinJoinQueue that = (CoinJoinQueue) o;

        if (denomination != that.denomination) return false;
        if (time != that.time) return false;
        if (ready != that.ready) return false;
        if (tried != that.tried) return false;
        if (!Objects.equals(proTxHash, that.proTxHash)) return false;
        return Objects.equals(signature, that.signature);
    }

    @Override
    public int hashCode() {
        return proTxHash.hashCode();
    }
}