FileSystemStore.java

/***********************************************************************
 *
 * $CVSHeader$
 *
 * This file is part of WebScarab, an Open Web Application Security
 * Project utility. For details, please see http://www.owasp.org/
 *
 * Copyright (c) 2002 - 2004 Rogan Dawes
 *
 * 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 2
 * 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * Getting Source
 * ==============
 *
 * Source for this application is maintained at Sourceforge.net, a
 * repository for free software projects.
 * 
 * For details, please see http://www.sourceforge.net/projects/owasp
 *
 */

/*
 * FileSystemStore.java
 *
 * Created on September 14, 2004, 4:09 PM
 */

package org.owasp.webscarab.plugin.sessionid;

import org.owasp.webscarab.model.StoreException;

import java.util.List;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.SortedMap;
import java.util.TreeMap;

import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;

/**
 *
 * @author  knoppix
 */
public class FileSystemStore implements SessionIDStore {
    
    private SortedMap<String, List<SessionID>> _sessionIDs = new TreeMap<String, List<SessionID>>();
    
    private File _dir;
    
    /** Creates a new instance of FileSystemStore */
    public FileSystemStore(File dir) throws StoreException {
        _dir = dir;
        File f = new File(_dir, "sessionids");
        if (f.exists()) {
            load();
        }
    }
    
    private void load() throws StoreException {
        File f = new File(_dir, "sessionids");
        if (!f.isFile()) {
            return;
        }
        try {
            BufferedReader br = new BufferedReader(new FileReader(f));
            for (String key = br.readLine(); key != null; key = br.readLine()) {
                ArrayList<SessionID> list = new ArrayList<SessionID>();
                _sessionIDs.put(key, list);
                for (String line = br.readLine(); line != null && ! line.equals(""); line=br.readLine()) {
                    SessionID id = new SessionID(line);
                    list.add(id);
                }
            }
        } catch (IOException ioe) {
            throw new StoreException("Error reading sessionids: " + ioe);
        }
    }
    
    public int addSessionID(String key, SessionID id) {
        List<SessionID> list = _sessionIDs.get(key);
        if (list == null) {
            list = new ArrayList<SessionID>();
            _sessionIDs.put(key, list);
        }
        int insert = Collections.binarySearch(list, id);
        if (insert<0) insert = -insert-1;
        list.add(insert, id);
        return insert;
    }
    
    public void clearSessionIDs(String key) {
        _sessionIDs.remove(key);
    }
    
    public int getSessionIDNameCount() {
        return _sessionIDs.size();
    }
    
    public String getSessionIDName(int index) {
        Iterator<String> it = _sessionIDs.keySet().iterator();
        while (it.hasNext()) {
            String key = it.next();
            if (index==0) {
                return key;
            } else {
                index--;
            }
        }
        return null;
    }
    
    public int getSessionIDCount(String key) {
        List<SessionID> list = _sessionIDs.get(key);
        if (list == null) return 0;
        return list.size();
    }
    
    public SessionID getSessionIDAt(String key, int index) {
        List<SessionID> list = _sessionIDs.get(key);
        if (list == null) return null;
        return list.get(index);
    }
    
    public void flush() throws StoreException {
        File f = new File(_dir, "sessionids");
        if (f.exists() && !f.isFile()) {
            throw new StoreException("Couldn't create output file " + f);
        }
        try {
            BufferedWriter bw = new BufferedWriter(new FileWriter(f));
            Iterator<String> it = _sessionIDs.keySet().iterator();
            while (it.hasNext()) {
                String key = it.next();
                bw.write(key + "\r\n");
                List<SessionID> list = _sessionIDs.get(key);
                Iterator<SessionID> it2 = list.iterator();
                while (it2.hasNext()) {
                    SessionID id = it2.next();
                    bw.write(id.toString() + "\r\n");
                }
                bw.write("\r\n");
            }
            bw.flush();
            bw.close();
        } catch (IOException ioe) {
            throw new StoreException("IOException: " + ioe);
        }
    }
    
}