1 /*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms of the
5 * Common Development and Distribution License (the "License").
6 * You may not use this file except in compliance with the License.
7 *
8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9 * or http://www.opensolaris.org/os/licensing.
10 * See the License for the specific language governing permissions
11 * and limitations under the License.
12 *
13 * When distributing Covered Code, include this CDDL HEADER in each
14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15 * If applicable, add the following below this CDDL HEADER, with the
16 * fields enclosed by brackets "[]" replaced with your own identifying
17 * information: Portions Copyright [yyyy] [name of copyright owner]
18 *
19 * CDDL HEADER END
20 */
21 /*
22 * Copyright 2009 Sun Microsystems, Inc. All rights reserved.
23 * Use is subject to license terms.
24 */
25
26 #include <synch.h>
27 #include <stdio.h>
28 #include <syslog.h>
29 #include <stdlib.h>
30 #include <arpa/inet.h>
31 #include <netdb.h>
32
33 #include <smbsrv/libsmbns.h>
34 #include <smbns_netbios.h>
35
36 #define NETBIOS_HTAB_SZ 128
37 #define NETBIOS_HKEY_SZ (NETBIOS_NAME_SZ + NETBIOS_DOMAIN_NAME_MAX)
38
39 #define NETBIOS_SAME_IP(addr1, addr2) \
40 ((addr1)->sin.sin_addr.s_addr == (addr2)->sin.sin_addr.s_addr)
41
42 typedef char nb_key_t[NETBIOS_HKEY_SZ];
43 static HT_HANDLE *smb_netbios_cache = 0;
44 static rwlock_t nb_cache_lock;
45
46 static void smb_strname(struct name_entry *name, char *buf, int bufsize);
47 static void hash_callback(HT_ITEM *item);
48 static int smb_netbios_match(const char *key1, const char *key2, size_t n);
49 static void smb_netbios_cache_key(char *key, unsigned char *name,
50 unsigned char *scope);
51
52 int
smb_netbios_cache_init(void)53 smb_netbios_cache_init(void)
54 {
55 (void) rw_wrlock(&nb_cache_lock);
56 if (smb_netbios_cache == NULL) {
57 smb_netbios_cache = ht_create_table(NETBIOS_HTAB_SZ,
58 NETBIOS_HKEY_SZ, HTHF_FIXED_KEY);
59 if (smb_netbios_cache == NULL) {
60 syslog(LOG_ERR, "nbns: cannot create name cache");
61 (void) rw_unlock(&nb_cache_lock);
62 return (-1);
63 }
64 (void) ht_register_callback(smb_netbios_cache, hash_callback);
65 ht_set_cmpfn(smb_netbios_cache, smb_netbios_match);
66 }
67 (void) rw_unlock(&nb_cache_lock);
68
69 return (0);
70 }
71
72 void
smb_netbios_cache_fini(void)73 smb_netbios_cache_fini(void)
74 {
75 (void) rw_wrlock(&nb_cache_lock);
76 ht_destroy_table(smb_netbios_cache);
77 smb_netbios_cache = NULL;
78 (void) rw_unlock(&nb_cache_lock);
79 }
80
81 void
smb_netbios_cache_clean(void)82 smb_netbios_cache_clean(void)
83 {
84 (void) rw_wrlock(&nb_cache_lock);
85 (void) ht_clean_table(smb_netbios_cache);
86 (void) rw_unlock(&nb_cache_lock);
87 }
88
89 int
smb_netbios_cache_getfirst(nbcache_iter_t * iter)90 smb_netbios_cache_getfirst(nbcache_iter_t *iter)
91 {
92 HT_ITEM *item;
93 struct name_entry *entry;
94
95 (void) rw_rdlock(&nb_cache_lock);
96 item = ht_findfirst(smb_netbios_cache, &iter->nbc_hti);
97 if (item == NULL || item->hi_data == NULL) {
98 (void) rw_unlock(&nb_cache_lock);
99 return (-1);
100 }
101
102 entry = (struct name_entry *)item->hi_data;
103 (void) mutex_lock(&entry->mtx);
104 iter->nbc_entry = smb_netbios_name_dup(entry, 1);
105 (void) mutex_unlock(&entry->mtx);
106
107 (void) rw_unlock(&nb_cache_lock);
108
109 return ((iter->nbc_entry) ? 0 : -1);
110 }
111
112 int
smb_netbios_cache_getnext(nbcache_iter_t * iter)113 smb_netbios_cache_getnext(nbcache_iter_t *iter)
114 {
115 HT_ITEM *item;
116 struct name_entry *entry;
117
118 (void) rw_rdlock(&nb_cache_lock);
119 item = ht_findnext(&iter->nbc_hti);
120 if (item == NULL || item->hi_data == NULL) {
121 (void) rw_unlock(&nb_cache_lock);
122 return (-1);
123 }
124
125 entry = (struct name_entry *)item->hi_data;
126 (void) mutex_lock(&entry->mtx);
127 iter->nbc_entry = smb_netbios_name_dup(entry, 1);
128 (void) mutex_unlock(&entry->mtx);
129
130 (void) rw_unlock(&nb_cache_lock);
131
132 return ((iter->nbc_entry) ? 0 : -1);
133 }
134
135 /*
136 * smb_netbios_cache_lookup
137 *
138 * Searches the name cache for the given entry, if found
139 * the entry will be locked before returning to caller
140 * so caller MUST unlock the entry after it's done with it.
141 */
142 struct name_entry *
smb_netbios_cache_lookup(struct name_entry * name)143 smb_netbios_cache_lookup(struct name_entry *name)
144 {
145 HT_ITEM *item;
146 nb_key_t key;
147 struct name_entry *entry = NULL;
148 unsigned char hostname[MAXHOSTNAMELEN];
149
150 if (NETBIOS_NAME_IS_STAR(name->name)) {
151 /* Return our address */
152 if (smb_getnetbiosname((char *)hostname, sizeof (hostname))
153 != 0)
154 return (NULL);
155
156 smb_encode_netbios_name(hostname, 0x00, NULL, name);
157 }
158
159 (void) rw_rdlock(&nb_cache_lock);
160
161 smb_netbios_cache_key(key, name->name, name->scope);
162 item = ht_find_item(smb_netbios_cache, key);
163 if (item) {
164 entry = (struct name_entry *)item->hi_data;
165 (void) mutex_lock(&entry->mtx);
166 if ((entry->attributes & NAME_ATTR_CONFLICT) != 0) {
167 (void) mutex_unlock(&entry->mtx);
168 entry = NULL;
169 }
170 }
171
172 (void) rw_unlock(&nb_cache_lock);
173 return (entry);
174 }
175
176 void
smb_netbios_cache_unlock_entry(struct name_entry * name)177 smb_netbios_cache_unlock_entry(struct name_entry *name)
178 {
179 if (name)
180 (void) mutex_unlock(&name->mtx);
181 }
182
183 /*
184 * smb_netbios_cache_lookup_addr
185 *
186 * lookup the given 'name' in the cache and then checks
187 * if the address also matches with the found entry.
188 * 'name' is supposed to contain only one address.
189 *
190 * The found entry will be locked before returning to caller
191 * so caller MUST unlock the entry after it's done with it.
192 */
193 struct name_entry *
smb_netbios_cache_lookup_addr(struct name_entry * name)194 smb_netbios_cache_lookup_addr(struct name_entry *name)
195 {
196 struct name_entry *entry = 0;
197 addr_entry_t *addr;
198 addr_entry_t *name_addr;
199 HT_ITEM *item;
200 nb_key_t key;
201
202 (void) rw_rdlock(&nb_cache_lock);
203 smb_netbios_cache_key(key, name->name, name->scope);
204 item = ht_find_item(smb_netbios_cache, key);
205
206 if (item && item->hi_data) {
207 name_addr = &name->addr_list;
208 entry = (struct name_entry *)item->hi_data;
209 (void) mutex_lock(&entry->mtx);
210 addr = &entry->addr_list;
211 do {
212 if (NETBIOS_SAME_IP(addr, name_addr)) {
213 /* note that entry lock isn't released here */
214 (void) rw_unlock(&nb_cache_lock);
215 return (entry);
216 }
217 addr = addr->forw;
218 } while (addr != &entry->addr_list);
219 (void) mutex_unlock(&entry->mtx);
220 }
221
222 (void) rw_unlock(&nb_cache_lock);
223 return (0);
224 }
225
226 int
smb_netbios_cache_insert(struct name_entry * name)227 smb_netbios_cache_insert(struct name_entry *name)
228 {
229 struct name_entry *entry;
230 addr_entry_t *addr;
231 addr_entry_t *name_addr;
232 HT_ITEM *item;
233 nb_key_t key;
234 int rc;
235
236 /* No point in adding a name with IP address 255.255.255.255 */
237 if (name->addr_list.sin.sin_addr.s_addr == 0xffffffff)
238 return (0);
239
240 (void) rw_wrlock(&nb_cache_lock);
241 smb_netbios_cache_key(key, name->name, name->scope);
242 item = ht_find_item(smb_netbios_cache, key);
243
244 if (item && item->hi_data) {
245 /* Name already exists */
246 entry = (struct name_entry *)item->hi_data;
247 (void) mutex_lock(&entry->mtx);
248
249 name_addr = &name->addr_list;
250 addr = &entry->addr_list;
251 if (NETBIOS_SAME_IP(addr, name_addr) &&
252 (addr->sin.sin_port == name_addr->sin.sin_port)) {
253 entry->attributes |=
254 name_addr->attributes & NAME_ATTR_LOCAL;
255 (void) mutex_unlock(&entry->mtx);
256 (void) rw_unlock(&nb_cache_lock);
257 return (0);
258 }
259
260 /* Was not primary: looks for others */
261 for (addr = entry->addr_list.forw;
262 addr != &entry->addr_list; addr = addr->forw) {
263 if (NETBIOS_SAME_IP(addr, name_addr) &&
264 (addr->sin.sin_port == name_addr->sin.sin_port)) {
265 (void) mutex_unlock(&entry->mtx);
266 (void) rw_unlock(&nb_cache_lock);
267 return (0);
268 }
269 }
270
271 if ((addr = malloc(sizeof (addr_entry_t))) != NULL) {
272 *addr = name->addr_list;
273 entry->attributes |= addr->attributes;
274 QUEUE_INSERT_TAIL(&entry->addr_list, addr);
275 rc = 0;
276 } else {
277 rc = -1;
278 }
279
280 (void) mutex_unlock(&entry->mtx);
281 (void) rw_unlock(&nb_cache_lock);
282 return (rc);
283 }
284
285 if ((entry = malloc(sizeof (struct name_entry))) == NULL) {
286 (void) rw_unlock(&nb_cache_lock);
287 return (-1);
288 }
289
290 *entry = *name;
291 entry->addr_list.forw = entry->addr_list.back = &entry->addr_list;
292 entry->attributes |= entry->addr_list.attributes;
293 (void) mutex_init(&entry->mtx, 0, 0);
294 if (ht_replace_item(smb_netbios_cache, key, entry) == 0) {
295 free(entry);
296 (void) rw_unlock(&nb_cache_lock);
297 return (-1);
298 }
299
300 (void) rw_unlock(&nb_cache_lock);
301 return (0);
302 }
303
304
305 void
smb_netbios_cache_delete(struct name_entry * name)306 smb_netbios_cache_delete(struct name_entry *name)
307 {
308 nb_key_t key;
309 HT_ITEM *item;
310 struct name_entry *entry;
311
312 (void) rw_wrlock(&nb_cache_lock);
313 smb_netbios_cache_key(key, name->name, name->scope);
314 item = ht_find_item(smb_netbios_cache, key);
315 if (item && item->hi_data) {
316 entry = (struct name_entry *)item->hi_data;
317 (void) mutex_lock(&entry->mtx);
318 ht_mark_delete(smb_netbios_cache, item);
319 (void) mutex_unlock(&entry->mtx);
320 }
321 (void) rw_unlock(&nb_cache_lock);
322 }
323
324 /*
325 * smb_netbios_cache_insert_list
326 *
327 * Insert a name with multiple addresses
328 */
329 int
smb_netbios_cache_insert_list(struct name_entry * name)330 smb_netbios_cache_insert_list(struct name_entry *name)
331 {
332 struct name_entry entry;
333 addr_entry_t *addr;
334
335 addr = &name->addr_list;
336 do {
337 smb_init_name_struct(NETBIOS_EMPTY_NAME, 0, name->scope,
338 addr->sin.sin_addr.s_addr,
339 addr->sin.sin_port,
340 name->attributes,
341 addr->attributes,
342 &entry);
343 (void) memcpy(entry.name, name->name, NETBIOS_NAME_SZ);
344 entry.addr_list.refresh_ttl = entry.addr_list.ttl =
345 addr->refresh_ttl;
346 (void) smb_netbios_cache_insert(&entry);
347 addr = addr->forw;
348 } while (addr != &name->addr_list);
349
350 return (0);
351 }
352
353 void
smb_netbios_cache_update_entry(struct name_entry * entry,struct name_entry * name)354 smb_netbios_cache_update_entry(struct name_entry *entry,
355 struct name_entry *name)
356 {
357 addr_entry_t *addr;
358 addr_entry_t *name_addr;
359
360 addr = &entry->addr_list;
361 name_addr = &name->addr_list;
362
363 if (IS_UNIQUE(entry->attributes)) {
364 do {
365 addr->ttl = name_addr->ttl;
366 addr = addr->forw;
367 } while (addr != &entry->addr_list);
368
369 } else {
370 do {
371 if (NETBIOS_SAME_IP(addr, name_addr) &&
372 (addr->sin.sin_port == name_addr->sin.sin_port)) {
373 addr->ttl = name_addr->ttl;
374 return;
375 }
376 addr = addr->forw;
377 } while (addr != &entry->addr_list);
378 }
379 }
380
381 /*
382 * smb_netbios_cache_status
383 *
384 * Scan the name cache and gather status for
385 * Node Status response for names in the given scope
386 */
387 unsigned char *
smb_netbios_cache_status(unsigned char * buf,int bufsize,unsigned char * scope)388 smb_netbios_cache_status(unsigned char *buf, int bufsize, unsigned char *scope)
389 {
390 HT_ITERATOR hti;
391 HT_ITEM *item;
392 struct name_entry *name;
393 unsigned char *numnames;
394 unsigned char *scan;
395 unsigned char *scan_end;
396
397 scan = buf;
398 scan_end = scan + bufsize;
399
400 numnames = scan++;
401 *numnames = 0;
402
403 (void) rw_rdlock(&nb_cache_lock);
404 item = ht_findfirst(smb_netbios_cache, &hti);
405 do {
406 if (item == 0)
407 break;
408
409 if (item->hi_data == 0)
410 continue;
411
412 if ((scan + NETBIOS_NAME_SZ + 2) >= scan_end)
413 /* no room for adding next entry */
414 break;
415
416 name = (struct name_entry *)item->hi_data;
417 (void) mutex_lock(&name->mtx);
418
419 if (IS_LOCAL(name->attributes) &&
420 (strcasecmp((char *)scope, (char *)name->scope) == 0)) {
421 bcopy(name->name, scan, NETBIOS_NAME_SZ);
422 scan += NETBIOS_NAME_SZ;
423 *scan++ = (PUBLIC_BITS(name->attributes) >> 8) & 0xff;
424 *scan++ = PUBLIC_BITS(name->attributes) & 0xff;
425 (*numnames)++;
426 }
427
428 (void) mutex_unlock(&name->mtx);
429 } while ((item = ht_findnext(&hti)) != 0);
430 (void) rw_unlock(&nb_cache_lock);
431
432 return (scan);
433 }
434
435 void
smb_netbios_cache_reset_ttl()436 smb_netbios_cache_reset_ttl()
437 {
438 addr_entry_t *addr;
439 struct name_entry *name;
440 HT_ITERATOR hti;
441 HT_ITEM *item;
442
443 (void) rw_rdlock(&nb_cache_lock);
444 item = ht_findfirst(smb_netbios_cache, &hti);
445 do {
446 if (item == 0)
447 break;
448
449 if (item->hi_data == 0)
450 continue;
451
452 name = (struct name_entry *)item->hi_data;
453 (void) mutex_lock(&name->mtx);
454
455 addr = &name->addr_list;
456 do {
457 if (addr->ttl < 1) {
458 if (addr->refresh_ttl)
459 addr->ttl = addr->refresh_ttl;
460 else
461 addr->refresh_ttl = addr->ttl =
462 TO_SECONDS(DEFAULT_TTL);
463 }
464 addr = addr->forw;
465 } while (addr != &name->addr_list);
466
467 (void) mutex_unlock(&name->mtx);
468 } while ((item = ht_findnext(&hti)) != 0);
469 (void) rw_unlock(&nb_cache_lock);
470 }
471
472 /*
473 * Returns TRUE when given name is added to the refresh queue
474 * FALSE if not.
475 */
476 static boolean_t
smb_netbios_cache_insrefq(name_queue_t * refq,HT_ITEM * item)477 smb_netbios_cache_insrefq(name_queue_t *refq, HT_ITEM *item)
478 {
479 struct name_entry *name;
480 struct name_entry *refent;
481
482 name = (struct name_entry *)item->hi_data;
483
484 if (IS_LOCAL(name->attributes)) {
485 if (IS_UNIQUE(name->attributes)) {
486 refent = smb_netbios_name_dup(name, 1);
487 if (refent)
488 QUEUE_INSERT_TAIL(&refq->head, refent)
489
490 /* next name */
491 return (B_TRUE);
492 }
493 } else {
494 ht_mark_delete(smb_netbios_cache, item);
495 refent = smb_netbios_name_dup(name, 0);
496 if (refent)
497 QUEUE_INSERT_TAIL(&refq->head, refent)
498
499 /* next name */
500 return (B_TRUE);
501 }
502
503 return (B_FALSE);
504 }
505
506 /*
507 * smb_netbios_cache_refresh
508 *
509 * Scans the name cache and add all local unique names
510 * and non-local names the passed refresh queue. Non-
511 * local names will also be marked as deleted.
512 *
513 * NOTE that the caller MUST protect the queue using
514 * its mutex
515 */
516 void
smb_netbios_cache_refresh(name_queue_t * refq)517 smb_netbios_cache_refresh(name_queue_t *refq)
518 {
519 struct name_entry *name;
520 addr_entry_t *addr;
521 HT_ITERATOR hti;
522 HT_ITEM *item;
523
524 bzero(&refq->head, sizeof (refq->head));
525 refq->head.forw = refq->head.back = &refq->head;
526
527 (void) rw_rdlock(&nb_cache_lock);
528 item = ht_findfirst(smb_netbios_cache, &hti);
529 do { /* name loop */
530 if (item == 0)
531 break;
532
533 if (item->hi_data == 0)
534 continue;
535
536 name = (struct name_entry *)item->hi_data;
537 (void) mutex_lock(&name->mtx);
538
539 addr = &name->addr_list;
540 do { /* address loop */
541 if (addr->ttl > 0) {
542 addr->ttl--;
543 if (addr->ttl == 0) {
544 if (smb_netbios_cache_insrefq(refq,
545 item))
546 break;
547 }
548 }
549 addr = addr->forw;
550 } while (addr != &name->addr_list);
551
552 (void) mutex_unlock(&name->mtx);
553 } while ((item = ht_findnext(&hti)) != 0);
554 (void) rw_unlock(&nb_cache_lock);
555 }
556
557 /*
558 * smb_netbios_cache_delete_locals
559 *
560 * Scans the name cache and add all local names to
561 * the passed delete queue.
562 *
563 * NOTE that the caller MUST protect the queue using
564 * its mutex
565 */
566 void
smb_netbios_cache_delete_locals(name_queue_t * delq)567 smb_netbios_cache_delete_locals(name_queue_t *delq)
568 {
569 struct name_entry *entry;
570 struct name_entry *delent;
571 HT_ITERATOR hti;
572 HT_ITEM *item;
573
574 bzero(&delq->head, sizeof (delq->head));
575 delq->head.forw = delq->head.back = &delq->head;
576
577 (void) rw_wrlock(&nb_cache_lock);
578 item = ht_findfirst(smb_netbios_cache, &hti);
579 do {
580 if (item == 0)
581 break;
582
583 if (item->hi_data == 0)
584 continue;
585
586 entry = (struct name_entry *)item->hi_data;
587 (void) mutex_lock(&entry->mtx);
588
589 if (IS_LOCAL(entry->attributes)) {
590 ht_mark_delete(smb_netbios_cache, item);
591 delent = smb_netbios_name_dup(entry, 1);
592 if (delent)
593 QUEUE_INSERT_TAIL(&delq->head, delent)
594 }
595
596 (void) mutex_unlock(&entry->mtx);
597 } while ((item = ht_findnext(&hti)) != 0);
598 (void) rw_unlock(&nb_cache_lock);
599 }
600
601 void
smb_netbios_name_freeaddrs(struct name_entry * entry)602 smb_netbios_name_freeaddrs(struct name_entry *entry)
603 {
604 addr_entry_t *addr;
605
606 if (entry == 0)
607 return;
608
609 while ((addr = entry->addr_list.forw) != &entry->addr_list) {
610 QUEUE_CLIP(addr);
611 free(addr);
612 }
613 }
614
615 /*
616 * smb_netbios_cache_count
617 *
618 * Returns the number of names in the cache
619 */
620 int
smb_netbios_cache_count()621 smb_netbios_cache_count()
622 {
623 int cnt;
624
625 (void) rw_rdlock(&nb_cache_lock);
626 cnt = ht_get_total_items(smb_netbios_cache);
627 (void) rw_unlock(&nb_cache_lock);
628
629 return (cnt);
630 }
631
632 void
smb_netbios_cache_dump(FILE * fp)633 smb_netbios_cache_dump(FILE *fp)
634 {
635 struct name_entry *name;
636 HT_ITERATOR hti;
637 HT_ITEM *item;
638
639 (void) rw_rdlock(&nb_cache_lock);
640
641 if (ht_get_total_items(smb_netbios_cache) != 0) {
642 (void) fprintf(fp, "\n%-22s %-16s %-16s %s\n",
643 "Name", "Type", "Address", "TTL");
644 (void) fprintf(fp, "%s%s\n",
645 "-------------------------------",
646 "------------------------------");
647 }
648
649 item = ht_findfirst(smb_netbios_cache, &hti);
650 while (item) {
651 if (item->hi_data) {
652 name = (struct name_entry *)item->hi_data;
653 (void) mutex_lock(&name->mtx);
654 smb_netbios_name_dump(fp, name);
655 (void) mutex_unlock(&name->mtx);
656 }
657 item = ht_findnext(&hti);
658 }
659 (void) rw_unlock(&nb_cache_lock);
660 }
661
662 void
smb_netbios_name_dump(FILE * fp,struct name_entry * entry)663 smb_netbios_name_dump(FILE *fp, struct name_entry *entry)
664 {
665 char buf[MAXHOSTNAMELEN];
666 addr_entry_t *addr;
667 char *type;
668 int count = 0;
669
670 smb_strname(entry, buf, sizeof (buf));
671 type = (IS_UNIQUE(entry->attributes)) ? "UNIQUE" : "GROUP";
672
673 (void) fprintf(fp, "%s %-6s (0x%04x) ", buf, type, entry->attributes);
674
675 addr = &entry->addr_list;
676 do {
677 if (count == 0)
678 (void) fprintf(fp, "%-16s %d\n",
679 inet_ntoa(addr->sin.sin_addr), addr->ttl);
680 else
681 (void) fprintf(fp, "%-28s (0x%04x) %-16s %d\n",
682 " ", addr->attributes,
683 inet_ntoa(addr->sin.sin_addr), addr->ttl);
684 ++count;
685 addr = addr->forw;
686 } while (addr != &entry->addr_list);
687 }
688
689 void
smb_netbios_name_logf(struct name_entry * entry)690 smb_netbios_name_logf(struct name_entry *entry)
691 {
692 char namebuf[MAXHOSTNAMELEN];
693 addr_entry_t *addr;
694
695 smb_strname(entry, namebuf, sizeof (namebuf));
696 syslog(LOG_DEBUG, "%s flags=0x%x\n", namebuf, entry->attributes);
697 addr = &entry->addr_list;
698 do {
699 syslog(LOG_DEBUG, " %s ttl=%d flags=0x%x port=%d",
700 inet_ntoa(addr->sin.sin_addr),
701 addr->ttl, addr->attributes,
702 addr->sin.sin_port);
703 addr = addr->forw;
704 } while (addr && (addr != &entry->addr_list));
705 }
706
707 /*
708 * smb_netbios_name_dup
709 *
710 * Duplicate the given name entry. If 'alladdr' is 0 only
711 * copy the primary address otherwise duplicate all the
712 * addresses. NOTE that the duplicate structure is not
713 * like a regular cache entry i.e. it's a contiguous block
714 * of memory and each addr structure doesn't have it's own
715 * allocated memory. So, the returned structure can be freed
716 * by one free call.
717 */
718 struct name_entry *
smb_netbios_name_dup(struct name_entry * entry,int alladdr)719 smb_netbios_name_dup(struct name_entry *entry, int alladdr)
720 {
721 addr_entry_t *addr;
722 addr_entry_t *dup_addr;
723 struct name_entry *dup;
724 int addr_cnt = 0;
725 int size = 0;
726
727 if (alladdr) {
728 addr = entry->addr_list.forw;
729 while (addr && (addr != &entry->addr_list)) {
730 addr_cnt++;
731 addr = addr->forw;
732 }
733 }
734
735 size = sizeof (struct name_entry) +
736 (addr_cnt * sizeof (addr_entry_t));
737 dup = (struct name_entry *)malloc(size);
738 if (dup == 0)
739 return (0);
740
741 bzero(dup, size);
742
743 dup->forw = dup->back = dup;
744 dup->attributes = entry->attributes;
745 (void) memcpy(dup->name, entry->name, NETBIOS_NAME_SZ);
746 (void) strlcpy((char *)dup->scope, (char *)entry->scope,
747 NETBIOS_DOMAIN_NAME_MAX);
748 dup->addr_list = entry->addr_list;
749 dup->addr_list.forw = dup->addr_list.back = &dup->addr_list;
750
751 if (alladdr == 0)
752 return (dup);
753
754 /* LINTED - E_BAD_PTR_CAST_ALIGN */
755 dup_addr = (addr_entry_t *)((unsigned char *)dup +
756 sizeof (struct name_entry));
757
758 addr = entry->addr_list.forw;
759 while (addr && (addr != &entry->addr_list)) {
760 *dup_addr = *addr;
761 QUEUE_INSERT_TAIL(&dup->addr_list, dup_addr);
762 addr = addr->forw;
763 dup_addr++;
764 }
765
766 return (dup);
767 }
768
769 static void
smb_strname(struct name_entry * entry,char * buf,int bufsize)770 smb_strname(struct name_entry *entry, char *buf, int bufsize)
771 {
772 char tmp[MAXHOSTNAMELEN];
773 char *p;
774
775 (void) snprintf(tmp, MAXHOSTNAMELEN, "%15.15s", entry->name);
776 if ((p = strchr(tmp, ' ')) != NULL)
777 *p = '\0';
778
779 if (entry->scope[0] != '\0') {
780 (void) strlcat(tmp, ".", MAXHOSTNAMELEN);
781 (void) strlcat(tmp, (char *)entry->scope, MAXHOSTNAMELEN);
782 }
783
784 (void) snprintf(buf, bufsize, "%-16s <%02X>", tmp, entry->name[15]);
785 }
786
787 static void
hash_callback(HT_ITEM * item)788 hash_callback(HT_ITEM *item)
789 {
790 struct name_entry *entry;
791
792 if (item && item->hi_data) {
793 entry = (struct name_entry *)item->hi_data;
794 smb_netbios_name_freeaddrs(entry);
795 free(entry);
796 }
797 }
798
799
800 /*ARGSUSED*/
801 static int
smb_netbios_match(const char * key1,const char * key2,size_t n)802 smb_netbios_match(const char *key1, const char *key2, size_t n)
803 {
804 int res;
805
806 res = bcmp(key1, key2, NETBIOS_NAME_SZ);
807 if (res == 0) {
808 /* Names are the same, compare scopes */
809 res = strcmp(key1 + NETBIOS_NAME_SZ, key2 + NETBIOS_NAME_SZ);
810 }
811
812 return (res);
813 }
814
815 static void
smb_netbios_cache_key(char * key,unsigned char * name,unsigned char * scope)816 smb_netbios_cache_key(char *key, unsigned char *name, unsigned char *scope)
817 {
818 bzero(key, NETBIOS_HKEY_SZ);
819 (void) memcpy(key, name, NETBIOS_NAME_SZ);
820 (void) memcpy(key + NETBIOS_NAME_SZ, scope,
821 strlen((const char *)scope));
822 }
823