initial commit

Author: 317787106

framework/src/main/java/org/tron/common/logsfilter/DesensitizedConverter.java

@@ -0,0 +1,50 @@
+package org.tron.common.logsfilter;
+
+import ch.qos.logback.classic.pattern.ClassicConverter;
+import ch.qos.logback.classic.spi.ILoggingEvent;
+import com.google.common.cache.Cache;
+import com.google.common.cache.CacheBuilder;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+import lombok.extern.slf4j.Slf4j;
+import org.tron.core.config.args.Args;
+
+@Slf4j(topic = "Parser")
+public class DesensitizedConverter extends ClassicConverter {
+
+  private static final int SENSITIVE_WORD_SIZE = 1_000;
+
+  private static final Pattern pattern = Pattern.compile(
+      "/(((25[0-5]|2[0-4]\\d|((1\\d{2})|([1-9]?\\d)))\\.){3}(25[0-5]|2[0-4]\\d|((1\\d{2})|"
+          + "([1-9]?\\d))))");
+
+  private static final Cache<String, String> sensitiveCache = CacheBuilder.newBuilder()
+      .maximumSize(SENSITIVE_WORD_SIZE)
+      .recordStats().build();
+
+  public static void addSensitive(String key, String value) {
+    sensitiveCache.put(key, value);
+  }
+
+  public String desensitization(String content) {
+    if (sensitiveCache.size() > 0) {
+      Matcher matcher = pattern.matcher(content);
+      while (matcher.find()) {
+        String key = matcher.group();
+        String value = sensitiveCache.getIfPresent(key);
+        if (value != null) {
+          content = content.replaceAll(key, value);
+        } else {
+          content = content.replaceAll(key, "unknown");
+        }
+      }
+    }
+    return content;
+  }
+
+  @Override
+  public String convert(ILoggingEvent iLoggingEvent) {
+    String source = iLoggingEvent.getFormattedMessage();
+    return Args.getInstance().isFastForward() ? desensitization(source) : source;
+  }
+}

framework/src/main/java/org/tron/core/net/service/relay/RelayService.java

@@ -18,8 +18,11 @@
 import org.tron.common.crypto.SignInterface;
 import org.tron.common.crypto.SignUtils;
 import org.tron.common.es.ExecutorServiceManager;
+import org.tron.common.logsfilter.DesensitizedConverter;
 import org.tron.common.parameter.CommonParameter;
 import org.tron.common.utils.ByteArray;
+import org.tron.common.utils.ByteUtil;
+import org.tron.common.utils.DecodeUtil;
 import org.tron.common.utils.Sha256Hash;
 import org.tron.core.ChainBaseManager;
 import org.tron.core.capsule.TransactionCapsule;
@@ -156,6 +159,10 @@ public boolean checkHelloMessage(HelloMessage message, Channel channel) {
       }
       if (flag) {
         TronNetService.getP2pConfig().getTrustNodes().add(channel.getInetAddress());
+        byte[] addressByte = ByteUtil.merge(new byte[] {DecodeUtil.addressPreFixByte},
+            msg.getAddress().toByteArray());
+        DesensitizedConverter.addSensitive(channel.getInetAddress().toString(),
+            ByteArray.toHexString(addressByte));
       }
       return flag;
     } catch (Exception e) {

framework/src/main/resources/logback.xml

@@ -4,6 +4,8 @@
 
   <!-- Be sure to flush latest logs on exit -->
   <shutdownHook class="org.tron.core.config.TronLogShutdownHook"/>
+  <conversionRule conversionWord="m"
+    converterClass="org.tron.common.logsfilter.DesensitizedConverter"/>
 
   <appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender">
     <encoder>

framework/src/test/java/org/tron/common/logsfilter/DesensitizedConverterTest.java

@@ -0,0 +1,22 @@
+package org.tron.common.logsfilter;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+public class DesensitizedConverterTest {
+
+  @Test
+  public void testReplace() {
+    DesensitizedConverter converter = new DesensitizedConverter();
+    DesensitizedConverter.addSensitive("/192.168.1.10", "address1");
+    DesensitizedConverter.addSensitive("/197.168.1.10", "address2");
+
+    String logStr1 = "This is test log /192.168.1.10:100, /197.168.1.10:200, /197.168.1.10:100";
+    Assert.assertEquals("This is test log address1:100, address2:200, address2:100",
+        converter.desensitization(logStr1));
+
+    String logStr2 = "This is test log /192.168.1.100:100, /197.168.1.10:200, /197.168.1.10:100";
+    Assert.assertEquals("This is test log unknown:100, address2:200, address2:100",
+        converter.desensitization(logStr2));
+  }
+}