#include <config.h>

#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#ifdef _WIN32
 #include <winsock2.h>
 #include <ws2tcpip.h>
 #include <windows.h>
#else
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <netdb.h>
 #include <unistd.h>
#endif

#include <pcap.h>

#include "varattrs.h"
#include "pcap/funcattrs.h"

static int ifprint(pcap_if_t *d);
static char *iptos(bpf_u_int32 in);

#ifdef _WIN32
#include "portability.h"

/*
* Generate a string for a Win32-specific error (i.e. an error generated when
* calling a Win32 API).
* For errors occurred during standard C calls, we still use pcap_strerror()
*/
#define ERRBUF_SIZE     1024
static const char *
win32_strerror(DWORD error)
{
 static char errbuf[ERRBUF_SIZE+1];
 size_t errlen;

 FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, error, 0, errbuf,
               ERRBUF_SIZE, NULL);

 /*
  * "FormatMessage()" "helpfully" sticks CR/LF at the end of the
  * message.  Get rid of it.
  */
 errlen = strlen(errbuf);
 if (errlen >= 2) {
   errbuf[errlen - 1] = '\0';
   errbuf[errlen - 2] = '\0';
   errlen -= 2;
 }
 return errbuf;
}

static char *
getpass(const char *prompt)
{
 HANDLE console_handle = GetStdHandle(STD_INPUT_HANDLE);
 DWORD console_mode, save_console_mode;
 static char password[128+1];
 char *p;

 fprintf(stderr, "%s", prompt);

 /*
  * Turn off echoing.
  */
 if (!GetConsoleMode(console_handle, &console_mode)) {
   fprintf(stderr, "Can't get console mode: %s\n",
           win32_strerror(GetLastError()));
   exit(1);
 }
 save_console_mode = console_mode;
 console_mode &= ~ENABLE_ECHO_INPUT;
 if (!SetConsoleMode(console_handle, console_mode)) {
   fprintf(stderr, "Can't set console mode: %s\n",
           win32_strerror(GetLastError()));
   exit(1);
 }
 if (fgets(password, sizeof password, stdin) == NULL) {
   fprintf(stderr, "\n");
   SetConsoleMode(console_handle, save_console_mode);
   exit(1);
 }
 fprintf(stderr, "\n");
 SetConsoleMode(console_handle, save_console_mode);
 p = strchr(password, '\n');
 if (p != NULL)
   *p = '\0';
return password;
}
#endif

#ifdef ENABLE_REMOTE
int main(int argc, char **argv)
#else
int main(int argc _U_, char **argv _U_)
#endif
{
 pcap_if_t *alldevs;
 pcap_if_t *d;
 bpf_u_int32 net, mask;
 int exit_status = 0;
 char errbuf[PCAP_ERRBUF_SIZE+1];
#ifdef ENABLE_REMOTE
 struct pcap_rmtauth auth;
 char username[128+1];
 char *p;
 char *password;
#endif

#ifdef ENABLE_REMOTE
 if (argc >= 2)
 {
   if (pcap_findalldevs_ex(argv[1], NULL, &alldevs, errbuf) == -1)
   {
     /*
      * OK, try it with a user name and password.
      */
     fprintf(stderr, "User name: ");
     if (fgets(username, sizeof username, stdin) == NULL)
       exit(1);
     p = strchr(username, '\n');
     if (p != NULL)
       *p = '\0';
     password = getpass("Password: ");
     auth.type = RPCAP_RMTAUTH_PWD;
     auth.username = username;
     auth.password = password;
     if (pcap_findalldevs_ex(argv[1], &auth, &alldevs, errbuf) == -1)
     {
       fprintf(stderr,"Error in pcap_findalldevs: %s\n",errbuf);
       exit(1);
     }
   }
 }
 else
#endif
 {
   if (pcap_findalldevs(&alldevs, errbuf) == -1)
   {
     fprintf(stderr,"Error in pcap_findalldevs: %s\n",errbuf);
     exit(1);
   }
 }
 for(d=alldevs;d;d=d->next)
 {
   if (!ifprint(d))
     exit_status = 2;
 }

 if (alldevs != NULL)
 {
   if (pcap_lookupnet(alldevs->name, &net, &mask, errbuf) < 0)
   {
     /*
      * XXX - this doesn't distinguish between "a real error
      * occurred" and "this interface doesn't *have* an IPv4
      * address".  The latter shouldn't be treated as an error.
      *
      * We look for the interface name, followed by a colon and
      * a space, and, if we find it,w e see if what follows it
      * is "no IPv4 address assigned".
      */
     size_t devnamelen = strlen(alldevs->name);
     if (strncmp(errbuf, alldevs->name, devnamelen) == 0 &&
         strncmp(errbuf + devnamelen, ": ", 2) == 0 &&
         strcmp(errbuf + devnamelen + 2, "no IPv4 address assigned") == 0)
       printf("Preferred device is not on an IPv4 network\n");
     else {
       fprintf(stderr,"Error in pcap_lookupnet: %s\n",errbuf);
       exit_status = 2;
     }
   }
   else
   {
     printf("Preferred device is on network: %s/%s\n",iptos(net), iptos(mask));
   }
 }

 pcap_freealldevs(alldevs);
 exit(exit_status);
}

static int ifprint(pcap_if_t *d)
{
 pcap_addr_t *a;
 char ipv4_buf[INET_ADDRSTRLEN];
#ifdef INET6
 char ipv6_buf[INET6_ADDRSTRLEN];
#endif
 const char *sep;
 int status = 1; /* success */

 printf("%s\n",d->name);
 if (d->description)
   printf("\tDescription: %s\n",d->description);
 printf("\tFlags: ");
 sep = "";
 if (d->flags & PCAP_IF_UP) {
   printf("%sUP", sep);
   sep = ", ";
 }
 if (d->flags & PCAP_IF_RUNNING) {
   printf("%sRUNNING", sep);
   sep = ", ";
 }
 if (d->flags & PCAP_IF_LOOPBACK) {
   printf("%sLOOPBACK", sep);
   sep = ", ";
 }
 if (d->flags & PCAP_IF_WIRELESS) {
   printf("%sWIRELESS", sep);
   switch (d->flags & PCAP_IF_CONNECTION_STATUS) {

   case PCAP_IF_CONNECTION_STATUS_UNKNOWN:
     printf(" (association status unknown)");
     break;

   case PCAP_IF_CONNECTION_STATUS_CONNECTED:
     printf(" (associated)");
     break;

   case PCAP_IF_CONNECTION_STATUS_DISCONNECTED:
     printf(" (not associated)");
     break;

   case PCAP_IF_CONNECTION_STATUS_NOT_APPLICABLE:
     break;
   }
 } else {
   switch (d->flags & PCAP_IF_CONNECTION_STATUS) {

   case PCAP_IF_CONNECTION_STATUS_UNKNOWN:
     printf(" (connection status unknown)");
     break;

   case PCAP_IF_CONNECTION_STATUS_CONNECTED:
     printf(" (connected)");
     break;

   case PCAP_IF_CONNECTION_STATUS_DISCONNECTED:
     printf(" (disconnected)");
     break;

   case PCAP_IF_CONNECTION_STATUS_NOT_APPLICABLE:
     break;
   }
 }
 sep = ", ";
 printf("\n");

 for(a=d->addresses;a;a=a->next) {
   if (a->addr != NULL)
     switch(a->addr->sa_family) {
     case AF_INET:
       printf("\tAddress Family: AF_INET (%d)\n", a->addr->sa_family);
       if (a->addr)
         printf("\t\tAddress: %s\n",
           inet_ntop(AF_INET,
              &((struct sockaddr_in *)(a->addr))->sin_addr,
              ipv4_buf, sizeof ipv4_buf));
       if (a->netmask)
         printf("\t\tNetmask: %s\n",
           inet_ntop(AF_INET,
              &((struct sockaddr_in *)(a->netmask))->sin_addr,
              ipv4_buf, sizeof ipv4_buf));
       if (a->broadaddr)
         printf("\t\tBroadcast Address: %s\n",
           inet_ntop(AF_INET,
              &((struct sockaddr_in *)(a->broadaddr))->sin_addr,
              ipv4_buf, sizeof ipv4_buf));
       if (a->dstaddr)
         printf("\t\tDestination Address: %s\n",
           inet_ntop(AF_INET,
              &((struct sockaddr_in *)(a->dstaddr))->sin_addr,
              ipv4_buf, sizeof ipv4_buf));
       break;
#ifdef INET6
     case AF_INET6:
       printf("\tAddress Family: AF_INET6 (%d)\n", a->addr->sa_family);
       if (a->addr)
         printf("\t\tAddress: %s\n",
           inet_ntop(AF_INET6,
              ((struct sockaddr_in6 *)(a->addr))->sin6_addr.s6_addr,
              ipv6_buf, sizeof ipv6_buf));
       if (a->netmask)
         printf("\t\tNetmask: %s\n",
           inet_ntop(AF_INET6,
             ((struct sockaddr_in6 *)(a->netmask))->sin6_addr.s6_addr,
              ipv6_buf, sizeof ipv6_buf));
       if (a->broadaddr)
         printf("\t\tBroadcast Address: %s\n",
           inet_ntop(AF_INET6,
             ((struct sockaddr_in6 *)(a->broadaddr))->sin6_addr.s6_addr,
              ipv6_buf, sizeof ipv6_buf));
       if (a->dstaddr)
         printf("\t\tDestination Address: %s\n",
           inet_ntop(AF_INET6,
             ((struct sockaddr_in6 *)(a->dstaddr))->sin6_addr.s6_addr,
              ipv6_buf, sizeof ipv6_buf));
       break;
#endif
     default:
       printf("\tAddress Family: Unknown (%d)\n", a->addr->sa_family);
       break;
     }
   else
   {
     fprintf(stderr, "\tWarning: a->addr is NULL, skipping this address.\n");
     status = 0;
   }
 }
 printf("\n");
 return status;
}

/* From tcptraceroute */
#define IPTOSBUFFERS    12
static char *iptos(bpf_u_int32 in)
{
       static char output[IPTOSBUFFERS][sizeof("255.255.255.255")];
       static short which;
       u_char *p;

       p = (u_char *)&in;
       which = (which + 1 == IPTOSBUFFERS ? 0 : which + 1);
       snprintf(output[which], sizeof(output[which]), "%d.%d.%d.%d", p[0], p[1], p[2], p[3]);
       return output[which];
}