/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2022. 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.
 * You may obtain a copy of the License at
 *
 *     http://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.
 */

#include "type_headers.h"

static constexpr int ARGS_POINT_ZERO = 0;
static constexpr int ARGS_POINT_ONE = 1;
static constexpr int ARGS_POINT_TWO = 2;
static constexpr int ARGS_POINT_THREE = 3;

static int Openat2ArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->openat2_args.dfd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->openat2_args.filename;
    args[ARGS_POINT_TWO] = (__u64) startEvent->openat2_args.how;
    args[ARGS_POINT_THREE] = 0xFFFFFFFFFFFFFFFF;
    return 0;
}

static int ReadArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->read_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->read_args.buf;
    args[ARGS_POINT_TWO] = (__u64) startEvent->read_args.count;
    args[ARGS_POINT_THREE] = 0xFFFFFFFFFFFFFFFF;
    return 0;
}

static int WriteArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->write_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->write_args.buf;
    args[ARGS_POINT_TWO] = (__u64) startEvent->write_args.count;
    args[ARGS_POINT_THREE] = 0xFFFFFFFFFFFFFFFF;
    return 0;
}

static int Pread64ArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->pread64_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->pread64_args.buf;
    args[ARGS_POINT_TWO] = (__u64) startEvent->pread64_args.count;
    args[ARGS_POINT_THREE] = (__u64) startEvent->pread64_args.pos;
    return 0;
}

static int Pwrite64ArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->pwrite64_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->pwrite64_args.buf;
    args[ARGS_POINT_TWO] = (__u64) startEvent->pwrite64_args.count;
    args[ARGS_POINT_THREE] = (__u64) startEvent->pwrite64_args.pos;
    return 0;
}

static int ReadvArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->readv_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->readv_args.vec;
    args[ARGS_POINT_TWO] = (__u64) startEvent->readv_args.vlen;
    args[ARGS_POINT_THREE] = (__u64) startEvent->readv_args.flags;
    return 0;
}

static int WritevArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->writev_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->writev_args.vec;
    args[ARGS_POINT_TWO] = (__u64) startEvent->writev_args.vlen;
    args[ARGS_POINT_THREE] = (__u64) startEvent->writev_args.flags;
    return 0;
}

static int PreadvArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->preadv_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->preadv_args.vec;
    args[ARGS_POINT_TWO] = (__u64) startEvent->preadv_args.vlen;
    args[ARGS_POINT_THREE] = (__u64) startEvent->preadv_args.pos;
    return 0;
}

static int PwritevArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->pwritev_args.fd;
    args[ARGS_POINT_ONE] = (__u64) startEvent->pwritev_args.vec;
    args[ARGS_POINT_TWO] = (__u64) startEvent->pwritev_args.vlen;
    args[ARGS_POINT_THREE] = (__u64) startEvent->pwritev_args.pos;
    return 0;
}

static int CloseArgsConverter(__u64* args, const struct fstrace_start_event_t* startEvent)
{
    args[ARGS_POINT_ZERO] = (__u64) startEvent->close_args.files;
    args[ARGS_POINT_ONE] = (__u64) startEvent->close_args.fd;
    args[ARGS_POINT_TWO] = 0xFFFFFFFFFFFFFFFF;
    args[ARGS_POINT_THREE] = 0xFFFFFFFFFFFFFFFF;
    return 0;
}

using ConverterType = int (*) (__u64*, const struct fstrace_start_event_t *);
ConverterType g_argsConverterTable[11] = {
    nullptr,
    &Openat2ArgsConverter,
    &ReadArgsConverter,
    &WriteArgsConverter,
    &Pread64ArgsConverter,
    &Pwrite64ArgsConverter,
    &ReadvArgsConverter,
    &WritevArgsConverter,
    &PreadvArgsConverter,
    &PwritevArgsConverter,
    &CloseArgsConverter
};