Examples with InsufficientComplexityException com.idealista.tlsh.exceptions.InsufficientComplexityException used on opensource projects

Example 1 with InsufficientComplexityException

use of com.idealista.tlsh.exceptions.InsufficientComplexityException in project nifi by apache.

the class FuzzyHashContent method onTrigger.

@Override
public void onTrigger(final ProcessContext context, final ProcessSession session) throws ProcessException {
    FlowFile flowFile = session.get();
    if (flowFile == null) {
        return;
    }
    final ComponentLog logger = getLogger();
    String algorithm = context.getProperty(HASH_ALGORITHM).getValue();
    if (checkMinimumAlgorithmRequirements(algorithm, flowFile) == false) {
        logger.error("The content of '{}' is smaller than the minimum required by {}, routing to failure", new Object[] { flowFile, algorithm });
        session.transfer(flowFile, REL_FAILURE);
        return;
    }
    final AtomicReference<String> hashValueHolder = new AtomicReference<>(null);
    try {
        session.read(flowFile, new InputStreamCallback() {

            @Override
            public void process(final InputStream in) throws IOException {
                try (ByteArrayOutputStream holder = new ByteArrayOutputStream()) {
                    StreamUtils.copy(in, holder);
                    String hashValue = generateHash(algorithm, holder.toString());
                    if (StringUtils.isBlank(hashValue) == false) {
                        hashValueHolder.set(hashValue);
                    }
                }
            }
        });
        final String attributeName = context.getProperty(ATTRIBUTE_NAME).getValue();
        flowFile = session.putAttribute(flowFile, attributeName, hashValueHolder.get());
        logger.info("Successfully added attribute '{}' to {} with a value of {}; routing to success", new Object[] { attributeName, flowFile, hashValueHolder.get() });
        session.getProvenanceReporter().modifyAttributes(flowFile);
        session.transfer(flowFile, REL_SUCCESS);
    } catch (final InsufficientComplexityException | ProcessException e) {
        logger.error("Failed to process {} due to {}; routing to failure", new Object[] { flowFile, e });
        session.transfer(flowFile, REL_FAILURE);
    }
}