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 }