org.eclipse.leshan.server.impl.FileSecurityStore Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of leshan-all Show documentation
A LWM2M client and server based on Californium (CoAP) all in one.
There is a newer version: 1.0.0-M11
/*******************************************************************************
 * Copyright (c) 2013-2015 Sierra Wireless and others.
 * 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * and Eclipse Distribution License v1.0 which accompany this distribution.
 * 
 * The Eclipse Public License is available at
 *    http://www.eclipse.org/legal/epl-v10.html
 * and the Eclipse Distribution License is available at
 *    http://www.eclipse.org/org/documents/edl-v10.html.
 * 
 * Contributors:
 *     Sierra Wireless - initial API and implementation
 *******************************************************************************/
package org.eclipse.leshan.server.impl;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;

import org.eclipse.leshan.server.security.NonUniqueSecurityInfoException;
import org.eclipse.leshan.server.security.SecurityInfo;
import org.eclipse.leshan.server.security.SecurityStore;
import org.eclipse.leshan.util.Validate;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * A {@link SecurityStore} which persists {@link SecurityInfo} in a file.
 * 
 * This implementation serializes the store content into a file to be able to re-load the {@link SecurityInfo} when the
 * server is restarted.
 * 
 */
public class FileSecurityStore extends InMemorySecurityStore {

    private static final Logger LOG = LoggerFactory.getLogger(FileSecurityStore.class);

    // the name of the file used to persist the store content
    private final String filename;

    // default location for persistence
    private static final String DEFAULT_FILE = "data/security.data";

    public FileSecurityStore() {
        this(DEFAULT_FILE);
    }

    public FileSecurityStore(String file) {
        Validate.notEmpty(file);
        filename = file;
        loadFromFile();
    }

    protected SecurityInfo addToStore(SecurityInfo info) throws NonUniqueSecurityInfoException {
        return super.add(info);
    }

    @Override
    public SecurityInfo add(SecurityInfo info) throws NonUniqueSecurityInfoException {
        writeLock.lock();
        try {
            SecurityInfo previous = addToStore(info);
            saveToFile();
            return previous;
        } finally {
            writeLock.unlock();
        }
    }

    @Override
    public SecurityInfo remove(String endpoint) {
        writeLock.lock();
        try {
            SecurityInfo info = super.remove(endpoint);
            if (info != null) {
                saveToFile();
            }
            return info;
        } finally {
            writeLock.unlock();
        }
    }

    protected void loadFromFile() {
        File file = new File(filename);
        if (!file.exists()) {
            return;
        }

        try (ObjectInputStream in = new ObjectInputStream(new FileInputStream(file));) {
            SecurityInfo[] infos = (SecurityInfo[]) in.readObject();

            if (infos != null) {
                for (SecurityInfo info : infos) {
                    addToStore(info);
                }
                if (infos.length > 0) {
                    LOG.debug("{} security infos loaded", infos.length);
                }
            }
        } catch (NonUniqueSecurityInfoException | IOException | ClassNotFoundException e) {
            LOG.error("Could not load security infos from file", e);
        }
    }

    protected void saveToFile() {
        try {
            File file = new File(filename);
            if (!file.exists()) {
                File parent = file.getParentFile();
                if (parent != null) {
                    parent.mkdirs();
                }
                file.createNewFile();
            }
            try (ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream(filename));) {
                out.writeObject(this.getAll().toArray(new SecurityInfo[0]));
            }
        } catch (IOException e) {
            LOG.error("Could not save security infos to file", e);
        }
    }
}