com.facebook.buck.tools.dxanalysis.DxAnalysisMain.java Source code

Introduction

Here is the source code for com.facebook.buck.tools.dxanalysis.DxAnalysisMain.java

Source

/*
 * Copyright 2014-present Facebook, Inc.
 *
 * 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 com.facebook.buck.tools.dxanalysis;

import com.google.common.collect.ImmutableMap;
import com.google.common.io.ByteStreams;

import org.objectweb.asm.ClassReader;
import org.objectweb.asm.tree.ClassNode;

import java.io.IOException;
import java.util.Collections;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

/**
 * This is a command-line tool for performing ad-hoc static analysis of the DX tool.
 * Specifically, it is intended to find structures that would make it unsafe to run
 * within the buckd process.
 */
public class DxAnalysisMain {
    private static final boolean DEBUG_MUTABILITY = false;

    private DxAnalysisMain() {
    }

    /**
     * @param args  One argument is expected: the path to dx.jar.
     */
    public static void main(String[] args) throws IOException {
        String zipFileName = args[0];
        ImmutableMap<String, ClassNode> allClasses = loadAllClasses(zipFileName);

        MutabilityAnalyzer mutability = MutabilityAnalyzer.analyze(allClasses);
        StaticStateAnalyzer staticState = StaticStateAnalyzer.analyze(allClasses, mutability);
        RegisterSpecAnalyzer registerSpec = RegisterSpecAnalyzer.analyze(allClasses);

        if (DEBUG_MUTABILITY) {
            for (String l : mutability.getLog()) {
                System.out.println(l);
            }
            System.out.println();
            System.out.println();
            System.out.println();

            for (String ic : mutability.getImmutableClasses()) {
                System.out.println("Immutable: " + ic);
            }
            System.out.println();
            System.out.println();
            System.out.println();
        }

        for (String l : staticState.getLog()) {
            System.out.println(l);
        }
        System.out.println();
        System.out.println();
        System.out.println();

        for (String l : registerSpec.getLog()) {
            System.out.println(l);
        }
        System.out.println();
        System.out.println();
        System.out.println();

    }

    private static ImmutableMap<String, ClassNode> loadAllClasses(String zipFileName) throws IOException {
        ImmutableMap.Builder<String, ClassNode> allClassesBuilder = ImmutableMap.builder();
        try (ZipFile inJar = new ZipFile(zipFileName)) {
            for (ZipEntry entry : Collections.list(inJar.entries())) {
                if (!entry.getName().endsWith(".class")) {
                    continue;
                }
                // Skip external libraries.
                if (entry.getName().startsWith("junit/") || entry.getName().startsWith("org/junit/")
                        || entry.getName().startsWith("com/google/common/")) {
                    continue;
                }
                byte[] rawClass = ByteStreams.toByteArray(inJar.getInputStream(entry));
                ClassNode klass = new ClassNode();
                new ClassReader(rawClass).accept(klass, ClassReader.EXPAND_FRAMES);
                allClassesBuilder.put(klass.name, klass);
            }
        }
        return allClassesBuilder.build();
    }
}