source: trunk/src/reglookup.c @ 61

Last change on this file since 61 was 61, checked in by tim, 18 years ago

various minor speedups and memory leak fixes. Added support for NONE and LINK types.

  • Property svn:keywords set to Id
File size: 20.8 KB
Line 
1/*
2 * A utility to read a Windows NT/2K/XP/2K3 registry file, using
3 * Gerald Carter''s regfio interface.
4 *
5 * Copyright (C) 2005-2006 Timothy D. Morgan
6 * Copyright (C) 2002 Richard Sharpe, rsharpe@richardsharpe.com
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; version 2 of the License.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. 
20 *
21 * $Id: reglookup.c 61 2006-07-18 01:56:59Z tim $
22 */
23
24
25#include <stdlib.h>
26#include <stdio.h>
27#include <string.h>
28#include <strings.h>
29#include <time.h>
30#include <iconv.h>
31#include "../include/regfio.h"
32#include "../include/void_stack.h"
33
34/* Globals, influenced by command line parameters */
35bool print_verbose = false;
36bool print_security = false;
37bool print_header = true;
38bool path_filter_enabled = false;
39bool type_filter_enabled = false;
40char* path_filter = NULL;
41int type_filter;
42char* registry_file = NULL;
43
44/* Other globals */
45const char* special_chars = ",\"\\";
46iconv_t conv_desc;
47
48
49void bailOut(int code, char* message)
50{
51  fprintf(stderr, message);
52  exit(code);
53}
54
55
56/* Returns a newly malloc()ed string which contains original buffer,
57 * except for non-printable or special characters are quoted in hex
58 * with the syntax '\xQQ' where QQ is the hex ascii value of the quoted
59 * character.  A null terminator is added, since only ascii, not binary,
60 * is returned.
61 */
62static char* quote_buffer(const unsigned char* str, 
63                          unsigned int len, const char* special)
64{
65  unsigned int i, added_len;
66  unsigned int num_written = 0;
67
68  unsigned int buf_len = sizeof(char)*(len+1);
69  char* ret_val = malloc(buf_len);
70  char* tmp_buf;
71
72  if(ret_val == NULL)
73    return NULL;
74
75  for(i=0; i<len; i++)
76  {
77    if(buf_len <= (num_written+5))
78    {
79      /* Expand the buffer by the memory consumption rate seen so far
80       * times the amount of input left to process.  The expansion is bounded
81       * below by a minimum safety increase, and above by the maximum possible
82       * output string length.
83       */
84      added_len = (len-i)*num_written/(i+1);
85      if((buf_len+added_len) > (len*4+1))
86        buf_len = len*4+1;
87      else
88      {
89        if (added_len < 5)
90          buf_len += 5;
91        else
92          buf_len += added_len;
93      }
94
95      tmp_buf = realloc(ret_val, buf_len);
96      if(tmp_buf == NULL)
97      {
98        free(ret_val);
99        return NULL;
100      }
101      ret_val = tmp_buf;
102    }
103   
104    if(str[i] < 32 || str[i] > 126 || strchr(special, str[i]) != NULL)
105    {
106      num_written += snprintf(ret_val + num_written, buf_len - num_written,
107                              "\\x%.2X", str[i]);
108    }
109    else
110      ret_val[num_written++] = str[i];
111  }
112  ret_val[num_written] = '\0';
113
114  return ret_val;
115}
116
117
118/* Returns a newly malloc()ed string which contains original string,
119 * except for non-printable or special characters are quoted in hex
120 * with the syntax '\xQQ' where QQ is the hex ascii value of the quoted
121 * character.
122 */
123static char* quote_string(const char* str, const char* special)
124{
125  unsigned int len;
126
127  if(str == NULL)
128    return NULL;
129
130  len = strlen(str);
131  return quote_buffer((const unsigned char*)str, len, special);
132}
133
134
135/*
136 * Convert from Unicode to ASCII.
137 */
138static int uni_to_ascii(unsigned char* uni, char* ascii, 
139                        unsigned int uni_max, unsigned int ascii_max)
140{
141  char* inbuf = (char*)uni;
142  char* outbuf = ascii;
143  int ret;
144
145  /* Set up conversion descriptor. */
146  conv_desc = iconv_open("US-ASCII", "UTF-16LE");
147
148  ret = iconv(conv_desc, &inbuf, &uni_max, &outbuf, &ascii_max);
149  if(ret == -1)
150  {
151    iconv_close(conv_desc);
152    return -1;
153  }
154
155  iconv_close(conv_desc); 
156  return strlen(ascii);
157}
158
159
160/*
161 * Convert a data value to a string for display
162 */
163static char* data_to_ascii(unsigned char *datap, int len, int type)
164{
165  char* asciip;
166  unsigned int i;
167  unsigned short num_nulls;
168  char* ascii;
169  unsigned char* cur_str;
170  char* cur_ascii;
171  char* cur_quoted;
172  unsigned int cur_str_len;
173  unsigned int ascii_max, cur_str_max;
174  unsigned int str_rem, cur_str_rem, alen;
175
176  switch (type) 
177  {
178    /* REG_LINK is a symbolic link, stored as a unicode string. */
179  case REG_LINK:
180  case REG_SZ:
181    ascii_max = sizeof(char)*len;
182    ascii = malloc(ascii_max+4);
183    if(ascii == NULL)
184      return NULL;
185   
186    if(uni_to_ascii(datap, ascii, len, ascii_max) < 0)
187    {
188      free(ascii);
189      return NULL;
190    }
191    cur_quoted = quote_string(ascii, special_chars);
192    free(ascii);
193    return cur_quoted;
194    break;
195
196    /* XXX: This could be Unicode or ASCII.  Are we processing it
197     *      correctly in both cases?
198     */
199  case REG_EXPAND_SZ:
200    ascii_max = sizeof(char)*len;
201    ascii = malloc(ascii_max+2);
202    if(ascii == NULL)
203      return NULL;
204
205    if(uni_to_ascii(datap, ascii, len, ascii_max) < 0)
206    {
207      free(ascii);
208      return NULL;
209    }
210    cur_quoted = quote_string(ascii, special_chars);
211    free(ascii);
212    return cur_quoted;
213    break;
214
215  case REG_DWORD:
216    ascii_max = sizeof(char)*11;
217    ascii = malloc(ascii_max);
218    if(ascii == NULL)
219      return NULL;
220
221    snprintf(ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
222             datap[0], datap[1], datap[2], datap[3]);
223    return ascii;
224    break;
225
226  case REG_DWORD_BE:
227    ascii_max = sizeof(char)*11;
228    ascii = malloc(ascii_max);
229    if(ascii == NULL)
230      return NULL;
231
232    snprintf(ascii, ascii_max, "0x%.2X%.2X%.2X%.2X", 
233             datap[3], datap[2], datap[1], datap[0]);
234    return ascii;
235    break;
236
237  /* XXX: this MULTI_SZ parser is pretty inefficient.  Should be
238   *      redone with fewer malloc calls and better string concatenation.
239   */
240  case REG_MULTI_SZ:
241    ascii_max = sizeof(char)*len*4;
242    cur_str_max = sizeof(char)*len+1;
243    cur_str = malloc(cur_str_max);
244    cur_ascii = malloc(cur_str_max);
245    ascii = malloc(ascii_max+4);
246    if(ascii == NULL || cur_str == NULL || cur_ascii == NULL)
247      return NULL;
248
249    /* Reads until it reaches 4 consecutive NULLs,
250     * which is two nulls in unicode, or until it reaches len, or until we
251     * run out of buffer.  The latter should never happen, but we shouldn't
252     * trust our file to have the right lengths/delimiters.
253     */
254    asciip = ascii;
255    num_nulls = 0;
256    str_rem = ascii_max;
257    cur_str_rem = cur_str_max;
258    cur_str_len = 0;
259
260    for(i=0; (i < len) && str_rem > 0; i++)
261    {
262      *(cur_str+cur_str_len) = *(datap+i);
263      if(*(cur_str+cur_str_len) == 0)
264        num_nulls++;
265      else
266        num_nulls = 0;
267      cur_str_len++;
268
269      if(num_nulls == 2)
270      {
271        if(uni_to_ascii(cur_str, cur_ascii, cur_str_max, cur_str_max) < 0)
272        {
273          free(ascii);
274          free(cur_ascii);
275          free(cur_str);
276          return NULL;
277        }
278
279        cur_quoted = quote_string(cur_ascii, ",|\"\\");
280        alen = snprintf(asciip, str_rem, "%s", cur_quoted);
281        asciip += alen;
282        str_rem -= alen;
283        free(cur_quoted);
284
285        if(*(datap+i+1) == 0 && *(datap+i+2) == 0)
286          break;
287        else
288        {
289          if(str_rem > 0)
290          {
291            asciip[0] = '|';
292            asciip[1] = '\0';
293            asciip++;
294            str_rem--;
295          }
296          memset(cur_str, 0, cur_str_max);
297          cur_str_len = 0;
298          num_nulls = 0;
299          /* To eliminate leading nulls in subsequent strings. */
300          i++;
301        }
302      }
303    }
304    *asciip = 0;
305    free(cur_str);
306    free(cur_ascii);
307    return ascii;
308    break;
309
310  /* XXX: Dont know what to do with these yet, just print as binary... */
311  case REG_NONE:
312  case REG_RESOURCE_LIST:
313  case REG_FULL_RESOURCE_DESCRIPTOR:
314  case REG_RESOURCE_REQUIREMENTS_LIST:
315
316  case REG_BINARY:
317    return quote_buffer(datap, len, special_chars);
318    break;
319
320  default:
321    return NULL;
322    break;
323  } 
324
325  return NULL;
326}
327
328
329void_stack* path2Stack(const char* s)
330{
331  void_stack* ret_val;
332  void_stack* rev_ret = void_stack_new(1024);
333  const char* cur = s;
334  char* next = NULL;
335  char* copy;
336
337  if (rev_ret == NULL)
338    return NULL;
339  if (s == NULL)
340    return rev_ret;
341 
342  while((next = strchr(cur, '/')) != NULL)
343  {
344    if ((next-cur) > 0)
345    {
346      copy = (char*)malloc((next-cur+1)*sizeof(char));
347      if(copy == NULL)
348        bailOut(2, "ERROR: Memory allocation problem.\n");
349         
350      memcpy(copy, cur, next-cur);
351      copy[next-cur] = '\0';
352      void_stack_push(rev_ret, copy);
353    }
354    cur = next+1;
355  }
356  if(strlen(cur) > 0)
357  {
358    copy = strdup(cur);
359    void_stack_push(rev_ret, copy);
360  }
361
362  ret_val = void_stack_copy_reverse(rev_ret);
363  void_stack_destroy(rev_ret);
364
365  return ret_val;
366}
367
368
369char* stack2Path(void_stack* nk_stack)
370{
371  const REGF_NK_REC* cur;
372  uint32 buf_left = 127;
373  uint32 buf_len = buf_left+1;
374  uint32 name_len = 0;
375  uint32 grow_amt;
376  char* buf; 
377  char* new_buf;
378  void_stack_iterator* iter;
379 
380  buf = (char*)malloc((buf_len)*sizeof(char));
381  if (buf == NULL)
382    return NULL;
383  buf[0] = '\0';
384
385  iter = void_stack_iterator_new(nk_stack);
386  if (iter == NULL)
387  {
388    free(buf);
389    return NULL;
390  }
391
392  /* skip root element */
393  cur = void_stack_iterator_next(iter);
394
395  while((cur = void_stack_iterator_next(iter)) != NULL)
396  {
397    buf[buf_len-buf_left-1] = '/';
398    buf_left -= 1;
399    name_len = strlen(cur->keyname);
400    if(name_len+1 > buf_left)
401    {
402      grow_amt = (uint32)(buf_len/2);
403      buf_len += name_len+1+grow_amt-buf_left;
404      if((new_buf = realloc(buf, buf_len)) == NULL)
405      {
406        free(buf);
407        free(iter);
408        return NULL;
409      }
410      buf = new_buf;
411      buf_left = grow_amt + name_len + 1;
412    }
413    strncpy(buf+(buf_len-buf_left-1), cur->keyname, name_len);
414    buf_left -= name_len;
415    buf[buf_len-buf_left-1] = '\0';
416  }
417
418  return buf;
419}
420
421
422void printValue(REGF_VK_REC* vk, char* prefix)
423{
424  uint32 size;
425  uint8 tmp_buf[4];
426  char* quoted_value;
427  char* quoted_prefix;
428  char* quoted_name;
429
430  /* Thanks Microsoft for making this process so straight-forward!!! */
431  size = (vk->data_size & ~VK_DATA_IN_OFFSET);
432  if(vk->data_size & VK_DATA_IN_OFFSET)
433  {
434    tmp_buf[0] = (uint8)((vk->data_off >> 3) & 0xFF);
435    tmp_buf[1] = (uint8)((vk->data_off >> 2) & 0xFF);
436    tmp_buf[2] = (uint8)((vk->data_off >> 1) & 0xFF);
437    tmp_buf[3] = (uint8)(vk->data_off & 0xFF);
438    if(size > 4)
439      size = 4;
440    quoted_value = data_to_ascii(tmp_buf, 4, vk->type);
441  }
442  else
443  {
444    /* XXX: This is a safety hack.  No data fields have yet been found
445     * larger, but length limits are probably better got from fields
446     * in the registry itself, within reason.
447     */
448    if(size > 16384)
449    {
450      fprintf(stderr, "WARNING: key size %d larger than "
451              "16384, truncating...\n", size);
452      size = 16384;
453    }
454
455    quoted_value = data_to_ascii(vk->data, vk->data_size, vk->type);
456  }
457 
458  /* XXX: Sometimes value names can be NULL in registry.  Need to
459   *      figure out why and when, and generate the appropriate output
460   *      for that condition.
461   */
462  quoted_prefix = quote_string(prefix, special_chars);
463  quoted_name = quote_string(vk->valuename, special_chars);
464 
465  if(print_security)
466    printf("%s/%s,%s,%s,,,,,\n", quoted_prefix, quoted_name,
467           regfio_type_val2str(vk->type), quoted_value);
468  else
469    printf("%s/%s,%s,%s,\n", quoted_prefix, quoted_name,
470           regfio_type_val2str(vk->type), quoted_value);
471 
472  if(quoted_value != NULL)
473    free(quoted_value);
474  if(quoted_prefix != NULL)
475    free(quoted_prefix);
476  if(quoted_name != NULL)
477    free(quoted_name);
478}
479
480
481void printValueList(REGF_NK_REC* nk, char* prefix)
482{
483  uint32 i;
484 
485  for(i=0; i < nk->num_values; i++)
486    if(!type_filter_enabled || (nk->values[i].type == type_filter))
487      printValue(&nk->values[i], prefix);
488}
489
490
491void printKey(REGF_NK_REC* k, char* full_path)
492{
493  static char empty_str[1] = "";
494  char* owner = NULL;
495  char* group = NULL;
496  char* sacl = NULL;
497  char* dacl = NULL;
498  char mtime[20];
499  time_t tmp_time[1];
500  struct tm* tmp_time_s = NULL;
501  char* quoted_path = NULL;
502
503  /* XXX: it would be faster to always pass a quoted string to this
504   *      function, instead of re-quoting it every call. */
505  /* XXX: check return of quote_string */
506  quoted_path = quote_string(full_path, special_chars);
507  *tmp_time = nt_time_to_unix(&k->mtime);
508  tmp_time_s = gmtime(tmp_time);
509  strftime(mtime, sizeof(mtime), "%Y-%m-%d %H:%M:%S", tmp_time_s);
510
511  if(print_security)
512  {
513    owner = regfio_get_owner(k->sec_desc->sec_desc);
514    group = regfio_get_group(k->sec_desc->sec_desc);
515    sacl = regfio_get_sacl(k->sec_desc->sec_desc);
516    dacl = regfio_get_dacl(k->sec_desc->sec_desc);
517    if(owner == NULL)
518      owner = empty_str;
519    if(group == NULL)
520      group = empty_str;
521    if(sacl == NULL)
522      sacl = empty_str;
523    if(dacl == NULL)
524      dacl = empty_str;
525
526    printf("%s,KEY,,%s,%s,%s,%s,%s\n", quoted_path, mtime, 
527           owner, group, sacl, dacl);
528
529    if(owner != empty_str)
530      free(owner);
531    if(group != empty_str)
532      free(group);
533    if(sacl != empty_str)
534      free(sacl);
535    if(dacl != empty_str)
536      free(dacl);
537  }
538  else
539    printf("%s,KEY,,%s\n", quoted_path, mtime);
540
541  free(quoted_path);
542}
543
544
545/* XXX: this function is awful.  Needs to be re-designed. */
546void printKeyTree(REGF_FILE* f, void_stack* nk_stack, char* prefix)
547{
548  REGF_NK_REC* cur = NULL;
549  REGF_NK_REC* sub = NULL;
550  char* path = NULL;
551  char* val_path = NULL;
552  int key_type = regfio_type_str2val("KEY");
553 
554  if((cur = (REGF_NK_REC*)void_stack_cur(nk_stack)) != NULL)
555  {
556    cur->subkey_index = 0;
557    path = stack2Path(nk_stack);
558
559    if(print_verbose)
560    {
561      if(prefix[0] == '\0')
562        fprintf(stderr, "VERBOSE: Printing key tree under path: /\n");
563      else
564        fprintf(stderr, "VERBOSE: Printing key tree under path: %s\n",
565                prefix);
566    }
567
568    val_path = (char*)malloc(strlen(prefix)+strlen(path)+1);
569    sprintf(val_path, "%s%s", prefix, path);
570    if(val_path[0] == '\0')
571    {
572      val_path[0] = '/';
573      val_path[1] = '\0';
574    }
575
576    if(!type_filter_enabled || (key_type == type_filter))
577      printKey(cur, val_path);
578    if(!type_filter_enabled || (key_type != type_filter))
579      printValueList(cur, val_path);
580    if(val_path != NULL)
581      free(val_path);
582    while((cur = (REGF_NK_REC*)void_stack_cur(nk_stack)) != NULL)
583    {
584      if((sub = regfio_fetch_subkey(f, cur)) != NULL)
585      {
586        sub->subkey_index = 0;
587        void_stack_push(nk_stack, sub);
588        path = stack2Path(nk_stack);
589        if(path != NULL)
590        {
591          val_path = (char*)malloc(strlen(prefix)+strlen(path)+1);
592          sprintf(val_path, "%s%s", prefix, path);
593          if(!type_filter_enabled || (key_type == type_filter))
594            printKey(sub, val_path);
595          if(!type_filter_enabled || (key_type != type_filter))
596            printValueList(sub, val_path);
597          if(val_path != NULL)
598            free(val_path);
599        }
600      }
601      else
602      {
603        cur = void_stack_pop(nk_stack);
604        /* XXX: This is just a shallow free.  Need to write deep free
605         * routines to replace the Samba code for this.
606         */ 
607        if(cur != NULL)
608          free(cur);
609      }
610    }
611  }
612  if(print_verbose)
613    fprintf(stderr, "VERBOSE: Finished printing key tree.\n");
614}
615
616
617/*
618 * Returns 0 if path was found.
619 * Returns 1 if path was not found.
620 * Returns less than 0 on other error.
621 */
622int retrievePath(REGF_FILE* f, void_stack* nk_stack,
623                 void_stack* path_stack)
624{
625  REGF_NK_REC* sub = NULL; 
626  REGF_NK_REC* cur = NULL;
627  void_stack* sub_nk_stack;
628  char* prefix;
629  uint32 prefix_len;
630  char* cur_str = NULL;
631  char* path = NULL;
632  bool found_cur = true;
633  uint32 i;
634  uint16 path_depth;
635  if(path_stack == NULL)
636    return -1;
637
638  path_depth = void_stack_size(path_stack);
639  if(path_depth < 1)
640    return -2;
641
642  if(void_stack_size(nk_stack) < 1)
643    return -3;
644  cur = (REGF_NK_REC*)void_stack_cur(nk_stack);
645
646  if(print_verbose)
647    fprintf(stderr, "VERBOSE: Beginning retrieval of specified path: %s\n", 
648            path_filter);
649
650  while(void_stack_size(path_stack) > 1)
651  {
652    /* Search key records only */
653    cur_str = (char*)void_stack_pop(path_stack);
654
655    found_cur = false;
656    while(!found_cur &&
657          (sub = regfio_fetch_subkey(f, cur)) != NULL)
658    {
659      if(strcasecmp(sub->keyname, cur_str) == 0)
660      {
661        cur = sub;
662        void_stack_push(nk_stack, sub);
663        found_cur = true;
664      }
665    }
666    if(print_verbose && !found_cur)
667      fprintf(stderr, "VERBOSE: Could not find KEY '%s' in specified path.\n", 
668              cur_str);
669
670    free(cur_str);
671    if(!found_cur)
672      return 1;
673  }
674
675  /* Last round, search value and key records */
676  cur_str = (char*)void_stack_pop(path_stack);
677
678  if(print_verbose)
679    fprintf(stderr, "VERBOSE: Searching values for last component"
680                    " of specified path.\n");
681
682  for(i=0; (i < cur->num_values); i++)
683  {
684    /* XXX: Not sure when/why this can be NULL, but it's happened. */
685    if(sub->values[i].valuename != NULL 
686       && strcasecmp(sub->values[i].valuename, cur_str) == 0)
687    {
688      path = stack2Path(nk_stack);
689
690      if(print_verbose)
691        fprintf(stderr, "VERBOSE: Found final path element as value.\n");
692
693      if(!type_filter_enabled || (sub->values[i].type == type_filter))
694        printValue(&sub->values[i], path);
695      if(path != NULL)
696        free(path);
697
698      return 0;
699    }
700  }
701
702  if(print_verbose)
703    fprintf(stderr, "VERBOSE: Searching keys for last component"
704                    " of specified path.\n");
705
706  while((sub = regfio_fetch_subkey(f, cur)) != NULL)
707  {
708    if(strcasecmp(sub->keyname, cur_str) == 0)
709    {
710      sub_nk_stack = void_stack_new(1024);
711      void_stack_push(sub_nk_stack, sub);
712      prefix = stack2Path(nk_stack);
713      prefix_len = strlen(prefix);
714      prefix = realloc(prefix, prefix_len+strlen(sub->keyname)+2);
715      if(prefix == NULL)
716        return -1;
717      strcat(prefix, "/");
718      strcat(prefix, sub->keyname);
719
720      if(print_verbose)
721        fprintf(stderr, "VERBOSE: Found final path element as key.\n");
722
723      printKeyTree(f, sub_nk_stack, prefix);
724
725      return 0;
726    }
727  }
728
729  if(print_verbose)
730    fprintf(stderr, "VERBOSE: Could not find last element of path.\n");
731
732  return 1;
733}
734
735
736static void usage(void)
737{
738  fprintf(stderr, "Usage: reglookup [-v] [-s]"
739          " [-p <PATH_FILTER>] [-t <TYPE_FILTER>]"
740          " <REGISTRY_FILE>\n");
741  /* XXX: replace version string with Subversion property? */
742  fprintf(stderr, "Version: 0.3.0\n");
743  fprintf(stderr, "Options:\n");
744  fprintf(stderr, "\t-v\t sets verbose mode.\n");
745  fprintf(stderr, "\t-h\t enables header row. (default)\n");
746  fprintf(stderr, "\t-H\t disables header row.\n");
747  fprintf(stderr, "\t-s\t enables security descriptor output.\n");
748  fprintf(stderr, "\t-S\t disables security descriptor output. (default)\n");
749  fprintf(stderr, "\t-p\t restrict output to elements below this path.\n");
750  fprintf(stderr, "\t-t\t restrict results to this specific data type.\n");
751  fprintf(stderr, "\n");
752}
753
754
755int main(int argc, char** argv)
756{
757  void_stack* nk_stack;
758  void_stack* path_stack;
759  REGF_FILE* f;
760  REGF_NK_REC* root;
761  int retr_path_ret;
762  uint32 argi, arge;
763
764  /* Process command line arguments */
765  if(argc < 2)
766  {
767    usage();
768    bailOut(1, "ERROR: Requires at least one argument.\n");
769  }
770 
771  arge = argc-1;
772  for(argi = 1; argi < arge; argi++)
773  {
774    if (strcmp("-p", argv[argi]) == 0)
775    {
776      if(++argi >= arge)
777      {
778        usage();
779        bailOut(1, "ERROR: '-p' option requires parameter.\n");
780      }
781      if((path_filter = strdup(argv[argi])) == NULL)
782        bailOut(2, "ERROR: Memory allocation problem.\n");
783
784      path_filter_enabled = true;
785    }
786    else if (strcmp("-t", argv[argi]) == 0)
787    {
788      if(++argi >= arge)
789      {
790        usage();
791        bailOut(1, "ERROR: '-t' option requires parameter.\n");
792      }
793      if((type_filter = regfio_type_str2val(argv[argi])) < 0)
794      {
795        fprintf(stderr, "ERROR: Invalid type specified: %s.\n", argv[argi]);
796        bailOut(1, "");
797      }
798      type_filter_enabled = true;
799    }
800    else if (strcmp("-h", argv[argi]) == 0)
801      print_header = true;
802    else if (strcmp("-H", argv[argi]) == 0)
803      print_header = false;
804    else if (strcmp("-s", argv[argi]) == 0)
805      print_security = true;
806    else if (strcmp("-S", argv[argi]) == 0)
807      print_security = false;
808    else if (strcmp("-v", argv[argi]) == 0)
809      print_verbose = true;
810    else
811    {
812      usage();
813      fprintf(stderr, "ERROR: Unrecognized option: %s\n", argv[argi]);
814      bailOut(1, "");
815    }
816  }
817  if((registry_file = strdup(argv[argi])) == NULL)
818    bailOut(2, "ERROR: Memory allocation problem.\n");
819
820  f = regfio_open(registry_file);
821  if(f == NULL)
822  {
823    fprintf(stderr, "ERROR: Couldn't open registry file: %s\n", registry_file);
824    bailOut(3, "");
825  }
826
827  root = regfio_rootkey(f);
828  nk_stack = void_stack_new(1024);
829
830  if(void_stack_push(nk_stack, root))
831  {
832    if(print_header)
833    {
834      if(print_security)
835        printf("PATH,TYPE,VALUE,MTIME,OWNER,GROUP,SACL,DACL\n");
836      else
837        printf("PATH,TYPE,VALUE,MTIME\n");
838    }
839
840    path_stack = path2Stack(path_filter);
841    if(void_stack_size(path_stack) < 1)
842      printKeyTree(f, nk_stack, "");
843    else
844    {
845      retr_path_ret = retrievePath(f, nk_stack, path_stack);
846      if(retr_path_ret == 1)
847        fprintf(stderr, "WARNING: specified path not found.\n");
848      else if(retr_path_ret != 0)
849        bailOut(4, "ERROR:\n");
850    }
851  }
852  else
853    bailOut(2, "ERROR: Memory allocation problem.\n");
854
855  void_stack_destroy_deep(nk_stack);
856  regfio_close(f);
857
858  return 0;
859}
Note: See TracBrowser for help on using the repository browser.