/*
 * This file and its contents are supplied under the terms of the
 * Common Development and Distribution License ("CDDL"), version 1.0.
 * You may only use this file in accordance with the terms of version
 * 1.0 of the CDDL.
 *
 * A full copy of the text of the CDDL should have accompanied this
 * source.  A copy of the CDDL is also available via the Internet at
 * http://www.illumos.org/license/CDDL.
 */

/*
 * Copyright 2018 Nexenta Systems, Inc. All rights reserved.
 */

/*
 * walker for libsmb : smb_ht.c (hash tables)
 */

#include <mdb/mdb_modapi.h>
#include <mdb/mdb_ks.h>
#include <mdb/mdb_ctf.h>

#include <smbsrv/hash_table.h>

/* smb_ht_walk info */
struct hw_info {
	HT_HANDLE hw_handle;	/* struct ht_handle being walked */
	HT_TABLE_ENTRY hw_tblent;
	HT_ITEM hw_item;
	int hw_idx;
};

/*
 * Walker for libsmb/smb_ht.c code.  Calls the call-back function with
 * each HT_ITEM object.  Top-level is HT_HANDLE, passed to _walk_init.
 */
int
smb_ht_walk_init(mdb_walk_state_t *wsp)
{
	struct hw_info *hw;
	uintptr_t addr = wsp->walk_addr;
	HT_HANDLE *ht;

	if (addr == 0) {
		mdb_printf("require address of an HT_HANDLE\n");
		return (WALK_ERR);
	}

	/*
	 * allocate the AVL walk data
	 */
	wsp->walk_data = hw = mdb_zalloc(sizeof (*hw), UM_GC|UM_SLEEP);

	/*
	 * get an mdb copy of the HT_HANDLE being walked
	 */
	ht = &hw->hw_handle;
	if (mdb_vread(ht, sizeof (*ht), wsp->walk_addr) == -1) {
		mdb_warn("failed to read %s at %#lx",
		    "HT_HANDLE", wsp->walk_addr);
		return (WALK_ERR);
	}

	hw->hw_idx = -1;
	wsp->walk_addr = 0;
	wsp->walk_data = hw;

	return (WALK_NEXT);
}

int
smb_ht_walk_step(mdb_walk_state_t *wsp)
{
	struct hw_info *hw = wsp->walk_data;
	HT_TABLE_ENTRY *he = &hw->hw_tblent;
	HT_ITEM *hi = &hw->hw_item;
	uintptr_t he_addr;
	int rv;

	while (wsp->walk_addr == 0) {
		if (++hw->hw_idx >= hw->hw_handle.ht_table_size)
			return (WALK_DONE);
		he_addr = (uintptr_t)hw->hw_handle.ht_table +
		    (hw->hw_idx * sizeof (HT_TABLE_ENTRY));
		if (mdb_vread(he, sizeof (*he), he_addr) == -1) {
			mdb_warn("failed to read %s at %p",
			    "HT_TABLE_ENTRY", wsp->walk_addr);
			return (WALK_ERR);
		}
		wsp->walk_addr = (uintptr_t)he->he_head;
	}

	if (mdb_vread(hi, sizeof (*hi), wsp->walk_addr) == -1) {
		mdb_warn("failed to read %s at %p",
		    "HT_ITEM", wsp->walk_addr);
		return (WALK_ERR);
	}

	rv = wsp->walk_callback(wsp->walk_addr, hi,
	    wsp->walk_cbdata);

	wsp->walk_addr = (uintptr_t)hi->hi_next;

	return (rv);
}