/*
* JMule - Java file sharing client
* Copyright (C) 2007-2008 JMule team ( jmule@jmule.org / 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.FileHash;
import org.jmule.core.edonkey.PartHashSet;
import org.jmule.core.utils.MD4;
/**
*
* @author javajox
* @version $$Revision: 1.4 $$
* Last changed by $$Author: binary255 $$ on $$Date: 2009/12/19 19:31:54 $$
*/
public class FileHashing extends JMFileTask {
private FileChannel fileChannel;
private PartHashSet fileHashSet;
private long position = 0L;
private boolean close_file_on_finish = false;
public FileHashing(File file) throws FileNotFoundException {
super("ED2K MD4 file hashing process");
fileChannel = ( new FileInputStream(file) ).getChannel();
close_file_on_finish = true;
}
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();
}
if (close_file_on_finish)
try {
fileChannel.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}