2 * Copyright (c) 2000, Boris Popov
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by Boris Popov.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * $Id: nbns_rq.c,v 1.5 2001/02/17 03:07:24 bp Exp $
34 #include <sys/param.h>
35 #include <sys/socket.h>
47 #define NB_NEEDRESOLVER
48 #include <netsmb/netbios.h>
49 #include <netsmb/smb_lib.h>
50 #include <netsmb/nb_lib.h>
53 static int nbns_rq_create(int opcode, struct nb_ctx *ctx, struct nbns_rq **rqpp);
54 static void nbns_rq_done(struct nbns_rq *rqp);
55 static int nbns_rq_getrr(struct nbns_rq *rqp, struct nbns_rr *rrp);
56 static int nbns_rq_prepare(struct nbns_rq *rqp);
57 static int nbns_rq(struct nbns_rq *rqp);
59 static struct nb_ifdesc *nb_iflist;
62 nbns_resolvename(const char *name, struct nb_ctx *ctx, struct sockaddr **adpp)
67 struct sockaddr_in *dest;
68 int error, rdrcount, len;
70 if (strlen(name) > NB_NAMELEN)
71 return NBERROR(NBERR_NAMETOOLONG);
72 error = nbns_rq_create(NBNS_OPCODE_QUERY, ctx, &rqp);
75 bzero(&nn, sizeof(nn));
76 strcpy(nn.nn_name, name);
77 nn.nn_scope = ctx->nb_scope;
78 nn.nn_type = NBT_SERVER;
79 rqp->nr_nmflags = NBNS_NMFLAG_RD;
81 rqp->nr_qdtype = NBNS_QUESTION_TYPE_NB;
82 rqp->nr_qdclass = NBNS_QUESTION_CLASS_IN;
86 dest->sin_family = AF_INET;
87 dest->sin_len = sizeof(*dest);
88 if (dest->sin_port == 0)
89 dest->sin_port = htons(137);
90 if (dest->sin_addr.s_addr == INADDR_ANY)
91 dest->sin_addr.s_addr = htonl(INADDR_BROADCAST);
92 if (dest->sin_addr.s_addr == INADDR_BROADCAST)
93 rqp->nr_flags |= NBRQF_BROADCAST;
94 error = nbns_rq_prepare(rqp);
99 rdrcount = NBNS_MAXREDIRECTS;
101 error = nbns_rq(rqp);
104 if ((rqp->nr_rpnmflags & NBNS_NMFLAG_AA) == 0) {
105 if (rdrcount-- == 0) {
106 error = NBERROR(NBERR_TOOMANYREDIRECTS);
109 error = nbns_rq_getrr(rqp, &rr);
112 error = nbns_rq_getrr(rqp, &rr);
115 bcopy(rr.rr_data, &dest->sin_addr, 4);
116 rqp->nr_flags &= ~NBRQF_BROADCAST;
119 if (rqp->nr_rpancount == 0) {
120 error = NBERROR(NBERR_HOSTNOTFOUND);
123 error = nbns_rq_getrr(rqp, &rr);
126 len = sizeof(struct sockaddr_in);
132 dest->sin_family = AF_INET;
133 bcopy(rr.rr_data + 2, &dest->sin_addr.s_addr, 4);
134 dest->sin_port = htons(SMB_TCP_PORT);
135 *adpp = (struct sockaddr*)dest;
136 ctx->nb_lastns = rqp->nr_sender;
144 nbns_rq_create(int opcode, struct nb_ctx *ctx, struct nbns_rq **rqpp)
147 static u_int16_t trnid;
150 rqp = malloc(sizeof(*rqp));
153 bzero(rqp, sizeof(*rqp));
154 error = mb_init(&rqp->nr_rq, NBDG_MAXSIZE);
159 rqp->nr_opcode = opcode;
161 rqp->nr_trnid = trnid++;
167 nbns_rq_done(struct nbns_rq *rqp)
173 mb_done(&rqp->nr_rq);
174 mb_done(&rqp->nr_rp);
179 * Extract resource record from the packet. Assume that there is only
183 nbns_rq_getrr(struct nbns_rq *rqp, struct nbns_rr *rrp)
185 struct mbdata *mbp = &rqp->nr_rp;
189 bzero(rrp, sizeof(*rrp));
191 len = nb_encname_len(cp);
193 return NBERROR(NBERR_INVALIDRESPONSE);
195 error = mb_get_mem(mbp, NULL, len);
198 mb_get_uint16be(mbp, &rrp->rr_type);
199 mb_get_uint16be(mbp, &rrp->rr_class);
200 mb_get_uint32be(mbp, &rrp->rr_ttl);
201 mb_get_uint16be(mbp, &rrp->rr_rdlength);
202 rrp->rr_data = mbp->mb_pos;
203 error = mb_get_mem(mbp, NULL, rrp->rr_rdlength);
208 nbns_rq_prepare(struct nbns_rq *rqp)
210 struct nb_ctx *ctx = rqp->nr_nbd;
211 struct mbdata *mbp = &rqp->nr_rq;
216 error = mb_init(&rqp->nr_rp, NBDG_MAXSIZE);
219 if (rqp->nr_dest.sin_addr.s_addr == INADDR_BROADCAST) {
220 rqp->nr_nmflags |= NBNS_NMFLAG_BCAST;
221 if (nb_iflist == NULL) {
222 error = nb_enum_if(&nb_iflist, 100);
227 rqp->nr_nmflags &= ~NBNS_NMFLAG_BCAST;
228 mb_put_uint16be(mbp, rqp->nr_trnid);
229 nmflags = ((rqp->nr_opcode & 0x1F) << 3) | ((rqp->nr_nmflags & 0x70) >> 4);
230 mb_put_uint8(mbp, nmflags);
231 mb_put_uint8(mbp, (rqp->nr_nmflags & 0x0f) << 4 /* rcode */);
232 mb_put_uint16be(mbp, rqp->nr_qdcount);
233 mb_put_uint16be(mbp, rqp->nr_ancount);
234 mb_put_uint16be(mbp, rqp->nr_nscount);
235 mb_put_uint16be(mbp, rqp->nr_arcount);
236 if (rqp->nr_qdcount) {
237 if (rqp->nr_qdcount > 1)
239 len = nb_name_len(rqp->nr_qdname);
240 error = mb_fit(mbp, len, (char**)&cp);
243 nb_name_encode(rqp->nr_qdname, cp);
244 mb_put_uint16be(mbp, rqp->nr_qdtype);
245 mb_put_uint16be(mbp, rqp->nr_qdclass);
247 m_lineup(mbp->mb_top, &mbp->mb_top);
248 if (ctx->nb_timo == 0)
249 ctx->nb_timo = 1; /* by default 1 second */
254 nbns_rq_recv(struct nbns_rq *rqp)
256 struct mbdata *mbp = &rqp->nr_rp;
257 void *rpdata = mtod(mbp->mb_top, void *);
260 struct sockaddr_in sender;
269 tv.tv_sec = rqp->nr_nbd->nb_timo;
272 n = select(s + 1, &rd, &wr, &ex, &tv);
277 if (FD_ISSET(s, &rd) == 0)
279 len = sizeof(sender);
280 n = recvfrom(s, rpdata, mbp->mb_top->m_maxlen, 0,
281 (struct sockaddr*)&sender, &len);
284 mbp->mb_top->m_len = mbp->mb_count = n;
285 rqp->nr_sender = sender;
290 nbns_rq_opensocket(struct nbns_rq *rqp)
292 struct sockaddr_in locaddr;
295 s = rqp->nr_fd = socket(AF_INET, SOCK_DGRAM, 0);
298 if (rqp->nr_flags & NBRQF_BROADCAST) {
300 if (setsockopt(s, SOL_SOCKET, SO_BROADCAST, &opt, sizeof(opt)) < 0)
302 if (rqp->nr_if == NULL)
303 return NBERROR(NBERR_NOBCASTIFS);
304 bzero(&locaddr, sizeof(locaddr));
305 locaddr.sin_family = AF_INET;
306 locaddr.sin_len = sizeof(locaddr);
307 locaddr.sin_addr = rqp->nr_if->id_addr;
308 rqp->nr_dest.sin_addr.s_addr = rqp->nr_if->id_addr.s_addr | ~rqp->nr_if->id_mask.s_addr;
309 if (bind(s, (struct sockaddr*)&locaddr, sizeof(locaddr)) < 0)
316 nbns_rq_send(struct nbns_rq *rqp)
318 struct mbdata *mbp = &rqp->nr_rq;
321 if (sendto(s, mtod(mbp->mb_top, char *), mbp->mb_count, 0,
322 (struct sockaddr*)&rqp->nr_dest, sizeof(rqp->nr_dest)) < 0)
328 nbns_rq(struct nbns_rq *rqp)
330 struct mbdata *mbp = &rqp->nr_rq;
333 int error, retrycount;
335 rqp->nr_if = nb_iflist;
337 error = nbns_rq_opensocket(rqp);
340 retrycount = 3; /* XXX - configurable */
342 error = nbns_rq_send(rqp);
345 error = nbns_rq_recv(rqp);
347 if (error != ETIMEDOUT || retrycount == 0) {
348 if ((rqp->nr_nmflags & NBNS_NMFLAG_BCAST) &&
349 rqp->nr_if != NULL &&
350 rqp->nr_if->id_next != NULL) {
351 rqp->nr_if = rqp->nr_if->id_next;
361 if (mbp->mb_count < 12)
362 return NBERROR(NBERR_INVALIDRESPONSE);
363 mb_get_uint16be(mbp, &rpid);
364 if (rpid != rqp->nr_trnid)
365 return NBERROR(NBERR_INVALIDRESPONSE);
368 mb_get_uint8(mbp, &nmflags);
369 rqp->nr_rpnmflags = (nmflags & 7) << 4;
370 mb_get_uint8(mbp, &nmflags);
371 rqp->nr_rpnmflags |= (nmflags & 0xf0) >> 4;
372 rqp->nr_rprcode = nmflags & 0xf;
374 return NBERROR(rqp->nr_rprcode);
375 mb_get_uint16be(mbp, &rpid); /* QDCOUNT */
376 mb_get_uint16be(mbp, &rqp->nr_rpancount);
377 mb_get_uint16be(mbp, &rqp->nr_rpnscount);
378 mb_get_uint16be(mbp, &rqp->nr_rparcount);