proguard.analysis.cpa.defaults.ProgramLocationDependentReachedSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of proguard-core Show documentation
ProGuardCORE is a free library to read, analyze, modify, and write Java class files.
There is a newer version: 9.1.7
/*
 * ProGuardCORE -- library to process Java bytecode.
 *
 * Copyright (c) 2002-2022 Guardsquare NV
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package proguard.analysis.cpa.defaults;

import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.stream.Collectors;
import proguard.classfile.Signature;
import proguard.analysis.cpa.interfaces.AbstractState;
import proguard.analysis.cpa.interfaces.CfaEdge;
import proguard.analysis.cpa.interfaces.CfaNode;
import proguard.analysis.cpa.interfaces.ProgramLocationDependent;
import proguard.analysis.cpa.interfaces.ReachedSet;

/**
 * This {@link ReachedSet} stores {@link ProgramLocationDependent} {@link AbstractState}s.
 * It assumes the analysis does merge the {@link AbstractState}s belonging to different {@link CfaNode}s and stores them in separate bins.
 *
 * @author Dmitry Ivanov
 */
public final class ProgramLocationDependentReachedSet,
                                                      CfaEdgeT extends CfaEdge,
                                                      AbstractStateT extends AbstractState & ProgramLocationDependent,
                                                      SignatureT extends Signature>
    implements ReachedSet
{

    private Map> locationToStates = new HashMap<>();

    // implementations for ReachedSet

    @Override
    public boolean add(AbstractState abstractState)
    {
        final boolean[]     result   = {false};
        Set newValue = new HashSet<>();
        AbstractStateT      state    = (AbstractStateT) abstractState;
        newValue.add(state);
        locationToStates.merge(state.getProgramLocation(), newValue, (oldStates, newStates) ->
        {
            result[0] = oldStates.addAll(newStates);
            return oldStates;
        });
        return result[0];
    }

    @Override
    public boolean addAll(Collection abstractStates)
    {
        return abstractStates.stream().map(this::add).collect(Collectors.toCollection(HashSet::new)).stream().anyMatch(b -> b);
    }

    @Override
    public boolean remove(AbstractState abstractState)
    {
        AtomicBoolean       result   = new AtomicBoolean(false);
        AbstractStateT      state    = (AbstractStateT) abstractState;
        Set newValue = new HashSet<>();
        locationToStates.merge(state.getProgramLocation(), newValue, (oldStates, newStates) ->
        {
            result.set(oldStates.remove(state));
            return oldStates;
        });
        return result.get();
    }

    @Override
    public boolean removeAll(Collection abstractStates)
    {
        return abstractStates.stream().map(state -> remove((AbstractState) state)).collect(Collectors.toCollection(HashSet::new)).stream().anyMatch(b -> b);
    }

    @Override
    public Collection asCollection()
    {
        return locationToStates.values().stream().reduce(new HashSet<>(), (x, y) ->
        {
            x.addAll(y);
            return x;
        });
    }

    @Override
    public Collection getReached(AbstractState abstractState)
    {
        return getReached(((AbstractStateT) abstractState).getProgramLocation());
    }

    /**
     * Returns a collection of abstract states belonging to the given {@code location}.
     */
    public Collection getReached(CfaNodeT location)
    {
        return locationToStates.getOrDefault(location, Collections.emptySet());
    }
}