dist/smbfs/lib/smb/nb.c

   1 /*
   2  * Copyright (c) 2000, 2001 Boris Popov
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *    This product includes software developed by Boris Popov.
  16  * 4. Neither the name of the author nor the names of any co-contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  *
  32  * from: Id: nb.c,v 1.4 2001/04/16 04:33:01 bp Exp
  33  */
  34
  35 #include <sys/cdefs.h>
  36 __RCSID("$NetBSD: nb.c,v 1.4 2006/03/22 02:29:42 christos Exp $");
  37
  38 #include <sys/param.h>
  39 #include <sys/socket.h>
  40
  41 #include <ctype.h>
  42 #include <netdb.h>
  43 #include <err.h>
  44 #include <errno.h>
  45 #include <stdlib.h>
  46 #include <string.h>
  47 #include <stdio.h>
  48 #include <unistd.h>
  49 #include <cflib.h>
  50
  51 #include <netsmb/netbios.h>
  52 #include <netsmb/smb_lib.h>
  53 #include <netsmb/nb_lib.h>
  54
  55 int
  56 nb_ctx_create(struct nb_ctx **ctxpp)
  57 {
  58         struct nb_ctx *ctx;
  59
  60         ctx = malloc(sizeof(struct nb_ctx));
  61         if (ctx == NULL)
  62                 return ENOMEM;
  63         bzero(ctx, sizeof(struct nb_ctx));
  64         *ctxpp = ctx;
  65         return 0;
  66 }
  67
  68 void
  69 nb_ctx_done(struct nb_ctx *ctx)
  70 {
  71         if (ctx == NULL)
  72                 return;
  73         if (ctx->nb_scope)
  74                 free(ctx->nb_scope);
  75 }
  76
  77 int
  78 nb_ctx_setns(struct nb_ctx *ctx, const char *addr)
  79 {
  80         if (addr == NULL || addr[0] == 0)
  81                 return EINVAL;
  82         if (ctx->nb_nsname)
  83                 free(ctx->nb_nsname);
  84         if ((ctx->nb_nsname = strdup(addr)) == NULL)
  85                 return ENOMEM;
  86         return 0;
  87 }
  88
  89 int
  90 nb_ctx_setscope(struct nb_ctx *ctx, const char *scope)
  91 {
  92         size_t slen = strlen(scope);
  93
  94         if (slen >= 128) {
  95                 smb_error("scope '%s' is too long", 0, scope);
  96                 return ENAMETOOLONG;
  97         }
  98         if (ctx->nb_scope)
  99                 free(ctx->nb_scope);
 100         ctx->nb_scope = malloc(slen + 1);
 101         if (ctx->nb_scope == NULL)
 102                 return ENOMEM;
 103         nls_str_upper(ctx->nb_scope, scope);
 104         return 0;
 105 }
 106
 107 int
 108 nb_ctx_resolve(struct nb_ctx *ctx)
 109 {
 110         struct sockaddr *sap;
 111         int error;
 112
 113         ctx->nb_flags &= ~NBCF_RESOLVED;
 114
 115         if (ctx->nb_nsname == NULL) {
 116                 ctx->nb_ns.sin_addr.s_addr = htonl(INADDR_BROADCAST);
 117         } else {
 118                 error = nb_resolvehost_in(ctx->nb_nsname, &sap);
 119                 if (error) {
 120                         smb_error("can't resolve %s", error, ctx->nb_nsname);
 121                         return error;
 122                 }
 123                 if (sap->sa_family != AF_INET) {
 124                         smb_error("unsupported address family %d", 0,
 125                             sap->sa_family);
 126                         free(sap);
 127                         return EINVAL;
 128                 }
 129                 bcopy(sap, &ctx->nb_ns, sizeof(ctx->nb_ns));
 130                 free(sap);
 131         }
 132         ctx->nb_ns.sin_port = htons(137);
 133         ctx->nb_ns.sin_family = AF_INET;
 134         ctx->nb_ns.sin_len = sizeof(ctx->nb_ns);
 135         ctx->nb_flags |= NBCF_RESOLVED;
 136         return 0;
 137 }
 138
 139 /*
 140  * used level values:
 141  * 0 - default
 142  * 1 - server
 143  */
 144 int
 145 nb_ctx_readrcsection(struct rcfile *rcfile, struct nb_ctx *ctx,
 146         const char *sname, int level)
 147 {
 148         char *p;
 149         int error;
 150
 151         if (level > 1)
 152                 return EINVAL;
 153         rc_getint(rcfile, sname, "nbtimeout", &ctx->nb_timo);
 154         rc_getstringptr(rcfile, sname, "nbns", &p);
 155         if (p) {
 156                 error = nb_ctx_setns(ctx, p);
 157                 if (error) {
 158                         smb_error("invalid address specified in the section %s", 0, sname);
 159                         return error;
 160                 }
 161         }
 162         rc_getstringptr(rcfile, sname, "nbscope", &p);
 163         if (p)
 164                 nb_ctx_setscope(ctx, p);
 165         return 0;
 166 }
 167
 168 static const char * const nb_err_rcode[] = {
 169         "bad request/response format",
 170         "NBNS server failure",
 171         "no such name",
 172         "unsupported request",
 173         "request rejected",
 174         "name already registered"
 175 };
 176
 177 static const char * const nb_err[] = {
 178         "host not found",
 179         "too many redirects",
 180         "invalid response",
 181         "NETBIOS name too long",
 182         "no interface to broadcast on and no NBNS server specified"
 183 };
 184
 185 const char *
 186 nb_strerror(int error)
 187 {
 188         if (error == 0)
 189                 return NULL;
 190         if (error <= NBERR_ACTIVE)
 191                 return nb_err_rcode[error - 1];
 192         else if (error >= NBERR_HOSTNOTFOUND && error < NBERR_MAX)
 193                 return nb_err[error - NBERR_HOSTNOTFOUND];
 194         else
 195                 return NULL;
 196 }
 197