| 1 | /*
|
|---|
| 2 | * Alternate testbrowse utility provided by Mikhail Kshevetskiy.
|
|---|
| 3 | * This version tests use of multiple contexts.
|
|---|
| 4 | */
|
|---|
| 5 |
|
|---|
| 6 | #include <stdio.h>
|
|---|
| 7 | #include <stdlib.h>
|
|---|
| 8 | #include <string.h>
|
|---|
| 9 | #include <libsmbclient.h>
|
|---|
| 10 |
|
|---|
| 11 | int debuglevel = 0;
|
|---|
| 12 | const char *workgroup = "NT";
|
|---|
| 13 | const char *username = "guest";
|
|---|
| 14 | const char *password = "";
|
|---|
| 15 |
|
|---|
| 16 | typedef struct smbitem smbitem;
|
|---|
| 17 | typedef int(*qsort_cmp)(const void *, const void *);
|
|---|
| 18 |
|
|---|
| 19 | struct smbitem{
|
|---|
| 20 | smbitem *next;
|
|---|
| 21 | int type;
|
|---|
| 22 | char name[1];
|
|---|
| 23 | };
|
|---|
| 24 |
|
|---|
| 25 | static void smbc_auth_fn(
|
|---|
| 26 | const char *server,
|
|---|
| 27 | const char *share,
|
|---|
| 28 | char *wrkgrp, int wrkgrplen,
|
|---|
| 29 | char *user, int userlen,
|
|---|
| 30 | char *passwd, int passwdlen){
|
|---|
| 31 |
|
|---|
| 32 | (void) server;
|
|---|
| 33 | (void) share;
|
|---|
| 34 | (void) wrkgrp;
|
|---|
| 35 | (void) wrkgrplen;
|
|---|
| 36 |
|
|---|
| 37 | strncpy(wrkgrp, workgroup, wrkgrplen - 1); wrkgrp[wrkgrplen - 1] = 0;
|
|---|
| 38 | strncpy(user, username, userlen - 1); user[userlen - 1] = 0;
|
|---|
| 39 | strncpy(passwd, password, passwdlen - 1); passwd[passwdlen - 1] = 0;
|
|---|
| 40 | }
|
|---|
| 41 |
|
|---|
| 42 | static SMBCCTX* create_smbctx(void){
|
|---|
| 43 | SMBCCTX *ctx;
|
|---|
| 44 |
|
|---|
| 45 | if ((ctx = smbc_new_context()) == NULL) return NULL;
|
|---|
| 46 |
|
|---|
| 47 | smbc_setDebug(ctx, debuglevel);
|
|---|
| 48 | smbc_setFunctionAuthData(ctx, smbc_auth_fn);
|
|---|
| 49 |
|
|---|
| 50 | if (smbc_init_context(ctx) == NULL){
|
|---|
| 51 | smbc_free_context(ctx, 1);
|
|---|
| 52 | return NULL;
|
|---|
| 53 | }
|
|---|
| 54 |
|
|---|
| 55 | return ctx;
|
|---|
| 56 | }
|
|---|
| 57 |
|
|---|
| 58 | static void delete_smbctx(SMBCCTX* ctx){
|
|---|
| 59 | smbc_getFunctionPurgeCachedServers(ctx)(ctx);
|
|---|
| 60 | smbc_free_context(ctx, 1);
|
|---|
| 61 | }
|
|---|
| 62 |
|
|---|
| 63 | static smbitem* get_smbitem_list(SMBCCTX *ctx, char *smb_path){
|
|---|
| 64 | SMBCFILE *fd;
|
|---|
| 65 | struct smbc_dirent *dirent;
|
|---|
| 66 | smbitem *list = NULL, *item;
|
|---|
| 67 |
|
|---|
| 68 | if ((fd = smbc_getFunctionOpendir(ctx)(ctx, smb_path)) == NULL)
|
|---|
| 69 | return NULL;
|
|---|
| 70 | while((dirent = smbc_getFunctionReaddir(ctx)(ctx, fd)) != NULL){
|
|---|
| 71 | if (strcmp(dirent->name, "") == 0) continue;
|
|---|
| 72 | if (strcmp(dirent->name, ".") == 0) continue;
|
|---|
| 73 | if (strcmp(dirent->name, "..") == 0) continue;
|
|---|
| 74 |
|
|---|
| 75 | if ((item = malloc(sizeof(smbitem) + strlen(dirent->name))) == NULL)
|
|---|
| 76 | continue;
|
|---|
| 77 |
|
|---|
| 78 | item->next = list;
|
|---|
| 79 | item->type = dirent->smbc_type;
|
|---|
| 80 | strcpy(item->name, dirent->name);
|
|---|
| 81 | list = item;
|
|---|
| 82 | }
|
|---|
| 83 | smbc_getFunctionClose(ctx)(ctx, fd);
|
|---|
| 84 | return /* smbitem_list_sort */ (list);
|
|---|
| 85 |
|
|---|
| 86 | }
|
|---|
| 87 |
|
|---|
| 88 | static void print_smb_path(const char *group, const char *path){
|
|---|
| 89 | if ((strlen(group) == 0) && (strlen(path) == 0)) printf("/\n");
|
|---|
| 90 | else if (strlen(path) == 0) printf("/%s\n", group);
|
|---|
| 91 | else{
|
|---|
| 92 | if (strlen(group) == 0) group = "(unknown_group)";
|
|---|
| 93 | printf("/%s/%s\n", group, path);
|
|---|
| 94 | }
|
|---|
| 95 | }
|
|---|
| 96 |
|
|---|
| 97 | static void recurse(SMBCCTX *ctx, const char *smb_group, char *smb_path, int maxlen){
|
|---|
| 98 | int len;
|
|---|
| 99 | smbitem *list, *item;
|
|---|
| 100 | SMBCCTX *ctx1;
|
|---|
| 101 |
|
|---|
| 102 | len = strlen(smb_path);
|
|---|
| 103 |
|
|---|
| 104 | list = get_smbitem_list(ctx, smb_path);
|
|---|
| 105 | while(list != NULL){
|
|---|
| 106 | switch(list->type){
|
|---|
| 107 | case SMBC_WORKGROUP:
|
|---|
| 108 | case SMBC_SERVER:
|
|---|
| 109 | if (list->type == SMBC_WORKGROUP){
|
|---|
| 110 | print_smb_path(list->name, "");
|
|---|
| 111 | smb_group = list->name;
|
|---|
| 112 | }
|
|---|
| 113 | else print_smb_path(smb_group, list->name);
|
|---|
| 114 |
|
|---|
| 115 | if (maxlen < 7 + strlen(list->name)) break;
|
|---|
| 116 | strcpy(smb_path + 6, list->name);
|
|---|
| 117 | if ((ctx1 = create_smbctx()) != NULL){
|
|---|
| 118 | recurse(ctx1, smb_group, smb_path, maxlen);
|
|---|
| 119 | delete_smbctx(ctx1);
|
|---|
| 120 | }else{
|
|---|
| 121 | recurse(ctx, smb_group, smb_path, maxlen);
|
|---|
| 122 | smbc_getFunctionPurgeCachedServers(ctx)(ctx);
|
|---|
| 123 | }
|
|---|
| 124 | break;
|
|---|
| 125 | case SMBC_FILE_SHARE:
|
|---|
| 126 | case SMBC_DIR:
|
|---|
| 127 | case SMBC_FILE:
|
|---|
| 128 | if (maxlen < len + strlen(list->name) + 2) break;
|
|---|
| 129 |
|
|---|
| 130 | smb_path[len] = '/';
|
|---|
| 131 | strcpy(smb_path + len + 1, list->name);
|
|---|
| 132 | print_smb_path(smb_group, smb_path + 6);
|
|---|
| 133 | if (list->type != SMBC_FILE){
|
|---|
| 134 | recurse(ctx, smb_group, smb_path, maxlen);
|
|---|
| 135 | if (list->type == SMBC_FILE_SHARE)
|
|---|
| 136 | smbc_getFunctionPurgeCachedServers(ctx)(ctx);
|
|---|
| 137 | }
|
|---|
| 138 | break;
|
|---|
| 139 | }
|
|---|
| 140 | item = list;
|
|---|
| 141 | list = list->next;
|
|---|
| 142 | free(item);
|
|---|
| 143 | }
|
|---|
| 144 | smb_path[len] = '\0';
|
|---|
| 145 | }
|
|---|
| 146 |
|
|---|
| 147 | int main(int argc, char *argv[]){
|
|---|
| 148 | int i;
|
|---|
| 149 | SMBCCTX *ctx;
|
|---|
| 150 | char smb_path[32768] = "smb://";
|
|---|
| 151 |
|
|---|
| 152 | if ((ctx = create_smbctx()) == NULL){
|
|---|
| 153 | perror("Cant create samba context.");
|
|---|
| 154 | return 1;
|
|---|
| 155 | }
|
|---|
| 156 |
|
|---|
| 157 | if (argc == 1) recurse(ctx, "", smb_path, sizeof(smb_path));
|
|---|
| 158 | else for(i = 1; i < argc; i++){
|
|---|
| 159 | strncpy(smb_path + 6, argv[i], sizeof(smb_path) - 7);
|
|---|
| 160 | smb_path[sizeof(smb_path) - 1] = '\0';
|
|---|
| 161 | recurse(ctx, "", smb_path, sizeof(smb_path));
|
|---|
| 162 | }
|
|---|
| 163 |
|
|---|
| 164 | delete_smbctx(ctx);
|
|---|
| 165 | return 0;
|
|---|
| 166 | }
|
|---|