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