blob: 7262964688921087cd4b30a0ad509f50a761b98f (
plain)
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
|
/*
* Copyright (C) 2012 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.ide.eclipse.ddms.systrace;
import com.android.ddmlib.IDevice;
import com.android.ddmlib.IShellOutputReceiver;
import com.google.common.primitives.Bytes;
public class SystraceTask implements Runnable {
private final IDevice mDevice;
private final String mOptions;
private volatile boolean mCancel;
private final Object mLock = new Object();
private String errorMessage;
private boolean mTraceComplete;
private byte[] mBuffer = new byte[1024];
private int mDataLength = 0;
public SystraceTask(IDevice device, String options) {
mDevice = device;
mOptions = options;
}
@Override
public void run() {
try {
mDevice.executeShellCommand("atrace " + mOptions, new Receiver(), 0);
} catch (Exception e) {
synchronized (mLock) {
errorMessage = "Unexpected error while running atrace on device: " + e;
}
}
}
public void cancel() {
mCancel = true;
}
public String getError() {
synchronized (mLock) {
return errorMessage;
}
}
public byte[] getAtraceOutput() {
synchronized (mLock) {
return mTraceComplete ? mBuffer : null;
}
}
private class Receiver implements IShellOutputReceiver {
@Override
public void addOutput(byte[] data, int offset, int length) {
synchronized (mLock) {
if (mDataLength + length > mBuffer.length) {
mBuffer = Bytes.ensureCapacity(mBuffer, mDataLength + length + 1, 1024);
}
for (int i = 0; i < length; i++) {
mBuffer[mDataLength + i] = data[offset + i];
}
mDataLength += length;
}
}
@Override
public void flush() {
synchronized (mLock) {
// trim mBuffer to its final size
byte[] copy = new byte[mDataLength];
for (int i = 0; i < mDataLength; i++) {
copy[i] = mBuffer[i];
}
mBuffer = copy;
mTraceComplete = true;
}
}
@Override
public boolean isCancelled() {
return mCancel;
}
}
}
|
