drm-lease-manager/lease-config.c

   1 /* Copyright 2022 IGEL Co., Ltd.
   2  *
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include "lease-config.h"
  17 #include "log.h"
  18 #include <errno.h>
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22 #include <toml.h>
  23
  24 #define CONFIG_ERROR(x, ...) ERROR_LOG("%s: " x, filename, ##__VA_ARGS__)
  25
  26 static bool populate_connector_names(struct lease_config *config,
  27                                      toml_array_t *conns)
  28 {
  29         int cnames = toml_array_nelem(conns);
  30         config->connector_names = calloc(cnames, sizeof(char *));
  31         if (!config->connector_names) {
  32                 DEBUG_LOG("Memory allocation failed: %s\n", strerror(errno));
  33                 return false;
  34         }
  35
  36         config->cnames = cnames;
  37
  38         for (int i = 0; i < config->cnames; i++) {
  39                 toml_datum_t conn = toml_string_at(conns, i);
  40                 if (!conn.ok) {
  41                         return false;
  42                 }
  43                 config->connector_names[i] = conn.u.s;
  44         }
  45         return true;
  46 }
  47
  48 int parse_config(char *filename, struct lease_config **parsed_config)
  49 {
  50         struct lease_config *config = NULL;
  51         int nconfigs, i, ret = 0;
  52         char parse_error[160];
  53
  54         FILE *fp = fopen(filename, "r");
  55         if (!fp)
  56                 return 0;
  57
  58         toml_table_t *t_config =
  59             toml_parse_file(fp, parse_error, sizeof parse_error);
  60         if (!t_config) {
  61                 CONFIG_ERROR("configuration file parse error: %s\n",
  62                              parse_error);
  63                 fclose(fp);
  64                 return 0;
  65         }
  66
  67         toml_array_t *leases = toml_array_in(t_config, "lease");
  68         if (!leases) {
  69                 CONFIG_ERROR(
  70                     "Invalid config - cannot find any 'lease' configs");
  71                 goto err;
  72         }
  73         nconfigs = toml_array_nelem(leases);
  74         config = calloc(nconfigs, sizeof *config);
  75
  76         if (!config) {
  77                 DEBUG_LOG("Memory allocation failed: %s\n", strerror(errno));
  78                 goto err;
  79         }
  80
  81         for (i = 0; i < toml_array_nelem(leases); i++) {
  82                 toml_table_t *lease = toml_table_at(leases, i);
  83
  84                 toml_datum_t name = toml_string_in(lease, "name");
  85                 if (!name.ok) {
  86                         CONFIG_ERROR("Invalid lease name in entry #%d\n", i);
  87                         goto err_free_config;
  88                 }
  89
  90                 config[i].lease_name = name.u.s;
  91
  92                 toml_array_t *conns = toml_array_in(lease, "connectors");
  93                 if (conns && !populate_connector_names(&config[i], conns)) {
  94                         CONFIG_ERROR("Non string connector name in lease: %s\n",
  95                                      config[i].lease_name);
  96                         goto err_free_config;
  97                 }
  98         }
  99
 100         *parsed_config = config;
 101         ret = nconfigs;
 102 err:
 103         toml_free(t_config);
 104         fclose(fp);
 105         return ret;
 106 err_free_config:
 107         release_config(i, config);
 108         goto err;
 109 }
 110
 111 void release_config(int num_leases, struct lease_config *config)
 112 {
 113         for (int i = 0; i < num_leases; i++) {
 114                 struct lease_config *c = &config[i];
 115                 free(c->lease_name);
 116                 for (int j = 0; j < c->cnames; j++)
 117                         free(c->connector_names[j]);
 118                 free(c->connector_names);
 119         }
 120         free(config);
 121 }