xref: /openbmc/linux/fs/befs/btree.c (revision 73919280)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * linux/fs/befs/btree.c
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  * Copyright (C) 2001-2002 Will Dyson <will_dyson@pobox.com>
51da177e4SLinus Torvalds  *
61da177e4SLinus Torvalds  * Licensed under the GNU GPL. See the file COPYING for details.
71da177e4SLinus Torvalds  *
825985edcSLucas De Marchi  * 2002-02-05: Sergey S. Kostyliov added binary search within
91da177e4SLinus Torvalds  * 		btree nodes.
101da177e4SLinus Torvalds  *
111da177e4SLinus Torvalds  * Many thanks to:
121da177e4SLinus Torvalds  *
131da177e4SLinus Torvalds  * Dominic Giampaolo, author of "Practical File System
141da177e4SLinus Torvalds  * Design with the Be File System", for such a helpful book.
151da177e4SLinus Torvalds  *
161da177e4SLinus Torvalds  * Marcus J. Ranum, author of the b+tree package in
171da177e4SLinus Torvalds  * comp.sources.misc volume 10. This code is not copied from that
181da177e4SLinus Torvalds  * work, but it is partially based on it.
191da177e4SLinus Torvalds  *
201da177e4SLinus Torvalds  * Makoto Kato, author of the original BeFS for linux filesystem
211da177e4SLinus Torvalds  * driver.
221da177e4SLinus Torvalds  */
231da177e4SLinus Torvalds 
241da177e4SLinus Torvalds #include <linux/kernel.h>
251da177e4SLinus Torvalds #include <linux/string.h>
261da177e4SLinus Torvalds #include <linux/slab.h>
271da177e4SLinus Torvalds #include <linux/mm.h>
281da177e4SLinus Torvalds #include <linux/buffer_head.h>
291da177e4SLinus Torvalds 
301da177e4SLinus Torvalds #include "befs.h"
311da177e4SLinus Torvalds #include "btree.h"
321da177e4SLinus Torvalds #include "datastream.h"
331da177e4SLinus Torvalds 
341da177e4SLinus Torvalds /*
351da177e4SLinus Torvalds  * The btree functions in this file are built on top of the
361da177e4SLinus Torvalds  * datastream.c interface, which is in turn built on top of the
371da177e4SLinus Torvalds  * io.c interface.
381da177e4SLinus Torvalds  */
391da177e4SLinus Torvalds 
401da177e4SLinus Torvalds /* Befs B+tree structure:
411da177e4SLinus Torvalds  *
421da177e4SLinus Torvalds  * The first thing in the tree is the tree superblock. It tells you
431da177e4SLinus Torvalds  * all kinds of useful things about the tree, like where the rootnode
441da177e4SLinus Torvalds  * is located, and the size of the nodes (always 1024 with current version
451da177e4SLinus Torvalds  * of BeOS).
461da177e4SLinus Torvalds  *
471da177e4SLinus Torvalds  * The rest of the tree consists of a series of nodes. Nodes contain a header
481da177e4SLinus Torvalds  * (struct befs_btree_nodehead), the packed key data, an array of shorts
491da177e4SLinus Torvalds  * containing the ending offsets for each of the keys, and an array of
501da177e4SLinus Torvalds  * befs_off_t values. In interior nodes, the keys are the ending keys for
511da177e4SLinus Torvalds  * the childnode they point to, and the values are offsets into the
521da177e4SLinus Torvalds  * datastream containing the tree.
531da177e4SLinus Torvalds  */
541da177e4SLinus Torvalds 
551da177e4SLinus Torvalds /* Note:
561da177e4SLinus Torvalds  *
571da177e4SLinus Torvalds  * The book states 2 confusing things about befs b+trees. First,
581da177e4SLinus Torvalds  * it states that the overflow field of node headers is used by internal nodes
591da177e4SLinus Torvalds  * to point to another node that "effectively continues this one". Here is what
601da177e4SLinus Torvalds  * I believe that means. Each key in internal nodes points to another node that
611da177e4SLinus Torvalds  * contains key values less than itself. Inspection reveals that the last key
621da177e4SLinus Torvalds  * in the internal node is not the last key in the index. Keys that are
631da177e4SLinus Torvalds  * greater than the last key in the internal node go into the overflow node.
641da177e4SLinus Torvalds  * I imagine there is a performance reason for this.
651da177e4SLinus Torvalds  *
661da177e4SLinus Torvalds  * Second, it states that the header of a btree node is sufficient to
671da177e4SLinus Torvalds  * distinguish internal nodes from leaf nodes. Without saying exactly how.
681da177e4SLinus Torvalds  * After figuring out the first, it becomes obvious that internal nodes have
691da177e4SLinus Torvalds  * overflow nodes and leafnodes do not.
701da177e4SLinus Torvalds  */
711da177e4SLinus Torvalds 
721da177e4SLinus Torvalds /*
731da177e4SLinus Torvalds  * Currently, this code is only good for directory B+trees.
741da177e4SLinus Torvalds  * In order to be used for other BFS indexes, it needs to be extended to handle
751da177e4SLinus Torvalds  * duplicate keys and non-string keytypes (int32, int64, float, double).
761da177e4SLinus Torvalds  */
771da177e4SLinus Torvalds 
781da177e4SLinus Torvalds /*
791da177e4SLinus Torvalds  * In memory structure of each btree node
801da177e4SLinus Torvalds  */
810f2a84f4SHimangi Saraogi struct befs_btree_node {
82a9721f31SAl Viro 	befs_host_btree_nodehead head;	/* head of node converted to cpu byteorder */
831da177e4SLinus Torvalds 	struct buffer_head *bh;
841da177e4SLinus Torvalds 	befs_btree_nodehead *od_node;	/* on disk node */
850f2a84f4SHimangi Saraogi };
861da177e4SLinus Torvalds 
871da177e4SLinus Torvalds /* local constants */
884bb59432SSalah Triki static const befs_off_t BEFS_BT_INVAL = 0xffffffffffffffffULL;
891da177e4SLinus Torvalds 
901da177e4SLinus Torvalds /* local functions */
9122341d8fSAl Viro static int befs_btree_seekleaf(struct super_block *sb, const befs_data_stream *ds,
921da177e4SLinus Torvalds 			       befs_btree_super * bt_super,
930f2a84f4SHimangi Saraogi 			       struct befs_btree_node *this_node,
941da177e4SLinus Torvalds 			       befs_off_t * node_off);
951da177e4SLinus Torvalds 
9622341d8fSAl Viro static int befs_bt_read_super(struct super_block *sb, const befs_data_stream *ds,
971da177e4SLinus Torvalds 			      befs_btree_super * sup);
981da177e4SLinus Torvalds 
9922341d8fSAl Viro static int befs_bt_read_node(struct super_block *sb, const befs_data_stream *ds,
1000f2a84f4SHimangi Saraogi 			     struct befs_btree_node *node,
1010f2a84f4SHimangi Saraogi 			     befs_off_t node_off);
1021da177e4SLinus Torvalds 
1030f2a84f4SHimangi Saraogi static int befs_leafnode(struct befs_btree_node *node);
1041da177e4SLinus Torvalds 
1050f2a84f4SHimangi Saraogi static fs16 *befs_bt_keylen_index(struct befs_btree_node *node);
1061da177e4SLinus Torvalds 
1070f2a84f4SHimangi Saraogi static fs64 *befs_bt_valarray(struct befs_btree_node *node);
1081da177e4SLinus Torvalds 
1090f2a84f4SHimangi Saraogi static char *befs_bt_keydata(struct befs_btree_node *node);
1101da177e4SLinus Torvalds 
1110f2a84f4SHimangi Saraogi static int befs_find_key(struct super_block *sb,
1120f2a84f4SHimangi Saraogi 			 struct befs_btree_node *node,
1131da177e4SLinus Torvalds 			 const char *findkey, befs_off_t * value);
1141da177e4SLinus Torvalds 
1150f2a84f4SHimangi Saraogi static char *befs_bt_get_key(struct super_block *sb,
1160f2a84f4SHimangi Saraogi 			     struct befs_btree_node *node,
1171da177e4SLinus Torvalds 			     int index, u16 * keylen);
1181da177e4SLinus Torvalds 
1191da177e4SLinus Torvalds static int befs_compare_strings(const void *key1, int keylen1,
1201da177e4SLinus Torvalds 				const void *key2, int keylen2);
1211da177e4SLinus Torvalds 
1221da177e4SLinus Torvalds /**
123799ce1dbSTommy Nguyen  * befs_bt_read_super() - read in btree superblock convert to cpu byteorder
1241da177e4SLinus Torvalds  * @sb:        Filesystem superblock
1251da177e4SLinus Torvalds  * @ds:        Datastream to read from
1261da177e4SLinus Torvalds  * @sup:       Buffer in which to place the btree superblock
1271da177e4SLinus Torvalds  *
1281da177e4SLinus Torvalds  * Calls befs_read_datastream to read in the btree superblock and
1291da177e4SLinus Torvalds  * makes sure it is in cpu byteorder, byteswapping if necessary.
130799ce1dbSTommy Nguyen  * Return: BEFS_OK on success and if *@sup contains the btree superblock in cpu
131799ce1dbSTommy Nguyen  * byte order. Otherwise return BEFS_ERR on error.
1321da177e4SLinus Torvalds  */
1331da177e4SLinus Torvalds static int
befs_bt_read_super(struct super_block * sb,const befs_data_stream * ds,befs_btree_super * sup)13422341d8fSAl Viro befs_bt_read_super(struct super_block *sb, const befs_data_stream *ds,
1351da177e4SLinus Torvalds 		   befs_btree_super * sup)
1361da177e4SLinus Torvalds {
1379f0f564aSFabian Frederick 	struct buffer_head *bh;
1389f0f564aSFabian Frederick 	befs_disk_btree_super *od_sup;
1391da177e4SLinus Torvalds 
140dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s", __func__);
1411da177e4SLinus Torvalds 
1421da177e4SLinus Torvalds 	bh = befs_read_datastream(sb, ds, 0, NULL);
1431da177e4SLinus Torvalds 
1441da177e4SLinus Torvalds 	if (!bh) {
1451da177e4SLinus Torvalds 		befs_error(sb, "Couldn't read index header.");
1461da177e4SLinus Torvalds 		goto error;
1471da177e4SLinus Torvalds 	}
148a9721f31SAl Viro 	od_sup = (befs_disk_btree_super *) bh->b_data;
1491da177e4SLinus Torvalds 	befs_dump_index_entry(sb, od_sup);
1501da177e4SLinus Torvalds 
1511da177e4SLinus Torvalds 	sup->magic = fs32_to_cpu(sb, od_sup->magic);
1521da177e4SLinus Torvalds 	sup->node_size = fs32_to_cpu(sb, od_sup->node_size);
1531da177e4SLinus Torvalds 	sup->max_depth = fs32_to_cpu(sb, od_sup->max_depth);
1541da177e4SLinus Torvalds 	sup->data_type = fs32_to_cpu(sb, od_sup->data_type);
1551da177e4SLinus Torvalds 	sup->root_node_ptr = fs64_to_cpu(sb, od_sup->root_node_ptr);
1561da177e4SLinus Torvalds 
1571da177e4SLinus Torvalds 	brelse(bh);
1581da177e4SLinus Torvalds 	if (sup->magic != BEFS_BTREE_MAGIC) {
1591da177e4SLinus Torvalds 		befs_error(sb, "Index header has bad magic.");
1601da177e4SLinus Torvalds 		goto error;
1611da177e4SLinus Torvalds 	}
1621da177e4SLinus Torvalds 
163dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s", __func__);
1641da177e4SLinus Torvalds 	return BEFS_OK;
1651da177e4SLinus Torvalds 
1661da177e4SLinus Torvalds       error:
167dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s ERROR", __func__);
1681da177e4SLinus Torvalds 	return BEFS_ERR;
1691da177e4SLinus Torvalds }
1701da177e4SLinus Torvalds 
1711da177e4SLinus Torvalds /**
1721da177e4SLinus Torvalds  * befs_bt_read_node - read in btree node and convert to cpu byteorder
1731da177e4SLinus Torvalds  * @sb: Filesystem superblock
1741da177e4SLinus Torvalds  * @ds: Datastream to read from
1751da177e4SLinus Torvalds  * @node: Buffer in which to place the btree node
1761da177e4SLinus Torvalds  * @node_off: Starting offset (in bytes) of the node in @ds
1771da177e4SLinus Torvalds  *
1781da177e4SLinus Torvalds  * Calls befs_read_datastream to read in the indicated btree node and
1791da177e4SLinus Torvalds  * makes sure its header fields are in cpu byteorder, byteswapping if
1801da177e4SLinus Torvalds  * necessary.
1812dfa8a6eSLuis de Bethencourt  * Note: node->bh must be NULL when this function is called the first time.
1822dfa8a6eSLuis de Bethencourt  * Don't forget brelse(node->bh) after last call.
1831da177e4SLinus Torvalds  *
1841da177e4SLinus Torvalds  * On success, returns BEFS_OK and *@node contains the btree node that
1851da177e4SLinus Torvalds  * starts at @node_off, with the node->head fields in cpu byte order.
1861da177e4SLinus Torvalds  *
1871da177e4SLinus Torvalds  * On failure, BEFS_ERR is returned.
1881da177e4SLinus Torvalds  */
1891da177e4SLinus Torvalds 
1901da177e4SLinus Torvalds static int
befs_bt_read_node(struct super_block * sb,const befs_data_stream * ds,struct befs_btree_node * node,befs_off_t node_off)19122341d8fSAl Viro befs_bt_read_node(struct super_block *sb, const befs_data_stream *ds,
1920f2a84f4SHimangi Saraogi 		  struct befs_btree_node *node, befs_off_t node_off)
1931da177e4SLinus Torvalds {
1941da177e4SLinus Torvalds 	uint off = 0;
1951da177e4SLinus Torvalds 
196dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s", __func__);
1971da177e4SLinus Torvalds 
1981da177e4SLinus Torvalds 	if (node->bh)
1991da177e4SLinus Torvalds 		brelse(node->bh);
2001da177e4SLinus Torvalds 
2011da177e4SLinus Torvalds 	node->bh = befs_read_datastream(sb, ds, node_off, &off);
2021da177e4SLinus Torvalds 	if (!node->bh) {
203dac52fc1SFabian Frederick 		befs_error(sb, "%s failed to read "
204dac52fc1SFabian Frederick 			   "node at %llu", __func__, node_off);
205dac52fc1SFabian Frederick 		befs_debug(sb, "<--- %s ERROR", __func__);
2061da177e4SLinus Torvalds 
2071da177e4SLinus Torvalds 		return BEFS_ERR;
2081da177e4SLinus Torvalds 	}
2091da177e4SLinus Torvalds 	node->od_node =
2101da177e4SLinus Torvalds 	    (befs_btree_nodehead *) ((void *) node->bh->b_data + off);
2111da177e4SLinus Torvalds 
2121da177e4SLinus Torvalds 	befs_dump_index_node(sb, node->od_node);
2131da177e4SLinus Torvalds 
2141da177e4SLinus Torvalds 	node->head.left = fs64_to_cpu(sb, node->od_node->left);
2151da177e4SLinus Torvalds 	node->head.right = fs64_to_cpu(sb, node->od_node->right);
2161da177e4SLinus Torvalds 	node->head.overflow = fs64_to_cpu(sb, node->od_node->overflow);
2171da177e4SLinus Torvalds 	node->head.all_key_count =
2181da177e4SLinus Torvalds 	    fs16_to_cpu(sb, node->od_node->all_key_count);
2191da177e4SLinus Torvalds 	node->head.all_key_length =
2201da177e4SLinus Torvalds 	    fs16_to_cpu(sb, node->od_node->all_key_length);
2211da177e4SLinus Torvalds 
222dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s", __func__);
2231da177e4SLinus Torvalds 	return BEFS_OK;
2241da177e4SLinus Torvalds }
2251da177e4SLinus Torvalds 
2261da177e4SLinus Torvalds /**
2271da177e4SLinus Torvalds  * befs_btree_find - Find a key in a befs B+tree
2281da177e4SLinus Torvalds  * @sb: Filesystem superblock
2291da177e4SLinus Torvalds  * @ds: Datastream containing btree
2301da177e4SLinus Torvalds  * @key: Key string to lookup in btree
2311da177e4SLinus Torvalds  * @value: Value stored with @key
2321da177e4SLinus Torvalds  *
233c78bad11SJoe Perches  * On success, returns BEFS_OK and sets *@value to the value stored
2341da177e4SLinus Torvalds  * with @key (usually the disk block number of an inode).
2351da177e4SLinus Torvalds  *
2361da177e4SLinus Torvalds  * On failure, returns BEFS_ERR or BEFS_BT_NOT_FOUND.
2371da177e4SLinus Torvalds  *
2381da177e4SLinus Torvalds  * Algorithm:
2391da177e4SLinus Torvalds  *   Read the superblock and rootnode of the b+tree.
2401da177e4SLinus Torvalds  *   Drill down through the interior nodes using befs_find_key().
2411da177e4SLinus Torvalds  *   Once at the correct leaf node, use befs_find_key() again to get the
24202d91f97SLuis de Bethencourt  *   actual value stored with the key.
2431da177e4SLinus Torvalds  */
2441da177e4SLinus Torvalds int
befs_btree_find(struct super_block * sb,const befs_data_stream * ds,const char * key,befs_off_t * value)24522341d8fSAl Viro befs_btree_find(struct super_block *sb, const befs_data_stream *ds,
2461da177e4SLinus Torvalds 		const char *key, befs_off_t * value)
2471da177e4SLinus Torvalds {
2489f0f564aSFabian Frederick 	struct befs_btree_node *this_node;
2491da177e4SLinus Torvalds 	befs_btree_super bt_super;
2501da177e4SLinus Torvalds 	befs_off_t node_off;
2511da177e4SLinus Torvalds 	int res;
2521da177e4SLinus Torvalds 
253dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s Key: %s", __func__, key);
2541da177e4SLinus Torvalds 
2551da177e4SLinus Torvalds 	if (befs_bt_read_super(sb, ds, &bt_super) != BEFS_OK) {
2561da177e4SLinus Torvalds 		befs_error(sb,
2571da177e4SLinus Torvalds 			   "befs_btree_find() failed to read index superblock");
2581da177e4SLinus Torvalds 		goto error;
2591da177e4SLinus Torvalds 	}
2601da177e4SLinus Torvalds 
2610f2a84f4SHimangi Saraogi 	this_node = kmalloc(sizeof(struct befs_btree_node),
2621da177e4SLinus Torvalds 						GFP_NOFS);
2631da177e4SLinus Torvalds 	if (!this_node) {
264dac52fc1SFabian Frederick 		befs_error(sb, "befs_btree_find() failed to allocate %zu "
2650f2a84f4SHimangi Saraogi 			   "bytes of memory", sizeof(struct befs_btree_node));
2661da177e4SLinus Torvalds 		goto error;
2671da177e4SLinus Torvalds 	}
2681da177e4SLinus Torvalds 
2691da177e4SLinus Torvalds 	this_node->bh = NULL;
2701da177e4SLinus Torvalds 
2711da177e4SLinus Torvalds 	/* read in root node */
2721da177e4SLinus Torvalds 	node_off = bt_super.root_node_ptr;
2731da177e4SLinus Torvalds 	if (befs_bt_read_node(sb, ds, this_node, node_off) != BEFS_OK) {
2741da177e4SLinus Torvalds 		befs_error(sb, "befs_btree_find() failed to read "
275dac52fc1SFabian Frederick 			   "node at %llu", node_off);
2761da177e4SLinus Torvalds 		goto error_alloc;
2771da177e4SLinus Torvalds 	}
2781da177e4SLinus Torvalds 
2791da177e4SLinus Torvalds 	while (!befs_leafnode(this_node)) {
2801da177e4SLinus Torvalds 		res = befs_find_key(sb, this_node, key, &node_off);
281672a8515SLuis de Bethencourt 		/* if no key set, try the overflow node */
282672a8515SLuis de Bethencourt 		if (res == BEFS_BT_OVERFLOW)
2831da177e4SLinus Torvalds 			node_off = this_node->head.overflow;
2841da177e4SLinus Torvalds 		if (befs_bt_read_node(sb, ds, this_node, node_off) != BEFS_OK) {
2851da177e4SLinus Torvalds 			befs_error(sb, "befs_btree_find() failed to read "
286dac52fc1SFabian Frederick 				   "node at %llu", node_off);
2871da177e4SLinus Torvalds 			goto error_alloc;
2881da177e4SLinus Torvalds 		}
2891da177e4SLinus Torvalds 	}
2901da177e4SLinus Torvalds 
291672a8515SLuis de Bethencourt 	/* at a leaf node now, check if it is correct */
2921da177e4SLinus Torvalds 	res = befs_find_key(sb, this_node, key, value);
2931da177e4SLinus Torvalds 
2941da177e4SLinus Torvalds 	brelse(this_node->bh);
2951da177e4SLinus Torvalds 	kfree(this_node);
2961da177e4SLinus Torvalds 
2971da177e4SLinus Torvalds 	if (res != BEFS_BT_MATCH) {
2984c3897ccSLuis de Bethencourt 		befs_error(sb, "<--- %s Key %s not found", __func__, key);
2994c3897ccSLuis de Bethencourt 		befs_debug(sb, "<--- %s ERROR", __func__);
3001da177e4SLinus Torvalds 		*value = 0;
3011da177e4SLinus Torvalds 		return BEFS_BT_NOT_FOUND;
3021da177e4SLinus Torvalds 	}
303dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s Found key %s, value %llu", __func__,
3041da177e4SLinus Torvalds 		   key, *value);
3051da177e4SLinus Torvalds 	return BEFS_OK;
3061da177e4SLinus Torvalds 
3071da177e4SLinus Torvalds       error_alloc:
3081da177e4SLinus Torvalds 	kfree(this_node);
3091da177e4SLinus Torvalds       error:
3101da177e4SLinus Torvalds 	*value = 0;
311dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s ERROR", __func__);
3121da177e4SLinus Torvalds 	return BEFS_ERR;
3131da177e4SLinus Torvalds }
3141da177e4SLinus Torvalds 
3151da177e4SLinus Torvalds /**
3161da177e4SLinus Torvalds  * befs_find_key - Search for a key within a node
3171da177e4SLinus Torvalds  * @sb: Filesystem superblock
3181da177e4SLinus Torvalds  * @node: Node to find the key within
319817e1d90SFabian Frederick  * @findkey: Keystring to search for
3201da177e4SLinus Torvalds  * @value: If key is found, the value stored with the key is put here
3211da177e4SLinus Torvalds  *
322672a8515SLuis de Bethencourt  * Finds exact match if one exists, and returns BEFS_BT_MATCH.
323672a8515SLuis de Bethencourt  * If there is no match and node's value array is too small for key, return
324672a8515SLuis de Bethencourt  * BEFS_BT_OVERFLOW.
325672a8515SLuis de Bethencourt  * If no match and node should countain this key, return BEFS_BT_NOT_FOUND.
3261da177e4SLinus Torvalds  *
327672a8515SLuis de Bethencourt  * Uses binary search instead of a linear.
3281da177e4SLinus Torvalds  */
3291da177e4SLinus Torvalds static int
befs_find_key(struct super_block * sb,struct befs_btree_node * node,const char * findkey,befs_off_t * value)3300f2a84f4SHimangi Saraogi befs_find_key(struct super_block *sb, struct befs_btree_node *node,
3311da177e4SLinus Torvalds 	      const char *findkey, befs_off_t * value)
3321da177e4SLinus Torvalds {
3331da177e4SLinus Torvalds 	int first, last, mid;
3341da177e4SLinus Torvalds 	int eq;
3351da177e4SLinus Torvalds 	u16 keylen;
3361da177e4SLinus Torvalds 	int findkey_len;
3371da177e4SLinus Torvalds 	char *thiskey;
338a9721f31SAl Viro 	fs64 *valarray;
3391da177e4SLinus Torvalds 
340dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s %s", __func__, findkey);
3411da177e4SLinus Torvalds 
3421da177e4SLinus Torvalds 	findkey_len = strlen(findkey);
3431da177e4SLinus Torvalds 
344bb75e666SLuis de Bethencourt 	/* if node can not contain key, just skip this node */
3451da177e4SLinus Torvalds 	last = node->head.all_key_count - 1;
3461da177e4SLinus Torvalds 	thiskey = befs_bt_get_key(sb, node, last, &keylen);
3471da177e4SLinus Torvalds 
3481da177e4SLinus Torvalds 	eq = befs_compare_strings(thiskey, keylen, findkey, findkey_len);
3491da177e4SLinus Torvalds 	if (eq < 0) {
350672a8515SLuis de Bethencourt 		befs_debug(sb, "<--- node can't contain %s", findkey);
351672a8515SLuis de Bethencourt 		return BEFS_BT_OVERFLOW;
3521da177e4SLinus Torvalds 	}
3531da177e4SLinus Torvalds 
3541da177e4SLinus Torvalds 	valarray = befs_bt_valarray(node);
3551da177e4SLinus Torvalds 
3561da177e4SLinus Torvalds 	/* simple binary search */
3571da177e4SLinus Torvalds 	first = 0;
3581da177e4SLinus Torvalds 	mid = 0;
3591da177e4SLinus Torvalds 	while (last >= first) {
3601da177e4SLinus Torvalds 		mid = (last + first) / 2;
3611da177e4SLinus Torvalds 		befs_debug(sb, "first: %d, last: %d, mid: %d", first, last,
3621da177e4SLinus Torvalds 			   mid);
3631da177e4SLinus Torvalds 		thiskey = befs_bt_get_key(sb, node, mid, &keylen);
3641da177e4SLinus Torvalds 		eq = befs_compare_strings(thiskey, keylen, findkey,
3651da177e4SLinus Torvalds 					  findkey_len);
3661da177e4SLinus Torvalds 
3671da177e4SLinus Torvalds 		if (eq == 0) {
368dac52fc1SFabian Frederick 			befs_debug(sb, "<--- %s found %s at %d",
369dac52fc1SFabian Frederick 				   __func__, thiskey, mid);
3701da177e4SLinus Torvalds 
3711da177e4SLinus Torvalds 			*value = fs64_to_cpu(sb, valarray[mid]);
3721da177e4SLinus Torvalds 			return BEFS_BT_MATCH;
3731da177e4SLinus Torvalds 		}
3741da177e4SLinus Torvalds 		if (eq > 0)
3751da177e4SLinus Torvalds 			last = mid - 1;
3761da177e4SLinus Torvalds 		else
3771da177e4SLinus Torvalds 			first = mid + 1;
3781da177e4SLinus Torvalds 	}
379672a8515SLuis de Bethencourt 
380672a8515SLuis de Bethencourt 	/* return an existing value so caller can arrive to a leaf node */
3811da177e4SLinus Torvalds 	if (eq < 0)
3821da177e4SLinus Torvalds 		*value = fs64_to_cpu(sb, valarray[mid + 1]);
3831da177e4SLinus Torvalds 	else
3841da177e4SLinus Torvalds 		*value = fs64_to_cpu(sb, valarray[mid]);
385672a8515SLuis de Bethencourt 	befs_error(sb, "<--- %s %s not found", __func__, findkey);
386672a8515SLuis de Bethencourt 	befs_debug(sb, "<--- %s ERROR", __func__);
387672a8515SLuis de Bethencourt 	return BEFS_BT_NOT_FOUND;
3881da177e4SLinus Torvalds }
3891da177e4SLinus Torvalds 
3901da177e4SLinus Torvalds /**
3911da177e4SLinus Torvalds  * befs_btree_read - Traverse leafnodes of a btree
3921da177e4SLinus Torvalds  * @sb: Filesystem superblock
3931da177e4SLinus Torvalds  * @ds: Datastream containing btree
3941da177e4SLinus Torvalds  * @key_no: Key number (alphabetical order) of key to read
3951da177e4SLinus Torvalds  * @bufsize: Size of the buffer to return key in
3961da177e4SLinus Torvalds  * @keybuf: Pointer to a buffer to put the key in
3971da177e4SLinus Torvalds  * @keysize: Length of the returned key
3981da177e4SLinus Torvalds  * @value: Value stored with the returned key
3991da177e4SLinus Torvalds  *
40002d91f97SLuis de Bethencourt  * Here's how it works: Key_no is the index of the key/value pair to
4011da177e4SLinus Torvalds  * return in keybuf/value.
4021da177e4SLinus Torvalds  * Bufsize is the size of keybuf (BEFS_NAME_LEN+1 is a good size). Keysize is
4036cb103b6SFabian Frederick  * the number of characters in the key (just a convenience).
4041da177e4SLinus Torvalds  *
4051da177e4SLinus Torvalds  * Algorithm:
4061da177e4SLinus Torvalds  *   Get the first leafnode of the tree. See if the requested key is in that
4071da177e4SLinus Torvalds  *   node. If not, follow the node->right link to the next leafnode. Repeat
4081da177e4SLinus Torvalds  *   until the (key_no)th key is found or the tree is out of keys.
4091da177e4SLinus Torvalds  */
4101da177e4SLinus Torvalds int
befs_btree_read(struct super_block * sb,const befs_data_stream * ds,loff_t key_no,size_t bufsize,char * keybuf,size_t * keysize,befs_off_t * value)41122341d8fSAl Viro befs_btree_read(struct super_block *sb, const befs_data_stream *ds,
4121da177e4SLinus Torvalds 		loff_t key_no, size_t bufsize, char *keybuf, size_t * keysize,
4131da177e4SLinus Torvalds 		befs_off_t * value)
4141da177e4SLinus Torvalds {
4150f2a84f4SHimangi Saraogi 	struct befs_btree_node *this_node;
4161da177e4SLinus Torvalds 	befs_btree_super bt_super;
417143d2a61SSalah Triki 	befs_off_t node_off;
4181da177e4SLinus Torvalds 	int cur_key;
419a9721f31SAl Viro 	fs64 *valarray;
4201da177e4SLinus Torvalds 	char *keystart;
4211da177e4SLinus Torvalds 	u16 keylen;
4221da177e4SLinus Torvalds 	int res;
4231da177e4SLinus Torvalds 
4241da177e4SLinus Torvalds 	uint key_sum = 0;
4251da177e4SLinus Torvalds 
426dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s", __func__);
4271da177e4SLinus Torvalds 
4281da177e4SLinus Torvalds 	if (befs_bt_read_super(sb, ds, &bt_super) != BEFS_OK) {
4291da177e4SLinus Torvalds 		befs_error(sb,
4301da177e4SLinus Torvalds 			   "befs_btree_read() failed to read index superblock");
4311da177e4SLinus Torvalds 		goto error;
4321da177e4SLinus Torvalds 	}
4331da177e4SLinus Torvalds 
4340f2a84f4SHimangi Saraogi 	this_node = kmalloc(sizeof(struct befs_btree_node), GFP_NOFS);
4350f2a84f4SHimangi Saraogi 	if (this_node == NULL) {
436dac52fc1SFabian Frederick 		befs_error(sb, "befs_btree_read() failed to allocate %zu "
4370f2a84f4SHimangi Saraogi 			   "bytes of memory", sizeof(struct befs_btree_node));
4381da177e4SLinus Torvalds 		goto error;
4391da177e4SLinus Torvalds 	}
4401da177e4SLinus Torvalds 
4411da177e4SLinus Torvalds 	node_off = bt_super.root_node_ptr;
4421da177e4SLinus Torvalds 	this_node->bh = NULL;
4431da177e4SLinus Torvalds 
4441da177e4SLinus Torvalds 	/* seeks down to first leafnode, reads it into this_node */
4451da177e4SLinus Torvalds 	res = befs_btree_seekleaf(sb, ds, &bt_super, this_node, &node_off);
4461da177e4SLinus Torvalds 	if (res == BEFS_BT_EMPTY) {
4471da177e4SLinus Torvalds 		brelse(this_node->bh);
4481da177e4SLinus Torvalds 		kfree(this_node);
4491da177e4SLinus Torvalds 		*value = 0;
4501da177e4SLinus Torvalds 		*keysize = 0;
451dac52fc1SFabian Frederick 		befs_debug(sb, "<--- %s Tree is EMPTY", __func__);
4521da177e4SLinus Torvalds 		return BEFS_BT_EMPTY;
4531da177e4SLinus Torvalds 	} else if (res == BEFS_ERR) {
4541da177e4SLinus Torvalds 		goto error_alloc;
4551da177e4SLinus Torvalds 	}
4561da177e4SLinus Torvalds 
4571da177e4SLinus Torvalds 	/* find the leaf node containing the key_no key */
4581da177e4SLinus Torvalds 
4591da177e4SLinus Torvalds 	while (key_sum + this_node->head.all_key_count <= key_no) {
4601da177e4SLinus Torvalds 
4611da177e4SLinus Torvalds 		/* no more nodes to look in: key_no is too large */
4624bb59432SSalah Triki 		if (this_node->head.right == BEFS_BT_INVAL) {
4631da177e4SLinus Torvalds 			*keysize = 0;
4641da177e4SLinus Torvalds 			*value = 0;
4651da177e4SLinus Torvalds 			befs_debug(sb,
466dac52fc1SFabian Frederick 				   "<--- %s END of keys at %llu", __func__,
467dac52fc1SFabian Frederick 				   (unsigned long long)
4681da177e4SLinus Torvalds 				   key_sum + this_node->head.all_key_count);
4691da177e4SLinus Torvalds 			brelse(this_node->bh);
4701da177e4SLinus Torvalds 			kfree(this_node);
4711da177e4SLinus Torvalds 			return BEFS_BT_END;
4721da177e4SLinus Torvalds 		}
4731da177e4SLinus Torvalds 
4741da177e4SLinus Torvalds 		key_sum += this_node->head.all_key_count;
4751da177e4SLinus Torvalds 		node_off = this_node->head.right;
4761da177e4SLinus Torvalds 
4771da177e4SLinus Torvalds 		if (befs_bt_read_node(sb, ds, this_node, node_off) != BEFS_OK) {
478dac52fc1SFabian Frederick 			befs_error(sb, "%s failed to read node at %llu",
479dac52fc1SFabian Frederick 				  __func__, (unsigned long long)node_off);
4801da177e4SLinus Torvalds 			goto error_alloc;
4811da177e4SLinus Torvalds 		}
4821da177e4SLinus Torvalds 	}
4831da177e4SLinus Torvalds 
4841da177e4SLinus Torvalds 	/* how many keys into this_node is key_no */
4851da177e4SLinus Torvalds 	cur_key = key_no - key_sum;
4861da177e4SLinus Torvalds 
4871da177e4SLinus Torvalds 	/* get pointers to datastructures within the node body */
4881da177e4SLinus Torvalds 	valarray = befs_bt_valarray(this_node);
4891da177e4SLinus Torvalds 
4901da177e4SLinus Torvalds 	keystart = befs_bt_get_key(sb, this_node, cur_key, &keylen);
4911da177e4SLinus Torvalds 
492dac52fc1SFabian Frederick 	befs_debug(sb, "Read [%llu,%d]: keysize %d",
493dac52fc1SFabian Frederick 		   (long long unsigned int)node_off, (int)cur_key,
494dac52fc1SFabian Frederick 		   (int)keylen);
4951da177e4SLinus Torvalds 
4961da177e4SLinus Torvalds 	if (bufsize < keylen + 1) {
497dac52fc1SFabian Frederick 		befs_error(sb, "%s keybuf too small (%zu) "
498dac52fc1SFabian Frederick 			   "for key of size %d", __func__, bufsize, keylen);
4991da177e4SLinus Torvalds 		brelse(this_node->bh);
5001da177e4SLinus Torvalds 		goto error_alloc;
5016cb103b6SFabian Frederick 	}
5021da177e4SLinus Torvalds 
503*73919280SAzeem Shaikh 	strscpy(keybuf, keystart, keylen + 1);
5041da177e4SLinus Torvalds 	*value = fs64_to_cpu(sb, valarray[cur_key]);
5051da177e4SLinus Torvalds 	*keysize = keylen;
5061da177e4SLinus Torvalds 
507dac52fc1SFabian Frederick 	befs_debug(sb, "Read [%llu,%d]: Key \"%.*s\", Value %llu", node_off,
5081da177e4SLinus Torvalds 		   cur_key, keylen, keybuf, *value);
5091da177e4SLinus Torvalds 
5101da177e4SLinus Torvalds 	brelse(this_node->bh);
5111da177e4SLinus Torvalds 	kfree(this_node);
5121da177e4SLinus Torvalds 
513dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s", __func__);
5141da177e4SLinus Torvalds 
5151da177e4SLinus Torvalds 	return BEFS_OK;
5161da177e4SLinus Torvalds 
5171da177e4SLinus Torvalds       error_alloc:
5181da177e4SLinus Torvalds 	kfree(this_node);
5191da177e4SLinus Torvalds 
5201da177e4SLinus Torvalds       error:
5211da177e4SLinus Torvalds 	*keysize = 0;
5221da177e4SLinus Torvalds 	*value = 0;
523dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s ERROR", __func__);
5241da177e4SLinus Torvalds 	return BEFS_ERR;
5251da177e4SLinus Torvalds }
5261da177e4SLinus Torvalds 
5271da177e4SLinus Torvalds /**
5281da177e4SLinus Torvalds  * befs_btree_seekleaf - Find the first leafnode in the btree
5291da177e4SLinus Torvalds  * @sb: Filesystem superblock
5301da177e4SLinus Torvalds  * @ds: Datastream containing btree
5311da177e4SLinus Torvalds  * @bt_super: Pointer to the superblock of the btree
5321da177e4SLinus Torvalds  * @this_node: Buffer to return the leafnode in
5331da177e4SLinus Torvalds  * @node_off: Pointer to offset of current node within datastream. Modified
5341da177e4SLinus Torvalds  * 		by the function.
5351da177e4SLinus Torvalds  *
5361da177e4SLinus Torvalds  * Helper function for btree traverse. Moves the current position to the
5371da177e4SLinus Torvalds  * start of the first leaf node.
5381da177e4SLinus Torvalds  *
5391da177e4SLinus Torvalds  * Also checks for an empty tree. If there are no keys, returns BEFS_BT_EMPTY.
5401da177e4SLinus Torvalds  */
5411da177e4SLinus Torvalds static int
befs_btree_seekleaf(struct super_block * sb,const befs_data_stream * ds,befs_btree_super * bt_super,struct befs_btree_node * this_node,befs_off_t * node_off)54222341d8fSAl Viro befs_btree_seekleaf(struct super_block *sb, const befs_data_stream *ds,
5430f2a84f4SHimangi Saraogi 		    befs_btree_super *bt_super,
5440f2a84f4SHimangi Saraogi 		    struct befs_btree_node *this_node,
5451da177e4SLinus Torvalds 		    befs_off_t * node_off)
5461da177e4SLinus Torvalds {
5471da177e4SLinus Torvalds 
548dac52fc1SFabian Frederick 	befs_debug(sb, "---> %s", __func__);
5491da177e4SLinus Torvalds 
5501da177e4SLinus Torvalds 	if (befs_bt_read_node(sb, ds, this_node, *node_off) != BEFS_OK) {
551dac52fc1SFabian Frederick 		befs_error(sb, "%s failed to read "
552dac52fc1SFabian Frederick 			   "node at %llu", __func__, *node_off);
5531da177e4SLinus Torvalds 		goto error;
5541da177e4SLinus Torvalds 	}
555dac52fc1SFabian Frederick 	befs_debug(sb, "Seekleaf to root node %llu", *node_off);
5561da177e4SLinus Torvalds 
5571da177e4SLinus Torvalds 	if (this_node->head.all_key_count == 0 && befs_leafnode(this_node)) {
558dac52fc1SFabian Frederick 		befs_debug(sb, "<--- %s Tree is EMPTY", __func__);
5591da177e4SLinus Torvalds 		return BEFS_BT_EMPTY;
5601da177e4SLinus Torvalds 	}
5611da177e4SLinus Torvalds 
5621da177e4SLinus Torvalds 	while (!befs_leafnode(this_node)) {
5631da177e4SLinus Torvalds 
5641da177e4SLinus Torvalds 		if (this_node->head.all_key_count == 0) {
565dac52fc1SFabian Frederick 			befs_debug(sb, "%s encountered "
566dac52fc1SFabian Frederick 				   "an empty interior node: %llu. Using Overflow "
567dac52fc1SFabian Frederick 				   "node: %llu", __func__, *node_off,
5681da177e4SLinus Torvalds 				   this_node->head.overflow);
5691da177e4SLinus Torvalds 			*node_off = this_node->head.overflow;
5701da177e4SLinus Torvalds 		} else {
571a9721f31SAl Viro 			fs64 *valarray = befs_bt_valarray(this_node);
5721da177e4SLinus Torvalds 			*node_off = fs64_to_cpu(sb, valarray[0]);
5731da177e4SLinus Torvalds 		}
5741da177e4SLinus Torvalds 		if (befs_bt_read_node(sb, ds, this_node, *node_off) != BEFS_OK) {
575dac52fc1SFabian Frederick 			befs_error(sb, "%s failed to read "
576dac52fc1SFabian Frederick 				   "node at %llu", __func__, *node_off);
5771da177e4SLinus Torvalds 			goto error;
5781da177e4SLinus Torvalds 		}
5791da177e4SLinus Torvalds 
580dac52fc1SFabian Frederick 		befs_debug(sb, "Seekleaf to child node %llu", *node_off);
5811da177e4SLinus Torvalds 	}
582dac52fc1SFabian Frederick 	befs_debug(sb, "Node %llu is a leaf node", *node_off);
5831da177e4SLinus Torvalds 
5841da177e4SLinus Torvalds 	return BEFS_OK;
5851da177e4SLinus Torvalds 
5861da177e4SLinus Torvalds       error:
587dac52fc1SFabian Frederick 	befs_debug(sb, "<--- %s ERROR", __func__);
5881da177e4SLinus Torvalds 	return BEFS_ERR;
5891da177e4SLinus Torvalds }
5901da177e4SLinus Torvalds 
5911da177e4SLinus Torvalds /**
5921da177e4SLinus Torvalds  * befs_leafnode - Determine if the btree node is a leaf node or an
5931da177e4SLinus Torvalds  * interior node
5941da177e4SLinus Torvalds  * @node: Pointer to node structure to test
5951da177e4SLinus Torvalds  *
5961da177e4SLinus Torvalds  * Return 1 if leaf, 0 if interior
5971da177e4SLinus Torvalds  */
5981da177e4SLinus Torvalds static int
befs_leafnode(struct befs_btree_node * node)5990f2a84f4SHimangi Saraogi befs_leafnode(struct befs_btree_node *node)
6001da177e4SLinus Torvalds {
6011da177e4SLinus Torvalds 	/* all interior nodes (and only interior nodes) have an overflow node */
6024bb59432SSalah Triki 	if (node->head.overflow == BEFS_BT_INVAL)
6031da177e4SLinus Torvalds 		return 1;
6041da177e4SLinus Torvalds 	else
6051da177e4SLinus Torvalds 		return 0;
6061da177e4SLinus Torvalds }
6071da177e4SLinus Torvalds 
6081da177e4SLinus Torvalds /**
6091da177e4SLinus Torvalds  * befs_bt_keylen_index - Finds start of keylen index in a node
6101da177e4SLinus Torvalds  * @node: Pointer to the node structure to find the keylen index within
6111da177e4SLinus Torvalds  *
6121da177e4SLinus Torvalds  * Returns a pointer to the start of the key length index array
6131da177e4SLinus Torvalds  * of the B+tree node *@node
6141da177e4SLinus Torvalds  *
6151da177e4SLinus Torvalds  * "The length of all the keys in the node is added to the size of the
6161da177e4SLinus Torvalds  * header and then rounded up to a multiple of four to get the beginning
6171da177e4SLinus Torvalds  * of the key length index" (p.88, practical filesystem design).
6181da177e4SLinus Torvalds  *
6191da177e4SLinus Torvalds  * Except that rounding up to 8 works, and rounding up to 4 doesn't.
6201da177e4SLinus Torvalds  */
621a9721f31SAl Viro static fs16 *
befs_bt_keylen_index(struct befs_btree_node * node)6220f2a84f4SHimangi Saraogi befs_bt_keylen_index(struct befs_btree_node *node)
6231da177e4SLinus Torvalds {
6241da177e4SLinus Torvalds 	const int keylen_align = 8;
6251da177e4SLinus Torvalds 	unsigned long int off =
6261da177e4SLinus Torvalds 	    (sizeof (befs_btree_nodehead) + node->head.all_key_length);
6271da177e4SLinus Torvalds 	ulong tmp = off % keylen_align;
6281da177e4SLinus Torvalds 
6291da177e4SLinus Torvalds 	if (tmp)
6301da177e4SLinus Torvalds 		off += keylen_align - tmp;
6311da177e4SLinus Torvalds 
632a9721f31SAl Viro 	return (fs16 *) ((void *) node->od_node + off);
6331da177e4SLinus Torvalds }
6341da177e4SLinus Torvalds 
6351da177e4SLinus Torvalds /**
6361da177e4SLinus Torvalds  * befs_bt_valarray - Finds the start of value array in a node
6371da177e4SLinus Torvalds  * @node: Pointer to the node structure to find the value array within
6381da177e4SLinus Torvalds  *
6391da177e4SLinus Torvalds  * Returns a pointer to the start of the value array
6401da177e4SLinus Torvalds  * of the node pointed to by the node header
6411da177e4SLinus Torvalds  */
642a9721f31SAl Viro static fs64 *
befs_bt_valarray(struct befs_btree_node * node)6430f2a84f4SHimangi Saraogi befs_bt_valarray(struct befs_btree_node *node)
6441da177e4SLinus Torvalds {
6451da177e4SLinus Torvalds 	void *keylen_index_start = (void *) befs_bt_keylen_index(node);
646a9721f31SAl Viro 	size_t keylen_index_size = node->head.all_key_count * sizeof (fs16);
6471da177e4SLinus Torvalds 
648a9721f31SAl Viro 	return (fs64 *) (keylen_index_start + keylen_index_size);
6491da177e4SLinus Torvalds }
6501da177e4SLinus Torvalds 
6511da177e4SLinus Torvalds /**
6521da177e4SLinus Torvalds  * befs_bt_keydata - Finds start of keydata array in a node
6531da177e4SLinus Torvalds  * @node: Pointer to the node structure to find the keydata array within
6541da177e4SLinus Torvalds  *
6551da177e4SLinus Torvalds  * Returns a pointer to the start of the keydata array
6561da177e4SLinus Torvalds  * of the node pointed to by the node header
6571da177e4SLinus Torvalds  */
6581da177e4SLinus Torvalds static char *
befs_bt_keydata(struct befs_btree_node * node)6590f2a84f4SHimangi Saraogi befs_bt_keydata(struct befs_btree_node *node)
6601da177e4SLinus Torvalds {
6611da177e4SLinus Torvalds 	return (char *) ((void *) node->od_node + sizeof (befs_btree_nodehead));
6621da177e4SLinus Torvalds }
6631da177e4SLinus Torvalds 
6641da177e4SLinus Torvalds /**
6651da177e4SLinus Torvalds  * befs_bt_get_key - returns a pointer to the start of a key
6661da177e4SLinus Torvalds  * @sb: filesystem superblock
6671da177e4SLinus Torvalds  * @node: node in which to look for the key
6681da177e4SLinus Torvalds  * @index: the index of the key to get
6691da177e4SLinus Torvalds  * @keylen: modified to be the length of the key at @index
6701da177e4SLinus Torvalds  *
6711da177e4SLinus Torvalds  * Returns a valid pointer into @node on success.
6721da177e4SLinus Torvalds  * Returns NULL on failure (bad input) and sets *@keylen = 0
6731da177e4SLinus Torvalds  */
6741da177e4SLinus Torvalds static char *
befs_bt_get_key(struct super_block * sb,struct befs_btree_node * node,int index,u16 * keylen)6750f2a84f4SHimangi Saraogi befs_bt_get_key(struct super_block *sb, struct befs_btree_node *node,
6761da177e4SLinus Torvalds 		int index, u16 * keylen)
6771da177e4SLinus Torvalds {
6781da177e4SLinus Torvalds 	int prev_key_end;
6791da177e4SLinus Torvalds 	char *keystart;
680a9721f31SAl Viro 	fs16 *keylen_index;
6811da177e4SLinus Torvalds 
6821da177e4SLinus Torvalds 	if (index < 0 || index > node->head.all_key_count) {
6831da177e4SLinus Torvalds 		*keylen = 0;
6841da177e4SLinus Torvalds 		return NULL;
6851da177e4SLinus Torvalds 	}
6861da177e4SLinus Torvalds 
6871da177e4SLinus Torvalds 	keystart = befs_bt_keydata(node);
6881da177e4SLinus Torvalds 	keylen_index = befs_bt_keylen_index(node);
6891da177e4SLinus Torvalds 
6901da177e4SLinus Torvalds 	if (index == 0)
6911da177e4SLinus Torvalds 		prev_key_end = 0;
6921da177e4SLinus Torvalds 	else
6931da177e4SLinus Torvalds 		prev_key_end = fs16_to_cpu(sb, keylen_index[index - 1]);
6941da177e4SLinus Torvalds 
6951da177e4SLinus Torvalds 	*keylen = fs16_to_cpu(sb, keylen_index[index]) - prev_key_end;
6961da177e4SLinus Torvalds 
6971da177e4SLinus Torvalds 	return keystart + prev_key_end;
6981da177e4SLinus Torvalds }
6991da177e4SLinus Torvalds 
7001da177e4SLinus Torvalds /**
7011da177e4SLinus Torvalds  * befs_compare_strings - compare two strings
7021da177e4SLinus Torvalds  * @key1: pointer to the first key to be compared
7031da177e4SLinus Torvalds  * @keylen1: length in bytes of key1
7041da177e4SLinus Torvalds  * @key2: pointer to the second key to be compared
705817e1d90SFabian Frederick  * @keylen2: length in bytes of key2
7061da177e4SLinus Torvalds  *
7071da177e4SLinus Torvalds  * Returns 0 if @key1 and @key2 are equal.
7081da177e4SLinus Torvalds  * Returns >0 if @key1 is greater.
70902d91f97SLuis de Bethencourt  * Returns <0 if @key2 is greater.
7101da177e4SLinus Torvalds  */
7111da177e4SLinus Torvalds static int
befs_compare_strings(const void * key1,int keylen1,const void * key2,int keylen2)7121da177e4SLinus Torvalds befs_compare_strings(const void *key1, int keylen1,
7131da177e4SLinus Torvalds 		     const void *key2, int keylen2)
7141da177e4SLinus Torvalds {
7151da177e4SLinus Torvalds 	int len = min_t(int, keylen1, keylen2);
7161da177e4SLinus Torvalds 	int result = strncmp(key1, key2, len);
7171da177e4SLinus Torvalds 	if (result == 0)
7181da177e4SLinus Torvalds 		result = keylen1 - keylen2;
7191da177e4SLinus Torvalds 	return result;
7201da177e4SLinus Torvalds }
7211da177e4SLinus Torvalds 
7221da177e4SLinus Torvalds /* These will be used for non-string keyed btrees */
7231da177e4SLinus Torvalds #if 0
7241da177e4SLinus Torvalds static int
7251da177e4SLinus Torvalds btree_compare_int32(cont void *key1, int keylen1, const void *key2, int keylen2)
7261da177e4SLinus Torvalds {
7271da177e4SLinus Torvalds 	return *(int32_t *) key1 - *(int32_t *) key2;
7281da177e4SLinus Torvalds }
7291da177e4SLinus Torvalds 
7301da177e4SLinus Torvalds static int
7311da177e4SLinus Torvalds btree_compare_uint32(cont void *key1, int keylen1,
7321da177e4SLinus Torvalds 		     const void *key2, int keylen2)
7331da177e4SLinus Torvalds {
7341da177e4SLinus Torvalds 	if (*(u_int32_t *) key1 == *(u_int32_t *) key2)
7351da177e4SLinus Torvalds 		return 0;
7361da177e4SLinus Torvalds 	else if (*(u_int32_t *) key1 > *(u_int32_t *) key2)
7371da177e4SLinus Torvalds 		return 1;
7381da177e4SLinus Torvalds 
7391da177e4SLinus Torvalds 	return -1;
7401da177e4SLinus Torvalds }
7411da177e4SLinus Torvalds static int
7421da177e4SLinus Torvalds btree_compare_int64(cont void *key1, int keylen1, const void *key2, int keylen2)
7431da177e4SLinus Torvalds {
7441da177e4SLinus Torvalds 	if (*(int64_t *) key1 == *(int64_t *) key2)
7451da177e4SLinus Torvalds 		return 0;
7461da177e4SLinus Torvalds 	else if (*(int64_t *) key1 > *(int64_t *) key2)
7471da177e4SLinus Torvalds 		return 1;
7481da177e4SLinus Torvalds 
7491da177e4SLinus Torvalds 	return -1;
7501da177e4SLinus Torvalds }
7511da177e4SLinus Torvalds 
7521da177e4SLinus Torvalds static int
7531da177e4SLinus Torvalds btree_compare_uint64(cont void *key1, int keylen1,
7541da177e4SLinus Torvalds 		     const void *key2, int keylen2)
7551da177e4SLinus Torvalds {
7561da177e4SLinus Torvalds 	if (*(u_int64_t *) key1 == *(u_int64_t *) key2)
7571da177e4SLinus Torvalds 		return 0;
7581da177e4SLinus Torvalds 	else if (*(u_int64_t *) key1 > *(u_int64_t *) key2)
7591da177e4SLinus Torvalds 		return 1;
7601da177e4SLinus Torvalds 
7611da177e4SLinus Torvalds 	return -1;
7621da177e4SLinus Torvalds }
7631da177e4SLinus Torvalds 
7641da177e4SLinus Torvalds static int
7651da177e4SLinus Torvalds btree_compare_float(cont void *key1, int keylen1, const void *key2, int keylen2)
7661da177e4SLinus Torvalds {
7671da177e4SLinus Torvalds 	float result = *(float *) key1 - *(float *) key2;
7681da177e4SLinus Torvalds 	if (result == 0.0f)
7691da177e4SLinus Torvalds 		return 0;
7701da177e4SLinus Torvalds 
7711da177e4SLinus Torvalds 	return (result < 0.0f) ? -1 : 1;
7721da177e4SLinus Torvalds }
7731da177e4SLinus Torvalds 
7741da177e4SLinus Torvalds static int
7751da177e4SLinus Torvalds btree_compare_double(cont void *key1, int keylen1,
7761da177e4SLinus Torvalds 		     const void *key2, int keylen2)
7771da177e4SLinus Torvalds {
7781da177e4SLinus Torvalds 	double result = *(double *) key1 - *(double *) key2;
7791da177e4SLinus Torvalds 	if (result == 0.0)
7801da177e4SLinus Torvalds 		return 0;
7811da177e4SLinus Torvalds 
7821da177e4SLinus Torvalds 	return (result < 0.0) ? -1 : 1;
7831da177e4SLinus Torvalds }
7841da177e4SLinus Torvalds #endif				//0
785