r9272: Patch for fixing unused variables warning from Jason Mader. Fixes
[ira/wip.git] / source / utils / nmblookup.c
index 4e85f6ac4568b9b85402492bfe27f32939ad5805..09148ad37c30df2204f921d3a6f73cfd50720eb8 100644 (file)
@@ -1,8 +1,8 @@
 /* 
-   Unix SMB/Netbios implementation.
-   Version 1.9.
+   Unix SMB/CIFS implementation.
    NBT client - used to lookup netbios names
    Copyright (C) Andrew Tridgell 1994-1998
+   Copyright (C) Jelmer Vernooij 2003 (Conversion to popt)
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    
 */
 
-#define NO_SYSLOG
-
 #include "includes.h"
 
-extern int DEBUGLEVEL;
-
-extern struct in_addr ipzero;
+extern BOOL AllowDebugChange;
 
+static BOOL give_flags = False;
 static BOOL use_bcast = True;
 static BOOL got_bcast = False;
 static struct in_addr bcast_addr;
@@ -56,30 +53,6 @@ static BOOL open_sockets(void)
   return True;
 }
 
-
-/****************************************************************************
-usage on the program
-****************************************************************************/
-static void usage(void)
-{
-  printf("Usage: nmblookup [-M] [-B bcast address] [-d debuglevel] name\n");
-  printf("Version %s\n",VERSION);
-  printf("\t-d debuglevel         set the debuglevel\n");
-  printf("\t-B broadcast address  the address to use for broadcasts\n");
-  printf("\t-U unicast   address  the address to use for unicast\n");
-  printf("\t-M                    searches for a master browser\n");
-  printf("\t-R                    set recursion desired in packet\n");
-  printf("\t-S                    lookup node status as well\n");
-  printf("\t-T                    translate IP addresses into names\n");
-  printf("\t-r                    Use root port 137 (Win95 only replies to this)\n");
-  printf("\t-A                    Do a node status on <name> as an IP Address\n");
-  printf("\t-i NetBIOS scope      Use the given NetBIOS scope for name queries\n");
-  printf("\t-s smb.conf file      Use the given path to the smb.conf file\n");
-  printf("\t-h                    Print this help message.\n");
-  printf("\n  If you specify -M and name is \"-\", nmblookup looks up __MSBROWSE__<01>\n");
-  printf("\n");
-}
-
 /****************************************************************************
 turn a node status flags field into a string
 ****************************************************************************/
@@ -101,73 +74,101 @@ static char *node_status_flags(unsigned char flags)
        return ret;
 }
 
+/****************************************************************************
+turn the NMB Query flags into a string
+****************************************************************************/
+static char *query_flags(int flags)
+{
+       static fstring ret1;
+       fstrcpy(ret1, "");
+
+       if (flags & NM_FLAGS_RS) fstrcat(ret1, "Response ");
+       if (flags & NM_FLAGS_AA) fstrcat(ret1, "Authoritative ");
+       if (flags & NM_FLAGS_TC) fstrcat(ret1, "Truncated ");
+       if (flags & NM_FLAGS_RD) fstrcat(ret1, "Recursion_Desired ");
+       if (flags & NM_FLAGS_RA) fstrcat(ret1, "Recursion_Available ");
+       if (flags & NM_FLAGS_B)  fstrcat(ret1, "Broadcast ");
+
+       return ret1;
+}
+
 /****************************************************************************
 do a node status query
 ****************************************************************************/
-static void do_node_status(int fd, char *name, int type, struct in_addr ip)
+static void do_node_status(int fd, const char *name, int type, struct in_addr ip)
 {
        struct nmb_name nname;
        int count, i, j;
-       struct node_status *status;
+       NODE_STATUS_STRUCT *status;
+       struct node_status_extra extra;
        fstring cleanname;
 
-       printf("Looking up status of %s\n",inet_ntoa(ip));
+       d_printf("Looking up status of %s\n",inet_ntoa(ip));
        make_nmb_name(&nname, name, type);
-       status = name_status_query(fd,&nname,ip, &count);
+       status = node_status_query(fd,&nname,ip, &count, &extra);
        if (status) {
                for (i=0;i<count;i++) {
-                       fstrcpy(cleanname, status[i].name);
+                       pull_ascii_fstring(cleanname, status[i].name);
                        for (j=0;cleanname[j];j++) {
-                               if (!isprint(cleanname[j])) cleanname[j] = '.';
+                               if (!isprint((int)cleanname[j])) cleanname[j] = '.';
                        }
-                       printf("\t%-15s <%02x> - %s\n",
+                       d_printf("\t%-15s <%02x> - %s\n",
                               cleanname,status[i].type,
                               node_status_flags(status[i].flags));
                }
-               free(status);
+               d_printf("\n\tMAC Address = %02X-%02X-%02X-%02X-%02X-%02X\n",
+                               extra.mac_addr[0], extra.mac_addr[1],
+                               extra.mac_addr[2], extra.mac_addr[3],
+                               extra.mac_addr[4], extra.mac_addr[5]);
+               d_printf("\n");
+               SAFE_FREE(status);
+       } else {
+               d_printf("No reply from %s\n\n",inet_ntoa(ip));
        }
-       printf("\n");
 }
 
 
 /****************************************************************************
 send out one query
 ****************************************************************************/
-static BOOL query_one(char *lookup, unsigned int lookup_type)
+static BOOL query_one(const char *lookup, unsigned int lookup_type)
 {
-       int j, count;
+       int j, count, flags = 0;
        struct in_addr *ip_list=NULL;
 
        if (got_bcast) {
-               printf("querying %s on %s\n", lookup, inet_ntoa(bcast_addr));
+               d_printf("querying %s on %s\n", lookup, inet_ntoa(bcast_addr));
                ip_list = name_query(ServerFD,lookup,lookup_type,use_bcast,
                                     use_bcast?True:recursion_desired,
-                                    bcast_addr,&count);
+                                    bcast_addr,&count, &flags, NULL);
        } else {
                struct in_addr *bcast;
                for (j=iface_count() - 1;
                     !ip_list && j >= 0;
                     j--) {
                        bcast = iface_n_bcast(j);
-                       printf("querying %s on %s\n", 
+                       d_printf("querying %s on %s\n", 
                               lookup, inet_ntoa(*bcast));
                        ip_list = name_query(ServerFD,lookup,lookup_type,
                                             use_bcast,
                                             use_bcast?True:recursion_desired,
-                                            *bcast,&count);
+                                            *bcast,&count, &flags, NULL);
                }
        }
 
        if (!ip_list) return False;
 
+       if (give_flags)
+         d_printf("Flags: %s\n", query_flags(flags));
+
        for (j=0;j<count;j++) {
                if (translate_addresses) {
                        struct hostent *host = gethostbyaddr((char *)&ip_list[j], sizeof(ip_list[j]), AF_INET);
                        if (host) {
-                               printf("%s, ", host -> h_name);
+                               d_printf("%s, ", host -> h_name);
                        }
                }
-               printf("%s %s<%02x>\n",inet_ntoa(ip_list[j]),lookup, lookup_type);
+               d_printf("%s %s<%02x>\n",inet_ntoa(ip_list[j]),lookup, lookup_type);
        }
 
        /* We can only do find_status if the ip address returned
@@ -190,134 +191,107 @@ int main(int argc,char *argv[])
 {
   int opt;
   unsigned int lookup_type = 0x0;
-  pstring lookup;
-  extern int optind;
-  extern char *optarg;
-  BOOL find_master=False;
-  int i;
-  static pstring servicesf = CONFIGFILE;
-  BOOL lookup_by_ip = False;
-  int commandline_debuglevel = -2;
-
-  DEBUGLEVEL = 1;
+  fstring lookup;
+  static BOOL find_master=False;
+  static BOOL lookup_by_ip = False;
+  poptContext pc;
+
+  struct poptOption long_options[] = {
+         POPT_AUTOHELP
+         { "broadcast", 'B', POPT_ARG_STRING, NULL, 'B', "Specify address to use for broadcasts", "BROADCAST-ADDRESS" },
+         { "flags", 'f', POPT_ARG_VAL, &give_flags, True, "List the NMB flags returned" },
+         { "unicast", 'U', POPT_ARG_STRING, NULL, 'U', "Specify address to use for unicast" },
+         { "master-browser", 'M', POPT_ARG_VAL, &find_master, True, "Search for a master browser" },
+         { "recursion", 'R', POPT_ARG_VAL, &recursion_desired, True, "Set recursion desired in package" },
+         { "status", 'S', POPT_ARG_VAL, &find_status, True, "Lookup node status as well" },
+         { "translate", 'T', POPT_ARG_NONE, NULL, 'T', "Translate IP addresses into names" },
+         { "root-port", 'r', POPT_ARG_VAL, &RootPort, True, "Use root port 137 (Win95 only replies to this)" },
+         { "lookup-by-ip", 'A', POPT_ARG_VAL, &lookup_by_ip, True, "Do a node status on <name> as an IP Address" },
+         POPT_COMMON_SAMBA
+         POPT_COMMON_CONNECTION
+         { 0, 0, 0, 0 }
+  };
+       
   *lookup = 0;
 
-  TimeInit();
-
   setup_logging(argv[0],True);
 
-  charset_initialise();
-
-  while ((opt = getopt(argc, argv, "d:B:U:i:s:SMrhART")) != EOF)
-    switch (opt)
-      {
-      case 'B':
-       bcast_addr = *interpret_addr2(optarg);
-       got_bcast = True;
-       use_bcast = True;
-       break;
-      case 'U':
-       bcast_addr = *interpret_addr2(optarg);
-       got_bcast = True;
-       use_bcast = False;
-       break;
-      case 'T':
-        translate_addresses = !translate_addresses;
-       break;
-      case 'i':
-             {
-                     extern pstring global_scope;
-                     pstrcpy(global_scope,optarg);
-                     strupper(global_scope);
-             }
-             break;
-      case 'M':
-       find_master = True;
-       break;
-      case 'S':
-       find_status = True;
-       break;
-      case 'R':
-       recursion_desired = True;
-       break;
-      case 'd':
-       commandline_debuglevel = DEBUGLEVEL = atoi(optarg);
-       break;
-      case 's':
-       pstrcpy(servicesf, optarg);
-       break;
-      case 'r':
-        RootPort = True;
-        break;
-      case 'h':
-       usage();
-       exit(0);
-       break;
-      case 'A':
-        lookup_by_ip = True;
-        break;
-      default:
-       usage();
-       exit(1);
-      }
-
-  if (argc < 2) {
-    usage();
-    exit(1);
+  pc = poptGetContext("nmblookup", argc, (const char **)argv, long_options, 
+                                         POPT_CONTEXT_KEEP_FIRST);
+
+  poptSetOtherOptionHelp(pc, "<NODE> ...");
+
+  while ((opt = poptGetNextOpt(pc)) != -1) {
+         switch (opt) {
+         case 'B':
+                 bcast_addr = *interpret_addr2(poptGetOptArg(pc));
+                 got_bcast = True;
+                 use_bcast = True;
+                 break;
+         case 'U':
+                 bcast_addr = *interpret_addr2(poptGetOptArg(pc));
+                 got_bcast = True;
+                 use_bcast = False;
+                 break;
+         case 'T':
+                 translate_addresses = !translate_addresses;
+                 break;
+         }
   }
 
-  if (!lp_load(servicesf,True,False,False)) {
-    fprintf(stderr, "Can't load %s - run testparm to debug it\n", servicesf);
-  }
+  poptGetArg(pc); /* Remove argv[0] */
 
-  /*
-   * Ensure we reset DEBUGLEVEL if someone specified it
-   * on the command line.
-   */
+  if(!poptPeekArg(pc)) { 
+         poptPrintUsage(pc, stderr, 0);
+         exit(1);
+  }
 
-  if(commandline_debuglevel != -2)
-    DEBUGLEVEL = commandline_debuglevel;
+  if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
+         fprintf(stderr, "Can't load %s - run testparm to debug it\n", dyn_CONFIGFILE);
+  }
 
   load_interfaces();
   if (!open_sockets()) return(1);
 
-  for (i=optind;i<argc;i++)
+  while(poptPeekArg(pc))
   {
-      char *p;
-      struct in_addr ip;
-
-      fstrcpy(lookup,argv[i]);
-
-      if(lookup_by_ip)
-      {
-        fstrcpy(lookup,"*");
-        ip = *interpret_addr2(argv[i]);
-       do_node_status(ServerFD, lookup, lookup_type, ip);
-        continue;
-      }
-
-      if (find_master) {
-       if (*lookup == '-') {
-         fstrcpy(lookup,"\01\02__MSBROWSE__\02");
-         lookup_type = 1;
-       } else {
-         lookup_type = 0x1d;
-       }
-      }
-
-      p = strchr(lookup,'#');
-      if (p) {
-        *p = '\0';
-        sscanf(++p,"%x",&lookup_type);
-      }
-
-      if (!query_one(lookup, lookup_type)) {
-       printf( "name_query failed to find name %s", lookup );
-        if( 0 != lookup_type )
-          printf( "#%02x", lookup_type );
-        printf( "\n" );
-      }
+         char *p;
+         struct in_addr ip;
+
+         fstrcpy(lookup,poptGetArg(pc));
+
+         if(lookup_by_ip)
+         {
+                 ip = *interpret_addr2(lookup);
+                 fstrcpy(lookup,"*");
+                 do_node_status(ServerFD, lookup, lookup_type, ip);
+                 continue;
+         }
+
+         if (find_master) {
+                 if (*lookup == '-') {
+                         fstrcpy(lookup,"\01\02__MSBROWSE__\02");
+                         lookup_type = 1;
+                 } else {
+                         lookup_type = 0x1d;
+                 }
+         }
+
+         p = strchr_m(lookup,'#');
+         if (p) {
+                 *p = '\0';
+                 sscanf(++p,"%x",&lookup_type);
+         }
+
+         if (!query_one(lookup, lookup_type)) {
+                 d_printf( "name_query failed to find name %s", lookup );
+                 if( 0 != lookup_type )
+                         d_printf( "#%02x", lookup_type );
+                 d_printf( "\n" );
+         }
   }
-  
+
+  poptFreeContext(pc);
+
   return(0);
 }