47668f7c9c4af4f829b121f7b06487dd90d0d8f3
[samba.git] / examples / libsmbclient / testacl.c
1 #include <stdlib.h>
2 #include <string.h>
3 #include <errno.h>
4 #include <popt.h>
5 #include "libsmbclient.h"
6
7 enum acl_mode
8 {
9     SMB_ACL_GET,
10     SMB_ACL_SET,
11     SMB_ACL_DELETE,
12     SMB_ACL_MODIFY,
13     SMB_ACL_ADD,
14     SMB_ACL_CHOWN,
15     SMB_ACL_CHGRP
16 };
17
18 static void
19 get_auth_data_fn(const char * pServer,
20                  const char * pShare,
21                  char * pWorkgroup,
22                  int maxLenWorkgroup,
23                  char * pUsername,
24                  int maxLenUsername,
25                  char * pPassword,
26                  int maxLenPassword)
27     
28 {
29     char temp[128];
30     
31     fprintf(stdout, "Workgroup: [%s] ", pWorkgroup);
32     fgets(temp, sizeof(temp), stdin);
33     
34     if (temp[strlen(temp) - 1] == '\n') /* A new line? */
35     {
36         temp[strlen(temp) - 1] = '\0';
37     }
38     
39     if (temp[0] != '\0')
40     {
41         strncpy(pWorkgroup, temp, maxLenWorkgroup - 1);
42     }
43     
44     fprintf(stdout, "Username: [%s] ", pUsername);
45     fgets(temp, sizeof(temp), stdin);
46     
47     if (temp[strlen(temp) - 1] == '\n') /* A new line? */
48     {
49         temp[strlen(temp) - 1] = '\0';
50     }
51     
52     if (temp[0] != '\0')
53     {
54         strncpy(pUsername, temp, maxLenUsername - 1);
55     }
56     
57     fprintf(stdout, "Password: ");
58     fgets(temp, sizeof(temp), stdin);
59     
60     if (temp[strlen(temp) - 1] == '\n') /* A new line? */
61     {
62         temp[strlen(temp) - 1] = '\0';
63     }
64     
65     if (temp[0] != '\0')
66     {
67         strncpy(pPassword, temp, maxLenPassword - 1);
68     }
69 }
70
71
72 int main(int argc, const char *argv[])
73 {
74     int opt;
75     int flags;
76     int debug = 0;
77     int numeric = 0;
78     enum acl_mode mode = SMB_ACL_GET;
79     static char *the_acl = NULL;
80     int ret;
81     char *p;
82     char *debugstr;
83     char path[1024];
84     char value[1024];
85     poptContext pc;
86     struct poptOption long_options[] =
87         {
88             POPT_AUTOHELP
89             {
90                 "numeric", 'n', POPT_ARG_NONE, &numeric,
91                 1, "Don't resolve sids or masks to names"
92             },
93             {
94                 "debug", 'd', POPT_ARG_INT, &debug,
95                 0, "Set debug level (0-100)"
96             },
97             {
98                 "delete", 'D', POPT_ARG_STRING, NULL,
99                 'D', "Delete an acl", "ACL"
100             },
101             {
102                 "modify", 'M', POPT_ARG_STRING, NULL,
103                 'M', "Modify an acl", "ACL"
104             },
105             {
106                 "add", 'a', POPT_ARG_STRING, NULL,
107                 'a', "Add an acl", "ACL"
108             },
109             {
110                 "set", 'S', POPT_ARG_STRING, NULL,
111                 'S', "Set acls", "ACLS"
112             },
113             {
114                 "chown", 'C', POPT_ARG_STRING, NULL,
115                 'C', "Change ownership of a file", "USERNAME"
116             },
117             {
118                 "chgrp", 'G', POPT_ARG_STRING, NULL,
119                 'G', "Change group ownership of a file", "GROUPNAME"
120             },
121             {
122                 "get", 'g', POPT_ARG_STRING, NULL,
123                 'g', "Get a specific acl attribute", "ACL"
124             },
125             {
126                 NULL
127             }
128         };
129     
130     setbuf(stdout, NULL);
131
132     pc = poptGetContext("smbcacls", argc, argv, long_options, 0);
133     
134     poptSetOtherOptionHelp(pc, "smb://server1/share1/filename");
135     
136     while ((opt = poptGetNextOpt(pc)) != -1) {
137         switch (opt) {
138         case 'S':
139             the_acl = strdup(poptGetOptArg(pc));
140             mode = SMB_ACL_SET;
141             break;
142             
143         case 'D':
144             the_acl = strdup(poptGetOptArg(pc));
145             mode = SMB_ACL_DELETE;
146             break;
147             
148         case 'M':
149             the_acl = strdup(poptGetOptArg(pc));
150             mode = SMB_ACL_MODIFY;
151             break;
152             
153         case 'a':
154             the_acl = strdup(poptGetOptArg(pc));
155             mode = SMB_ACL_ADD;
156             break;
157
158         case 'g':
159             the_acl = strdup(poptGetOptArg(pc));
160             mode = SMB_ACL_GET;
161             break;
162
163         case 'C':
164             the_acl = strdup(poptGetOptArg(pc));
165             mode = SMB_ACL_CHOWN;
166             break;
167
168         case 'G':
169             the_acl = strdup(poptGetOptArg(pc));
170             mode = SMB_ACL_CHGRP;
171             break;
172         }
173     }
174     
175     /* Make connection to server */
176     if(!poptPeekArg(pc)) { 
177         poptPrintUsage(pc, stderr, 0);
178         return 1;
179     }
180     
181     strcpy(path, poptGetArg(pc));
182     
183     if (smbc_init(get_auth_data_fn, debug) != 0)
184     {
185         printf("Could not initialize smbc_ library\n");
186         return 1;
187     }
188     
189     /* Perform requested action */
190     
191     switch(mode)
192     {
193     case SMB_ACL_GET:
194         if (the_acl == NULL)
195         {
196             if (numeric)
197             {
198                 the_acl = "system.nt_sec_desc.*";
199             }
200             else
201             {
202                 the_acl = "system.nt_sec_desc.*+";
203             }
204         }
205         ret = smbc_getxattr(path, the_acl, value, sizeof(value));
206         if (ret < 0)
207         {
208             printf("Could not get attributes for [%s] %d: %s\n",
209                    path, errno, strerror(errno));
210             return 1;
211         }
212         
213         printf("Attributes for [%s] are:\n%s\n", path, value);
214         break;
215
216     case SMB_ACL_ADD:
217         flags = SMBC_XATTR_FLAG_CREATE;
218         debugstr = "add attributes";
219         goto do_set;
220         
221     case SMB_ACL_MODIFY:
222         flags = SMBC_XATTR_FLAG_REPLACE;
223         debugstr = "modify attributes";
224         goto do_set;
225
226     case SMB_ACL_CHOWN:
227         snprintf(value, sizeof(value),
228                  "system.nt_sec_desc.owner%s:%s",
229                  numeric ? "" : "+", the_acl);
230         the_acl = value;
231         debugstr = "chown owner";
232         goto do_set;
233
234     case SMB_ACL_CHGRP:
235         snprintf(value, sizeof(value),
236                  "system.nt_sec_desc.group%s:%s",
237                  numeric ? "" : "+", the_acl);
238         the_acl = value;
239         debugstr = "change group";
240         goto do_set;
241
242     case SMB_ACL_SET:
243         flags = 0;
244         debugstr = "set attributes";
245         
246       do_set:
247         if ((p = strchr(the_acl, ':')) == NULL)
248         {
249             printf("Missing value.  ACL must be name:value pair\n");
250             return 1;
251         }
252
253         *p++ = '\0';
254         
255         ret = smbc_setxattr(path, the_acl, p, strlen(p), flags);
256         if (ret < 0)
257         {
258             printf("Could not %s for [%s] %d: %s\n",
259                    debugstr, path, errno, strerror(errno));
260             return 1;
261         }
262         break;
263
264     case SMB_ACL_DELETE:
265         ret = smbc_removexattr(path, the_acl);
266         if (ret < 0)
267         {
268             printf("Could not remove attribute %s for [%s] %d:%s\n",
269                    the_acl, path, errno, strerror(errno));
270             return 1;
271         }
272         break;
273
274     default:
275         printf("operation not yet implemented\n");
276         break;
277     }
278     
279     return 0;
280 }