001/*
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *   https://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing,
013 * software distributed under the License is distributed on an
014 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
015 * KIND, either express or implied.  See the License for the
016 * specific language governing permissions and limitations
017 * under the License.
018 */
019
020package org.apache.bcel.classfile;
021
022import java.io.DataInput;
023import java.io.DataOutputStream;
024import java.io.IOException;
025
026import org.apache.bcel.Const;
027import org.apache.bcel.util.Args;
028
029/**
030 * This class is derived from <em>Attribute</em> and indicates the main class of a module. There may be at most one
031 * ModuleMainClass attribute in a ClassFile structure.
032 *
033 * @see Attribute
034 */
035public final class ModuleMainClass extends Attribute {
036
037    private int mainClassIndex;
038
039    /**
040     * Constructs object from input stream.
041     *
042     * @param nameIndex Index in constant pool.
043     * @param length Content length in bytes.
044     * @param input Input stream.
045     * @param constantPool Array of constants.
046     * @throws IOException Thrown if an I/O error occurs.
047     */
048    ModuleMainClass(final int nameIndex, final int length, final DataInput input, final ConstantPool constantPool) throws IOException {
049        this(nameIndex, length, 0, constantPool);
050        mainClassIndex = input.readUnsignedShort();
051    }
052
053    /**
054     * Constructs a new ModuleMainClass.
055     *
056     * @param nameIndex Index in constant pool.
057     * @param length Content length in bytes.
058     * @param mainClassIndex Host class index.
059     * @param constantPool Array of constants.
060     */
061    public ModuleMainClass(final int nameIndex, final int length, final int mainClassIndex, final ConstantPool constantPool) {
062        super(Const.ATTR_MODULE_MAIN_CLASS, nameIndex, length, constantPool);
063        this.mainClassIndex = Args.requireU2(mainClassIndex, "mainClassIndex");
064    }
065
066    /**
067     * Initialize from another object. Note that both objects use the same references (shallow copy). Use copy() for a
068     * physical copy.
069     *
070     * @param c Source to copy.
071     */
072    public ModuleMainClass(final ModuleMainClass c) {
073        this(c.getNameIndex(), c.getLength(), c.getHostClassIndex(), c.getConstantPool());
074    }
075
076    /**
077     * Called by objects that are traversing the nodes of the tree implicitly defined by the contents of a Java class. I.e.,
078     * the hierarchy of methods, fields, attributes, etc. spawns a tree of objects.
079     *
080     * @param v Visitor object.
081     */
082    @Override
083    public void accept(final Visitor v) {
084        v.visitModuleMainClass(this);
085    }
086
087    /**
088     * @return deep copy of this attribute.
089     */
090    @Override
091    public Attribute copy(final ConstantPool constantPool) {
092        final ModuleMainClass c = (ModuleMainClass) clone();
093        c.setConstantPool(constantPool);
094        return c;
095    }
096
097    /**
098     * Dumps ModuleMainClass attribute to file stream in binary format.
099     *
100     * @param file Output file stream.
101     * @throws IOException Thrown if an I/O error occurs.
102     */
103    @Override
104    public void dump(final DataOutputStream file) throws IOException {
105        super.dump(file);
106        file.writeShort(mainClassIndex);
107    }
108
109    /**
110     * Gets the host class index.
111     *
112     * @return index into constant pool of host class name.
113     */
114    public int getHostClassIndex() {
115        return mainClassIndex;
116    }
117
118    /**
119     * Sets the host class index.
120     *
121     * @param mainClassIndex The host class index.
122     */
123    public void setHostClassIndex(final int mainClassIndex) {
124        this.mainClassIndex = mainClassIndex;
125    }
126
127    /**
128     * @return String representation.
129     */
130    @Override
131    public String toString() {
132        final StringBuilder buf = new StringBuilder();
133        buf.append("ModuleMainClass: ");
134        final String className = super.getConstantPool().getConstantString(mainClassIndex, Const.CONSTANT_Class);
135        buf.append(Utility.compactClassName(className, false));
136        return buf.toString();
137    }
138}