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_planes(struct connector_config *config,
  27                                       toml_array_t *planes)
  28 {
  29         config->nplanes = toml_array_nelem(planes);
  30         config->planes = calloc(config->nplanes, sizeof(uint32_t));
  31         if (!config->planes) {
  32                 DEBUG_LOG("Memory allocation failed: %s\n", strerror(errno));
  33                 return false;
  34         }
  35
  36         for (int j = 0; j < config->nplanes; j++) {
  37                 toml_datum_t plane = toml_int_at(planes, j);
  38                 if (!plane.ok) {
  39                         return false;
  40                 }
  41                 config->planes[j] = plane.u.i;
  42         }
  43         return true;
  44 }
  45
  46 static bool populate_connector_config(struct lease_config *config,
  47                                       toml_table_t *global_table,
  48                                       toml_array_t *conns)
  49 {
  50         int nconnectors = toml_array_nelem(conns);
  51         config->connectors = calloc(nconnectors, sizeof(*config->connectors));
  52         if (!config->connectors) {
  53                 DEBUG_LOG("Memory allocation failed: %s\n", strerror(errno));
  54                 return false;
  55         }
  56
  57         config->nconnectors = nconnectors;
  58
  59         for (int i = 0; i < config->nconnectors; i++) {
  60                 struct connector_config *conn_config = &config->connectors[i];
  61                 toml_datum_t conn = toml_string_at(conns, i);
  62                 if (!conn.ok) {
  63                         ERROR_LOG("Invalid connector in lease %s: idx:%d\n",
  64                                   config->lease_name, i);
  65                         return false;
  66                 }
  67                 conn_config->name = conn.u.s;
  68
  69                 toml_table_t *conn_config_data =
  70                     toml_table_in(global_table, conn.u.s);
  71                 if (!conn_config_data)
  72                         continue;
  73
  74                 toml_datum_t optional =
  75                     toml_bool_in(conn_config_data, "optional");
  76                 if (optional.ok)
  77                         config->connectors[i].optional = optional.u.b;
  78
  79                 toml_array_t *planes =
  80                     toml_array_in(conn_config_data, "planes");
  81                 if (planes && !populate_connector_planes(conn_config, planes)) {
  82                         ERROR_LOG("Invalid plane id for connector: %s\n",
  83                                   conn_config->name);
  84                         return false;
  85                 }
  86         }
  87         return true;
  88 }
  89
  90 int parse_config(char *filename, struct lease_config **parsed_config)
  91 {
  92         struct lease_config *config = NULL;
  93         int nconfigs, i, ret = 0;
  94         char parse_error[160];
  95
  96         FILE *fp = fopen(filename, "r");
  97         if (!fp)
  98                 return 0;
  99
 100         toml_table_t *t_config =
 101             toml_parse_file(fp, parse_error, sizeof parse_error);
 102         if (!t_config) {
 103                 CONFIG_ERROR("configuration file parse error: %s\n",
 104                              parse_error);
 105                 fclose(fp);
 106                 return 0;
 107         }
 108
 109         toml_array_t *leases = toml_array_in(t_config, "lease");
 110         if (!leases) {
 111                 CONFIG_ERROR(
 112                     "Invalid config - cannot find any 'lease' configs");
 113                 goto err;
 114         }
 115         nconfigs = toml_array_nelem(leases);
 116         config = calloc(nconfigs, sizeof *config);
 117
 118         if (!config) {
 119                 DEBUG_LOG("Memory allocation failed: %s\n", strerror(errno));
 120                 goto err;
 121         }
 122
 123         for (i = 0; i < toml_array_nelem(leases); i++) {
 124                 toml_table_t *lease = toml_table_at(leases, i);
 125
 126                 toml_datum_t name = toml_string_in(lease, "name");
 127                 if (!name.ok) {
 128                         CONFIG_ERROR("Invalid lease name in entry #%d\n", i);
 129                         goto err_free_config;
 130                 }
 131
 132                 config[i].lease_name = name.u.s;
 133
 134                 toml_array_t *conns = toml_array_in(lease, "connectors");
 135                 if (conns &&
 136                     !populate_connector_config(&config[i], t_config, conns)) {
 137                         CONFIG_ERROR("Error configuring lease: %s\n",
 138                                      config[i].lease_name);
 139                         goto err_free_config;
 140                 }
 141         }
 142
 143         *parsed_config = config;
 144         ret = nconfigs;
 145 err:
 146         toml_free(t_config);
 147         fclose(fp);
 148         return ret;
 149 err_free_config:
 150         release_config(i, config);
 151         goto err;
 152 }
 153
 154 void release_config(int num_leases, struct lease_config *config)
 155 {
 156         for (int i = 0; i < num_leases; i++) {
 157                 struct lease_config *c = &config[i];
 158                 free(c->lease_name);
 159                 for (int j = 0; j < c->nconnectors; j++) {
 160                         free(c->connectors[j].name);
 161                         free(c->connectors[j].planes);
 162                 }
 163                 free(c->connectors);
 164         }
 165         free(config);
 166 }