PD Guard: модуль безопасности персональных данных

2026-09-22 17:43:33 +03:00
commit 41052fac6a
80 changed files with 91230 additions and 0 deletions
@@ -0,0 +1,82 @@
+package ru.pdguard.mask;
+
+import org.springframework.stereotype.Component;
+import ru.pdguard.detect.RuleRegistry;
+
+import java.util.Map;
+import java.util.function.UnaryOperator;
+
+/**
+ * Превращает найденное значение в замену согласно настройкам системы.
+ *
+ * <p>Тип, для которого вид маски не задан, скрывается звёздочками целиком —
+ * безопасное поведение по умолчанию для вновь добавленных правил.
+ */
+@Component
+public class Masker {
+
+    private static final UnaryOperator<String> EDGES = v -> Strategies.keepEdges(v, 2, 2);
+    private static final UnaryOperator<String> SHORT_SERIES = v -> Strategies.keepEdges(v, 0, 2);
+
+    private static final Map<String, UnaryOperator<String>> BY_TYPE = Map.ofEntries(
+            Map.entry(RuleRegistry.EMAIL, Strategies::email),
+            Map.entry(RuleRegistry.PHONE, EDGES),
+            Map.entry(RuleRegistry.CARD, EDGES),
+            Map.entry(RuleRegistry.INN, EDGES),
+            Map.entry(RuleRegistry.SNILS, EDGES),
+            Map.entry(RuleRegistry.PASSPORT, EDGES),
+            Map.entry(RuleRegistry.DRIVER_LICENSE, EDGES),
+            Map.entry(RuleRegistry.DEPT_CODE, EDGES),
+            // У этих документов серия короткая — две цифры или две буквы. Оставь мы
+            // первые два знака, серия оказалась бы открыта целиком, поэтому видны
+            // только последние. У паспорта РФ и водительского удостоверения серия
+            // из четырёх знаков, там открывается половина.
+            Map.entry(RuleRegistry.FOREIGN_PASSPORT, SHORT_SERIES),
+            Map.entry(RuleRegistry.MILITARY_ID, SHORT_SERIES),
+            Map.entry(RuleRegistry.BIRTH_CERTIFICATE, SHORT_SERIES),
+            Map.entry(RuleRegistry.MEDICAL_POLICY, EDGES),
+            Map.entry(RuleRegistry.CARDHOLDER, Strategies::initials),
+            Map.entry(RuleRegistry.FIO, Strategies::initials),
+
+            // Код проверки и пин-код не показываем даже частично: у них слишком
+            // мало знаков, чтобы открывать хотя бы один.
+            Map.entry(RuleRegistry.CVV, Strategies::stars),
+            Map.entry(RuleRegistry.PIN, Strategies::stars),
+
+            Map.entry(RuleRegistry.PASSPORT_ISSUER, Strategies::stars),
+
+            // У дат сохраняем разделители: модель видит, что это дата, но не какая.
+            Map.entry(RuleRegistry.BIRTH_DATE, Strategies::starsKeepingPunctuation),
+            Map.entry(RuleRegistry.PASSPORT_DATE, Strategies::starsKeepingPunctuation),
+            Map.entry(RuleRegistry.DATE, Strategies::starsKeepingPunctuation),
+
+            Map.entry(RuleRegistry.ADDRESS_COUNTRY, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_POSTCODE, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_CITY, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_STREET, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_HOUSE, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_FLAT, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_REGION, Strategies::stars),
+            Map.entry(RuleRegistry.ADDRESS_DISTRICT, Strategies::stars),
+            Map.entry(RuleRegistry.BIRTH_PLACE, Strategies::stars),
+            Map.entry(RuleRegistry.CITIZENSHIP, Strategies::stars),
+
+            Map.entry(RuleRegistry.ACCOUNT_NUMBER, EDGES),
+            Map.entry(RuleRegistry.OGRN, EDGES),
+            Map.entry(RuleRegistry.OGRNIP, EDGES),
+            Map.entry(RuleRegistry.KPP, EDGES),
+            // Срок действия карты — разделитель виден, сам месяц/год нет.
+            Map.entry(RuleRegistry.CARD_EXPIRY, Strategies::starsKeepingPunctuation),
+            Map.entry(RuleRegistry.BIK, Strategies::stars),
+            Map.entry(RuleRegistry.INCOME, Strategies::stars),
+            Map.entry(RuleRegistry.BIOMETRIC, Strategies::stars)
+    );
+
+    public String mask(String type, String value, MaskMode mode, MaskContext context) {
+        return switch (mode) {
+            case MASK -> BY_TYPE.getOrDefault(type, Strategies::stars).apply(value);
+            case TOKEN -> context.resolve(type, value, (t, n) -> "[" + t + "_" + n + "]");
+            case SYNTHETIC -> context.resolve(type, value, (t, n) -> Synthetic.forType(t, value, n));
+        };
+    }
+}