/*
 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
 * 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.
 */

/*
 * @file commands.h
 * @brief All hicn-light commands: 14 in total.
 *
 * Header and payload in binary format.
 */

#ifndef commands_h
#define commands_h

#ifndef _WIN32
#include <netinet/in.h>
#include <sys/socket.h>
#endif

#include <stdint.h>
#include <stdlib.h>

#include <hicn/util/ip_address.h>
#ifdef WITH_POLICY
#include <hicn/policy.h>
#endif /* WITH_POLICY */

#define SYMBOLIC_NAME_LEN 16
#define MAX_FWD_STRATEGY_RELATED_PREFIXES 10

typedef struct in6_addr ipv6_addr_t;
typedef uint32_t ipv4_addr_t;

typedef enum {
  ADDR_INET = 1,
  ADDR_INET6,
  ADDR_LINK,
  ADDR_IFACE,
  ADDR_UNIX /* PF_UNIX */
} address_type;

typedef enum {
  UDP_CONN,
  TCP_CONN,
  GRE_CONN,  // not implemented
  HICN_CONN
} connection_type;

typedef struct in6_addr ipv6_addr_t;
typedef uint32_t ipv4_addr_t;

typedef enum {
    MESSAGE_COMMAND_SUBTYPE_UNDEFINED,
    REQUEST_LIGHT = 0xc0, // this is a command
    RESPONSE_LIGHT,
    ACK_LIGHT,
    NACK_LIGHT,
    MESSAGE_COMMAND_SUBTYPE_N
} message_command_subtype_t;

#define message_type_is_valid(message_type) \
    ((message_type != MESSAGE_TYPE_UNDEFINED) && (message_type != MESSAGE_COMMAND_SUBTYPE_N))

#define message_type_from_uchar(x) \
    (((x) < REQUEST_LIGHT) || (((x) >= MESSAGE_COMMAND_SUBTYPE_N)) ? MESSAGE_COMMAND_SUBTYPE_N : (message_command_subtype_t)(x))

#define foreach_command_type                                    \
    _(listener_add, LISTENER_ADD)                               \
    _(listener_remove, LISTENER_REMOVE)                         \
    _(listener_list, LISTENER_LIST)                             \
    _(connection_add, CONNECTION_ADD)                           \
    _(connection_remove, CONNECTION_REMOVE)                     \
    _(connection_list, CONNECTION_LIST)                         \
    _(connection_set_admin_state, CONNECTION_SET_ADMIN_STATE)   \
    _(connection_update, CONNECTION_UPDATE)                     \
    _(connection_set_priority, CONNECTION_SET_PRIORITY)         \
    _(connection_set_tags, CONNECTION_SET_TAGS)                 \
    _(route_add, ROUTE_ADD)                                     \
    _(route_remove, ROUTE_REMOVE)                               \
    _(route_list, ROUTE_LIST)                                   \
    _(cache_set_store, CACHE_SET_STORE)                         \
    _(cache_set_serve, CACHE_SET_SERVE)                         \
    _(cache_clear, CACHE_CLEAR)                                 \
    _(strategy_set, STRATEGY_SET)                               \
    _(wldr_set, WLDR_SET)                                       \
    _(punting_add, PUNTING_ADD)                                 \
    _(mapme_enable, MAPME_ENABLE)                               \
    _(mapme_set_discovery, MAPME_SET_DISCOVERY)                 \
    _(mapme_set_timescale, MAPME_SET_TIMESCALE)                 \
    _(mapme_set_retx, MAPME_SET_RETX)                           \
    _(mapme_send_update, MAPME_SEND_UPDATE)                     \
    _(policy_add, POLICY_ADD)                                   \
    _(policy_remove, POLICY_REMOVE)                             \
    _(policy_list, POLICY_LIST)                                 \

typedef enum {
    COMMAND_TYPE_UNDEFINED,
#define _(l, u) COMMAND_TYPE_ ## u,
    foreach_command_type
#undef _
    COMMAND_TYPE_N,
} command_type_t;

#define command_type_is_valid(command_type) \
    ((command_type != COMMAND_TYPE_UNDEFINED) && (command_type != COMMAND_TYPE_N))

#define command_type_from_uchar(x) \
    (((x) >= COMMAND_TYPE_N) ? COMMAND_TYPE_N : (command_type_t)(x))

/* Header */

/* Should be at least 8 bytes */
typedef struct {
    uint8_t messageType;
    uint8_t commandID ;
    uint16_t length;    /* Number of structures in the payload */
    uint32_t seqNum;
} cmd_header_t;

typedef struct {
    cmd_header_t header;
} msg_header_t;

/* Listener */

typedef enum { ETHER_MODE, IP_MODE, HICN_MODE } listener_mode;

typedef struct {
    char symbolic[SYMBOLIC_NAME_LEN];
    char interfaceName[SYMBOLIC_NAME_LEN];
    ip_address_t address;
    uint16_t port;
    // uint16_t etherType;
    uint8_t addressType;
    uint8_t listenerMode;
    uint8_t connectionType;
    uint8_t family;
    uint8_t listenerType;
} cmd_listener_add_t;

typedef struct {
    char symbolicOrListenerid[SYMBOLIC_NAME_LEN];
} cmd_listener_remove_t;

typedef struct {
    ip_address_t address;
    char listenerName[SYMBOLIC_NAME_LEN];
    char interfaceName[SYMBOLIC_NAME_LEN];
    uint32_t connid;
    uint16_t port;
    uint8_t addressType;
    uint8_t encapType;
} cmd_listener_list_t;

typedef struct {
    ip_address_t address;
    char name[SYMBOLIC_NAME_LEN];
    char interface_name[SYMBOLIC_NAME_LEN];
    uint32_t id;
    uint16_t port;
    uint8_t family;
    uint8_t type;
} cmd_listener_list_item_t;

/* Connection */

typedef struct {
    char symbolic[SYMBOLIC_NAME_LEN];
    //char interface_name[SYMBOLIC_NAME_LEN];
    ip_address_t remote_ip;
    ip_address_t local_ip;
    uint16_t remote_port;
    uint16_t local_port;
    uint8_t family;
    uint8_t type;
    uint8_t connection_type;
    uint8_t admin_state;
#ifdef WITH_POLICY
    uint32_t priority;
    policy_tags_t tags;
#endif /* WITH_POLICY */
} cmd_connection_add_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
} cmd_connection_remove_t;

typedef enum {
  CONN_GRE,
  CONN_TCP,
  CONN_UDP,
  CONN_MULTICAST,
  CONN_L2,
  CONN_HICN
} list_connections_type;

typedef enum {
  IFACE_UP = 0,
  IFACE_DOWN = 1,
  IFACE_UNKNOWN = 2  // not used actually
} connection_state;

typedef struct {
  cmd_connection_add_t connectionData;
  uint32_t connid;
  uint8_t state;
  char interfaceName[SYMBOLIC_NAME_LEN];
  char connectionName[SYMBOLIC_NAME_LEN];
} cmd_connection_list_t;

typedef struct {
    char symbolic[SYMBOLIC_NAME_LEN];
    //char interface_name[SYMBOLIC_NAME_LEN];
    ip_address_t remote_ip;
    ip_address_t local_ip;
    uint16_t remote_port;
    uint16_t local_port;
    uint8_t family;
    uint8_t type;
    uint8_t admin_state;
#ifdef WITH_POLICY
    uint32_t priority;
    policy_tags_t tags;
#endif /* WITH_POLICY */
    uint32_t id;
    uint8_t state;
    char interface_name[SYMBOLIC_NAME_LEN];
    char name[SYMBOLIC_NAME_LEN];
} cmd_connection_list_item_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    uint8_t admin_state;
    uint8_t pad8[3];
} cmd_connection_set_admin_state_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    uint8_t admin_state;
    uint32_t priority;
    policy_tags_t tags;
} cmd_connection_update_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    uint32_t priority;
} cmd_connection_set_priority_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    policy_tags_t tags;
} cmd_connection_set_tags_t;

/* Route */

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    ip_address_t address;
    uint16_t cost;
    uint8_t addressType;
    uint8_t family;
    uint8_t len;
} cmd_route_add_t;

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    ip_address_t address;
    uint8_t addressType;
    uint8_t family;
    uint8_t len;
} cmd_route_remove_t;

typedef struct {
    ip_address_t address;
    uint32_t connid;
    uint16_t cost;
    uint8_t addressType;
    uint8_t len;
} cmd_route_list_t;

typedef struct {
    ip_address_t address;
    uint32_t connection_id;
    uint16_t cost;
    uint8_t family;
    uint8_t len;
} cmd_route_list_item_t;

/* Cache */

typedef struct {
    uint8_t activate;
} cmd_cache_set_store_t;

typedef struct {
    uint8_t activate;
} cmd_cache_set_serve_t;

typedef struct {
} cmd_cache_clear_t;

/* WLDR */

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    uint8_t activate;
} cmd_wldr_set_t;

/* Strategy */

typedef struct {
    ip_address_t address;
    uint8_t strategy_type;
    uint8_t address_type;
    uint8_t family;
    uint8_t len;
    uint8_t related_prefixes;
    union {
        struct {
            ip_address_t addresses[MAX_FWD_STRATEGY_RELATED_PREFIXES];
            uint8_t lens[MAX_FWD_STRATEGY_RELATED_PREFIXES];
            uint8_t families[MAX_FWD_STRATEGY_RELATED_PREFIXES];
        } low_latency;
    };
} cmd_strategy_set_t;

/* Punting */

typedef struct {
    char symbolicOrConnid[SYMBOLIC_NAME_LEN];
    ip_address_t address;
    uint8_t addressType;
    uint8_t family;
    uint8_t len;
} cmd_punting_add_t;

/* MAP-Me */

typedef struct {
    uint8_t activate;
} cmd_mapme_activator_t;

typedef cmd_mapme_activator_t cmd_mapme_enable_t;
typedef cmd_mapme_activator_t cmd_mapme_set_discovery_t;

typedef struct {
    uint32_t timePeriod;
} cmd_mapme_timing_t;

typedef cmd_mapme_timing_t cmd_mapme_set_timescale_t;
typedef cmd_mapme_timing_t cmd_mapme_set_retx_t;

typedef struct {
    ip_address_t address;
    uint8_t family;
    uint8_t len;
} cmd_mapme_send_update_t;

#ifdef WITH_POLICY
/* Policy */

typedef struct {
    ip_address_t address;
    uint8_t addressType;
    uint8_t family;
    uint8_t len;
    policy_t policy;
} cmd_policy_add_t;

typedef struct {
    ip_address_t address;
    uint8_t addressType;
    uint8_t family;
    uint8_t len;
} cmd_policy_remove_t;

typedef struct {
    ip_address_t address;
    uint8_t addressType;
    uint8_t len;
    policy_t policy;
} cmd_policy_list_t;

typedef struct {
    ip_address_t address;
    uint8_t family;
    uint8_t len;
    policy_t policy;
} cmd_policy_list_item_t;

#endif /* WITH_POLICY */

/* Full messages */

#define _(l, u)                         \
typedef struct {                        \
    cmd_header_t header;                \
    cmd_ ## l ## _t payload;            \
} msg_ ## l ## _t;
    foreach_command_type
#undef _

typedef struct {
    cmd_header_t header;
    cmd_listener_list_item_t payload;
} msg_listener_list_reply_t;

typedef struct {
    cmd_header_t header;
    cmd_connection_list_item_t payload;
} msg_connection_list_reply_t;

typedef struct {
    cmd_header_t header;
    cmd_route_list_item_t payload;
} msg_route_list_reply_t;

typedef struct {
    cmd_header_t header;
    cmd_policy_list_item_t payload;
} msg_policy_list_reply_t;

//===== size of commands ======
// REMINDER: when a new_command is added, the following switch has to be
// updated.
static inline int command_get_payload_len(command_type_t command_type) {
    switch (command_type) {
#define _(l, u)                                                         \
        case COMMAND_TYPE_ ## u:                                        \
            return sizeof(cmd_## l ## _t);
    foreach_command_type
#undef _
        case COMMAND_TYPE_UNDEFINED:
        case COMMAND_TYPE_N:
            return 0;
    }
}
#endif