ba5464a0f0aabf8166887073e17366630ec64126
[kai/samba.git] / source4 / param / generic.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  Copyright (C) Jelmer Vernooij                       2005
4  *  
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; either version 3 of the License, or
8  *  (at your option) any later version.
9  *  
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU General Public License for more details.
14  *  
15  *  You should have received a copy of the GNU General Public License
16  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
17  */
18
19 #include "includes.h"
20 #include "../lib/util/dlinklist.h"
21 #include "param/param.h"
22 #include "param/loadparm.h"
23 #include "system/filesys.h"
24
25 struct param_section *param_get_section(struct param_context *ctx, const char *name)
26 {
27         struct param_section *sect;
28
29         if (name == NULL) 
30                 name = GLOBAL_NAME;
31
32         for (sect = ctx->sections; sect; sect = sect->next) {
33                 if (!strcasecmp_m(sect->name, name)) 
34                         return sect;
35         }
36
37         return NULL;
38 }
39
40 struct param_opt *param_section_get(struct param_section *section, 
41                                     const char *name)
42 {
43         struct param_opt *p;
44
45         for (p = section->parameters; p; p = p->next) {
46                 if (strcasecmp_m(p->key, name) == 0) 
47                         return p;
48         }
49
50         return NULL;
51 }
52
53 struct param_opt *param_get (struct param_context *ctx, const char *name, const char *section_name)
54 {
55         struct param_section *section = param_get_section(ctx, section_name);
56         if (section == NULL)
57                 return NULL;
58
59         return param_section_get(section, name);
60 }
61
62 struct param_section *param_add_section(struct param_context *ctx, const char *section_name)
63 {
64         struct param_section *section;
65         section = talloc_zero(ctx, struct param_section);
66         if (section == NULL)
67                 return NULL;
68
69         section->name = talloc_strdup(section, section_name);
70         DLIST_ADD_END(ctx->sections, section, struct param_section *);
71         return section;
72 }
73
74 /* Look up parameter. If it is not found, add it */
75 struct param_opt *param_get_add(struct param_context *ctx, const char *name, const char *section_name)
76 {
77         struct param_section *section;
78         struct param_opt *p;
79
80         SMB_ASSERT(section_name != NULL);
81         SMB_ASSERT(name != NULL);
82
83         section = param_get_section(ctx, section_name);
84
85         if (section == NULL) {
86                 section = param_add_section(ctx, section_name);
87         }
88
89         p = param_section_get(section, name);
90         if (p == NULL) {
91                 p = talloc_zero(section, struct param_opt);
92                 if (p == NULL)
93                         return NULL;
94
95                 p->key = talloc_strdup(p, name);
96                 DLIST_ADD_END(section->parameters, p, struct param_opt *);
97         }
98         
99         return p;
100 }
101
102 const char *param_get_string(struct param_context *ctx, const char *param, const char *section)
103 {
104         struct param_opt *p = param_get(ctx, param, section);
105
106         if (p == NULL)
107                 return NULL;
108
109         return p->value;
110 }
111
112 int param_set_string(struct param_context *ctx, const char *param, const char *value, const char *section)
113 {
114         struct param_opt *p = param_get_add(ctx, param, section);
115
116         if (p == NULL)
117                 return -1;
118
119         p->value = talloc_strdup(p, value);
120
121         return 0;
122 }
123
124 const char **param_get_string_list(struct param_context *ctx, const char *param, const char *separator, const char *section)
125 {
126         struct param_opt *p = param_get(ctx, param, section);
127         
128         if (p == NULL)
129                 return NULL;
130
131         return (const char **)str_list_make(ctx, p->value, separator);
132 }
133
134 int param_set_string_list(struct param_context *ctx, const char *param, const char **list, const char *section)
135 {
136         struct param_opt *p = param_get_add(ctx, param, section);       
137
138         p->value = str_list_join(p, list, ' ');
139
140         return 0;
141 }
142
143 int param_get_int(struct param_context *ctx, const char *param, int default_v, const char *section)
144 {
145         const char *value = param_get_string(ctx, param, section);
146         
147         if (value)
148                 return strtol(value, NULL, 0); 
149
150         return default_v;
151 }
152
153 void param_set_int(struct param_context *ctx, const char *param, int value, const char *section)
154 {
155         struct param_opt *p = param_get_add(ctx, section, param);
156
157         if (!p) 
158                 return;
159
160         p->value = talloc_asprintf(p, "%d", value);
161 }
162
163 unsigned long param_get_ulong(struct param_context *ctx, const char *param, unsigned long default_v, const char *section)
164 {
165         const char *value = param_get_string(ctx, param, section);
166         
167         if (value)
168                 return strtoul(value, NULL, 0);
169
170         return default_v;
171 }
172
173 void param_set_ulong(struct param_context *ctx, const char *name, unsigned long value, const char *section)
174 {
175         struct param_opt *p = param_get_add(ctx, name, section);
176
177         if (!p)
178                 return;
179
180         p->value = talloc_asprintf(p, "%lu", value);
181 }
182
183 static bool param_sfunc (const char *name, void *_ctx)
184 {
185         struct param_context *ctx = (struct param_context *)_ctx;
186         struct param_section *section = param_get_section(ctx, name);
187
188         if (section == NULL) {
189                 section = talloc_zero(ctx, struct param_section);
190                 if (section == NULL)
191                         return false;
192
193                 section->name = talloc_strdup(section, name);
194
195                 DLIST_ADD_END(ctx->sections, section, struct param_section *);
196         }
197
198         /* Make sure this section is on top of the list for param_pfunc */
199         DLIST_PROMOTE(ctx->sections, section);
200
201         return true;
202 }
203
204 static bool param_pfunc (const char *name, const char *value, void *_ctx)
205 {
206         struct param_context *ctx = (struct param_context *)_ctx;
207         struct param_opt *p = param_section_get(ctx->sections, name);
208
209         if (!p) {
210                 p = talloc_zero(ctx->sections, struct param_opt);
211                 if (p == NULL)
212                         return false;
213
214                 p->key = talloc_strdup(p, name);
215                 p->value = talloc_strdup(p, value);
216                 DLIST_ADD(ctx->sections->parameters, p);
217         } else { /* Replace current value */
218                 talloc_free(p->value);
219                 p->value = talloc_strdup(p, value);
220         }
221
222         return true;
223 }
224
225 struct param_context *param_init(TALLOC_CTX *mem_ctx)
226 {
227         return talloc_zero(mem_ctx, struct param_context);
228 }
229
230
231 int param_read(struct param_context *ctx, const char *fn)
232 {
233         ctx->sections = talloc_zero(ctx, struct param_section);
234         if (ctx->sections == NULL)
235                 return -1;
236
237         ctx->sections->name = talloc_strdup(ctx->sections, "global");
238         if (!pm_process( fn, param_sfunc, param_pfunc, ctx)) {
239                 return -1;
240         }
241
242         return 0;
243 }
244
245 int param_use(struct loadparm_context *lp_ctx, struct param_context *ctx)
246 {
247         struct param_section *section;
248
249         for (section = ctx->sections; section; section = section->next) {
250                 struct param_opt *param;
251                 bool isglobal = strcmp(section->name, "global") == 0;
252                 for (param = section->parameters; param; param = param->next) {
253                         if (isglobal)
254                                 lp_do_global_parameter(lp_ctx, param->key,
255                                                        param->value);
256                         else {
257                                 struct loadparm_service *service = 
258                                                         lp_service(lp_ctx, section->name);
259                                 if (service == NULL)
260                                         service = lp_add_service(lp_ctx, lp_default_service(lp_ctx), section->name);
261                                 lp_do_service_parameter(lp_ctx, service, param->key, param->value);
262                         }
263                 }
264         }
265         return 0;
266 }
267
268 int param_write(struct param_context *ctx, const char *fn)
269 {
270         int file;
271         struct param_section *section;
272
273         if (fn == NULL || ctx == NULL)
274                 return -1;
275
276         file = open(fn, O_WRONLY|O_CREAT, 0755);
277
278         if (file == -1)
279                 return -1;
280         
281         for (section = ctx->sections; section; section = section->next) {
282                 struct param_opt *param;
283                 
284                 fdprintf(file, "[%s]\n", section->name);
285                 for (param = section->parameters; param; param = param->next) {
286                         fdprintf(file, "\t%s = %s\n", param->key, param->value);
287                 }
288                 fdprintf(file, "\n");
289         }
290
291         close(file);
292
293         return 0;
294 }