r148: Ensure we do not dereference a null pointer when we return the user
[Samba/gebeck_regimport.git] / source3 / nsswitch / winbindd_wins.c
bloba1eef159c0a8d1a13df9d270fb2a494b9302d560
1 /*
2 Unix SMB/CIFS implementation.
4 Winbind daemon - WINS related functions
6 Copyright (C) Andrew Tridgell 1999
7 Copyright (C) Herb Lewis 2002
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "includes.h"
25 #include "winbindd.h"
27 #undef DBGC_CLASS
28 #define DBGC_CLASS DBGC_WINBIND
30 /* Use our own create socket code so we don't recurse.... */
32 static int wins_lookup_open_socket_in(void)
34 struct sockaddr_in sock;
35 int val=1;
36 int res;
38 memset((char *)&sock,'\0',sizeof(sock));
40 #ifdef HAVE_SOCK_SIN_LEN
41 sock.sin_len = sizeof(sock);
42 #endif
43 sock.sin_port = 0;
44 sock.sin_family = AF_INET;
45 sock.sin_addr.s_addr = interpret_addr("0.0.0.0");
46 res = socket(AF_INET, SOCK_DGRAM, 0);
47 if (res == -1)
48 return -1;
50 setsockopt(res,SOL_SOCKET,SO_REUSEADDR,(char *)&val,sizeof(val));
51 #ifdef SO_REUSEPORT
52 setsockopt(res,SOL_SOCKET,SO_REUSEPORT,(char *)&val,sizeof(val));
53 #endif /* SO_REUSEPORT */
55 /* now we've got a socket - we need to bind it */
57 if (bind(res, (struct sockaddr * ) &sock,sizeof(sock)) < 0) {
58 close(res);
59 return(-1);
62 set_socket_options(res,"SO_BROADCAST");
64 return res;
68 static struct node_status *lookup_byaddr_backend(char *addr, int *count)
70 int fd;
71 struct in_addr ip;
72 struct nmb_name nname;
73 struct node_status *status;
75 fd = wins_lookup_open_socket_in();
76 if (fd == -1)
77 return NULL;
79 make_nmb_name(&nname, "*", 0);
80 ip = *interpret_addr2(addr);
81 status = node_status_query(fd,&nname,ip, count);
83 close(fd);
84 return status;
87 static struct in_addr *lookup_byname_backend(const char *name, int *count)
89 int fd;
90 struct ip_service *ret = NULL;
91 struct in_addr *return_ip;
92 int j, i, flags = 0;
94 *count = 0;
96 /* always try with wins first */
97 if (resolve_wins(name,0x20,&ret,count)) {
98 if ( count == 0 )
99 return NULL;
100 if ( (return_ip = (struct in_addr *)malloc((*count)*sizeof(struct in_addr))) == NULL ) {
101 free( ret );
102 return NULL;
105 /* copy the IP addresses */
106 for ( i=0; i<(*count); i++ )
107 return_ip[i] = ret[i].ip;
109 return return_ip;
112 fd = wins_lookup_open_socket_in();
113 if (fd == -1) {
114 return NULL;
117 /* uggh, we have to broadcast to each interface in turn */
118 for (j=iface_count() - 1;
119 j >= 0;
120 j--) {
121 struct in_addr *bcast = iface_n_bcast(j);
122 return_ip = name_query(fd,name,0x20,True,True,*bcast,count, &flags, NULL);
123 if (return_ip) break;
126 close(fd);
127 return return_ip;
130 /* Get hostname from IP */
132 enum winbindd_result winbindd_wins_byip(struct winbindd_cli_state *state)
134 fstring response;
135 int i, count, maxlen, size;
136 struct node_status *status;
138 /* Ensure null termination */
139 state->request.data.winsreq[sizeof(state->request.data.winsreq)-1]='\0';
141 DEBUG(3, ("[%5lu]: wins_byip %s\n", (unsigned long)state->pid,
142 state->request.data.winsreq));
144 *response = '\0';
145 maxlen = sizeof(response) - 1;
147 if ((status = lookup_byaddr_backend(state->request.data.winsreq, &count))){
148 size = strlen(state->request.data.winsreq);
149 if (size > maxlen) {
150 SAFE_FREE(status);
151 return WINBINDD_ERROR;
153 fstrcat(response,state->request.data.winsreq);
154 fstrcat(response,"\t");
155 for (i = 0; i < count; i++) {
156 /* ignore group names */
157 if (status[i].flags & 0x80) continue;
158 if (status[i].type == 0x20) {
159 size = sizeof(status[i].name) + strlen(response);
160 if (size > maxlen) {
161 SAFE_FREE(status);
162 return WINBINDD_ERROR;
164 fstrcat(response, status[i].name);
165 fstrcat(response, " ");
168 /* make last character a newline */
169 response[strlen(response)-1] = '\n';
170 SAFE_FREE(status);
172 fstrcpy(state->response.data.winsresp,response);
173 return WINBINDD_OK;
176 /* Get IP from hostname */
178 enum winbindd_result winbindd_wins_byname(struct winbindd_cli_state *state)
180 struct in_addr *ip_list;
181 int i, count, maxlen, size;
182 fstring response;
183 char * addr;
185 /* Ensure null termination */
186 state->request.data.winsreq[sizeof(state->request.data.winsreq)-1]='\0';
188 DEBUG(3, ("[%5lu]: wins_byname %s\n", (unsigned long)state->pid,
189 state->request.data.winsreq));
191 *response = '\0';
192 maxlen = sizeof(response) - 1;
194 if ((ip_list = lookup_byname_backend(state->request.data.winsreq,&count))){
195 for (i = count; i ; i--) {
196 addr = inet_ntoa(ip_list[i-1]);
197 size = strlen(addr);
198 if (size > maxlen) {
199 SAFE_FREE(ip_list);
200 return WINBINDD_ERROR;
202 if (i != 0) {
203 /* Clear out the newline character */
204 /* But only if there is something in there,
205 otherwise we clobber something in the stack */
206 if (strlen(response))
207 response[strlen(response)-1] = ' ';
209 fstrcat(response,addr);
210 fstrcat(response,"\t");
212 size = strlen(state->request.data.winsreq) + strlen(response);
213 if (size > maxlen) {
214 SAFE_FREE(ip_list);
215 return WINBINDD_ERROR;
217 fstrcat(response,state->request.data.winsreq);
218 fstrcat(response,"\n");
219 SAFE_FREE(ip_list);
220 } else
221 return WINBINDD_ERROR;
223 fstrcpy(state->response.data.winsresp,response);
225 return WINBINDD_OK;