s3-masktest: Fix cli_errstr() usage (part of bug #7864)
[ira/wip.git] / source3 / utils / nmblookup.c
index 7e0ed4a2030287d5176de9d57e3c69c35e5d6962..bd150df74131c82ef1417316b279e030e9323d20 100644 (file)
@@ -1,82 +1,65 @@
-/* 
+/*
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
-   
-*/
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-#define NO_SYSLOG
+*/
 
 #include "includes.h"
-
-extern BOOL AllowDebugChange;
-
-static BOOL give_flags = False;
-static BOOL use_bcast = True;
-static BOOL got_bcast = False;
-static struct in_addr bcast_addr;
-static BOOL recursion_desired = False;
-static BOOL translate_addresses = False;
+#include "popt_common.h"
+
+static bool give_flags = false;
+static bool use_bcast = true;
+static bool got_bcast = false;
+static struct sockaddr_storage bcast_addr;
+static bool recursion_desired = false;
+static bool translate_addresses = false;
 static int ServerFD= -1;
-static int RootPort = False;
-static BOOL find_status=False;
+static bool RootPort = false;
+static bool find_status = false;
 
 /****************************************************************************
-  open the socket communication
-  **************************************************************************/
-static BOOL open_sockets(void)
+ Open the socket communication.
+**************************************************************************/
+
+static bool open_sockets(void)
 {
-  ServerFD = open_socket_in( SOCK_DGRAM,
-                             (RootPort ? 137 : 0),
-                             (RootPort ?   0 : 3),
-                             interpret_addr(lp_socket_address()), True );
+       struct sockaddr_storage ss;
+       const char *sock_addr = lp_socket_address();
+
+       if (!interpret_string_addr(&ss, sock_addr,
+                               AI_NUMERICHOST|AI_PASSIVE)) {
+               DEBUG(0,("open_sockets: unable to get socket address "
+                                       "from string %s", sock_addr));
+               return false;
+       }
+       ServerFD = open_socket_in( SOCK_DGRAM,
+                               (RootPort ? 137 : 0),
+                               (RootPort ?   0 : 3),
+                               &ss, true );
 
-  if (ServerFD == -1)
-    return(False);
+       if (ServerFD == -1) {
+               return false;
+       }
 
-  set_socket_options( ServerFD, "SO_BROADCAST" );
+       set_socket_options( ServerFD, "SO_BROADCAST" );
 
-  DEBUG(3, ("Socket opened.\n"));
-  return True;
-}
-
-
-/****************************************************************************
-usage on the program
-****************************************************************************/
-static void usage(void)
-{
-  d_printf("Usage: nmblookup [options] name\n");
-  d_printf("Version %s\n",VERSION);
-  d_printf("\t-d debuglevel         set the debuglevel\n");
-  d_printf("\t-B broadcast address  the address to use for broadcasts\n");
-  d_printf("\t-f                    list the NMB flags returned\n");
-  d_printf("\t-U unicast   address  the address to use for unicast\n");
-  d_printf("\t-M                    searches for a master browser\n");
-  d_printf("\t-R                    set recursion desired in packet\n");
-  d_printf("\t-S                    lookup node status as well\n");
-  d_printf("\t-T                    translate IP addresses into names\n");
-  d_printf("\t-r                    Use root port 137 (Win95 only replies to this)\n");
-  d_printf("\t-A                    Do a node status on <name> as an IP Address\n");
-  d_printf("\t-i NetBIOS scope      Use the given NetBIOS scope for name queries\n");
-  d_printf("\t-s smb.conf file      Use the given path to the smb.conf file\n");
-  d_printf("\t-h                    Print this help message.\n");
-  d_printf("\n  If you specify -M and name is \"-\", nmblookup looks up __MSBROWSE__<01>\n");
-  d_printf("\n");
+       DEBUG(3, ("Socket opened.\n"));
+       return true;
 }
 
 /****************************************************************************
@@ -86,7 +69,7 @@ static char *node_status_flags(unsigned char flags)
 {
        static fstring ret;
        fstrcpy(ret,"");
-       
+
        fstrcat(ret, (flags & 0x80) ? "<GROUP> " : "        ");
        if ((flags & 0x60) == 0x00) fstrcat(ret,"B ");
        if ((flags & 0x60) == 0x20) fstrcat(ret,"P ");
@@ -96,13 +79,14 @@ static char *node_status_flags(unsigned char flags)
        if (flags & 0x08) fstrcat(ret,"<CONFLICT> ");
        if (flags & 0x04) fstrcat(ret,"<ACTIVE> ");
        if (flags & 0x02) fstrcat(ret,"<PERMANENT> ");
-       
+
        return ret;
 }
 
 /****************************************************************************
-turn the NMB Query flags into a string
+ Turn the NMB Query flags into a string.
 ****************************************************************************/
+
 static char *query_flags(int flags)
 {
        static fstring ret1;
@@ -119,87 +103,128 @@ static char *query_flags(int flags)
 }
 
 /****************************************************************************
-do a node status query
+ Do a node status query.
 ****************************************************************************/
-static void do_node_status(int fd, const char *name, int type, struct in_addr ip)
+
+static void do_node_status(const char *name,
+               int type,
+               struct sockaddr_storage *pss)
 {
        struct nmb_name nname;
        int count, i, j;
-       struct node_status *status;
+       struct node_status *addrs;
+       struct node_status_extra extra;
        fstring cleanname;
+       char addr[INET6_ADDRSTRLEN];
+       NTSTATUS status;
 
-       d_printf("Looking up status of %s\n",inet_ntoa(ip));
+       print_sockaddr(addr, sizeof(addr), pss);
+       d_printf("Looking up status of %s\n",addr);
        make_nmb_name(&nname, name, type);
-       status = node_status_query(fd,&nname,ip, &count);
-       if (status) {
+       status = node_status_query(talloc_tos(), &nname, pss,
+                                  &addrs, &count, &extra);
+       if (NT_STATUS_IS_OK(status)) {
                for (i=0;i<count;i++) {
-                       fstrcpy(cleanname, status[i].name);
+                       pull_ascii_fstring(cleanname, addrs[i].name);
                        for (j=0;cleanname[j];j++) {
-                               if (!isprint((int)cleanname[j])) cleanname[j] = '.';
+                               if (!isprint((int)cleanname[j])) {
+                                       cleanname[j] = '.';
+                               }
                        }
                        d_printf("\t%-15s <%02x> - %s\n",
-                              cleanname,status[i].type,
-                              node_status_flags(status[i].flags));
+                              cleanname,addrs[i].type,
+                              node_status_flags(addrs[i].flags));
                }
-               SAFE_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");
+               TALLOC_FREE(addrs);
+       } else {
+               d_printf("No reply from %s\n\n",addr);
        }
-       d_printf("\n");
 }
 
 
 /****************************************************************************
-send out one query
+ Send out one query.
 ****************************************************************************/
-static BOOL query_one(const char *lookup, unsigned int lookup_type)
+
+static bool query_one(const char *lookup, unsigned int lookup_type)
 {
-       int j, count, flags = 0;
-       struct in_addr *ip_list=NULL;
+       int j, count;
+       uint8_t flags;
+       struct sockaddr_storage *ip_list=NULL;
+       NTSTATUS status = NT_STATUS_NOT_FOUND;
 
        if (got_bcast) {
-               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, &flags, NULL);
+               char addr[INET6_ADDRSTRLEN];
+               print_sockaddr(addr, sizeof(addr), &bcast_addr);
+               d_printf("querying %s on %s\n", lookup, addr);
+               status = name_query(lookup,lookup_type,use_bcast,
+                                   use_bcast?true:recursion_desired,
+                                   &bcast_addr, talloc_tos(),
+                                   &ip_list, &count, &flags);
        } else {
-               struct in_addr *bcast;
+               const struct in_addr *bcast;
                for (j=iface_count() - 1;
                     !ip_list && j >= 0;
                     j--) {
-                       bcast = iface_n_bcast(j);
-                       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, &flags, NULL);
+                       char addr[INET6_ADDRSTRLEN];
+                       struct sockaddr_storage bcast_ss;
+
+                       bcast = iface_n_bcast_v4(j);
+                       if (!bcast) {
+                               continue;
+                       }
+                       in_addr_to_sockaddr_storage(&bcast_ss, *bcast);
+                       print_sockaddr(addr, sizeof(addr), &bcast_ss);
+                       d_printf("querying %s on %s\n",
+                              lookup, addr);
+                       status = name_query(lookup,lookup_type,
+                                           use_bcast,
+                                           use_bcast?True:recursion_desired,
+                                           &bcast_ss, talloc_tos(),
+                                           &ip_list, &count, &flags);
                }
        }
 
-       if (!ip_list) return False;
+       if (!NT_STATUS_IS_OK(status)) {
+               return false;
+       }
 
-       if (give_flags)
-         d_printf("Flags: %s\n", query_flags(flags));
+       if (give_flags) {
+               d_printf("Flags: %s\n", query_flags(flags));
+       }
 
        for (j=0;j<count;j++) {
+               char addr[INET6_ADDRSTRLEN];
                if (translate_addresses) {
-                       struct hostent *host = gethostbyaddr((char *)&ip_list[j], sizeof(ip_list[j]), AF_INET);
-                       if (host) {
-                               d_printf("%s, ", host -> h_name);
+                       char h_name[MAX_DNS_NAME_LENGTH];
+                       h_name[0] = '\0';
+                       if (sys_getnameinfo((const struct sockaddr *)&ip_list[j],
+                                       sizeof(struct sockaddr_storage),
+                                       h_name, sizeof(h_name),
+                                       NULL, 0,
+                                       NI_NAMEREQD)) {
+                               continue;
                        }
+                       d_printf("%s, ", h_name);
+               }
+               print_sockaddr(addr, sizeof(addr), &ip_list[j]);
+               d_printf("%s %s<%02x>\n", addr,lookup, lookup_type);
+               /* We can only do find_status if the ip address returned
+                  was valid - ie. name_query returned true.
+                */
+               if (find_status) {
+                       do_node_status(lookup, lookup_type, &ip_list[j]);
                }
-               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
-          was valid - ie. name_query returned true.
-       */
-       if (find_status) {
-               do_node_status(ServerFD, lookup, lookup_type, ip_list[0]);
        }
 
-       safe_free(ip_list);
+       TALLOC_FREE(ip_list);
 
-       return (ip_list != NULL);
+       return NT_STATUS_IS_OK(status);
 }
 
 
@@ -208,133 +233,140 @@ static BOOL query_one(const char *lookup, unsigned int lookup_type)
 ****************************************************************************/
 int main(int argc,char *argv[])
 {
-  int opt;
-  unsigned int lookup_type = 0x0;
-  fstring lookup;
-  extern int optind;
-  extern char *optarg;
-  BOOL find_master=False;
-  int i;
-  BOOL lookup_by_ip = False;
-  int commandline_debuglevel = -2;
-
-  DEBUGLEVEL = 1;
-  /* Prevent smb.conf setting from overridding */
-  AllowDebugChange = False;
-
-  *lookup = 0;
-
-  setup_logging(argv[0],True);
-
-  while ((opt = getopt(argc, argv, "d:fB:U:i:s:SMrhART")) != EOF)
-    switch (opt)
-      {
-      case 'B':
-       bcast_addr = *interpret_addr2(optarg);
-       got_bcast = True;
-       use_bcast = True;
-       break;
-      case 'f':
-       give_flags = 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':
-       set_global_scope(optarg);
-       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(dyn_CONFIGFILE, 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);
-  }
-
-  if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
-    fprintf(stderr, "Can't load %s - run testparm to debug it\n", dyn_CONFIGFILE);
-  }
-
-  /*
-   * Ensure we reset DEBUGLEVEL if someone specified it
-   * on the command line.
-   */
-
-  if(commandline_debuglevel != -2)
-    DEBUGLEVEL = commandline_debuglevel;
-
-  load_interfaces();
-  if (!open_sockets()) return(1);
-
-  for (i=optind;i<argc;i++)
-  {
-      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;
+       int opt;
+       unsigned int lookup_type = 0x0;
+       fstring lookup;
+       static bool find_master=False;
+       static bool lookup_by_ip = False;
+       poptContext pc;
+       TALLOC_CTX *frame = talloc_stackframe();
+
+       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_NONE, NULL, 'f', "List the NMB flags returned" },
+               { "unicast", 'U', POPT_ARG_STRING, NULL, 'U', "Specify address to use for unicast" },
+               { "master-browser", 'M', POPT_ARG_NONE, NULL, 'M', "Search for a master browser" },
+               { "recursion", 'R', POPT_ARG_NONE, NULL, 'R', "Set recursion desired in package" },
+               { "status", 'S', POPT_ARG_NONE, NULL, 'S', "Lookup node status as well" },
+               { "translate", 'T', POPT_ARG_NONE, NULL, 'T', "Translate IP addresses into names" },
+               { "root-port", 'r', POPT_ARG_NONE, NULL, 'r', "Use root port 137 (Win95 only replies to this)" },
+               { "lookup-by-ip", 'A', POPT_ARG_NONE, NULL, 'A', "Do a node status on <name> as an IP Address" },
+               POPT_COMMON_SAMBA
+               POPT_COMMON_CONNECTION
+               { 0, 0, 0, 0 }
+       };
+
+       *lookup = 0;
+
+       load_case_tables();
+
+       setup_logging(argv[0], DEBUG_STDOUT);
+
+       pc = poptGetContext("nmblookup", argc, (const char **)argv,
+                       long_options, POPT_CONTEXT_KEEP_FIRST);
+
+       poptSetOtherOptionHelp(pc, "<NODE> ...");
+
+       while ((opt = poptGetNextOpt(pc)) != -1) {
+               switch (opt) {
+               case 'f':
+                       give_flags = true;
+                       break;
+               case 'M':
+                       find_master = true;
+                       break;
+               case 'R':
+                       recursion_desired = true;
+                       break;
+               case 'S':
+                       find_status = true;
+                       break;
+               case 'r':
+                       RootPort = true;
+                       break;
+               case 'A':
+                       lookup_by_ip = true;
+                       break;
+               case 'B':
+                       if (interpret_string_addr(&bcast_addr,
+                                       poptGetOptArg(pc),
+                                       NI_NUMERICHOST)) {
+                               got_bcast = True;
+                               use_bcast = True;
+                       }
+                       break;
+               case 'U':
+                       if (interpret_string_addr(&bcast_addr,
+                                       poptGetOptArg(pc),
+                                       0)) {
+                               got_bcast = True;
+                               use_bcast = False;
+                       }
+                       break;
+               case 'T':
+                       translate_addresses = !translate_addresses;
+                       break;
+               }
+       }
+
+       poptGetArg(pc); /* Remove argv[0] */
+
+       if(!poptPeekArg(pc)) {
+               poptPrintUsage(pc, stderr, 0);
+               exit(1);
+       }
+
+       if (!lp_load(get_dyn_CONFIGFILE(),True,False,False,True)) {
+               fprintf(stderr, "Can't load %s - run testparm to debug it\n",
+                               get_dyn_CONFIGFILE());
        }
-      }
-
-      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" );
-      }
-  }
-  
-  return(0);
+
+       load_interfaces();
+       if (!open_sockets()) {
+               return(1);
+       }
+
+       while(poptPeekArg(pc)) {
+               char *p;
+               struct in_addr ip;
+
+               fstrcpy(lookup,poptGetArg(pc));
+
+               if(lookup_by_ip) {
+                       struct sockaddr_storage ss;
+                       ip = interpret_addr2(lookup);
+                       in_addr_to_sockaddr_storage(&ss, ip);
+                       fstrcpy(lookup,"*");
+                       do_node_status(lookup, lookup_type, &ss);
+                       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);
+       TALLOC_FREE(frame);
+       return(0);
 }