+/* Copyright (c) 2012 Nicira, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License. */
+
+#include <config.h>
+#include "smap.h"
+
+#include <assert.h>
+
+#include "hash.h"
+
+static struct smap_node *smap_add__(struct smap *, char *, void *,
+ size_t hash);
+static struct smap_node *smap_find__(const struct smap *, const char *key,
+ size_t key_len, size_t hash);
+static int compare_nodes_by_key(const void *, const void *);
+\f
+/* Public Functions. */
+
+void
+smap_init(struct smap *smap)
+{
+ hmap_init(&smap->map);
+}
+
+void
+smap_destroy(struct smap *smap)
+{
+ if (smap) {
+ smap_clear(smap);
+ hmap_destroy(&smap->map);
+ }
+}
+
+/* Adds 'key' paired with 'value' to 'smap'. It is the caller's responsibility
+ * to avoid duplicate keys if desirable. */
+struct smap_node *
+smap_add(struct smap *smap, const char *key, const char *value)
+{
+ size_t key_len = strlen(key);
+ return smap_add__(smap, xmemdup0(key, key_len), xstrdup(value),
+ hash_bytes(key, key_len, 0));
+}
+
+/* Attempts to add 'key' to 'smap' associated with 'value'. If 'key' already
+ * exists in 'smap', does nothing and returns false. Otherwise, performs the
+ * addition and returns true. */
+bool
+smap_add_once(struct smap *smap, const char *key, const char *value)
+{
+ if (!smap_get(smap, key)) {
+ smap_add(smap, key, value);
+ return true;
+ } else {
+ return false;
+ }
+}
+
+/* Adds 'key' paired with a value derived from 'format' (similar to printf).
+ * It is the caller's responsibility to avoid duplicate keys if desirable. */
+void
+smap_add_format(struct smap *smap, const char *key, const char *format, ...)
+{
+ size_t key_len;
+ va_list args;
+ char *value;
+
+ va_start(args, format);
+ value = xvasprintf(format, args);
+ va_end(args);
+
+ key_len = strlen(key);
+ smap_add__(smap, xmemdup0(key, key_len), value,
+ hash_bytes(key, key_len, 0));
+}
+
+/* Searches for 'key' in 'smap'. If it does not already exists, adds it.
+ * Otherwise, changes its value to 'value'. */
+void
+smap_replace(struct smap *smap, const char *key, const char *value)
+{
+ size_t key_len = strlen(key);
+ size_t hash = hash_bytes(key, key_len, 0);
+
+ struct smap_node *node;
+
+ node = smap_find__(smap, key, key_len, hash);
+ if (node) {
+ free(node->value);
+ node->value = xstrdup(value);
+ } else {
+ smap_add__(smap, xmemdup0(key, key_len), xstrdup(value), hash);
+ }
+}
+
+/* If 'key' is in 'smap', removes it. Otherwise does nothing. */
+void
+smap_remove(struct smap *smap, const char *key)
+{
+ struct smap_node *node = smap_get_node(smap, key);
+
+ if (node) {
+ smap_remove_node(smap, node);
+ }
+}
+
+/* Removes 'node' from 'smap'. */
+void
+smap_remove_node(struct smap *smap, struct smap_node *node)
+{
+ hmap_remove(&smap->map, &node->node);
+ free(node->key);
+ free(node->value);
+ free(node);
+}
+
+/* Removes all key-value pairs from 'smap'. */
+void
+smap_clear(struct smap *smap)
+{
+ struct smap_node *node, *next;
+
+ SMAP_FOR_EACH_SAFE (node, next, smap) {
+ smap_remove_node(smap, node);
+ }
+}
+
+/* Returns the value associated with 'key' in 'smap', or NULL. */
+const char *
+smap_get(const struct smap *smap, const char *key)
+{
+ struct smap_node *node = smap_get_node(smap, key);
+ return node ? node->value : NULL;
+}
+
+/* Returns the node associated with 'key' in 'smap', or NULL. */
+struct smap_node *
+smap_get_node(const struct smap *smap, const char *key)
+{
+ size_t key_len = strlen(key);
+ return smap_find__(smap, key, key_len, hash_bytes(key, key_len, 0));
+}
+
+/* Gets the value associated with 'key' in 'smap' and converts it to a boolean.
+ * If 'key' is not in 'smap', or its value is neither "true" nor "false",
+ * returns 'def'. */
+bool
+smap_get_bool(const struct smap *smap, const char *key, bool def)
+{
+ const char *value = smap_get(smap, key);
+
+ if (!value) {
+ return def;
+ }
+
+ if (def) {
+ return strcasecmp("false", value) != 0;
+ } else {
+ return !strcasecmp("true", value);
+ }
+}
+
+/* Gets the value associated with 'key' in 'smap' and converts it to an int
+ * using atoi(). If 'key' is not in 'smap', returns 'def'. */
+int
+smap_get_int(const struct smap *smap, const char *key, int def)
+{
+ const char *value = smap_get(smap, key);
+
+ return value ? atoi(value) : def;
+}
+
+/* Returns true of there are no elements in 'smap'. */
+bool
+smap_is_empty(const struct smap *smap)
+{
+ return hmap_is_empty(&smap->map);
+}
+
+/* Returns the number of elements in 'smap'. */
+size_t
+smap_count(const struct smap *smap)
+{
+ return hmap_count(&smap->map);
+}
+
+/* Initializes 'dst' as a clone of 'src. */
+void
+smap_clone(struct smap *dst, struct smap *src)
+{
+ struct smap_node *node;
+
+ smap_init(dst);
+ SMAP_FOR_EACH (node, src) {
+ smap_add__(dst, xstrdup(node->key), xstrdup(node->value),
+ node->node.hash);
+ }
+}
+
+/* Returns an array of nodes sorted on key or NULL if 'smap' is empty. The
+ * caller is responsible for freeing this array. */
+const struct smap_node **
+smap_sort(const struct smap *smap)
+{
+ if (smap_is_empty(smap)) {
+ return NULL;
+ } else {
+ const struct smap_node **nodes;
+ struct smap_node *node;
+ size_t i, n;
+
+ n = smap_count(smap);
+ nodes = xmalloc(n * sizeof *nodes);
+ i = 0;
+ SMAP_FOR_EACH (node, smap) {
+ nodes[i++] = node;
+ }
+ assert(i == n);
+
+ qsort(nodes, n, sizeof *nodes, compare_nodes_by_key);
+
+ return nodes;
+ }
+}
+\f
+/* Private Helpers. */
+
+static struct smap_node *
+smap_add__(struct smap *smap, char *key, void *value, size_t hash)
+{
+ struct smap_node *node = xmalloc(sizeof *node);
+ node->key = key;
+ node->value = value;
+ hmap_insert(&smap->map, &node->node, hash);
+ return node;
+}
+
+static struct smap_node *
+smap_find__(const struct smap *smap, const char *key, size_t key_len,
+ size_t hash)
+{
+ struct smap_node *node;
+
+ HMAP_FOR_EACH_WITH_HASH (node, node, hash, &smap->map) {
+ if (!strncmp(node->key, key, key_len) && !node->key[key_len]) {
+ return node;
+ }
+ }
+
+ return NULL;
+}
+
+static int
+compare_nodes_by_key(const void *a_, const void *b_)
+{
+ const struct smap_node *const *a = a_;
+ const struct smap_node *const *b = b_;
+ return strcmp((*a)->key, (*b)->key);
+}