blob: 518e5b434df90dd08cb977800c2a289af40f709f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
#include <u.h>
#include <libc.h>
#include <ip.h>
#include <thread.h>
#include "netbios.h"
int
nbnsfindname(uchar *serveripaddr, NbName name, uchar *ipaddr, ulong *ttlp)
{
NbnsMessage *nq;
Alt aa[3];
int tries = NbnsRetryBroadcast;
NbnsAlarm *a;
int rv;
NbnsMessage *response;
nq = nbnsmessagenamequeryrequestnew(0, serveripaddr == nil, name);
if (nq == nil)
return -1;
a = nbnsalarmnew();
if (a == nil) {
free(nq);
return -1;
}
aa[0].c = a->c;
aa[0].v = nil;
aa[0].op = CHANRCV;
aa[1].op = CHANRCV;
aa[2].op = CHANEND;
while (tries > 0) {
NbnsTransaction *t;
nq->id = nbnsnextid();
t = nbnstransactionnew(nq, serveripaddr);
aa[1].c = t->c;
aa[1].v = &response;
nbnsalarmset(a, NbnsTimeoutBroadcast);
for (;;) {
int i;
i = alt(aa);
if (i == 0) {
tries--;
break;
}
else if (i == 1) {
if (response->opcode == NbnsOpQuery) {
nbnstransactionfree(&t);
goto done;
}
nbnsmessagefree(&response);
}
}
nbnstransactionfree(&t);
}
done:
if (tries == 0)
rv = -1;
else {
if (response->rcode != 0)
rv = response->rcode;
else if (response->an == nil)
rv = -1;
else {
rv = 0;
v4tov6(ipaddr, response->an->rdata + 2);
if (ttlp)
*ttlp = response->an->ttl;
}
nbnsmessagefree(&response);
}
nbnsalarmfree(&a);
nbnsmessagefree(&nq);
return rv;
}
|
