1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*******************************************************************************
* Copyright (c) 2012, 2015 Wind River Systems, Inc. and others. All rights reserved.
* This program and the accompanying materials are made available under the terms
* of the Eclipse Public License v1.0 which accompanies this distribution, and is
* available at http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Wind River Systems - initial API and implementation
*******************************************************************************/
package org.eclipse.tcf.te.tcf.filesystem.core.internal.operations;
import static java.text.MessageFormat.format;
import java.io.BufferedInputStream;
import java.io.InputStream;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.tcf.protocol.IToken;
import org.eclipse.tcf.protocol.Protocol;
import org.eclipse.tcf.services.IFileSystem;
import org.eclipse.tcf.services.IFileSystem.DoneOpen;
import org.eclipse.tcf.services.IFileSystem.FileSystemException;
import org.eclipse.tcf.services.IFileSystem.IFileHandle;
import org.eclipse.tcf.te.tcf.core.concurrent.TCFOperationMonitor;
import org.eclipse.tcf.te.tcf.filesystem.core.internal.FSTreeNode;
import org.eclipse.tcf.te.tcf.filesystem.core.internal.utils.StatusHelper;
import org.eclipse.tcf.te.tcf.filesystem.core.nls.Messages;
import org.eclipse.tcf.util.TCFFileInputStream;
/**
* The operation that computes the digest of the cache file in the background.
*/
public class OpTargetFileDigest extends AbstractOperation {
FSTreeNode node;
byte[] digest;
public OpTargetFileDigest(FSTreeNode node) {
this.node = node;
}
@Override
public IStatus doRun(IProgressMonitor monitor) {
long totalSize = node.getSize();
monitor.beginTask(getName(), 100);
final String path = node.getLocation(true);
final TCFOperationMonitor<InputStream> result = new TCFOperationMonitor<InputStream>();
Protocol.invokeLater(new Runnable() {
@Override
public void run() {
IFileSystem fs = node.getRuntimeModel().getFileSystem();
if (fs == null) {
result.setCancelled();
} else {
tcfGetInputStream(fs, path, result);
}
}
});
IStatus status = result.waitDone(monitor);
if (!status.isOK())
return status;
InputStream in = new BufferedInputStream(result.getValue());
try {
int chunk_size = (int) totalSize / 100;
int percentRead = 0;
long bytesRead = 0;
MessageDigest digest = MessageDigest.getInstance(MD_ALG);
in = new DigestInputStream(in, digest);
// The buffer used to download the file.
byte[] data = new byte[DEFAULT_CHUNK_SIZE];
int length;
while ((length = in.read(data)) >= 0){
bytesRead += length;
if (chunk_size != 0) {
int percent = (int) bytesRead / chunk_size;
if (percent != percentRead) { // Update the progress.
monitor.worked(percent - percentRead);
percentRead = percent; // Remember the percentage.
}
}
if (monitor.isCanceled())
return Status.CANCEL_STATUS;
}
this.digest = digest.digest();
return Status.OK_STATUS;
} catch (Exception e) {
return StatusHelper.createStatus(format(Messages.OpTargetFileDigest_error_download, path), e);
} finally {
if (in != null) {
try {
in.close();
} catch (Exception e) {
}
}
}
}
protected void tcfGetInputStream(IFileSystem fileSystem, final String path, final TCFOperationMonitor<InputStream> result) {
int flags = IFileSystem.TCF_O_READ;
if (!result.checkCancelled()) {
fileSystem.open(path, flags, null, new DoneOpen() {
@Override
public void doneOpen(IToken token, FileSystemException error, IFileHandle handle) {
if (error != null) {
result.setError(format(Messages.OpTargetFileDigest_error_openFile, path), error);
} else {
result.setDone(new TCFFileInputStream(handle));
}
}
});
}
}
public byte[] getDigest() {
return digest;
}
@Override
public String getName() {
return "Update target digest"; //$NON-NLS-1$
}
}
|