/*
 * Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"). You may not use
 * this file except in compliance with the License. A copy of the License is
 * located at
 *
 *     http://aws.amazon.com/apache2.0/
 *
 * or in the "license" file accompanying this file. This file 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.
 */

/**
 * Invalid CMMs for testing virtual function error handling.
 */

#include "bad_cmm.h"
#include <stdio.h>
#include <stdlib.h>

#ifdef _MSC_VER
// We ignore a lot of parameters in this source file.
#    pragma warning(disable : 4100)
#endif

int aws_cryptosdk_cmm_release_with_failed_return_value(struct aws_cryptosdk_cmm *cmm) {
    aws_reset_error();
    aws_cryptosdk_cmm_release(cmm);
    return AWS_OP_ERR;
}

/**
 * VFs which will never get called because of the failed check on the vt_size field.
 */
void destroy_abort(struct aws_cryptosdk_cmm *cmm) {
    fprintf(stderr, "%s's destroy VF was called when it should not have been\n", cmm->vtable->name);
    abort();
}

int generate_abort(
    struct aws_cryptosdk_cmm *cmm,
    struct aws_cryptosdk_enc_materials **output,
    struct aws_cryptosdk_enc_request *request) {
    fprintf(stderr, "%s's generate_enc_materials VF was called when it should not have been\n", cmm->vtable->name);
    abort();
}

int decrypt_abort(
    struct aws_cryptosdk_cmm *cmm,
    struct aws_cryptosdk_dec_materials **output,
    struct aws_cryptosdk_dec_request *request) {
    fprintf(stderr, "%s's decrypt_materials VF was called when it should not have been\n", cmm->vtable->name);
    abort();
}

/**
 * A totally correct VT except for the zero size.
 */
static const struct aws_cryptosdk_cmm_vt zero_size_cmm_vt = { .vt_size                = 0,
                                                              .name                   = "zero size cmm",
                                                              .destroy                = destroy_abort,
                                                              .generate_enc_materials = generate_abort,
                                                              .decrypt_materials      = decrypt_abort };

struct aws_cryptosdk_cmm aws_cryptosdk_zero_size_cmm() {
    struct aws_cryptosdk_cmm cmm;

    aws_cryptosdk_cmm_base_init(&cmm, &zero_size_cmm_vt);

    return cmm;
}

static const struct aws_cryptosdk_cmm_vt null_cmm_vt = { .vt_size                = sizeof(struct aws_cryptosdk_cmm_vt),
                                                         .name                   = "null cmm",
                                                         .destroy                = NULL,
                                                         .generate_enc_materials = NULL,
                                                         .decrypt_materials      = NULL };

struct aws_cryptosdk_cmm aws_cryptosdk_null_cmm() {
    struct aws_cryptosdk_cmm cmm;

    aws_cryptosdk_cmm_base_init(&cmm, &null_cmm_vt);

    return cmm;
}