org.telegram.mtproto.state.AbsMTProtoState Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mtproto Show documentation
Telegram MTProto implementation for java
The newest version!
package org.telegram.mtproto.state;

import org.telegram.mtproto.time.TimeOverlord;

import java.util.HashMap;

/**
 * MTProto state
 */
public abstract class AbsMTProtoState {

    public abstract byte[] getAuthKey();

    public abstract ConnectionInfo[] getAvailableConnections();

    public abstract KnownSalt[] readKnownSalts();

    protected abstract void writeKnownSalts(KnownSalt[] salts);

    public void mergeKnownSalts(int currentTime, KnownSalt[] salts) {
        KnownSalt[] knownSalts = readKnownSalts();
        HashMap ids = new HashMap();
        for (KnownSalt s : knownSalts) {
            if (s.getValidUntil() < currentTime) {
                continue;
            }
            ids.put(s.getSalt(), s);
        }
        for (KnownSalt s : salts) {
            if (s.getValidUntil() < currentTime) {
                continue;
            }
            ids.put(s.getSalt(), s);
        }
        writeKnownSalts(ids.values().toArray(new KnownSalt[0]));
    }

    public void addCurrentSalt(long salt) {
        int time = (int) (TimeOverlord.getInstance().getServerTime() / 1000);
        mergeKnownSalts(time, new KnownSalt[]{new KnownSalt(time, time + 30 * 60, salt)});
    }

    public void badServerSalt(long salt) {
        int time = (int) (TimeOverlord.getInstance().getServerTime() / 1000);
        writeKnownSalts(new KnownSalt[]{new KnownSalt(time, time + 30 * 60, salt)});
    }

    public void initialServerSalt(long salt) {
        int time = (int) (TimeOverlord.getInstance().getServerTime() / 1000);
        writeKnownSalts(new KnownSalt[]{new KnownSalt(time, time + 30 * 60, salt)});
    }

    public long findActualSalt(int time) {
        KnownSalt[] knownSalts = readKnownSalts();
        for (KnownSalt salt : knownSalts) {
            if (salt.getValidSince() <= time && time <= salt.getValidUntil()) {
                return salt.getSalt();
            }
        }

        return 0;
    }

    public int maximumCachedSalts(int time) {
        int count = 0;
        for (KnownSalt salt : readKnownSalts()) {
            if (salt.getValidSince() > time) {
                count++;
            }
        }
        return count;
    }

    public int maximumCachedSaltsTime() {
        int max = 0;
        for (KnownSalt salt : readKnownSalts()) {
            max = Math.max(max, salt.getValidUntil());
        }
        return max;
    }
}