dist/smbfs/smbutil/login.c

   1 /*
   2  * Copyright (c) 2000, 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: login.c,v 1.6 2001/08/22 03:33:38 bp Exp
  33  */
  34 #include <sys/param.h>
  35 #include <sys/errno.h>
  36 #include <sys/stat.h>
  37 #include <err.h>
  38 #include <stdio.h>
  39 #include <unistd.h>
  40 #include <strings.h>
  41 #include <stdlib.h>
  42 #include <sysexits.h>
  43
  44 #include <cflib.h>
  45
  46 #include <netsmb/smb_lib.h>
  47 #include <netsmb/smb_conn.h>
  48
  49 #include "common.h"
  50
  51
  52 int
  53 cmd_login(int argc, char *argv[])
  54 {
  55         struct smb_ctx sctx, *ctx = &sctx;
  56         int error, opt, setprimary = 0, level;
  57
  58         if (argc < 2)
  59                 login_usage();
  60         if (smb_ctx_init(ctx, argc, argv, SMBL_VC, SMBL_SHARE, SMB_ST_ANY) != 0)
  61                 exit(1);
  62         if (smb_ctx_readrc(ctx) != 0)
  63                 exit(1);
  64         if (smb_rc)
  65                 rc_close(smb_rc);
  66         while ((opt = getopt(argc, argv, STDPARAM_OPT"D")) != EOF) {
  67                 switch(opt){
  68                     case STDPARAM_ARGS:
  69                         error = smb_ctx_opt(ctx, opt, optarg);
  70                         if (error)
  71                                 exit(1);
  72                         break;
  73                     case 'D':
  74                         setprimary = 1;
  75                         break;
  76                     default:
  77                         login_usage();
  78                         /*NOTREACHED*/
  79                 }
  80         }
  81 #ifdef APPLE
  82         if (loadsmbvfs())
  83                 errx(EX_OSERR, "SMB filesystem is not available");
  84 #endif
  85         if (smb_ctx_resolve(ctx) != 0)
  86                 exit(1);
  87         level = ctx->ct_parsedlevel;
  88         error = smb_ctx_lookup(ctx, level, 0);
  89         if (error == 0) {
  90                 smb_error("connection already exists", error);
  91                 exit(0);
  92         }
  93         error = smb_ctx_lookup(ctx, level, SMBLK_CREATE);
  94         if (error) {
  95                 smb_error("could not login to server %s", error, ctx->ct_ssn.ioc_srvname);
  96                 exit(1);
  97         }
  98         switch (level) {
  99             case SMBL_VC:
 100                 opt = SMBV_PERMANENT;
 101                 break;
 102             case SMBL_SHARE:
 103                 opt = SMBS_PERMANENT;
 104                 break;
 105             default:
 106                 smb_error("unknown connection level %d", 0, level);
 107                 exit(1);
 108         }
 109         error = smb_ctx_setflags(ctx, level, opt, opt);
 110         if (error && error != EACCES) {
 111                 smb_error("Can't make connection permanent", error);
 112                 exit(1);
 113         }
 114         printf("Connected to %s%s%s\n", ctx->ct_ssn.ioc_user,
 115             level == SMBL_SHARE ? "@" : "",
 116             level == SMBL_SHARE ? ctx->ct_sh.ioc_share : "");
 117         return 0;
 118 }
 119
 120 int
 121 cmd_logout(int argc, char *argv[])
 122 {
 123         struct smb_ctx sctx, *ctx = &sctx;
 124         int error, opt, level;
 125
 126         if (argc < 2)
 127                 logout_usage();
 128         if (smb_ctx_init(ctx, argc, argv, SMBL_VC, SMBL_SHARE, SMB_ST_ANY) != 0)
 129                 exit(1);
 130         if (smb_ctx_readrc(ctx) != 0)
 131                 exit(1);
 132         if (smb_rc)
 133                 rc_close(smb_rc);
 134         while ((opt = getopt(argc, argv, STDPARAM_OPT)) != EOF){
 135                 switch (opt) {
 136                     case STDPARAM_ARGS:
 137                         error = smb_ctx_opt(ctx, opt, optarg);
 138                         if (error)
 139                                 exit(1);
 140                         break;
 141                     default:
 142                         logout_usage();
 143                         /*NOTREACHED*/
 144                 }
 145         }
 146 #ifdef APPLE
 147         error = loadsmbvfs();
 148         if (error)
 149                 errx(EX_OSERR, "SMB filesystem is not available");
 150 #endif
 151         ctx->ct_ssn.ioc_opt &= ~SMBVOPT_CREATE;
 152         ctx->ct_sh.ioc_opt &= ~SMBSOPT_CREATE;
 153         if (smb_ctx_resolve(ctx) != 0)
 154                 exit(1);
 155         level = ctx->ct_parsedlevel;
 156         error = smb_ctx_lookup(ctx, level, 0);
 157         if (error == ENOENT) {
 158 /*              ctx->ct_ssn.ioc_opt |= SMBCOPT_SINGLE;
 159                 error = smb_ctx_login(ctx);
 160                 if (error == ENOENT) {
 161                         ctx->ct_ssn.ioc_opt |= SMBCOPT_PRIVATE;
 162                         error = smb_ctx_login(ctx);
 163                         if (error == ENOENT) {
 164                                 ctx->ct_ssn.ioc_opt &= ~SMBCOPT_SINGLE;
 165                                 error = smb_ctx_login(ctx);
 166                         }
 167                 }*/
 168                 if (error) {
 169                         smb_error("There is no connection to %s", error, ctx->ct_ssn.ioc_srvname);
 170                         exit(1);
 171                 }
 172         }
 173         if (error)
 174                 exit(1);
 175         switch (level) {
 176             case SMBL_VC:
 177                 opt = SMBV_PERMANENT;
 178                 break;
 179             case SMBL_SHARE:
 180                 opt = SMBS_PERMANENT;
 181                 break;
 182             default:
 183                 smb_error("unknown connection level %d", 0, level);
 184                 exit(1);
 185         }
 186         error = smb_ctx_setflags(ctx, level, opt, 0);
 187         if (error && error != EACCES) {
 188                 smb_error("Can't release connection", error);
 189                 exit(1);
 190         }
 191         printf("Connection unmarked as permanent and will be closed when possible\n");
 192         exit(0);
 193 }
 194
 195 void
 196 login_usage(void)
 197 {
 198         printf(
 199         "usage: smbutil login [-E cs1:cs2] [-I host] [-L locale] [-M crights:srights]\n"
 200         "               [-N cowner:cgroup/sowner:sgroup] [-P]\n"
 201         "               [-R retrycount] [-T timeout]\n"
 202         "               [-W workgroup] //user@server\n");
 203         exit(1);
 204 }
 205
 206 void
 207 logout_usage(void)
 208 {
 209         printf("usage: smbutil logout [user@server]\n");
 210         exit(1);
 211 }