ed3045605db6b3ef43e81e3ad4f93c5b0bd9ed07
[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 "system/filesys.h"
23
24 struct param_section *param_get_section(struct param_context *ctx, const char *name)
25 {
26         struct param_section *sect;
27
28         if (name == NULL) 
29                 name = GLOBAL_NAME;
30
31         for (sect = ctx->sections; sect; sect = sect->next) {
32                 if (!strcasecmp_m(sect->name, name)) 
33                         return sect;
34         }
35
36         return NULL;
37 }
38
39 struct param_opt *param_section_get(struct param_section *section, 
40                                     const char *name)
41 {
42         struct param_opt *p;
43
44         for (p = section->parameters; p; p = p->next) {
45                 if (strcasecmp_m(p->key, name) == 0) 
46                         return p;
47         }
48
49         return NULL;
50 }
51
52 struct param_opt *param_get (struct param_context *ctx, const char *name, const char *section_name)
53 {
54         struct param_section *section = param_get_section(ctx, section_name);
55         if (section == NULL)
56                 return NULL;
57
58         return param_section_get(section, name);
59 }
60
61 struct param_section *param_add_section(struct param_context *ctx, const char *section_name)
62 {
63         struct param_section *section;
64         section = talloc_zero(ctx, struct param_section);
65         if (section == NULL)
66                 return NULL;
67
68         section->name = talloc_strdup(section, section_name);
69         DLIST_ADD_END(ctx->sections, section, struct param_section *);
70         return section;
71 }
72
73 /* Look up parameter. If it is not found, add it */
74 struct param_opt *param_get_add(struct param_context *ctx, const char *name, const char *section_name)
75 {
76         struct param_section *section;
77         struct param_opt *p;
78
79         SMB_ASSERT(section_name != NULL);
80         SMB_ASSERT(name != NULL);
81
82         section = param_get_section(ctx, section_name);
83
84         if (section == NULL) {
85                 section = param_add_section(ctx, section_name);
86         }
87
88         p = param_section_get(section, name);
89         if (p == NULL) {
90                 p = talloc_zero(section, struct param_opt);
91                 if (p == NULL)
92                         return NULL;
93
94                 p->key = talloc_strdup(p, name);
95                 DLIST_ADD_END(section->parameters, p, struct param_opt *);
96         }
97         
98         return p;
99 }
100
101 const char *param_get_string(struct param_context *ctx, const char *param, const char *section)
102 {
103         struct param_opt *p = param_get(ctx, param, section);
104
105         if (p == NULL)
106                 return NULL;
107
108         return p->value;
109 }
110
111 int param_set_string(struct param_context *ctx, const char *param, const char *value, const char *section)
112 {
113         struct param_opt *p = param_get_add(ctx, param, section);
114
115         if (p == NULL)
116                 return -1;
117
118         p->value = talloc_strdup(p, value);
119
120         return 0;
121 }
122
123 const char **param_get_string_list(struct param_context *ctx, const char *param, const char *separator, const char *section)
124 {
125         struct param_opt *p = param_get(ctx, param, section);
126         
127         if (p == NULL)
128                 return NULL;
129
130         if (separator == NULL)
131                 separator = LIST_SEP;
132         
133         return (const char **)str_list_make(ctx, p->value, separator);
134 }
135
136 int param_set_string_list(struct param_context *ctx, const char *param, const char **list, const char *section)
137 {
138         struct param_opt *p = param_get_add(ctx, param, section);       
139
140         p->value = str_list_join(p, list, ' ');
141
142         return 0;
143 }
144
145 int param_get_int(struct param_context *ctx, const char *param, int default_v, const char *section)
146 {
147         const char *value = param_get_string(ctx, param, section);
148         
149         if (value)
150                 return strtol(value, NULL, 0); 
151
152         return default_v;
153 }
154
155 void param_set_int(struct param_context *ctx, const char *param, int value, const char *section)
156 {
157         struct param_opt *p = param_get_add(ctx, section, param);
158
159         if (!p) 
160                 return;
161
162         p->value = talloc_asprintf(p, "%d", value);
163 }
164
165 unsigned long param_get_ulong(struct param_context *ctx, const char *param, unsigned long default_v, const char *section)
166 {
167         const char *value = param_get_string(ctx, param, section);
168         
169         if (value)
170                 return strtoul(value, NULL, 0);
171
172         return default_v;
173 }
174
175 void param_set_ulong(struct param_context *ctx, const char *name, unsigned long value, const char *section)
176 {
177         struct param_opt *p = param_get_add(ctx, name, section);
178
179         if (!p)
180                 return;
181
182         p->value = talloc_asprintf(p, "%lu", value);
183 }
184
185 static bool param_sfunc (const char *name, void *_ctx)
186 {
187         struct param_context *ctx = (struct param_context *)_ctx;
188         struct param_section *section = param_get_section(ctx, name);
189
190         if (section == NULL) {
191                 section = talloc_zero(ctx, struct param_section);
192                 if (section == NULL)
193                         return false;
194
195                 section->name = talloc_strdup(section, name);
196
197                 DLIST_ADD_END(ctx->sections, section, struct param_section *);
198         }
199
200         /* Make sure this section is on top of the list for param_pfunc */
201         DLIST_PROMOTE(ctx->sections, section);
202
203         return true;
204 }
205
206 static bool param_pfunc (const char *name, const char *value, void *_ctx)
207 {
208         struct param_context *ctx = (struct param_context *)_ctx;
209         struct param_opt *p = param_section_get(ctx->sections, name);
210
211         if (!p) {
212                 p = talloc_zero(ctx->sections, struct param_opt);
213                 if (p == NULL)
214                         return false;
215
216                 p->key = talloc_strdup(p, name);
217                 p->value = talloc_strdup(p, value);
218                 DLIST_ADD(ctx->sections->parameters, p);
219         } else { /* Replace current value */
220                 talloc_free(p->value);
221                 p->value = talloc_strdup(p, value);
222         }
223
224         return true;
225 }
226
227 struct param_context *param_init(TALLOC_CTX *mem_ctx)
228 {
229         return talloc_zero(mem_ctx, struct param_context);
230 }
231
232
233 int param_read(struct param_context *ctx, const char *fn)
234 {
235         ctx->sections = talloc_zero(ctx, struct param_section);
236         if (ctx->sections == NULL)
237                 return -1;
238
239         ctx->sections->name = talloc_strdup(ctx->sections, "global");
240         if (!pm_process( fn, param_sfunc, param_pfunc, ctx)) {
241                 return -1;
242         }
243
244         return 0;
245 }
246
247 int param_use(struct loadparm_context *lp_ctx, struct param_context *ctx)
248 {
249         struct param_section *section;
250
251         for (section = ctx->sections; section; section = section->next) {
252                 struct param_opt *param;
253                 bool isglobal = strcmp(section->name, "global") == 0;
254                 for (param = section->parameters; param; param = param->next) {
255                         if (isglobal)
256                                 lp_do_global_parameter(lp_ctx, param->key,
257                                                        param->value);
258                         else {
259                                 struct loadparm_service *service = 
260                                                         lp_service(lp_ctx, section->name);
261                                 if (service == NULL)
262                                         service = lp_add_service(lp_ctx, lp_default_service(lp_ctx), section->name);
263                                 lp_do_service_parameter(lp_ctx, service, param->key, param->value);
264                         }
265                 }
266         }
267         return 0;
268 }
269
270 int param_write(struct param_context *ctx, const char *fn)
271 {
272         int file;
273         struct param_section *section;
274
275         if (fn == NULL || ctx == NULL)
276                 return -1;
277
278         file = open(fn, O_WRONLY|O_CREAT, 0755);
279
280         if (file == -1)
281                 return -1;
282         
283         for (section = ctx->sections; section; section = section->next) {
284                 struct param_opt *param;
285                 
286                 fdprintf(file, "[%s]\n", section->name);
287                 for (param = section->parameters; param; param = param->next) {
288                         fdprintf(file, "\t%s = %s\n", param->key, param->value);
289                 }
290                 fdprintf(file, "\n");
291         }
292
293         close(file);
294
295         return 0;
296 }