Add the Inferno license to files we got from Inferno
[akaros.git] / kern / src / ns / allocb.c
index 28879fc..ab68534 100644 (file)
@@ -1,4 +1,31 @@
-// INFERNO
+/* Copyright © 1994-1999 Lucent Technologies Inc.  All rights reserved.
+ * Portions Copyright © 1997-1999 Vita Nuova Limited
+ * Portions Copyright © 2000-2007 Vita Nuova Holdings Limited
+ *                                (www.vitanuova.com)
+ * Revisions Copyright © 2000-2007 Lucent Technologies Inc. and others
+ *
+ * Modified for the Akaros operating system:
+ * Copyright (c) 2013-2014 The Regents of the University of California
+ * Copyright (c) 2013-2015 Google Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE. */
+
 #include <vfs.h>
 #include <kfs.h>
 #include <slab.h>
@@ -79,8 +106,9 @@ struct block *allocb(int size)
  * extra_data array.  When growing, it'll copy over the old entries.  All new
  * entries will be zeroed.  mem_flags determines if we'll block on kmallocs.
  *
+ * Return 0 on success or -1 on error.
  * Caller is responsible for concurrent access to the block's metadata. */
-void block_add_extd(struct block *b, unsigned int nr_bufs, int mem_flags)
+int block_add_extd(struct block *b, unsigned int nr_bufs, int mem_flags)
 {
        unsigned int old_nr_bufs = b->nr_extra_bufs;
        size_t old_amt = sizeof(struct extra_bdata) * old_nr_bufs;
@@ -88,19 +116,58 @@ void block_add_extd(struct block *b, unsigned int nr_bufs, int mem_flags)
        void *new_bdata;
 
        if (old_nr_bufs >= nr_bufs)
-               return;
+               return 0;
        if (b->extra_data) {
                new_bdata = krealloc(b->extra_data, new_amt, mem_flags);
                if (!new_bdata)
-                       return;
+                       return -1;
                memset(new_bdata + old_amt, 0, new_amt - old_amt);
        } else {
                new_bdata = kzmalloc(new_amt, mem_flags);
                if (!new_bdata)
-                       return;
+                       return - 1;
        }
        b->extra_data = new_bdata;
        b->nr_extra_bufs = nr_bufs;
+       return 0;
+}
+
+/* Go backwards from the end of the list, remember the last unused slot, and
+ * stop when a used slot is encountered. */
+static struct extra_bdata *next_unused_slot(struct block *b)
+{
+       struct extra_bdata *ebd = NULL;
+
+       for (int i = b->nr_extra_bufs - 1; i >= 0; i--) {
+               if (b->extra_data[i].base)
+                       break;
+               ebd = &b->extra_data[i];
+       }
+       return ebd;
+}
+
+/* Append a zero-filled extra data buffer of length @len to block @b.
+ * Reuse an unused extra data slot if there's any.
+ * Return 0 on success or -1 on error. */
+int block_append_extra(struct block *b, int len, int mem_flags)
+{
+       unsigned int nr_bufs = b->nr_extra_bufs + 1;
+       struct extra_bdata *ebd;
+
+       ebd = next_unused_slot(b);
+       if (!ebd) {
+               if (block_add_extd(b, nr_bufs, mem_flags) != 0)
+                       return -1;
+               ebd = next_unused_slot(b);
+               assert(ebd);
+       }
+       ebd->base = (uintptr_t)kzmalloc(len, mem_flags);
+       if (!ebd->base)
+               return -1;
+       ebd->off = 0;
+       ebd->len = len;
+       b->extra_len += ebd->len;
+       return 0;
 }
 
 /*
@@ -131,14 +198,10 @@ struct block *iallocb(int size)
        return b;
 }
 
-void freeb(struct block *b)
+void free_block_extra(struct block *b)
 {
-       void *dead = (void *)Bdead;
        struct extra_bdata *ebd;
 
-       if (b == NULL)
-               return;
-
        /* assuming our release method is kfree, which will change when we support
         * user buffers */
        for (int i = 0; i < b->nr_extra_bufs; i++) {
@@ -146,8 +209,20 @@ void freeb(struct block *b)
                if (ebd->base)
                        kfree((void*)ebd->base);
        }
+       b->extra_len = 0;
+       b->nr_extra_bufs = 0;
        kfree(b->extra_data);   /* harmless if it is 0 */
        b->extra_data = 0;              /* in case the block is reused by a free override */
+}
+
+void freeb(struct block *b)
+{
+       void *dead = (void *)Bdead;
+
+       if (b == NULL)
+               return;
+
+       free_block_extra(b);
        /*
         * drivers which perform non cache coherent DMA manage their own buffer
         * pool of uncached buffers and provide their own free routine.