/*      $OpenBSD: agentx_internal.h,v 1.3 2021/10/23 14:39:35 martijn Exp $ */
/*
 * Copyright (c) 2020 Martijn van Duren <martijn@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
#include <sys/queue.h>
#include <sys/time.h>
#include <sys/tree.h>

#include "ax.h"

enum agentx_cstate {            /* Current state */
        AX_CSTATE_CLOSE,        /* Closed */
        AX_CSTATE_WAITOPEN,     /* Connection requested */
        AX_CSTATE_OPEN,         /* Open */
        AX_CSTATE_WAITCLOSE     /* Close requested */
};

enum agentx_dstate {            /* Desired state */
        AX_DSTATE_OPEN,         /* Open */
        AX_DSTATE_CLOSE         /* Close/free */
};

struct agentx {
        void (*ax_nofd)(struct agentx *, void *, int);
        void *ax_cookie;
        int ax_fd;
        enum agentx_cstate ax_cstate;
        enum agentx_dstate ax_dstate;
        int ax_free;            /* Freeing already planned */
        struct ax *ax_ax;
        TAILQ_HEAD(, agentx_session) ax_sessions;
        TAILQ_HEAD(, agentx_get) ax_getreqs;
        RB_HEAD(ax_requests, agentx_request) ax_requests;
};

struct agentx_session {
        struct agentx *axs_ax;
        uint32_t axs_id;
        uint32_t axs_timeout;
        struct ax_oid axs_oid;
        struct ax_ostring axs_descr;
        enum agentx_cstate axs_cstate;
        enum agentx_dstate axs_dstate;
        uint32_t axs_packetid;
        TAILQ_HEAD(, agentx_context) axs_contexts;
        TAILQ_ENTRY(agentx_session) axs_ax_sessions;
};

struct agentx_context {
        struct agentx_session *axc_axs;
        int axc_name_default;
        struct ax_ostring axc_name;
        uint32_t axc_sysuptime;
        struct timespec axc_sysuptimespec;
        enum agentx_cstate axc_cstate;
        enum agentx_dstate axc_dstate;
        TAILQ_HEAD(, agentx_agentcaps) axc_agentcaps;
        TAILQ_HEAD(, agentx_region) axc_regions;
        RB_HEAD(axc_objects, agentx_object) axc_objects;
        TAILQ_ENTRY(agentx_context) axc_axs_contexts;
};

struct agentx_get {
        struct agentx_context *axg_axc;
        int axg_fd;                             /* Only used for logging */
        uint32_t axg_sessionid;
        uint32_t axg_transactionid;
        uint32_t axg_packetid;
        int axg_context_default;
        struct ax_ostring axg_context;
        enum ax_pdu_type axg_type;
        uint16_t axg_nonrep;
        uint16_t axg_maxrep;
        size_t axg_nvarbind;
        struct agentx_varbind *axg_varbind;
        TAILQ_ENTRY(agentx_get) axg_ax_getreqs;
};

__dead void agentx_log_ax_fatalx(struct agentx *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_ax_warn(struct agentx *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_ax_warnx(struct agentx *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_ax_info(struct agentx *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_ax_debug(struct agentx *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
__dead void agentx_log_axs_fatalx(struct agentx_session *, const char *,
    ...) __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axs_warnx(struct agentx_session *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axs_warn(struct agentx_session *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axs_info(struct agentx_session *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
__dead void agentx_log_axc_fatalx(struct agentx_context *, const char *,
    ...) __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axc_warnx(struct agentx_context *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axc_warn(struct agentx_context *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axc_info(struct agentx_context *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axc_debug(struct agentx_context *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
__dead void agentx_log_axg_fatalx(struct agentx_get *, const char *,
    ...) __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axg_warnx(struct agentx_get *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axg_warn(struct agentx_get *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));
void agentx_log_axg_debug(struct agentx_get *, const char *, ...)
    __attribute__((__format__ (printf, 2, 3)));