bluesky/util.c

   1 /* Blue Sky: File Systems in the Cloud
   2  *
   3  * Copyright (C) 2009  The Regents of the University of California
   4  * Written by Michael Vrable <mvrable@cs.ucsd.edu>
   5  *
   6  * TODO: Licensing
   7  */
   8
   9 #include <stdio.h>
  10 #include <stdint.h>
  11 #include <glib.h>
  12 #include <string.h>
  13
  14 #include "bluesky-private.h"
  15
  16 /* Miscellaneous useful functions that don't really fit anywhere else. */
  17
  18 bluesky_time_hires bluesky_now_hires()
  19 {
  20     struct timespec time;
  21
  22     if (clock_gettime(CLOCK_REALTIME, &time) != 0) {
  23         perror("clock_gettime");
  24         return 0;
  25     }
  26
  27     return (int64_t)(time.tv_sec) * 1000000000 + time.tv_nsec;
  28 }
  29
  30 /* Convert a UTF-8 string to lowercase.  This can be used to implement
  31  * case-insensitive lookups and comparisons, by normalizing all values to
  32  * lowercase first.  Returns a newly-allocated string as a result. */
  33 gchar *bluesky_lowercase(const gchar *s)
  34 {
  35     /* TODO: Unicode handling; for now just do ASCII. */
  36     return g_ascii_strdown(s, -1);
  37 }
  38
  39 /**** Reference-counted strings. ****/
  40
  41 /* Create and return a new reference-counted string.  The reference count is
  42  * initially one.  The newly-returned string takes ownership of the memory
  43  * pointed at by data, and will call g_free on it when the reference count
  44  * drops to zero. */
  45 BlueSkyRCStr *bluesky_string_new(gpointer data, gsize len)
  46 {
  47     BlueSkyRCStr *string = g_new(BlueSkyRCStr, 1);
  48     string->data = data;
  49     string->len = len;
  50     g_atomic_int_set(&string->refcount, 1);
  51     return string;
  52 }
  53
  54 /* Create a new BlueSkyRCStr from a GString.  The GString is destroyed. */
  55 BlueSkyRCStr *bluesky_string_new_from_gstring(GString *s)
  56 {
  57     gsize len = s->len;
  58     return bluesky_string_new(g_string_free(s, FALSE), len);
  59 }
  60
  61 void bluesky_string_ref(BlueSkyRCStr *string)
  62 {
  63     if (string == NULL)
  64         return;
  65
  66     g_atomic_int_inc(&string->refcount);
  67 }
  68
  69 void bluesky_string_unref(BlueSkyRCStr *string)
  70 {
  71     if (string == NULL)
  72         return;
  73
  74     if (g_atomic_int_dec_and_test(&string->refcount)) {
  75         g_free(string->data);
  76         g_free(string);
  77     }
  78 }
  79
  80 /* Duplicate and return a new reference-counted string, containing a copy of
  81  * the original data, with a reference count of 1.  As an optimization, if the
  82  * passed-in string already has a reference count of 1, the original is
  83  * returned.   Can be used to make a mutable copy of a shared string.  For this
  84  * to truly be safe, it is probably needed that there be some type of lock
  85  * protecting access to the string. */
  86 BlueSkyRCStr *bluesky_string_dup(BlueSkyRCStr *string)
  87 {
  88     if (string == NULL)
  89         return NULL;
  90
  91     if (g_atomic_int_dec_and_test(&string->refcount)) {
  92         /* There are no other shared copies, so return this one. */
  93         g_atomic_int_inc(&string->refcount);
  94         return string;
  95     } else {
  96         return bluesky_string_new(g_memdup(string->data, string->len),
  97                                   string->len);
  98     }
  99 }
 100
 101 /* Resize the data block used by a BlueSkyRCStr.  The data pointer might change
 102  * after making this call, so it should not be cached across calls to this
 103  * function.  To avoid confusing any other users, the caller probably ought to
 104  * hold the only reference to the string (by calling bluesky_string_dup first
 105  * if needed). */
 106 void bluesky_string_resize(BlueSkyRCStr *string, gsize len)
 107 {
 108     if (string->len == len)
 109         return;
 110
 111     string->data = g_realloc(string->data, len);
 112 }