summaryrefslogtreecommitdiff
path: root/contrib/smbfs/lib/smb/rap.c
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/smbfs/lib/smb/rap.c')
-rw-r--r--contrib/smbfs/lib/smb/rap.c404
1 files changed, 404 insertions, 0 deletions
diff --git a/contrib/smbfs/lib/smb/rap.c b/contrib/smbfs/lib/smb/rap.c
new file mode 100644
index 000000000000..952f66670b9e
--- /dev/null
+++ b/contrib/smbfs/lib/smb/rap.c
@@ -0,0 +1,404 @@
+/*
+ * Copyright (c) 2000, Boris Popov
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * 3. All advertising materials mentioning features or use of this software
+ * must display the following acknowledgement:
+ * This product includes software developed by Boris Popov.
+ * 4. Neither the name of the author nor the names of any co-contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $Id: rap.c,v 1.8 2001/02/24 15:56:05 bp Exp $
+ *
+ * This is very simple implementation of RAP protocol.
+ */
+#include <sys/param.h>
+#include <sys/errno.h>
+#include <sys/stat.h>
+#include <ctype.h>
+#include <err.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <strings.h>
+#include <stdlib.h>
+#include <sysexits.h>
+
+#include <sys/mchain.h>
+
+#include <netsmb/smb_lib.h>
+#include <netsmb/smb_conn.h>
+#include <netsmb/smb_rap.h>
+
+/*#include <sys/ioctl.h>*/
+
+static int
+smb_rap_parserqparam(const char *s, char **next, int *rlen)
+{
+ char *np;
+ int len, m;
+
+ m = 1;
+ switch (*s++) {
+ case 'L':
+ case 'T':
+ case 'W':
+ len = 2;
+ break;
+ case 'D':
+ case 'O':
+ len = 4;
+ break;
+ case 'b':
+ case 'F':
+ len = 1;
+ break;
+ case 'r':
+ case 's':
+ len = 0;
+ break;
+ default:
+ return EINVAL;
+ }
+ if (isdigit(*s)) {
+ len *= strtoul(s, &np, 10);
+ s = np;
+ }
+ *rlen = len;
+ *(const char**)next = s;
+ return 0;
+}
+
+static int
+smb_rap_parserpparam(const char *s, char **next, int *rlen)
+{
+ char *np;
+ int len, m;
+
+ m = 1;
+ switch (*s++) {
+ case 'e':
+ case 'h':
+ len = 2;
+ break;
+ case 'i':
+ len = 4;
+ break;
+ case 'g':
+ len = 1;
+ break;
+ default:
+ return EINVAL;
+ }
+ if (isdigit(*s)) {
+ len *= strtoul(s, &np, 10);
+ s = np;
+ }
+ *rlen = len;
+ *(const char**)next = s;
+ return 0;
+}
+
+static int
+smb_rap_parserpdata(const char *s, char **next, int *rlen)
+{
+ char *np;
+ int len, m;
+
+ m = 1;
+ switch (*s++) {
+ case 'B':
+ len = 1;
+ break;
+ case 'W':
+ len = 2;
+ break;
+ case 'D':
+ case 'O':
+ case 'z':
+ len = 4;
+ break;
+ default:
+ return EINVAL;
+ }
+ if (isdigit(*s)) {
+ len *= strtoul(s, &np, 10);
+ s = np;
+ }
+ *rlen = len;
+ *(const char**)next = s;
+ return 0;
+}
+
+static int
+smb_rap_rqparam_z(struct smb_rap *rap, const char *value)
+{
+ int len = strlen(value) + 1;
+
+ bcopy(value, rap->r_npbuf, len);
+ rap->r_npbuf += len;
+ rap->r_plen += len;
+ return 0;
+}
+
+static int
+smb_rap_rqparam(struct smb_rap *rap, char ptype, char plen, long value)
+{
+ char *p = rap->r_npbuf;
+ int len;
+
+ switch (ptype) {
+ case 'L':
+ case 'W':
+ setwle(p, 0, value);
+ len = 2;
+ break;
+ case 'D':
+ setdle(p, 0, value);
+ len = 4;
+ break;
+ case 'b':
+ memset(p, value, plen);
+ len = plen;
+ default:
+ return EINVAL;
+ }
+ rap->r_npbuf += len;
+ rap->r_plen += len;
+ return 0;
+}
+
+int
+smb_rap_create(int fn, const char *param, const char *data,
+ struct smb_rap **rapp)
+{
+ struct smb_rap *rap;
+ char *p;
+ int plen, len;
+
+ rap = malloc(sizeof(*rap));
+ if (rap == NULL)
+ return NULL;
+ bzero(rap, sizeof(*rap));
+ p = rap->r_sparam = rap->r_nparam = strdup(param);
+ rap->r_sdata = rap->r_ndata = strdup(data);
+ /*
+ * Calculate length of request parameter block
+ */
+ len = 2 + strlen(param) + 1 + strlen(data) + 1;
+
+ while (*p) {
+ if (smb_rap_parserqparam(p, &p, &plen) != 0)
+ break;
+ len += plen;
+ }
+ rap->r_pbuf = rap->r_npbuf = malloc(len);
+ smb_rap_rqparam(rap, 'W', 1, fn);
+ smb_rap_rqparam_z(rap, rap->r_sparam);
+ smb_rap_rqparam_z(rap, rap->r_sdata);
+ *rapp = rap;
+ return 0;
+}
+
+void
+smb_rap_done(struct smb_rap *rap)
+{
+ if (rap->r_sparam)
+ free(rap->r_sparam);
+ if (rap->r_sdata)
+ free(rap->r_sdata);
+ free(rap);
+}
+
+int
+smb_rap_setNparam(struct smb_rap *rap, long value)
+{
+ char *p = rap->r_nparam;
+ char ptype = *p;
+ int error, plen;
+
+ error = smb_rap_parserqparam(p, &p, &plen);
+ if (error)
+ return error;
+ switch (ptype) {
+ case 'L':
+ rap->r_rcvbuflen = value;
+ /* FALLTHROUGH */
+ case 'W':
+ case 'D':
+ case 'b':
+ error = smb_rap_rqparam(rap, ptype, plen, value);
+ break;
+ default:
+ return EINVAL;
+ }
+ rap->r_nparam = p;
+ return 0;
+}
+
+int
+smb_rap_setPparam(struct smb_rap *rap, void *value)
+{
+ char *p = rap->r_nparam;
+ char ptype = *p;
+ int error, plen;
+
+ error = smb_rap_parserqparam(p, &p, &plen);
+ if (error)
+ return error;
+ switch (ptype) {
+ case 'r':
+ rap->r_rcvbuf = value;
+ break;
+ default:
+ return EINVAL;
+ }
+ rap->r_nparam = p;
+ return 0;
+}
+
+static int
+smb_rap_getNparam(struct smb_rap *rap, long *value)
+{
+ char *p = rap->r_nparam;
+ char ptype = *p;
+ int error, plen;
+
+ error = smb_rap_parserpparam(p, &p, &plen);
+ if (error)
+ return error;
+ switch (ptype) {
+ case 'h':
+ *value = letohs(*(u_int16_t*)rap->r_npbuf);
+ break;
+ default:
+ return EINVAL;
+ }
+ rap->r_npbuf += plen;
+ rap->r_nparam = p;
+ return 0;
+}
+
+int
+smb_rap_request(struct smb_rap *rap, struct smb_ctx *ctx)
+{
+ u_int16_t *rp, conv;
+ u_int32_t *p32;
+ char *dp, *p = rap->r_nparam;
+ char ptype;
+ int error, rdatacnt, rparamcnt, entries, done, dlen;
+
+ rdatacnt = rap->r_rcvbuflen;
+ rparamcnt = rap->r_plen;
+ error = smb_t2_request(ctx, 0, 0, "\\PIPE\\LANMAN",
+ rap->r_plen, rap->r_pbuf, /* int tparamcnt, void *tparam */
+ 0, NULL, /* int tdatacnt, void *tdata */
+ &rparamcnt, rap->r_pbuf, /* rparamcnt, void *rparam */
+ &rdatacnt, rap->r_rcvbuf /* int *rdatacnt, void *rdata */
+ );
+ if (error)
+ return error;
+ rp = (u_int16_t*)rap->r_pbuf;
+ rap->r_result = letohs(*rp++);
+ conv = letohs(*rp++);
+ rap->r_npbuf = (char*)rp;
+ rap->r_entries = entries = 0;
+ done = 0;
+ while (!done && *p) {
+ ptype = *p;
+ switch (ptype) {
+ case 'e':
+ rap->r_entries = entries = letohs(*(u_int16_t*)rap->r_npbuf);
+ rap->r_npbuf += 2;
+ p++;
+ break;
+ default:
+ done = 1;
+ }
+/* error = smb_rap_parserpparam(p, &p, &plen);
+ if (error) {
+ smb_error("reply parameter mismath %s", 0, p);
+ return EBADRPC;
+ }*/
+ }
+ rap->r_nparam = p;
+ /*
+ * In general, unpacking entries we may need to relocate
+ * entries for proper alingning. For now use them as is.
+ */
+ dp = rap->r_rcvbuf;
+ while (entries--) {
+ p = rap->r_sdata;
+ while (*p) {
+ ptype = *p;
+ error = smb_rap_parserpdata(p, &p, &dlen);
+ if (error) {
+ smb_error("reply data mismath %s", 0, p);
+ return EBADRPC;
+ }
+ switch (ptype) {
+ case 'z':
+ p32 = (u_int32_t*)dp;
+ *p32 = (*p32 & 0xffff) - conv;
+ break;
+ }
+ dp += dlen;
+ }
+ }
+ return error;
+}
+
+int
+smb_rap_error(struct smb_rap *rap, int error)
+{
+ if (error)
+ return error;
+ if (rap->r_result == 0)
+ return 0;
+ return rap->r_result | SMB_RAP_ERROR;
+}
+
+int
+smb_rap_NetShareEnum(struct smb_ctx *ctx, int sLevel, void *pbBuffer,
+ int cbBuffer, int *pcEntriesRead, int *pcTotalAvail)
+{
+ struct smb_rap *rap;
+ long lval;
+ int error;
+
+ error = smb_rap_create(0, "WrLeh", "B13BWz", &rap);
+ if (error)
+ return error;
+ smb_rap_setNparam(rap, sLevel); /* W - sLevel */
+ smb_rap_setPparam(rap, pbBuffer); /* r - pbBuffer */
+ smb_rap_setNparam(rap, cbBuffer); /* L - cbBuffer */
+ error = smb_rap_request(rap, ctx);
+ if (error == 0) {
+ *pcEntriesRead = rap->r_entries;
+ error = smb_rap_getNparam(rap, &lval);
+ *pcTotalAvail = lval;
+ }
+ error = smb_rap_error(rap, error);
+ smb_rap_done(rap);
+ return error;
+}