/*
 * 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 (c) 2010, Oracle and/or its affiliates. All rights reserved.
 */

#include <netdb.h>
#include <sys/types.h>
#include <string.h>
#include <strings.h>
#include <smbsrv/libsmb.h>
#include <smbsrv/libmlsvc.h>
#include <smbsrv/smbinfo.h>
#include <smbsrv/nmpipes.h>
#include <smbsrv/ndl/srvsvc.ndl>

static int wkssvc_s_NetWkstaGetInfo(void *, ndr_xa_t *);
static int wkssvc_s_NetWkstaTransportEnum(void *, ndr_xa_t *);

static ndr_stub_table_t wkssvc_stub_table[] = {
	{ wkssvc_s_NetWkstaGetInfo,	WKSSVC_OPNUM_NetWkstaGetInfo },
	{ wkssvc_s_NetWkstaTransportEnum, WKSSVC_OPNUM_NetWkstaTransportEnum },
	{0}
};

static ndr_service_t wkssvc_service = {
	"Workstation",			/* name (WKSSVC or WKSTA) */
	"Workstation services",		/* desc */
	"\\wkssvc",			/* endpoint */
	PIPE_NTSVCS,			/* sec_addr_port */
	"6bffd098-a112-3610-9833-46c3f87e345a", 1,	/* abstract */
	NDR_TRANSFER_SYNTAX_UUID,		2,	/* transfer */
	0,				/* no bind_instance_size */
	0,				/* no bind_req() */
	0,				/* no unbind_and_close() */
	0,				/* use generic_call_stub() */
	&TYPEINFO(wkssvc_interface),	/* interface ti */
	wkssvc_stub_table		/* stub_table */
};

void
wkssvc_initialize(void)
{
	(void) ndr_svc_register(&wkssvc_service);
}

/*
 * WKSSVC NetWkstaGetInfo
 */
static int
wkssvc_s_NetWkstaGetInfo(void *arg, ndr_xa_t *mxa)
{
	struct mslm_NetWkstaGetInfo *param = arg;
	mslm_NetWkstaGetInfo_rb *rb;
	char hostname[MAXHOSTNAMELEN];
	char resource_domain[SMB_PI_MAX_DOMAIN];
	smb_version_t version;
	char *name;
	char *domain;
	DWORD status;
	int rc;

	(void) smb_getdomainname(resource_domain, SMB_PI_MAX_DOMAIN);

	rb = NDR_NEW(mxa, mslm_NetWkstaGetInfo_rb);

	if ((rc = smb_getnetbiosname(hostname, MAXHOSTNAMELEN)) == 0) {
		name = NDR_STRDUP(mxa, hostname);
		domain = NDR_STRDUP(mxa, resource_domain);
	}

	if ((rc != 0) || (rb == NULL) || (name == NULL) || (domain == NULL)) {
		bzero(param, sizeof (struct mslm_NetWkstaGetInfo));
		param->status = ERROR_NOT_ENOUGH_MEMORY;
		return (NDR_DRC_OK);
	}

	smb_config_get_version(&version);

	param->result.level = param->level;
	param->result.bufptr.nullptr = (void *)rb;

	switch (param->level) {
	case 100:
		rb->buf100.wki100_platform_id = SV_PLATFORM_ID_NT;
		rb->buf100.wki100_ver_major = version.sv_major;
		rb->buf100.wki100_ver_minor = version.sv_minor;
		rb->buf100.wki100_computername = (unsigned char *)name;
		rb->buf100.wki100_langroup = (unsigned char *)domain;
		status = ERROR_SUCCESS;
		break;

	case 101:
		rb->buf101.wki101_platform_id = SV_PLATFORM_ID_NT;
		rb->buf101.wki101_ver_major = version.sv_major;
		rb->buf101.wki101_ver_minor = version.sv_minor;
		rb->buf101.wki101_computername = (unsigned char *)name;
		rb->buf101.wki101_langroup = (unsigned char *)domain;
		rb->buf101.wki101_lanroot = (unsigned char *)"";
		status = ERROR_SUCCESS;
		break;

	case 102:
		rb->buf102.wki102_platform_id = SV_PLATFORM_ID_NT;
		rb->buf102.wki102_ver_major = version.sv_major;
		rb->buf102.wki102_ver_minor = version.sv_minor;
		rb->buf102.wki102_computername = (unsigned char *)name;
		rb->buf102.wki102_langroup = (unsigned char *)domain;
		rb->buf102.wki102_lanroot = (unsigned char *)"";
		rb->buf102.wki102_logged_on_users = 1;
		status = ERROR_SUCCESS;
		break;

	case 502:
		bzero(&rb->buf502, sizeof (struct mslm_WKSTA_INFO_502));
		rb->buf502.keep_connection = 600;
		rb->buf502.max_commands = 1024;
		rb->buf502.session_timeout = 5400;
		rb->buf502.size_char_buf = 1024;
		rb->buf502.max_threads = 1024;
		rb->buf502.use_opportunistic_locking = 1;
		rb->buf502.use_unlock_behind = 1;
		rb->buf502.use_close_behind = 1;
		rb->buf502.buf_named_pipes = 1;
		rb->buf502.use_lock_read_unlock = 1;
		rb->buf502.utilize_nt_caching = 1;
		rb->buf502.use_raw_read = 1;
		rb->buf502.use_raw_write = 1;
		status = ERROR_SUCCESS;
		break;

	default:
		param->result.bufptr.nullptr = 0;
		status = ERROR_INVALID_LEVEL;
		break;
	}

	if (status != ERROR_SUCCESS) {
		bzero(param, sizeof (struct mslm_NetWkstaGetInfo));
		param->status = status;
	}

	return (NDR_DRC_OK);
}

/*
 * WKSSVC NetWkstaTransportEnum
 */
static int
wkssvc_s_NetWkstaTransportEnum(void *arg, ndr_xa_t *mxa)
{
	struct mslm_NetWkstaTransportEnum *param = arg;
	struct mslm_NetWkstaTransportCtr0 *info0;
	struct mslm_NetWkstaTransportInfo0 *ti0;

	switch (param->info.level) {
	case 0:
		info0 = NDR_NEW(mxa, struct mslm_NetWkstaTransportCtr0);
		ti0 = NDR_NEW(mxa, struct mslm_NetWkstaTransportInfo0);

		if (info0 == NULL || ti0 == NULL) {
			bzero(param, sizeof (struct mslm_NetWkstaGetInfo));
			param->status = ERROR_NOT_ENOUGH_MEMORY;
			break;
		}

		ti0->quality_of_service = 65535;
		ti0->num_vcs = 0;
		ti0->transport_name = (unsigned char *)"\\Device\\NetbiosSmb";
		ti0->transport_address = (unsigned char *)"000000000000";
		ti0->wan_ish = 1024;

		info0->count = 1;
		info0->ti0 = ti0;
		param->info.ru.info0 = info0;
		param->total_entries = 1;

		if (param->resume_handle)
			*param->resume_handle = 0;

		param->status = ERROR_SUCCESS;
		break;

	default:
		bzero(param, sizeof (struct mslm_NetWkstaGetInfo));
		param->status = ERROR_INVALID_LEVEL;
	}

	return (NDR_DRC_OK);
}