| /* GIO - GLib Input, Output and Streaming Library |
| * |
| * Copyright (C) 2013 Emmanuele Bassi <ebassi@gnome.org> |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General |
| * Public License along with this library; if not, write to the |
| * Free Software Foundation, Inc., 59 Temple Place, Suite 330, |
| * Boston, MA 02111-1307, USA. |
| */ |
| |
| /** |
| * SECTION:gencoder |
| * @Title: GEncoder |
| * @Short_Description: Encodes and decodes key and value pairs |
| * |
| * #GEncoder is an abstract class that provides an API to store (encoder) and |
| * retrieve (decode) key, value pairs from memory or disk. |
| * |
| * Implementations of #GEncoder are required to provide the code to read a |
| * data storage in the form of a #GBytes and place its contents into the |
| * #GEncoder, and the code to write the contents of the #GEncoder into a |
| * #GBytes. It is not necessary for a #GEncoder to provide both encoding |
| * and decoding: if the #GEncoder sub-class provides only the implementation |
| * of the #GEncoderClass.read_from_bytes() virtual function it is called a |
| * "decoder"; alternatively, if it only provides the implementation of the |
| * the #GEncoderClass.write_to_bytes() virtual function it is called an |
| * "encoder". |
| */ |
| |
| #include "config.h" |
| |
| #include "gencoder.h" |
| #include "glibintl.h" |
| |
| #include <string.h> |
| |
| #define G_ENCODER_PRIVATE(obj) (&G_STRUCT_MEMBER (GEncoderPrivate, (obj), g_encoder_private_offset)) |
| |
| typedef struct _GEncoderPrivate GEncoderPrivate; |
| typedef struct _EncoderValue EncoderValue; |
| |
| #define ENCODER_LOCKED 1 |
| #define ENCODER_CLOSED 2 |
| |
| struct _GEncoderPrivate |
| { |
| GHashTable *values; |
| GVariant *encoded; |
| |
| gint flags; |
| }; |
| |
| static gint g_encoder_private_offset = 0; |
| |
| G_DEFINE_ABSTRACT_TYPE (GEncoder, g_encoder, G_TYPE_OBJECT) |
| |
| static inline void |
| g_encoder_lock (GEncoder *encoder) |
| { |
| GEncoderPrivate *priv = G_ENCODER_PRIVATE (encoder); |
| |
| g_bit_lock (&priv->flags, ENCODER_LOCKED); |
| } |
| |
| static inline void |
| g_encoder_unlock (GEncoder *encoder) |
| { |
| GEncoderPrivate *priv = G_ENCODER_PRIVATE (encoder); |
| |
| g_bit_unlock (&priv->flags, ENCODER_LOCKED); |
| } |
| |
| static inline gboolean |
| g_encoder_is_closed (GEncoder *encoder) |
| { |
| return (G_ENCODER_PRIVATE (encoder)->flags & ENCODER_CLOSED) != FALSE; |
| } |
| |
| static void |
| g_encoder_finalize (GObject *gobject) |
| { |
| GEncoderPrivate *priv = G_ENCODER_PRIVATE (gobject); |
| |
| if (priv->values != NULL) |
| g_hash_table_unref (priv->values); |
| |
| if (priv->encoded != NULL) |
| g_variant_unref (priv->encoded); |
| |
| G_OBJECT_CLASS (g_encoder_parent_class)->finalize (gobject); |
| } |
| |
| static void |
| g_encoder_real_closed (GEncoder *encoder, |
| GVariant *variant) |
| { |
| } |
| |
| static void |
| g_encoder_real_value_encoded (GEncoder *encoder, |
| const char *key, |
| GVariant *value) |
| { |
| } |
| |
| static gboolean |
| g_encoder_real_read_from_bytes (GEncoder *encoder, |
| GBytes *bytes, |
| GError **error) |
| { |
| return FALSE; |
| } |
| |
| static GBytes * |
| g_encoder_real_write_to_bytes (GEncoder *encoder, |
| GError **error) |
| { |
| return NULL; |
| } |
| |
| static void |
| g_encoder_class_init (GEncoderClass *klass) |
| { |
| GObjectClass *gobject_class = G_OBJECT_CLASS (klass); |
| |
| klass->closed = g_encoder_real_closed; |
| klass->value_encoded = g_encoder_real_value_encoded; |
| klass->read_from_bytes = g_encoder_real_read_from_bytes; |
| klass->write_to_bytes = g_encoder_real_write_to_bytes; |
| |
| gobject_class->finalize = g_encoder_finalize; |
| |
| g_type_class_add_private (klass, sizeof (GEncoderPrivate)); |
| g_encoder_private_offset = g_type_class_get_instance_private_offset (klass); |
| } |
| |
| static void |
| g_encoder_init (GEncoder *self) |
| { |
| GEncoderPrivate *priv = G_ENCODER_PRIVATE (self); |
| |
| priv->values = |
| g_hash_table_new_full (g_str_hash, g_str_equal, |
| g_free, |
| (GDestroyNotify) g_variant_unref); |
| } |
| |
| static inline void |
| g_encoder_value_encoded (GEncoder *encoder, |
| const char *key, |
| GVariant *value) |
| { |
| G_ENCODER_GET_CLASS (encoder)->value_encoded (encoder, key, value); |
| } |
| |
| static inline gboolean |
| g_encoder_add_key_value (GEncoder *encoder, |
| const char *key, |
| GVariant *value) |
| { |
| GEncoderPrivate *priv = G_ENCODER_PRIVATE (encoder); |
| gboolean res = TRUE; |
| |
| g_encoder_lock (encoder); |
| |
| if (g_hash_table_lookup (priv->values, key) != NULL) |
| res = FALSE; |
| |
| g_hash_table_replace (priv->values, g_strdup (key), value); |
| |
| g_encoder_value_encoded (encoder, key, value); |
| |
| g_encoder_unlock (encoder); |
| |
| return res; |
| } |
| |
| gboolean |
| g_encoder_add_key (GEncoder *encoder, |
| const char *key, |
| GVariant *value) |
| { |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (value != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| return g_encoder_add_key_value (encoder, key, g_variant_ref (value)); |
| } |
| |
| /** |
| * g_encoder_add_key_data: |
| * @encoder: a #GEncoder |
| * @key: the key for the data |
| * @value: (array length=value_len): the data to store for @key |
| * @value_len: the length of the @value array |
| * |
| * Stores an array of bytes inside @encoder, replacing the current |
| * value for @key if necessary. |
| * |
| * The @encoder makes a copy of the passed @value. |
| * |
| * Return value: %TRUE if the key was newly added, and %FALSE |
| * if the value was replaced. |
| * |
| * Since: 2.38 |
| */ |
| gboolean |
| g_encoder_add_key_data (GEncoder *encoder, |
| const char *key, |
| const guint8 *value, |
| gsize value_len) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_fixed_array (G_VARIANT_TYPE ("y"), value, value_len, sizeof (guint8)); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_add_key_string (GEncoder *encoder, |
| const char *key, |
| const char *value) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_string (value); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_add_key_int64 (GEncoder *encoder, |
| const char *key, |
| gint64 value) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_int64 (value); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_add_key_int32 (GEncoder *encoder, |
| const char *key, |
| int value) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_int32 (value); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_add_key_double (GEncoder *encoder, |
| const char *key, |
| double value) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_double (value); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_add_key_bool (GEncoder *encoder, |
| const char *key, |
| gboolean value) |
| { |
| GVariant *ev; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| g_return_val_if_fail (!g_encoder_is_closed (encoder), FALSE); |
| |
| ev = g_variant_new_boolean (value); |
| g_variant_ref_sink (ev); |
| |
| return g_encoder_add_key_value (encoder, key, ev); |
| } |
| |
| gboolean |
| g_encoder_get_key_data (GEncoder *encoder, |
| const char *key, |
| guint8 **res, |
| gsize *res_len) |
| { |
| char *byte_string; |
| GVariant *ev; |
| gsize len; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_BYTESTRING)) |
| { |
| g_encoder_unlock (encoder); |
| |
| if (res != NULL) |
| *res = NULL; |
| |
| if (res_len != NULL) |
| *res_len = 0; |
| |
| return FALSE; |
| } |
| |
| byte_string = g_variant_dup_bytestring (ev, &len); |
| if (res != NULL) |
| *res = (guint8 *) byte_string; |
| else |
| g_free (byte_string); |
| |
| if (res_len != NULL) |
| *res_len = len; |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| gboolean |
| g_encoder_get_key_string (GEncoder *encoder, |
| const char *key, |
| char **res) |
| { |
| const char *str; |
| GVariant *ev; |
| gsize len; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_STRING)) |
| { |
| g_encoder_unlock (encoder); |
| |
| if (res != NULL) |
| *res = NULL; |
| |
| return FALSE; |
| } |
| |
| str = g_variant_get_string (ev, &len); |
| if (res != NULL) |
| *res = g_strndup (str, len); |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| gboolean |
| g_encoder_get_key_int64 (GEncoder *encoder, |
| const char *key, |
| gint64 *res) |
| { |
| GVariant *ev; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_INT64)) |
| { |
| g_encoder_unlock (encoder); |
| return FALSE; |
| } |
| |
| if (res != NULL) |
| *res = g_variant_get_int64 (ev); |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| gboolean |
| g_encoder_get_key_int32 (GEncoder *encoder, |
| const char *key, |
| gint32 *res) |
| { |
| GVariant *ev; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_INT32)) |
| { |
| g_encoder_unlock (encoder); |
| |
| if (res != NULL) |
| *res = 0; |
| |
| return FALSE; |
| } |
| |
| if (res != NULL) |
| *res = g_variant_get_int32 (ev); |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| gboolean |
| g_encoder_get_key_double (GEncoder *encoder, |
| const char *key, |
| double *res) |
| { |
| GVariant *ev; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_DOUBLE)) |
| { |
| g_encoder_unlock (encoder); |
| |
| if (res != NULL) |
| *res = 0; |
| |
| return FALSE; |
| } |
| |
| if (res != NULL) |
| *res = g_variant_get_double (ev); |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| gboolean |
| g_encoder_get_key_bool (GEncoder *encoder, |
| const char *key, |
| gboolean *res) |
| { |
| GVariant *ev; |
| |
| g_encoder_lock (encoder); |
| |
| ev = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key); |
| if (ev == NULL || !g_variant_is_of_type (ev, G_VARIANT_TYPE_BOOLEAN)) |
| { |
| g_encoder_unlock (encoder); |
| |
| if (res != NULL) |
| *res = FALSE; |
| |
| return FALSE; |
| } |
| |
| if (res != NULL) |
| *res = g_variant_get_boolean (ev); |
| |
| g_encoder_unlock (encoder); |
| |
| return TRUE; |
| } |
| |
| /** |
| * g_encoder_has_key: |
| * @encoder: a #GEncoder |
| * @key: the key to look up |
| * |
| * Checks whether @key is set inside @encoder. |
| * |
| * It is usually more performant to call any of the g_encoder_get_key_* |
| * family of functions without checking for the existence of the key |
| * beforehand. |
| * |
| * Return value: %TRUE if the key exists inside @encoder, and %FALSE |
| * otherwise. |
| * |
| * Since: 2.38 |
| */ |
| gboolean |
| g_encoder_has_key (GEncoder *encoder, |
| const char *key) |
| { |
| gboolean res; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (key != NULL, FALSE); |
| |
| g_encoder_lock (encoder); |
| |
| res = g_hash_table_lookup (G_ENCODER_PRIVATE (encoder)->values, key) != NULL; |
| |
| g_encoder_unlock (encoder); |
| |
| return res; |
| } |
| |
| /** |
| * g_encoder_close: |
| * @encoder: a #GEncoder |
| * |
| * Closes a @encoder. |
| * |
| * It is not possible to add or modify a key in @encoder after calling |
| * this function. |
| * |
| * This function should only be called when writing #GEncoder sub-classes |
| * from the implementation of the #GEncoderClass.write_to_bytes() virtual |
| * function. |
| * |
| * Return value: (transfer none): The encoded representation of @encoder, |
| * stored inside a #GVariant with type 'a{sv}'. The returned #GVariant |
| * is owned by the #GEncoder and should not be modified or freed. |
| * |
| * Since: 2.38 |
| */ |
| GVariant * |
| g_encoder_close (GEncoder *encoder) |
| { |
| GVariantBuilder builder; |
| GEncoderPrivate *priv; |
| GHashTableIter iter; |
| gpointer key, value; |
| GVariant *res; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), NULL); |
| |
| priv = G_ENCODER_PRIVATE (encoder); |
| |
| g_encoder_lock (encoder); |
| |
| priv->flags |= ENCODER_CLOSED; |
| |
| g_variant_builder_init (&builder, G_VARIANT_TYPE ("a{sv}")); |
| |
| if (priv->values == NULL) |
| goto out; |
| |
| g_hash_table_iter_init (&iter, priv->values); |
| while (g_hash_table_iter_next (&iter, &key, &value)) |
| { |
| GVariant *vkey = g_variant_new_string (key); |
| GVariant *vvalue = g_variant_new_variant (value); |
| |
| g_variant_builder_open (&builder, G_VARIANT_TYPE ("{sv}")); |
| g_variant_builder_add_value (&builder, vkey); |
| g_variant_builder_add_value (&builder, vvalue); |
| g_variant_builder_close (&builder); |
| } |
| |
| out: |
| res = g_variant_builder_end (&builder); |
| priv->encoded = g_variant_ref_sink (res); |
| |
| g_encoder_unlock (encoder); |
| |
| G_ENCODER_GET_CLASS (encoder)->closed (encoder, res); |
| |
| return res; |
| } |
| |
| /** |
| * g_encoder_read_from_bytes: |
| * @encoder: a #GEncoder |
| * @bytes: a data buffer |
| * @error: (allow-none): return location for a #GError, or %NULL |
| * |
| * Reads the contents of @bytes and decodes them into an @encoder. |
| * |
| * This function calls the #GEncoderClass.read_from_bytes virtual |
| * function implementation for the @encode class. |
| * |
| * Return value: %TRUE if the @bytes buffer was successfully read, |
| * and %FALSE otherwise. |
| * |
| * Since: 2.38 |
| */ |
| gboolean |
| g_encoder_read_from_bytes (GEncoder *encoder, |
| GBytes *bytes, |
| GError **error) |
| { |
| GEncoderPrivate *priv; |
| |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| g_return_val_if_fail (bytes != NULL, FALSE); |
| |
| priv = G_ENCODER_PRIVATE (encoder); |
| |
| priv->flags &= ~ENCODER_CLOSED; |
| |
| if (priv->values != NULL) |
| g_hash_table_unref (priv->values); |
| |
| priv->values = |
| g_hash_table_new_full (g_str_hash, g_str_equal, |
| g_free, |
| (GDestroyNotify) g_variant_unref); |
| |
| return G_ENCODER_GET_CLASS (encoder)->read_from_bytes (encoder, bytes, error); |
| } |
| |
| /** |
| * g_encoder_write_to_bytes: |
| * @encode: a #GEncoder |
| * @error: (allow-none): return location for a #GError, or %NULL |
| * |
| * Encodes the contents of @encoder and writes them into a #GBytes |
| * buffer. |
| * |
| * This function calls the #GEncoderClass.write_to_bytes virtual |
| * function implementation for the @encode class. |
| * |
| * Return value: (transfer full): a #GBytes containing the encoded |
| * contents of @encoder, or %NULL |
| * |
| * Since: 2.38 |
| */ |
| GBytes * |
| g_encoder_write_to_bytes (GEncoder *encoder, |
| GError **error) |
| { |
| g_return_val_if_fail (G_IS_ENCODER (encoder), FALSE); |
| |
| return G_ENCODER_GET_CLASS (encoder)->write_to_bytes (encoder, error); |
| } |