The Design and Implementation of the FreeBSD Operating System, Second Edition
Now available: The Design and Implementation of the FreeBSD Operating System (Second Edition)


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]

FreeBSD/Linux Kernel Cross Reference
sys/netsmb/smb_conn.c

Version: -  FREEBSD  -  FREEBSD-13-STABLE  -  FREEBSD-13-0  -  FREEBSD-12-STABLE  -  FREEBSD-12-0  -  FREEBSD-11-STABLE  -  FREEBSD-11-0  -  FREEBSD-10-STABLE  -  FREEBSD-10-0  -  FREEBSD-9-STABLE  -  FREEBSD-9-0  -  FREEBSD-8-STABLE  -  FREEBSD-8-0  -  FREEBSD-7-STABLE  -  FREEBSD-7-0  -  FREEBSD-6-STABLE  -  FREEBSD-6-0  -  FREEBSD-5-STABLE  -  FREEBSD-5-0  -  FREEBSD-4-STABLE  -  FREEBSD-3-STABLE  -  FREEBSD22  -  l41  -  OPENBSD  -  linux-2.6  -  MK84  -  PLAN9  -  xnu-8792 
SearchContext: -  none  -  3  -  10 

    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  *
   14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
   15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
   18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   24  * SUCH DAMAGE.
   25  */
   26 
   27 /*
   28  * Connection engine.
   29  */
   30 
   31 #include <sys/cdefs.h>
   32 __FBSDID("$FreeBSD: releng/9.0/sys/netsmb/smb_conn.c 206361 2010-04-07 16:50:38Z joel $");
   33 
   34 #include <sys/param.h>
   35 #include <sys/systm.h>
   36 #include <sys/kernel.h>
   37 #include <sys/malloc.h>
   38 #include <sys/priv.h>
   39 #include <sys/proc.h>
   40 #include <sys/lock.h>
   41 #include <sys/sysctl.h>
   42 #include <sys/socketvar.h>
   43 
   44 #include <sys/iconv.h>
   45 
   46 #include <netsmb/smb.h>
   47 #include <netsmb/smb_subr.h>
   48 #include <netsmb/smb_conn.h>
   49 #include <netsmb/smb_tran.h>
   50 #include <netsmb/smb_trantcp.h>
   51 
   52 static struct smb_connobj smb_vclist;
   53 static int smb_vcnext = 1;      /* next unique id for VC */
   54 
   55 SYSCTL_NODE(_net, OID_AUTO, smb, CTLFLAG_RW, NULL, "SMB protocol");
   56 
   57 MALLOC_DEFINE(M_SMBCONN, "smb_conn", "SMB connection");
   58 
   59 static void smb_co_init(struct smb_connobj *cp, int level, char *ilockname,
   60     char *lockname);
   61 static void smb_co_done(struct smb_connobj *cp);
   62 static int  smb_co_lockstatus(struct smb_connobj *cp);
   63 
   64 static int  smb_vc_disconnect(struct smb_vc *vcp);
   65 static void smb_vc_free(struct smb_connobj *cp);
   66 static void smb_vc_gone(struct smb_connobj *cp, struct smb_cred *scred);
   67 static smb_co_free_t smb_share_free;
   68 static smb_co_gone_t smb_share_gone;
   69 
   70 static int  smb_sysctl_treedump(SYSCTL_HANDLER_ARGS);
   71 
   72 SYSCTL_PROC(_net_smb, OID_AUTO, treedump, CTLFLAG_RD | CTLTYPE_OPAQUE,
   73             NULL, 0, smb_sysctl_treedump, "S,treedump", "Requester tree");
   74 
   75 int
   76 smb_sm_init(void)
   77 {
   78 
   79         smb_co_init(&smb_vclist, SMBL_SM, "smbsm ilock", "smbsm");
   80         smb_co_unlock(&smb_vclist, 0);
   81         return 0;
   82 }
   83 
   84 int
   85 smb_sm_done(void)
   86 {
   87 
   88         /* XXX: hold the mutex */
   89         if (smb_vclist.co_usecount > 1) {
   90                 SMBERROR("%d connections still active\n", smb_vclist.co_usecount - 1);
   91                 return EBUSY;
   92         }
   93         lockmgr(&smb_vclist.co_lock, LK_DRAIN, NULL);
   94         smb_co_done(&smb_vclist);
   95         return 0;
   96 }
   97 
   98 static int
   99 smb_sm_lockvclist(int flags)
  100 {
  101 
  102         return smb_co_lock(&smb_vclist, flags | LK_CANRECURSE);
  103 }
  104 
  105 static void
  106 smb_sm_unlockvclist(void)
  107 {
  108 
  109         smb_co_unlock(&smb_vclist, LK_RELEASE);
  110 }
  111 
  112 static int
  113 smb_sm_lookupint(struct smb_vcspec *vcspec, struct smb_sharespec *shspec,
  114         struct smb_cred *scred, struct smb_vc **vcpp)
  115 {
  116         struct smb_connobj *scp;
  117         struct smb_vc *vcp;
  118         int exact = 1;
  119         int error;
  120 
  121         vcspec->shspec = shspec;
  122         error = ENOENT;
  123         vcp = NULL;
  124         SMBCO_FOREACH(scp, &smb_vclist) {
  125                 vcp = (struct smb_vc *)scp;
  126                 error = smb_vc_lock(vcp, LK_EXCLUSIVE);
  127                 if (error)
  128                         continue;
  129 
  130                 if ((vcp->obj.co_flags & SMBV_PRIVATE) ||
  131                     !CONNADDREQ(vcp->vc_paddr, vcspec->sap) ||
  132                     strcmp(vcp->vc_username, vcspec->username) != 0)
  133                         goto err1;
  134                 if (vcspec->owner != SMBM_ANY_OWNER) {
  135                         if (vcp->vc_uid != vcspec->owner)
  136                                 goto err1;
  137                 } else
  138                         exact = 0;
  139                 if (vcspec->group != SMBM_ANY_GROUP) {
  140                         if (vcp->vc_grp != vcspec->group)
  141                                 goto err1;
  142                 } else
  143                         exact = 0;
  144                 if (vcspec->mode & SMBM_EXACT) {
  145                         if (!exact || (vcspec->mode & SMBM_MASK) !=
  146                             vcp->vc_mode)
  147                                 goto err1;
  148                 }
  149                 if (smb_vc_access(vcp, scred, vcspec->mode) != 0)
  150                         goto err1;
  151                 vcspec->ssp = NULL;
  152                 if (shspec) {
  153                         error = (int)smb_vc_lookupshare(vcp, shspec, scred,
  154                             &vcspec->ssp);
  155                         if (error)
  156                                 goto fail;
  157                 }
  158                 error = 0;
  159                 break;
  160         err1:
  161                 error = 1;
  162         fail:
  163                 smb_vc_unlock(vcp, 0);
  164         }
  165         if (vcp) {
  166                 smb_vc_ref(vcp);
  167                 *vcpp = vcp;
  168         }
  169         return (error);
  170 }
  171 
  172 int
  173 smb_sm_lookup(struct smb_vcspec *vcspec, struct smb_sharespec *shspec,
  174         struct smb_cred *scred, struct smb_vc **vcpp)
  175 {
  176         struct smb_vc *vcp;
  177         struct smb_share *ssp = NULL;
  178         int error;
  179 
  180         *vcpp = vcp = NULL;
  181 
  182         error = smb_sm_lockvclist(LK_EXCLUSIVE);
  183         if (error)
  184                 return error;
  185         error = smb_sm_lookupint(vcspec, shspec, scred, vcpp);
  186         if (error == 0 || (vcspec->flags & SMBV_CREATE) == 0) {
  187                 smb_sm_unlockvclist();
  188                 return error;
  189         }
  190         error = smb_sm_lookupint(vcspec, NULL, scred, &vcp);
  191         if (error) {
  192                 error = smb_vc_create(vcspec, scred, &vcp);
  193                 if (error)
  194                         goto out;
  195                 error = smb_vc_connect(vcp, scred);
  196                 if (error)
  197                         goto out;
  198         }
  199         if (shspec == NULL)
  200                 goto out;
  201         error = smb_share_create(vcp, shspec, scred, &ssp);
  202         if (error)
  203                 goto out;
  204         error = smb_smb_treeconnect(ssp, scred);
  205         if (error == 0)
  206                 vcspec->ssp = ssp;
  207         else
  208                 smb_share_put(ssp, scred);
  209 out:
  210         smb_sm_unlockvclist();
  211         if (error == 0)
  212                 *vcpp = vcp;
  213         else if (vcp) {
  214                 smb_vc_lock(vcp, LK_EXCLUSIVE);
  215                 smb_vc_put(vcp, scred);
  216         }
  217         return error;
  218 }
  219 
  220 /*
  221  * Common code for connection object
  222  */
  223 static void
  224 smb_co_init(struct smb_connobj *cp, int level, char *ilockname, char *lockname)
  225 {
  226         SLIST_INIT(&cp->co_children);
  227         smb_sl_init(&cp->co_interlock, ilockname);
  228         lockinit(&cp->co_lock, PZERO, lockname, 0, 0);
  229         cp->co_level = level;
  230         cp->co_usecount = 1;
  231         if (smb_co_lock(cp, LK_EXCLUSIVE) != 0)
  232             panic("smb_co_init: lock failed");
  233 }
  234 
  235 static void
  236 smb_co_done(struct smb_connobj *cp)
  237 {
  238         smb_sl_destroy(&cp->co_interlock);
  239         lockmgr(&cp->co_lock, LK_RELEASE, NULL);
  240         lockdestroy(&cp->co_lock);
  241 }
  242 
  243 static void
  244 smb_co_gone(struct smb_connobj *cp, struct smb_cred *scred)
  245 {
  246         struct smb_connobj *parent;
  247 
  248         if (cp->co_gone)
  249                 cp->co_gone(cp, scred);
  250         parent = cp->co_parent;
  251         if (parent) {
  252                 smb_co_lock(parent, LK_EXCLUSIVE);
  253                 SLIST_REMOVE(&parent->co_children, cp, smb_connobj, co_next);
  254                 smb_co_put(parent, scred);
  255         }
  256         if (cp->co_free)
  257                 cp->co_free(cp);
  258 }
  259 
  260 void
  261 smb_co_ref(struct smb_connobj *cp)
  262 {
  263 
  264         SMB_CO_LOCK(cp);
  265         cp->co_usecount++;
  266         SMB_CO_UNLOCK(cp);
  267 }
  268 
  269 void
  270 smb_co_rele(struct smb_connobj *cp, struct smb_cred *scred)
  271 {
  272 
  273         SMB_CO_LOCK(cp);
  274         if (cp->co_usecount > 1) {
  275                 cp->co_usecount--;
  276                 SMB_CO_UNLOCK(cp);
  277                 return;
  278         }
  279         if (cp->co_usecount == 0) {
  280                 SMBERROR("negative use_count for object %d", cp->co_level);
  281                 SMB_CO_UNLOCK(cp);
  282                 return;
  283         }
  284         cp->co_usecount--;
  285         cp->co_flags |= SMBO_GONE;
  286 
  287         lockmgr(&cp->co_lock, LK_DRAIN | LK_INTERLOCK, &cp->co_interlock);
  288         smb_co_gone(cp, scred);
  289 }
  290 
  291 int
  292 smb_co_get(struct smb_connobj *cp, int flags, struct smb_cred *scred)
  293 {
  294         int error;
  295 
  296         if ((flags & LK_INTERLOCK) == 0)
  297                 SMB_CO_LOCK(cp);
  298         cp->co_usecount++;
  299         error = smb_co_lock(cp, flags | LK_INTERLOCK);
  300         if (error) {
  301                 SMB_CO_LOCK(cp);
  302                 cp->co_usecount--;
  303                 SMB_CO_UNLOCK(cp);
  304                 return error;
  305         }
  306         return 0;
  307 }
  308 
  309 void
  310 smb_co_put(struct smb_connobj *cp, struct smb_cred *scred)
  311 {
  312 
  313         SMB_CO_LOCK(cp);
  314         if (cp->co_usecount > 1) {
  315                 cp->co_usecount--;
  316         } else if (cp->co_usecount == 1) {
  317                 cp->co_usecount--;
  318                 cp->co_flags |= SMBO_GONE;
  319         } else {
  320                 SMBERROR("negative usecount");
  321         }
  322         lockmgr(&cp->co_lock, LK_RELEASE | LK_INTERLOCK, &cp->co_interlock);
  323         if ((cp->co_flags & SMBO_GONE) == 0)
  324                 return;
  325         lockmgr(&cp->co_lock, LK_DRAIN, NULL);
  326         smb_co_gone(cp, scred);
  327 }
  328 
  329 int
  330 smb_co_lockstatus(struct smb_connobj *cp)
  331 {
  332         return lockstatus(&cp->co_lock);
  333 }
  334 
  335 int
  336 smb_co_lock(struct smb_connobj *cp, int flags)
  337 {
  338 
  339         if (cp->co_flags & SMBO_GONE)
  340                 return EINVAL;
  341         if ((flags & LK_TYPE_MASK) == 0)
  342                 flags |= LK_EXCLUSIVE;
  343         if (smb_co_lockstatus(cp) == LK_EXCLUSIVE && 
  344             (flags & LK_CANRECURSE) == 0) {
  345                 SMBERROR("recursive lock for object %d\n", cp->co_level);
  346                 return 0;
  347         }
  348         return lockmgr(&cp->co_lock, flags, &cp->co_interlock);
  349 }
  350 
  351 void
  352 smb_co_unlock(struct smb_connobj *cp, int flags)
  353 {
  354         (void)lockmgr(&cp->co_lock, flags | LK_RELEASE, &cp->co_interlock);
  355 }
  356 
  357 static void
  358 smb_co_addchild(struct smb_connobj *parent, struct smb_connobj *child)
  359 {
  360 
  361         KASSERT(smb_co_lockstatus(parent) == LK_EXCLUSIVE,
  362             ("smb_co_addchild: parent not locked"));
  363         KASSERT(smb_co_lockstatus(child) == LK_EXCLUSIVE,
  364             ("smb_co_addchild: child not locked"));
  365 
  366         smb_co_ref(parent);
  367         SLIST_INSERT_HEAD(&parent->co_children, child, co_next);
  368         child->co_parent = parent;
  369 }
  370 
  371 /*
  372  * Session implementation
  373  */
  374 
  375 int
  376 smb_vc_create(struct smb_vcspec *vcspec,
  377         struct smb_cred *scred, struct smb_vc **vcpp)
  378 {
  379         struct smb_vc *vcp;
  380         struct ucred *cred = scred->scr_cred;
  381         uid_t uid = vcspec->owner;
  382         gid_t gid = vcspec->group;
  383         uid_t realuid = cred->cr_uid;
  384         char *domain = vcspec->domain;
  385         int error, isroot;
  386 
  387         isroot = smb_suser(cred) == 0;
  388         /*
  389          * Only superuser can create VCs with different uid and gid
  390          */
  391         if (uid != SMBM_ANY_OWNER && uid != realuid && !isroot)
  392                 return EPERM;
  393         if (gid != SMBM_ANY_GROUP && !groupmember(gid, cred) && !isroot)
  394                 return EPERM;
  395 
  396         vcp = smb_zmalloc(sizeof(*vcp), M_SMBCONN, M_WAITOK);
  397         smb_co_init(VCTOCP(vcp), SMBL_VC, "smb_vc ilock", "smb_vc");
  398         vcp->obj.co_free = smb_vc_free;
  399         vcp->obj.co_gone = smb_vc_gone;
  400         vcp->vc_number = smb_vcnext++;
  401         vcp->vc_timo = SMB_DEFRQTIMO;
  402         vcp->vc_smbuid = SMB_UID_UNKNOWN;
  403         vcp->vc_mode = vcspec->rights & SMBM_MASK;
  404         vcp->obj.co_flags = vcspec->flags & (SMBV_PRIVATE | SMBV_SINGLESHARE);
  405         vcp->vc_tdesc = &smb_tran_nbtcp_desc;
  406         vcp->vc_seqno = 0;
  407         vcp->vc_mackey = NULL;
  408         vcp->vc_mackeylen = 0;
  409 
  410         if (uid == SMBM_ANY_OWNER)
  411                 uid = realuid;
  412         if (gid == SMBM_ANY_GROUP)
  413                 gid = cred->cr_groups[0];
  414         vcp->vc_uid = uid;
  415         vcp->vc_grp = gid;
  416 
  417         smb_sl_init(&vcp->vc_stlock, "vcstlock");
  418         error = ENOMEM;
  419 
  420         vcp->vc_paddr = sodupsockaddr(vcspec->sap, M_WAITOK);
  421         if (vcp->vc_paddr == NULL)
  422                 goto fail;
  423         vcp->vc_laddr = sodupsockaddr(vcspec->lap, M_WAITOK);
  424         if (vcp->vc_laddr == NULL)
  425                 goto fail;
  426         vcp->vc_pass = smb_strdup(vcspec->pass);
  427         if (vcp->vc_pass == NULL)
  428                 goto fail;
  429         vcp->vc_domain = smb_strdup((domain && domain[0]) ? domain :
  430             "NODOMAIN");
  431         if (vcp->vc_domain == NULL)
  432                 goto fail;
  433         vcp->vc_srvname = smb_strdup(vcspec->srvname);
  434         if (vcp->vc_srvname == NULL)
  435                 goto fail;
  436         vcp->vc_username = smb_strdup(vcspec->username);
  437         if (vcp->vc_username == NULL)
  438                 goto fail;
  439         error = (int)iconv_open("tolower", vcspec->localcs, &vcp->vc_tolower);
  440         if (error)
  441                 goto fail;
  442         error = (int)iconv_open("toupper", vcspec->localcs, &vcp->vc_toupper);
  443         if (error)
  444                 goto fail;
  445         if (vcspec->servercs[0]) {
  446                 error = (int)iconv_open(vcspec->servercs, vcspec->localcs,
  447                     &vcp->vc_toserver);
  448                 if (error)
  449                         goto fail;
  450                 error = (int)iconv_open(vcspec->localcs, vcspec->servercs,
  451                     &vcp->vc_tolocal);
  452                 if (error)
  453                         goto fail;
  454         }
  455         error = (int)smb_iod_create(vcp);
  456         if (error)
  457                 goto fail;
  458         *vcpp = vcp;
  459         smb_co_addchild(&smb_vclist, VCTOCP(vcp));
  460         return (0);
  461 
  462  fail:
  463         smb_vc_put(vcp, scred);
  464         return (error);
  465 }
  466 
  467 static void
  468 smb_vc_free(struct smb_connobj *cp)
  469 {
  470         struct smb_vc *vcp = CPTOVC(cp);
  471 
  472         if (vcp->vc_iod)
  473                 smb_iod_destroy(vcp->vc_iod);
  474         SMB_STRFREE(vcp->vc_username);
  475         SMB_STRFREE(vcp->vc_srvname);
  476         SMB_STRFREE(vcp->vc_pass);
  477         SMB_STRFREE(vcp->vc_domain);
  478         if (vcp->vc_mackey)
  479                 free(vcp->vc_mackey, M_SMBTEMP);
  480         if (vcp->vc_paddr)
  481                 free(vcp->vc_paddr, M_SONAME);
  482         if (vcp->vc_laddr)
  483                 free(vcp->vc_laddr, M_SONAME);
  484         if (vcp->vc_tolower)
  485                 iconv_close(vcp->vc_tolower);
  486         if (vcp->vc_toupper)
  487                 iconv_close(vcp->vc_toupper);
  488         if (vcp->vc_tolocal)
  489                 iconv_close(vcp->vc_tolocal);
  490         if (vcp->vc_toserver)
  491                 iconv_close(vcp->vc_toserver);
  492         smb_co_done(VCTOCP(vcp));
  493         smb_sl_destroy(&vcp->vc_stlock);
  494         free(vcp, M_SMBCONN);
  495 }
  496 
  497 /*
  498  * Called when use count of VC dropped to zero.
  499  * VC should be locked on enter with LK_DRAIN.
  500  */
  501 static void
  502 smb_vc_gone(struct smb_connobj *cp, struct smb_cred *scred)
  503 {
  504         struct smb_vc *vcp = CPTOVC(cp);
  505 
  506         smb_vc_disconnect(vcp);
  507 }
  508 
  509 void
  510 smb_vc_ref(struct smb_vc *vcp)
  511 {
  512         smb_co_ref(VCTOCP(vcp));
  513 }
  514 
  515 void
  516 smb_vc_rele(struct smb_vc *vcp, struct smb_cred *scred)
  517 {
  518         smb_co_rele(VCTOCP(vcp), scred);
  519 }
  520 
  521 int
  522 smb_vc_get(struct smb_vc *vcp, int flags, struct smb_cred *scred)
  523 {
  524         return smb_co_get(VCTOCP(vcp), flags, scred);
  525 }
  526 
  527 void
  528 smb_vc_put(struct smb_vc *vcp, struct smb_cred *scred)
  529 {
  530         smb_co_put(VCTOCP(vcp), scred);
  531 }
  532 
  533 int
  534 smb_vc_lock(struct smb_vc *vcp, int flags)
  535 {
  536         return smb_co_lock(VCTOCP(vcp), flags);
  537 }
  538 
  539 void
  540 smb_vc_unlock(struct smb_vc *vcp, int flags)
  541 {
  542         smb_co_unlock(VCTOCP(vcp), flags);
  543 }
  544 
  545 int
  546 smb_vc_access(struct smb_vc *vcp, struct smb_cred *scred, mode_t mode)
  547 {
  548         struct ucred *cred = scred->scr_cred;
  549 
  550         if (smb_suser(cred) == 0 || cred->cr_uid == vcp->vc_uid)
  551                 return 0;
  552         mode >>= 3;
  553         if (!groupmember(vcp->vc_grp, cred))
  554                 mode >>= 3;
  555         return (vcp->vc_mode & mode) == mode ? 0 : EACCES;
  556 }
  557 
  558 static int
  559 smb_vc_cmpshare(struct smb_share *ssp, struct smb_sharespec *dp)
  560 {
  561         int exact = 1;
  562 
  563         if (strcmp(ssp->ss_name, dp->name) != 0)
  564                 return 1;
  565         if (dp->owner != SMBM_ANY_OWNER) {
  566                 if (ssp->ss_uid != dp->owner)
  567                         return 1;
  568         } else
  569                 exact = 0;
  570         if (dp->group != SMBM_ANY_GROUP) {
  571                 if (ssp->ss_grp != dp->group)
  572                         return 1;
  573         } else
  574                 exact = 0;
  575 
  576         if (dp->mode & SMBM_EXACT) {
  577                 if (!exact)
  578                         return 1;
  579                 return (dp->mode & SMBM_MASK) == ssp->ss_mode ? 0 : 1;
  580         }
  581         if (smb_share_access(ssp, dp->scred, dp->mode) != 0)
  582                 return 1;
  583         return 0;
  584 }
  585 
  586 /*
  587  * Lookup share in the given VC. Share referenced and locked on return.
  588  * VC expected to be locked on entry and will be left locked on exit.
  589  */
  590 int
  591 smb_vc_lookupshare(struct smb_vc *vcp, struct smb_sharespec *dp,
  592         struct smb_cred *scred, struct smb_share **sspp)
  593 {
  594         struct smb_connobj *scp = NULL;
  595         struct smb_share *ssp = NULL;
  596         int error;
  597 
  598         *sspp = NULL;
  599         dp->scred = scred;
  600         SMBCO_FOREACH(scp, VCTOCP(vcp)) {
  601                 ssp = (struct smb_share *)scp;
  602                 error = smb_share_lock(ssp, LK_EXCLUSIVE);
  603                 if (error)
  604                         continue;
  605                 if (smb_vc_cmpshare(ssp, dp) == 0)
  606                         break;
  607                 smb_share_unlock(ssp, 0);
  608         }
  609         if (ssp) {
  610                 smb_share_ref(ssp);
  611                 *sspp = ssp;
  612                 error = 0;
  613         } else
  614                 error = ENOENT;
  615         return error;
  616 }
  617 
  618 int
  619 smb_vc_connect(struct smb_vc *vcp, struct smb_cred *scred)
  620 {
  621 
  622         return smb_iod_request(vcp->vc_iod, SMBIOD_EV_CONNECT | SMBIOD_EV_SYNC, NULL);
  623 }
  624 
  625 /*
  626  * Destroy VC to server, invalidate shares linked with it.
  627  * Transport should be locked on entry.
  628  */
  629 int
  630 smb_vc_disconnect(struct smb_vc *vcp)
  631 {
  632 
  633         smb_iod_request(vcp->vc_iod, SMBIOD_EV_DISCONNECT | SMBIOD_EV_SYNC, NULL);
  634         return 0;
  635 }
  636 
  637 static char smb_emptypass[] = "";
  638 
  639 const char *
  640 smb_vc_getpass(struct smb_vc *vcp)
  641 {
  642         if (vcp->vc_pass)
  643                 return vcp->vc_pass;
  644         return smb_emptypass;
  645 }
  646 
  647 static int
  648 smb_vc_getinfo(struct smb_vc *vcp, struct smb_vc_info *vip)
  649 {
  650         bzero(vip, sizeof(struct smb_vc_info));
  651         vip->itype = SMB_INFO_VC;
  652         vip->usecount = vcp->obj.co_usecount;
  653         vip->uid = vcp->vc_uid;
  654         vip->gid = vcp->vc_grp;
  655         vip->mode = vcp->vc_mode;
  656         vip->flags = vcp->obj.co_flags;
  657         vip->sopt = vcp->vc_sopt;
  658         vip->iodstate = vcp->vc_iod->iod_state;
  659         bzero(&vip->sopt.sv_skey, sizeof(vip->sopt.sv_skey));
  660         snprintf(vip->srvname, sizeof(vip->srvname), "%s", vcp->vc_srvname);
  661         snprintf(vip->vcname, sizeof(vip->vcname), "%s", vcp->vc_username);
  662         return 0;
  663 }
  664 
  665 u_short
  666 smb_vc_nextmid(struct smb_vc *vcp)
  667 {
  668         u_short r;
  669 
  670         SMB_CO_LOCK(&vcp->obj);
  671         r = vcp->vc_mid++;
  672         SMB_CO_UNLOCK(&vcp->obj);
  673         return r;
  674 }
  675 
  676 /*
  677  * Share implementation
  678  */
  679 /*
  680  * Allocate share structure and attach it to the given VC
  681  * Connection expected to be locked on entry. Share will be returned
  682  * in locked state.
  683  */
  684 int
  685 smb_share_create(struct smb_vc *vcp, struct smb_sharespec *shspec,
  686         struct smb_cred *scred, struct smb_share **sspp)
  687 {
  688         struct smb_share *ssp;
  689         struct ucred *cred = scred->scr_cred;
  690         uid_t realuid = cred->cr_uid;
  691         uid_t uid = shspec->owner;
  692         gid_t gid = shspec->group;
  693         int error, isroot;
  694 
  695         isroot = smb_suser(cred) == 0;
  696         /*
  697          * Only superuser can create shares with different uid and gid
  698          */
  699         if (uid != SMBM_ANY_OWNER && uid != realuid && !isroot)
  700                 return EPERM;
  701         if (gid != SMBM_ANY_GROUP && !groupmember(gid, cred) && !isroot)
  702                 return EPERM;
  703         error = smb_vc_lookupshare(vcp, shspec, scred, &ssp);
  704         if (!error) {
  705                 smb_share_put(ssp, scred);
  706                 return EEXIST;
  707         }
  708         if (uid == SMBM_ANY_OWNER)
  709                 uid = realuid;
  710         if (gid == SMBM_ANY_GROUP)
  711                 gid = cred->cr_groups[0];
  712         ssp = smb_zmalloc(sizeof(*ssp), M_SMBCONN, M_WAITOK);
  713         smb_co_init(SSTOCP(ssp), SMBL_SHARE, "smbss ilock", "smbss");
  714         ssp->obj.co_free = smb_share_free;
  715         ssp->obj.co_gone = smb_share_gone;
  716         smb_sl_init(&ssp->ss_stlock, "ssstlock");
  717         ssp->ss_name = smb_strdup(shspec->name);
  718         if (shspec->pass && shspec->pass[0])
  719                 ssp->ss_pass = smb_strdup(shspec->pass);
  720         ssp->ss_type = shspec->stype;
  721         ssp->ss_tid = SMB_TID_UNKNOWN;
  722         ssp->ss_uid = uid;
  723         ssp->ss_grp = gid;
  724         ssp->ss_mode = shspec->rights & SMBM_MASK;
  725         smb_co_addchild(VCTOCP(vcp), SSTOCP(ssp));
  726         *sspp = ssp;
  727         return 0;
  728 }
  729 
  730 static void
  731 smb_share_free(struct smb_connobj *cp)
  732 {
  733         struct smb_share *ssp = CPTOSS(cp);
  734 
  735         SMB_STRFREE(ssp->ss_name);
  736         SMB_STRFREE(ssp->ss_pass);
  737         smb_sl_destroy(&ssp->ss_stlock);
  738         smb_co_done(SSTOCP(ssp));
  739         free(ssp, M_SMBCONN);
  740 }
  741 
  742 static void
  743 smb_share_gone(struct smb_connobj *cp, struct smb_cred *scred)
  744 {
  745         struct smb_share *ssp = CPTOSS(cp);
  746 
  747         smb_smb_treedisconnect(ssp, scred);
  748 }
  749 
  750 void
  751 smb_share_ref(struct smb_share *ssp)
  752 {
  753         smb_co_ref(SSTOCP(ssp));
  754 }
  755 
  756 void
  757 smb_share_rele(struct smb_share *ssp, struct smb_cred *scred)
  758 {
  759         smb_co_rele(SSTOCP(ssp), scred);
  760 }
  761 
  762 int
  763 smb_share_get(struct smb_share *ssp, int flags, struct smb_cred *scred)
  764 {
  765         return smb_co_get(SSTOCP(ssp), flags, scred);
  766 }
  767 
  768 void
  769 smb_share_put(struct smb_share *ssp, struct smb_cred *scred)
  770 {
  771         smb_co_put(SSTOCP(ssp), scred);
  772 }
  773 
  774 int
  775 smb_share_lock(struct smb_share *ssp, int flags)
  776 {
  777         return smb_co_lock(SSTOCP(ssp), flags);
  778 }
  779 
  780 void
  781 smb_share_unlock(struct smb_share *ssp, int flags)
  782 {
  783         smb_co_unlock(SSTOCP(ssp), flags);
  784 }
  785 
  786 int
  787 smb_share_access(struct smb_share *ssp, struct smb_cred *scred, mode_t mode)
  788 {
  789         struct ucred *cred = scred->scr_cred;
  790 
  791         if (smb_suser(cred) == 0 || cred->cr_uid == ssp->ss_uid)
  792                 return 0;
  793         mode >>= 3;
  794         if (!groupmember(ssp->ss_grp, cred))
  795                 mode >>= 3;
  796         return (ssp->ss_mode & mode) == mode ? 0 : EACCES;
  797 }
  798 
  799 void
  800 smb_share_invalidate(struct smb_share *ssp)
  801 {
  802         ssp->ss_tid = SMB_TID_UNKNOWN;
  803 }
  804 
  805 int
  806 smb_share_valid(struct smb_share *ssp)
  807 {
  808         return ssp->ss_tid != SMB_TID_UNKNOWN &&
  809             ssp->ss_vcgenid == SSTOVC(ssp)->vc_genid;
  810 }
  811 
  812 const char*
  813 smb_share_getpass(struct smb_share *ssp)
  814 {
  815         struct smb_vc *vcp;
  816 
  817         if (ssp->ss_pass)
  818                 return ssp->ss_pass;
  819         vcp = SSTOVC(ssp);
  820         if (vcp->vc_pass)
  821                 return vcp->vc_pass;
  822         return smb_emptypass;
  823 }
  824 
  825 static int
  826 smb_share_getinfo(struct smb_share *ssp, struct smb_share_info *sip)
  827 {
  828         bzero(sip, sizeof(struct smb_share_info));
  829         sip->itype = SMB_INFO_SHARE;
  830         sip->usecount = ssp->obj.co_usecount;
  831         sip->tid  = ssp->ss_tid;
  832         sip->type= ssp->ss_type;
  833         sip->uid = ssp->ss_uid;
  834         sip->gid = ssp->ss_grp;
  835         sip->mode= ssp->ss_mode;
  836         sip->flags = ssp->obj.co_flags;
  837         snprintf(sip->sname, sizeof(sip->sname), "%s", ssp->ss_name);
  838         return 0;
  839 }
  840 
  841 /*
  842  * Dump an entire tree into sysctl call
  843  */
  844 static int
  845 smb_sysctl_treedump(SYSCTL_HANDLER_ARGS)
  846 {
  847         struct thread *td = req->td;
  848         struct smb_cred scred;
  849         struct smb_connobj *scp1, *scp2;
  850         struct smb_vc *vcp;
  851         struct smb_share *ssp;
  852         struct smb_vc_info vci;
  853         struct smb_share_info ssi;
  854         int error, itype;
  855 
  856         smb_makescred(&scred, td, td->td_ucred);
  857         error = sysctl_wire_old_buffer(req, 0);
  858         if (error)
  859                 return (error);
  860         error = smb_sm_lockvclist(LK_SHARED);
  861         if (error)
  862                 return error;
  863         SMBCO_FOREACH(scp1, &smb_vclist) {
  864                 vcp = (struct smb_vc *)scp1;
  865                 error = smb_vc_lock(vcp, LK_SHARED);
  866                 if (error)
  867                         continue;
  868                 smb_vc_getinfo(vcp, &vci);
  869                 error = SYSCTL_OUT(req, &vci, sizeof(struct smb_vc_info));
  870                 if (error) {
  871                         smb_vc_unlock(vcp, 0);
  872                         break;
  873                 }
  874                 SMBCO_FOREACH(scp2, VCTOCP(vcp)) {
  875                         ssp = (struct smb_share *)scp2;
  876                         error = smb_share_lock(ssp, LK_SHARED);
  877                         if (error) {
  878                                 error = 0;
  879                                 continue;
  880                         }
  881                         smb_share_getinfo(ssp, &ssi);
  882                         smb_share_unlock(ssp, 0);
  883                         error = SYSCTL_OUT(req, &ssi, sizeof(struct smb_share_info));
  884                         if (error)
  885                                 break;
  886                 }
  887                 smb_vc_unlock(vcp, 0);
  888                 if (error)
  889                         break;
  890         }
  891         if (!error) {
  892                 itype = SMB_INFO_NONE;
  893                 error = SYSCTL_OUT(req, &itype, sizeof(itype));
  894         }
  895         smb_sm_unlockvclist();
  896         return error;
  897 }

Cache object: 197c9c15f0c2d950351c8f9d08be4dd1


[ source navigation ] [ diff markup ] [ identifier search ] [ freetext search ] [ file search ] [ list types ] [ track identifier ]


This page is part of the FreeBSD/Linux Linux Kernel Cross-Reference, and was automatically generated using a modified version of the LXR engine.