/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "prio.h"
#include "prenv.h"
#include "prmem.h"
#include "prlink.h"
#include "prsystem.h"
#include "prnetdb.h"
#include "prprf.h"
#include "prvrsion.h"

#include "plerror.h"
#include "plgetopt.h"
#include "obsolete/probslet.h"

#include <string.h>

#define DNS_BUFFER 100
#define ADDR_BUFFER 100
#define HOST_BUFFER 1024
#define PROTO_BUFFER 1500

#define NETADDR_SIZE(addr)                                 \
  (PR_AF_INET == (addr)->raw.family ? sizeof((addr)->inet) \
                                    : sizeof((addr)->ipv6))

static PRFileDesc* err = NULL;

static void Help(void) {
  PR_fprintf(err, "Usage: [-V] [-h]\n");
  PR_fprintf(err,
             "\t<nul>    Name of host to lookup          (default: self)\n");
  PR_fprintf(err,
             "\t-V       Display runtime version info    (default: FALSE)\n");
  PR_fprintf(err, "\t-h       This message and nothing else\n");
} /* Help */

static void DumpAddr(const PRNetAddr* address, const char* msg) {
  PRUint32* word = (PRUint32*)address;
  PRUint32 addr_len = sizeof(PRNetAddr);
  PR_fprintf(err, "%s[%d]\t", msg, NETADDR_SIZE(address));
  while (addr_len > 0) {
    PR_fprintf(err, " %08x", *word++);
    addr_len -= sizeof(PRUint32);
  }
  PR_fprintf(err, "\n");
} /* DumpAddr */

static PRStatus PrintAddress(const PRNetAddr* address) {
  PRNetAddr translation;
  char buffer[ADDR_BUFFER];
  PRStatus rv = PR_NetAddrToString(address, buffer, sizeof(buffer));
  if (PR_FAILURE == rv) {
    PL_FPrintError(err, "PR_NetAddrToString");
  } else {
    PR_fprintf(err, "\t%s\n", buffer);
    memset(&translation, 0, sizeof(translation));
    rv = PR_StringToNetAddr(buffer, &translation);
    if (PR_FAILURE == rv) {
      PL_FPrintError(err, "PR_StringToNetAddr");
    } else {
      PRSize addr_len = NETADDR_SIZE(address);
      if (0 != memcmp(address, &translation, addr_len)) {
        PR_fprintf(err, "Address translations do not match\n");
        DumpAddr(address, "original");
        DumpAddr(&translation, "translate");
        rv = PR_FAILURE;
      }
    }
  }
  return rv;
} /* PrintAddress */

int main(int argc, char** argv) {
  PRStatus rv;
  PLOptStatus os;
  PRHostEnt host;
  PRProtoEnt proto;
  const char* name = NULL;
  PRBool failed = PR_FALSE, version = PR_FALSE;
  PLOptState* opt = PL_CreateOptState(argc, argv, "Vh");

  err = PR_GetSpecialFD(PR_StandardError);

  while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
    if (PL_OPT_BAD == os) {
      continue;
    }
    switch (opt->option) {
      case 0: /* Name of host to lookup */
        name = opt->value;
        break;
      case 'V': /* Do version discovery */
        version = PR_TRUE;
        break;
      case 'h': /* user wants some guidance */
      default:
        Help();   /* so give him an earful */
        return 2; /* but not a lot else */
    }
  }
  PL_DestroyOptState(opt);

  if (version) {
#if defined(WINNT)
#  define NSPR_LIB "libnspr4"
#else
#  define NSPR_LIB "nspr4"
#endif
    const PRVersionDescription* version_info;
    char* nspr_path = PR_GetEnv("LD_LIBRARY_PATH");
    char* nspr_name = PR_GetLibraryName(nspr_path, NSPR_LIB);
    PRLibrary* runtime = PR_LoadLibrary(nspr_name);
    if (NULL == runtime) {
      PL_FPrintError(err, "PR_LoadLibrary");
    } else {
      versionEntryPointType versionPoint =
          (versionEntryPointType)PR_FindSymbol(runtime, "libVersionPoint");
      if (NULL == versionPoint) {
        PL_FPrintError(err, "PR_FindSymbol");
      } else {
        char buffer[100];
        PRExplodedTime exploded;
        version_info = versionPoint();
        (void)PR_fprintf(err, "Runtime library version information\n");
        PR_ExplodeTime(version_info->buildTime, PR_GMTParameters, &exploded);
        (void)PR_FormatTime(buffer, sizeof(buffer), "%d %b %Y %H:%M:%S",
                            &exploded);
        (void)PR_fprintf(err, "  Build time: %s GMT\n", buffer);
        (void)PR_fprintf(err, "  Build time: %s\n",
                         version_info->buildTimeString);
        (void)PR_fprintf(err, "  %s V%u.%u.%u (%s%s%s)\n",
                         version_info->description, version_info->vMajor,
                         version_info->vMinor, version_info->vPatch,
                         (version_info->beta ? " beta " : ""),
                         (version_info->debug ? " debug " : ""),
                         (version_info->special ? " special" : ""));
        (void)PR_fprintf(err, "  filename: %s\n", version_info->filename);
        (void)PR_fprintf(err, "  security: %s\n", version_info->security);
        (void)PR_fprintf(err, "  copyright: %s\n", version_info->copyright);
        (void)PR_fprintf(err, "  comment: %s\n", version_info->comment);
      }
    }
    if (NULL != nspr_name) {
      PR_FreeLibraryName(nspr_name);
    }
  }

  {
    if (NULL == name) {
      char* me = (char*)PR_MALLOC(DNS_BUFFER);
      rv = PR_GetSystemInfo(PR_SI_HOSTNAME, me, DNS_BUFFER);
      if (PR_FAILURE == rv) {
        failed = PR_TRUE;
        PL_FPrintError(err, "PR_GetSystemInfo");
        return 2;
      }
      name = me; /* just leak the storage */
    }
  }

  {
    char buffer[HOST_BUFFER];
    PR_fprintf(err, "Translating the name %s ...", name);

    rv = PR_GetHostByName(name, buffer, sizeof(buffer), &host);
    if (PR_FAILURE == rv) {
      failed = PR_TRUE;
      PL_FPrintError(err, "PR_GetHostByName");
    } else {
      PRIntn index = 0;
      PRNetAddr address;
      memset(&address, 0, sizeof(PRNetAddr));
      PR_fprintf(err, "success .. enumerating results\n");
      do {
        index = PR_EnumerateHostEnt(index, &host, 0, &address);
        if (index > 0) {
          PrintAddress(&address);
        } else if (-1 == index) {
          failed = PR_TRUE;
          PL_FPrintError(err, "PR_EnumerateHostEnt");
        }
      } while (index > 0);
    }
  }

  {
    char buffer[PROTO_BUFFER];
    /*
    ** Get Proto by name/number
    */
    rv = PR_GetProtoByName("tcp", &buffer[1], sizeof(buffer) - 1, &proto);
    rv = PR_GetProtoByNumber(6, &buffer[3], sizeof(buffer) - 3, &proto);
  }

  return (failed) ? 1 : 0;
}