bluesky/cloudlog.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 // Rough size limit for a log segment.  This is not a firm limit and there are
  17 // no absolute guarantees on the size of a log segment.
  18 #define CLOUDLOG_SEGMENT_SIZE (4 << 20)
  19
  20 BlueSkyCloudID bluesky_cloudlog_new_id()
  21 {
  22     BlueSkyCloudID id;
  23     bluesky_crypt_random_bytes((uint8_t *)&id.bytes, sizeof(id));
  24     return id;
  25 }
  26
  27 gchar *bluesky_cloudlog_id_to_string(BlueSkyCloudID id)
  28 {
  29     char buf[sizeof(BlueSkyCloudID) * 2 + 1];
  30     buf[0] = '\0';
  31
  32     for (int i = 0; i < sizeof(BlueSkyCloudID); i++) {
  33         sprintf(&buf[2*i], "%02x", (uint8_t)(id.bytes[i]));
  34     }
  35
  36     return g_strdup(buf);
  37 }
  38
  39 BlueSkyCloudID bluesky_cloudlog_id_from_string(const gchar *idstr)
  40 {
  41     BlueSkyCloudID id;
  42     memset(&id, 0, sizeof(id));
  43     for (int i = 0; i < 2*sizeof(BlueSkyCloudID); i++) {
  44         char c = idstr[i];
  45         if (c == '\0') {
  46             g_warning("Short cloud id: %s\n", idstr);
  47             break;
  48         }
  49         int val = 0;
  50         if (c >= '0' && c <= '9')
  51             val = c - '0';
  52         else if (c >= 'a' && c <= 'f')
  53             val = c - 'a' + 10;
  54         else
  55             g_warning("Bad character in cloud id: %s\n", idstr);
  56         id.bytes[i / 2] += val << (i % 2 ? 0 : 4);
  57     }
  58     return id;
  59 }
  60
  61 gboolean bluesky_cloudlog_equal(gconstpointer a, gconstpointer b)
  62 {
  63     BlueSkyCloudID *id1 = (BlueSkyCloudID *)a, *id2 = (BlueSkyCloudID *)b;
  64
  65     return memcmp(id1, id2, sizeof(BlueSkyCloudID)) == 0;
  66 }
  67
  68 guint bluesky_cloudlog_hash(gconstpointer a)
  69 {
  70     BlueSkyCloudID *id = (BlueSkyCloudID *)a;
  71
  72     // Assume that bits in the ID are randomly chosen so that any subset of the
  73     // bits can be used as a hash key.
  74     return *(guint *)(&id->bytes);
  75 }
  76
  77 /* Formatting of cloud log segments.  This handles grouping items together
  78  * before writing a batch to the cloud, handling indirection through items like
  79  * the inode map, etc. */
  80
  81 BlueSkyCloudLog *bluesky_cloudlog_new(BlueSkyFS *fs, const BlueSkyCloudID *id)
  82 {
  83     BlueSkyCloudLog *log = g_new0(BlueSkyCloudLog, 1);
  84
  85     log->lock = g_mutex_new();
  86     log->cond = g_cond_new();
  87     log->fs = fs;
  88     log->type = LOGTYPE_UNKNOWN;
  89     if (id != NULL)
  90         memcpy(&log->id, id, sizeof(BlueSkyCloudID));
  91     else
  92         log->id = bluesky_cloudlog_new_id();
  93     log->links = g_array_new(FALSE, TRUE, sizeof(BlueSkyCloudLog *));
  94     g_atomic_int_set(&log->refcount, 1);
  95
  96     return log;
  97 }
  98
  99 /* Helper function for updating memory usage statistics for a filesystem (the
 100  * cache_log_* variables).  This will increment (type=1) or decrement (type=-1)
 101  * the counter associated with the current state of the cloud log item.  The
 102  * item should be locked or otherwise protected from concurrent access. */
 103 void bluesky_cloudlog_stats_update(BlueSkyCloudLog *log, int type)
 104 {
 105     BlueSkyFS *fs = log->fs;
 106
 107     if (log->location_flags & CLOUDLOG_CLOUD) {
 108         g_atomic_int_add(&fs->cache_log_cloud, type);
 109     } else if (log->location_flags & CLOUDLOG_JOURNAL) {
 110         g_atomic_int_add(&fs->cache_log_journal, type);
 111     } else if (log->pending_write & CLOUDLOG_JOURNAL) {
 112         g_atomic_int_add(&fs->cache_log_journal, type);
 113     } else if (log->data != NULL) {
 114         g_atomic_int_add(&fs->cache_log_dirty, type);
 115     }
 116 }
 117
 118 /* The reference held by the hash table does not count towards the reference
 119  * count.  When a new object is created, it initially has a reference count of
 120  * 1 for the creator, and similarly fetching an item from the hash table will
 121  * also create a reference.  If the reference count drops to zero,
 122  * bluesky_cloudlog_unref attempts to remove the object from the hash
 123  * table--but there is a potential race since another thread might read the
 124  * object from the hash table at the same time.  So an object with a reference
 125  * count of zero may still be resurrected, in which case we need to abort the
 126  * destruction.  Once the object is gone from the hash table, and if the
 127  * reference count is still zero, it can actually be deleted. */
 128 void bluesky_cloudlog_ref(BlueSkyCloudLog *log)
 129 {
 130     if (log == NULL)
 131         return;
 132
 133     g_atomic_int_inc(&log->refcount);
 134 }
 135
 136 void bluesky_cloudlog_unref(BlueSkyCloudLog *log)
 137 {
 138     if (log == NULL)
 139         return;
 140
 141     if (g_atomic_int_dec_and_test(&log->refcount)) {
 142         BlueSkyFS *fs = log->fs;
 143
 144         g_mutex_lock(fs->lock);
 145         if (g_atomic_int_get(&log->refcount) > 0) {
 146             g_mutex_unlock(fs->lock);
 147             return;
 148         }
 149
 150         if (!g_hash_table_remove(fs->locations, &log->id)) {
 151             if (bluesky_verbose)
 152                 g_warning("Could not find and remove cloud log item from hash table!");
 153         }
 154         g_mutex_unlock(fs->lock);
 155
 156         bluesky_cloudlog_stats_update(log, -1);
 157         log->type = LOGTYPE_INVALID;
 158         g_mutex_free(log->lock);
 159         g_cond_free(log->cond);
 160         for (int i = 0; i < log->links->len; i++) {
 161             BlueSkyCloudLog *c = g_array_index(log->links,
 162                                                BlueSkyCloudLog *, i);
 163             bluesky_cloudlog_unref(c);
 164         }
 165         g_array_unref(log->links);
 166         bluesky_string_unref(log->data);
 167         g_free(log);
 168     }
 169 }
 170
 171 /* For locking reasons cloudlog unrefs may sometimes need to be performed in
 172  * the future.  We launch a thread for handling these delayed unreference
 173  * requests. */
 174 static gpointer cloudlog_unref_thread(gpointer q)
 175 {
 176     GAsyncQueue *queue = (GAsyncQueue *)q;
 177
 178     while (TRUE) {
 179         BlueSkyCloudLog *item = (BlueSkyCloudLog *)g_async_queue_pop(queue);
 180         bluesky_cloudlog_unref(item);
 181     }
 182
 183     return NULL;
 184 }
 185
 186 void bluesky_cloudlog_unref_delayed(BlueSkyCloudLog *log)
 187 {
 188     if (log != NULL)
 189         g_async_queue_push(log->fs->unref_queue, log);
 190 }
 191
 192 void bluesky_cloudlog_threads_init(BlueSkyFS *fs)
 193 {
 194     fs->unref_queue = g_async_queue_new();
 195     g_thread_create(cloudlog_unref_thread, fs->unref_queue, FALSE, NULL);
 196 }
 197
 198 /* Erase the information contained within the in-memory cloud log
 199  * representation.  This does not free up the item itself, but frees the data
 200  * and references to other log items and resets the type back to unknown.  If
 201  * the object was written out to persistent storage, all state about it can be
 202  * recovered by loading the object back in.  The object must be locked before
 203  * calling this function. */
 204 void bluesky_cloudlog_erase(BlueSkyCloudLog *log)
 205 {
 206     g_assert(log->data_lock_count == 0);
 207
 208     if (log->type == LOGTYPE_UNKNOWN)
 209         return;
 210
 211     log->type = LOGTYPE_UNKNOWN;
 212     log->data_size = 0;
 213     bluesky_string_unref(log->data);
 214     log->data = NULL;
 215     log->data_lock_count = 0;
 216
 217     for (int i = 0; i < log->links->len; i++) {
 218         BlueSkyCloudLog *c = g_array_index(log->links,
 219                                            BlueSkyCloudLog *, i);
 220         bluesky_cloudlog_unref(c);
 221     }
 222     g_array_unref(log->links);
 223     log->links = g_array_new(FALSE, TRUE, sizeof(BlueSkyCloudLog *));
 224 }
 225
 226 /* Start a write of the object to the local log. */
 227 void bluesky_cloudlog_sync(BlueSkyCloudLog *log)
 228 {
 229     bluesky_log_item_submit(log, log->fs->log);
 230 }
 231
 232 /* Add the given entry to the global hash table containing cloud log entries.
 233  * Takes ownership of the caller's reference. */
 234 void bluesky_cloudlog_insert_locked(BlueSkyCloudLog *log)
 235 {
 236     g_hash_table_insert(log->fs->locations, &log->id, log);
 237 }
 238
 239 void bluesky_cloudlog_insert(BlueSkyCloudLog *log)
 240 {
 241     g_mutex_lock(log->fs->lock);
 242     bluesky_cloudlog_insert_locked(log);
 243     g_mutex_unlock(log->fs->lock);
 244 }
 245
 246 /* Look up the cloud log entry for the given ID.  If create is TRUE and the
 247  * item does not exist, create a special pending entry that can later be filled
 248  * in when the real item is loaded.  The returned item has a reference held.
 249  * As a special case, if a null ID is provided then NULL is returned. */
 250 BlueSkyCloudLog *bluesky_cloudlog_get(BlueSkyFS *fs, BlueSkyCloudID id)
 251 {
 252     static BlueSkyCloudID id0 = {{0}};
 253
 254     if (memcmp(&id, &id0, sizeof(BlueSkyCloudID)) == 0)
 255         return NULL;
 256
 257     g_mutex_lock(fs->lock);
 258     BlueSkyCloudLog *item;
 259     item = g_hash_table_lookup(fs->locations, &id);
 260     if (item == NULL) {
 261         item = bluesky_cloudlog_new(fs, &id);
 262         bluesky_cloudlog_stats_update(item, 1);
 263         bluesky_cloudlog_insert_locked(item);
 264     } else {
 265         bluesky_cloudlog_ref(item);
 266     }
 267     g_mutex_unlock(fs->lock);
 268     return item;
 269 }
 270
 271 /* Attempt to prefetch a cloud log item.  This does not guarantee that it will
 272  * be made available, but does make it more likely that a future call to
 273  * bluesky_cloudlog_fetch will complete quickly.  Item must be locked? */
 274 void bluesky_cloudlog_prefetch(BlueSkyCloudLog *item)
 275 {
 276     if (item->data != NULL)
 277         return;
 278
 279     /* TODO: Some of the code here is duplicated with bluesky_log_map_object.
 280      * Refactor to fix that. */
 281     BlueSkyFS *fs = item->fs;
 282     BlueSkyCacheFile *map = NULL;
 283
 284     /* First, check to see if the journal still contains a copy of the item and
 285      * if so update the atime on the journal so it is likely to be kept around
 286      * until we need it. */
 287     if ((item->location_flags | item->pending_write) & CLOUDLOG_JOURNAL) {
 288         map = bluesky_cachefile_lookup(fs, -1, item->log_seq, TRUE);
 289         if (map != NULL) {
 290             map->atime = bluesky_get_current_time();
 291             bluesky_cachefile_unref(map);
 292             g_mutex_unlock(map->lock);
 293             return;
 294         }
 295     }
 296
 297     item->location_flags &= ~CLOUDLOG_JOURNAL;
 298     if (!(item->location_flags & CLOUDLOG_CLOUD))
 299         return;
 300
 301     map = bluesky_cachefile_lookup(fs,
 302                                    item->location.directory,
 303                                    item->location.sequence,
 304                                    FALSE);
 305     if (map == NULL)
 306         return;
 307
 308     /* At this point, we have information about the log segment containing the
 309      * item we need.  If our item is already fetched, we have nothing to do
 310      * except update the atime.  If not, queue up a fetch of our object. */
 311     const BlueSkyRangesetItem *rangeitem;
 312     rangeitem = bluesky_rangeset_lookup(map->items,
 313                                         item->location.offset);
 314     if (rangeitem == NULL) {
 315         if (map->prefetches == NULL)
 316             map->prefetches = bluesky_rangeset_new();
 317
 318         gchar *id = bluesky_cloudlog_id_to_string(item->id);
 319         g_print("Need to prefetch %s\n", id);
 320         g_free(id);
 321
 322         bluesky_rangeset_insert(map->prefetches,
 323                                 item->location.offset,
 324                                 item->location.size, NULL);
 325
 326         uint64_t start, length;
 327         bluesky_rangeset_get_extents(map->prefetches, &start, &length);
 328         g_print("Range to prefetch: %"PRIu64" + %"PRIu64"\n", start, length);
 329     }
 330
 331     bluesky_cachefile_unref(map);
 332     g_mutex_unlock(map->lock);
 333 }
 334
 335 /* Ensure that a cloud log item is loaded in memory, and if not read it in.
 336  * TODO: Make asynchronous, and make this also fetch from the cloud.  Right now
 337  * we only read from the log.  Log item must be locked. */
 338 void bluesky_cloudlog_fetch(BlueSkyCloudLog *log)
 339 {
 340     if (log->data != NULL)
 341         return;
 342
 343     BlueSkyProfile *profile = bluesky_profile_get();
 344     if (profile != NULL)
 345         bluesky_profile_add_event(profile, g_strdup_printf("Fetch log entry"));
 346
 347     /* There are actually two cases: a full deserialization if we have not ever
 348      * read the object before, and a partial deserialization where the metadata
 349      * is already in memory and we just need to remap the data.  If the object
 350      * type has not yet been set, we'll need to read and parse the metadata.
 351      * Once that is done, we can fall through the case of remapping the data
 352      * itself. */
 353     if (log->type == LOGTYPE_UNKNOWN) {
 354         BlueSkyRCStr *raw = bluesky_log_map_object(log, FALSE);
 355         g_assert(raw != NULL);
 356         bluesky_deserialize_cloudlog(log, raw->data, raw->len);
 357         bluesky_string_unref(raw);
 358     }
 359
 360     /* At this point all metadata should be available and we need only remap
 361      * the object data. */
 362     log->data = bluesky_log_map_object(log, TRUE);
 363
 364     if (log->data == NULL) {
 365         g_error("Unable to fetch cloudlog entry!");
 366     }
 367
 368     if (profile != NULL)
 369         bluesky_profile_add_event(profile, g_strdup_printf("Fetch complete"));
 370     g_cond_broadcast(log->cond);
 371 }
 372
 373 BlueSkyCloudPointer bluesky_cloudlog_serialize(BlueSkyCloudLog *log,
 374                                                BlueSkyFS *fs)
 375 {
 376     BlueSkyCloudLogState *state = fs->log_state;
 377
 378     if ((log->location_flags | log->pending_write) & CLOUDLOG_CLOUD) {
 379         return log->location;
 380     }
 381
 382     for (int i = 0; i < log->links->len; i++) {
 383         BlueSkyCloudLog *ref = g_array_index(log->links,
 384                                              BlueSkyCloudLog *, i);
 385         if (ref != NULL)
 386             bluesky_cloudlog_serialize(ref, fs);
 387     }
 388
 389     g_mutex_lock(log->lock);
 390     bluesky_cloudlog_fetch(log);
 391     g_assert(log->data != NULL);
 392
 393     bluesky_cloudlog_stats_update(log, -1);
 394
 395     GString *data1 = g_string_new("");
 396     GString *data2 = g_string_new("");
 397     GString *data3 = g_string_new("");
 398     bluesky_serialize_cloudlog(log, data1, data2, data3);
 399
 400     log->location = state->location;
 401     log->location.offset = state->data->len;
 402     log->data_size = data1->len;
 403
 404     struct cloudlog_header header;
 405     memcpy(header.magic, CLOUDLOG_MAGIC, 4);
 406     memset(header.crypt_auth, sizeof(header.crypt_auth), 0);
 407     memset(header.crypt_iv, sizeof(header.crypt_iv), 0);
 408     header.type = log->type + '0';
 409     header.size1 = GUINT32_TO_LE(data1->len);
 410     header.size2 = GUINT32_TO_LE(data2->len);
 411     header.size3 = GUINT32_TO_LE(data3->len);
 412     header.id = log->id;
 413     header.inum = GUINT64_TO_LE(log->inum);
 414
 415     g_string_append_len(state->data, (const char *)&header, sizeof(header));
 416     g_string_append_len(state->data, data1->str, data1->len);
 417     g_string_append_len(state->data, data2->str, data2->len);
 418     g_string_append_len(state->data, data3->str, data3->len);
 419
 420     log->location.size = state->data->len - log->location.offset;
 421
 422     g_string_free(data1, TRUE);
 423     g_string_free(data2, TRUE);
 424     g_string_free(data3, TRUE);
 425
 426     /* If the object we flushed was an inode, update the inode map. */
 427     if (log->type == LOGTYPE_INODE) {
 428         g_mutex_lock(fs->lock);
 429         InodeMapEntry *entry = bluesky_inode_map_lookup(fs->inode_map,
 430                                                         log->inum, 1);
 431         bluesky_cloudlog_unref_delayed(entry->item);
 432         entry->item = log;
 433         bluesky_cloudlog_ref(entry->item);
 434         g_mutex_unlock(fs->lock);
 435     }
 436
 437     /* TODO: We should mark the objects as committed on the cloud until the
 438      * data is flushed and acknowledged. */
 439     log->pending_write |= CLOUDLOG_CLOUD;
 440     bluesky_cloudlog_stats_update(log, 1);
 441     state->writeback_list = g_slist_prepend(state->writeback_list, log);
 442     bluesky_cloudlog_ref(log);
 443     g_mutex_unlock(log->lock);
 444
 445     if (state->data->len > CLOUDLOG_SEGMENT_SIZE)
 446         bluesky_cloudlog_flush(fs);
 447
 448     return log->location;
 449 }
 450
 451 static void cloudlog_flush_complete(BlueSkyStoreAsync *async,
 452                                     SerializedRecord *record)
 453 {
 454     g_print("Write of %s to cloud complete, status = %d\n",
 455             async->key, async->result);
 456
 457     g_mutex_lock(record->lock);
 458     if (async->result >= 0) {
 459         while (record->items != NULL) {
 460             BlueSkyCloudLog *item = (BlueSkyCloudLog *)record->items->data;
 461             g_mutex_lock(item->lock);
 462             bluesky_cloudlog_stats_update(item, -1);
 463             item->pending_write &= ~CLOUDLOG_CLOUD;
 464             item->location_flags |= CLOUDLOG_CLOUD;
 465             bluesky_cloudlog_stats_update(item, 1);
 466             g_mutex_unlock(item->lock);
 467             bluesky_cloudlog_unref(item);
 468
 469             record->items = g_slist_delete_link(record->items, record->items);
 470         }
 471
 472         bluesky_string_unref(record->data);
 473         record->data = NULL;
 474         g_slist_free(record->items);
 475         record->items = NULL;
 476         record->complete = TRUE;
 477         g_cond_broadcast(record->cond);
 478     } else {
 479         g_print("Write should be resubmitted...\n");
 480
 481         BlueSkyStoreAsync *async2 = bluesky_store_async_new(async->store);
 482         async2->op = STORE_OP_PUT;
 483         async2->key = g_strdup(async->key);
 484         async2->data = record->data;
 485         bluesky_string_ref(record->data);
 486         bluesky_store_async_submit(async2);
 487         bluesky_store_async_add_notifier(async2,
 488                                          (GFunc)cloudlog_flush_complete,
 489                                          record);
 490         bluesky_store_async_unref(async2);
 491     }
 492     g_mutex_unlock(record->lock);
 493 }
 494
 495 /* Finish up a partially-written cloud log segment and flush it to storage. */
 496 void bluesky_cloudlog_flush(BlueSkyFS *fs)
 497 {
 498     BlueSkyCloudLogState *state = fs->log_state;
 499     if (state->data == NULL || state->data->len == 0)
 500         return;
 501
 502     /* TODO: Append some type of commit record to the log segment? */
 503
 504     g_print("Serializing %zd bytes of data to cloud\n", state->data->len);
 505     SerializedRecord *record = g_new0(SerializedRecord, 1);
 506     bluesky_cloudlog_encrypt(state->data, fs->keys);
 507     record->data = bluesky_string_new_from_gstring(state->data);
 508     record->items = state->writeback_list;
 509     record->lock = g_mutex_new();
 510     record->cond = g_cond_new();
 511     state->writeback_list = NULL;
 512
 513     BlueSkyStoreAsync *async = bluesky_store_async_new(fs->store);
 514     async->op = STORE_OP_PUT;
 515     async->key = g_strdup_printf("log-%08d-%08d",
 516                                  state->location.directory,
 517                                  state->location.sequence);
 518     async->data = record->data;
 519     bluesky_string_ref(record->data);
 520     bluesky_store_async_submit(async);
 521     bluesky_store_async_add_notifier(async,
 522                                      (GFunc)cloudlog_flush_complete,
 523                                      record);
 524     bluesky_store_async_unref(async);
 525
 526     state->pending_segments = g_list_prepend(state->pending_segments, record);
 527
 528     state->location.sequence++;
 529     state->location.offset = 0;
 530     state->data = g_string_new("");
 531 }
 532
 533 /* Make an encryption pass over a cloud log segment to encrypt private data in
 534  * it. */
 535 void bluesky_cloudlog_encrypt(GString *segment, BlueSkyCryptKeys *keys)
 536 {
 537     char *data = segment->str;
 538     size_t remaining_size = segment->len;
 539
 540     while (remaining_size >= sizeof(struct cloudlog_header)) {
 541         struct cloudlog_header *header = (struct cloudlog_header *)data;
 542         size_t item_size = sizeof(struct cloudlog_header)
 543                            + GUINT32_FROM_LE(header->size1)
 544                            + GUINT32_FROM_LE(header->size2)
 545                            + GUINT32_FROM_LE(header->size3);
 546         if (item_size > remaining_size)
 547             break;
 548         bluesky_crypt_block_encrypt(data, item_size, keys);
 549
 550         data += item_size;
 551         remaining_size -= item_size;
 552     }
 553 }
 554
 555 /* Make an decryption pass over a cloud log segment to decrypt items which were
 556  * encrypted.  Also computes a list of all offsets which at which valid
 557  * cloud log items are found and adds those offsets to items (if non-NULL). */
 558 void bluesky_cloudlog_decrypt(char *segment, size_t len,
 559                               BlueSkyCryptKeys *keys,
 560                               BlueSkyRangeset *items)
 561 {
 562     char *data = segment;
 563     size_t remaining_size = len;
 564     size_t offset = 0;
 565
 566     while (remaining_size >= sizeof(struct cloudlog_header)) {
 567         struct cloudlog_header *header = (struct cloudlog_header *)data;
 568         size_t item_size = sizeof(struct cloudlog_header)
 569                            + GUINT32_FROM_LE(header->size1)
 570                            + GUINT32_FROM_LE(header->size2)
 571                            + GUINT32_FROM_LE(header->size3);
 572         if (item_size > remaining_size)
 573             break;
 574         if (bluesky_crypt_block_decrypt(data, item_size, keys)) {
 575             if (items != NULL) {
 576                 g_print("  data item at %zx\n", offset);
 577                 bluesky_rangeset_insert(items, offset, item_size,
 578                                         GINT_TO_POINTER(TRUE));
 579             }
 580         } else {
 581             g_warning("Unauthenticated data at offset %zd", offset);
 582             if (items != NULL) {
 583                 bluesky_rangeset_insert(items, offset, item_size,
 584                                         GINT_TO_POINTER(TRUE));
 585             }
 586         }
 587
 588         data += item_size;
 589         offset += item_size;
 590         remaining_size -= item_size;
 591     }
 592 }