c3633ef0af19a25603a5368ec1b1eab0a98befc1
[kai/samba.git] / source4 / lib / registry / tools / regdiff.c
1 /* 
2    Unix SMB/CIFS implementation.
3    simple registry frontend
4    
5    Copyright (C) Jelmer Vernooij 2004
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
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
22 #include "includes.h"
23
24 void writediff(REG_KEY *oldkey, REG_KEY *newkey, FILE *out)
25 {
26         int i;
27         REG_KEY *t1,*t2;
28         REG_VAL *v1, *v2;
29         WERROR error1, error2;
30
31         for(i = 0; W_ERROR_IS_OK(error1 = reg_key_get_subkey_by_index(oldkey, i, &t1)); i++) {
32                 error2 = reg_key_get_subkey_by_name(newkey, reg_key_name(t1), &t2);
33                 if(W_ERROR_EQUAL(error2, WERR_DEST_NOT_FOUND)) {
34                         fprintf(out, "-%s\n", reg_key_get_path(t1)+1);
35                 } else if(!W_ERROR_IS_OK(error2)) {
36                         DEBUG(0, ("Error occured while getting subkey by name: %d\n", W_ERROR_V(error2)));
37                 }
38         }
39
40         if(!W_ERROR_EQUAL(error1, WERR_NO_MORE_ITEMS)) {
41                 DEBUG(0, ("Error occured while getting subkey by index: %d\n", W_ERROR_V(error1)));
42                 return;
43         }
44
45         for(i = 0; W_ERROR_IS_OK(error1 = reg_key_get_subkey_by_index(newkey, i, &t1)); i++) {
46                 error2 = reg_key_get_subkey_by_name(oldkey, reg_key_name(t1), &t2);
47                 if(W_ERROR_EQUAL(error2, WERR_DEST_NOT_FOUND)) {
48                         fprintf(out, "\n[%s]\n", reg_key_get_path(t1)+1);
49                 } else if(!W_ERROR_IS_OK(error2)) {
50                         DEBUG(0, ("Error occured while getting subkey by name: %d\n", W_ERROR_V(error2)));
51                 }
52                 writediff(t2, t1, out);
53         }
54
55         if(!W_ERROR_EQUAL(error1, WERR_NO_MORE_ITEMS)) {
56                 DEBUG(0, ("Error occured while getting subkey by index: %d\n", W_ERROR_V(error1)));
57                 return;
58         }
59
60         for(i = 0; W_ERROR_IS_OK(error1 = reg_key_get_value_by_index(newkey, i, &v1)); i++) {
61                 error2 = reg_key_get_value_by_name(oldkey, reg_val_name(v1), &v2);
62                 if (((W_ERROR_IS_OK(error2) && reg_val_size(v2) != reg_val_size(v1)) || memcmp(reg_val_data_blk(v1), reg_val_data_blk(v2), reg_val_size(v1))) 
63                         || W_ERROR_EQUAL(error2, WERR_DEST_NOT_FOUND)) {
64                         fprintf(out, "\"%s\"=%s:%s\n", reg_val_name(v1), str_regtype(reg_val_type(v1)), reg_val_data_string(v1));
65                 }
66
67                 if(!W_ERROR_IS_OK(error2) && !W_ERROR_EQUAL(error2, WERR_DEST_NOT_FOUND)) {
68                         DEBUG(0, ("Error occured while getting value by name: %d\n", W_ERROR_V(error2)));
69                 }
70         }
71
72         if(!W_ERROR_EQUAL(error1, WERR_NO_MORE_ITEMS)) {
73                 DEBUG(0, ("Error occured while getting value by index: %d\n", W_ERROR_V(error1)));
74                 return;
75         }
76
77
78         for(i = 0; W_ERROR_IS_OK(error1 = reg_key_get_value_by_index(oldkey, i, &v1)); i++) {
79                 error2 = reg_key_get_value_by_name(newkey, reg_val_name(v1), &v2);
80                 if(W_ERROR_IS_OK(error2)) {
81                 } else if(W_ERROR_EQUAL(error2, WERR_DEST_NOT_FOUND)) {
82                         fprintf(out, "\"%s\"=-\n", reg_val_name(v1));
83                 } else {
84                         DEBUG(0, ("Error occured while getting value by name: %d\n", W_ERROR_V(error2)));
85                 }
86         }
87
88         if(!W_ERROR_EQUAL(error1, WERR_NO_MORE_ITEMS)) {
89                 DEBUG(0, ("Error occured while getting value by index: %d\n", W_ERROR_V(error1)));
90                 return;
91         }
92 }
93
94  int main(int argc, char **argv)
95 {
96         int opt;
97         poptContext pc;
98         const char *backend1 = NULL, *backend2 = NULL;
99         const char *location2;
100         const char *credentials1= NULL, *credentials2 = NULL;
101         char *outputfile = NULL;
102         FILE *fd = stdout;
103         REG_HANDLE *h2;
104         REG_KEY *root1 = NULL, *root2;
105         int from_null = 0;
106         WERROR error;
107         struct poptOption long_options[] = {
108                 POPT_AUTOHELP
109                 {"backend", 'b', POPT_ARG_STRING, NULL, 'b', "backend to use", NULL},
110                 {"credentials", 'c', POPT_ARG_STRING, NULL, 'c', "credentials", NULL},
111                 {"output", 'o', POPT_ARG_STRING, &outputfile, 'o', "output file to use", NULL },
112                 {"null", 'n', POPT_ARG_NONE, &from_null, 'n', "Diff from NULL" },
113                 POPT_TABLEEND
114         };
115
116         pc = poptGetContext(argv[0], argc, (const char **) argv, long_options,0);
117         
118         while((opt = poptGetNextOpt(pc)) != -1) {
119                 switch(opt)     {
120                 case 'c':
121                         if(!credentials1 && !from_null) credentials1 = poptGetOptArg(pc);
122                         else if(!credentials2) credentials2 = poptGetOptArg(pc);
123                         break;
124                 case 'b':
125                                 if(!backend1 && !from_null) backend1 = poptGetOptArg(pc);
126                                 else if(!backend2) backend2 = poptGetOptArg(pc);
127                                 break;
128                 }
129         }
130         setup_logging(argv[0], True);
131
132         if(!from_null) {
133                 REG_HANDLE *h1;
134                 const char *location1;
135                 location1 = poptGetArg(pc);
136                 if(!location1) {
137                         poptPrintUsage(pc, stderr, 0);
138                         return 1;
139                 }
140
141                 if(!backend1) backend1 = "dir";
142
143                 error = reg_open(backend1, location1, credentials1, &h1);
144                 if(!W_ERROR_IS_OK(error)) {
145                         fprintf(stderr, "Unable to open '%s' with backend '%s'\n", location1, backend1);
146                         return 1;
147                 }
148
149                 if(!W_ERROR_IS_OK(reg_get_root(h1, &root1))) return 1;
150         }
151
152         location2 = poptGetArg(pc);
153         if(!location2) {
154                 poptPrintUsage(pc, stderr, 0);
155                 return 2;
156         }
157
158         if(!backend2) backend2 = "dir";
159         
160         error = reg_open(backend2, location2, credentials2, &h2);
161         if(!W_ERROR_IS_OK(error)) {
162                 fprintf(stderr, "Unable to open '%s' with backend '%s'\n", location2, backend2);
163                 return 1;
164         }
165         
166         error = reg_get_root(h2, &root2);
167         if(!W_ERROR_IS_OK(error)) {
168                 fprintf(stderr, "Can't open root key for '%s:%s'\n", backend2, location2);
169                 return 1;
170         }
171
172         poptFreeContext(pc);
173
174         if(outputfile) {
175                 fd = fopen(outputfile, "w+");
176                 if(!fd) {
177                         fprintf(stderr, "Unable to open '%s'\n", outputfile);
178                         return 1;
179                 }
180         }
181
182         fprintf(fd, "REGEDIT4\n\n");
183         fprintf(fd, "; Generated using regdiff\n");
184
185         writediff(root1, root2, fd); 
186
187         fclose(fd);
188         
189         return 0;
190 }