[206] | 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 0x%lx", 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 |
|
---|