src/java.base/share/classes/java/lang/classfile/MethodBuilder.java - master - wstartsev/jdk

src/java.base/share/classes/java/lang/classfile/MethodBuilder.java
111 строк4 KB
Chen Liang
8361614: Missing sub-int value validation in the Class-File API
29 авг 2025, 17:35
29 авг 2025, 17:35ae96077
/*
 * Copyright (c) 2022, 2025, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package java.lang.classfile;

import java.lang.constant.MethodTypeDesc;
import java.lang.reflect.AccessFlag;
import java.util.function.Consumer;

import jdk.internal.classfile.impl.AccessFlagsImpl;
import jdk.internal.classfile.impl.ChainedMethodBuilder;
import jdk.internal.classfile.impl.TerminalMethodBuilder;

/**
 * A builder for methods.  The main way to obtain a method builder is via {@link
 * ClassBuilder#withMethod(String, MethodTypeDesc, int, Consumer)}.  {@link
 * ClassBuilder#withMethodBody(String, MethodTypeDesc, int, Consumer)} is
 * useful if no attribute on the method except {@link CodeModel Code} needs to
 * be configured, skipping the method handler.
 * <p>
 * Refer to {@link ClassFileBuilder} for general guidance and caution around
 * the use of builders for structures in the {@code class} file format.
 *
 * @see MethodModel
 * @see MethodTransform
 * @jvms 4.6 Methods
 * @since 24
 */
public sealed interface MethodBuilder
        extends ClassFileBuilder<MethodElement, MethodBuilder>
        permits ChainedMethodBuilder, TerminalMethodBuilder {

    /**
     * Sets the method access flags.  The {@link AccessFlag#STATIC} flag cannot
     * be modified after the builder is created.
     *
     * @param flags the access flags, as a bit mask
     * @return this builder
     * @throws IllegalArgumentException if {@code flags} is not {@link
     *         java.lang.classfile##u2 u2}, or the {@link ClassFile#ACC_STATIC
     *         ACC_STATIC} flag is modified
     * @see AccessFlags
     * @see AccessFlag.Location#METHOD
     */
    default MethodBuilder withFlags(int flags) {
        return with(new AccessFlagsImpl(AccessFlag.Location.METHOD, flags));
    }

    /**
     * Sets the method access flags.  The {@link AccessFlag#STATIC} flag cannot
     * be modified after the builder is created.
     *
     * @param flags the access flags, as a bit mask
     * @return this builder
     * @throws IllegalArgumentException if the {@link ClassFile#ACC_STATIC
     *         ACC_STATIC} flag is modified, or if any flag cannot be applied to
     *         the {@link AccessFlag.Location#METHOD} location
     * @see AccessFlags
     * @see AccessFlag.Location#METHOD
     */
    default MethodBuilder withFlags(AccessFlag... flags) {
        return with(new AccessFlagsImpl(AccessFlag.Location.METHOD, flags));
    }

    /**
     * Build the method body for this method.
     *
     * @param code a handler receiving a {@link CodeBuilder}
     * @return this builder
     * @see CodeModel
     */
    MethodBuilder withCode(Consumer<? super CodeBuilder> code);

    /**
     * Build the method body for this method by transforming the body of another
     * method.
     * <p>
     * This method behaves as if:
     * {@snippet lang=java :
     * withCode(cob -> cob.transform(code, transform));
     * }
     *
     * @param code the method body to be transformed
     * @param transform the transform to apply to the method body
     * @return this builder
     * @see CodeTransform
     */
    MethodBuilder transformCode(CodeModel code, CodeTransform transform);
}