Examples with StepConfigGraphNode org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode used on opensource projects

Example 1 with StepConfigGraphNode

use of org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode in project carbon-identity-framework by wso2.

the class JsGraphBuilder method executeStep.

/**
 * Adds the step given by step ID tp the authentication graph.
 *
 * @param stepId Step Id
 * @param params params
 */
@SuppressWarnings("unchecked")
public final void executeStep(int stepId, Object... params) {
    StepConfig stepConfig;
    stepConfig = stepNamedMap.get(stepId);
    if (stepConfig == null) {
        log.error("Given Authentication Step :" + stepId + " is not in Environment");
        return;
    }
    StepConfigGraphNode newNode = wrap(stepConfig);
    if (currentNode == null) {
        result.setStartNode(newNode);
    } else {
        attachToLeaf(currentNode, newNode);
    }
    currentNode = newNode;
    if (params.length > 0) {
        // if there are any params provided, last one is assumed to be the event listeners
        if (params[params.length - 1] instanceof Map) {
            attachEventListeners((Map<String, Object>) params[params.length - 1]);
        } else {
            log.error("Invalid argument and hence ignored. Last argument should be a Map of event listeners.");
        }
    }
    if (params.length == 2) {
        // There is an argument with options present
        if (params[0] instanceof Map) {
            Map<String, Object> options = (Map<String, Object>) params[0];
            handleOptions(options, stepConfig);
        }
    }
}

Example 2 with StepConfigGraphNode

use of org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode in project carbon-identity-framework by wso2.

the class JsGraphBuilder method executeStepInAsyncEvent.

/**
 * Adds the step given by step ID tp the authentication graph.
 *
 * @param params params
 */
@SuppressWarnings("unchecked")
public void executeStepInAsyncEvent(int stepId, Object... params) {
    AuthenticationContext context = contextForJs.get();
    AuthGraphNode currentNode = dynamicallyBuiltBaseNode.get();
    if (log.isDebugEnabled()) {
        log.debug("Execute Step on async event. Step ID : " + stepId);
    }
    AuthenticationGraph graph = context.getSequenceConfig().getAuthenticationGraph();
    if (graph == null) {
        log.error("The graph happens to be null on the sequence config. Can not execute step : " + stepId);
        return;
    }
    StepConfig stepConfig = graph.getStepMap().get(stepId);
    if (stepConfig == null) {
        if (log.isDebugEnabled()) {
            log.debug("The stepConfig of the step ID : " + stepId + " is null");
        }
        return;
    }
    // Inorder to keep original stepConfig as a backup in AuthenticationGraph.
    StepConfig clonedStepConfig = new StepConfig(stepConfig);
    StepConfig stepConfigFromContext = null;
    if (MapUtils.isNotEmpty(context.getSequenceConfig().getStepMap())) {
        stepConfigFromContext = context.getSequenceConfig().getStepMap().values().stream().filter(contextStepConfig -> (stepConfig.getOrder() == contextStepConfig.getOrder())).findFirst().orElse(null);
    }
    clonedStepConfig.applyStateChangesToNewObjectFromContextStepMap(stepConfigFromContext);
    if (log.isDebugEnabled()) {
        log.debug("Found step for the Step ID : " + stepId + ", Step Config " + clonedStepConfig);
    }
    StepConfigGraphNode newNode = wrap(clonedStepConfig);
    if (currentNode == null) {
        if (log.isDebugEnabled()) {
            log.debug("Setting a new node at the first time. Node : " + newNode.getName());
        }
        dynamicallyBuiltBaseNode.set(newNode);
    } else {
        attachToLeaf(currentNode, newNode);
    }
    if (params.length > 0) {
        // if there is only one param, it is assumed to be the event listeners
        if (params[params.length - 1] instanceof Map) {
            attachEventListeners((Map<String, Object>) params[params.length - 1], newNode);
        } else {
            log.error("Invalid argument and hence ignored. Last argument should be a Map of event listeners.");
        }
    }
    if (params.length == 2) {
        // There is an argument with options present
        if (params[0] instanceof Map) {
            Map<String, Object> options = (Map<String, Object>) params[0];
            handleOptions(options, clonedStepConfig);
        }
    }
}

Example 3 with StepConfigGraphNode

use of org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode in project carbon-identity-framework by wso2.

the class JsGraphBuilderTest method testCreateJavascript.

@Test
public void testCreateJavascript() throws Exception {
    String script = "var onLoginRequest = function(context) { executeStep(1, { onSuccess : function(context) {" + "executeStep(2);}})};";
    ServiceProvider sp1 = getTestServiceProvider("js-sp-1.xml");
    AuthenticationContext context = getAuthenticationContext(sp1);
    Map<Integer, StepConfig> stepConfigMap = new HashMap<>();
    stepConfigMap.put(1, new StepConfig());
    stepConfigMap.put(2, new StepConfig());
    JsGraphBuilder jsGraphBuilder = jsGraphBuilderFactory.createBuilder(context, stepConfigMap);
    jsGraphBuilder.createWith(script);
    AuthenticationGraph graph = jsGraphBuilder.build();
    assertNotNull(graph.getStartNode());
    assertTrue(graph.getStartNode() instanceof StepConfigGraphNode);
    StepConfigGraphNode firstStep = (StepConfigGraphNode) graph.getStartNode();
    assertNotNull(firstStep.getNext());
    assertTrue(firstStep.getNext() instanceof DynamicDecisionNode);
}

Example 4 with StepConfigGraphNode

use of org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode in project carbon-identity-framework by wso2.

the class JsGraphBuilderTest method testCreateDirectJava.

@Test
public void testCreateDirectJava() throws Exception {
    ServiceProvider sp1 = getTestServiceProvider("js-sp-1.xml");
    AuthenticationContext context = getAuthenticationContext(sp1);
    Map<Integer, StepConfig> stepConfigMap = new HashMap<>();
    stepConfigMap.put(1, new StepConfig());
    stepConfigMap.put(2, new StepConfig());
    JsGraphBuilder jsGraphBuilder = jsGraphBuilderFactory.createBuilder(context, stepConfigMap);
    jsGraphBuilder.executeStep(1);
    jsGraphBuilder.executeStep(2);
    AuthenticationGraph graph = jsGraphBuilder.build();
    assertNotNull(graph.getStartNode());
    assertTrue(graph.getStartNode() instanceof StepConfigGraphNode);
    StepConfigGraphNode firstStep = (StepConfigGraphNode) graph.getStartNode();
    assertNotNull(firstStep.getNext());
    assertTrue(firstStep.getNext() instanceof StepConfigGraphNode);
}

Example 5 with StepConfigGraphNode

use of org.wso2.carbon.identity.application.authentication.framework.config.model.graph.StepConfigGraphNode in project carbon-identity-framework by wso2.

the class GraphBasedSequenceHandler method handleAuthenticationStep.

private boolean handleAuthenticationStep(HttpServletRequest request, HttpServletResponse response, AuthenticationContext context, SequenceConfig sequenceConfig, StepConfigGraphNode stepConfigGraphNode) throws FrameworkException {
    StepConfig stepConfig = stepConfigGraphNode.getStepConfig();
    if (stepConfig == null) {
        throw new FrameworkException("StepConfig not found while handling the step. Service Provider : " + context.getServiceProviderName());
    }
    // if the current step is completed
    if (stepConfig.isCompleted()) {
        stepConfig.setCompleted(false);
        stepConfig.setRetrying(false);
        // if the request didn't fail during the step execution
        if (context.isRequestAuthenticated()) {
            if (log.isDebugEnabled()) {
                log.debug("Step " + stepConfig.getOrder() + " is completed. Going to get the next one.");
            }
        } else {
            if (log.isDebugEnabled()) {
                log.debug("Authentication has failed in the Step " + (context.getCurrentStep()));
            }
            // authentication
            if (stepConfig.isMultiOption() && !context.isPassiveAuthenticate()) {
                stepConfig.setRetrying(true);
                context.setRequestAuthenticated(true);
            } else {
                FrameworkUtils.resetAuthenticationContext(context);
            }
        }
        FrameworkUtils.resetAuthenticationContext(context);
    }
    // if the sequence is not completed, we have work to do.
    if (log.isDebugEnabled()) {
        log.debug("Starting Step: " + stepConfig.getOrder());
    }
    AuthenticatorFlowStatus flowStatus = (AuthenticatorFlowStatus) request.getAttribute(FrameworkConstants.RequestParams.FLOW_STATUS);
    int stepNumber = context.getCurrentStep();
    if (!context.isReturning()) {
        if (stepNumber <= 0) {
            stepNumber = 1;
        } else if (flowStatus != FAIL_COMPLETED) {
            stepNumber++;
        }
        context.setCurrentStep(stepNumber);
        context.getSequenceConfig().getStepMap().put(stepNumber, stepConfig);
    }
    FrameworkUtils.getStepHandler().handle(request, response, context);
    flowStatus = (AuthenticatorFlowStatus) request.getAttribute(FrameworkConstants.RequestParams.FLOW_STATUS);
    if (flowStatus != SUCCESS_COMPLETED && flowStatus != INCOMPLETE && !(FAIL_COMPLETED.equals(flowStatus) && context.isRetrying())) {
        stepConfig.setSubjectAttributeStep(false);
        stepConfig.setSubjectIdentifierStep(false);
    }
    if (flowStatus == FAIL_COMPLETED) {
        if (!(stepConfigGraphNode.getNext() instanceof DynamicDecisionNode)) {
            if (context.isRetrying()) {
                StepConfigGraphNode newNextNode = new StepConfigGraphNode(stepConfigGraphNode.getStepConfig());
                newNextNode.setNext(stepConfigGraphNode.getNext());
                AuthGraphNode parentNode = stepConfigGraphNode.getParent();
                if (parentNode == null) {
                    parentNode = sequenceConfig.getAuthenticationGraph().getStartNode();
                }
                newNextNode.setParent(parentNode);
                if (parentNode instanceof DynamicDecisionNode) {
                    ((DynamicDecisionNode) parentNode).setDefaultEdge(newNextNode);
                } else if (parentNode instanceof StepConfigGraphNode) {
                    ((StepConfigGraphNode) parentNode).setNext(newNextNode);
                }
                stepConfigGraphNode.setNext(newNextNode);
            } else {
                stepConfigGraphNode.setNext(new FailNode());
            }
        }
    }
    // if step is not completed, that means step wants to redirect to outside
    if (!stepConfig.isCompleted()) {
        if (log.isDebugEnabled()) {
            log.debug("Step is not complete yet. Redirecting to outside.");
        }
        return true;
    }
    if (context.isPassiveAuthenticate() && !context.isRequestAuthenticated()) {
        return true;
    }
    context.setReturning(false);
    return false;
}