Add locking, thread-ID tracking to profiling.
[bluesky.git] / bluesky / bluesky.h
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 #ifndef _BLUESKY_H
10 #define _BLUESKY_H
11
12 #include <stdint.h>
13 #include <inttypes.h>
14 #include <glib.h>
15
16 #ifdef __cplusplus
17 extern "C" {
18 #endif
19
20 /* Various options to tweak for performance benchmarking purposes. */
21 typedef struct {
22     /* Perform all get/put operations synchronously. */
23     int synchronous_stores;
24
25     /* Write data in cache immediately after file is modified. */
26     int writethrough_cache;
27
28     /* Can inodes be fetched asynchronously?  (Inode object is initially
29      * created in a pending state, and not unlocked until the data is actually
30      * available.) */
31     int sync_inode_fetches;
32
33     /* Should frontends handle requests serially or allow operations to proceed
34      * in parallel? */
35     int sync_frontends;
36
37     /* Target size of the disk cache at the proxy, in kilobytes. */
38     int cache_size;
39 } BlueSkyOptions;
40
41 extern BlueSkyOptions bluesky_options;
42
43 /* Maximum number of threads to use in any particular thread pool, or -1 for no
44  * limit */
45 extern int bluesky_max_threads;
46
47 /* A general-purpose counter for gathering run-time statistics. */
48 struct bluesky_stats {
49     const char *name;
50     int64_t count;
51     int64_t sum;
52 };
53 struct bluesky_stats *bluesky_stats_new(const char *name);
54 void bluesky_stats_add(struct bluesky_stats *stats, int64_t value);
55 void bluesky_stats_dump_all();
56
57 /* BlueSky status and error codes.  Various frontends should translate these to
58  * the appropriate error code for whatever protocol they implement. */
59 typedef enum {
60     BSTATUS_OK = 0,             /* No error */
61     BSTATUS_IOERR,              /* I/O error of some form */
62     BSTATUS_NOENT,              /* File does not exist */
63 } BlueSkyStatus;
64
65 void bluesky_init(void);
66
67 gchar *bluesky_lowercase(const gchar *s);
68
69 struct BlueSkyCacheFile;
70 typedef struct BlueSkyCacheFile BlueSkyCacheFile;
71
72 typedef struct {
73     gint refcount;
74     BlueSkyCacheFile *mmap;
75     gchar *data;
76     gsize len;
77 } BlueSkyRCStr;
78
79 BlueSkyRCStr *bluesky_string_new(gpointer data, gsize len);
80 BlueSkyRCStr *bluesky_string_new_from_gstring(GString *s);
81 BlueSkyRCStr *bluesky_string_new_from_mmap(BlueSkyCacheFile *mmap,
82                                            int offset, gsize len);
83 void bluesky_string_ref(BlueSkyRCStr *string);
84 void bluesky_string_unref(BlueSkyRCStr *string);
85 BlueSkyRCStr *bluesky_string_dup(BlueSkyRCStr *string);
86 void bluesky_string_resize(BlueSkyRCStr *string, gsize len);
87
88 struct BlueSkyRangeset;
89 typedef struct BlueSkyRangeset BlueSkyRangeset;
90 typedef struct {
91     uint64_t start, length;
92     gpointer data;
93 } BlueSkyRangesetItem;
94
95 BlueSkyRangeset *bluesky_rangeset_new();
96 void bluesky_rangeset_free(BlueSkyRangeset *rangeset);
97 gboolean bluesky_rangeset_insert(BlueSkyRangeset *rangeset,
98                                  uint64_t start, uint64_t length,
99                                  gpointer data);
100 const BlueSkyRangesetItem *bluesky_rangeset_lookup(BlueSkyRangeset *rangeset,
101                                                    uint64_t offset);
102 const BlueSkyRangesetItem *bluesky_rangeset_lookup_next(BlueSkyRangeset *rangeset, uint64_t offset);
103 void bluesky_rangeset_get_extents(BlueSkyRangeset *rangeset,
104                                   uint64_t *start, uint64_t *length);
105
106 /* Storage interface.  This presents a key-value store abstraction, and can
107  * have multiple implementations: in-memory, on-disk, in-cloud. */
108 struct BlueSkyStore;
109 typedef struct BlueSkyStore BlueSkyStore;
110
111 struct BlueSkyLog;
112 typedef struct BlueSkyLog BlueSkyLog;
113
114 struct BlueSkyCloudLogState;
115 typedef struct BlueSkyCloudLogState BlueSkyCloudLogState;
116
117 struct BlueSkyCloudLog;
118 typedef struct BlueSkyCloudLog BlueSkyCloudLog;
119
120 void bluesky_store_init();
121 BlueSkyStore *bluesky_store_new(const gchar *type);
122 void bluesky_store_free(BlueSkyStore *store);
123 BlueSkyRCStr *bluesky_store_get(BlueSkyStore *store, const gchar *key);
124 void bluesky_store_put(BlueSkyStore *store,
125                        const gchar *key, BlueSkyRCStr *val);
126
127 /* File types.  The numeric values are chosen to match with those used in
128  * NFSv3. */
129 typedef enum {
130     BLUESKY_REGULAR = 1,
131     BLUESKY_DIRECTORY = 2,
132     BLUESKY_BLOCK = 3,
133     BLUESKY_CHARACTER = 4,
134     BLUESKY_SYMLINK = 5,
135     BLUESKY_SOCKET = 6,
136     BLUESKY_FIFO = 7,
137
138     /* Special types used only internally. */
139     BLUESKY_PENDING = 0,    /* Inode being loaded; type not yet determined */
140     BLUESKY_INVALID = -1,   /* Inode is invalid (failed to load) */
141 } BlueSkyFileType;
142
143 /* Filesystem state.  Each filesystem which is exported is represented by a
144  * single bluesky_fs structure in memory. */
145 struct BlueSkyCryptKeys;
146
147 typedef struct {
148     GMutex *lock;
149
150     gchar *name;                /* Descriptive name for the filesystem */
151     GHashTable *inodes;         /* Cached inodes */
152     uint64_t next_inum;         /* Next available inode for allocation */
153
154     BlueSkyStore *store;
155     BlueSkyLog *log;
156     BlueSkyCloudLogState *log_state;
157
158     /* Filesystem crypto keys */
159     char *master_key;
160     struct BlueSkyCryptKeys *keys;
161
162     /* Accounting for memory used for caches.  Space is measured in blocks, not
163      * bytes.  Updates to these variables must be made atomically. */
164     gint cache_dirty;
165
166     /* Like above, but tracking data stored in the cloudlog entries
167      * specifically:
168      *  - cache_log_dirty: data uncommitted to journal and cloud
169      *  - cache_log_writeback: data being written to journal
170      *  - cache_log_journal: data committed to journal
171      *  - cache_log_cloud: data written to cloud as well
172      * Log entries should progress from the top state to the bottom, and are
173      * only ever counted in one category at a time. */
174     gint cache_log_dirty, cache_log_writeback,
175          cache_log_journal, cache_log_cloud;
176
177     /* Linked list of inodes, sorted by access/modification times for cache
178      * management.  Editing these lists is protected by the filesystem lock; to
179      * avoid deadlock do not attempt to take any other locks while the FS lock
180      * is held for list editing purposes.  Items at the head of the list are
181      * most recently accessed/modified. */
182     GList unlogged_list;        // Changes not yet synced to journal
183     GList dirty_list;           // Not yet written to cloud storage
184     GList accessed_list;        // All in-memory inodes
185
186     /* Mutex for the flush daemon, to prevent concurrent execution. */
187     GMutex *flushd_lock;
188
189     /* Used to wait for the cache daemon to free up space */
190     GCond *flushd_cond;
191
192     /* Mapping of object identifiers (blocks, inodes) to physical location (in
193      * the local cache or in the logs in the cloud). */
194     GHashTable *locations;
195
196     /* The inode map, which maps inode numbers to the location of the most
197      * recent version. */
198     GSequence *inode_map;
199
200     /* Queue for asynchronous cloudlog unrefs, where needed. */
201     GAsyncQueue *unref_queue;
202
203     /* Thread pool for asynchronous inode fetches */
204     GThreadPool *inode_fetch_thread_pool;
205 } BlueSkyFS;
206
207 /* Inode number of the root directory. */
208 #define BLUESKY_ROOT_INUM 1
209
210 /* Timestamp, measured in microseconds since the Unix epoch. */
211 typedef int64_t bluesky_time;
212
213 /* High-resolution timer, measured in nanoseconds. */
214 typedef int64_t bluesky_time_hires;
215 bluesky_time_hires bluesky_now_hires();
216
217 /* In-memory representation of an inode within a Blue Sky server.  This
218  * corresponds roughly with information that is committed to persistent
219  * storage.  Locking/refcounting rules:
220  *   - To access or modify any data fields, the lock must be held.  This
221  *     includes file blocks.
222  *   - One reference is held by the BlueSkyFS inode hash table.  If that is the
223  *     only reference (and the inode is unlocked), the inode is subject to
224  *     dropping from the cache.
225  *   - Any pending operations should hold extra references to the inode as
226  *     appropriate to keep it available until the operation completes.
227  *   - Locking dependency order is, when multiple locks are to be acquired, to
228  *     acquire locks on parents in the filesystem tree before children.
229  *     (TODO: What about rename when we acquire locks in unrelated parts of the
230  *     filesystem?)
231  *   - An inode should not be locked while the filesystem lock is already held,
232  *     since some code may do an inode lookup (which acquires the filesystem
233  *     lock) while a different inode is locked.
234  * */
235 typedef struct {
236     GMutex *lock;
237     gint refcount;
238
239     BlueSkyFS *fs;
240
241     BlueSkyFileType type;
242     uint32_t mode;
243     uint32_t uid, gid;
244     uint32_t nlink;
245
246     /* Rather than track an inode number and generation number, we will simply
247      * never re-use a fileid after a file is deleted.  64 bits should be enough
248      * that we don't exhaust the identifier space. */
249     uint64_t inum;
250
251     /* change_count is increased with every operation which modifies the inode,
252      * and can be used to determine if cached data is still valid.
253      * change_commit is the value of change_count when the inode was last
254      * committed to stable storage (the log).
255      * change_cloud tracks which version was last commited to cloud storage. */
256     uint64_t change_count, change_commit, change_cloud;
257
258     /* Timestamp for controlling when modified data is flushed to stable
259      * storage.  When an inode is first modified from a clean state, this is
260      * set to the current time.  If the inode is clean, it is set to zero. */
261     int64_t change_time;
262
263     /* Last access time to this inode, for controlling cache evictions. */
264     int64_t access_time;
265
266     /* Version of the object last serialized and committed to storage. */
267     BlueSkyCloudLog *committed_item;
268
269     /* Pointers to the linked-list elements for this inode in the accessed and
270      * dirty linked lists.  We re-use the GList structure, using ->next to
271      * point to the head of the list and ->prev to point to the tail.  The data
272      * element is unused. */
273     GList *unlogged_list, *accessed_list, *dirty_list;
274
275     int64_t atime;              /* Microseconds since the Unix epoch */
276     int64_t ctime;
277     int64_t mtime;
278     int64_t ntime;              /* "new time": time object was created */
279
280     /* File-specific fields */
281     uint64_t size;
282     GArray *blocks;
283
284     /* Directory-specific fields */
285     GSequence *dirents;         /* List of entries for READDIR */
286     GHashTable *dirhash;        /* Hash table by name for LOOKUP */
287     GHashTable *dirhash_folded; /* As above, but case-folded */
288     uint64_t parent_inum;       /* inode for ".."; 0 if the root directory */
289
290     /* Symlink-specific fields */
291     gchar *symlink_contents;
292
293     /* A field for short-term use internally while the lock is held. */
294     gpointer private_data;
295 } BlueSkyInode;
296
297 /* A directory entry.  The name is UTF-8 and is a freshly-allocated string.
298  * Each directory entry is listed in two indices: dirents is indexed by cookie
299  * and dirhash by name.  The cookie is a randomly-assigned 32-bit value, unique
300  * within the directory, that remains unchanged until the entry is deleted.  It
301  * is used to provide a stable key for restarting a READDIR call. */
302 typedef struct {
303     gchar *name;
304     gchar *name_folded;         /* Name, folded for case-insensitive lookup */
305     uint32_t cookie;
306     uint64_t inum;
307 } BlueSkyDirent;
308
309 /* File data is divided into fixed-size blocks (except the last block which may
310  * be short?).  These blocks are backed by storage in a key/value store, but
311  * may also be dirty if modifications have been made in-memory that have not
312  * been committed. */
313 #define BLUESKY_BLOCK_SIZE 32768ULL
314 #define BLUESKY_MAX_FILE_SIZE (BLUESKY_BLOCK_SIZE << 24)
315 typedef enum {
316     BLUESKY_BLOCK_ZERO = 0,     /* Data is all zeroes, not explicitly stored */
317     BLUESKY_BLOCK_REF = 1,      /* Reference to cloud log item, data clean */
318     BLUESKY_BLOCK_DIRTY = 2,    /* Data needs to be committed to store */
319 } BlueSkyBlockType;
320
321 typedef struct {
322     BlueSkyBlockType type;
323     BlueSkyCloudLog *ref;       /* if REF: cloud log entry with data */
324     BlueSkyRCStr *dirty;        /* if DIRTY: raw data in memory */
325 } BlueSkyBlock;
326
327 BlueSkyFS *bluesky_init_fs(gchar *name, BlueSkyStore *store,
328                            const gchar *master_key);
329
330 gboolean bluesky_inode_is_ready(BlueSkyInode *inode);
331
332 int64_t bluesky_get_current_time();
333 void bluesky_inode_update_ctime(BlueSkyInode *inode, gboolean update_mtime);
334 uint64_t bluesky_fs_alloc_inode(BlueSkyFS *fs);
335 void bluesky_init_inode(BlueSkyInode *i, BlueSkyFileType type);
336 BlueSkyInode *bluesky_new_inode(uint64_t inum, BlueSkyFS *fs, BlueSkyFileType type);
337
338 BlueSkyInode *bluesky_get_inode(BlueSkyFS *fs, uint64_t inum);
339 void bluesky_inode_ref(BlueSkyInode *inode);
340 void bluesky_inode_unref(BlueSkyInode *inode);
341 void bluesky_insert_inode(BlueSkyFS *fs, BlueSkyInode *inode);
342
343 void bluesky_dirent_destroy(gpointer dirent);
344 uint64_t bluesky_directory_lookup(BlueSkyInode *inode, gchar *name);
345 uint64_t bluesky_directory_ilookup(BlueSkyInode *inode, gchar *name);
346 BlueSkyDirent *bluesky_directory_read(BlueSkyInode *dir, uint32_t cookie);
347 gboolean bluesky_directory_insert(BlueSkyInode *dir, const gchar *name,
348                                   uint64_t inum);
349 void bluesky_directory_dump(BlueSkyInode *dir);
350
351 void bluesky_file_truncate(BlueSkyInode *inode, uint64_t size);
352 void bluesky_file_write(BlueSkyInode *inode, uint64_t offset,
353                         const char *data, gint len);
354 void bluesky_file_read(BlueSkyInode *inode, uint64_t offset,
355                        char *buf, gint len);
356
357 void bluesky_inode_fetch(BlueSkyFS *fs, uint64_t inum);
358
359 gint bluesky_dirent_compare(gconstpointer a, gconstpointer b,
360                             gpointer unused);
361
362 void bluesky_flushd_invoke(BlueSkyFS *fs);
363 void bluesky_flushd_invoke_conditional(BlueSkyFS *fs);
364 void bluesky_inode_do_sync(BlueSkyInode *inode);
365 void bluesky_flushd_thread_launch(BlueSkyFS *fs);
366
367 void bluesky_debug_dump(BlueSkyFS *fs);
368
369 /* Request response time tracking. */
370 typedef struct BlueSkyProfile {
371     GMutex *lock;
372     char *description;
373     GList *events;
374 } BlueSkyProfile;
375
376 BlueSkyProfile *bluesky_profile_new();
377 void bluesky_profile_free(BlueSkyProfile *profile);
378 void bluesky_profile_add_event(BlueSkyProfile *profile, char *message);
379 void bluesky_profile_print(BlueSkyProfile *profile);
380 BlueSkyProfile *bluesky_profile_get();
381 void bluesky_profile_set(BlueSkyProfile *profile);
382
383 #ifdef __cplusplus
384 }
385 #endif
386
387 #endif