harvey/sys/src/cmd/aquarela/message.c

/*
 * This file is part of the UCB release of Plan 9. It is subject to the license
 * terms in the LICENSE file found in the top-level directory of this
 * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
 * part of the UCB release of Plan 9, including this file, may be copied,
 * modified, propagated, or distributed except according to the terms contained
 * in the LICENSE file.
 */

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

NbnsMessage *
nbnsmessagenamequeryrequestnew(uint16_t id, int broadcast, NbName name)
{
	NbnsMessage *s;
	NbnsMessageQuestion *q;
	s = nbnsmessagenew();
	if (s == nil)
		return nil;
	s->id = id;
	s->opcode = NbnsOpQuery;
	s->broadcast = broadcast;
	s->recursiondesired = 1;
	q =  nbnsmessagequestionnew(name, NbnsQuestionTypeNb, NbnsQuestionClassIn);
	if (q == nil) {
		nbnsmessagefree(&s);
		return nil;
	}
	nbnsmessageaddquestion(s, q);
	return s;
}

NbnsMessage *
nbnsmessagenameregistrationrequestnew(uint16_t id, int broadcast,
				      NbName name,
				      uint32_t ttl, uint8_t *ipaddr)
{
	NbnsMessage *s;
	NbnsMessageQuestion *q;
	uint8_t rdata[6];
	NbnsMessageResource *r;

	s = nbnsmessagenew();
	if (s == nil)
		return nil;
	s->id = id;
	s->opcode = NbnsOpRegistration;
	s->broadcast = broadcast;
	s->recursiondesired = 1;
	q =  nbnsmessagequestionnew(name, NbnsQuestionTypeNb, NbnsQuestionClassIn);
	if (q == nil) {
		nbnsmessagefree(&s);
		return nil;
	}
	nbnsmessageaddquestion(s, q);
	rdata[0] = 0;
	rdata[1] = 0;
	v6tov4(rdata + 2, ipaddr);
	r = nbnsmessageresourcenew(name, NbnsResourceTypeNb, NbnsResourceClassIn, ttl, 6, rdata);
	nbnsmessageaddresource(&s->ar, r);
	return s;
}