Merge branch 'v3-2-test' of ssh://git.samba.org/data/git/samba into v3-2-test
[samba.git] / examples / libsmbclient / testbrowse.c
1 #include <sys/types.h>
2 #include <unistd.h>
3 #include <dirent.h>
4 #include <errno.h>
5 #include <stdio.h>
6 #include <string.h>
7 #include <popt.h>
8 #include <stdlib.h>
9 #include <libsmbclient.h>
10 #include "get_auth_data_fn.h"
11
12 static void
13 no_auth_data_fn(const char * pServer,
14                 const char * pShare,
15                 char * pWorkgroup,
16                 int maxLenWorkgroup,
17                 char * pUsername,
18                 int maxLenUsername,
19                 char * pPassword,
20                 int maxLenPassword);
21
22 static void browse(char * path,
23                    int scan,
24                    int indent);
25
26
27 static void
28 get_auth_data_with_context_fn(SMBCCTX * context,
29                               const char * pServer,
30                               const char * pShare,
31                               char * pWorkgroup,
32                               int maxLenWorkgroup,
33                               char * pUsername,
34                               int maxLenUsername,
35                               char * pPassword,
36                               int maxLenPassword);
37
38 int
39 main(int argc, char * argv[])
40 {
41     int                         debug = 0;
42     int                         debug_stderr = 0;
43     int                         no_auth = 0;
44     int                         context_auth = 0;
45     int                         scan = 0;
46     int                         iterations = -1;
47     int                         again;
48     int                         opt;
49     char *                      p;
50     char *                      q;
51     char                        buf[1024];
52     poptContext                 pc;
53     SMBCCTX *                   context;
54     struct poptOption           long_options[] =
55         {
56             POPT_AUTOHELP
57             {
58                 "debug", 'd', POPT_ARG_INT, &debug,
59                 0, "Set debug level", "integer"
60             },
61             {
62                 "stderr", 'e', POPT_ARG_NONE, &debug_stderr,
63                 0, "Debug log to stderr instead of stdout", "integer"
64             },
65             {
66                 "scan", 's', POPT_ARG_NONE, &scan,
67                 0, "Scan for servers and shares", "integer"
68             },
69             {
70                 "iterations", 'i', POPT_ARG_INT, &iterations,
71                 0, "Iterations", "integer"
72             },
73             {
74                 "noauth", 'A', POPT_ARG_NONE, &no_auth,
75                 0, "Do not request authentication data", "integer"
76             },
77             {
78                 "contextauth", 'C', POPT_ARG_NONE, &context_auth,
79                 0, "Use new authentication function with context", "integer"
80             },
81             {
82                 NULL
83             }
84         };
85     
86     setbuf(stdout, NULL);
87
88     pc = poptGetContext("opendir", argc, (const char **)argv, long_options, 0);
89     
90     poptSetOtherOptionHelp(pc, "");
91     
92     while ((opt = poptGetNextOpt(pc)) != -1) {
93         printf("Got option %d = %c\n", opt, opt);
94         switch (opt) {
95         }
96     }
97
98     /* Allocate a new context */
99     context = smbc_new_context();
100     if (!context) {
101         printf("Could not allocate new smbc context\n");
102         return 1;
103     }
104         
105     /* If we're scanning, do no requests for authentication data */
106     if (scan) {
107         no_auth = 1;
108     }
109
110     /* Set mandatory options (is that a contradiction in terms?) */
111     context->debug = debug;
112     if (context_auth) {
113         context->callbacks.auth_fn = NULL;
114         smbc_option_set(context,
115                         "auth_function",
116                         (void *) get_auth_data_with_context_fn);
117         smbc_option_set(context, "user_data", "hello world");
118     } else {
119         context->callbacks.auth_fn =
120             (no_auth ? no_auth_data_fn : get_auth_data_fn);
121     }
122
123     /* If we've been asked to log to stderr instead of stdout, ... */
124     if (debug_stderr) {
125         /* ... then set the option to do so */
126         smbc_option_set(context, "debug_to_stderr", 1);
127     }
128         
129     /* Initialize the context using the previously specified options */
130     if (!smbc_init_context(context)) {
131         smbc_free_context(context, 0);
132         printf("Could not initialize smbc context\n");
133         return 1;
134     }
135
136     /* Tell the compatibility layer to use this context */
137     smbc_set_context(context);
138
139     if (scan)
140     {
141         for (;
142              iterations == -1 || iterations > 0;
143              iterations = (iterations == -1 ? iterations : --iterations))
144         {
145             snprintf(buf, sizeof(buf), "smb://");
146             browse(buf, scan, 0);
147         }
148     }
149     else
150     {
151         for (;
152              iterations == -1 || iterations > 0;
153              iterations = (iterations == -1 ? iterations : --iterations))
154         {
155             fputs("url: ", stdout);
156             p = fgets(buf, sizeof(buf), stdin);
157             if (! p)
158             {
159                 break;
160             }
161
162             if ((p = strchr(buf, '\n')) != NULL)
163             {
164                 *p = '\0';
165             }
166             
167             browse(buf, scan, 0);
168         }
169     }
170
171     exit(0);
172 }
173
174
175 static void
176 no_auth_data_fn(const char * pServer,
177                 const char * pShare,
178                 char * pWorkgroup,
179                 int maxLenWorkgroup,
180                 char * pUsername,
181                 int maxLenUsername,
182                 char * pPassword,
183                 int maxLenPassword)
184 {
185     return;
186 }
187
188
189 static void
190 get_auth_data_with_context_fn(SMBCCTX * context,
191                               const char * pServer,
192                               const char * pShare,
193                               char * pWorkgroup,
194                               int maxLenWorkgroup,
195                               char * pUsername,
196                               int maxLenUsername,
197                               char * pPassword,
198                               int maxLenPassword)
199 {
200     printf("Authenticating with context 0x%lx", context);
201     if (context != NULL) {
202         char *user_data = smbc_option_get(context, "user_data");
203         printf(" with user data %s", user_data);
204     }
205     printf("\n");
206
207     get_auth_data_fn(pServer, pShare, pWorkgroup, maxLenWorkgroup,
208                      pUsername, maxLenUsername, pPassword, maxLenPassword);
209 }
210
211 static void browse(char * path, int scan, int indent)
212 {
213     char *                      p;
214     char                        buf[1024];
215     int                         dir;
216     struct stat                 stat;
217     struct smbc_dirent *        dirent;
218
219     if (! scan)
220     {
221         printf("Opening (%s)...\n", path);
222     }
223         
224     if ((dir = smbc_opendir(path)) < 0)
225     {
226         printf("Could not open directory [%s] (%d:%s)\n",
227                path, errno, strerror(errno));
228         return;
229     }
230
231     while ((dirent = smbc_readdir(dir)) != NULL)
232     {
233         printf("%*.*s%-30s", indent, indent, "", dirent->name);
234
235         switch(dirent->smbc_type)
236         {
237         case SMBC_WORKGROUP:
238             printf("WORKGROUP");
239             break;
240             
241         case SMBC_SERVER:
242             printf("SERVER");
243             break;
244             
245         case SMBC_FILE_SHARE:
246             printf("FILE_SHARE");
247             break;
248             
249         case SMBC_PRINTER_SHARE:
250             printf("PRINTER_SHARE");
251             break;
252             
253         case SMBC_COMMS_SHARE:
254             printf("COMMS_SHARE");
255             break;
256             
257         case SMBC_IPC_SHARE:
258             printf("IPC_SHARE");
259             break;
260             
261         case SMBC_DIR:
262             printf("DIR");
263             break;
264             
265         case SMBC_FILE:
266             printf("FILE");
267
268             p = path + strlen(path);
269             strcat(p, "/");
270             strcat(p+1, dirent->name);
271             if (smbc_stat(path, &stat) < 0)
272             {
273                 printf(" unknown size (reason %d: %s)",
274                        errno, strerror(errno));
275             }
276             else
277             {
278                 printf(" size %lu", (unsigned long) stat.st_size);
279             }
280             *p = '\0';
281
282             break;
283             
284         case SMBC_LINK:
285             printf("LINK");
286             break;
287         }
288
289         printf("\n");
290
291         if (scan &&
292             (dirent->smbc_type == SMBC_WORKGROUP ||
293              dirent->smbc_type == SMBC_SERVER))
294         {
295             /*
296              * don't append server name to workgroup; what we want is:
297              *
298              *   smb://workgroup_name
299              * or
300              *   smb://server_name
301              *
302              */
303             snprintf(buf, sizeof(buf), "smb://%s", dirent->name);
304             browse(buf, scan, indent + 2);
305         }
306     }
307
308     smbc_closedir(dir);
309 }
310