View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *   https://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing,
13   * software distributed under the License is distributed on an
14   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15   * KIND, either express or implied.  See the License for the
16   * specific language governing permissions and limitations
17   * under the License.
18   */
19  
20  package org.apache.bcel.classfile;
21  
22  import java.io.DataInput;
23  import java.io.DataOutputStream;
24  import java.io.IOException;
25  import java.util.Arrays;
26  
27  import org.apache.bcel.Const;
28  import org.apache.bcel.util.Args;
29  import org.apache.commons.lang3.ArrayUtils;
30  
31  /**
32   * This class is derived from <em>Attribute</em> and represents the list of packages that are exported or opened by the
33   * Module attribute. There may be at most one ModulePackages attribute in a ClassFile structure.
34   *
35   * @see Attribute
36   */
37  public final class ModulePackages extends Attribute {
38  
39      private int[] packageIndexTable;
40  
41      /**
42       * Constructs object from input stream.
43       *
44       * @param nameIndex Index in constant pool.
45       * @param length Content length in bytes.
46       * @param dataInput Input stream.
47       * @param constantPool Array of constants.
48       * @throws IOException if an I/O error occurs.
49       */
50      ModulePackages(final int nameIndex, final int length, final DataInput dataInput, final ConstantPool constantPool) throws IOException {
51          this(nameIndex, length, (int[]) null, constantPool);
52          packageIndexTable = ClassParser.readU2U2Table(dataInput);
53      }
54  
55      /**
56       * @param nameIndex Index in constant pool.
57       * @param length Content length in bytes.
58       * @param packageIndexTable Table of indices in constant pool.
59       * @param constantPool Array of constants.
60       */
61      public ModulePackages(final int nameIndex, final int length, final int[] packageIndexTable, final ConstantPool constantPool) {
62          super(Const.ATTR_MODULE_PACKAGES, nameIndex, length, constantPool);
63          this.packageIndexTable = ArrayUtils.nullToEmpty(packageIndexTable);
64          Args.requireU2(this.packageIndexTable.length, "packageIndexTable.length");
65      }
66  
67      /**
68       * Initialize from another object. Note that both objects use the same references (shallow copy). Use copy() for a
69       * physical copy.
70       *
71       * @param c Source to copy.
72       */
73      public ModulePackages(final ModulePackages c) {
74          this(c.getNameIndex(), c.getLength(), c.getPackageIndexTable(), c.getConstantPool());
75      }
76  
77      /**
78       * Called by objects that are traversing the nodes of the tree implicitly defined by the contents of a Java class.
79       * I.e., the hierarchy of methods, fields, attributes, etc. spawns a tree of objects.
80       *
81       * @param v Visitor object.
82       */
83      @Override
84      public void accept(final Visitor v) {
85          v.visitModulePackages(this);
86      }
87  
88      /**
89       * @return deep copy of this attribute.
90       */
91      @Override
92      public Attribute copy(final ConstantPool constantPool) {
93          final ModulePackages c = (ModulePackages) clone();
94          if (packageIndexTable != null) {
95              c.packageIndexTable = packageIndexTable.clone();
96          }
97          c.setConstantPool(constantPool);
98          return c;
99      }
100 
101     /**
102      * Dumps ModulePackages attribute to file stream in binary format.
103      *
104      * @param file Output file stream.
105      * @throws IOException if an I/O error occurs.
106      */
107     @Override
108     public void dump(final DataOutputStream file) throws IOException {
109         super.dump(file);
110         file.writeShort(packageIndexTable.length);
111         for (final int index : packageIndexTable) {
112             file.writeShort(index);
113         }
114     }
115 
116     /**
117      * @return Length of package table.
118      */
119     public int getNumberOfPackages() {
120         return packageIndexTable == null ? 0 : packageIndexTable.length;
121     }
122 
123     /**
124      * @return array of indices into constant pool of package names.
125      */
126     public int[] getPackageIndexTable() {
127         return packageIndexTable;
128     }
129 
130     /**
131      * @return string array of package names.
132      */
133     public String[] getPackageNames() {
134         final String[] names = new String[packageIndexTable.length];
135         Arrays.setAll(names, i -> Utility.pathToPackage(super.getConstantPool().getConstantString(packageIndexTable[i], Const.CONSTANT_Package)));
136         return names;
137     }
138 
139     /**
140      * @param packageIndexTable the list of package indexes Also redefines number_of_packages according to table length.
141      */
142     public void setPackageIndexTable(final int[] packageIndexTable) {
143         this.packageIndexTable = ArrayUtils.nullToEmpty(packageIndexTable);
144     }
145 
146     /**
147      * @return String representation, that is, a list of packages.
148      */
149     @Override
150     public String toString() {
151         final StringBuilder buf = new StringBuilder();
152         buf.append("ModulePackages(");
153         buf.append(packageIndexTable.length);
154         buf.append("):\n");
155         for (final int index : packageIndexTable) {
156             final String packageName = super.getConstantPool().getConstantString(index, Const.CONSTANT_Package);
157             buf.append("  ").append(Utility.compactClassName(packageName, false)).append("\n");
158         }
159         return buf.substring(0, buf.length() - 1); // remove the last newline
160     }
161 }