2 Unix SMB/CIFS implementation.
3 Samba database functions
4 Copyright (C) Andrew Tridgell 1999-2000
5 Copyright (C) Paul `Rusty' Russell 2000
6 Copyright (C) Jeremy Allison 2000
7 Copyright (C) Andrew Esh 2001
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
39 /* a tdb tool for manipulating a tdb database */
41 #define FSTRING_LEN 256
42 typedef char fstring
[FSTRING_LEN
];
44 typedef struct connections_key
{
50 typedef struct connections_data
{
62 static TDB_CONTEXT
*tdb
;
64 static int print_rec(TDB_CONTEXT
*the_tdb
, TDB_DATA key
, TDB_DATA dbuf
, void *state
);
66 static void print_asc(unsigned char *buf
,int len
)
70 /* We're probably printing ASCII strings so don't try to display
71 the trailing NULL character. */
73 if (buf
[len
- 1] == 0)
77 printf("%c",isprint(buf
[i
])?buf
[i
]:'.');
80 static void print_data(unsigned char *buf
,int len
)
86 printf("%02X ",(int)buf
[i
]);
88 if (i
%8 == 0) printf(" ");
90 print_asc(&buf
[i
-16],8); printf(" ");
91 print_asc(&buf
[i
-8],8); printf("\n");
92 if (i
<len
) printf("[%03X] ",i
);
100 if (n
>8) printf(" ");
101 while (n
--) printf(" ");
105 print_asc(&buf
[i
-(i
%16)],n
); printf(" ");
107 if (n
>0) print_asc(&buf
[i
-n
],n
);
112 static void help(void)
116 " create dbname : create a database\n"
117 " open dbname : open an existing database\n"
118 " erase : erase the database\n"
119 " dump : dump the database as strings\n"
120 " insert key data : insert a record\n"
121 " store key data : store a record (replace)\n"
122 " show key : show a record by key\n"
123 " delete key : delete a record by key\n"
124 " list : print the database hash table and freelist\n"
125 " free : print the database freelist\n"
126 " 1 | first : print the first record\n"
127 " n | next : print the next record\n"
128 " q | quit : terminate\n"
129 " \\n : repeat 'next' command\n"
133 static void terror(char *why
)
138 static char *get_token(int startover
)
140 static char tmp
[1024];
141 static char *cont
= NULL
;
142 char *insert
, *start
;
143 char *k
= strtok(NULL
, " ");
154 insert
= start
+ strlen(start
) - 1;
155 while (*insert
== '\\') {
157 k
= strtok(NULL
, " ");
161 insert
= start
+ strlen(start
) - 1;
164 /* Get ready for next call */
165 cont
= start
+ strlen(start
) + 1;
169 static void create_tdb(void)
171 char *tok
= get_token(1);
176 if (tdb
) tdb_close(tdb
);
177 tdb
= tdb_open(tok
, 0, TDB_CLEAR_IF_FIRST
,
178 O_RDWR
| O_CREAT
| O_TRUNC
, 0600);
180 printf("Could not create %s: %s\n", tok
, strerror(errno
));
184 static void open_tdb(void)
186 char *tok
= get_token(1);
191 if (tdb
) tdb_close(tdb
);
192 tdb
= tdb_open(tok
, 0, 0, O_RDWR
, 0600);
194 printf("Could not open %s: %s\n", tok
, strerror(errno
));
198 static void insert_tdb(void)
200 char *k
= get_token(1);
201 char *d
= get_token(0);
210 key
.dsize
= strlen(k
)+1;
212 dbuf
.dsize
= strlen(d
)+1;
214 if (tdb_store(tdb
, key
, dbuf
, TDB_INSERT
) == -1) {
215 terror("insert failed");
219 static void store_tdb(void)
221 char *k
= get_token(1);
222 char *d
= get_token(0);
231 key
.dsize
= strlen(k
)+1;
233 dbuf
.dsize
= strlen(d
)+1;
235 printf("Storing key:\n");
236 print_rec(tdb
, key
, dbuf
, NULL
);
238 if (tdb_store(tdb
, key
, dbuf
, TDB_REPLACE
) == -1) {
239 terror("store failed");
243 static void show_tdb(void)
245 char *k
= get_token(1);
254 /* key.dsize = strlen(k)+1;*/
255 key
.dsize
= strlen(k
);
257 dbuf
= tdb_fetch(tdb
, key
);
259 terror("fetch failed");
262 /* printf("%s : %*.*s\n", k, (int)dbuf.dsize, (int)dbuf.dsize, dbuf.dptr); */
263 print_rec(tdb
, key
, dbuf
, NULL
);
266 static void delete_tdb(void)
268 char *k
= get_token(1);
277 key
.dsize
= strlen(k
)+1;
279 if (tdb_delete(tdb
, key
) != 0) {
280 terror("delete failed");
285 static int print_conn_key(TDB_DATA key
)
287 printf( "pid =%5d ", ((connections_key
*)key
.dptr
)->pid
);
288 printf( "cnum =%10d ", ((connections_key
*)key
.dptr
)->cnum
);
289 printf( "name =[%s]\n", ((connections_key
*)key
.dptr
)->name
);
293 static int print_conn_data(TDB_DATA dbuf
)
295 printf( "pid =%5d ", ((connections_data
*)dbuf
.dptr
)->pid
);
296 printf( "cnum =%10d ", ((connections_data
*)dbuf
.dptr
)->cnum
);
297 printf( "name =[%s]\n", ((connections_data
*)dbuf
.dptr
)->name
);
299 printf( "uid =%5d ", ((connections_data
*)dbuf
.dptr
)->uid
);
300 printf( "addr =[%s]\n", ((connections_data
*)dbuf
.dptr
)->addr
);
301 printf( "gid =%5d ", ((connections_data
*)dbuf
.dptr
)->gid
);
302 printf( "machine=[%s]\n", ((connections_data
*)dbuf
.dptr
)->machine
);
303 printf( "start = %s\n", ctime(&((connections_data
*)dbuf
.dptr
)->start
));
308 static int print_rec(TDB_CONTEXT
*the_tdb
, TDB_DATA key
, TDB_DATA dbuf
, void *state
)
312 print_conn_data(dbuf
);
315 printf("\nkey %d bytes\n", key
.dsize
);
316 print_asc(key
.dptr
, key
.dsize
);
317 printf("\ndata %d bytes\n", dbuf
.dsize
);
318 print_data(dbuf
.dptr
, dbuf
.dsize
);
323 static int print_key(TDB_CONTEXT
*the_tdb
, TDB_DATA key
, TDB_DATA dbuf
, void *state
)
325 print_asc(key
.dptr
, key
.dsize
);
330 static int total_bytes
;
332 static int traverse_fn(TDB_CONTEXT
*the_tdb
, TDB_DATA key
, TDB_DATA dbuf
, void *state
)
334 total_bytes
+= dbuf
.dsize
;
338 static void info_tdb(void)
342 if ((count
= tdb_traverse(tdb
, traverse_fn
, NULL
) == -1))
343 printf("Error = %s\n", tdb_errorstr(tdb
));
345 printf("%d records totalling %d bytes\n", count
, total_bytes
);
348 static char *tdb_getline(char *prompt
)
350 static char line
[1024];
352 fputs(prompt
, stdout
);
354 p
= fgets(line
, sizeof(line
)-1, stdin
);
355 if (p
) p
= strchr(p
, '\n');
360 static int do_delete_fn(TDB_CONTEXT
*the_tdb
, TDB_DATA key
, TDB_DATA dbuf
,
363 return tdb_delete(the_tdb
, key
);
366 static void first_record(TDB_CONTEXT
*the_tdb
, TDB_DATA
*pkey
)
369 *pkey
= tdb_firstkey(the_tdb
);
371 dbuf
= tdb_fetch(the_tdb
, *pkey
);
372 if (!dbuf
.dptr
) terror("fetch failed");
374 /* printf("%s : %*.*s\n", k, (int)dbuf.dsize, (int)dbuf.dsize, dbuf.dptr); */
375 print_rec(the_tdb
, *pkey
, dbuf
, NULL
);
379 static void next_record(TDB_CONTEXT
*the_tdb
, TDB_DATA
*pkey
)
382 *pkey
= tdb_nextkey(the_tdb
, *pkey
);
384 dbuf
= tdb_fetch(the_tdb
, *pkey
);
386 terror("fetch failed");
388 /* printf("%s : %*.*s\n", k, (int)dbuf.dsize, (int)dbuf.dsize, dbuf.dptr); */
389 print_rec(the_tdb
, *pkey
, dbuf
, NULL
);
392 int main(int argc
, char *argv
[])
397 TDB_DATA iterate_kbuf
;
400 static char tmp
[1024];
401 sprintf(tmp
, "open %s", argv
[1]);
406 while ((line
= tdb_getline("tdb> "))) {
410 if (line
[0] == '!') {
415 if ((tok
= strtok(line
," "))==NULL
) {
417 next_record(tdb
, &iterate_kbuf
);
420 if (strcmp(tok
,"create") == 0) {
424 } else if (strcmp(tok
,"open") == 0) {
427 } else if ((strcmp(tok
, "q") == 0) ||
428 (strcmp(tok
, "quit") == 0)) {
432 /* all the rest require a open database */
435 terror("database not open");
440 if (strcmp(tok
,"insert") == 0) {
443 } else if (strcmp(tok
,"store") == 0) {
446 } else if (strcmp(tok
,"show") == 0) {
449 } else if (strcmp(tok
,"erase") == 0) {
451 tdb_traverse(tdb
, do_delete_fn
, NULL
);
452 } else if (strcmp(tok
,"delete") == 0) {
455 } else if (strcmp(tok
,"dump") == 0) {
457 tdb_traverse(tdb
, print_rec
, NULL
);
458 } else if (strcmp(tok
,"list") == 0) {
460 } else if (strcmp(tok
, "free") == 0) {
461 tdb_printfreelist(tdb
);
462 } else if (strcmp(tok
,"info") == 0) {
464 } else if ( (strcmp(tok
, "1") == 0) ||
465 (strcmp(tok
, "first") == 0)) {
467 first_record(tdb
, &iterate_kbuf
);
468 } else if ((strcmp(tok
, "n") == 0) ||
469 (strcmp(tok
, "next") == 0)) {
470 next_record(tdb
, &iterate_kbuf
);
471 } else if ((strcmp(tok
, "keys") == 0)) {
473 tdb_traverse(tdb
, print_key
, NULL
);
479 if (tdb
) tdb_close(tdb
);