moonlight-stream/moonlight-android
1
0
Fork 0
mirror of https://github.com/moonlight-stream/moonlight-android.git synced 2025-07-19 19:13:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add double-free detection code to the pools

Browse Source
This commit is contained in:
Cameron Gutman 2013-11-21 12:55:05 -05:00
parent 7951c0fa13
commit acb3dc8881
3 changed files with 50 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/com/limelight/nvstream/av/AvByteBufferPool.java
View File

@ -6,6 +6,8 @@ public class AvByteBufferPool {
private ConcurrentLinkedQueue<byte[]> bufferList = new ConcurrentLinkedQueue<byte[]>();
private int bufferSize;
private static final boolean doubleFreeDebug = true;
public AvByteBufferPool(int size)
{
this.bufferSize = size;
@ -18,7 +20,13 @@ public class AvByteBufferPool {
public byte[] allocate()
{
byte[] buff = bufferList.poll();
byte[] buff;
if (doubleFreeDebug) {
buff = null;
}
else {
buff = bufferList.poll();
}
if (buff == null) {
buff = new byte[bufferSize];
}
@ -27,6 +35,14 @@ public class AvByteBufferPool {
public void free(byte[] buffer)
{
if (doubleFreeDebug) {
for (byte[] buf : bufferList) {
if (buf == buffer) {
throw new IllegalStateException("Double free detected");
}
}
}
bufferList.add(buffer);
}
}

17
src/com/limelight/nvstream/av/AvObjectPool.java
View File

@ -5,6 +5,8 @@ import java.util.concurrent.ConcurrentLinkedQueue;
public class AvObjectPool<T> {
private ConcurrentLinkedQueue<T> objectList = new ConcurrentLinkedQueue<T>();
private static final boolean doubleFreeDebug = true;
public void purge()
{
objectList.clear();
@ -12,11 +14,24 @@ public class AvObjectPool<T> {
public T tryAllocate()
{
return objectList.poll();
if (doubleFreeDebug) {
return null;
}
else {
return objectList.poll();
}
}
public void free(T object)
{
if (doubleFreeDebug) {
for (T obj : objectList) {
if (obj == object) {
throw new IllegalStateException("Double free detected");
}
}
}
objectList.add(object);
}
}

18
src/com/limelight/nvstream/av/AvShortBufferPool.java
View File

@ -6,6 +6,8 @@ public class AvShortBufferPool {
private ConcurrentLinkedQueue<short[]> bufferList = new ConcurrentLinkedQueue<short[]>();
private int bufferSize;
private static final boolean doubleFreeDebug = true;
public AvShortBufferPool(int size)
{
this.bufferSize = size;
@ -18,7 +20,13 @@ public class AvShortBufferPool {
public short[] allocate()
{
short[] buff = bufferList.poll();
short[] buff;
if (doubleFreeDebug) {
buff = null;
}
else {
buff = bufferList.poll();
}
if (buff == null) {
buff = new short[bufferSize];
}
@ -27,6 +35,14 @@ public class AvShortBufferPool {
public void free(short[] buffer)
{
if (doubleFreeDebug) {
for (short[] buf : bufferList) {
if (buf == buffer) {
throw new IllegalStateException("Double free detected");
}
}
}
bufferList.add(buffer);
}
}