#include "speer.h"

#include "core/lmem.h"
#include "net/lpacket.h"
#include "sterm.h"

tShell_peer *shellP_newPeer(PEERTYPE type, OSTYPE osType, uint8_t *pubKey, char *hostname,
                            char *inet, char *ipStr)
{
    tShell_peer *peer = (tShell_peer *)laikaM_malloc(sizeof(tShell_peer));
    peer->type = type;
    peer->osType = osType;

    /* copy pubKey to peer's pubKey */
    memcpy(peer->pub, pubKey, crypto_kx_PUBLICKEYBYTES);

    /* copy hostname & ip */
    memcpy(peer->hostname, hostname, LAIKA_HOSTNAME_LEN);
    memcpy(peer->inet, inet, LAIKA_INET_LEN);
    memcpy(peer->ipStr, ipStr, LAIKA_IPSTR_LEN);

    /* restore NULL terminators */
    peer->hostname[LAIKA_HOSTNAME_LEN - 1] = '\0';
    peer->inet[LAIKA_INET_LEN - 1] = '\0';
    peer->ipStr[LAIKA_IPSTR_LEN - 1] = '\0';

    return peer;
}

void shellP_freePeer(tShell_peer *peer)
{
    laikaM_free(peer);
}

char *shellP_typeStr(tShell_peer *peer)
{
    switch (peer->type) {
    case PEER_PEER:
        return "Peer";
    case PEER_BOT:
        return "Bot";
    case PEER_CNC:
        return "CNC";
    case PEER_AUTH:
        return "Auth";
    default:
        return "err";
    }
}

char *shellP_osTypeStr(tShell_peer *peer)
{
    switch (peer->osType) {
    case OS_WIN:
        return "Windows";
    case OS_LIN:
        return "Linux";
    default:
        return "unkn";
    }
}

void shellP_printInfo(tShell_peer *peer)
{
    char buf[128]; /* i don't expect bin2hex to write outside this, but it's only user-info and
                      doesn't break anything (ie doesn't write outside the buffer) */

    sodium_bin2hex(buf, sizeof(buf), peer->pub, crypto_kx_PUBLICKEYBYTES);
    shellT_printf("\t%s-%s\n\tOS: %s\n\tINET: %s\n\tPUBKEY: %s\n", peer->ipStr, peer->hostname,
                  shellP_osTypeStr(peer), peer->inet, buf);
}