smb_msgbuf.c revision b819cea2f73f98c5662230cc9affc8cc84f77fcf
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * CDDL HEADER START
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * The contents of this file are subject to the terms of the
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Common Development and Distribution License (the "License").
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * You may not use this file except in compliance with the License.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * or http://www.opensolaris.org/os/licensing.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * See the License for the specific language governing permissions
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * and limitations under the License.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * When distributing Covered Code, include this CDDL HEADER in each
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * If applicable, add the following below this CDDL HEADER, with the
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * fields enclosed by brackets "[]" replaced with your own identifying
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * information: Portions Copyright [yyyy] [name of copyright owner]
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * CDDL HEADER END
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown * Copyright 2009 Sun Microsystems, Inc. All rights reserved.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Use is subject to license terms.
b819cea2f73f98c5662230cc9affc8cc84f77fcfGordon Ross *
b819cea2f73f98c5662230cc9affc8cc84f77fcfGordon Ross * Copyright 2014 Nexenta Systems, Inc. All rights reserved.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Msgbuf buffer management implementation. The smb_msgbuf interface is
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * typically used to encode or decode SMB data using sprintf/scanf
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * style operations. It contains special handling for the SMB header.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * It can also be used for general purpose encoding and decoding.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <sys/types.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <sys/varargs.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <sys/byteorder.h>
b819cea2f73f98c5662230cc9affc8cc84f77fcfGordon Ross#if !defined(_KERNEL) && !defined(_FAKE_KERNEL)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <stdlib.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <syslog.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <string.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <strings.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#else
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <sys/sunddi.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <sys/kmem.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#endif
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <smbsrv/string.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <smbsrv/msgbuf.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#include <smbsrv/smb.h>
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int buf_decode(smb_msgbuf_t *, char *, va_list ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int buf_encode(smb_msgbuf_t *, char *, va_list ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic void *smb_msgbuf_malloc(smb_msgbuf_t *, size_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int smb_msgbuf_chkerc(char *text, int erc);
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brownstatic void buf_decode_wcs(smb_wchar_t *, smb_wchar_t *, int wcstrlen);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Returns the offset or number of bytes used within the buffer.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsize_t
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_used(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*LINTED E_PTRDIFF_OVERFLOW*/
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (mb->scan - mb->base);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Returns the actual buffer size.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsize_t
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_size(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (mb->max);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwuint8_t *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_base(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (mb->base);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Ensure that the scan is aligned on a word (16-bit) boundary.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_word_align(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan = (uint8_t *)((uintptr_t)(mb->scan + 1) & ~1);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Ensure that the scan is aligned on a dword (32-bit) boundary.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_dword_align(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan = (uint8_t *)((uintptr_t)(mb->scan + 3) & ~3);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Checks whether or not the buffer has space for the amount of data
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * specified. Returns 1 if there is space, otherwise returns 0.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwint
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_has_space(smb_msgbuf_t *mb, size_t size)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (size > mb->max || (mb->scan + size) > mb->end)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (0);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (1);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Set flags the smb_msgbuf.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_fset(smb_msgbuf_t *mb, uint32_t flags)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->flags |= flags;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Clear flags the smb_msgbuf.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_fclear(smb_msgbuf_t *mb, uint32_t flags)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->flags &= ~flags;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_init
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Initialize a smb_msgbuf_t structure based on the buffer and size
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * specified. Both scan and base initially point to the beginning
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * of the buffer and end points to the limit of the buffer. As
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * data is added scan should be incremented to point to the next
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * offset at which data will be written. Max and count are set
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * to the actual buffer size.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_init(smb_msgbuf_t *mb, uint8_t *buf, size_t size, uint32_t flags)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan = mb->base = buf;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->max = mb->count = size;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->end = &buf[size];
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->flags = flags;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->mlist.next = 0;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_term
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Destruct a smb_msgbuf_t. Free any memory hanging off the mlist.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwvoid
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_term(smb_msgbuf_t *mb)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw smb_msgbuf_mlist_t *item = mb->mlist.next;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw smb_msgbuf_mlist_t *tmp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (item) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw tmp = item;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw item = item->next;
b819cea2f73f98c5662230cc9affc8cc84f77fcfGordon Ross#if !defined(_KERNEL) && !defined(_FAKE_KERNEL)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw free(tmp);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#else
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw kmem_free(tmp, tmp->size);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#endif
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_decode
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Decode a smb_msgbuf buffer as indicated by the format string into
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * the variable arg list. This is similar to a scanf operation.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * On success, returns the number of bytes encoded. Otherwise
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * returns a -ve error code.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwint
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_decode(smb_msgbuf_t *mb, char *fmt, ...)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int rc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t *orig_scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_list ap;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_start(ap, fmt);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw orig_scan = mb->scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = buf_decode(mb, fmt, ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_end(ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc != SMB_MSGBUF_SUCCESS) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw (void) smb_msgbuf_chkerc("smb_msgbuf_decode", rc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan = orig_scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (rc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*LINTED E_PTRDIFF_OVERFLOW*/
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (mb->scan - orig_scan);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * buf_decode
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Private decode function, where the real work of decoding the smb_msgbuf
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * is done. This function should only be called via smb_msgbuf_decode to
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * ensure correct behaviour and error handling.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwbuf_decode(smb_msgbuf_t *mb, char *fmt, va_list ap)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint32_t ival;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t c;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t *cvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t **cvalpp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint16_t *wvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint32_t *lvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint64_t *llvalp;
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown smb_wchar_t *wcs;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int repc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int rc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while ((c = *fmt++) != 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (c == ' ' || c == '\t')
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw continue;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (c == '(') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (((c = *fmt++) != 0) && c != ')')
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw ;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (!c)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_SUCCESS);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw continue;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ('0' <= c && c <= '9') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = 0;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw do {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = repc * 10 + c - '0';
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw c = *fmt++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } while ('0' <= c && c <= '9');
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } else if (c == '#') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = va_arg(ap, int);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw c = *fmt++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw switch (c) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case '.':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += repc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'c':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp = va_arg(ap, uint8_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw bcopy(mb->scan, cvalp, repc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += repc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'b':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp = va_arg(ap, uint8_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *cvalp++ = *mb->scan++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'w':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (uint16_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw wvalp = va_arg(ap, uint16_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *wvalp++ = LE_IN16(mb->scan);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (uint16_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'l':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (int32_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw lvalp = va_arg(ap, uint32_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *lvalp++ = LE_IN32(mb->scan);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (int32_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'q':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (int64_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw llvalp = va_arg(ap, uint64_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *llvalp++ = LE_IN64(mb->scan);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (int64_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'u': /* Convert from unicode if flags are set */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (mb->flags & SMB_MSGBUF_UNICODE)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw goto unicode_translation;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*FALLTHROUGH*/
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 's':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw ival = strlen((const char *)mb->scan) + 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, ival) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ((cvalp = smb_msgbuf_malloc(mb, ival * 2)) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown if ((ival = smb_stombs((char *)cvalp,
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw (char *)mb->scan, ival * 2)) ==
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw (uint32_t)-1) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_DATA_ERROR);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalpp = va_arg(ap, uint8_t **);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *cvalpp = cvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += (ival+1);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'U': /* Convert from unicode */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwunicode_translation:
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Unicode strings are always word aligned.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * The malloc'd area is larger than the
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * original string because the UTF-8 chars
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * may be longer than the wide-chars.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw smb_msgbuf_word_align(mb);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*LINTED E_BAD_PTR_CAST_ALIGN*/
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown wcs = (smb_wchar_t *)mb->scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* count the null wchar */
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown repc = sizeof (smb_wchar_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (*wcs++)
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown repc += sizeof (smb_wchar_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* Decode wchar string into host byte-order */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ((wcs = smb_msgbuf_malloc(mb, repc)) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*LINTED E_BAD_PTR_CAST_ALIGN*/
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown buf_decode_wcs(wcs, (smb_wchar_t *)mb->scan,
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown repc / sizeof (smb_wchar_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* Get space for translated string */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ((cvalp = smb_msgbuf_malloc(mb, repc * 2)) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* Translate string */
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown (void) smb_wcstombs((char *)cvalp, wcs, repc * 2);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalpp = va_arg(ap, uint8_t **);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *cvalpp = cvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += repc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'M':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, 4) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_UNDERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (mb->scan[0] != 0xFF ||
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan[1] != 'S' ||
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan[2] != 'M' ||
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan[3] != 'B') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_INVALID_HEADER);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += 4;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw default:
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_INVALID_FORMAT);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_SUCCESS);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_encode
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Encode a smb_msgbuf buffer as indicated by the format string using
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * the variable arg list. This is similar to a sprintf operation.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * On success, returns the number of bytes encoded. Otherwise
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * returns a -ve error code.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwint
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_encode(smb_msgbuf_t *mb, char *fmt, ...)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int rc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t *orig_scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_list ap;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_start(ap, fmt);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw orig_scan = mb->scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = buf_encode(mb, fmt, ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw va_end(ap);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc != SMB_MSGBUF_SUCCESS) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw (void) smb_msgbuf_chkerc("smb_msgbuf_encode", rc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan = orig_scan;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (rc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*LINTED E_PTRDIFF_OVERFLOW*/
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (mb->scan - orig_scan);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * buf_encode
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Private encode function, where the real work of encoding the smb_msgbuf
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * is done. This function should only be called via smb_msgbuf_encode to
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * ensure correct behaviour and error handling.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwbuf_encode(smb_msgbuf_t *mb, char *fmt, va_list ap)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t cval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint16_t wval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint32_t lval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint64_t llval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint32_t ival;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t *cvalp;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw uint8_t c;
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown smb_wchar_t wcval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int count;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int repc = 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int rc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while ((c = *fmt++) != 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (c == ' ' || c == '\t')
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw continue;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (c == '(') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (((c = *fmt++) != 0) && c != ')')
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw ;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (!c)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_SUCCESS);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw continue;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ('0' <= c && c <= '9') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = 0;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw do {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = repc * 10 + c - '0';
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw c = *fmt++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } while ('0' <= c && c <= '9');
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } else if (c == '#') {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw repc = va_arg(ap, int);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw c = *fmt++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw switch (c) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case '.':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = 0;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'c':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp = va_arg(ap, uint8_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw bcopy(cvalp, mb->scan, repc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += repc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'b':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, repc) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cval = va_arg(ap, int);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = cval;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'w':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (uint16_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw wval = va_arg(ap, int);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw LE_OUT16(mb->scan, wval);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (uint16_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'l':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (int32_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw lval = va_arg(ap, uint32_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw LE_OUT32(mb->scan, lval);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (int32_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'q':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb, repc * sizeof (int64_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw while (repc-- > 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw llval = va_arg(ap, uint64_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw LE_OUT64(mb->scan, llval);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += sizeof (uint64_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'u': /* conditional unicode */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (mb->flags & SMB_MSGBUF_UNICODE)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw goto unicode_translation;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* FALLTHROUGH */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 's':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp = va_arg(ap, uint8_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw ival = strlen((const char *)cvalp) + 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, ival) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw ival =
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown smb_mbstos((char *)mb->scan, (const char *)cvalp);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan += ival + 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'U': /* unicode */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwunicode_translation:
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Unicode strings are always word aligned.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw smb_msgbuf_word_align(mb);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp = va_arg(ap, uint8_t *);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw for (;;) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw rc = smb_msgbuf_has_space(mb,
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown sizeof (smb_wchar_t));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (rc == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown count = smb_mbtowc(&wcval, (const char *)cvalp,
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw MTS_MB_CHAR_MAX);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (count < 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_DATA_ERROR);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } else if (count == 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * No longer need to do this now that
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * mbtowc correctly writes the null
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * before returning zero but paranoia
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * wins.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw wcval = 0;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw count = 1;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /* Write wchar in wire-format */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw LE_OUT16(mb->scan, wcval);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (*cvalp == 0) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * End of string. Check to see whether
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * or not to include the null
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * terminator.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ((mb->flags & SMB_MSGBUF_NOTERM) ==
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->scan +=
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown sizeof (smb_wchar_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brown mb->scan += sizeof (smb_wchar_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw cvalp += count;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw case 'M':
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (smb_msgbuf_has_space(mb, 4) == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_OVERFLOW);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = 0xFF;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = 'S';
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = 'M';
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *mb->scan++ = 'B';
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw default:
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_INVALID_FORMAT);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (SMB_MSGBUF_SUCCESS);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_malloc
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Allocate some memory for use with this smb_msgbuf. We increase the
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * requested size to hold the list pointer and return a pointer
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * to the area for use by the caller.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic void *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_malloc(smb_msgbuf_t *mb, size_t size)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw smb_msgbuf_mlist_t *item;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw size += sizeof (smb_msgbuf_mlist_t);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
b819cea2f73f98c5662230cc9affc8cc84f77fcfGordon Ross#if !defined(_KERNEL) && !defined(_FAKE_KERNEL)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if ((item = malloc(size)) == NULL)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (NULL);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#else
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw item = kmem_alloc(size, KM_SLEEP);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw#endif
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw item->next = mb->mlist.next;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw item->size = size;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw mb->mlist.next = item;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw /*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * The caller gets a pointer to the address
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * immediately after the smb_msgbuf_mlist_t.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return ((void *)(item + 1));
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw/*
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * smb_msgbuf_chkerc
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw * Diagnostic function to write an appropriate message to the system log.
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw */
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic int
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwsmb_msgbuf_chkerc(char *text, int erc)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw static struct {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int erc;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw char *name;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw } etable[] = {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw { SMB_MSGBUF_SUCCESS, "success" },
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw { SMB_MSGBUF_UNDERFLOW, "overflow/underflow" },
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw { SMB_MSGBUF_INVALID_FORMAT, "invalid format" },
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw { SMB_MSGBUF_INVALID_HEADER, "invalid header" },
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw { SMB_MSGBUF_DATA_ERROR, "data error" }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw };
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int i;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw for (i = 0; i < sizeof (etable)/sizeof (etable[0]); ++i) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (etable[i].erc == erc) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw if (text == 0)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw text = "smb_msgbuf_chkerc";
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw break;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw return (erc);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amwstatic void
bbf6f00c25b6a2bed23c35eac6d62998ecdb338cJordan Brownbuf_decode_wcs(smb_wchar_t *dst_wcstr, smb_wchar_t *src_wcstr, int wcstrlen)
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw{
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw int i;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw for (i = 0; i < wcstrlen; i++) {
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw *dst_wcstr = LE_IN16(src_wcstr);
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw dst_wcstr++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw src_wcstr++;
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw }
da6c28aaf62fa55f0fdb8004aa40f88f23bf53f0amw}