Package org.gudy.azureus2.core3.ipfilter.impl

Source Code of org.gudy.azureus2.core3.ipfilter.impl.IpFilterManagerImpl

/*
* Created on 19-Jul-2004
* Created by Paul Gardner
* Copyright (C) 2004, 2005, 2006 Aelitis, All Rights Reserved.
*
* 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.
*
* AELITIS, SAS au capital de 46,603.30 euros
* 8 Allee Lenotre, La Grille Royale, 78600 Le Mesnil le Roi, France.
*
*/

package org.gudy.azureus2.core3.ipfilter.impl;

/**
* @author parg
*
*/

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.RandomAccessFile;

import org.gudy.azureus2.core3.config.COConfigurationManager;
import org.gudy.azureus2.core3.config.ParameterListener;
import org.gudy.azureus2.core3.ipfilter.BadIps;
import org.gudy.azureus2.core3.ipfilter.IpFilter;
import org.gudy.azureus2.core3.ipfilter.IpFilterManager;
import org.gudy.azureus2.core3.ipfilter.IpRange;
import org.gudy.azureus2.core3.util.FileUtil;

public class
IpFilterManagerImpl
  implements IpFilterManager, ParameterListener
{
  protected static IpFilterManagerImpl    singleton  = new IpFilterManagerImpl();

  private RandomAccessFile rafDescriptions = null;
 
  /**
   *
   */
  public IpFilterManagerImpl() {
    COConfigurationManager.addAndFireParameterListener(
        "Ip Filter Enable Description Cache", this);
  }
 
  public Object addDescription(IpRange range, byte[] description) {
    //if (true) return;
    if (rafDescriptions == null) {
      return null;
    }
   
    try {
      if (description == null || description.length == 0)
        return null;

      int start;
      int end;
      start = (int)rafDescriptions.getFilePointer();
      int len = (int)rafDescriptions.length();

      //System.out.println(len - 0x1FFFFFF);
      if (len + 61 >= 0x1FFFFFF) {
        // we could try to fit a desc < 61, but why bother.. at this point
        // we have at least 550,072 ranges
        return null;
      }
     
      if (start != len) {
        rafDescriptions.seek(len);
        start = (int)rafDescriptions.getFilePointer();
      }
     
      // last 25: position
      // 26 - 31 (6, 61 chars max): len
     
      if (description.length <= 61) {
        rafDescriptions.write(description);
      } else {
        rafDescriptions.write(description, 061);
      }
      end = (int)rafDescriptions.getFilePointer();
     
      //System.out.println("add " + new String(description, 0, (end - start)) + "; " + start + " - " + end);
     
      int info = start + ((end - start) << 25);
     
      return new Integer(info);
    } catch (Exception e) {
      // TODO Auto-generated catch block
      e.printStackTrace();
    }
   
    return null;
  }
 
  public byte[] getDescription(Object info) {
    // if cached, info is an object array, with the first index being the descr
    if (info instanceof Object[]) {
      return (byte[])(((Object[])info)[0]);
    }
   
    if (rafDescriptions == null || !(info instanceof Integer)) {
      return "".getBytes();
    }
   
    try {
      int posInfo = ((Integer)info).intValue();
      int pos = posInfo & 0x1FFFFFF;
      int len = posInfo >> 25;
     
      if (len < 0) {
        throw new IllegalArgumentException(getClass().getName() + ": invalid posInfo [" + posInfo +"], pos [" + pos + "], len [" + len + "]");
      }

      if (rafDescriptions.getFilePointer() != pos) {
        rafDescriptions.seek(pos);
      }

      byte[] bytes = new byte[len];
      rafDescriptions.read(bytes);
     
      return bytes;
    } catch (IOException e) {
      return "".getBytes();
    }
  }
 
  public void cacheAllDescriptions() {
    IpRange[] ranges = getIPFilter().getRanges();
    for (int i = 0; i < ranges.length; i++) {
      Object info = ((IpRangeImpl)ranges[i]).getDescRef();
      if (info instanceof Integer) {
        byte[] desc = getDescription(info);
        Object[] data = { desc, info };
        ((IpRangeImpl)ranges[i]).setDescRef(data);
      }
    }
  }
 
  public void clearDescriptionCache() {
    IpRange[] ranges = getIPFilter().getRanges();
    for (int i = 0; i < ranges.length; i++) {
      Object info = ((IpRangeImpl)ranges[i]).getDescRef();
      if (info instanceof Object[]) {
        Integer data = (Integer)((Object[])info)[1];
        ((IpRangeImpl)ranges[i]).setDescRef(data);
      }
    }
  }
 
  public void deleteAllDescriptions() {
    if (rafDescriptions != null) {
      try {
        rafDescriptions.close();
      } catch (IOException e) {
      }
      rafDescriptions = null;
    }
   
    parameterChanged(null);
  }

  public static IpFilterManager
  getSingleton()
  {
    return( singleton );
  }
 
  public IpFilter
  getIPFilter()
  {
    return( IpFilterImpl.getInstance());
  }
 
  public BadIps
  getBadIps()
  {
    return (BadIpsImpl.getInstance());
  }

  public void parameterChanged(String parameterName) {
    boolean enable = COConfigurationManager.getBooleanParameter("Ip Filter Enable Description Cache");
    if (enable && rafDescriptions == null) {
      File fDescriptions = FileUtil.getUserFile("ipfilter.cache");
      try {
        if (fDescriptions.exists()) {
          fDescriptions.delete();
        }
        rafDescriptions = new RandomAccessFile(fDescriptions, "rw");
      } catch (FileNotFoundException e) {
        // TODO Auto-generated catch block
        e.printStackTrace();
      }
    } else if (!enable && rafDescriptions != null) {
      try {
        rafDescriptions.close();
      } catch (IOException e) {
      }
      rafDescriptions = null;
    }
  }
}
TOP

Related Classes of org.gudy.azureus2.core3.ipfilter.impl.IpFilterManagerImpl

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.