xref: /illumos-gate/usr/src/lib/smbsrv/libsmbns/common/smbns_netbios.c (revision da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0)

/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

/*
 * Main startup code for SMB/NETBIOS and some utility routines
 * for the NETBIOS layer.
 */

#include <synch.h>
#include <unistd.h>
#include <syslog.h>
#include <string.h>
#include <strings.h>
#include <sys/socket.h>

#include <smbns_netbios.h>

netbios_status_t nb_status;

static pthread_t smb_nbns_thr; /* name service */
static pthread_t smb_nbds_thr; /* dgram service */
static pthread_t smb_nbts_thr; /* timer */
static pthread_t smb_nbbs_thr; /* browser */

static void *smb_netbios_timer(void *);

void
smb_netbios_chg_status(uint32_t status, int set)
{
	(void) mutex_lock(&nb_status.mtx);
	if (set)
		nb_status.state |= status;
	else
		nb_status.state &= ~status;
	(void) cond_broadcast(&nb_status.cv);
	(void) mutex_unlock(&nb_status.mtx);
}

void
smb_netbios_shutdown(void)
{
	smb_netbios_chg_status(NETBIOS_SHUTTING_DOWN, 1);

	(void) pthread_join(smb_nbts_thr, 0);
	(void) pthread_join(smb_nbbs_thr, 0);
	(void) pthread_join(smb_nbns_thr, 0);
	(void) pthread_join(smb_nbds_thr, 0);

	nb_status.state = NETBIOS_SHUT_DOWN;
}

void
smb_netbios_start()
{
	int rc;
	mutex_t *mp;
	cond_t *cvp;

	/* Startup Netbios named; port 137 */
	rc = pthread_create(&smb_nbns_thr, 0,
	    smb_netbios_name_service_daemon, 0);
	if (rc)
		return;

	mp = &nb_status.mtx;
	cvp = &nb_status.cv;

	(void) mutex_lock(mp);

	while (!(nb_status.state & (NETBIOS_NAME_SVC_RUNNING |
	    NETBIOS_NAME_SVC_FAILED))) {
		(void) cond_wait(cvp, mp);
	}

	if (nb_status.state & NETBIOS_NAME_SVC_FAILED) {
		(void) mutex_unlock(mp);
		(void) fprintf(stderr,
		    "smbd: Netbios Name service startup failed!");
		smb_netbios_shutdown();
		return;
	}
	(void) mutex_unlock(mp);

	(void) fprintf(stderr, "smbd: Netbios Name service started.");
	smb_netbios_name_config();

	/* Startup Netbios datagram service; port 138 */
	rc = pthread_create(&smb_nbds_thr, 0,
	    smb_netbios_datagram_service_daemon, 0);
	if (rc == 0) {
		(void) mutex_lock(mp);
		while (!(nb_status.state & (NETBIOS_DATAGRAM_SVC_RUNNING |
		    NETBIOS_DATAGRAM_SVC_FAILED))) {
			(void) cond_wait(cvp, mp);
		}

		if (nb_status.state & NETBIOS_DATAGRAM_SVC_FAILED) {
			(void) mutex_unlock(mp);
			(void) fprintf(stderr, "smbd: Netbios Datagram service "
			    "startup failed!");
			smb_netbios_shutdown();
			return;
		}
		(void) mutex_unlock(mp);
	} else {
		smb_netbios_shutdown();
		return;
	}

	(void) fprintf(stderr, "smbd: Netbios Datagram service started.");

	/* Startup Netbios browser service */
	rc = pthread_create(&smb_nbbs_thr, 0, smb_browser_daemon, 0);
	if (rc) {
		smb_netbios_shutdown();
		return;
	}

	(void) fprintf(stderr, "smbd: Netbios Browser client started.");

	/* Startup Our internal, 1 second resolution, timer */
	rc = pthread_create(&smb_nbts_thr, 0, smb_netbios_timer, 0);
	if (rc == 0) {
		(void) mutex_lock(mp);
		while (!(nb_status.state & (NETBIOS_TIMER_RUNNING |
		    NETBIOS_TIMER_FAILED))) {
			(void) cond_wait(cvp, mp);
		}

		if (nb_status.state & NETBIOS_TIMER_FAILED) {
			(void) mutex_unlock(mp);
			smb_netbios_shutdown();
			return;
		}
		(void) mutex_unlock(mp);
	} else {
		smb_netbios_shutdown();
		return;
	}

	(void) fprintf(stderr, "smbd: Netbios Timer service started.");
}

/*ARGSUSED*/
static void *
smb_netbios_timer(void *arg)
{
	static unsigned int	ticks;

	smb_netbios_chg_status(NETBIOS_TIMER_RUNNING, 1);

	while ((nb_status.state & NETBIOS_SHUTTING_DOWN) == 0) {
		(void) sleep(1);

		if (nb_status.state & NETBIOS_DATAGRAM_SVC_RUNNING)
			smb_netbios_datagram_tick();
		else
			break;

		if (nb_status.state & NETBIOS_NAME_SVC_RUNNING) {
			smb_netbios_name_tick();

			/* every 10 minutes */
			if ((ticks % 600) == 0)
				smb_netbios_cache_clean();
		}
		else
			break;
	}

	nb_status.state &= ~NETBIOS_TIMER_RUNNING;
	if ((nb_status.state & NETBIOS_SHUTTING_DOWN) == 0) {
		/* either name or datagram service has failed */
		smb_netbios_shutdown();
	}

	return (0);
}

int
smb_first_level_name_encode(struct name_entry *name,
				unsigned char *out, int max_out)
{
	return (netbios_first_level_name_encode(name->name, name->scope,
	    out, max_out));
}

int
smb_first_level_name_decode(unsigned char *in, struct name_entry *name)
{
	return (netbios_first_level_name_decode((char *)in, (char *)name->name,
	    (char *)name->scope));
}

/*
 * smb_encode_netbios_name
 *
 * Set up the name and scope fields in the destination name_entry structure.
 * The name is padded with spaces to 15 bytes. The suffix is copied into the
 * last byte, i.e. "netbiosname    <suffix>". The scope is copied and folded
 * to uppercase.
 */
void
smb_encode_netbios_name(unsigned char *name, char suffix, unsigned char *scope,
    struct name_entry *dest)
{
	char tmp_name[NETBIOS_NAME_SZ];
	mts_wchar_t wtmp_name[NETBIOS_NAME_SZ];
	unsigned int cpid;
	int	len;
	size_t rc;

	len = 0;
	rc = mts_mbstowcs(wtmp_name, (const char *)name, NETBIOS_NAME_SZ);

	if (rc != (size_t)-1) {
		wtmp_name[NETBIOS_NAME_SZ - 1] = 0;
		cpid = oem_get_smb_cpid();
		rc = unicodestooems(tmp_name, wtmp_name, NETBIOS_NAME_SZ, cpid);
		if (rc > 0)
			len = strlen(tmp_name);
	}

	(void) memset(dest->name, ' ', NETBIOS_NAME_SZ - 1);
	if (len) {
		(void) utf8_strupr(tmp_name);
		(void) memcpy(dest->name, tmp_name, len);
	}
	dest->name[NETBIOS_NAME_SZ - 1] = suffix;

	if (scope == NULL) {
		smb_config_rdlock();
		(void) strlcpy((char *)dest->scope,
		    smb_config_getstr(SMB_CI_NBSCOPE), NETBIOS_DOMAIN_NAME_MAX);
		smb_config_unlock();
	} else {
		(void) strlcpy((char *)dest->scope, (const char *)scope,
		    NETBIOS_DOMAIN_NAME_MAX);
	}
	(void) utf8_strupr((char *)dest->scope);
}

void
smb_init_name_struct(unsigned char *name, char suffix, unsigned char *scope,
    uint32_t ipaddr, unsigned short port, uint32_t attr,
    uint32_t addr_attr, struct name_entry *dest)
{
	bzero(dest, sizeof (struct name_entry));
	smb_encode_netbios_name(name, suffix, scope, dest);

	switch (smb_node_type) {
	case 'H':
		dest->attributes = attr | NAME_ATTR_OWNER_TYPE_HNODE;
		break;
	case 'M':
		dest->attributes = attr | NAME_ATTR_OWNER_TYPE_MNODE;
		break;
	case 'P':
		dest->attributes = attr | NAME_ATTR_OWNER_TYPE_PNODE;
		break;
	case 'B':
	default:
		dest->attributes = attr | NAME_ATTR_OWNER_TYPE_BNODE;
		break;
	}

	dest->addr_list.refresh_ttl = dest->addr_list.ttl =
	    TO_SECONDS(DEFAULT_TTL);

	dest->addr_list.sin.sin_family = AF_INET;
	dest->addr_list.sinlen = sizeof (dest->addr_list.sin);
	dest->addr_list.sin.sin_addr.s_addr = ipaddr;
	dest->addr_list.sin.sin_port = port;
	dest->addr_list.attributes = addr_attr;
	dest->addr_list.forw = dest->addr_list.back = &dest->addr_list;
}