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