mamo.vanillaVotifier.AbstractConfig.java Source code

Introduction

Here is the source code for mamo.vanillaVotifier.AbstractConfig.java

Source

/*
 * Copyright (C) 2016  Matteo Morena
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package mamo.vanillaVotifier;

import mamo.vanillaVotifier.exception.InvalidPrivateKeyFileException;
import mamo.vanillaVotifier.exception.InvalidPublicKeyFileException;
import mamo.vanillaVotifier.exception.PrivateKeyFileNotFoundException;
import mamo.vanillaVotifier.exception.PublicKeyFileNotFoundException;
import mamo.vanillaVotifier.utils.RsaUtils;
import org.bouncycastle.util.io.pem.PemObject;
import org.bouncycastle.util.io.pem.PemReader;
import org.bouncycastle.util.io.pem.PemWriter;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.io.*;
import java.net.InetSocketAddress;
import java.security.KeyPair;
import java.security.spec.InvalidKeySpecException;
import java.util.ArrayList;
import java.util.List;

public abstract class AbstractConfig implements Config {
    @NotNull
    protected File configFile;
    protected int configVersion;
    @Nullable
    protected File logFile;
    @Nullable
    protected File logDirectory;
    @Nullable
    protected InetSocketAddress inetSocketAddress;
    @Nullable
    protected File publicKeyFile;
    @Nullable
    protected File privateKeyFile;
    @Nullable
    protected KeyPair keyPair;
    @NotNull
    protected ArrayList<VoteAction> voteActions = new ArrayList<VoteAction>();

    public AbstractConfig(@NotNull File configFile) {
        this.configFile = configFile;
    }

    @Override
    public File getConfigFile() {
        return configFile;
    }

    @Override
    public synchronized int getConfigVersion() {
        return configVersion;
    }

    @Override
    @Nullable
    public synchronized File getLogFile() {
        return logFile;
    }

    @Override
    public synchronized void setLogFile(@NotNull File location) {
        logFile = location;
    }

    @Override
    @Nullable
    public File getLogDirectory() {
        return logDirectory;
    }

    @Override
    public void setLogDirectory(@NotNull File location) {
        logDirectory = location;
    }

    @Override
    @Nullable
    public synchronized InetSocketAddress getInetSocketAddress() {
        return inetSocketAddress;
    }

    @Override
    public synchronized void setInetSocketAddress(@NotNull InetSocketAddress inetSocketAddress) {
        this.inetSocketAddress = inetSocketAddress;
    }

    @Override
    @Nullable
    public synchronized File getPublicKeyFile() {
        return publicKeyFile;
    }

    @Override
    public synchronized void setPublicKeyFile(@NotNull File location) {
        publicKeyFile = location;
    }

    @Override
    @Nullable
    public synchronized File getPrivateKeyFile() {
        return privateKeyFile;
    }

    @Override
    public synchronized void setPrivateKeyFile(@NotNull File location) {
        privateKeyFile = location;
    }

    @Override
    @Nullable
    public synchronized KeyPair getKeyPair() {
        return keyPair;
    }

    @Override
    public synchronized void setKeyPair(@NotNull KeyPair keyPair) {
        this.keyPair = keyPair;
    }

    @Override
    public synchronized void generateKeyPair() {
        generateKeyPair(2048);
    }

    @Override
    public synchronized void generateKeyPair(int keySize) {
        setKeyPair(RsaUtils.genKeyPair(keySize));
    }

    @Override
    @NotNull
    public List<VoteAction> getVoteActions() {
        return voteActions;
    }

    protected void copyDefaultConfig(InputStream in) throws IOException {
        BufferedReader reader = new BufferedReader(new InputStreamReader(in));
        StringBuilder stringBuilder = new StringBuilder();
        int i;
        while ((i = reader.read()) != -1) {
            stringBuilder.append((char) i);
        }
        reader.close();
        BufferedWriter writer = new BufferedWriter(new FileWriter(configFile));
        for (char c : stringBuilder.toString()
                .replaceAll("\\u000D\\u000A|[\\u000A\\u000B\\u000C\\u000D\\u0085\\u2028\\u2029]",
                        System.getProperty("line.separator"))
                .toCharArray()) {
            writer.write((int) c);
        }
        writer.flush();
        writer.close();
    }

    protected void loadKeyPair() throws IOException, InvalidKeySpecException {
        if (!publicKeyFile.exists() && !privateKeyFile.exists()) {
            generateKeyPair();
            saveKeyPair();
        }
        if (!publicKeyFile.exists()) {
            throw new PublicKeyFileNotFoundException();
        }
        if (!privateKeyFile.exists()) {
            throw new PrivateKeyFileNotFoundException();
        }
        PemReader publicKeyPemReader = new PemReader(new BufferedReader(new FileReader(publicKeyFile)));
        PemReader privateKeyPemReader = new PemReader(new BufferedReader(new FileReader(privateKeyFile)));
        PemObject publicPemObject = publicKeyPemReader.readPemObject();
        if (publicPemObject == null) {
            throw new InvalidPublicKeyFileException();
        }
        PemObject privatePemObject = privateKeyPemReader.readPemObject();
        if (privatePemObject == null) {
            throw new InvalidPrivateKeyFileException();
        }
        keyPair = new KeyPair(RsaUtils.bytesToPublicKey(publicPemObject.getContent()),
                RsaUtils.bytesToPrivateKey(privatePemObject.getContent()));
        publicKeyPemReader.close();
        privateKeyPemReader.close();
    }

    @Override
    public synchronized void saveKeyPair() throws IOException {
        PemWriter publicPemWriter = new PemWriter(new BufferedWriter(new FileWriter(getPublicKeyFile())));
        publicPemWriter.writeObject(new PemObject("PUBLIC KEY", getKeyPair().getPublic().getEncoded()));
        publicPemWriter.flush();
        publicPemWriter.close();
        PemWriter privatePemWriter = new PemWriter(new BufferedWriter(new FileWriter(getPrivateKeyFile())));
        privatePemWriter.writeObject(new PemObject("RSA PRIVATE KEY", getKeyPair().getPrivate().getEncoded()));
        privatePemWriter.flush();
        privatePemWriter.close();
    }
}