s3-talloc Change TALLOC_ARRAY() to talloc_array()
[samba.git] / source3 / winbindd / winbindd_wins.c
index 2ee6f69b66a77ff75b25ce2b8a3f7e06e3df60cf..7ed330f81bcc4e127a51858664be278f04b3152a 100644 (file)
 
 #include "includes.h"
 #include "winbindd.h"
+#include "libsmb/nmblib.h"
 
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_WINBIND
 
-/* Use our own create socket code so we don't recurse.... */
-
-static int wins_lookup_open_socket_in(void)
+static struct sockaddr_storage *lookup_byname_backend(TALLOC_CTX *mem_ctx,
+                                                     const char *name,
+                                                     int *count)
 {
-       struct sockaddr_in sock;
-       int val=1;
-       int res;
-
-       memset((char *)&sock,'\0',sizeof(sock));
-
-#ifdef HAVE_SOCK_SIN_LEN
-       sock.sin_len = sizeof(sock);
-#endif
-       sock.sin_port = 0;
-       sock.sin_family = AF_INET;
-       sock.sin_addr.s_addr = interpret_addr("0.0.0.0");
-       res = socket(AF_INET, SOCK_DGRAM, 0);
-       if (res == -1)
-               return -1;
-
-       setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val));
-#ifdef SO_REUSEPORT
-       setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val));
-#endif /* SO_REUSEPORT */
-
-       /* now we've got a socket - we need to bind it */
-
-       if (bind(res, (struct sockaddr * ) &sock,sizeof(sock)) < 0) {
-               close(res);
-               return(-1);
-       }
-
-       set_socket_options(res,"SO_BROADCAST");
-
-       return res;
-}
-
-
-static NODE_STATUS_STRUCT *lookup_byaddr_backend(char *addr, int *count)
-{
-       int fd;
-       struct in_addr  ip;
-       struct nmb_name nname;
-       NODE_STATUS_STRUCT *status;
-
-       fd = wins_lookup_open_socket_in();
-       if (fd == -1)
-               return NULL;
-
-       make_nmb_name(&nname, "*", 0);
-       ip = *interpret_addr2(addr);
-       status = node_status_query(fd,&nname,ip, count, NULL);
-
-       close(fd);
-       return status;
-}
-
-static struct in_addr *lookup_byname_backend(const char *name, int *count)
-{
-       int fd;
        struct ip_service *ret = NULL;
-       struct in_addr *return_ip = NULL;
-       int j, i, flags = 0;
+       struct sockaddr_storage *return_ss = NULL;
+       int j, i;
+       NTSTATUS status;
 
        *count = 0;
 
@@ -96,142 +42,96 @@ static struct in_addr *lookup_byname_backend(const char *name, int *count)
        if (NT_STATUS_IS_OK(resolve_wins(name,0x20,&ret,count))) {
                if ( *count == 0 )
                        return NULL;
-               if ( (return_ip = SMB_MALLOC_ARRAY(struct in_addr, *count)) == NULL ) {
+               return_ss = talloc_array(mem_ctx, struct sockaddr_storage,
+                                        *count);
+               if (return_ss == NULL ) {
                        free( ret );
                        return NULL;
                }
 
                /* copy the IP addresses */
-               for ( i=0; i<(*count); i++ ) 
-                       return_ip[i] = ret[i].ip;
-               
-               free( ret );
-               return return_ip;
-       }
+               for ( i=0; i<(*count); i++ )
+                       return_ss[i] = ret[i].ss;
 
-       fd = wins_lookup_open_socket_in();
-       if (fd == -1) {
-               return NULL;
+               free( ret );
+               return return_ss;
        }
 
        /* uggh, we have to broadcast to each interface in turn */
        for (j=iface_count() - 1;
             j >= 0;
             j--) {
-               const struct in_addr *bcast = iface_n_bcast_v4(j);
-               if (!bcast) {
+               const struct sockaddr_storage *bcast_ss = iface_n_bcast(j);
+               if (!bcast_ss) {
                        continue;
                }
-               return_ip = name_query(fd,name,0x20,True,True,*bcast,count, &flags, NULL);
-               if (return_ip) {
+               status = name_query(name, 0x20, True, True,bcast_ss,
+                                   mem_ctx, &return_ss, count, NULL);
+               if (NT_STATUS_IS_OK(status)) {
                        break;
                }
        }
 
-       close(fd);
-       return return_ip;
-}
-
-/* Get hostname from IP  */
-
-void winbindd_wins_byip(struct winbindd_cli_state *state)
-{
-       fstring response;
-       int i, count, maxlen, size;
-       NODE_STATUS_STRUCT *status;
-
-       /* Ensure null termination */
-       state->request.data.winsreq[sizeof(state->request.data.winsreq)-1]='\0';
-
-       DEBUG(3, ("[%5lu]: wins_byip %s\n", (unsigned long)state->pid,
-               state->request.data.winsreq));
-
-       *response = '\0';
-       maxlen = sizeof(response) - 1;
-
-       if ((status = lookup_byaddr_backend(state->request.data.winsreq, &count))){
-           size = strlen(state->request.data.winsreq);
-           if (size > maxlen) {
-               SAFE_FREE(status);
-               request_error(state);
-               return;
-           }
-           fstrcat(response,state->request.data.winsreq);
-           fstrcat(response,"\t");
-           for (i = 0; i < count; i++) {
-               /* ignore group names */
-               if (status[i].flags & 0x80) continue;
-               if (status[i].type == 0x20) {
-                       size = sizeof(status[i].name) + strlen(response);
-                       if (size > maxlen) {
-                           SAFE_FREE(status);
-                           request_error(state);
-                           return;
-                       }
-                       fstrcat(response, status[i].name);
-                       fstrcat(response, " ");
-               }
-           }
-           /* make last character a newline */
-           response[strlen(response)-1] = '\n';
-           SAFE_FREE(status);
-       }
-       fstrcpy(state->response.data.winsresp,response);
-       request_ok(state);
+       return return_ss;
 }
 
 /* Get IP from hostname */
 
 void winbindd_wins_byname(struct winbindd_cli_state *state)
 {
-       struct in_addr *ip_list;
+       struct sockaddr_storage *ip_list = NULL;
        int i, count, maxlen, size;
        fstring response;
-       char * addr;
+       char addr[INET6_ADDRSTRLEN];
 
        /* Ensure null termination */
-       state->request.data.winsreq[sizeof(state->request.data.winsreq)-1]='\0';
+       state->request->data.winsreq[sizeof(state->request->data.winsreq)-1]='\0';
 
        DEBUG(3, ("[%5lu]: wins_byname %s\n", (unsigned long)state->pid,
-               state->request.data.winsreq));
+               state->request->data.winsreq));
 
        *response = '\0';
        maxlen = sizeof(response) - 1;
 
-       if ((ip_list = lookup_byname_backend(state->request.data.winsreq,&count))){
+       ip_list = lookup_byname_backend(
+               state->mem_ctx, state->request->data.winsreq, &count);
+       if (ip_list != NULL){
                for (i = count; i ; i--) {
-                   addr = inet_ntoa(ip_list[i-1]);
-                   size = strlen(addr);
-                   if (size > maxlen) {
-                       SAFE_FREE(ip_list);
-                       request_error(state);
-                       return;
-                   }
-                   if (i != 0) {
-                       /* Clear out the newline character */
-                       /* But only if there is something in there, 
-                          otherwise we clobber something in the stack */
-                       if (strlen(response))
-                               response[strlen(response)-1] = ' '; 
-                   }
-                   fstrcat(response,addr);
-                   fstrcat(response,"\t");
+                       print_sockaddr(addr, sizeof(addr), &ip_list[i-1]);
+                       size = strlen(addr);
+                       if (size > maxlen) {
+                               TALLOC_FREE(ip_list);
+                               request_error(state);
+                               return;
+                       }
+                       if (i != 0) {
+                               /* Clear out the newline character */
+                               /* But only if there is something in there,
+                               otherwise we clobber something in the stack */
+                               if (strlen(response)) {
+                                       response[strlen(response)-1] = ' ';
+                               }
+                       }
+                       strlcat(response,addr,sizeof(response));
+                       strlcat(response,"\t",sizeof(response));
                }
-               size = strlen(state->request.data.winsreq) + strlen(response);
+               size = strlen(state->request->data.winsreq) + strlen(response);
                if (size > maxlen) {
-                   SAFE_FREE(ip_list);
+                   TALLOC_FREE(ip_list);
                    request_error(state);
                    return;
-               }   
-               fstrcat(response,state->request.data.winsreq);
-               fstrcat(response,"\n");
-               SAFE_FREE(ip_list);
+               }
+               strlcat(response,state->request->data.winsreq,sizeof(response));
+               strlcat(response,"\n",sizeof(response));
+               TALLOC_FREE(ip_list);
        } else {
                request_error(state);
                return;
        }
 
-       fstrcpy(state->response.data.winsresp,response);
+       strlcpy(state->response->data.winsresp,
+               response,
+               sizeof(state->response->data.winsresp));
 
        request_ok(state);
 }