filehashing.java

JMule是一个基于Java开发

/*
 *  JMule - Java file sharing client
 *  Copyright (C) 2007-2008 JMule team ( [e-mail] / http://jmule.org )
 *
 *  Any parts of this program derived from other projects, or contributed
 *  by third-party developers are copyrighted by their respective authors.
 *
 *  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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 */
package org.jmule.core.sharingmanager;

import static org.jmule.core.edonkey.E2DKConstants.PARTSIZE;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.channels.FileChannel;

import org.jmule.core.edonkey.impl.FileHash;
import org.jmule.core.edonkey.impl.PartHashSet;
import org.jmule.core.utils.MD4;

/**
 * 
 * @author javajox
 * @version $$Revision: 1.2 $$
 * Last changed by $$Author: binary256_ $$ on $$Date: 2008/10/16 18:22:33 $$
 */
public class FileHashing extends JMFileTask {

	private FileChannel fileChannel;
	private PartHashSet fileHashSet;
	private long position = 0L;
	
	public FileHashing(File file) throws FileNotFoundException {
		
		super("ED2K MD4 file hashing process");
		
		fileChannel = ( new FileInputStream(file) ).getChannel();
		
	}
	
	public FileHashing(FileChannel fileChannel) {
		
		super("ED2K MD4 file hashing process");
		
		this.fileChannel = fileChannel;
		
	}
	
	public double getPercent() {
		
		double result = 0d;
		
		try {
			
			double rate = (double)(position) * 100d;
			
			result = (double)(rate) / (double)(fileChannel.size());
			
		} catch (IOException e) {
			e.printStackTrace();
			return 100d;
		} 
		
		return result;
		
	}
	

	public PartHashSet getFileHashSet() {
		
		return fileHashSet;
		
	}
	
	public void run() {
		
		try {
			
			isDone = false;
			
	    	fileChannel.position(0);
	    	
            int i;
            
            int c;
            
            ByteBuffer hashset;
            
            MD4 msgDigest = new MD4();
            
            ByteBuffer bb = ByteBuffer.allocateDirect(8192).order(ByteOrder.LITTLE_ENDIAN);
            
            ByteBuffer di = ByteBuffer.allocate(16).order(ByteOrder.LITTLE_ENDIAN);
            
            c = (int) ((fileChannel.size() + PARTSIZE - 1) / PARTSIZE);
            
            // we will need space for c Parts
            hashset = ByteBuffer.allocate(16 * (c>0?c:1)).order(ByteOrder.LITTLE_ENDIAN);
            
            for (i = 1; i < c; i++) {
            	
                while (position <= (i * PARTSIZE - bb.capacity())) {
                	
                	if( stop ) return;
                	
                    position+=fileChannel.read(bb, position);
                    
                    bb.flip();
                    
                    msgDigest.update(bb);
                    
                    bb.rewind();
                    
                }
            
                if (position < (i * PARTSIZE)) {
                	
                    bb.limit((int) ((i * PARTSIZE) - position));
                    
                    position+=fileChannel.read(bb, position);
                    
                    bb.flip();
                    
                    msgDigest.update(bb);
                    
                    bb.rewind();
                    
                }
                
                 hashset.limit(16 * i);
                 
                 //update hashset - add a hash
                 msgDigest.finalDigest(hashset);
                 
            }
            
            if (c > 0) {
            	
                while (position < (fileChannel.size())) {
                	
                	if(stop) return;
                	
                    position+=fileChannel.read(bb, position);
                    
                    bb.flip();
                    
                    msgDigest.update(bb);
                    
                    bb.rewind();
                    
                }
                
                hashset.limit(16 * i);
                
            }
            
            msgDigest.finalDigest(hashset);
           
            hashset.flip();
            
            if (c > 1) {
            	
                msgDigest.update(hashset);
                
                msgDigest.finalDigest(di);
                
            } else {
            	
                di.put(hashset);
                
            }
            
            di.rewind();
            
            hashset.rewind();
            
            //hashes 
            byte[][] hashes = new byte[(c!=1)?(c + 1):1][16];
            
            di.get(hashes[0]);
            
            for (int j = 1; j < hashes.length; j++) {
            	
            	if(stop) return;
            	
                hashset.get(hashes[j]);
                
            }
            
            hashset.rewind();
                       
            fileHashSet = new PartHashSet(new FileHash(hashes[0]));
            
            for(int j=1; j < hashes.length;j++){
            	
            	if(stop) return;
            	
            	fileHashSet.add(hashes[j]);
            	
            }
            
            isDone = true;
            
        } catch (Throwable e) {
        	
        	 e.printStackTrace();
        	 
          }
	}
		
}