2 * Copyright (c) 2000-2002, Boris Popov
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
32 * $Id: ctx.c,v 1.24 2002/04/13 14:35:28 bp Exp $
35 #include <sys/param.h>
36 #include <sys/sysctl.h>
37 #include <sys/ioctl.h>
39 #include <sys/mount.h>
49 #include <sys/iconv.h>
51 #define NB_NEEDRESOLVER
53 #include <netsmb/smb_lib.h>
54 #include <netsmb/netbios.h>
55 #include <netsmb/nb_lib.h>
56 #include <netsmb/smb_conn.h>
60 * Prescan command line for [-U user] argument
61 * and fill context with defaults
64 smb_ctx_init(struct smb_ctx *ctx, int argc, char *argv[],
65 int minlevel, int maxlevel, int sharetype)
72 bzero(ctx,sizeof(*ctx));
73 error = nb_ctx_create(&ctx->ct_nb);
77 ctx->ct_parsedlevel = SMBL_NONE;
78 ctx->ct_minlevel = minlevel;
79 ctx->ct_maxlevel = maxlevel;
80 ctx->ct_smbtcpport = SMB_TCP_PORT;
82 ctx->ct_ssn.ioc_opt = SMBVOPT_CREATE;
83 ctx->ct_ssn.ioc_timeout = 15;
84 ctx->ct_ssn.ioc_retrycount = 4;
85 ctx->ct_ssn.ioc_owner = SMBM_ANY_OWNER;
86 ctx->ct_ssn.ioc_group = SMBM_ANY_GROUP;
87 ctx->ct_ssn.ioc_mode = SMBM_EXEC;
88 ctx->ct_ssn.ioc_rights = SMBM_DEFAULT;
90 ctx->ct_sh.ioc_opt = SMBVOPT_CREATE;
91 ctx->ct_sh.ioc_owner = SMBM_ANY_OWNER;
92 ctx->ct_sh.ioc_group = SMBM_ANY_GROUP;
93 ctx->ct_sh.ioc_mode = SMBM_EXEC;
94 ctx->ct_sh.ioc_rights = SMBM_DEFAULT;
95 ctx->ct_sh.ioc_owner = SMBM_ANY_OWNER;
96 ctx->ct_sh.ioc_group = SMBM_ANY_GROUP;
98 nb_ctx_setscope(ctx->ct_nb, "");
100 if ((pwd = getpwuid(euid)) != NULL) {
101 smb_ctx_setuser(ctx, pwd->pw_name);
103 } else if (euid == 0)
104 smb_ctx_setuser(ctx, "root");
109 for (opt = 1; opt < argc; opt++) {
111 if (strncmp(cp, "//", 2) != 0)
113 error = smb_ctx_parseunc(ctx, cp, sharetype, (const char**)&cp);
116 ctx->ct_uncnext = cp;
119 while (error == 0 && (opt = cf_getopt(argc, argv, ":E:L:U:")) != -1) {
123 error = smb_ctx_setcharset(ctx, arg);
128 error = nls_setlocale(arg);
133 error = smb_ctx_setuser(ctx, arg);
137 cf_optind = cf_optreset = 1;
142 smb_ctx_done(struct smb_ctx *ctx)
144 if (ctx->ct_ssn.ioc_server)
145 nb_snbfree(ctx->ct_ssn.ioc_server);
146 if (ctx->ct_ssn.ioc_local)
147 nb_snbfree(ctx->ct_ssn.ioc_local);
149 free(ctx->ct_srvaddr);
151 nb_ctx_done(ctx->ct_nb);
155 getsubstring(const char *p, u_char sep, char *dest, int maxlen, const char **next)
160 for (len = 0; len < maxlen && *p != sep; p++, len++, dest++) {
166 *next = *p ? p + 1 : p;
171 * Here we expect something like "[proto:]//[user@]host[:psmb[:pnb]][/share][/path]"
174 smb_ctx_parseunc(struct smb_ctx *ctx, const char *unc, int sharetype,
178 char *p1, *psmb, *pnb;
182 ctx->ct_parsedlevel = SMBL_NONE;
183 if (*p++ != '/' || *p++ != '/') {
184 smb_error("UNC should start with '//'", 0);
188 error = getsubstring(p, '@', p1, sizeof(tmp), &p);
190 if (ctx->ct_maxlevel < SMBL_VC) {
191 smb_error("no user name required", 0);
194 error = smb_ctx_setuser(ctx, tmp);
197 ctx->ct_parsedlevel = SMBL_VC;
199 error = getsubstring(p, '/', p1, sizeof(tmp), &p);
201 error = getsubstring(p, '\0', p1, sizeof(tmp), &p);
203 smb_error("no server name found", 0);
208 smb_error("empty server name", 0);
212 * Check for port number specification.
214 psmb = strchr(tmp, ':');
217 pnb = strchr(psmb, ':');
220 error = smb_ctx_setnbport(ctx, atoi(pnb));
222 smb_error("Invalid NetBIOS port number", 0);
226 error = smb_ctx_setsmbport(ctx, atoi(psmb));
228 smb_error("Invalid SMB port number", 0);
232 error = smb_ctx_setserver(ctx, tmp);
235 if (sharetype == SMB_ST_NONE) {
239 if (*p != 0 && ctx->ct_maxlevel < SMBL_SHARE) {
240 smb_error("no share name required", 0);
243 error = getsubstring(p, '/', p1, sizeof(tmp), &p);
245 error = getsubstring(p, '\0', p1, sizeof(tmp), &p);
247 smb_error("unexpected end of line", 0);
251 if (*p1 == 0 && ctx->ct_minlevel >= SMBL_SHARE) {
252 smb_error("empty share name", 0);
258 error = smb_ctx_setshare(ctx, p1, sharetype);
263 smb_ctx_setcharset(struct smb_ctx *ctx, const char *arg)
265 char *cp, *servercs, *localcs;
266 int cslen = sizeof(ctx->ct_ssn.ioc_localcs);
267 int scslen, lcslen, error;
269 cp = strchr(arg, ':');
270 lcslen = cp ? (cp - arg) : 0;
271 if (lcslen == 0 || lcslen >= cslen) {
272 smb_error("invalid local charset specification (%s)", 0, arg);
275 scslen = (size_t)strlen(++cp);
276 if (scslen == 0 || scslen >= cslen) {
277 smb_error("invalid server charset specification (%s)", 0, arg);
280 localcs = memcpy(ctx->ct_ssn.ioc_localcs, arg, lcslen);
282 servercs = strcpy(ctx->ct_ssn.ioc_servercs, cp);
283 error = nls_setrecode(localcs, servercs);
286 smb_error("can't initialize iconv support (%s:%s)",
287 error, localcs, servercs);
294 smb_ctx_setserver(struct smb_ctx *ctx, const char *name)
296 if (strlen(name) > SMB_MAXSRVNAMELEN) {
297 smb_error("server name '%s' too long", 0, name);
300 nls_str_upper(ctx->ct_ssn.ioc_srvname, name);
305 smb_ctx_setnbport(struct smb_ctx *ctx, int port)
307 if (port < 1 || port > 0xffff)
309 ctx->ct_nb->nb_nmbtcpport = port;
314 smb_ctx_setsmbport(struct smb_ctx *ctx, int port)
316 if (port < 1 || port > 0xffff)
318 ctx->ct_smbtcpport = port;
319 ctx->ct_nb->nb_smbtcpport = port;
324 smb_ctx_setuser(struct smb_ctx *ctx, const char *name)
326 if (strlen(name) > SMB_MAXUSERNAMELEN) {
327 smb_error("user name '%s' too long", 0, name);
330 nls_str_upper(ctx->ct_ssn.ioc_user, name);
335 smb_ctx_setworkgroup(struct smb_ctx *ctx, const char *name)
337 if (strlen(name) > SMB_MAXUSERNAMELEN) {
338 smb_error("workgroup name '%s' too long", 0, name);
341 nls_str_upper(ctx->ct_ssn.ioc_workgroup, name);
346 smb_ctx_setpassword(struct smb_ctx *ctx, const char *passwd)
350 if (strlen(passwd) > SMB_MAXPASSWORDLEN) {
351 smb_error("password too long", 0);
354 if (strncmp(passwd, "$$1", 3) == 0)
355 smb_simpledecrypt(ctx->ct_ssn.ioc_password, passwd);
357 strcpy(ctx->ct_ssn.ioc_password, passwd);
358 strcpy(ctx->ct_sh.ioc_password, ctx->ct_ssn.ioc_password);
363 smb_ctx_setshare(struct smb_ctx *ctx, const char *share, int stype)
365 if (strlen(share) > SMB_MAXSHARENAMELEN) {
366 smb_error("share name '%s' too long", 0, share);
369 nls_str_upper(ctx->ct_sh.ioc_share, share);
371 ctx->ct_parsedlevel = SMBL_SHARE;
372 ctx->ct_sh.ioc_stype = stype;
377 smb_ctx_setsrvaddr(struct smb_ctx *ctx, const char *addr)
379 if (addr == NULL || addr[0] == 0)
382 free(ctx->ct_srvaddr);
383 if ((ctx->ct_srvaddr = strdup(addr)) == NULL)
389 smb_parse_owner(char *pair, uid_t *uid, gid_t *gid)
395 cp = strchr(pair, ':');
403 smb_error("Invalid group name %s, ignored",
412 smb_error("Invalid user name %s, ignored", 0, pair);
419 smb_ctx_opt(struct smb_ctx *ctx, int opt, const char *arg)
428 error = smb_ctx_setsrvaddr(ctx, arg);
431 ctx->ct_ssn.ioc_rights = strtol(arg, &cp, 8);
433 ctx->ct_sh.ioc_rights = strtol(cp + 1, &cp, 8);
434 ctx->ct_flags |= SMBCF_SRIGHTS;
438 ctx->ct_flags |= SMBCF_NOPWD;
445 error = smb_parse_owner(cp, &ctx->ct_sh.ioc_owner,
446 &ctx->ct_sh.ioc_group);
448 if (*p && error == 0) {
449 error = smb_parse_owner(p, &ctx->ct_ssn.ioc_owner,
450 &ctx->ct_ssn.ioc_group);
455 /* ctx->ct_ssn.ioc_opt |= SMBCOPT_PERMANENT;*/
458 ctx->ct_ssn.ioc_retrycount = atoi(arg);
461 ctx->ct_ssn.ioc_timeout = atoi(arg);
464 error = smb_ctx_setworkgroup(ctx, arg);
472 smb_hexdump(const u_char *buf, int len) {
477 printf("\n%02X: ", ofs);
478 printf("%02x ", *buf++);
487 smb_addiconvtbl(const char *to, const char *from, const u_char *tbl)
491 error = kiconv_add_xlat_table(to, from, tbl);
492 if (error && error != EEXIST) {
493 smb_error("can not setup kernel iconv table (%s:%s)", error,
501 * Verify context before connect operation(s),
502 * lookup specified server and try to fill all forgotten fields.
505 smb_ctx_resolve(struct smb_ctx *ctx)
507 struct smbioc_ossn *ssn = &ctx->ct_ssn;
508 struct smbioc_oshare *sh = &ctx->ct_sh;
510 struct sockaddr *sap;
511 struct sockaddr_nb *salocal, *saserver;
515 ctx->ct_flags &= ~SMBCF_RESOLVED;
516 if (ssn->ioc_srvname[0] == 0) {
517 smb_error("no server name specified", 0);
520 if (ctx->ct_minlevel >= SMBL_SHARE && sh->ioc_share[0] == 0) {
521 smb_error("no share name specified for %s@%s",
522 0, ssn->ioc_user, ssn->ioc_srvname);
525 error = nb_ctx_resolve(ctx->ct_nb);
528 if (ssn->ioc_localcs[0] == 0)
529 strcpy(ssn->ioc_localcs, "ISO8859-1");
530 error = smb_addiconvtbl("tolower", ssn->ioc_localcs, nls_lower);
533 error = smb_addiconvtbl("toupper", ssn->ioc_localcs, nls_upper);
536 if (ssn->ioc_servercs[0] != 0) {
537 error = kiconv_add_xlat16_cspairs
538 (ssn->ioc_servercs, ssn->ioc_localcs);
539 if (error) return error;
541 if (ctx->ct_srvaddr) {
542 error = nb_resolvehost_in(ctx->ct_srvaddr, &sap, ctx->ct_smbtcpport);
544 error = nbns_resolvename(ssn->ioc_srvname, ctx->ct_nb, &sap);
547 smb_error("can't get server address", error);
550 nn.nn_scope = ctx->ct_nb->nb_scope;
551 nn.nn_type = NBT_SERVER;
552 strcpy(nn.nn_name, ssn->ioc_srvname);
553 error = nb_sockaddr(sap, &nn, &saserver);
556 smb_error("can't allocate server address", error);
559 ssn->ioc_server = (struct sockaddr*)saserver;
560 if (ctx->ct_locname[0] == 0) {
561 error = nb_getlocalname(ctx->ct_locname);
563 smb_error("can't get local name", error);
566 nls_str_upper(ctx->ct_locname, ctx->ct_locname);
568 strcpy(nn.nn_name, ctx->ct_locname);
569 nn.nn_type = NBT_WKSTA;
570 nn.nn_scope = ctx->ct_nb->nb_scope;
571 error = nb_sockaddr(NULL, &nn, &salocal);
573 nb_snbfree((struct sockaddr*)saserver);
574 smb_error("can't allocate local address", error);
577 ssn->ioc_local = (struct sockaddr*)salocal;
578 ssn->ioc_lolen = salocal->snb_len;
579 ssn->ioc_svlen = saserver->snb_len;
580 if (ssn->ioc_password[0] == 0 && (ctx->ct_flags & SMBCF_NOPWD) == 0) {
581 cp = getpass("Password:");
582 error = smb_ctx_setpassword(ctx, cp);
586 ctx->ct_flags |= SMBCF_RESOLVED;
591 smb_ctx_gethandle(struct smb_ctx *ctx)
596 fd = open("/dev/"NSMB_NAME, O_RDWR);
605 smb_ctx_lookup(struct smb_ctx *ctx, int level, int flags)
607 struct smbioc_lookup rq;
610 if ((ctx->ct_flags & SMBCF_RESOLVED) == 0) {
611 smb_error("smb_ctx_lookup() data is not resolved", 0);
614 if (ctx->ct_fd != -1) {
618 error = smb_ctx_gethandle(ctx);
620 smb_error("can't get handle to requester (no /dev/"NSMB_NAME"* device)", 0);
623 bzero(&rq, sizeof(rq));
624 bcopy(&ctx->ct_ssn, &rq.ioc_ssn, sizeof(struct smbioc_ossn));
625 bcopy(&ctx->ct_sh, &rq.ioc_sh, sizeof(struct smbioc_oshare));
626 rq.ioc_flags = flags;
627 rq.ioc_level = level;
628 if (ioctl(ctx->ct_fd, SMBIOC_LOOKUP, &rq) == -1) {
630 if (flags & SMBLK_CREATE)
631 smb_error("unable to open connection", error);
638 smb_ctx_login(struct smb_ctx *ctx)
640 struct smbioc_ossn *ssn = &ctx->ct_ssn;
641 struct smbioc_oshare *sh = &ctx->ct_sh;
644 if ((ctx->ct_flags & SMBCF_RESOLVED) == 0) {
645 smb_error("smb_ctx_resolve() should be called first", 0);
648 if (ctx->ct_fd != -1) {
652 error = smb_ctx_gethandle(ctx);
654 smb_error("can't get handle to requester", 0);
657 if (ioctl(ctx->ct_fd, SMBIOC_OPENSESSION, ssn) == -1) {
659 smb_error("can't open session to server %s", error, ssn->ioc_srvname);
662 if (sh->ioc_share[0] == 0)
664 if (ioctl(ctx->ct_fd, SMBIOC_OPENSHARE, sh) == -1) {
666 smb_error("can't connect to share //%s/%s", error,
667 ssn->ioc_srvname, sh->ioc_share);
674 smb_ctx_setflags(struct smb_ctx *ctx, int level, int mask, int flags)
676 struct smbioc_flags fl;
678 if (ctx->ct_fd == -1)
680 fl.ioc_level = level;
682 fl.ioc_flags = flags;
683 if (ioctl(ctx->ct_fd, SMBIOC_SETFLAGS, &fl) == -1)
693 * 3 - server:user:share
696 smb_ctx_readrcsection(struct smb_ctx *ctx, const char *sname, int level)
702 rc_getstringptr(smb_rc, sname, "charsets", &p);
704 error = smb_ctx_setcharset(ctx, p);
706 smb_error("charset specification in the section '%s' ignored", error, sname);
710 rc_getint(smb_rc, sname, "timeout", &ctx->ct_ssn.ioc_timeout);
711 rc_getint(smb_rc, sname, "retry_count", &ctx->ct_ssn.ioc_retrycount);
714 rc_getstringptr(smb_rc, sname, "addr", &p);
716 error = smb_ctx_setsrvaddr(ctx, p);
718 smb_error("invalid address specified in the section %s", 0, sname);
724 rc_getstringptr(smb_rc, sname, "password", &p);
726 smb_ctx_setpassword(ctx, p);
728 rc_getstringptr(smb_rc, sname, "workgroup", &p);
730 smb_ctx_setworkgroup(ctx, p);
735 * read rc file as follows:
736 * 1. read [default] section
737 * 2. override with [server] section
738 * 3. override with [server:user:share] section
739 * Since abcence of rcfile is not fatal, silently ignore this fact.
740 * smb_rc file should be closed by caller.
743 smb_ctx_readrc(struct smb_ctx *ctx)
745 char sname[SMB_MAXSRVNAMELEN + SMB_MAXUSERNAMELEN + SMB_MAXSHARENAMELEN + 4];
748 if (smb_open_rcfile() != 0)
751 if (ctx->ct_ssn.ioc_user[0] == 0 || ctx->ct_ssn.ioc_srvname[0] == 0)
754 smb_ctx_readrcsection(ctx, "default", 0);
755 nb_ctx_readrcsection(smb_rc, ctx->ct_nb, "default", 0);
756 smb_ctx_readrcsection(ctx, ctx->ct_ssn.ioc_srvname, 1);
757 nb_ctx_readrcsection(smb_rc, ctx->ct_nb, ctx->ct_ssn.ioc_srvname, 1);
759 * SERVER:USER parameters
761 snprintf(sname, sizeof(sname), "%s:%s", ctx->ct_ssn.ioc_srvname,
762 ctx->ct_ssn.ioc_user);
763 smb_ctx_readrcsection(ctx, sname, 2);
765 if (ctx->ct_sh.ioc_share[0] != 0) {
767 * SERVER:USER:SHARE parameters
769 snprintf(sname, sizeof(sname), "%s:%s:%s", ctx->ct_ssn.ioc_srvname,
770 ctx->ct_ssn.ioc_user, ctx->ct_sh.ioc_share);
771 smb_ctx_readrcsection(ctx, sname, 3);