| 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 |     smbc_setDebug(context, debug);
 | 
|---|
| 112 |     if (context_auth) {
 | 
|---|
| 113 |         smbc_setFunctionAuthDataWithContext(context,
 | 
|---|
| 114 |                                             get_auth_data_with_context_fn);
 | 
|---|
| 115 |         smbc_setOptionUserData(context, "hello world");
 | 
|---|
| 116 |     } else {
 | 
|---|
| 117 |         smbc_setFunctionAuthData(context, get_auth_data_fn);
 | 
|---|
| 118 |     }
 | 
|---|
| 119 | 
 | 
|---|
| 120 |     smbc_setOptionUseKerberos(context, 1);
 | 
|---|
| 121 |     smbc_setOptionFallbackAfterKerberos(context, 1);
 | 
|---|
| 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_setOptionDebugToStderr(context, 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 %p", context);
 | 
|---|
| 201 |     if (context != NULL) {
 | 
|---|
| 202 |         char *user_data = smbc_getOptionUserData(context);
 | 
|---|
| 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 |     
 | 
|---|