add the 'y' and 'Y' options to ps, and add the 'iac' keyword. The 'y'
[dragonfly.git] / contrib / 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  * $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 }