/*
* Copyright (c) 2008-2013, Matthias Mann
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* * Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* * Neither the name of Matthias Mann nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package nginx.clojure;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashSet;
import java.util.Locale;
import java.util.concurrent.locks.Lock;
import junit.framework.TestCase;
import nginx.clojure.asm.ClassReader;
import nginx.clojure.asm.ClassWriter;
import nginx.clojure.logger.TinyLogService;
import nginx.clojure.wave.AlreadyInstrumented;
import nginx.clojure.wave.InstrumentClass;
import nginx.clojure.wave.MethodDatabase;
import nginx.clojure.wave.MethodDatabaseUtil;
import org.junit.Test;
/**
* Test to checking blocking call detection
*
* @author Matthias Mann
*/
@AlreadyInstrumented
public class BlockingTest extends TestCase {
@Test
public void testSuspend() throws IOException {
final String className = BlockingTest.class.getName().replace('.', '/');
final HashSet<String> msgs = new HashSet<String>();
msgs.add("Method "+className+"#t_wait()V contains potentially blocking call to java/lang/Object#wait()V");
msgs.add("Method "+className+"#t_sleep1()V contains potentially blocking call to java/lang/Thread#sleep(J)V");
msgs.add("Method "+className+"#t_sleep2()V contains potentially blocking call to java/lang/Thread#sleep(JI)V");
msgs.add("Method "+className+"#t_join1(Ljava/lang/Thread;)V contains potentially blocking call to java/lang/Thread#join()V");
msgs.add("Method "+className+"#t_join2(Ljava/lang/Thread;)V contains potentially blocking call to java/lang/Thread#join(J)V");
msgs.add("Method "+className+"#t_join3(Ljava/lang/Thread;)V contains potentially blocking call to java/lang/Thread#join(JI)V");
msgs.add("Method "+className+"#t_lock1(Ljava/util/concurrent/locks/Lock;)V contains potentially blocking call to java/util/concurrent/locks/Lock#lock()V");
msgs.add("Method "+className+"#t_lock2(Ljava/util/concurrent/locks/Lock;)V contains potentially blocking call to java/util/concurrent/locks/Lock#lockInterruptibly()V");
MethodDatabase db = new MethodDatabase(BlockingTest.class.getClassLoader());
MethodDatabaseUtil.load(db, "nginx/clojure/wave/coroutine-method-db.txt");
db.setAllowBlocking(true);
db.setLog(new TinyLogService() {
@Override
public void warn(String message, Object ...args) {
message = String.format(Locale.ENGLISH, message, args);
assertTrue("Unexpected message: " + message, msgs.remove(message));
}
});
InputStream in = BlockingTest.class.getResourceAsStream("BlockingTest.class");
try {
ClassReader r = new ClassReader(in);
ClassWriter cw = new ClassWriter(0);
InstrumentClass ic = new InstrumentClass(r.getClassName(), MethodDatabaseUtil.buildClassEntryFamily(db, r), cw, db, true);
r.accept(ic, ClassReader.SKIP_FRAMES);
} finally {
in.close();
}
assertTrue("Expected messages not generated: "+msgs.toString(), msgs.isEmpty());
}
public void t_wait() throws SuspendExecution, InterruptedException {
synchronized (this) {
wait();
}
}
public void t_notify() throws SuspendExecution {
synchronized (this) {
notify();
}
}
public void t_sleep1() throws SuspendExecution, InterruptedException {
Thread.sleep(1000);
}
public void t_sleep2() throws SuspendExecution, InterruptedException {
Thread.sleep(1000, 100);
}
public void t_join1(Thread t) throws SuspendExecution, InterruptedException {
t.join();
}
public void t_join2(Thread t) throws SuspendExecution, InterruptedException {
t.join(1000);
}
public void t_join3(Thread t) throws SuspendExecution, InterruptedException {
t.join(1, 100);
}
public void t_lock1(Lock lock) throws SuspendExecution {
lock.lock();
}
public void t_lock2(Lock lock) throws SuspendExecution, InterruptedException {
lock.lockInterruptibly();
}
public void t_lock3() throws SuspendExecution {
lock();
}
public void lock() {
System.out.println("Just a method which have similar signature");
}
}