Skip to content

Commit 6535639

Browse files
Merge pull request #179 from AikidoSec/debug-level-trace-enable-bytebuddy-logs
Allow logLevel = TRACE to enable the bytebuddy debug logs
2 parents da41b5f + c5bca09 commit 6535639

2 files changed

Lines changed: 11 additions & 2 deletions

File tree

agent/src/main/java/dev/aikido/agent/ByteBuddyInitializer.java

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
package dev.aikido.agent;
22

3+
import dev.aikido.agent_api.helpers.logging.LogManager;
4+
import dev.aikido.agent_api.helpers.logging.Logger;
35
import net.bytebuddy.ByteBuddy;
46
import net.bytebuddy.agent.builder.AgentBuilder;
57
import net.bytebuddy.dynamic.VisibilityBridgeStrategy;
@@ -8,12 +10,15 @@
810
import net.bytebuddy.matcher.ElementMatchers;
911

1012
public final class ByteBuddyInitializer {
11-
13+
private static final Logger logger = LogManager.getLogger(ByteBuddyInitializer.class);
1214
public static AgentBuilder createAgentBuilder() {
13-
return createAgentBuilder(false); // Set debug mode to false by default
15+
// byte buddy debug mode should be linked to the trace logs, since we rarely want to inspect them.
16+
boolean debugMode = logger.logsTraceLogs();
17+
return createAgentBuilder(debugMode);
1418
}
1519

1620
public static AgentBuilder createAgentBuilder(boolean debugMode) {
21+
logger.debug("Creating new ByteBuddy agent, with debugMode: %s", debugMode);
1722
AgentBuilder agentBuilder = new AgentBuilder.Default(
1823
// default method graph compiler inspects the class hierarchy, we don't need it, so
1924
// we use a simpler and faster strategy instead
@@ -42,6 +47,7 @@ public static AgentBuilder createAgentBuilder(boolean debugMode) {
4247

4348
agentBuilder = agentBuilder.with(AgentBuilder.TypeStrategy.Default.DECORATE);
4449

50+
logger.trace("Finished creating new ByteBuddy agent: %s", agentBuilder);
4551
return agentBuilder;
4652
}
4753
}

agent_api/src/main/java/dev/aikido/agent_api/helpers/logging/Logger.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,4 +106,7 @@ public void error(Object message, Object... args) {
106106
public void fatal(Object message, Object... args) {
107107
log(LogLevel.FATAL, message, args);
108108
}
109+
public boolean logsTraceLogs() {
110+
return LogLevel.TRACE.getLevel() >= logLevel.getLevel();
111+
}
109112
}

0 commit comments

Comments
 (0)