我正在尝试从内存中读取文件并将其拆分为1KB块。
程序的作用是从内存中读取文件(视频文件),然后将其拆分为1KB块。然后使用SHA-256散列最后一个块,并将散列附加到第二个最后一个块。然后,它计算第二个最后一个块上的散列和附加的散列,然后将此散列附加到其先前的块。这一直持续到第一个块,它将附加第二个块的散列。
我只需要第一个块的哈希值及其附加的哈希值。我试图以两种方式实现这一点,但我认为我做错了。有人可以告诉我我做错了什么。在没有解决方案的情况下,我被困在这里6天了。我已粘贴下面的两个实现。任何帮助将不胜感激。
我已经阅读了整个文件,并试图在下面的尝试中手动将字节数组拆分为1KB块。
package com.test;
import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.security.MessageDigest;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class ReadFileByteByByte {
public static void main(String[] args) throws Exception {
InputStream inStream = null;
BufferedInputStream bis = null;
try{
inStream = new FileInputStream("C:\\a.mp4");
bis = new BufferedInputStream(inStream);
int numByte = bis.available();
byte[] buf = new byte[numByte];
bis.read(buf, 0, numByte);
System.out.println(numByte/1024);
ArrayList<byte[]> a = new ArrayList<>();
ArrayList<byte[]> b = new ArrayList<>();
for(int i=0,j=0;i<buf.length;i++,j++){
byte[] buf2 = new byte[1057];
buf2[j] = buf[i];
if(i%1024==1023){
a.add(buf2);
j=0;
}
}
for(int i=a.size()-1,j=-1;i>=0;i--,j++){
MessageDigest digest = MessageDigest.getInstance("SHA-256");
if(i==a.size()-1){
byte[] hash = digest.digest(a.get(i));
byte[] dest = new byte[a.get(i).length+hash.length];
System.arraycopy(a.get(i-1), 0, dest, 0, a.get(i-1).length);
System.arraycopy(hash, 0, dest, a.get(i-1).length, hash.length);
b.add(dest);
}
else{
byte[] hash = digest.digest(b.get(0));
if(i!=0){
byte[] dest = new byte[a.get(i-1).length+hash.length];
System.arraycopy(a.get(i-1), 0, dest, 0, a.get(i-1).length);
System.arraycopy(hash, 0, dest, a.get(i-1).length, hash.length);
b.clear();
b.add(dest);
}else{
System.out.println(bytesToHex(hash));}
}
}
}catch(Exception e){
e.printStackTrace();
}finally{
if(inStream!=null)
inStream.close();
if(bis!=null)
bis.close();
}
}
final protected static char[] hexArray = "0123456789ABCDEF".toCharArray();
public static String bytesToHex(byte[] bytes) {
char[] hexChars = new char[bytes.length * 2];
for ( int j = 0; j < bytes.length; j++ ) {
int v = bytes[j] & 0xFF;
hexChars[j * 2] = hexArray[v >>> 4];
hexChars[j * 2 + 1] = hexArray[v & 0x0F];
}
return new String(hexChars);
}
}
我在这次尝试中直接将文件读取为1KB块。在这种尝试中,由于某种原因,哈希需要很长时间。
package com.test;
import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.security.MessageDigest;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class ReadFileByteByByte2 {
public static void main(String[] args) throws Exception {
InputStream inStream = null;
BufferedInputStream bis = null;
try{
inStream = new FileInputStream("C:\\aa.mp4");
bis = new BufferedInputStream(inStream);
int numByte = bis.available();
System.out.println(numByte/1024);
ArrayList<byte[]> a = new ArrayList<>();
ArrayList<byte[]> b = new ArrayList<>();
byte[] buf = new byte[numByte];
int ii=0;
while(bis.read(buf, ii, 1024)!=-1){
a.add(buf);
}
System.out.println(a.size());
for(int i=a.size()-1,j=-1;i>=0;i--,j++){
MessageDigest digest = MessageDigest.getInstance("SHA-256");
if(i==a.size()-1){
System.out.println(a.get(i).toString());
byte[] hash = digest.digest(a.get(i));
byte[] dest = new byte[a.get(i).length+hash.length];
System.arraycopy(a.get(i-1), 0, dest, 0, a.get(i-1).length);
System.arraycopy(hash, 0, dest, a.get(i-1).length, hash.length);
b.add(dest);
}
else{
System.out.println(i);
byte[] hash = digest.digest(b.get(0));
if(i!=0){
byte[] dest = new byte[a.get(i-1).length+hash.length];
System.arraycopy(a.get(i-1), 0, dest, 0, a.get(i-1).length);
System.arraycopy(hash, 0, dest, a.get(i-1).length, hash.length);
b.clear();
b.add(dest);
}else{
System.out.println(bytesToHex(hash));}
}
}
}catch(Exception e){
e.printStackTrace();
}finally{
if(inStream!=null)
inStream.close();
if(bis!=null)
bis.close();
}
}
final protected static char[] hexArray = "0123456789ABCDEF".toCharArray();
public static String bytesToHex(byte[] bytes) {
char[] hexChars = new char[bytes.length * 2];
for ( int j = 0; j < bytes.length; j++ ) {
int v = bytes[j] & 0xFF;
hexChars[j * 2] = hexArray[v >>> 4];
hexChars[j * 2 + 1] = hexArray[v & 0x0F];
}
return new String(hexChars);
}
}
非常感谢任何帮助。提前谢谢。
答案 0 :(得分:2)
首先,你必须使用DataInputStream.readFully()
来确保你真的获得1k块,并确保如果它比其他块短,则不要在最后一块上使用它。 read()
不保证填充缓冲区,或返回任何大于1的计数。见Javadoc。
其次,你是在滥用available().
它没有做你正在使用它的内容:它告诉你可以在不阻塞的情况下读取多少字节。它无效作为EOS测试,也不是获取流的长度的手段。见Javadoc。在这种情况下,您根本不需要它,只需File.length().
第三,你根本不需要将块的哈希附加到块中,因此可以计算下一个哈希值。只需在块数据上调用digest.update()
,然后digest.doFinal()
提供前一个哈希作为参数,您将获得完全相同的值。
第四,我想知道你是否正确理解了你的要求。计算向前方向的哈希值会更有意义。然后你根本不需要将整个文件读入内存。增加的完整性是相同的。