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 | }
|
---|