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 2 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, write to the Free Software
  17  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19
  20 #include "includes.h"
  21 #include "dlinklist.h"
  22 #include "param/generic.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 *param_section_get (struct param_section *section, const char *name)
  41 {
  42         struct param *p;
  43
  44         for (p = section->parameters; p; p = p->next) {
  45                 if (strcasecmp_m(p->name, name) == 0)
  46                         return p;
  47         }
  48
  49         return NULL;
  50 }
  51
  52 struct param *param_get (struct param_context *ctx, const char *section_name, const char *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 /* Look up parameter. If it is not found, add it */
  62 static struct param *param_get_add(struct param_context *ctx, const char *section_name, const char *name)
  63 {
  64         struct param_section *section;
  65         struct param *p;
  66
  67         section = param_get_section(ctx, section_name);
  68
  69         if (section == NULL) {
  70                 section = talloc_zero(ctx, struct param_section);
  71                 section->name = talloc_strdup(section, section_name);
  72                 DLIST_ADD(ctx->sections, section);
  73         }
  74
  75         p = param_section_get(section, name);
  76         if (p == NULL) {
  77                 p = talloc_zero(section, struct param);
  78                 p->name = talloc_strdup(p, name);
  79                 DLIST_ADD(section->parameters, p);
  80         }
  81
  82         return p;
  83 }
  84
  85 const char *param_get_string(struct param_context *ctx, const char *section, const char *param)
  86 {
  87         struct param *p = param_get(ctx, section, param);
  88
  89         if (p == NULL)
  90                 return NULL;
  91
  92         return p->value;
  93 }
  94
  95 int param_set_string(struct param_context *ctx, const char *section, const char *param, const char *value)
  96 {
  97         struct param *p = param_get_add(ctx, section, param);
  98
  99         if (p == NULL)
 100                 return -1;
 101
 102         p->value = talloc_strdup(p, value);
 103
 104         return 0;
 105 }
 106
 107 const char **param_get_string_list(struct param_context *ctx, const char *section, const char *param,
 108                                  const char *separator)
 109 {
 110         struct param *p = param_get(ctx, section, param);
 111
 112         if (p == NULL)
 113                 return NULL;
 114
 115         if (separator == NULL)
 116                 separator = LIST_SEP;
 117
 118         if (p->list_value == NULL) {
 119                 p->list_value = str_list_make(ctx, p->value, separator);
 120         }
 121
 122         return p->list_value;
 123 }
 124
 125 int param_set_string_list(struct param_context *ctx, const char *section, const char *param, const char **list)
 126 {
 127         struct param *p = param_get_add(ctx, section, param);
 128
 129         p->value = str_list_join(p, list, ' ');
 130         p->list_value = str_list_copy(p, list);
 131
 132         return 0;
 133 }
 134
 135 int param_get_int(struct param_context *ctx, const char *section, const char *param, int default_v)
 136 {
 137         const char *value = param_get_string(ctx, section, param);
 138
 139         if (value)
 140                 return strtol(value, NULL, 0);
 141
 142         return default_v;
 143 }
 144
 145 void param_set_int(struct param_context *ctx, const char *section, const char *param, int value)
 146 {
 147         struct param *p = param_get_add(ctx, section, param);
 148
 149         p->value = talloc_asprintf(p, "%d", value);
 150 }
 151
 152 unsigned long param_get_ulong(struct param_context *ctx, const char *section, const char *param, unsigned long default_v)
 153 {
 154         const char *value = param_get_string(ctx, section, param);
 155
 156         if (value)
 157                 return strtoul(value, NULL, 0);
 158
 159         return default_v;
 160 }
 161
 162 void param_set_ulong(struct param_context *ctx, const char *section, const char *name, unsigned long value)
 163 {
 164         struct param *p = param_get_add(ctx, section, name);
 165
 166         p->value = talloc_asprintf(p, "%lu", value);
 167 }
 168
 169 static BOOL param_sfunc (const char *name, void *_ctx)
 170 {
 171         struct param_context *ctx = _ctx;
 172         struct param_section *section = param_get_section(ctx, name);
 173
 174         if (section == NULL) {
 175                 section = talloc_zero(ctx, struct param_section);
 176                 section->name = talloc_strdup(section, name);
 177
 178                 DLIST_ADD(ctx->sections, section);
 179         }
 180
 181         /* Make sure this section is on top of the list for param_pfunc */
 182         DLIST_PROMOTE(ctx->sections, section);
 183
 184         return True;
 185 }
 186
 187 static BOOL param_pfunc (const char *name, const char *value, void *_ctx)
 188 {
 189         struct param_context *ctx = _ctx;
 190         struct param *p = param_section_get(ctx->sections, name);
 191
 192         if (!p) {
 193                 p = talloc_zero(ctx->sections, struct param);
 194                 p->name = talloc_strdup(p, name);
 195                 p->value = talloc_strdup(p, value);
 196                 DLIST_ADD(ctx->sections->parameters, p);
 197         } else { /* Replace current value */
 198                 talloc_free(p->value);
 199                 p->value = talloc_strdup(p, value);
 200         }
 201
 202         return True;
 203 }
 204
 205 struct param_context *param_init(TALLOC_CTX *mem_ctx)
 206 {
 207         return talloc_zero(mem_ctx, struct param_context);
 208 }
 209
 210
 211 int param_read(struct param_context *ctx, const char *fn)
 212 {
 213         ctx->sections = talloc_zero(ctx, struct param_section);
 214         ctx->sections->name = talloc_strdup(ctx->sections, "global");
 215
 216         if (!pm_process( fn, param_sfunc, param_pfunc, ctx)) {
 217                 return -1;
 218         }
 219
 220         return 0;
 221 }
 222
 223 int param_write(struct param_context *ctx, const char *fn)
 224 {
 225         XFILE *file;
 226         struct param_section *section;
 227
 228         if (fn == NULL || ctx == NULL)
 229                 return -1;
 230
 231         file = x_fopen(fn, O_WRONLY|O_CREAT, 0755);
 232
 233         if (file == NULL)
 234                 return -1;
 235
 236         for (section = ctx->sections; section; section = section->next) {
 237                 struct param *param;
 238
 239                 x_fprintf(file, "[%s]\n", section->name);
 240                 for (param = section->parameters; param; param = param->next) {
 241                         x_fprintf(file, "\t%s = %s\n", param->name, param->value);
 242                 }
 243                 x_fprintf(file, "\n");
 244         }
 245
 246         x_fclose(file);
 247
 248         return 0;
 249 }