Add auto-detect of time server. Uses master browser.
[kai/samba.git] / source3 / utils / nmblookup.c
index b14887afabea4fb7db8cd02c18796fe39e15349c..9549d16d044096d502264a683141873b03221d2f 100644 (file)
@@ -1,8 +1,7 @@
 /* 
-   Unix SMB/Netbios implementation.
-   Version 1.9.
+   Unix SMB/CIFS implementation.
    NBT client - used to lookup netbios names
-   Copyright (C) Andrew Tridgell 1994-1997
+   Copyright (C) Andrew Tridgell 1994-1998
    
    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
    
 */
 
-#ifdef SYSLOG
-#undef SYSLOG
-#endif
+#define NO_SYSLOG
 
 #include "includes.h"
 
-extern int DEBUGLEVEL;
+extern BOOL AllowDebugChange;
 
-extern pstring scope;
-
-extern pstring myhostname;
-extern struct in_addr ipzero;
-
-int ServerFD= -1;
-
-int RootPort = 0;
+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;
+static int ServerFD= -1;
+static int RootPort = False;
+static BOOL find_status=False;
 
 /****************************************************************************
   open the socket communication
@@ -43,14 +40,14 @@ int RootPort = 0;
 static BOOL open_sockets(void)
 {
   ServerFD = open_socket_in( SOCK_DGRAM,
-                             (RootPort ? 137 :0),
-                             3,
-                             interpret_addr(lp_socket_address()) );
+                             (RootPort ? 137 : 0),
+                             (RootPort ?   0 : 3),
+                             interpret_addr(lp_socket_address()), True );
 
   if (ServerFD == -1)
     return(False);
 
-  set_socket_options(ServerFD,"SO_BROADCAST");
+  set_socket_options( ServerFD, "SO_BROADCAST" );
 
   DEBUG(3, ("Socket opened.\n"));
   return True;
@@ -58,32 +55,128 @@ static BOOL open_sockets(void)
 
 
 /****************************************************************************
-  initialise connect, service and file structs
+usage on the program
 ****************************************************************************/
-static BOOL init_structs(void )
+static void usage(void)
 {
-  if (!get_myname(myhostname,NULL))
-    return(False);
+  d_printf("Usage: nmblookup [-M] [-B bcast address] [-d debuglevel] 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-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");
+}
 
-  return True;
+/****************************************************************************
+turn a node status flags field into a string
+****************************************************************************/
+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 ");
+       if ((flags & 0x60) == 0x40) fstrcat(ret,"M ");
+       if ((flags & 0x60) == 0x60) fstrcat(ret,"H ");
+       if (flags & 0x10) fstrcat(ret,"<DEREGISTERING> ");
+       if (flags & 0x08) fstrcat(ret,"<CONFLICT> ");
+       if (flags & 0x04) fstrcat(ret,"<ACTIVE> ");
+       if (flags & 0x02) fstrcat(ret,"<PERMANENT> ");
+       
+       return ret;
 }
 
 /****************************************************************************
-usage on the program
+do a node status query
 ****************************************************************************/
-static void usage(void)
+static void do_node_status(int fd, char *name, int type, struct in_addr ip)
+{
+       struct nmb_name nname;
+       int count, i, j;
+       struct node_status *status;
+       fstring cleanname;
+
+       d_printf("Looking up status of %s\n",inet_ntoa(ip));
+       make_nmb_name(&nname, name, type);
+       status = node_status_query(fd,&nname,ip, &count);
+       if (status) {
+               for (i=0;i<count;i++) {
+                       fstrcpy(cleanname, status[i].name);
+                       for (j=0;cleanname[j];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));
+               }
+               SAFE_FREE(status);
+       }
+       d_printf("\n");
+}
+
+
+/****************************************************************************
+send out one query
+****************************************************************************/
+static BOOL query_one(char *lookup, unsigned int lookup_type)
 {
-  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-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("\n");
+       int j, count;
+       struct in_addr *ip_list=NULL;
+
+       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);
+       } else {
+               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);
+               }
+       }
+
+       if (!ip_list) return False;
+
+       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) {
+                               d_printf("%s, ", host -> h_name);
+                       }
+               }
+               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);
+
+       return (ip_list != NULL);
 }
 
 
@@ -98,43 +191,41 @@ int main(int argc,char *argv[])
   extern int optind;
   extern char *optarg;
   BOOL find_master=False;
-  BOOL find_status=False;
   int i;
-  static pstring servicesf = CONFIGFILE;
-  struct in_addr bcast_addr;
-  BOOL use_bcast = True;
-  BOOL got_bcast = False;
   BOOL lookup_by_ip = False;
-  BOOL recursion_desired = False;
+  int commandline_debuglevel = -2;
 
   DEBUGLEVEL = 1;
-  *lookup = 0;
+  /* Prevent smb.conf setting from overridding */
+  AllowDebugChange = False;
 
-  TimeInit();
+  *lookup = 0;
 
   setup_logging(argv[0],True);
 
-  charset_initialise();
-
-  while ((opt = getopt(argc, argv, "d:B:U:i:s:SMrhAR")) != EOF)
+  while ((opt = getopt(argc, argv, "d:B:U:i:s:SMrhART")) != EOF)
     switch (opt)
       {
       case 'B':
-       iface_set_default(NULL,optarg,NULL);
        bcast_addr = *interpret_addr2(optarg);
        got_bcast = True;
        use_bcast = True;
        break;
       case 'U':
-       iface_set_default(NULL,optarg,NULL);
        bcast_addr = *interpret_addr2(optarg);
        got_bcast = True;
        use_bcast = False;
        break;
-      case 'i':
-       fstrcpy(scope,optarg);
-       strupper(scope);
+      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;
@@ -145,13 +236,13 @@ int main(int argc,char *argv[])
        recursion_desired = True;
        break;
       case 'd':
-       DEBUGLEVEL = atoi(optarg);
+       commandline_debuglevel = DEBUGLEVEL = atoi(optarg);
        break;
       case 's':
-       pstrcpy(servicesf, optarg);
+       pstrcpy(dyn_CONFIGFILE, optarg);
        break;
       case 'r':
-        RootPort = -1;
+        RootPort = True;
         break;
       case 'h':
        usage();
@@ -170,71 +261,56 @@ int main(int argc,char *argv[])
     exit(1);
   }
 
-  if (!lp_load(servicesf,True)) {
-    fprintf(stderr, "Can't load %s - run testparm to debug it\n", servicesf);
+  if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
+    fprintf(stderr, "Can't load %s - run testparm to debug it\n", dyn_CONFIGFILE);
   }
 
-  load_interfaces();
-  init_structs();
-  if (!open_sockets()) return(1);
-
-  if (!got_bcast)
-    bcast_addr = *iface_bcast(ipzero);
+  /*
+   * Ensure we reset DEBUGLEVEL if someone specified it
+   * on the command line.
+   */
 
-  DEBUG(1,("Sending queries to %s\n",inet_ntoa(bcast_addr)));
+  if(commandline_debuglevel != -2)
+    DEBUGLEVEL = commandline_debuglevel;
 
+  load_interfaces();
+  if (!open_sockets()) return(1);
 
   for (i=optind;i<argc;i++)
   {
-      int j, count, retries = 2;
       char *p;
       struct in_addr ip;
-      struct in_addr *ip_list;
 
       fstrcpy(lookup,argv[i]);
 
       if(lookup_by_ip)
       {
-        strcpy(lookup,"*");
+        fstrcpy(lookup,"*");
         ip = *interpret_addr2(argv[i]);
-        printf("Looking up status of %s\n",inet_ntoa(ip));
-        name_status(ServerFD,lookup,lookup_type,True,ip,NULL,NULL,NULL);
-        printf("\n");
+       do_node_status(ServerFD, lookup, lookup_type, ip);
         continue;
       }
 
       if (find_master) {
        if (*lookup == '-') {
-         strcpy(lookup,"\01\02__MSBROWSE__\02");
+         fstrcpy(lookup,"\01\02__MSBROWSE__\02");
          lookup_type = 1;
        } else {
          lookup_type = 0x1d;
        }
       }
 
-      p = strchr(lookup,'#');
-
+      p = strchr_m(lookup,'#');
       if (p) {
-       *p = 0;
-       sscanf(p+1,"%x",&lookup_type);
-       retries = 1;
+        *p = '\0';
+        sscanf(++p,"%x",&lookup_type);
       }
 
-      if ((ip_list = name_query(ServerFD,lookup,lookup_type,use_bcast,recursion_desired,
-                               bcast_addr,&count,NULL))) {
-             for (j=0;j<count;j++)
-                     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) {
-                     printf("Looking up status of %s\n",inet_ntoa(ip_list[0]));
-                     name_status(ServerFD,lookup,lookup_type,True,ip_list[0],NULL,NULL,NULL);
-                     printf("\n");
-             }
-      } else {
-             printf("name_query failed to find name %s\n", lookup);
+      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" );
       }
   }