Subversion Repositories planix.SVN

#include <u.h>
#include <libc.h>
#include <ip.h>
#include <thread.h>
#include "netbios.h"

static char *hmsg = "headers";
int chatty = 1;

void
warning(char *fmt, ...)
{
        char err[128];
        va_list arg;

        va_start(arg, fmt);
        vseprint(err, err+sizeof(err), fmt, arg);
        va_end(arg);
        syslog(1, "netbios-ns", err);
        if (chatty)
                print("%s\n", err);
}

static int
udpannounce(void)
{
        int data, ctl;
        char dir[64];
        char datafile[64+6];

        /* get a udp port */
        ctl = announce("udp!*!netbios-ns", dir);
        if(ctl < 0){
                warning("can't announce on netbios-ns udp port");
                return -1;
        }
        snprint(datafile, sizeof(datafile), "%s/data", dir);

        /* turn on header style interface */
        if(write(ctl, hmsg, strlen(hmsg)) , 0)
                abort(); /* hmsg */;
        data = open(datafile, ORDWR);
        if(data < 0){
                close(ctl);
                warning("can't announce on dns udp port");
                return -1;
        }

        close(ctl);
        return data;
}

#define BROADCAST 1


void
listen137(void *)
{       
        for (;;) {
                uchar msg[Udphdrsize + 576];
                int len = read(fd137, msg, sizeof(msg));
                if (len < 0)
                        break;
                if (len >= Udphdrsize) {
                        NbnsMessage *s;
                        Udphdr *uh;
                        uchar *p;

                        uh = (Udphdr*)msg;
                        p = msg + Udphdrsize;
                        len -= Udphdrsize;
                        s = nbnsconvM2S(p, len);
                        if (s) {
                                print("%I:%d -> %I:%d\n", uh->raddr, nhgets(uh->rport), uh->laddr, nhgets(uh->lport));
                                nbnsdumpmessage(s);
                                if (s->response) {
                                        NbnsTransaction *t;
                                        qlock(&transactionlist);
                                        for (t = transactionlist.head; t; t = t->next)
                                                if (t->id == s->id)
                                                        break;
                                        if (t) {
                                                sendp(t->c, s);
                                        }
                                        else
                                                nbnsmessagefree(&s);
                                        qunlock(&transactionlist);
                                }
                                else
                                        nbnsmessagefree(&s);
                        }
                }
        }
}

void
usage(void)
{
        fprint(2, "usage: %s [-u ipaddr] name\n", argv0);
        exits("usage");
}

void
threadmain(int argc, char **argv)
{
        int broadcast = 1, i, listen137thread, rv;
        char *ip;
        uchar ipaddr[IPaddrlen], serveripaddr[IPaddrlen];
        NbName nbname;

        ARGBEGIN {
        case 'u':
                broadcast = 0;
                ip = EARGF(usage());
                if (parseip(serveripaddr, ip) == -1)
                        sysfatal("bad ip address %s", ip);
                break;
        default:
                usage();
        } ARGEND;

        if (argc == 0)
                usage();

        nbmknamefromstring(nbname, argv[0]);

        ipifc = readipifc("/net", nil, 0);
        if (ipifc == nil || ipifc->lifc == nil)
                sysfatal("no network interface");
        fmtinstall('I', eipfmt);
        ipmove(bcastaddr, ipifc->lifc->ip);
        for (i = 0; i < IPaddrlen; i++)
                bcastaddr[i] |= ~ipifc->lifc->mask[i];
        print("broadcasting to %I\n", bcastaddr);
//      setnetmtpt("/net");
        fd137 = udpannounce();
        listen137thread = proccreate(listen137, nil, 16384);
        rv = nbnsaddname(broadcast ? nil : serveripaddr, nbname, 3000, ipifc->lifc->ip);
        if (rv != 0)
                print("error code %d\n", rv);
        else
                print("%I\n", ipaddr);
        nbnsalarmend();
        threadint(listen137thread);
}