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 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, &sDefault, section->name);
 263                                 lp_do_service_parameter(lp_ctx,
 264                                                         service,
 265                                                         param->key,
 266                                                         param->value);
 267                         }
 268                 }
 269         }
 270         return 0;
 271 }
 272
 273 int param_write(struct param_context *ctx, const char *fn)
 274 {
 275         int file;
 276         struct param_section *section;
 277
 278         if (fn == NULL || ctx == NULL)
 279                 return -1;
 280
 281         file = open(fn, O_WRONLY|O_CREAT, 0755);
 282
 283         if (file == -1)
 284                 return -1;
 285
 286         for (section = ctx->sections; section; section = section->next) {
 287                 struct param_opt *param;
 288
 289                 fdprintf(file, "[%s]\n", section->name);
 290                 for (param = section->parameters; param; param = param->next) {
 291                         fdprintf(file, "\t%s = %s\n", param->key, param->value);
 292                 }
 293                 fdprintf(file, "\n");
 294         }
 295
 296         close(file);
 297
 298         return 0;
 299 }