001/*
002 *  Licensed to the Apache Software Foundation (ASF) under one or more
003 *  contributor license agreements.  See the NOTICE file distributed with
004 *  this work for additional information regarding copyright ownership.
005 *  The ASF licenses this file to You under the Apache License, Version 2.0
006 *  (the "License"); you may not use this file except in compliance with
007 *  the License.  You may obtain a copy of the License at
008 *
009 *     http://www.apache.org/licenses/LICENSE-2.0
010 *
011 *  Unless required by applicable law or agreed to in writing, software
012 *  distributed under the License is distributed on an "AS IS" BASIS,
013 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
014 *  See the License for the specific language governing permissions and
015 *  limitations under the License.
016 */
017package org.apache.commons.compress.harmony.unpack200.bytecode;
018
019import java.io.DataOutputStream;
020import java.io.IOException;
021import java.util.ArrayList;
022import java.util.List;
023
024/**
025 * Inner classes class file attribute
026 */
027public class InnerClassesAttribute extends Attribute {
028
029    private static final class InnerClassesEntry {
030
031        CPClass innerClassInfo;
032        CPClass outerClassInfo;
033        CPUTF8 innerClassName;
034
035        int innerClassInfoIndex = -1;
036        int outerClassInfoIndex = -1;
037        int innerNameIndex = -1;
038        int innerClassAccessFlags = -1;
039
040        InnerClassesEntry(final CPClass innerClass, final CPClass outerClass, final CPUTF8 innerName, final int flags) {
041            this.innerClassInfo = innerClass;
042            this.outerClassInfo = outerClass;
043            this.innerClassName = innerName;
044            this.innerClassAccessFlags = flags;
045        }
046
047        /**
048         * Determine the indices of the things in the receiver which point to elements of the ClassConstantPool
049         *
050         * @param pool ClassConstantPool which holds the CPClass and CPUTF8 objects.
051         */
052        public void resolve(final ClassConstantPool pool) {
053            if (innerClassInfo != null) {
054                innerClassInfo.resolve(pool);
055                innerClassInfoIndex = pool.indexOf(innerClassInfo);
056            } else {
057                innerClassInfoIndex = 0;
058            }
059
060            if (innerClassName != null) {
061                innerClassName.resolve(pool);
062                innerNameIndex = pool.indexOf(innerClassName);
063            } else {
064                innerNameIndex = 0;
065            }
066
067            if (outerClassInfo != null) {
068                outerClassInfo.resolve(pool);
069                outerClassInfoIndex = pool.indexOf(outerClassInfo);
070            } else {
071                outerClassInfoIndex = 0;
072            }
073        }
074
075        public void write(final DataOutputStream dos) throws IOException {
076            dos.writeShort(innerClassInfoIndex);
077            dos.writeShort(outerClassInfoIndex);
078            dos.writeShort(innerNameIndex);
079            dos.writeShort(innerClassAccessFlags);
080        }
081
082    }
083
084    private static CPUTF8 attributeName;
085
086    public static void setAttributeName(final CPUTF8 cpUTF8Value) {
087        attributeName = cpUTF8Value;
088    }
089
090    private final List<InnerClassesEntry> innerClasses = new ArrayList<>();
091    private final List<ConstantPoolEntry> nestedClassFileEntries = new ArrayList<>();
092
093    public InnerClassesAttribute(final String name) {
094        super(attributeName);
095        nestedClassFileEntries.add(getAttributeName());
096    }
097
098    public void addInnerClassesEntry(final CPClass innerClass, final CPClass outerClass, final CPUTF8 innerName, final int flags) {
099        if (innerClass != null) {
100            nestedClassFileEntries.add(innerClass);
101        }
102        if (outerClass != null) {
103            nestedClassFileEntries.add(outerClass);
104        }
105        if (innerName != null) {
106            nestedClassFileEntries.add(innerName);
107        }
108        addInnerClassesEntry(new InnerClassesEntry(innerClass, outerClass, innerName, flags));
109    }
110
111    private void addInnerClassesEntry(final InnerClassesEntry innerClassesEntry) {
112        innerClasses.add(innerClassesEntry);
113    }
114
115    @Override
116    public boolean equals(final Object obj) {
117        if (this == obj) {
118            return true;
119        }
120        if (!super.equals(obj)) {
121            return false;
122        }
123        if (this.getClass() != obj.getClass()) {
124            return false;
125        }
126        final InnerClassesAttribute other = (InnerClassesAttribute) obj;
127        if (getAttributeName() == null) {
128            if (other.getAttributeName() != null) {
129                return false;
130            }
131        } else if (!getAttributeName().equals(other.getAttributeName())) {
132            return false;
133        }
134        return true;
135    }
136
137    @Override
138    protected int getLength() {
139        return 2 + (2 + 2 + 2 + 2) * innerClasses.size();
140    }
141
142    @Override
143    protected ClassFileEntry[] getNestedClassFileEntries() {
144        return nestedClassFileEntries.toArray(NONE);
145    }
146
147    @Override
148    public int hashCode() {
149        final int PRIME = 31;
150        int result = super.hashCode();
151        result = PRIME * result + (getAttributeName() == null ? 0 : getAttributeName().hashCode());
152        return result;
153    }
154
155    @Override
156    protected void resolve(final ClassConstantPool pool) {
157        super.resolve(pool);
158        for (final InnerClassesEntry entry : innerClasses) {
159            entry.resolve(pool);
160        }
161    }
162
163    @Override
164    public String toString() {
165        return "InnerClasses: " + getAttributeName();
166    }
167
168    @Override
169    protected void writeBody(final DataOutputStream dos) throws IOException {
170        dos.writeShort(innerClasses.size());
171
172        for (final InnerClassesEntry entry : innerClasses) {
173            entry.write(dos);
174        }
175    }
176}