diff options
Diffstat (limited to 'examples/libmsrpc/test/reg/regvalenum.c')
-rw-r--r-- | examples/libmsrpc/test/reg/regvalenum.c | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/examples/libmsrpc/test/reg/regvalenum.c b/examples/libmsrpc/test/reg/regvalenum.c new file mode 100644 index 0000000000..9778f4e2b3 --- /dev/null +++ b/examples/libmsrpc/test/reg/regvalenum.c @@ -0,0 +1,103 @@ +/*tests enumerating registry values*/ + +#include "libmsrpc.h" +#include "test_util.h" + +#define MAX_KEYS_PER_ENUM 3 + + +int main(int argc, char **argv) { + CacServerHandle *hnd = NULL; + TALLOC_CTX *mem_ctx = NULL; + + int num_keys; + + int max_enum; + + fstring *key_names; + + int i; + + mem_ctx = talloc_init("regvalenum"); + + hnd = cac_NewServerHandle(True); + + cac_parse_cmd_line(argc, argv, hnd); + + cac_SetAuthDataFn(hnd, cactest_GetAuthDataFn); + + if(!cac_Connect(hnd, NULL)) { + fprintf(stderr, "Could not connect to server.\n Error: %s.\n errno: %s\n", nt_errstr(hnd->status), strerror(errno)); + cac_FreeHandle(hnd); + exit(-1); + } + + printf("How many keys do you want to open?: "); + fscanf(stdin, "%d", &num_keys); + + printf("How many values per enum?: "); + fscanf(stdin, "%d", &max_enum); + + key_names = TALLOC_ARRAY(mem_ctx, fstring , num_keys); + if(!key_names) { + fprintf(stderr, "No memory\n"); + exit(-1); + } + + for(i = 0; i < num_keys; i++) { + printf("Enter key to open: "); + fscanf(stdin, "%s", key_names[i]); + } + + for(i = 0; i < num_keys; i++) { + printf("trying to open key %s...\n", key_names[i]); + + struct RegOpenKey rok; + ZERO_STRUCT(rok); + + rok.in.parent_key = NULL; + rok.in.name = key_names[i]; + rok.in.access = REG_KEY_ALL; + + if(!cac_RegOpenKey(hnd, mem_ctx, &rok)) { + fprintf(stderr, "Could not open key %s\n Error: %s\n", rok.in.name, nt_errstr(hnd->status)); + continue; + } + + /**enumerate all the subkeys*/ + printf("Enumerating all values:\n"); + + struct RegEnumValues rev; + ZERO_STRUCT(rev); + + rev.in.key = rok.out.key; + rev.in.max_values = max_enum; + + while(cac_RegEnumValues(hnd, mem_ctx, &rev)) { + int j; + + for(j = 0; j < rev.out.num_values; j++) { + printf(" Value name: %s\n", rev.out.value_names[j]); + print_value(rev.out.types[j], rev.out.values[j]); + } + } + + if(CAC_OP_FAILED(hnd->status)) { + fprintf(stderr, "Could not enumerate values: %s\n", nt_errstr(hnd->status)); + continue; + } + + printf("closing key %s...\n", key_names[i]); + + if(!cac_RegClose(hnd, mem_ctx, rok.out.key)) { + fprintf(stderr, "Could not close handle %s\n", nt_errstr(hnd->status)); + } + } + + cac_FreeHandle(hnd); + + talloc_destroy(mem_ctx); + + return 0; + +} |