com.authlete.cose.COSEEncrypt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cbor Show documentation
A Java library for CBOR, COSE, CWT and mdoc.
The newest version!
/*
 * Copyright (C) 2023 Authlete, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.authlete.cose;


import java.util.ArrayList;
import java.util.List;
import com.authlete.cbor.CBORByteArray;
import com.authlete.cbor.CBORItem;
import com.authlete.cbor.CBORItemList;
import com.authlete.cbor.CBORNull;
import com.authlete.cbor.CBORizer;


/**
 * COSE_Encrypt
 *
 * 
 * {@code COSE_Encrypt} is defined in 5.1. Enveloped
 * COSE Structure of RFC 9052 as follows.
 * 
 *
 *  * COSE_Encrypt = [
 *     Headers,
 *     ciphertext : bstr / nil,
 *     recipients : [+COSE_recipient]
 * ]
 * 
 *
 * @since 1.1
 *
 * @see RFC 9052, 5.1. Enveloped COSE Structure
 */
public class COSEEncrypt extends COSEMessage
{
    /**
     * A constructor with a protected header, an unprotected header, a
     * cipher text and recipients.
     *
     * @param protectedHeader
     *         A protected header. Must not be null.
     *
     * @param unprotectedHeader
     *         An unprotected header. Must not be null.
     *
     * @param ciphertext
     *         A cipher text. Must be either {@link CBORByteArray} or
     *         {@link CBORNull}.
     *
     * @param recipients
     *         Recipients. Must not be null. Items in the list must be
     *         instances of {@link COSERecipient}. At least one element
     *         must be contained.
     */
    public COSEEncrypt(
            COSEProtectedHeader protectedHeader,
            COSEUnprotectedHeader unprotectedHeader,
            CBORItem ciphertext,
            CBORItemList recipients)
    {
        super(COSEMessageType.COSE_ENCRYPT,
                protectedHeader, unprotectedHeader, ciphertext, recipients);

        validateRecipients(recipients);
    }


    private static void validateRecipients(CBORItemList recipients)
    {
        if (recipients == null)
        {
            throw new IllegalArgumentException(
                    "A recipient list given to COSEEncrypt's constructor must not be null.");
        }

        List items = recipients.getItems();

        if (items == null || items.size() == 0)
        {
            throw new IllegalArgumentException(
                    "A recipient list given to COSEEncrypt's constructor must not be empty.");
        }

        for (CBORItem item : items)
        {
            if (item instanceof COSERecipient)
            {
                continue;
            }

            throw new IllegalArgumentException(
                    "Items in the recipient list given to COSEEncrypt's constructor must be instances of COSERecipient.");
        }
    }


    /**
     * Get the cipher text.
     *
     * 
     * The type of the value is {@link CBORByteArray} or {@link CBORNull}.
     * 
     *
     * @return
     *         The cipher text.
     */
    public CBORItem getCiphertext()
    {
        return getItems().get(2);
    }


    /**
     * Get the list of recipients. Items in the list are {@link COSERecipient}
     * instances.
     *
     * @return
     *         The list of recipients.
     */
    public CBORItemList getRecipients()
    {
        return (CBORItemList)getItems().get(3);
    }


    /**
     * Build a {@code COSEEncrypt} instance from the given CBOR data item.
     * The data item must be a CBOR array.
     *
     * @param item
     *         A CBOR data item that represents {@code COSE_Encrypt}.
     *
     * @return
     *         A {@code COSEEncrypt} instance built from the given CBOR data item.
     *
     * @throws COSEException
     */
    public static COSEEncrypt build(CBORItem item) throws COSEException
    {
        // Build the protected header, the unprotected header and the content.
        List common = COSEObject.buildCommon(item, "COSE_Encrypt", 4);

        // The fourth element of COSE_Encrypt, which is a list of recipients.
        CBORItem fourth = common.get(3);

        // If the fourth element is not a CBOR array.
        if (!(fourth instanceof CBORItemList))
        {
            throw new COSEException(
                    "The fourth element (recipients) of COSE_Encrypt must be a CBOR array.");
        }

        // Recipients
        CBORItemList recipients = buildRecipients((CBORItemList)fourth);

        try
        {
            return new COSEEncrypt(
                    (COSEProtectedHeader)common.get(0),
                    (COSEUnprotectedHeader)common.get(1),
                    common.get(2),
                    recipients);
        }
        catch (Exception cause)
        {
            throw new COSEException(cause.getMessage(), cause);
        }
    }


    private static CBORItemList buildRecipients(CBORItemList list) throws COSEException
    {
        List items = list.getItems();

        if (items == null || items.size() == 0)
        {
            throw new COSEException(
                    "A list of recipients in COSE_Encrypt must contain at least one recipient.");
        }

        List recipients = new ArrayList<>();

        for (CBORItem item : items)
        {
            // Interpret the CBOR item as a recipient.
            COSERecipient recipient = COSERecipient.build(item);

            // Add the recipient.
            recipients.add(recipient);
        }

        return new CBORItemList(recipients);
    }


    /**
     * Build a {@code COSEEncrypt} instance from the given object list.
     *
     * @param list
     *         A object list that represents {@code COSE_Encrypt}.
     *
     * @return
     *         A {@code COSEEncrypt} instance built from the given object list.
     *
     * @throws COSEException
     */
    public static COSEEncrypt build(List